Своя гравитация для каждого тела в Box2d

Добрый вечер.
Это довольно короткий пост, потому я сначала поместил его в «Персональные».

Я только что добавил себе возможность применять к каждому телу свою статическую (только направление, нет притяжения к конкретной точке) гравитацию. Это занимает около минуты.
Решил выложить на случай, если кому-нибудь понадобится.

Итак, мы отредактируем 2 класса бокса, а именно:
Box2D\Dynamics\b2Island.as
Box2D\Dynamics\b2Body.as

В тело мы добавляем 2 переменные, одна из которых будет хранить саму гравитацию, а другая будет флагом, говорящим движку изменена-ли гравитация.
Следующее можно добавить в любое место b2Body.as

b2internal var m_useCustomGravity:Boolean = false;
        b2internal var m_customGravity:b2Vec2 = null;
        
        /**
        * Set custom Gravity. Use this to change the object's gravity.
        */
        public function setCustomGravity(newGravity:b2Vec2):void 
        {
                if (newGravity.IsValid())
                {
                        //trace("valid");
                        m_useCustomGravity = true;
                        m_customGravity = newGravity;
                }
                
        }
        
        /**
        * Get custom Gravity. Returns null is nocustom gravity attached.
        */
        public function getCustomGravity():b2Vec2 
        {
                return m_customGravity;
        }
        
        /**
        * Sets the object to be moved by default gravity.
        */
        public function clearCustomGravity():void 
        {
                m_useCustomGravity = false;
                m_customGravity = null;
        }

Замечание: Проверка newGravity.IsValid() обязательна, в противном случае можно получить 100% загрузку процессора и неработоспособность физики из-за того, что движок будет пытаться использовать нечисловые значения для рассчёта скорости.

Далее надо вставить всего 2 строки в b2ISland.
if (b.m_useCustomGravity)
gravity = b.m_customGravity;

Вот кусок класса со вставленной проверкой:

public function Solve(step:b2TimeStep, gravity:b2Vec2, allowSleep:Boolean) : void
        {
                var i:int;
                var j:int;
                var b:b2Body;
                var joint:b2Joint;
                
                // Integrate velocities and apply damping.
                for (i = 0; i < m_bodyCount; ++i)
                {
                        b = m_bodies[i];
                        
                        if (b.GetType() != b2Body.b2_dynamicBody)
                                continue;
                                
                        //Modified by Midnight to use custom gravity for each body.
                        if (b.m_useCustomGravity)
                                gravity = b.m_customGravity;
                                
                                
                        // Integrate velocities.
                        //b.m_linearVelocity += step.dt * (gravity + b.m_invMass * b.m_force);
                        b.m_linearVelocity.x += step.dt * (gravity.x + b.m_invMass * b.m_force.x);
                        b.m_linearVelocity.y += step.dt * (gravity.y + b.m_invMass * b.m_force.y);
                        b.m_angularVelocity += step.dt * b.m_invI * b.m_torque;
...Обрыв файла

Да, да, вы правильно заметили, это одна из самых важных частей box2d, будьте осторожны с её редактированием.

На всякий случай использование с объяснением:

//var body:b2Body;
body.setCustomGravity(tempVec);   //Изменяет гравитацию тела
body.clearCustomGravity(tempVec); //Возвращает телу стандартную гравитацию.
body.getCustomGravity(tempVec);   //возвращает B2Vec2 с заданной гравитацией, нулевой указатель, если она не задана.

Это и всё, спасибо за внимание.
Отзывы приветствуются.

PS: Метод является доработанной и более удобной версией этого:
www.emanueleferonato.com/2009/07/17/managing-multiple-gravities-with-box2d/