added method to convert polyline in floatGrid (#4556)

* added method to convert polyline in floatGrid

* added variants for more voxels storage types

* update interface

Author: ABSitf

source/MRVoxels/MRPolylineToVoxels.cpp

@@ -0,0 +1,76 @@
+#include "MRPolylineToVoxels.h"
+#include "MRFloatGrid.h"
+#include "MRMesh/MRPolyline.h"
+#include "MRMesh/MRVector3.h"
+#include "MRMesh/MRProgressCallback.h"
+#include "MRMesh/MRTimer.h"
+#include "MRMesh/MRMesh.h"
+#include "MRMesh/MRMeshFillHole.h"
+#include "MRVDBConversions.h"
+#include "MRVoxelsVolume.h"
+#include "MRMeshToDistanceVolume.h"
+
+namespace MR
+{
+
+Mesh polylineToDegenerateMesh( const Polyline3& polyline )
+{
+    Mesh mesh;
+    auto contours = polyline.topology.convertToContours<Vector3f>(
+        [&points = polyline.points] ( VertId v )
+    {
+        return points[v];
+    } );
+
+    std::vector<EdgeId> newHoles;
+    newHoles.reserve( contours.size() );
+    for ( auto& cont : contours )
+    {
+        if ( cont[0] != cont.back() )
+            cont.insert( cont.end(), cont.rbegin(), cont.rend() );
+        newHoles.push_back( mesh.addSeparateEdgeLoop( cont ) );
+    }
+
+    for ( auto h : newHoles )
+        makeDegenerateBandAroundHole( mesh, h );
+
+    return mesh;
+}
+
+Expected<FloatGrid> polylineToDistanceField( const Polyline3& polyline, const PolylineToDistanceVolumeParams& params )
+{
+    MR_TIMER;
+
+    const Mesh mesh = polylineToDegenerateMesh( polyline );
+    return meshToDistanceField( mesh, {}, params.voxelSize, params.offsetCount, params.cb );
+}
+
+Expected<VdbVolume> polylineToVdbVolume( const Polyline3& polyline, const PolylineToDistanceVolumeParams& params )
+{
+    const Mesh mesh = polylineToDegenerateMesh( polyline );
+    MeshToVolumeParams meshParams;
+    meshParams.voxelSize = params.voxelSize;
+    meshParams.surfaceOffset = params.offsetCount;
+    meshParams.cb = params.cb;
+    return meshToDistanceVdbVolume( mesh, meshParams );
+}
+
+Expected<SimpleVolume> polylineToSimpleVolume( const Polyline3& polyline, const PolylineToVolumeParams& params )
+{
+    const Mesh mesh = polylineToDegenerateMesh( polyline );
+    MeshToDistanceVolumeParams meshParams;
+    meshParams.vol = params.vol;
+    meshParams.dist = { params.dist };
+    return meshToDistanceVolume( mesh, meshParams );
+}
+
+Expected<FunctionVolume> polylineToFunctionVolume( const Polyline3& polyline, const PolylineToVolumeParams& params )
+{
+    const Mesh mesh = polylineToDegenerateMesh( polyline );
+    MeshToDistanceVolumeParams meshParams;
+    meshParams.vol = params.vol;
+    meshParams.dist = { params.dist };
+    return meshToDistanceFunctionVolume( mesh, meshParams );
+}
+
+}

source/MRVoxels/MRPolylineToVoxels.h

@@ -0,0 +1,41 @@
+#pragma once
+#include "MRMesh/MRMeshFwd.h"
+#include "MRVoxelsFwd.h"
+#include "MRMesh/MRExpected.h"
+#include "MRVDBConversions.h"
+#include "MRDistanceVolumeParams.h"
+#include "MRMesh/MRDistanceToMeshOptions.h"
+
+namespace MR
+{
+
+struct PolylineToDistanceVolumeParams
+{
+    Vector3f voxelSize = Vector3f::diagonal( 1.f );
+    /// offsetCount - the number of voxels around polyline to calculate distance in (should be positive)
+    float offsetCount = 3;
+    ProgressCallback cb = {};
+};
+
+/// convert polyline to voxels distance field
+MRVOXELS_API Expected<FloatGrid> polylineToDistanceField( const Polyline3& polyline, const PolylineToDistanceVolumeParams& params );
+
+/// convert polyline to VDB volume
+/// \param offsetCount - the number of voxels around polyline to calculate distance in (should be positive)
+MRVOXELS_API Expected<VdbVolume> polylineToVdbVolume( const Polyline3& polyline, const PolylineToDistanceVolumeParams& params );
+
+/// Settings to conversion polyline to volume
+struct PolylineToVolumeParams
+{
+    DistanceVolumeParams vol;
+
+    DistanceToMeshOptions dist;
+};
+
+/// convert polyline to simple volume
+MRVOXELS_API Expected<SimpleVolume> polylineToSimpleVolume( const Polyline3& polyline, const PolylineToVolumeParams& params );
+
+/// convert polyline to function volume
+MRVOXELS_API Expected<FunctionVolume> polylineToFunctionVolume( const Polyline3& polyline, const PolylineToVolumeParams& params );
+
+}

source/MRVoxels/MRVoxels.vcxproj

@@ -34,6 +34,7 @@
     <ClCompile Include="MRPartialOffset.cpp" />
     <ClCompile Include="MRPointsToDistanceVolume.cpp" />
     <ClCompile Include="MRPointsToMeshFusion.cpp" />
+    <ClCompile Include="MRPolylineToVoxels.cpp" />
     <ClCompile Include="MRRebuildMesh.cpp" />
     <ClCompile Include="MRScalarConvert.cpp" />
     <ClCompile Include="MRScanHelpers.cpp" />
@@ -73,6 +74,7 @@
     <ClInclude Include="MRPartialOffset.h" />
     <ClInclude Include="MRPointsToDistanceVolume.h" />
     <ClInclude Include="MRPointsToMeshFusion.h" />
+    <ClInclude Include="MRPolylineToVoxels.h" />
     <ClInclude Include="MRRebuildMesh.h" />
     <ClInclude Include="MRScalarConvert.h" />
     <ClInclude Include="MRScanHelpers.h" />