diff --git a/kyu_3/line_safari_is_that_a_line/walker_class.py b/kyu_3/line_safari_is_that_a_line/walker_class.py index db7d2962982..14707a95d1b 100644 --- a/kyu_3/line_safari_is_that_a_line/walker_class.py +++ b/kyu_3/line_safari_is_that_a_line/walker_class.py @@ -11,7 +11,6 @@ class Walker: """ def __init__(self, grid: list): - # print('__init__') self.__grid: list = grid self.__is_start: bool = True self.__position: dict = self.__get_start_point() @@ -158,6 +157,8 @@ def __set_direction(self) -> None: self.__reset_direction() print(f'prev: {previous_position}, pos: {self.position}') + case_i = self.__get_case_i(previous_position) + if self.position == '+' and previous_position in '-X': self.__direction['up'] = self.__test_up() self.__direction['down'] = self.__test_down() @@ -186,6 +187,46 @@ def __set_direction(self) -> None: elif self.__position['row'] > self.__position['prev_row']: self.__direction['down'] = self.__test_down() + def __get_case_i(self, previous_position) -> int: + """ + Get case i base on the current position vs previous_position + :param previous_position: + :return: + """ + i: int = 0 + + if self.position == '+' and previous_position in '-X': + i = 1 + + if self.position == '+' and previous_position == '|': + i = 2 + + if self.position == previous_position == '+' \ + and self.__position['col'] == self.__position['prev_col']: + i = 3 + + if self.position == previous_position == '+' \ + and self.__position['row'] == self.__position['prev_row']: + i = 4 + + if self.position == '-' and previous_position in '-X+' \ + and self.__position['col'] < self.__position['prev_col']: + i = 5 + + if self.position == '-' and previous_position in '-X+' \ + and self.__position['col'] > self.__position['prev_col']: + i = 6 + + if self.position == '|' and previous_position in '|X+' \ + and self.__position['row'] < self.__position['prev_row']: + i = 7 + + if self.position == '|' and previous_position in '|X+' \ + and self.__position['row'] > self.__position['prev_row']: + i = 8 + + return i + def __test_up(self) -> bool: row: int = self.__position['row'] col: int = self.__position['col']