43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
|
from direct.directnotify import DirectNotifyGlobal
|
||
|
|
import ActiveCellAI, CrateGlobals
|
||
|
|
from direct.task import Task
|
||
|
|
|
||
|
|
class DirectionalCellAI(ActiveCellAI.ActiveCellAI):
|
||
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('DirectionalCellAI')
|
||
|
|
|
||
|
|
def __init__(self, level, entId):
|
||
|
|
self.dir = [
|
||
|
|
0, 0]
|
||
|
|
ActiveCellAI.ActiveCellAI.__init__(self, level, entId)
|
||
|
|
self.moveTrack = None
|
||
|
|
return
|
||
|
|
|
||
|
|
def delete(self):
|
||
|
|
if self.moveTrack:
|
||
|
|
self.moveTrack.pause()
|
||
|
|
del self.moveTrack
|
||
|
|
self.moveTrack = None
|
||
|
|
taskMgr.remove(self.taskName('moveTask'))
|
||
|
|
return
|
||
|
|
|
||
|
|
def setState(self, state, objId=None):
|
||
|
|
ActiveCellAI.ActiveCellAI.setState(self, state, objId)
|
||
|
|
self.startMoveTask()
|
||
|
|
|
||
|
|
def taskName(self, name):
|
||
|
|
return self.level.taskName(name) + '-' + str(self.entId)
|
||
|
|
|
||
|
|
def startMoveTask(self):
|
||
|
|
taskMgr.remove(self.taskName('moveTask'))
|
||
|
|
taskMgr.doMethodLater(CrateGlobals.T_PUSH + CrateGlobals.T_PAUSE, self.moveTask, self.taskName('moveTask'))
|
||
|
|
|
||
|
|
def moveTask(self, task):
|
||
|
|
oldPos = self.grid.getObjPos(self.occupantId)
|
||
|
|
if self.grid.doMove(self.occupantId, self.dir[0], self.dir[1]):
|
||
|
|
newPos = self.grid.getObjPos(self.occupantId)
|
||
|
|
crate = simbase.air.doId2do.get(self.occupantId)
|
||
|
|
if crate:
|
||
|
|
crate.sendUpdate('setMoveTo', [
|
||
|
|
oldPos[0], oldPos[1], oldPos[2], newPos[0], newPos[1], newPos[2]])
|
||
|
|
return Task.done
|