qwt3d_color.h

#ifndef __COLORGENERATOR_H__
#define __COLORGENERATOR_H__
 
#include <qstring.h>
#include "qwt3d_global.h"
#include "qwt3d_types.h"
 
namespace Qwt3D {
 
class QWT3D_EXPORT Color
{
public:
    // Implement your color model here
    virtual Qwt3D::RGBA operator()(double x, double y,
                                   double z) const = 0;
    virtual Qwt3D::RGBA operator()(Qwt3D::Triple const &t) const
    {
        return this->operator()(t.x, t.y, t.z);
    }
    // Should create a color vector usable by ColorLegend. The default implementation returns its argument
    virtual Qwt3D::ColorVector &createVector(Qwt3D::ColorVector &vec) { return vec; }
 
    void destroy() const { delete this; }
 
protected:
    virtual ~Color() { }
};
 
class Plot3D;
 
class QWT3D_EXPORT StandardColor : public Color
{
public:
    // Initializes with data and set up a ColorVector with a size of 100 z values (default)
    explicit StandardColor(Qwt3D::Plot3D *data, unsigned size = 100);
    // Receives z-dependent color from ColorVector
    Qwt3D::RGBA operator()(double x, double y,
                           double z) const;
    void setColorVector(Qwt3D::ColorVector const &cv);
    // Resets the standard colors
    void reset(unsigned size = 100);
    // Sets unitary alpha value for all colors
    void setAlpha(double a);
    // Creates color vector for ColorLegend - essentially a copy from the internal vector
    Qwt3D::ColorVector &createVector(Qwt3D::ColorVector &vec)
    {
        vec = colors_;
        return vec;
    }
 
protected:
    Qwt3D::ColorVector colors_;
    Qwt3D::Plot3D *data_;
};
 
} // ns
 
#endif