MRTeethMaskToDirectionVolume.h

Go to the documentation of this file.

#pragma once
 
#include "MRVoxelsFwd.h"
 
#include "MRMesh/MRExpected.h"
#include "MRVoxelsVolume.h"
#include "MRMesh/MRAffineXf3.h"
#include "MRMesh/MRphmap.h"
 
#include <array>
#include <optional>
 
namespace MR
{
 
class DentalId
{
public:
    template <int id>
    static constexpr DentalId fromFDI()
    {
        const int t = id % 10;
        const int q = id / 10;
        static_assert( q >= 1 && q <= 4 && t >= 1 && t <= 8 );
 
        return DentalId( id );
    }
 
    MRVOXELS_API static std::optional<DentalId> fromFDI( int id );
 
    MRVOXELS_API int fdi() const;
 
    auto operator <=> ( const DentalId& other ) const = default;
 
private:
    constexpr explicit DentalId( int fdi ):
        fdi_( fdi )
    {}
 
    int fdi_;
};
 
}
 
template <>
struct std::hash<MR::DentalId>
{
    inline size_t operator() ( const MR::DentalId& id ) const noexcept
    {
        return hash<int>{}( id.fdi() );
    }
};
 
namespace MR
{
 
class TeethMaskToDirectionVolumeConvertor
{
public:
    MRVOXELS_API static Expected<TeethMaskToDirectionVolumeConvertor> create( const VdbVolume& volume, const std::vector<int>& additionalIds = {} );
 
    MRVOXELS_API const HashMap<int, Box3i>& getObjectBounds() const;
 
    using DirectionVolume = std::array<SimpleVolumeMinMax, 3>;
    struct ProcessResult
    {
        DirectionVolume volume;
        AffineXf3f xf;
    };
 
    MRVOXELS_API Expected<ProcessResult> convertObject( int id ) const;
 
    MRVOXELS_API Expected<ProcessResult> convertAll() const;
 
private:
    MRVOXELS_API TeethMaskToDirectionVolumeConvertor();
 
    HashMap<int, Box3i> presentObjects_;
    SimpleVolume mask_;
};
 
 
MRVOXELS_API Expected<std::array<SimpleVolumeMinMax, 3>> teethMaskToDirectionVolume( const VdbVolume& volume, const std::vector<int>& additionalIds = {} );
 
 
}