ColorMap.hpp

Go to the documentation of this file.

#ifndef COLORMAP_H_
#define COLORMAP_H_

#include <iostream>

/***
 * @brief Class to generate and handle color gradients
 *
 * @TODO: Integrated from show.
 */

namespace lvr2
{

enum GradientType
{
    SOLID = 0,
    GREY = 1,
    HSV = 2,
    JET = 3,
    HOT = 4,
    SHSV = 5,
    SIMPSONS = 6
};

class ColorMap
{
public:

    ColorMap(size_t buckets) : m_numBuckets(buckets) {}

    virtual ~ColorMap() {};

    void getColor(float* color, size_t bucket, GradientType gradient = GREY);

private:

    void convertHSVToRGB(float hue, float s, float v, float &r, float &g, float &b);

    void calcColorGrey(float* color, size_t bucket);

    void calcColorHSV(float* color, size_t bucket);

    void calcColorJet(float* color, size_t bucket);

    void calcColorHot(float* color, size_t bucket);

    void calcColorSHSV(float* color, size_t bucket);

    void calcColorSimpsons(float* color, size_t bucket);

    size_t      m_numBuckets;
};

} // namespace lvr2

#endif /* COLORMAP_H_ */