MRVoxelsVolumeAccess.h

Go to the documentation of this file.

#pragma once
 
#include "MRVoxelsFwd.h"
#include "MRVoxelsVolume.h"
#include "MRVDBFloatGrid.h"
#include "MRMesh/MRVolumeIndexer.h"
#include "MRMesh/MRIsNaN.h"
 
namespace MR
{
 
template <typename Volume>
class VoxelsVolumeAccessor;
 
template <>
class VoxelsVolumeAccessor<VdbVolume>
{
public:
    using VolumeType = VdbVolume;
    using ValueType = typename VolumeType::ValueType;
    static constexpr bool cacheEffective = true; 
 
    explicit VoxelsVolumeAccessor( const VolumeType& volume )
        : accessor_( volume.data->getConstAccessor() )
        , minCoord_( fromVdb( volume.data->evalActiveVoxelBoundingBox().min() ) )
    {}
 
    ValueType get( const Vector3i& pos ) const
    {
        ValueType res;
        if ( !accessor_.probeValue( toVdb( pos + minCoord_ ), res ) )
            return cQuietNan;
        return res;
    }
 
    ValueType get( const VoxelLocation & loc ) const
    {
        return get( loc.pos );
    }
 
    const Vector3i& minCoord() const { return minCoord_; }
 
    Vector3f shift() const { return Vector3f( minCoord_ ); }
 
private:
    openvdb::FloatGrid::ConstAccessor accessor_;
    Vector3i minCoord_;
};
 
template <typename T>
class VoxelsVolumeAccessor<VoxelsVolume<std::vector<T>>>
{
public:
    using VolumeType = VoxelsVolume<std::vector<T>>;
    using ValueType = typename VolumeType::ValueType;
    static constexpr bool cacheEffective = false; 
 
    explicit VoxelsVolumeAccessor( const VolumeType& volume )
        : data_( volume.data )
          , indexer_( volume.dims )
    {}
 
    ValueType get( const Vector3i& pos ) const
    {
        return data_[indexer_.toVoxelId( pos )];
    }
 
    ValueType get( const VoxelLocation & loc ) const
    {
        return data_[loc.id];
    }
 
    Vector3f shift() const { return Vector3f::diagonal( 0.5f ); }
 
private:
    const std::vector<T>& data_;
    VolumeIndexer indexer_;
};
 
template <typename T>
class VoxelsVolumeAccessor<VoxelsVolumeMinMax<std::vector<T>>> : public VoxelsVolumeAccessor<VoxelsVolume<std::vector<T>>>
{
    using Base = VoxelsVolumeAccessor<VoxelsVolume<std::vector<T>>>;
public:
    using VolumeType = VoxelsVolumeMinMax<std::vector<T>>;
    using ValueType = typename VolumeType::ValueType;
    using Base::cacheEffective;
    using Base::Base;
};
 
template <typename T>
class VoxelsVolumeAccessor<VoxelsVolume<VoxelValueGetter<T>>>
{
public:
    using VolumeType = VoxelsVolume<VoxelValueGetter<T>>;
    using ValueType = typename VolumeType::ValueType;
    static constexpr bool cacheEffective = true; 
 
    explicit VoxelsVolumeAccessor( const VolumeType& volume )
        : data_( volume.data )
    {}
 
    ValueType get( const Vector3i& pos ) const
    {
        return data_( pos );
    }
 
    ValueType get( const VoxelLocation & loc ) const
    {
        return get( loc.pos );
    }
 
    Vector3f shift() const { return Vector3f::diagonal( 0.5f ); }
 
private:
    const VoxelValueGetter<T>& data_;
};
 
} // namespace MR