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; 
        ElementBitSet elements; 
    };

    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 );
};
 
}