GradientColorGen.h

Go to the documentation of this file.

#pragma once
#include "ColorGen.h"

class GradientColorGen : public ColorGen
{
public:
    GradientColorGen() : _colorStart(osg::Vec4(1,1,1,1)), _colorEnd(osg::Vec4(1,1,1,1)) {}
    GradientColorGen(const osg::Vec4& colorStart, const osg::Vec4& colorEnd)
        : _colorStart(colorStart), _colorEnd(colorEnd) {}

    META_DAObject(GradientColorGen, "Gradient Brush")

    
    virtual Vec4ArrayPtr getColors( unsigned int leadSplineSize, unsigned int crossSectionSize )
    {
        _dirty = false;

        Vec4ArrayPtr result = new osg::Vec4Array;
        unsigned int arraySize = leadSplineSize * crossSectionSize;
        result->reserve(arraySize);
        
        osg::Vec4 gradientStep = (_colorEnd - _colorStart) / (double)leadSplineSize;
        osg::Vec4 actualColor;

        for (unsigned int i = 0; i < leadSplineSize; i++)
        {
            actualColor = _colorStart + gradientStep * i;
            for (unsigned int j = 0; j < crossSectionSize; j++)
            {
                result->push_back(actualColor);
            }
        }
        return result;
    }
    
    inline osg::Vec4 getColorEnd() const { return _colorEnd; }
    inline void setColorEnd(osg::Vec4 val) { _dirty = true; _colorEnd = val; }

    inline osg::Vec4 getColorStart() const { return _colorStart; }
    inline void setColorStart(osg::Vec4 val) { _dirty = true; _colorStart = val; }

protected:
    osg::Vec4    _colorStart;    
    osg::Vec4    _colorEnd;    
};