image-processing-tool/imageIO.py at main · MelekElloumi/image-processing-tool · 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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
import settings as s
import utils
import cv2

def read(filepath):
    type = readType(filepath)
    if (type == None):
        raise Exception
    if (type == "P2\n"):
        imageread, s.width, s.height, s.graylevel, s.colour = readPGMascii(filepath)
    elif (type == "P5\n"):
        imageread, s.width, s.height, s.graylevel, s.colour = readPGMbinary(filepath)
    else:
        imageread, s.width, s.height, s.graylevel, s.colour = readPPM(filepath)
    s.isread = True
    s.image_orig = utils.arrayToMatrix(imageread, s.height, s.width,s.colour)


def readType(filepath):
    try:
        file = open(filepath, 'r')
        type = file.readline()
    except UnicodeDecodeError:
        file = open(filepath, 'rb')
        type = file.readline().decode()
    if not ("P2" in type or "P5" in type or "P3" in type or "P6" in type):
        file.close()
        return None
    file.close()
    return type


def readPGMascii(filepath):
    file = open(filepath, 'r')
    file.readline()
    while True:
        line = file.readline()
        if line[0] != '#': break
    dimx, dimy = line.split()
    dimx, dimy = int(dimx), int(dimy)
    nivg = int(file.readline())
    imageread = []
    for line in file.readlines():
        for num in line.split():
            imageread.append(int(num))
    if len(imageread) != dimx * dimy:
        file.close()
        return None, -1, -1, -1
    file.close()
    return imageread, dimx, dimy, nivg, False

def readPGMbinary(filepath):
    file = open(filepath, 'rb')
    file.readline()
    while True:
        line = file.readline().decode()
        if line[0] != '#': break
    dimx, dimy = line.split()
    dimx, dimy = int(dimx), int(dimy)
    nivg = int(file.readline().decode())
    imageread = list(file.read(dimx * dimy))
    if len(imageread) != dimx * dimy:
        file.close()
        return None, -1, -1, -1
    file.close()
    return imageread, dimx, dimy, nivg, False

def readPPM(filepath):
    file = open(filepath, 'r')
    file.readline()
    while True:
        line = file.readline()
        if line[0] != '#': break
    dimx, dimy = line.split()
    dimx, dimy = int(dimx), int(dimy)
    nivg = int(file.readline())
    file.close()
    imageread = cv2.imread(filepath)
    return imageread, dimx, dimy, nivg, True


def write(filepath, image):
    data = utils.matrixToArray(image, s.height, s.width)
    writePGM(filepath, data)


def writePGM(filepath, image):
    # image is a tuple of (data,width,height,graylevel)
    file = open(filepath, "w")
    file.write("P2\n")
    file.write("#output image created by image-processing-tool, Melek Elloumi\n")
    file.write(str(s.width) + " " + str(s.height) + "\n")
    file.write(str(s.graylevel) + "\n")
    for num in range(0, len(image)):
        # file.write(str(num)+"\t")
        file.write(str(image[num]) + " ")
        if ((num + 1) % s.width == 0):
            file.write("\n")
    file.close()