import os
os.chdir("/Users/koc/Desktop/abc")
from cImage import *

#------------
def grayScale(imf):
    n = imf.getWidth()
    m = imf.getHeight()
    omf = EmptyImage(n,m)
    for i in range(n):
        for j in range(m):
            p = imf.getPixel(i,j)
            r = p.getRed()
            g = p.getGreen()
            b = p.getBlue()
            a = (r+g+b)//3
            q = Pixel(a,a,a)
            omf.setPixel(i,j,q)
    return(omf)

def mystery(imf):
    n = imf.getWidth()
    m = imf.getHeight()
    omf = EmptyImage(n,m)
    for i in range(n):
        for j in range(m):
            p = imf.getPixel(i,j)
            r = p.getRed()
            g = p.getGreen()
            b = p.getBlue()
            q = Pixel(255-r,255-g,255-b)
            omf.setPixel(i,j,q)
    return(omf)

def grayScale2(imf):
    n = imf.getWidth()
    m = imf.getHeight()
    for i in range(n):
        for j in range(m):
            p = imf.getPixel(i,j)
            r = p.getRed()
            g = p.getGreen()
            b = p.getBlue()
            a = (r+g+b)//3
            q = Pixel(a,a,a)
            imf.setPixel(i,j,q)
    return

def halving(imf):
    n = imf.getWidth()
    m = imf.getHeight()
    omf = EmptyImage(n//2,m//2)
    for i in range(0,n,2):
        for j in range(0,n,2):
            p = imf.getPixel(i,j)
            omf.setPixel(i//2, j//2, p)
    return(omf)


#------------
mf = FileImage("leo.gif")
mw1 = ImageWin("mywindow1", 500, 500)
mf.draw(mw1)

new = grayScale(mf)
mw2 = ImageWin("mywindow2", 500, 500)
new.draw(mw2)

mw3 = ImageWin("mywindow2", 500, 500)
new3 = mystery(mf)
new3.draw(mw3)