8.4

'''
Napraviti demo sliku dimenzije 256x256 sa dva kvadrata u sredini (jedan u drugom)
koji su sa nijansama 127 i 191. Postaviti pozadinu na 63. Prikazati njen histogram.

Na ovu sliku dodati impulsni:

so
biber
so i biber
šum gustine 20%. Prikazati zašumljene slike i njihove histograme.
'''

import plotly.graph_objects as go
from plotly.subplots import make_subplots
import numpy as np
import plotly.express as px

img = np.full((256,256), fill_value=63)
img[32:32+192,32:32+192] = 127
img[64:64+128,64:64+128] = 191
fig = px.imshow(img, color_continuous_scale='gray')
fig.show()

# SO

rng = np.random.default_rng()
p = rng.uniform(size=img.shape)

p0 = 0.2

img_so = img.copy()
img_so[p<p0] = 255
fig = px.imshow(img_so, color_continuous_scale='gray')
fig.show()

# BIBER

img_biber = img.copy()
img_biber[(rng.uniform(size=img.shape))<p0] = 0
fig = px.imshow(img_biber, color_continuous_scale='gray')
fig.show()

# SO I BIBER

img_sb = img.copy()
p = rng.uniform(size=img.shape)
img_sb[p<p0/2] = 0
img_sb[(p0/2<=p) & (p<p0)] = 255
fig = px.imshow(img_sb, color_continuous_scale='gray')
fig.show()