47 lines
1.1 KiB
Python
47 lines
1.1 KiB
Python
|
# Alexander Bass
|
||
|
|
# Created 4/12/24
|
||
|
|
# Last Edited 4/13/24
|
||
|
|
def load_ppm(file):
|
||
|
|
header = next(file).strip().upper()
|
||
|
|
assert header == "P3"
|
||
|
|
dimensions = next(file).strip()
|
||
|
|
split_dimensions = dimensions.split()
|
||
|
|
assert len(split_dimensions) == 2
|
||
|
|
width = split_dimensions[0]
|
||
|
|
height = split_dimensions[-1]
|
||
|
|
width = int(width)
|
||
|
|
height = int(height)
|
||
|
|
|
||
|
|
# not really doing anything with this
|
||
|
|
color_depth = next(file).strip()
|
||
|
|
assert color_depth == "255"
|
||
|
|
|
||
|
|
image = []
|
||
|
|
for y, line in enumerate(file):
|
||
|
|
assert y < height
|
||
|
|
t = [int(value) for value in line.strip().split()]
|
||
|
|
|
||
|
|
pixels = []
|
||
|
|
|
||
|
|
assert len(t) % 3 == 0
|
||
|
|
|
||
|
|
assert len(t) // 3 <= width
|
||
|
|
|
||
|
|
highest = 0
|
||
|
|
for x in range(0, len(t), 3):
|
||
|
|
red = t[0 + x]
|
||
|
|
green = t[1 + x]
|
||
|
|
blue = t[2 + x]
|
||
|
|
highest = x // 3
|
||
|
|
pixels.append((red, green, blue))
|
||
|
|
assert highest == width - 1
|
||
|
|
image.append(pixels)
|
||
|
|
|
||
|
|
assert len(image) == height
|
||
|
|
file.close()
|
||
|
|
return image
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
print(load_ppm(open("dirt.ppm")))
|