mirror of
				https://github.com/SoftFever/OrcaSlicer.git
				synced 2025-10-31 04:31:15 -06:00 
			
		
		
		
	
		
			
				
	
	
		
			136 lines
		
	
	
	
		
			4.8 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			136 lines
		
	
	
	
		
			4.8 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| #define NOMINMAX
 | |
| #include "OpenVDBUtils.hpp"
 | |
| 
 | |
| #ifdef _MSC_VER
 | |
| // Suppress warning C4146 in OpenVDB: unary minus operator applied to unsigned type, result still unsigned 
 | |
| #pragma warning(push)
 | |
| #pragma warning(disable : 4146)
 | |
| #endif // _MSC_VER
 | |
| #include <openvdb/tools/MeshToVolume.h>
 | |
| #ifdef _MSC_VER
 | |
| #pragma warning(pop)
 | |
| #endif // _MSC_VER
 | |
| 
 | |
| #include <openvdb/tools/VolumeToMesh.h>
 | |
| #include <openvdb/tools/Composite.h>
 | |
| #include <openvdb/tools/LevelSetRebuild.h>
 | |
| 
 | |
| //#include "MTUtils.hpp"
 | |
| 
 | |
| namespace Slic3r {
 | |
| 
 | |
| class TriangleMeshDataAdapter {
 | |
| public:
 | |
|     const indexed_triangle_set &its;
 | |
|     float voxel_scale;
 | |
| 
 | |
|     size_t polygonCount() const { return its.indices.size(); }
 | |
|     size_t pointCount() const   { return its.vertices.size(); }
 | |
|     size_t vertexCount(size_t) const { return 3; }
 | |
| 
 | |
|     // Return position pos in local grid index space for polygon n and vertex v
 | |
|     // The actual mesh will appear to openvdb as scaled uniformly by voxel_size
 | |
|     // And the voxel count per unit volume can be affected this way.
 | |
|     void getIndexSpacePoint(size_t n, size_t v, openvdb::Vec3d& pos) const
 | |
|     {
 | |
|         auto vidx = size_t(its.indices[n](Eigen::Index(v)));
 | |
|         Slic3r::Vec3d p = its.vertices[vidx].cast<double>() * voxel_scale;
 | |
|         pos = {p.x(), p.y(), p.z()};
 | |
|     }
 | |
| 
 | |
|     TriangleMeshDataAdapter(const indexed_triangle_set &m, float voxel_sc = 1.f)
 | |
|         : its{m}, voxel_scale{voxel_sc} {};
 | |
| };
 | |
| 
 | |
| // TODO: Do I need to call initialize? Seems to work without it as well but the
 | |
| // docs say it should be called ones. It does a mutex lock-unlock sequence all
 | |
| // even if was called previously.
 | |
| openvdb::FloatGrid::Ptr mesh_to_grid(const indexed_triangle_set &    mesh,
 | |
|                                      const openvdb::math::Transform &tr,
 | |
|                                      float voxel_scale,
 | |
|                                      float exteriorBandWidth,
 | |
|                                      float interiorBandWidth,
 | |
|                                      int   flags)
 | |
| {
 | |
|     openvdb::initialize();
 | |
| 
 | |
|     std::vector<indexed_triangle_set> meshparts = its_split(mesh);
 | |
| 
 | |
|     auto it = std::remove_if(meshparts.begin(), meshparts.end(),
 | |
|                              [](auto &m) { return its_volume(m) < EPSILON; });
 | |
| 
 | |
|     meshparts.erase(it, meshparts.end());
 | |
| 
 | |
|     openvdb::FloatGrid::Ptr grid;
 | |
|     for (auto &m : meshparts) {
 | |
|         auto subgrid = openvdb::tools::meshToVolume<openvdb::FloatGrid>(
 | |
|             TriangleMeshDataAdapter{m, voxel_scale}, tr, exteriorBandWidth,
 | |
|             interiorBandWidth, flags);
 | |
| 
 | |
|         if (grid && subgrid) openvdb::tools::csgUnion(*grid, *subgrid);
 | |
|         else if (subgrid) grid = std::move(subgrid);
 | |
|     }
 | |
| 
 | |
|     if (grid) {
 | |
|         grid = openvdb::tools::levelSetRebuild(*grid, 0., exteriorBandWidth,
 | |
|                                                interiorBandWidth);
 | |
|     } else if(meshparts.empty()) {
 | |
|         // Splitting failed, fall back to hollow the original mesh
 | |
|         grid = openvdb::tools::meshToVolume<openvdb::FloatGrid>(
 | |
|             TriangleMeshDataAdapter{mesh}, tr, exteriorBandWidth,
 | |
|             interiorBandWidth, flags);
 | |
|     }
 | |
| 
 | |
|     grid->insertMeta("voxel_scale", openvdb::FloatMetadata(voxel_scale));
 | |
| 
 | |
|     return grid;
 | |
| }
 | |
| 
 | |
| indexed_triangle_set grid_to_mesh(const openvdb::FloatGrid &grid,
 | |
|                           double                    isovalue,
 | |
|                           double                    adaptivity,
 | |
|                           bool                      relaxDisorientedTriangles)
 | |
| {
 | |
|     openvdb::initialize();
 | |
| 
 | |
|     std::vector<openvdb::Vec3s> points;
 | |
|     std::vector<openvdb::Vec3I> triangles;
 | |
|     std::vector<openvdb::Vec4I> quads;
 | |
| 
 | |
|     openvdb::tools::volumeToMesh(grid, points, triangles, quads, isovalue,
 | |
|                                  adaptivity, relaxDisorientedTriangles);
 | |
| 
 | |
|     float scale = 1.;
 | |
|     try {
 | |
|         scale = grid.template metaValue<float>("voxel_scale");
 | |
|     }  catch (...) { }
 | |
| 
 | |
|     indexed_triangle_set ret;
 | |
|     ret.vertices.reserve(points.size());
 | |
|     ret.indices.reserve(triangles.size() + quads.size() * 2);
 | |
| 
 | |
|     for (auto &v : points) ret.vertices.emplace_back(to_vec3f(v) / scale);
 | |
|     for (auto &v : triangles) ret.indices.emplace_back(to_vec3i(v));
 | |
|     for (auto &quad : quads) {
 | |
|         ret.indices.emplace_back(quad(0), quad(1), quad(2));
 | |
|         ret.indices.emplace_back(quad(2), quad(3), quad(0));
 | |
|     }
 | |
| 
 | |
|     return ret;
 | |
| }
 | |
| 
 | |
| openvdb::FloatGrid::Ptr redistance_grid(const openvdb::FloatGrid &grid,
 | |
|                                         double                    iso,
 | |
|                                         double                    er,
 | |
|                                         double                    ir)
 | |
| {
 | |
|     auto new_grid = openvdb::tools::levelSetRebuild(grid, float(iso),
 | |
|                                                     float(er), float(ir));
 | |
| 
 | |
|     // Copies voxel_scale metadata, if it exists.
 | |
|     new_grid->insertMeta(*grid.deepCopyMeta());
 | |
| 
 | |
|     return new_grid;
 | |
| }
 | |
| 
 | |
| } // namespace Slic3r
 | 
