day4.py

from HelperFunctions import readInputFile
from HelperFunctions import readExampleInput
from HelperFunctions import convertToInt

directions = {'UP':(0,-1), 'DOWN':(0,1), 'LEFT':(-1,0), 'RIGHT':(1,0), 'UPLEFT':(-1,-1), 'UPRIGHT':(1,-1), 'DOWNLEFT':(-1,1), 'DOWNRIGHT':(1,1)}
word = ['X','M','A','S']

def prepareBoard(board):

	newBoard = []
	newBoard.append(['.' for x in range(len(board[0])+2)])

	for line in board:
		newLine = ['.'] + [x for x in line] + ['.']
		newBoard.append(newLine)

	newBoard.append(['.' for x in range(len(board[0])+2)])

	return newBoard

def findWordInDirections(posX,posY,board):
	words = 0

	for direction in directions.values():
		currentPos = (posX,posY)
		for wordPos in range(1,5):
			if wordPos == 4:
				words = words + 1
				break
		
			currentPos = (currentPos[0] + direction[0], currentPos[1] + direction[1])
			if not(board[currentPos[1]][currentPos[0]] == word[wordPos]):
				break

	return words

def isXMAS(posX,posY,board):

	if (board[posY-1][posX-1] == 'M' and board[posY+1][posX+1] == 'S') or (board[posY-1][posX-1] == 'S' and board[posY+1][posX+1] == 'M'):
		if (board[posY+1][posX-1] == 'M' and board[posY-1][posX+1] == 'S') or (board[posY+1][posX-1] == 'S' and board[posY-1][posX+1] == 'M'):
			return 1
	
	return 0

def do1(board):
	
	words = 0

	for posY,line in enumerate(board):
		for posX,value in enumerate(line):
			if value == 'X':
				words = words + findWordInDirections(posX,posY,board)

	return words

def do2(board):
	
	occurrences = 0

	for posY,line in enumerate(board):
		for posX,value in enumerate(line):
			if value == 'A':
				occurrences = occurrences + isXMAS(posX,posY,board)	
	
	return occurrences

def do():
	strInput = readInputFile(4)

	board = prepareBoard(strInput.split('\n'))

	print(do1(board))
	print(do2(board))

	print('done')


do()