C:/Users/Marc/Desktop/Phystank/trunk/src/game/VisualComponent.h

//This file is part of PhysTank.
//
//PhysTank is free software: you can redistribute it and/or modify
//it under the terms of the GNU General Public License as published by
//the Free Software Foundation, either version 3 of the License, or
//(at your option) any later version.
//
//PhysTank is distributed in the hope that it will be useful,
//but WITHOUT ANY WARRANTY; without even the implied warranty of
//MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//GNU General Public License for more details.
//
//You should have received a copy of the GNU General Public License
//along with PhysTank.  If not, see <http://www.gnu.org/licenses/>.
//
// © Marc Johnson, 2009

#ifndef _VISUALCOMPONENT_
#define _VISUALCOMPONENT_

#include <map>

#include "game/BaseComponent.h"
#include "game/VisualFactory.h"

class AbstractRenderer;

class VisualComponent : public BaseComponent
{
private:
        
        DEFINE_STL_TYPE(std::map<CONCAT(std::string, BaseVisual*)>, VisualMap);

        VisualMap m_visuals;

public:
        DEFAULT_COMPONENT_CONSTRUCTOR(BaseComponent, VisualComponent, , ,);

        template<class T>
        T * CreateVisual(const std::string & type, const std::string & name);

        template<class T>
        T * GetVisual(const std::string & name);

        void RemoveVisual(const std::string & name);
        void Render(AbstractRenderer* pRenderer, const int nDelta);

        virtual void Update(const int nDelta) { }

        virtual void ToStream(std::ostream & stream) const;
        virtual void FromStream(std::istream & stream);
        virtual bool FromAttribute(const Attribute* pRoot);
};

template<class T>
T * VisualComponent::CreateVisual(const std::string & type, const std::string & name)
{
        if(m_visuals.find(name) != m_visuals.end())
        {
                assert(false && "Attempted to add visual with a name that has already been allocated");
                return false;
        }

        T* pVisual = VisualFactoryCreate<T>(type);

        assert(pVisual && "Invalid visual type");
        
        m_visuals[name] = (BaseVisual*)pVisual;

        pVisual->SetOwner(m_pGameEntity);

        return pVisual;
}

template<class T>
T * VisualComponent::GetVisual(const std::string & name)
{
        return m_visuals[name];
}

REGISTER_COMPONENT(VisualComponent, "component_visual", "visual");

#endif

---