126 lines
4.3 KiB
Python
126 lines
4.3 KiB
Python
from panda3d.core import *
|
|
from direct.distributed import DistributedObject
|
|
from . import SuitPlannerBase
|
|
from toontown.toonbase import ToontownGlobals
|
|
|
|
class DistributedSuitPlanner(DistributedObject.DistributedObject, SuitPlannerBase.SuitPlannerBase):
|
|
|
|
def __init__(self, cr):
|
|
DistributedObject.DistributedObject.__init__(self, cr)
|
|
SuitPlannerBase.SuitPlannerBase.__init__(self)
|
|
self.suitList = []
|
|
self.buildingList = [0,
|
|
0,
|
|
0,
|
|
0]
|
|
self.pathViz = None
|
|
return
|
|
|
|
def generate(self):
|
|
self.notify.info('DistributedSuitPlanner %d: generating' % self.getDoId())
|
|
DistributedObject.DistributedObject.generate(self)
|
|
base.cr.currSuitPlanner = self
|
|
|
|
def disable(self):
|
|
self.notify.info('DistributedSuitPlanner %d: disabling' % self.getDoId())
|
|
self.hidePaths()
|
|
DistributedObject.DistributedObject.disable(self)
|
|
base.cr.currSuitPlanner = None
|
|
return
|
|
|
|
def d_suitListQuery(self):
|
|
self.sendUpdate('suitListQuery')
|
|
|
|
def suitListResponse(self, suitList):
|
|
self.suitList = suitList
|
|
messenger.send('suitListResponse')
|
|
|
|
def d_buildingListQuery(self):
|
|
self.sendUpdate('buildingListQuery')
|
|
|
|
def buildingListResponse(self, buildingList):
|
|
self.buildingList = buildingList
|
|
messenger.send('buildingListResponse')
|
|
|
|
def hidePaths(self):
|
|
if self.pathViz:
|
|
self.pathViz.detachNode()
|
|
self.pathViz = None
|
|
return
|
|
|
|
def showPaths(self):
|
|
self.hidePaths()
|
|
vizNode = GeomNode(self.uniqueName('PathViz'))
|
|
lines = LineSegs()
|
|
self.pathViz = render.attachNewNode(vizNode)
|
|
points = self.frontdoorPointList + self.sidedoorPointList + self.cogHQDoorPointList + self.streetPointList
|
|
while len(points) > 0:
|
|
self.__doShowPoints(vizNode, lines, None, points)
|
|
|
|
cnode = CollisionNode('battleCells')
|
|
cnode.setCollideMask(BitMask32.allOff())
|
|
for zoneId, cellPos in list(self.battlePosDict.items()):
|
|
cnode.addSolid(CollisionSphere(cellPos, 9))
|
|
text = '%s' % zoneId
|
|
self.__makePathVizText(text, cellPos[0], cellPos[1], cellPos[2] + 9, (1, 1, 1, 1))
|
|
|
|
self.pathViz.attachNewNode(cnode).show()
|
|
return
|
|
|
|
def __doShowPoints(self, vizNode, lines, p, points):
|
|
if p == None:
|
|
pi = len(points) - 1
|
|
if pi < 0:
|
|
return
|
|
p = points[pi]
|
|
del points[pi]
|
|
else:
|
|
if p not in points:
|
|
return
|
|
pi = points.index(p)
|
|
del points[pi]
|
|
text = '%s' % p.getIndex()
|
|
pos = p.getPos()
|
|
if p.getPointType() == DNASuitPoint.FRONTDOORPOINT:
|
|
color = (1, 0, 0, 1)
|
|
elif p.getPointType() == DNASuitPoint.SIDEDOORPOINT:
|
|
color = (0, 0, 1, 1)
|
|
else:
|
|
color = (0, 1, 0, 1)
|
|
self.__makePathVizText(text, pos[0], pos[1], pos[2], color)
|
|
adjacent = self.dnaStore.getAdjacentPoints(p)
|
|
numPoints = adjacent.getNumPoints()
|
|
for i in range(numPoints):
|
|
qi = adjacent.getPointIndex(i)
|
|
q = self.dnaStore.getSuitPointWithIndex(qi)
|
|
pp = p.getPos()
|
|
qp = q.getPos()
|
|
v = Vec3(qp - pp)
|
|
v.normalize()
|
|
c = v.cross(Vec3.up())
|
|
p1a = pp + v * 2 + c * 0.5
|
|
p1b = pp + v * 3
|
|
p1c = pp + v * 2 - c * 0.5
|
|
lines.reset()
|
|
lines.moveTo(pp)
|
|
lines.drawTo(qp)
|
|
lines.moveTo(p1a)
|
|
lines.drawTo(p1b)
|
|
lines.drawTo(p1c)
|
|
lines.create(vizNode, 0)
|
|
self.__doShowPoints(vizNode, lines, q, points)
|
|
|
|
return
|
|
|
|
def __makePathVizText(self, text, x, y, z, color):
|
|
if not hasattr(self, 'debugTextNode'):
|
|
self.debugTextNode = TextNode('debugTextNode')
|
|
self.debugTextNode.setAlign(TextNode.ACenter)
|
|
self.debugTextNode.setFont(ToontownGlobals.getSignFont())
|
|
self.debugTextNode.setTextColor(*color)
|
|
self.debugTextNode.setText(text)
|
|
np = self.pathViz.attachNewNode(self.debugTextNode.generate())
|
|
np.setPos(x, y, z + 1)
|
|
np.setScale(1.0)
|
|
np.setBillboardPointEye(2)
|
|
np.node().setAttrib(TransparencyAttrib.make(TransparencyAttrib.MDual), 2)
|