import numpy as np
import matplotlib.pyplot as plt
from scipy import misc as mc
import math
import copy

def rgb2gray(img):
        r=img[:,:,0]
        g=img[:,:,1]
        b=img[:,:,2]
        s1=0.2126*r+0.7152*g+0.0722*b
        return s1


    #
def brightness(img):
        M,N=img.shape
        J=(1/(M*N))*(np.sum(np.sum(img)))
        return J


    #
def kontrast(img,J):
        M,N=img.shape
        pierwsze=np.sum(np.sum(np.power((img-J),2)))
        drugie=(1/(M*N))*pierwsze
        K=np.power(drugie,(1/2))
        return K


    #

imgp = mc.imread('kierowca.png')

#c=img[:,:,:]
s1=rgb2gray(imgp)

#imgp = imgp.astype(float)



M,N,O=imgp.shape
print (M)
print (N)

img=copy.copy(imgp)

J=brightness(img[:,:,0])
print('Jasnosc: %s' % (J))

K=kontrast(img[:,:,0],J)
print('Kontrast: %s' % (K))

print (img.dtype)
print (img.shape)

#plt.imshow(s1)
#plt.show()

img=img[:,:,0]
#const=50
#for x in range(0, M):
        #for y in range(0, N):
            #if((img[x, y] + const) > 255):
                #img[x, y] = 225
            #elif((img[x, y] + const) < 0):
                #img[x, y] = 0
            #else:
                #img[x, y] = img[x, y] + const

stala=50

for i in range(0, M):
    for j in range(0, N):
        img[i,j]=img[i,j]+stala
        if ((img[i,j])>255):
            img[i,j]=255
        if ((img[i,j])<0):
            img[i,j]=0

J=brightness(img)
K=kontrast(img, J)
print('Jasnosc: %s' % (J))
print('Kontrast: %s' % (K))

img1=copy.copy(imgp)
img1=img1[:,:,0]
stala=50
for i in range(0, M):
    for j in range(0, N):
        img1[i,j]=img1[i,j]*stala
        if ((img1[i,j])>255):
            img1[i,j]=255
        if ((img1[i,j])<0):
            img1[i,j]=0

J=brightness(img1)
K=kontrast(img1, J)
print('Jasnosc: %s' % (J))
print('Kontrast: %s' % (K))