5b.py

#Advent of code 2021
# 12/11/21 day 5b
# Joe McFarland
# import sys
# import re
# import copy

#max_x = 10
#max_y = 10
max_x = 1000
max_y = 1000
grid = [[0]*max_x for _ in range(max_y)] 
filename = "data5.txt"

file = open(filename)
filestr = file.read()
a_list = filestr.split("\n")
maxrows = len(a_list)
print(a_list)
#maxcols = len(a_list[0])

# [3, 4] : [1, 4]   # xy1[0] xy1[1] : xy2[0] xy2[1]  : [x1, y1] [x2, y2]
def incrRange(xy1, xy2):
    x1,y1,x2,y2 = xy1[0],xy1[1],xy2[0],xy2[1]
    if x1==x2:
        for y in range(min(y1,y2), max(y1,y2)+1):
            grid[x1][y] += 1
    elif y1==y2:
        for x in range(min(x1,x2), max(x1,x2)+1):
            grid[x][y1] += 1
    else:
        print(f"DIAGNOL, {line}")
        #print(f"  {min(x1,x2)}, {max(x1,x2)+1}, {min(y1,y2)}, {max(y1,y2)+1}")
        x, y = x1, y1
        if x1 > x2:
            xstep = -1
            len = x1 - x2
        else:         # 5,5 -> 8,2  x1,y1 -> x2,y2
            xstep = 1
            len = x2 - x1
        if y1 > y2:   # 5,5 -> 8,2  x1,y1 -> x2,y2
            ystep = -1
        else:
            ystep = 1
        for i in range(len+1):
            grid[x][y] += 1
            #print(f"{x},{y}")
            x += xstep
            y += ystep

for line in a_list:
    xy = line.split(" -> ")
    xy1str = xy[0].split(",")
    xy1 = [int(elem) for elem in xy1str]
    xy2str = xy[1].split(",")
    xy2 = [int(elem) for elem in xy2str]
    print(f"{xy1} : {xy2}")
    incrRange(xy1, xy2)

# dump the grid
# for y in range(0, max_y):
#     for x in range(0, max_x):
#         print(f"{grid[x][y]}",end="")
#     print("")

count = 0
for y in range(0, max_y):
    for x in range(0, max_x):
        if grid[x][y] >= 2:
            count += 1
print(f"count = {count}")