historical/toontown-classic.git/panda/direct/particles/GlobalForceGroup.py
2024-01-16 11:20:27 -06:00

23 lines
750 B
Python

from . import ForceGroup
class GlobalForceGroup(ForceGroup.ForceGroup):
def __init__(self, name = None):
ForceGroup.ForceGroup.__init__(self, name)
def addForce(self, force):
ForceGroup.ForceGroup.addForce(force)
if (force.isLinear() == 0):
# Physics manager will need an angular integrator
base.addAngularIntegrator()
if (force.isLinear() == 1):
physicsMgr.addLinearForce(force)
else:
physicsMgr.addAngularForce(force)
def removeForce(self, force):
ForceGroup.ForceGroup.removeForce(force)
if (force.isLinear() == 1):
physicsMgr.removeLinearForce(force)
else:
physicsMgr.removeAngularForce(force)