Function polyline2FromNP

source/mrmeshnumpy/MRPythonNumpyExposing.cpp:273–310

}

MR::Polyline2 polyline2FromNP( const pybind11::buffer& points )
{
    pybind11::buffer_info infoPoints = points.request();
    if ( infoPoints.ndim != 2 || infoPoints.shape[1] != 2 )
        throw std::runtime_error( "shape of input python vector 'points' should be (n,2)" );

    // verts to points part
    MR::Contour2f inputContour;

    auto stride0 = infoPoints.strides[0] / infoPoints.itemsize;
    auto stride1 = infoPoints.strides[1] / infoPoints.itemsize;
    inputContour.resize( infoPoints.shape[0] );
    auto fillPoints = [&] ( const auto* data )
    {
        for ( auto i = 0; i < infoPoints.shape[0]; i++ )
        {
            auto ind = stride0 * i;
            inputContour[i] = MR::Vector2f( float( data[ind] ), float( data[ind + stride1] ) );
        }
    };
    if ( infoPoints.format == pybind11::format_descriptor<double>::format() )
    {
        double* data = reinterpret_cast< double* >( infoPoints.ptr );
        fillPoints( data );
    }
    else if ( infoPoints.format == pybind11::format_descriptor<float>::format() )
    {
        float* data = reinterpret_cast< float* >( infoPoints.ptr );
        fillPoints( data );
    }
    else
        throw std::runtime_error( "dtype of input python vector should be float32 or float64" );

    MR::Polyline2 res;
    res.addFromPoints( inputContour.data(), inputContour.size() );

    return res;
}

// returns numpy array shapes [num faces,3] which represents vertices of mesh valid faces
pybind11::array_t<int> getNumpyFaces( const MR::MeshTopology& topology )