ImPy/ImAdd.py at master · mm10ws/ImPy · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
__author__ = 'Mayur M'
import ImgIO


def add(image1, image2):  # add two images together
    if image1.width == image2.width and image1.height == image2.height:
        return_red = []
        return_green = []
        return_blue = []
        for i in range(0, len(image1.red)):
            tmp_r = image1.red[i] + image2.red[i]  # adding the RGB values
            tmp_g = image1.green[i] + image2.green[i]
            tmp_b = image1.blue[i] + image2.blue[i]
            if 0 <= tmp_r <= 255:
                return_red.append(tmp_r)
            else:
                return_red.append(tmp_r % 255)  # loop values around if saturation
            if 0 <= tmp_g <= 255:
                return_green.append(tmp_g)
            else:
                return_green.append(tmp_g % 255)  # loop values around if saturation
            if 0 <= tmp_b <= 255:
                return_blue.append(tmp_b)
            else:
                return_blue.append(tmp_b % 255)  # loop values around if saturation
        return return_red, return_green, return_blue
    else:
        print "Error: image dimensions do not match!"


def main():  # test case
    print('start!!!!!')
    ima = ImgIO.ImgIO()
    imb = ImgIO.ImgIO()
    ima.read_image("y.jpg")
    imb.read_image("test1.png")
    add_r, add_g, add_b = add(ima, imb)
    imc = ImgIO.ImgIO()
    imc.read_list(add_r, add_g, add_b, "final1.png", ima.width, ima.height)
    imc.write_image("final1.png")


if __name__ == '__main__':
    main()