MRMarkedVoxelSlice.h

Go to the documentation of this file.

#pragma once
 
#include "MRViewerFwd.h"
#ifndef MRVIEWER_NO_VOXELS
 
#include "MRImGuiImage.h"
#include <MRMesh/MRColor.h>
#include <MRMesh/MRVector3.h>
#include <MRMesh/MRBitSet.h>
#include <MRMesh/MRBox.h>
#include <MRVoxels/MRFloatGrid.h>
#include <MRVoxels/MRVoxelPath.h>
 
namespace MR
{
 
 
class MarkedVoxelSlice : public ImGuiImage
{
public:
    struct Mark
    {
        Color color;
        VoxelBitSet mask;
    };
 
    MRVIEWER_API MarkedVoxelSlice( const ObjectVoxels& voxels );
    
    enum MaskType { Inside, Outside, Segment, Count };
 
    VoxelBitSet& getMask( MaskType type ) { return params_.marks[type].mask; }
    const VoxelBitSet& getMask( MaskType type ) const { return params_.marks[type].mask; }
    void setMask( const VoxelBitSet& mask, MaskType type ) { params_.marks[type].mask = mask; forceUpdate(); }
    
    const Color& getColor( MaskType type ) const { return params_.marks[type].color; }
    void setColor( const Color& color, MaskType type ) { params_.marks[type].color = color; forceUpdate(); }
 
 
    Mark& getMark( MaskType type ) { return params_.marks[type]; }
    const Mark& getMark( MaskType type ) const { return params_.marks[type]; }
    void setMark( const Mark& mark, MaskType type ) { params_.marks[type] = mark; forceUpdate(); }
 
    std::vector<Mark>& getCustomBackgroundMarks() { return params_.customBackgroundMarks; }
    const std::vector<Mark>& getCustomBackgroundMarks() const { return params_.customBackgroundMarks; }
    void setCustomBackgroundMarks( const std::vector<Mark>& backgroundMarks ) { params_.customBackgroundMarks = backgroundMarks; forceUpdate(); }
 
    std::vector<Mark>& getCustomForegroundMarks() { return params_.customForegroundMarks; }    
    const std::vector<Mark>& getCustomForegroundMarks() const { return params_.customForegroundMarks; }
    void setCustomForegroundMarks( const std::vector<Mark>& foregroundMarks ) { params_.customForegroundMarks = foregroundMarks; forceUpdate(); }
 
 
    SlicePlane getActivePlane() const { return params_.activePlane; }
    void setActivePlane( SlicePlane plane ) { params_.activePlane = plane; forceUpdate(); }
 
    const Vector3i& getActiveVoxel() const { return params_.activeVoxel; }
    void setActiveVoxel( const Vector3i& voxel ) { params_.activeVoxel = voxel; forceUpdate(); }
 
    float getMin() const { return params_.min; }
    void setMin( float min ) { params_.min = min; forceUpdate(); }
    float getMax() const { return params_.max; }
    void setMax( float max ) { params_.max = max; forceUpdate(); }
 
    const Box3i& getActiveBox() const { return params_.activeBox; }
    void setActiveBox( const Box3i& box ) { params_.activeBox = box; forceUpdate(); }
 
    struct Parameters
    {
        std::array<Mark, size_t( MaskType::Count )> marks = {Mark{Color::red()},Mark{Color::blue()},Mark{Color::yellow()}};
        std::vector<Mark> customBackgroundMarks;
        std::vector<Mark> customForegroundMarks;
        Vector3i activeVoxel;
        Box3i activeBox;
        float min{0.0f};
        float max{0.0f};
        SlicePlane activePlane{XY};
        std::optional<Color> inactiveVoxelColor;
    };
 
    const Parameters& getParameters() const { return params_; }
    void setParameters( const Parameters& params ) { params_ = params; forceUpdate(); }
 
    MRVIEWER_API void forceUpdate();
 
private:
    FloatGrid grid_;
    Vector3i dims_;
 
    Parameters params_;
 
};
 
} 
 
#endif