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

# -------- Test Functions --------
def MakeGrayImage(im1):
    n = im1.getWidth()
    m = im1.getHeight()
    im2 = EmptyImage(n,m)
    for i in range(n):
        for j in range(m):
            p = im1.getPixel(i,j)
            r = p.getRed()
            g = p.getGreen()
            b = p.getBlue()
            avg = (r+g+b)//3
            newp = Pixel(avg,avg,avg)
            im2.setPixel(i,j,newp)
    return(im2)

def MakeNegativeImage(im1):
    n = im1.getWidth()
    m = im1.getHeight()
    im2 = EmptyImage(n,m)
    for i in range(n):
        for j in range(m):
            p = im1.getPixel(i,j)
            r = p.getRed()
            g = p.getGreen()
            b = p.getBlue()
            newp = Pixel(255-r,255-g,255-b)
            im2.setPixel(i,j,newp)
    return(im2)

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

def MakeDoubleImage(im1):
    n = im1.getWidth()
    m = im1.getHeight()
    im2 = EmptyImage(2*n,2*m)
    for i in range(n):
        for j in range(m):
            p = im1.getPixel(i,j)
            im2.setPixel(2*i,2*j,p)
            im2.setPixel(2*i+1,2*j,p)
            im2.setPixel(2*i,2*j+1,p)
            im2.setPixel(2*i+1,2*j+1,p)
    return(im2)

# -------- Main Program --------

im1 = FileImage("leo1.gif")
n = im1.getWidth()
m = im1.getHeight()
mywin1 = ImageWin("file1",n,m)
im1.draw(mywin1)

im2 = MakeGrayImage(im1)
#im2 = MakeNegativeImage(im1)
#im2 = MakeHalfImage(im1)
#im2 = MakeDoubleImage(im1)

mywin2 = ImageWin("file2",n,m)
im2.draw(mywin2)