42 lines
1.0 KiB
Python
42 lines
1.0 KiB
Python
from Point import Point
|
|
|
|
class Trac:
|
|
def __init__(self, rotation, position):
|
|
self.__position = Point(position.get_cord())
|
|
self.__rotation = rotation
|
|
|
|
def get_symbol(self):
|
|
if self.__rotation == 'S':
|
|
return 'v'
|
|
elif self.__rotation == 'N':
|
|
return '^'
|
|
elif self.__rotation == 'E':
|
|
return '>'
|
|
else:
|
|
return '<'
|
|
|
|
def set_rotation(self, rotation):
|
|
self.__rotation = rotation
|
|
|
|
def set_position(self, position):
|
|
self.__position.set_cord(position)
|
|
|
|
def get_position(self):
|
|
return self.__position
|
|
|
|
def get_rotation(self):
|
|
return self.__rotation
|
|
|
|
def move(self):
|
|
x = self.get_position().get_x()
|
|
y = self.get_position().get_y()
|
|
if self.__rotation == 'N':
|
|
self.set_position((x - 1 ,y))
|
|
elif rotation == 'S':
|
|
self.set_position((x + 1 ,y))
|
|
elif rotation == 'W':
|
|
self.set_position((x, y - 1))
|
|
else:
|
|
self.set_position((x, y + 1))
|
|
|