add script to visualize CSG mesh with plotly

README.md

@@ -33,3 +33,12 @@ export TMP=/tmp/myname
 export GEOM=mygeom
 export OPTICKS_EVENT_MODE=DebugLite
 ```
+
+
+## Visualization
+
+Plot any volume serialized 
+
+```
+scripts/plot-csg.py ../out/csg/CSGFoundry/SSim/scene/meshmerge/0/
+```

plot_csg_tree.py → g4ox/csg.py

@@ -1,20 +1,8 @@
 import numpy as np
 import plotly.graph_objects as go
 
-from opticks.CSG.CSGFoundry import CSGFoundry
-
-mesh = None
-
-def set_csg(path):
-    global mesh
-    cf = CSGFoundry.Load(path)
-    mesh = cf.sim.stree.mesh
-
-
-def create_mesh(mesh_subdir):
-    vtx = mesh_subdir.vtx
-    tri = mesh_subdir.tri
 
+def mesh(tri, vtx):
     # close triangles and create edge coordinates
     tri_close = np.hstack((tri, tri[:,0][:,np.newaxis]))
     tri_sides = np.array([[None]*3 if i is None else vtx[i].tolist() for t in tri_close for i in list(t)+[None]], dtype=float)
@@ -23,14 +11,3 @@ def create_mesh(mesh_subdir):
     edges = go.Scatter3d(x=tri_sides.T[0], y=tri_sides.T[1], z=tri_sides.T[2], mode='lines', line_width=5)
 
     return faces, edges
-
-
-def fig_show(data):
-    fig = go.Figure(data=data)
-    fig.show()
-    return fig
-
-
-def fig_add(fig, data):
-    fig.add_traces(data)
-    return fig

pyproject.toml

@@ -0,0 +1,19 @@
+[tool.poetry]
+name = "g4ox"
+version = "0.0.0"
+description = ""
+authors = ["BNL NPPS"]
+readme = "README.md"
+packages = [{include = "g4ox"}]
+
+[tool.poetry.dependencies]
+python = "^3.10"
+numpy = "^2.1"
+plotly = "^5.22.0"
+nbformat = "^5.10.4"
+flask = "^3.1.0"
+flask-autoindex = "^0.6.6"
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

scripts/plot-csg.py

@@ -0,0 +1,40 @@
+#!/usr/bin/env python3
+
+import argparse
+import logging
+import pathlib as pl
+import plotly.graph_objects as go
+import numpy as np
+
+from g4ox import csg
+
+logging.basicConfig(level=logging.INFO)
+
+
+def read_mesh(path_csg, path_out):
+    tri = np.load(path_csg/'tri.npy')
+    vtx = np.load(path_csg/'vtx.npy')
+
+    faces, edges = csg.mesh(tri, vtx)
+
+    logging.info(path_csg.name)
+
+    fig = go.Figure(data=[faces, edges])
+    html_file = path_out / (str(path_csg).replace("/", "_").replace("\\", "_").replace(".", "") + '.html')
+    fig.write_html(html_file, include_plotlyjs='cdn')
+
+    logging.info(f'Saved image to: {html_file}')
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+    parser.add_argument('path', help="A path to CSG dir")
+    parser.add_argument('-o', '--outdir', default='/home/web/g4ox', help="A path to output dir")
+
+    args = parser.parse_args()
+
+    path_csg = pl.Path(args.path)
+    path_out = pl.Path(args.outdir)
+    logging.info(f'Reading data from dir: {path_csg}')
+
+    read_mesh(path_csg, path_out)

scripts/serve-path.py

@@ -0,0 +1,20 @@
+#!/usr/bin/env python3
+
+import argparse
+
+from flask import Flask
+from flask_autoindex import AutoIndex
+
+app = Flask(__name__)
+
+def run_autoindex(path):
+    AutoIndex(app, browse_root=path)
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+    parser.add_argument("path", nargs='?', default='iframe_figures/', help="A path to serve")
+
+    args = parser.parse_args()
+
+    run_autoindex(args.path)
+    app.run(host='0.0.0.0', port=8080)