Poodletooth-iLand/toontown/dna/DNASuitPoint.py
2015-03-03 17:10:12 -05:00

89 lines
3.1 KiB
Python

class DNASuitPoint:
COMPONENT_CODE = 20
STREET_POINT = 0
FRONT_DOOR_POINT = 1
SIDE_DOOR_POINT = 2
COGHQ_IN_POINT = 3
COGHQ_OUT_POINT = 4
def __init__(self, index, pointType, pos, landmarkBuildingIndex=-1):
self.index = index
self.pointType = pointType
self.pos = pos
self.graphId = 0
self.landmarkBuildingIndex = landmarkBuildingIndex
def __str__(self):
pointType = self.getPointType()
if pointType == DNASuitPoint.STREET_POINT:
pointTypeStr = 'STREET_POINT'
elif pointType == DNASuitPoint.FRONT_DOOR_POINT:
pointTypeStr = 'FRONT_DOOR_POINT'
elif pointType == DNASuitPoint.SIDE_DOOR_POINT:
pointTypeStr = 'SIDE_DOOR_POINT'
elif pointType == DNASuitPoint.COGHQ_IN_POINT:
pointTypeStr = 'COGHQ_IN_POINT'
elif pointType == DNASuitPoint.COGHQ_OUT_POINT:
pointTypeStr = 'COGHQ_OUT_POINT'
else:
pointTypeStr = '**invalid**'
return 'DNASuitPoint index: %d, pointType: %s, pos: %s' % (
self.getIndex(), pointTypeStr, self.getPos())
def setIndex(self, index):
self.index = index
def getIndex(self):
return self.index
def getGraphId(self):
return self.graphId
def getLandmarkBuildingIndex(self):
return self.landmarkBuildingIndex
def getPos(self):
return self.pos
def isTerminal(self):
return self.pointType <= 2
def setGraphId(self, id):
self.graphId = id
def setLandmarkBuildingIndex(self, index):
self.landmarkBuildingIndex = index
def setPointType(self, pointType):
if isinstance(pointType, int):
if type == DNASuitPoint.STREET_POINT:
self.pointType = DNASuitPoint.STREET_POINT
elif type == DNASuitPoint.FRONT_DOOR_POINT:
self.pointType = DNASuitPoint.FRONT_DOOR_POINT
elif pointType == DNASuitPoint.SIDE_DOOR_POINT:
self.pointType = DNASuitPoint.SIDE_DOOR_POINT
elif pointType == DNASuitPoint.COGHQ_IN_POINT:
self.pointType = DNASuitPoint.COGHQ_IN_POINT
elif pointType == DNASuitPoint.COGHQ_OUT_POINT:
self.pointType = DNASuitPoint.COGHQ_OUT_POINT
else:
raise TypeError('%i is not a valid DNASuitPointType' % pointType)
elif isinstance(pointType, str):
if type == 'STREET_POINT':
self.pointType = DNASuitPoint.STREET_POINT
elif type == 'FRONT_DOOR_POINT':
self.pointType = DNASuitPoint.FRONT_DOOR_POINT
elif pointType == 'SIDE_DOOR_POINT':
self.pointType = DNASuitPoint.SIDE_DOOR_POINT
elif pointType == 'COGHQ_IN_POINT':
self.pointType = DNASuitPoint.COGHQ_IN_POINT
elif pointType == 'COGHQ_OUT_POINT':
self.pointType = DNASuitPoint.COGHQ_OUT_POINT
else:
raise TypeError('%s is not a valid DNASuitPointType' % pointType)
def getPointType(self):
return self.pointType
def setPos(self, pos):
self.pos = pos