Function simpleVolumeFrom3Darray

source/mrmeshnumpy/MRPythonNumpyVoxels.cpp:11–50

using namespace MR;

MR::SimpleVolumeMinMax simpleVolumeFrom3Darray( const pybind11::buffer& voxelsArray )
{
    pybind11::buffer_info info = voxelsArray.request();
    if ( info.ndim != 3 )
        throw std::runtime_error( "shape of input python vector 'voxelsArray' should be (x,y,z)" );

    MR::SimpleVolumeMinMax res;
    res.dims = MR::Vector3i( int( info.shape[0] ), int( info.shape[1] ), int( info.shape[2] ) );
    size_t countPoints = size_t( res.dims.x ) * res.dims.y * res.dims.z;
    res.data.resize( countPoints );

    auto strideX = info.strides[0] / info.itemsize;
    auto strideY = info.strides[1] / info.itemsize;
    auto strideZ = info.strides[2] / info.itemsize;

    VolumeIndexer indexer( res.dims );
    if ( info.format == pybind11::format_descriptor<double>::format() )
    {
        double* data = reinterpret_cast< double* >( info.ptr );
        ParallelFor( 0_vox, indexer.endId(), [&] ( VoxelId i )
        {
            auto pos = indexer.toPos( i );
            res.data[i] = float( data[size_t( pos.x ) * strideX + size_t( pos.y ) * strideY + size_t( pos.z ) * strideZ] );
        } );
    }
    else if ( info.format == pybind11::format_descriptor<float>::format() )
    {
        float* data = reinterpret_cast< float* >( info.ptr );
        ParallelFor( 0_vox, indexer.endId(), [&] ( VoxelId i )
        {
            auto pos = indexer.toPos( i );
            res.data[i] = data[size_t( pos.x ) * strideX + size_t( pos.y ) * strideY + size_t( pos.z ) * strideZ];
        } );
    }
    else
        throw std::runtime_error( "dtype of input python vector should be float32 or float64" );

    std::tie( res.min, res.max ) = MR::parallelMinMax( res.data );
    return res;
}

pybind11::array_t<double> getNumpy3Darray( const MR::SimpleVolume& simpleVolume )
{