mirror of
https://github.com/panda3d/panda3d.git
synced 2025-10-04 19:08:55 -04:00
118 lines
4.2 KiB
Python
Executable File
118 lines
4.2 KiB
Python
Executable File
|
|
from pandac.PandaModules import *
|
|
|
|
from direct.task import Task
|
|
import DistributedNodeAI
|
|
import CartesianGridBase
|
|
|
|
class DistributedCartesianGridAI(DistributedNodeAI.DistributedNodeAI,
|
|
CartesianGridBase.CartesianGridBase):
|
|
|
|
notify = directNotify.newCategory("DistributedCartesianGridAI")
|
|
|
|
RuleSeparator = ":"
|
|
|
|
def __init__(self, air, startingZone, gridSize, gridRadius):
|
|
DistributedNodeAI.DistributedNodeAI.__init__(self, air)
|
|
self.style = "Cartesian"
|
|
self.startingZone = startingZone
|
|
self.gridSize = gridSize
|
|
self.gridRadius = gridRadius
|
|
|
|
# Keep track of all AI objects added to the grid
|
|
self.gridObjects = {}
|
|
self.updateTaskStarted = 0
|
|
|
|
def delete(self):
|
|
DistributedNodeAI.DistributedNodeAI.delete(self)
|
|
self.stopUpdateGridTask()
|
|
|
|
def isGridParent(self):
|
|
# If this distributed object is a DistributedGrid return 1. 0 by default
|
|
return 1
|
|
|
|
def getParentingRules(self):
|
|
print "calling getter"
|
|
rule = ("%i%s%i%s%i" % (self.startingZone, self.RuleSeparator,
|
|
self.gridSize, self.RuleSeparator,
|
|
self.gridRadius))
|
|
return [self.style, rule]
|
|
|
|
# Reparent and setLocation on av to DistributedOceanGrid
|
|
def addObjectToGrid(self, av):
|
|
print "setting parent to grid %s" % self
|
|
avId = av.doId
|
|
|
|
# Create a grid parent
|
|
#gridParent = self.attachNewNode("gridParent-%s" % avId)
|
|
#self.gridParents[avId] = gridParent
|
|
self.gridObjects[avId] = av
|
|
|
|
# Put the avatar on the grid
|
|
self.handleAvatarZoneChange(av)
|
|
|
|
if not self.updateTaskStarted:
|
|
self.startUpdateGridTask()
|
|
|
|
def removeObjectFromGrid(self, av):
|
|
# TODO: WHAT LOCATION SHOULD WE SET THIS TO?
|
|
#av.wrtReparentTo(self.parentNP)
|
|
#av.setLocation(self.air.districtId, 1000)
|
|
|
|
# Remove grid parent for this av
|
|
avId = av.doId
|
|
if self.gridObjects.has_key(avId):
|
|
del self.gridObjects[avId]
|
|
|
|
# Stop task if there are no more av's being managed
|
|
if len(self.gridObjects) == 0:
|
|
self.stopUpdateGridTask()
|
|
|
|
#####################################################################
|
|
# updateGridTask
|
|
# This task is similar to the processVisibility task for the local client.
|
|
# A couple differences:
|
|
# - we are not doing setInterest on the AI (that is a local client specific call).
|
|
# - we assume that the moving objects on the grid are parented to a gridParent,
|
|
# and are broadcasting their position relative to that gridParent. This
|
|
# makes the task's math easy. Just check to see when our position goes
|
|
# out of the current grid cell. When it does, call handleAvatarZoneChange
|
|
|
|
def startUpdateGridTask(self):
|
|
self.stopUpdateGridTask()
|
|
self.updateTaskStarted = 1
|
|
taskMgr.add(self.updateGridTask, self.taskName("updateGridTask"))
|
|
|
|
def stopUpdateGridTask(self):
|
|
taskMgr.remove(self.taskName("updateGridTask"))
|
|
self.updateTaskStarted = 0
|
|
|
|
def updateGridTask(self, task):
|
|
# Run through all grid objects and update their parents if needed
|
|
for avId in self.gridObjects:
|
|
av = self.gridObjects[avId]
|
|
pos = av.getPos()
|
|
if ((pos[0] < 0 or pos[1] < 0) or
|
|
(pos[0] > self.cellWidth or pos[1] > self.cellWidth)):
|
|
# we are out of the bounds of this current cell
|
|
self.handleAvatarZoneChange(av)
|
|
# Do this every second, not every frame
|
|
task.delayTime = 1.0
|
|
return Task.again
|
|
|
|
def handleAvatarZoneChange(self, av):
|
|
# Calculate zone id
|
|
# Get position of av relative to this grid
|
|
pos = av.getPos(self)
|
|
zoneId = self.getZoneFromXYZ(pos)
|
|
|
|
if not self.isValidZone(zoneId):
|
|
self.notify.warning("%s handleAvatarZoneChange %s: not a valid zone (%s) for pos %s" %
|
|
(self.doId, av.doId, zoneId, pos))
|
|
return
|
|
|
|
# Set the location on the server.
|
|
# setLocation will update the gridParent
|
|
av.b_setLocation(self.doId, zoneId)
|
|
|