Source code for direct.particles.GlobalForceGroup

from . import ForceGroup
from direct.showbase.PhysicsManagerGlobal import physicsMgr


[docs]class GlobalForceGroup(ForceGroup.ForceGroup):

[docs]    def __init__(self, name = None):
        ForceGroup.ForceGroup.__init__(self, name)

[docs]    def addForce(self, force):
        ForceGroup.ForceGroup.addForce(self, 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)

[docs]    def removeForce(self, force):
        ForceGroup.ForceGroup.removeForce(self, force)
        if (force.isLinear() == 1):
            physicsMgr.removeLinearForce(force)
        else:
            physicsMgr.removeAngularForce(force)