baseobjectscene.h

Go to the documentation of this file.

#pragma once
 
#include <osg/PositionAttitudeTransform>
 
#include "baseobject.h"
 
namespace adtf
{
 
namespace disptb
{
 
namespace mixinlib
{
 
namespace endor
{

class cBaseObjectScene : public osg::PositionAttitudeTransform
{
protected:
public:
    cBaseObjectScene(const tChar* strName = ""): m_strName(strName)
    {
    }

    virtual ~cBaseObjectScene()
    {
    }

    virtual const tChar* GetName()
    {
        return m_strName.c_str();
    }

    virtual tResult AddObject(cBaseObject* pObj)
    {
        // check if an object with this id already exists
        auto oObj = m_oObjects.find(pObj->GetId());
        if (oObj != m_oObjects.end())
        {
            RETURN_ERROR(ERR_DEVICE_IN_USE);
        }
 
        // add it to the scene
        addChild(pObj);
        // and store it in our map
        m_oObjects[pObj->GetId()] = pObj;
 
        RETURN_NOERROR;
    }

    virtual tResult RemoveObject(tUInt32 nId)
    {
        // find the object
        auto oObj = m_oObjects.find(nId);
        if (oObj != m_oObjects.end())
        {
            // remove the osg node (expensive operation)
            removeChild(oObj->second.get());
            // remove it from our map
            m_oObjects.erase(oObj);
        }
        else
        {
            RETURN_ERROR(ERR_NOT_FOUND);
        }
 
        RETURN_NOERROR;
    }

    virtual tResult GetObject(tUInt32 nId, cBaseObject** ppObj)
    {
        // find it in the map O(log N)
        auto oObj = m_oObjects.find(nId);
        if (oObj != m_oObjects.end())
        {
            *ppObj = oObj->second.get();
        }
        else
        {
            RETURN_ERROR(ERR_NOT_FOUND);
        }
 
        RETURN_NOERROR;
    }

    virtual tResult RemoveInvalid()
    {
        tUInt nChildren = getNumChildren();
        for (tUInt nChild = 0; nChild < nChildren; ++nChild)
        {
            // no dynamic cast is necessary we know that we only added cIdSwitch children
            cBaseObject* pObj = static_cast<cBaseObject*>(getChild(nChild));
            if (pObj->IsUpdated())
            {
                // if it has been updated during this cycle we reset it.
                pObj->Reset();
            }
            else
            {
                // if it has not been updated we remove it from the graph and the map
                m_oObjects.erase(m_oObjects.find(pObj->GetId()));
                removeChild(nChild); // expensive since osg uses a std::vector
                // this should also have free'd the object
                --nChildren;
                --nChild;
            }
        }
 
        RETURN_NOERROR;
    }

    virtual tResult SetPosition(tFloat64 fPosX, tFloat64 fPosY, tFloat64 fPosZ)
    {
        setPosition(osg::Vec3(fPosX, fPosY, fPosZ));
        RETURN_NOERROR;
    }

    virtual tResult SetRotation(tFloat64 fRotX, tFloat64 fRotY, tFloat64 fRotZ)
    {
        setAttitude(osg::Quat(fRotX, osg::Vec3(1, 0, 0), fRotY, osg::Vec3(0, 1, 0), fRotZ, osg::Vec3(0, 0, 1)));
        RETURN_NOERROR;
    }
 
protected:
    std::string m_strName;
    std::map<tUInt32, osg::ref_ptr<cBaseObject>> m_oObjects;
};
 
} // namespace endor

using endor::cBaseObjectScene;
 
} // namespace mixinlib
 
} // namespace disptb
 
} // namespace adtf