aoc22/day22-1.py

import re

#create a class to store the map
class plan:
    def __init__(self):
        self.grid = []
        self.goTo = [(0, 1), (1, 0), (0, -1), (-1, 0)]
    
    def addLine(self, inputLine):
        inputLine = inputLine.replace(' ', '+')
        self.grid.append(list(inputLine))

    def testPrint(self):
        for line in self.grid:
            for point in line:
                print(point, end = ' ')
            print('\n')

    def setStart(self):
        self.direction = 0
        self.column = self.grid[0].index('.')
        self.row = 0
        maxL = 0
        for line in self.grid:
            if len(line) > maxL:
                maxL = len(line)
        for i in range(len(self.grid)):
            while len(self.grid[i]) < maxL:
                self.grid[i].append('+')

    def getInstruction(self, instruction):
        if instruction.isnumeric():
            self.move(int(instruction))
        if instruction.isalpha():
            self.turn(instruction)

    def turn(self, direction):
        if direction == 'L':
            self.direction -= 1
            if self.direction < 0:
                self.direction = 3
        elif direction == 'R':
            self.direction = (self.direction + 1) % 4

    def move(self, distance):
        for _ in range(distance):
            nextRow = self.row + self.goTo[self.direction][0]
            nextCol = self.column + self.goTo[self.direction][1]
            if nextRow < 0 or nextCol < 0 or nextRow >= len(self.grid) or nextCol >= len(self.grid[nextRow]) or self.grid[nextRow][nextCol] == '+':
                nextRow, nextCol = self.wrapAround(nextRow, nextCol)
            if self.grid[nextRow][nextCol] == '#':
                return
            self.column = nextCol
            self.row = nextRow

    def wrapAround(self, nextRow, nextCol):
        if self.direction == 0:
            for i in range(len(self.grid[nextRow])):
                if self.grid[nextRow][i] in ['.', '#']:
                    nextCol = i
                    break
        elif self.direction == 1:
            for i in range(len(self.grid)):
                if self.grid[i][nextCol] in ['.', '#']:
                    nextRow = i
                    break
        elif self.direction == 2:
            for i in range(len(self.grid[nextRow])-1, 0, -1):
                if self.grid[nextRow][i] in ['.', '#']:
                    nextCol = i
                    break
        elif self.direction == 3:
            for i in range(len(self.grid)-1, 0, -1):
                if self.grid[i][nextCol] in ['.', '#']:
                    nextRow = i
                    break
        return nextRow, nextCol

    def getPassword(self):
        return (self.column + 1)*4 + (self.row + 1)*1000 + self.direction



#create an object of the map 
path = plan()

# reading  input and inserting it into the map
with open('input22.txt','r') as f:
    inp = f.read().splitlines(keepends=False)

for line in inp:
    if line == '':
        break
    path.addLine(line)

instructionSet = inp[-1]
instructionSet = re.split('(\d+)', instructionSet)

path.setStart()

#go through the instructions
for instruction in instructionSet:
    path.getInstruction(instruction)

#path.testPrint()
print(path.column + 1, path.row + 1, path.direction, sep = '---')
print(path.getPassword())