add support for partio; some minor change

Author: losanc

simloader/importer.py

@@ -153,26 +153,32 @@ def update_obj(scene, depsgraph=None):
                 show_message_box(traceback.format_exc(), "running script: " + obj.SIMLOADER.script_name + " failed: " + str(e),
                                  "ERROR")
                 continue
-        
+
         if 'process' in locals():
             user_process = locals()['process']
-            user_process(fs, current_frame, obj.data)
+            try:
+                user_process(fs, current_frame, obj.data)
+            except Exception as e:
+                show_message_box("Error when calling user process: " + traceback.format_exc(), icon="ERROR")
             continue
 
         elif 'preprocess' in locals():
             user_preprocess = locals()['preprocess']
-            meshio_mesh = user_preprocess(fs, current_frame)
+            try:
+                meshio_mesh = user_preprocess(fs, current_frame)
+            except Exception as e:
+                show_message_box("Error when calling user preprocess: " + traceback.format_exc(), icon="ERROR")
+                continue
         else:
             filepath = fs[current_frame % len(fs)]
             try:
                 meshio_mesh = meshio.read(filepath)
             except Exception as e:
                 show_message_box("Error when reading: " + filepath + ",\n" + traceback.format_exc(),
-                                "Meshio Loading Error" + str(e),
-                                icon="ERROR")
+                                 "Meshio Loading Error" + str(e),
+                                 icon="ERROR")
                 continue
 
-
         if not isinstance(meshio_mesh, meshio.Mesh):
             show_message_box('function preprocess does not return meshio object', "ERROR")
             continue

simloader/utils.py

@@ -13,6 +13,7 @@ def draw(self, context):
 
     print("Information: ", title)
     print(message)
+    print('End of simloader message box')
     print()
     stop_animation()
     bpy.context.window_manager.popup_menu(draw, title=title, icon=icon)

template/partio.py

@@ -0,0 +1,45 @@
+import meshio
+import fileseq
+import bpy
+import numpy as np
+
+
+def preprocess(fileseq: fileseq.FileSequence, frame_number: int) -> meshio.Mesh:
+    # type the path to the partio binary file
+    partio_pybind_path = ''
+    # here is an example
+    # partio_pybind_path = '/home/hui/Desktop/BlenderPartioTools/partio_extension_pybind/partio_pybind.cpython-39-x86_64-linux-gnu.so'
+
+    
+    # load partio_pybind module
+    import importlib
+    spec = importlib.util.spec_from_file_location('partio_pybind', partio_pybind_path)
+    partio_pybind = importlib.util.module_from_spec(spec)
+
+    # read particle data from file
+    frame_number = frame_number % len(fileseq)
+    file = fileseq[frame_number]
+    particle = partio_pybind.read(file)
+
+
+    # construct meshio.mesh
+    points = None
+    n_particles = particle.numParticles()
+    point_data = {}
+    for i in range(particle.numAttributes()):
+        attr = particle.attributeInfo(i)
+        attr_data = np.array(particle.data_buffer(attr), copy=True)
+        if attr.name == 'position':
+            points = attr_data
+        else:
+            point_data[attr.name] = attr_data
+    
+    
+    # release memory
+    particle.release()
+
+    return meshio.Mesh(
+        points,
+        # the cells is not important here
+        cells=[('vertex', [[x] for x in range(n_particles)])],
+        point_data=point_data)