ResInsight/Fwk/AppFwk/cafVizExtensions/cafCategoryMapper.h

60 lines
2.1 KiB
C++

#pragma once
#include "cvfBase.h"
#include "cvfObject.h"
#include "cvfScalarMapper.h"
#include "cvfString.h"
namespace caf {
//==================================================================================================
//
//
//==================================================================================================
class CategoryMapper : public cvf::ScalarMapper
{
public:
CategoryMapper();
void setCategories(const std::vector<int>& categoryValues);
void setCategoriesWithNames(const std::vector<int>& categoryValues, const std::vector<cvf::String>& categoryNames);
// Colors in color array are cycled, if category count is larger than color count, colors are reused
void setCycleColors(const cvf::Color3ubArray& colorArray);
// Colors are interpolated to make sure all categories get a unique color
void setInterpolateColors(const cvf::Color3ubArray& colorArray);
// Overrides used from legend
virtual cvf::Vec2f mapToTextureCoord(double scalarValue) const;
virtual bool updateTexture(cvf::TextureImage* image) const;
virtual cvf::Color3ub mapToColor(double normalizedValue) const;
virtual void majorTickValues(std::vector<double>* domainValues) const;
virtual double normalizedValue(double domainValue) const;
virtual double domainValue(double normalizedValue) const;
private:
friend class CategoryLegend;
size_t categoryCount() const;
const cvf::String textForCategoryIndex(size_t index) const;
int categoryIndexForCategory(double domainValue) const;
void recomputeMaxTexCoord();
private:
cvf::Color3ubArray m_colors;
cvf::uint m_textureSize; // The size of texture that updateTexture() is will produce.
double m_maxTexCoord; // The largest allowable s texture coordinate, scalar values >= m_rangeMax will get mapped to this coordinate
std::vector<int> m_categoryValues;
std::vector<cvf::String> m_categoryNames;
};
}