ColorGenAdapter.h

Go to the documentation of this file.

#pragma once
#include <osg\Referenced>
#include "Utils.h"
#include "ConstantColorGen.h"
#include "GradientColorGen.h"
#include "SectorColorGen.h"
#include "types.h"

namespace APDYNAMICART
{

class ColorGenAdapter : public PoAAdapter<ColorGen>
{
public:
    ColorGenAdapter() : _colorIndex(0) {}
    virtual void setColor(const osg::Vec4& color) {}
    virtual osg::Vec4 getColor() { return osg::Vec4(0.0, 1.0, 0.0, 1.0); }

    int getColorIndex() const { return _colorIndex; }
    void setColorIndex(int val) { _colorIndex = val; }

protected:
    int _colorIndex;
    
};

class ConstantColorAdapter : public ColorGenAdapter
{
public:
    ConstantColorAdapter() : _colorGen(new ConstantColorGen) {}
    ConstantColorAdapter(const osg::Vec4& color)
        : _colorGen(new ConstantColorGen(color)) {}

    META_DAObject(ConstantColorAdapter, "Constant")

    virtual void createMenu(MenuPtr menu);
    virtual int load( XERCES_CPP_NAMESPACE_QUALIFIER DOMNode* ParametersNode );
    virtual void save(XERCES_CPP_NAMESPACE_QUALIFIER DOMDocument* Document, XERCES_CPP_NAMESPACE_QUALIFIER DOMNode* poaNode);
    virtual void setColor( const osg::Vec4& color ) { _colorGen->setColorValue(color); }
    virtual ColorGenPtr getObject() { return _colorGen; }

    virtual osg::Vec4 getColor() { return _colorGen->getColorValue(); }

protected:
    osg::ref_ptr<ConstantColorGen> _colorGen;
};


class GradientColorAdapter : public ColorGenAdapter
{
public:
    GradientColorAdapter() : _colorGen(new GradientColorGen) {}
    GradientColorAdapter(const osg::Vec4& color1, const osg::Vec4& color2) 
        : _colorGen(new GradientColorGen(color1, color2)) {}

    META_DAObject(GradientColorAdapter, "Gradient")

    virtual void createMenu(MenuPtr menu);
    virtual int load( XERCES_CPP_NAMESPACE_QUALIFIER DOMNode* ParametersNode );
    virtual void save(XERCES_CPP_NAMESPACE_QUALIFIER DOMDocument* Document, XERCES_CPP_NAMESPACE_QUALIFIER DOMNode* poaNode);
    virtual ColorGenPtr getObject() { return _colorGen; }

    virtual void setColor( const osg::Vec4& color );
    virtual osg::Vec4 getColor() { return _colorGen->getColorStart(); }

protected:
    osg::ref_ptr<GradientColorGen> _colorGen;
};

//NOTE: Disabled due to revision in SectorColorGen class

//class SectorColorAdapter : public ColorGenAdapter
//{
//public:
//    SectorColorAdapter() : _colorGen(new SectorColorGen) {}
//
//    META_DAObject(SectorColorAdapter, "Sector coloring")
//
//    virtual ColorGenPtr getObject() { return _colorGen; }
//    virtual void setColor( const osg::Vec4& color ) { _colorGen->setDefaultColor(color); }
//    virtual osg::Vec4 getColor() { return _colorGen->getDefaultColor(); }
//    virtual void createMenu( MenuPtr menu );
//    virtual bool sliderChanged( const char* sliderID, float sliderPos );
//
//protected:
//    osg::ref_ptr<SectorColorGen> _colorGen;
//};

class ColorGenFactory : public AdapterFactory<ColorGenAdapter>
{
public:
    ColorGenFactory()
    {
        _adapters.push_back(new ConstantColorAdapter);
        _adapters.push_back(new GradientColorAdapter);
        //_adapters.push_back(new SectorColorAdapter);
    }

    ColorGenFactory( const ColorGenFactory& orig )
    {
        //TODO some more generic approach here would be nice
        //  but how to do it?


        std::vector<osg::ref_ptr<ColorGenAdapter>> origAdapters = orig.getSupportedAdapters();

        _adapters.push_back(new ConstantColorAdapter(origAdapters[0]->getColor()));

        ColorGen *tmpGen = (origAdapters[1]->getObject()).get();
        GradientColorGen *tmpGrdGen = dynamic_cast<GradientColorGen *>(tmpGen);
        _adapters.push_back(new GradientColorAdapter(tmpGrdGen->getColorStart(), tmpGrdGen->getColorEnd()));
    }
};

}