imstkColorFunction.h

/*
** This file is part of the Interactive Medical Simulation Toolkit (iMSTK)
** iMSTK is distributed under the Apache License, Version 2.0.
** See accompanying NOTICE for details.
*/

#pragma once

#include "imstkColor.h"
#include "imstkMath.h"

#include <vector>

namespace imstk
{
class ColorFunction
{
public:
    enum class ColorSpace
    {
        RGB,
        DIVERING,
        HSV,
        LAB
    };

public:
    ColorFunction() = default;
    virtual ~ColorFunction() = default;

public:
    int getNumberOfColors() const { return static_cast<int>(m_table.size()); }
    const Color& getColor(int i) const { return m_table[i]; }
    const Vec2d& getRange() const { return m_range; }
    const ColorSpace& getColorSpace() const { return m_colorSpace; }

    void setNumberOfColors(int numColors) { m_table.resize(numColors); }
    void setColor(int i, const Color& color) { m_table[i] = color; }
    void setRange(double min, double max) { setRange(Vec2d(min, max)); }
    void setRange(const Vec2d& range) { this->m_range = range; }
    void setColorSpace(const ColorSpace& space) { this->m_colorSpace = space; }

    void clear() { m_table.clear(); }

protected:
    std::vector<Color> m_table;
    Vec2d      m_range;
    ColorSpace m_colorSpace;
};
} // namespace imstk