VtkColorLookupTable.h

Go to the documentation of this file.

 
#pragma once
 
// ** INCLUDES **
#include <cassert>
#include <map>
#include <vector>
#include <vtkLookupTable.h>
 
#include "Applications/DataHolderLib/ColorLookupTable.h"
 
class VtkColorLookupTable : public vtkLookupTable
{
public:
    static const int DEFAULTMINVALUE = -9999;
    static const int DEFAULTMAXVALUE =  9999;
 
    static VtkColorLookupTable* New();
 
    vtkTypeMacro(VtkColorLookupTable,vtkLookupTable);
 
    void Build() override;
 
    /* \brief Sets the given colour as a constant in the colour lookup table.
     * The colour will subsequently be considered in the interpolation process when the lookup table is built. Note that pos is only a
     * relative position, i.e. pos in (0,1). The actual position of that colour in the table is dependent on the number of entries set
     * the SetRange() method.
     */
    void setColor(double pos, DataHolderLib::Color const& color);
 
    /* \brief Returns the colour at the given index from the colour lookup table.
     * The colour will be interpolated from the colour-dictionary entries before and after this index.
     * Make sure that Build() has been called before using this method.
     */
    void getColor(vtkIdType indx, unsigned char rgba[4]) const;
 
    DataHolderLib::LUTType getInterpolationType() const { return _type; }
 
    void setInterpolationType(DataHolderLib::LUTType type) { _type = type; }
 
    void setLookupTable(DataHolderLib::ColorLookupTable const& lut);
 
    void writeToFile(const std::string &filename);
 
    void SetTableValueRGBA(vtkIdType idx, unsigned char rgba[4]);
 
    void GetTableValue(vtkIdType idx, unsigned char rgba[4]);
 
protected:
    VtkColorLookupTable();
 
    ~VtkColorLookupTable() override;
 
private:
    unsigned char linInterpolation(unsigned char a, unsigned char b, double p) const;
 
    unsigned char expInterpolation(unsigned char a, unsigned char b, double gamma, double p) const;
 
    std::map<double, unsigned char*> _dict;
    DataHolderLib::LUTType _type{DataHolderLib::LUTType::LINEAR};
};