Function multiModelGridSampling

source/MRMesh/MRGridSampling.cpp:203–273

}

std::optional<std::vector<ObjVertId>> multiModelGridSampling( const Vector<ModelPointsData, ObjId>& models, float voxelSize, const ProgressCallback& cb )
{
    MR_TIMER;
    std::vector<ObjVertId> res;
    if ( voxelSize <= 0.f )
    {
        // return all input points as samples
        ObjId oId;
        auto sb = subprogress( cb, 0.1f, 0.8f );
        for ( const auto& model : models )
        {
            ++oId;
            if ( !model.points || !model.validPoints )
                continue;
            for ( auto v : *model.validPoints )
            {
                auto useOId = model.fakeObjId ? model.fakeObjId : oId;
                res.push_back( { useOId, v } );
            }
        }
        return res;
    }

    Box3f bbox;
    for ( const auto& model : models )
    {
        if ( !model.points || !model.validPoints )
            continue;
        bbox.include( computeBoundingBox( *model.points, *model.validPoints, model.xf ) );
    }
    const auto bboxSz = bbox.max - bbox.min;
    constexpr float maxVoxelsInOneDim = 1 << 10;
    const Vector3i dims
    {
        ( int )std::clamp( std::ceil( bboxSz.x / voxelSize ),1.0f, maxVoxelsInOneDim ),
        ( int )std::clamp( std::ceil( bboxSz.y / voxelSize ),1.0f, maxVoxelsInOneDim ),
        ( int )std::clamp( std::ceil( bboxSz.z / voxelSize ),1.0f, maxVoxelsInOneDim )
    };

    Grid grid( bbox, dims );
    if ( cb && !cb( 0.1f ) )
        return {};

    ObjId oId;
    auto sb = subprogress( cb, 0.1f, 0.8f );
    for ( const auto& model : models )
    {
        ++oId;
        if ( !model.points || !model.validPoints )
            continue;
        for ( auto v : *model.validPoints )
        {
            auto useOId = model.fakeObjId ? model.fakeObjId : oId;
            if ( model.xf )
                grid.addVertex( ( *model.xf )( ( *model.points )[v] ), v, useOId );
            else
                grid.addVertex( ( *model.points )[v], v, useOId );
        }