batch_processing generates hex-meshes using HexEx

on PolyCut outputs as well

Author: sebmestrallet

definitions/algorithms/batch_processing.py

@@ -61,9 +61,9 @@
 AUTOMATIC_POLYCUBE_OUTPUT_MISSING_POLICY = 'ask'
 EVOCUBE_OUTPUT_MISSING_POLICY            = 'ask'
 POLYCUT_OUTPUT_MISSING_POLICY            = 'pass' # on Windows only. selector for not only PolyCut itself, but the whole pipeline their provide (with hex-meshing & untangling)
-POLYCUBE_WITHHEXEX_OUTPUT_MISSING_POLICY = 'ask'
-GLOBAL_PADDING_OUTPUT_MISSING_POLICY     = 'ask'
-INNER_SMOOTHING_OUTPUT_MISSING_POLICY    = 'ask'
+POLYCUBE_WITHHEXEX_OUTPUT_MISSING_POLICY = 'run'
+GLOBAL_PADDING_OUTPUT_MISSING_POLICY     = 'run'
+INNER_SMOOTHING_OUTPUT_MISSING_POLICY    = 'run'
 
 RUNNING_ALGO_LINE_TEMPLATE            = "Running [green]{algo}[/] on [cyan]{path}[/]"
 EXISTING_OUTPUT_LINE_TEMPLATE         = "\[[bright_black]-[/]] [green]{algo}[/] on [cyan]{path}[/]" # type: ignore
@@ -77,6 +77,7 @@
 assert(AUTOMATIC_POLYCUBE_PATH is not None)
 AUTOMATIC_POLYCUBE_PATH = Path(expanduser(AUTOMATIC_POLYCUBE_PATH))
 
+TET_MESH_MEDIT,_ = translate_filename_keyword('TET_MESH_MEDIT')
 SURFACE_MESH_OBJ,_ = translate_filename_keyword('SURFACE_MESH_OBJ')
 SURFACE_MAP_TXT,_ = translate_filename_keyword('SURFACE_MAP_TXT')
 SURFACE_MESH_STATS_JSON,_ = translate_filename_keyword('SURFACE_MESH_STATS_JSON')
@@ -590,6 +591,19 @@ def process_step(step_object: DataFolder):
                     assert((coarser_tet_mesh_for_PolyCut.path / SURFACE_MESH_OBJ).exists())
                 # else: PolyCut failed to generate 'segmentation_XYZ.obj', we cannot extract a .txt labeling
 
+            # overwrite 'Gmsh_0.15' / `SURFACE_MAP_TXT`, which do not correspond to the converted, colorless .obj
+            # infer the surface map (association from surface triangles to tetrahedra facets) from the tetrahedral and triangle meshes
+            assert(AUTOMATIC_POLYCUBE_PATH is not None)
+            command_line = '{executable} {tet_mesh} {triangle_mesh} {output_surface_map}'.format(
+                executable = AUTOMATIC_POLYCUBE_PATH / 'extract_surface',
+                tet_mesh = coarser_tet_mesh_for_PolyCut.path / TET_MESH_MEDIT,
+                triangle_mesh = coarser_tet_mesh_for_PolyCut.path / SURFACE_MESH_OBJ,
+                output_surface_map = coarser_tet_mesh_for_PolyCut.path / SURFACE_MAP_TXT
+            )
+            with CONSOLE.status(RUNNING_ALGO_LINE_TEMPLATE.format(algo="extract_surface",path=collapseuser(coarser_tet_mesh_for_PolyCut.path))) as status:
+                subprocess.run(command_line, shell=True, capture_output=False)
+            CONSOLE.print(NEW_OUTPUT_LINE_TEMPLATE.format(algo='extract_surface', path=collapseuser(coarser_tet_mesh_for_PolyCut.path)))
+
             # sometimes there is no `HEX_MESH_MEDIT` in untangler/
             # (which was 'hex_highest_quality.mesh' before run_PolyCut_pipeline() renamed it)
             # -> if there is a 'untangler.hex_smooth.mesh', rename it `HEX_MESH_MEDIT`
@@ -601,6 +615,12 @@ def process_step(step_object: DataFolder):
                         coarser_tet_mesh_for_PolyCut.path / 'PolyCut_3' / 'optimizer_100' / HEX_MESH_MEDIT
                     )
 
+            # alongside the hex-mesh generated by the set of executables in the polycut demo,
+            # generate a hex-mesh with the same process as Evocube/automatic_polycube outputs: polycube_withHexEx, then global padding, then inner smoothing
+
+            if (coarser_tet_mesh_for_PolyCut.path / 'PolyCut_3' / SURFACE_LABELING_TXT).exists():
+                labeling_object: DataFolder = DataFolder(coarser_tet_mesh_for_PolyCut.path / 'PolyCut_3')
+                process_labeling(labeling_object)
 
 
 def main(input_folder: Path, arguments: list):