Method applyCompensation

source/MRMesh/MRRadiusCompensation.cpp:173–255

}

Expected<void> RadiusCompensator::applyCompensation()
{
    MR_TIMER;
    MR_WRITER( mesh_ );

    Mesh cpyMesh = mesh_; // for projecting
    VertBitSet updatedVerts( planeVerts_.size() );

    int maxIters = std::max( 1, params_.maxIterations );

    auto sb = subprogress( params_.callback, 0.4f, 1.0f );
    auto maxAllowedShiftSq = sqr( std::min( mesh_.computeBoundingBox( params_.region ).diagonal() * 1e-2f, 2 * params_.toolRadius / float( maxIters ) ) );
    for ( int i = 1; i <= maxIters; ++i )
    {
        if ( i == maxIters )
            maxAllowedShiftSq = radiusSq_; // allow full move on last iteration

        float maxShiftSq = -FLT_MAX;
        updatedVerts.reset();
        for ( auto [cost, cId] : costs_ )
        {
            if ( cId < 0 )
                continue;
            auto planeToolCenter = toPlaneXf_( toolCenters_[cId] );

            findPointsInBall( *planeTree_, { .center = to3dim( to2dim( planeToolCenter ) ),.radiusSq = radiusSq_ },
                [&] ( const PointsProjectionResult & found, const Vector3f &, Ball3f & )
            {
                const auto v = found.vId;
                if ( !vertRegion_.test( v ) )
                    return Processing::Continue; // boundary vert
                auto planePoint = toPlaneXf_( mesh_.points[v] );
                auto shift = calcCompensationMovement_( planePoint, planeToolCenter );
                if ( shift == Vector3f() )
                    return Processing::Continue;

                auto shiftLenSq = shift.lengthSq();
                if ( shiftLenSq > maxShiftSq )
                    maxShiftSq = shiftLenSq;
                if ( shiftLenSq < maxAllowedShiftSq )
                {
                    mesh_.points[v] += shift;
                    if ( shiftLenSq < maxAllowedShiftSq * 1e-4f )
                        return Processing::Continue;
                }
                else
                {
                    mesh_.points[v] += shift.normalized() * maxAllowedShiftSq;
                }
                updatedVerts.set( v );
                return Processing::Continue;
            } );
        }

        if ( updatedVerts.none() )
            break; // fast return on finish

        expand( mesh_.topology, updatedVerts, params_.relaxExpansion );