add prim output

zenustech · Dec 18, 2023 · 2df27a2 · 2df27a2
1 parent 34f7625
commit 2df27a2
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/projects/Geometry/fTetWild b/projects/Geometry/fTetWild
diff --git a/projects/Geometry/tetmesh.cpp b/projects/Geometry/tetmesh.cpp
@@ -16,6 +16,7 @@ struct FTetWild : INode {
     virtual void apply() override {
         auto prim = get_input<PrimitiveObject>("prim");
         auto input_dir = get_input2<std::string>("input_dir");
+        input_dir.erase(input_dir.find_last_not_of(" ")+1);
         auto output_dir = get_input2<std::string>("output_dir");
         auto tag = get_input2<std::string>("tag");
         tag.erase(tag.find_last_not_of(" ")+1);
@@ -208,13 +209,25 @@ struct FTetWild : INode {
                 verts[i].push_back(prim->verts[i][j]);
 
         runFTetWild(faces, verts, argc, argv);
+
+        prim->quads.clear();
+        prim->quads.resize(faces.size());
+        prim->verts.clear();
+        prim->verts.resize(verts.size());
+        for (int i = 0; i < faces.size(); ++i)
+            for (int j = 0; j < 4; ++j)
+                prim->quads[i][j] = faces[i][j];
+        for (int i = 0; i < verts.size(); ++i)
+            for (int j = 0; j < 3; ++j)
+                prim->verts[i][j] = verts[i][j];
+        set_output("prim", std::move(prim));
     }
 };
 
 ZENO_DEFNODE(FTetWild)
 ({
     {{"prim"},
-    {"readpath", "input_dir"},
+    {"readpath", "input_dir", " "},
     {"string", "output_dir"},
     {"readpath", "tag", " "},
     {"enum none union intersection difference", "operation", "none"},
@@ -234,7 +247,7 @@ ZENO_DEFNODE(FTetWild)
     {"bool", "use_general_wn", "0"},
     {"bool", "use_input_for_wn", "0"},
     {"readpath", "bg_mesh", " "}},
-    {},
+    {("prim")},
     {},
     {"primitive"},
 });
+18 −13		api.cpp
+52 −0		src/MeshIO.cpp
+1 −1		src/MeshIO.hpp
+2 −1		src/MeshImprovement.cpp