MRVolumeSegment.h

Go to the documentation of this file.

#pragma once
#include "MRVoxelsFwd.h"
 
#include "MRMesh/MRVector3.h"
#include "MRMesh/MRExpected.h"
#include "MRVoxelsVolume.h"
#include "MRMesh/MRBitSet.h"
#include "MRMesh/MRProgressCallback.h"
#include <string>
 
namespace MR
{
 
MRVOXELS_API Expected<MR::Mesh> meshFromVoxelsMask( const VdbVolume& volume, const VoxelBitSet& mask );
 
struct VolumeSegmentationParameters
{
    float buildPathExponentModifier{-1.0f};
    float segmentationExponentModifier{3000.0f};
    int voxelsExpansion{25};
};
 
MRVOXELS_API Expected<MR::Mesh> segmentVolume( const VdbVolume& volume, const std::vector<std::pair<Vector3f, Vector3f>>& pairs,
                                                              const VolumeSegmentationParameters& params = VolumeSegmentationParameters() );
 
struct VoxelMetricParameters;
 
class VolumeSegmenter
{
public:
    MRVOXELS_API VolumeSegmenter( const VdbVolume& volume );
 
    enum SeedType
    {
        Inside,
        Outside,
        Count
    };
 
    MRVOXELS_API void addPathSeeds( const VoxelMetricParameters& metricParameters, SeedType seedType, float exponentModifier = -1.0f );
    
    MRVOXELS_API void setSeeds( const std::vector<Vector3i>& seeds, SeedType seedType );
    
    MRVOXELS_API void addSeeds( const std::vector<Vector3i>& seeds, SeedType seedType );
 
    MRVOXELS_API const std::vector<Vector3i>& getSeeds( SeedType seedType ) const;
 
    MRVOXELS_API Expected<VoxelBitSet> segmentVolume( float segmentationExponentModifier = 3000.0f, int voxelsExpansion = 25, ProgressCallback cb = {} );
    
    MRVOXELS_API Expected<MR::Mesh> createMeshFromSegmentation( const VoxelBitSet& segmentation ) const;
 
    MRVOXELS_API const Vector3i& getVolumePartDimensions() const;
 
    MRVOXELS_API const Vector3i& getMinVoxel() const;
private:
    const VdbVolume& volume_;
 
    SimpleVolumeMinMax volumePart_;
 
    Vector3i minVoxel_;
    Vector3i maxVoxel_;
 
    std::array<std::vector<Vector3i>, size_t( SeedType::Count )> seeds_;
    std::array<VoxelBitSet, size_t( SeedType::Count )> seedsInVolumePartSpace_;
 
    bool seedsChanged_{true};
 
    void setupVolumePart_( int voxelsExpansion );
};
 
MRVOXELS_API Expected<std::vector<Mesh>> segmentVoxelMaskToInstances( const VdbVolume& mask, size_t minSize = 100, ProgressCallback cb = {} );
 
 
}