Environment.cpp

Go to the documentation of this file.

/* 
*
*/

#include "Environment.h"
#include "CentralBody/CentralBody.h"
namespace O_SESSAME {
    
Environment::Environment() : m_CalculatedForces(3), 
                             m_CalculatedTorques(3), 
                             m_ForcesFunctor(const_cast<Environment*>(this), &Environment::GetForces),
                             m_TorquesFunctor(const_cast<Environment*>(this), &Environment::GetTorques)
{
    m_pCB = NULL;
}
    
Environment::~Environment()
{
    if(m_pCB)
        m_pCB->ReleaseReference();
}

Vector Environment::GetTorques(const ssfTime& _currentTime, const OrbitState& _currentOrbitState, const AttitudeState& _currentAttitudeState)
{
    m_CalculatedTorques.setToValue(0);
    m_funcIterator = m_TorqueFunctions.begin();
    while (m_funcIterator != m_TorqueFunctions.end())
    {
        m_CalculatedTorques += (*m_funcIterator).Evaluate(_currentTime, _currentOrbitState, _currentAttitudeState);
        ++m_funcIterator;
    }
    return m_CalculatedTorques;
}

Vector Environment::GetForces(const ssfTime& _currentTime, const OrbitState& _currentOrbitState, const AttitudeState& _currentAttitudeState)
{
    m_CalculatedForces.setToValue(0);
    m_funcIterator = m_ForceFunctions.begin();
    while (m_funcIterator != m_ForceFunctions.end())
    {
        m_CalculatedForces += (*m_funcIterator).Evaluate(_currentTime, _currentOrbitState, _currentAttitudeState);
        ++m_funcIterator;
    }
    return m_CalculatedForces;
}
    
void Environment::SetCentralBody(CentralBody *_pCB)
{
    m_pCB = _pCB;
    return;
}

const CentralBody* Environment::GetCentralBody() const
{
    return m_pCB;
}
    
void Environment::AddForceFunction(const EnvFunction &_forceFunc)
{
    // iterate through the set of Force Functions to prevent duplication./*
    m_funcIterator = m_ForceFunctions.begin();
    while (m_funcIterator != m_ForceFunctions.end())
    {
        if(*m_funcIterator == _forceFunc)
            return;
    }*/
    m_ForceFunctions.push_back(_forceFunc);
    return;
}
void Environment::AddTorqueFunction(const EnvFunction &_torqueFunc)
{
    // iterate through the set of Torque Functions to prevent duplication./*
    m_funcIterator = m_TorqueFunctions.begin();
    while (m_funcIterator != m_TorqueFunctions.end())
    {
        if(*m_funcIterator == _torqueFunc)
            return;
    }*/
    m_TorqueFunctions.push_back(_torqueFunc);
    return;
}
    
const Functor& Environment::GetForceFunction() const
{
    return m_ForcesFunctor;
}

const Functor& Environment::GetTorqueFunction() const
{
    return m_TorquesFunctor;
}
// EnvFunction Member Functions

EnvFunction::EnvFunction(pt2EnvFunc _EnvFuncPtr) : m_NumParameters(0) , m_EnvFuncParameters(m_NumParameters)
{
    m_EnvFuncPtr = _EnvFuncPtr;
}
EnvFunction::EnvFunction(pt2EnvFunc _EnvFuncPtr, const int &_numParameters) : m_NumParameters(_numParameters) , m_EnvFuncParameters(m_NumParameters)
{
    m_EnvFuncPtr = _EnvFuncPtr;
}
Vector EnvFunction::Evaluate(const ssfTime& _currentTime, const OrbitState& _currentOrbitState, const AttitudeState& _currentAttitudeState)
{
    return m_EnvFuncPtr(_currentTime, _currentOrbitState, _currentAttitudeState, m_EnvFuncParameters);
}

void EnvFunction::AddParameter(void* _parameter, int _paramNumber)
{
    if (_paramNumber > m_NumParameters)
    {    
        // Calculate the number of parameters "missing" to be temporarily filled
        int delta = _paramNumber - m_NumParameters;
        if(delta > 1)
        {
            for(int ii = 1; ii < delta; ++ii)
            {
                m_EnvFuncParameters.push_back(0);
            }
        }
        m_EnvFuncParameters.push_back(_parameter);
    }
    else
        m_EnvFuncParameters.insert(m_EnvFuncParameters.begin() + _paramNumber - 1, _parameter);
    
    m_NumParameters = _paramNumber;

}
    
void EnvFunction::ChangeParameter(void* _parameter, int _paramNumber)
{
    RemoveParameter(_paramNumber);
    AddParameter(_parameter, _paramNumber);
}

void EnvFunction::RemoveParameter(int _paramNumber)
{
    if(_paramNumber <= m_NumParameters)
        m_EnvFuncParameters.erase(m_EnvFuncParameters.begin()+_paramNumber);
    else if(_paramNumber == -1)
        m_EnvFuncParameters.pop_back();
    else
        return;
    
    --m_NumParameters;
}
} // close namespace O_SESSAME

// Do not change the comments below - they will be added automatically by CVS
/*****************************************************************************
*       $Log: Environment.cpp,v $
*       Revision 1.8  2003/06/12 18:06:35  nilspace
*       .
*       
*       Revision 1.7  2003/06/10 15:58:52  nilspace
*       GetTorques now incrementes the function iterator.
*       
*       Revision 1.6  2003/05/22 03:42:32  nilspace
*       Moved documentation to the implementation file.
*       
*       Revision 1.5  2003/05/13 18:59:16  nilspace
*       Fixed some comments.
*       
*       Revision 1.4  2003/05/05 20:45:46  nilspace
*       Changed the pass-by-address parameters to pass-by-reference.
*       
*       Revision 1.3  2003/05/02 02:16:29  nilspace
*       Documented the API.
*       Added a GetTorquesFunction().
*       
*       Revision 1.2  2003/04/27 22:04:33  nilspace
*       Created the namespace O_SESSAME.
*       
*       Revision 1.1  2003/04/08 22:37:22  nilspace
*       Initial Submission.
*       
*
******************************************************************************/

---