MRTeethMaskToDirectionVolume.h

Go to the documentation of this file.

#pragma once
 
#include "MRMeshFwd.h"
#ifndef MRMESH_NO_OPENVDB
#include "MRExpected.h"
#include "MRVoxelsVolume.h"
#include "MRAffineXf3.h"
#include "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 );
    }
 
    MRMESH_API static std::optional<DentalId> fromFDI( int id );
 
    MRMESH_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:
    MRMESH_API static Expected<TeethMaskToDirectionVolumeConvertor> create( const VdbVolume& volume );
 
    MRMESH_API const HashMap<DentalId, Box3i>& getToothBounds() const;
 
    using DirectionVolume = std::array<SimpleVolume, 3>;
    struct ProcessResult
    {
        DirectionVolume volume;
        AffineXf3f xf;
    };
 
    MRMESH_API Expected<ProcessResult> convertTooth( DentalId id ) const;
 
    MRMESH_API Expected<ProcessResult> convertAll() const;
 
private:
    MRMESH_API TeethMaskToDirectionVolumeConvertor();
 
    HashMap<DentalId, Box3i> presentTeeth_;
    SimpleVolume mask_;
};
 
}
#endif