def StopTransform( self ):
"""
Stop the transfrom operation by removing the selection modified
message task. Also create a transform action and push it onto the undo
queue.
"""
# Remove the transform task
if self._xformTask in taskMgr.getAllTasks():
taskMgr.remove( self._xformTask )
self._xformTask = None
actGizmo = self.gizmoMgr.GetActiveGizmo()
actns = []
for i, np in enumerate( actGizmo.attachedNps ):
actns.append( actions.Transform( np, np.getTransform(), actGizmo.initNpXforms[i] ) )
actn = actions.Composite( actns )
self.actnMgr.Push( actn )
self.gizmo = False
# Make sure to mark the NodePath as dirty in case it is a child of a
# model root.
wrpr = base.game.nodeMgr.Wrap( np )
wrpr.SetModified( True )
# Call OnModified next frame. Not sure why but if we call it straight
# away it causes a small jitter when xforming...
taskMgr.doMethodLater( 0, self.doc.OnModified, 'dragDrop',
[actGizmo.attachedNps] )
def Ungroup(nps):
"""
Create the ungroup action, execute it and push it onto the undo queue.
"""
pNps = []
cNpSets = []
for np in nps:
wrpr = base.game.nodeMgr.Wrap(np)
pNps.append(wrpr.GetParent().data)
cNpSets.append([cWrpr.data for cWrpr in wrpr.GetChildren()])
# Remove those nodes which were empty NodePaths.
rmvNps = [np for np in nps if np.node().isExactType(pm.PandaNode)]
actns = []
actns.append(actions.Deselect(nps))
for i, cNps in enumerate(cNpSets):
actns.extend([actions.Parent(cNp, pNps[i]) for cNp in cNps])
actns.extend([actions.Remove(np) for np in rmvNps])
actns.append(actions.Select([cNp for cNps in cNpSets for cNp in cNps]))
actn = actions.Composite(actns)
wx.GetApp().actnMgr.Push(actn)
actn()
wx.GetApp().doc.OnModified(nps.append(rmvNps))
def SetConnections(tgtComps, cnnctns):
"""
Create the connect action, execute it and push it onto the undo queue.
"""
actns = [actions.SetConnections(tgtComps, cnnctn) for cnnctn in cnnctns]
actn = actions.Composite(actns)
wx.GetApp().actnMgr.Push(actn)
actn()
wx.GetApp().doc.OnModified()
def Parent(comps, pComp):
"""
Create the parent action, execute it and push it onto the undo queue.
"""
actns = [actions.Parent(comp, pComp) for comp in comps]
actn = actions.Composite(actns)
wx.GetApp().actnMgr.Push(actn)
actn()
wx.GetApp().doc.OnModified(comps)
def Select(comps):
"""
Create the select composite action, execute it and push it onto the
undo queue.
"""
actns = [actions.Deselect(base.selection.comps), actions.Select(comps)]
actn = actions.Composite(actns)
wx.GetApp().actnMgr.Push(actn)
actn()
wx.GetApp().doc.OnRefresh(comps)
def Remove(comps):
"""
Create the remove composite action, execute it and push it onto the undo
queue.
"""
actns = []
actns.append(actions.Deselect(comps))
actns.extend([actions.Remove(comp) for comp in comps])
actn = actions.Composite(actns)
wx.GetApp().actnMgr.Push(actn)
actn()
wx.GetApp().doc.OnModified(comps)
def SetAttribute(comps, attrs, val):
"""
Create the set attribute action, execute it and push it onto the undo
queue.
"""
actns = [
actions.SetAttribute(comps[i], attrs[i], val)
for i in range(len(comps))
]
actn = actions.Composite(actns)
wx.GetApp().actnMgr.Push(actn)
actn()
wx.GetApp().doc.OnModified(comps)
def Replace(fromComp, toComp):
"""
"""
actns = [
actions.Deselect([fromComp]),
actions.Remove(fromComp),
actions.Add(toComp),
actions.Select([toComp])
]
actn = actions.Composite(actns)
wx.GetApp().actnMgr.Push(actn)
actn()
wx.GetApp().doc.OnModified([fromComp, toComp])
def Duplicate(comps):
"""
Create the duplicate composite action, execute it and push it onto the
undo queue.
"""
selComps = base.selection.comps
base.selection.Clear()
dupeComps = []
for comp in comps:
wrpr = base.game.nodeMgr.Wrap(comp)
dupeComps.append(wrpr.Duplicate())
actns = []
actns.append(actions.Deselect(selComps))
actns.extend([actions.Add(dupeComp) for dupeComp in dupeComps])
actns.append(actions.Select(dupeComps))
actn = actions.Composite(actns)
wx.GetApp().actnMgr.Push(actn)
actn()
wx.GetApp().doc.OnModified(dupeComps)
def Group(nps):
"""
Create the group action, execute it and push it onto the undo queue.
"""
# Find the lowest common ancestor for all NodePaths - this will be the
# parent for the group NodePath.
cmmnNp = nps[0].getParent()
for np in nps:
cmmnNp = cmmnNp.getCommonAncestor(np)
grpNp = pm.NodePath('group')
grpNp.reparentTo(cmmnNp)
actns = []
actns.append(actions.Add(grpNp))
actns.extend([actions.Parent(np, grpNp) for np in nps])
actns.append(actions.Deselect(nps))
actns.append(actions.Select([grpNp]))
actn = actions.Composite(actns)
wx.GetApp().actnMgr.Push(actn)
actn()
wx.GetApp().doc.OnModified(nps.append(grpNp))
