import numpy as np
from brainplotlib import brain_plot

rng = np.random.default_rng(0)
v = rng.random((1175, ))

img = brain_plot(v, vmax=1, vmin=0, cmap='viridis')

print(img.dtype, img.shape)
print(img.max(), img.min())

:tags: ["remove-cell"]
import matplotlib.pyplot as plt
fig = plt.figure(figsize=(img.shape[1] / 200, img.shape[0] / 200), dpi=200)
ax = fig.add_axes([0, 0, 1, 1])
ax.imshow(img)
ax.axis('off')
plt.savefig('save_image_matplotlib.png')
plt.close()

:tags: ["remove-cell"]
from PIL import Image
im = Image.fromarray(
    np.round(img * 255).astype(np.uint8))
im.save('save_image_pillow.png')

:tags: ["remove-cell"]
from myst_nb import glue
glue('save_image', im, display=False)

:tags: ["remove-cell"]
import cv2
## The default channel order of OpenCV is BGR rather than RGB.
reorder = {3: [2, 1, 0], 4: [2, 1, 0, 3]}[img.shape[2]]
cv2.imwrite(
    'save_image_opencv.png',
    np.round(img[:, :, reorder] * 255).astype(np.uint8))

````{tab-item} Using matplotlib
```python
import matplotlib.pyplot as plt
fig = plt.figure(
    figsize=(img.shape[1] / 200, img.shape[0] / 200), dpi=200)
ax = fig.add_axes([0, 0, 1, 1])
ax.imshow(img)
ax.axis('off')
plt.savefig('save_image_matplotlib.png')
plt.close()
```
````
````{tab-item} Using Pillow
```python
from PIL import Image
im = Image.fromarray(
    np.round(img * 255).astype(np.uint8))
im.save('save_image_pillow.png')
```
```{note}
The code block above requires that the [Pillow package](https://pillow.readthedocs.io/en/stable/index.html) has been installed.
```
````
````{tab-item} Using OpenCV
```python
import cv2
## The default channel order of OpenCV is BGR rather than RGB.
reorder = {3: [2, 1, 0], 4: [2, 1, 0, 3]}[img.shape[2]]
cv2.imwrite(
    'save_image_opencv.png',
    np.round(img[:, :, reorder] * 255).astype(np.uint8))
```
```{note}
The code block above requires that [OpenCV](https://opencv.org/) and its Python bindings to be installed.
```
````

```{grid-item-card}
:class-header: bg-light text-center
Using matplotlib
^^^
![](save_image_matplotlib.png)
```
```{grid-item-card}
:class-header: bg-light text-center
Using Pillow
^^^
![](save_image_pillow.png)
```
```{grid-item-card}
:class-header: bg-light text-center
Using OpenCV
^^^
![](save_image_opencv.png)
```