def handle(self, msgType, di):
"""
Send data from the net on the local netMessenger.
"""
assert self.notify.debugCall()
if msgType not in self.__type2message:
self.notify.warning('Received unknown message: %d' % msgType)
return
message = self.__type2message[msgType]
sentArgs = loads(di.getString())
if type(sentArgs) != list:
self.notify.warning('Received non-list item in %s message: %r' %
(message, sentArgs))
return
Messenger.send(self, message, sentArgs=sentArgs)
def __init__(self, air, baseChannel=20000, baseMsgType=20000):
"""
air is the AI Repository.
baseChannel is the channel that the first message is sent on.
baseMsgType is the MsgType of the same.
"""
assert self.notify.debugCall()
Messenger.__init__(self)
self.air = air
self.baseChannel = baseChannel
self.baseMsgType = baseMsgType
self.__message2type = {}
self.__type2message = {}
self.__message2channel = {}
def clear(self):
assert self.notify.debugCall()
for i in self.channels:
self.air.unRegisterChannel(i)
del self.air
del self.channels
Messenger.clear(self)
def clear(self):
assert self.notify.debugCall()
for i in self.channels:
self.air.unRegisterChannel(i)
del self.air
del self.channels
Messenger.clear(self)
def __init__(self, air, baseChannel=20000, baseMsgType=20000):
Messenger.__init__(self)
self.air = air
self.baseChannel = baseChannel
self.baseMsgType = baseMsgType
self.__message2type = {}
self.__type2message = {}
self.__message2channel = {}
def send(self, message, sentArgs=[]):
"""
Send message to anything that's listening for it.
"""
assert self.notify.debugCall()
datagram = self.prepare(message, sentArgs)
self.air.send(datagram)
Messenger.send(self, message, sentArgs=sentArgs)
def send(self, message, sentArgs=[]):
"""
Send message to anything that's listening for it.
"""
assert self.notify.debugCall()
datagram = self.prepare(message, sentArgs)
self.air.send(datagram)
Messenger.send(self, message, sentArgs=sentArgs)
def accept(self, message, *args):
if message not in self.__message2channel:
self.notify.error('Tried to accept unregistered message %s!' %
message)
return
anyAccepting = bool(self.whoAccepts(message))
if not anyAccepting:
self.air.registerForChannel(self.__message2channel[message])
Messenger.accept(self, message, *args)
def accept(self, message, *args):
if message not in self.__message2channel:
self.notify.error('Tried to accept unregistered message %s!' % message)
return
anyAccepting = bool(self.whoAccepts(message))
if not anyAccepting:
self.air.registerForChannel(self.__message2channel[message])
Messenger.accept(self, message, *args)
def __init__(self, air, channels):
"""
air is the AI Repository.
channels is a list of channel IDs (uint32 values)
"""
assert self.notify.debugCall()
Messenger.__init__(self)
self.air = air
self.channels = channels
for i in self.channels:
self.air.registerForChannel(i)
def __init__(self, air, channels):
"""
air is the AI Repository.
channels is a list of channel IDs (uint32 values)
"""
assert self.notify.debugCall()
Messenger.__init__(self)
self.air=air
self.channels=channels
for i in self.channels:
self.air.registerForChannel(i)
def handle(self, msgType, di):
if msgType not in self.__type2message:
self.notify.warning('Received unknown message: %d' % msgType)
return
message = self.__type2message[msgType]
sentArgs = loads(di.getString())
if type(sentArgs) != list:
self.notify.warning('Received non-list item in %s message: %r' %
(message, sentArgs))
return
Messenger.send(self, message, sentArgs=sentArgs)
def handle(self, pickleData):
"""
Send pickleData from the net on the local netMessenger.
The internal data in pickleData should have a tuple of
(messageString, sendArgsList).
"""
assert self.notify.debugCall()
messageType=self.air.getMsgType()
if messageType:
message=MESSAGE_TYPES[messageType-1]
sentArgs=loads(pickleData)
else:
(message, sentArgs) = loads(pickleData)
Messenger.send(self, message, sentArgs=sentArgs)
def handle(self, pickleData):
"""
Send pickleData from the net on the local netMessenger.
The internal data in pickleData should have a tuple of
(messageString, sendArgsList).
"""
assert self.notify.debugCall()
messageType = self.air.getMsgType()
if messageType:
message = MESSAGE_TYPES[messageType - 1]
sentArgs = loads(pickleData)
else:
(message, sentArgs) = loads(pickleData)
Messenger.send(self, message, sentArgs=sentArgs)
def __init__(self, air, baseChannel=20000, baseMsgType=20000):
"""
air is the AI Repository.
baseChannel is the channel that the first message is sent on.
baseMsgType is the MsgType of the same.
"""
assert self.notify.debugCall()
Messenger.__init__(self)
self.air=air
self.baseChannel = baseChannel
self.baseMsgType = baseMsgType
self.__message2type = {}
self.__type2message = {}
self.__message2channel = {}
def handle(self, msgType, di):
"""
Send data from the net on the local netMessenger.
"""
assert self.notify.debugCall()
if msgType not in self.__type2message:
self.notify.warning('Received unknown message: %d' % msgType)
return
message = self.__type2message[msgType]
sentArgs=loads(di.getString())
if type(sentArgs) != list:
self.notify.warning('Received non-list item in %s message: %r' %
(message, sentArgs))
return
Messenger.send(self, message, sentArgs=sentArgs)
def __init__(self):
DirectObject.__init__(self)
self.filename = None
self.unsaved = False
self.idGenerator = IDGenerator()
self.world = None
self.isOpen = False
self.gsg = None
self.page = None
self.shaderGenerator = None
self.numlights = 0
# Each document has its own message bus, task manager, and event manager.
self.taskMgr = TaskManager()
self.taskMgr.mgr = AsyncTaskManager("documentTaskMgr")
self.messenger = Messenger(self.taskMgr)
self.eventMgr = EventManager(messenger=self.messenger,
taskMgr=self.taskMgr)
self.messenger.setEventMgr(self.eventMgr)
self.render = NodePath("docRender")
self.render.setAttrib(LightRampAttrib.makeIdentity())
#self.render.setShaderAuto()
self.viewportMgr = ViewportManager(self)
self.toolMgr = ToolManager(self)
self.selectionMgr = SelectionManager(self)
self.actionMgr = ActionManager(self)
# Create the page that the document is viewed in.
self.page = DocumentWindow(self)
#self.createShaderGenerator()
self.toolMgr.addTools()
self.eventMgr.restart()
def clear(self):
assert self.notify.debugCall()
Messenger.clear(self)
def send(self, message, sentArgs=[]):
datagram = self.prepare(message, sentArgs)
self.air.send(datagram)
Messenger.send(self, message, sentArgs=sentArgs)
class Document(DirectObject):
def __init__(self):
DirectObject.__init__(self)
self.filename = None
self.unsaved = False
self.idGenerator = IDGenerator()
self.world = None
self.isOpen = False
self.gsg = None
self.gsg2D = None
self.shaderGenerators = []
self.numlights = 0
# Each document has its own message bus, task manager, and event manager.
self.taskMgr = TaskManager()
self.taskMgr.mgr = AsyncTaskManager("documentTaskMgr")
self.messenger = Messenger(self.taskMgr)
self.eventMgr = EventManager(messenger = self.messenger, taskMgr = self.taskMgr)
self.messenger.setEventMgr(self.eventMgr)
self.render = NodePath("docRender")
self.render.setAttrib(LightRampAttrib.makeIdentity())
self.render.setShaderAuto()
self.viewportMgr = ViewportManager(self)
self.toolMgr = ToolManager(self)
self.selectionMgr = SelectionManager(self)
self.actionMgr = ActionManager(self)
# Create the page that the document is viewed in.
self.page = DocumentWindow(self)
self.createShaderGenerator()
self.toolMgr.addTools()
self.eventMgr.restart()
def step(self):
#print("Stepping", self)
self.taskMgr.step()
# Called when the document's tab has been switched into.
def activated(self):
# Move document constructs into global space so we don't have to directly
# reference the document for them.
base.render = self.render
base.viewportMgr = self.viewportMgr
base.toolMgr = self.toolMgr
base.selectionMgr = self.selectionMgr
base.actionMgr = self.actionMgr
builtins.render = self.render
messenger.send('documentActivated', [self])
# Called when the document's tab has been switched out of.
def deactivated(self):
messenger.send('documentDeactivated', [self])
def getNextID(self):
return self.idGenerator.getNextID()
def reserveID(self, id):
self.idGenerator.reserveID(id)
def getNextFaceID(self):
return self.idGenerator.getNextFaceID()
def reserveFaceID(self, id):
self.idGenerator.reserveFaceID(id)
def save(self, filename = None):
# if filename is not none, this is a save-as
if not filename:
filename = self.filename
kv = CKeyValues()
self.world.doWriteKeyValues(kv)
kv.write(filename, 4)
self.filename = filename
self.unsaved = False
base.actionMgr.documentSaved()
self.updateTabText()
def close(self):
if not self.isOpen:
return
self.toolMgr.cleanup()
self.toolMgr = None
self.world.delete()
self.world = None
self.idGenerator.cleanup()
self.idGenerator = None
self.filename = None
self.unsaved = None
self.isOpen = None
self.gsg = None
self.gsg2D = None
self.shaderGenerators = None
self.viewportMgr.cleanup()
self.viewportMgr = None
self.actionMgr.cleanup()
self.actionMgr = None
self.selectionMgr.cleanup()
self.selectionMgr = None
self.eventMgr.shutdown()
self.eventMgr = None
self.messenger = None
self.taskMgr.destroy()
self.taskMgr = None
self.render.removeNode()
self.render = None
self.page.cleanup()
self.page = None
def __newMap(self):
self.unsaved = False
self.idGenerator.reset()
self.world = World(self.getNextID())
self.world.reparentTo(self.render)
self.isOpen = True
self.toolMgr.switchToSelectTool()
self.updateTabText()
def updateTabText(self):
title = self.getMapTitle()
self.page.setWindowTitle(title)
base.setWindowSubTitle(title)
def r_open(self, kv, parent = None):
classDef = MapObjectFactory.MapObjectsByName.get(kv.getName())
if not classDef:
return
id = int(kv.getValue("id"))
self.reserveID(id)
obj = classDef(id)
obj.readKeyValues(kv)
obj.reparentTo(parent)
if classDef is World:
self.world = obj
for i in range(kv.getNumChildren()):
self.r_open(kv.getChild(i), obj)
def getViewport(self, type):
return self.page.docWidget.viewports[type]
def createShaderGenerator(self):
for gsg, vp in [(self.gsg, self.getViewport(VIEWPORT_3D)), (self.gsg2D, self.getViewport(VIEWPORT_2D_FRONT))]:
shgen = BSPShaderGenerator(vp.win, gsg, vp.camera, self.render)
gsg.setShaderGenerator(shgen)
for shader in ShaderGlobals.getShaders():
shgen.addShader(shader)
self.shaderGenerators.append(shgen)
def open(self, filename = None):
# if filename is none, this is a new document/map
if not filename:
self.__newMap()
return
# opening a map from disk, read through the keyvalues and
# generate the objects
self.idGenerator.reset()
root = CKeyValues.load(filename)
for i in range(root.getNumChildren()):
self.r_open(root.getChild(i))
self.unsaved = False
self.filename = filename
self.isOpen = True
self.toolMgr.switchToSelectTool()
self.updateTabText()
def markSaved(self):
self.unsaved = False
self.updateTabText()
def markUnsaved(self):
self.unsaved = True
self.updateTabText()
def isUnsaved(self):
return self.unsaved
def getMapName(self):
if not self.filename:
return "Untitled"
return self.filename.getBasename()
def getMapTitle(self):
name = self.getMapName()
if self.unsaved:
name += " *"
return name
def updateAllViews(self, now = False):
for vp in self.viewportMgr.viewports:
vp.updateView(now)
def update2DViews(self, now = False):
for vp in self.viewportMgr.viewports:
if vp.is2D():
vp.updateView(now)
def update3DViews(self, now = False):
for vp in self.viewportMgr.viewports:
if vp.is3D():
vp.updateView(now)
self.air.send(datagram)
def handle(self, pickleData):
"""
Send pickleData from the net on the local netMessenger.
The internal data in pickleData should have a tuple of
(messageString, sendArgsList).
"""
assert self.notify.debugCall()
messageType=self.air.getMsgType()
if messageType:
message=MESSAGE_TYPES[messageType-1]
sentArgs=loads(pickleData)
else:
(message, sentArgs) = loads(pickleData)
Messenger.send(self, message, sentArgs=sentArgs)
def register(self, code, message):
assert self.notify.debugCall()
channel = self.baseChannel + code
msgType = self.baseMsgType + code
if message in self.__message2type:
self.notify.error('Tried to register message %s twice!' % message)
return
self.__message2type[message] = msgType
self.__type2message[msgType] = message
self.__message2channel[message] = channel
def clear(self):
assert self.notify.debugCall()
Messenger.clear(self)
def clear(self):
Messenger.clear(self)
