"""
Program rysujący histogramy barw obrazka
"""
#!/usr/bin/python3
import numpy as np
import matplotlib.pyplot as plt
import sys

im = plt.imread(sys.argv[1])

#Ustawienie skali w zależności czy png czy jpg
if im.dtype == np.uint8:
    scale=255
else:
    scale=1

reds   = im[:,:,0].flatten()
greens = im[:,:,1].flatten()
blues  = im[:,:,2].flatten()
values = im.max(axis=2).flatten()

fig,ax = plt.subplots(4,1)
ax[0].hist(reds,bins=255,range=(0,scale),color = 'red')
ax[1].hist(greens,255,range=(0,scale),color = 'green')
ax[2].hist(blues,255,range=(0,scale),color = 'blue')
ax[3].hist(values,255,range=(0,scale),color = 'black')
plt.show()