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)