MRColorMapAggregator.h

Go to the documentation of this file.

#pragma once
#include "MRMeshFwd.h"
#include "MRVector.h"
#include "MRColor.h"
#include "MRId.h"
#include "MRBitSet.h"
 
 
namespace MR
{
template<typename Tag>
class ColorMapAggregator
{
public:
    using ColorMap = Vector<Color, Id<Tag>>;
    using ElementBitSet = TaggedBitSet<Tag>;
 
    ColorMapAggregator() = default;
 
    MRMESH_API void setDefaultColor( const Color& color );
 
    struct PartialColorMap
    {
        ColorMap colorMap; // color map
        ElementBitSet elements; // bitset of elements for which the color map is applied
    };
 
    MRMESH_API void pushBack( const PartialColorMap& partitialColorMap );
 
    MRMESH_API void insert( int i, const PartialColorMap& partitialColorMap );
 
    MRMESH_API void replace( int i, const PartialColorMap& partitialColorMap );
 
    MRMESH_API void reset();
 
    size_t getColorMapNumber() { return dataSet_.size(); };
 
    const PartialColorMap& getPartialColorMap( int i ) { return dataSet_[i]; }
 
    MRMESH_API void erase( int i, int n = 1 );
 
    enum class AggregateMode
    {
        Overlay, 
        Blending 
    };
 
    MRMESH_API void setMode( AggregateMode mode );
 
    MRMESH_API ColorMap aggregate( const ElementBitSet& elementBitSet );
private:
    Color defaultColor_;
 
    std::vector<PartialColorMap> dataSet_;
 
    ColorMap aggregatedColorMap_;
    bool needUpdate_{ true };
    AggregateMode mode_{ AggregateMode::Overlay };
 
    bool checkInputData_( const PartialColorMap& partitialColorMap );
    void updateAggregated_( int newSize );
};
 
}