Physics Simulation Forum

//===========================================================================
class IMotionStateCallback
{
public:
   virtual void      OnMotionStateChanged ( const btTransform& worldTrans ) = 0;
};

//===========================================================================
class MotionState : public btMotionState
{
public:
                  MotionState         ( const btTransform& startTrans = btTransform::getIdentity())
                  : worldTrans_      ( startTrans )
                  , stateChangedCallbacks_ ()
                  { }

   //---------------------------------------------------------------------------
   // btMotionState interface.

   /// Synchronizes world transform from user to physics
   virtual void      getWorldTransform   ( btTransform& centerOfMassWorldTrans ) const
   {
      centerOfMassWorldTrans = worldTrans_;
   }

   ///synchronizes world transform from physics to user
   ///Bullet only calls the update of worldtransform for active objects
   virtual void      setWorldTransform   ( const btTransform& centerOfMassWorldTrans )
   {
      worldTrans_ = centerOfMassWorldTrans;
      if (!stateChangedCallbacks_.empty())
      {
         for (MotionStateCallbacks_t::const_iterator it = stateChangedCallbacks_.begin(),
               it_end = stateChangedCallbacks_.end(); it != it_end; ++it)
         {
            (*it)->OnMotionStateChanged (worldTrans_);
         }
      }
   }

   // Internal interface.
   const btTransform&   getWorldTransform   () const
   {
      return worldTrans_;
   }

   void            setWorldTransform   ( const btTransform& centerOfMassWorldTrans
                                 , bool invokeCallbacks = true )
   {
      worldTrans_ = centerOfMassWorldTrans;
      if (invokeCallbacks && !stateChangedCallbacks_.empty())
      {
         for (MotionStateCallbacks_t::const_iterator it = stateChangedCallbacks_.begin(),
               it_end = stateChangedCallbacks_.end(); it != it_end; ++it)
         {
            (*it)->OnMotionStateChanged (worldTrans_);
         }
      }
   }

   void            AddMotionStateCallback ( IMotionStateCallback* callback )
   {
      IwAssert(PHYSICS, stateChangedCallbacks_.find ( callback ) < 0);
      stateChangedCallbacks_.push_back ( callback );
   }

   void            RemoveMotionStateCallback ( IMotionStateCallback* callback )
   {
      IwAssert(PHYSICS, stateChangedCallbacks_.find (callback) >= 0);
      stateChangedCallbacks_.find_and_remove_fast ( callback );
   }

private:
   typedef CIwArray<IMotionStateCallback*>   MotionStateCallbacks_t;

   btTransform                        worldTrans_;
   MotionStateCallbacks_t               stateChangedCallbacks_;
};