traktor/Trac.py
marcinljablonski bf2ecb7de5 .
2019-04-29 06:52:57 +02:00

30 lines
688 B
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