added support for bgeo

Author: losanc

mzd/__init__.py

@@ -1,4 +1,4 @@
 from .mzd import readMZD_to_bpymesh,readMZD_to_meshio
+from .bgeo import readbgeo_to_meshio
 
-
-__all__ = [readMZD_to_bpymesh,readMZD_to_meshio]
+__all__ = [readMZD_to_bpymesh,readMZD_to_meshio,readbgeo_to_meshio]

mzd/bgeo.py

@@ -0,0 +1,107 @@
+import gzip
+import numpy as np
+import meshio
+
+
+def readbgeo_to_meshio(filepath):
+    with gzip.open(filepath, 'r') as file:
+        byte = file.read(5)
+        if byte != b"BgeoV":
+            raise Exception('not bgeo file format')
+        byte = file.read(4)
+        version = int.from_bytes(byte, byteorder="big")
+        if version != 5:
+            raise Exception('bgeo file not version 5')
+
+        header = {}
+        point_attributes = {}
+        point_attributes_names = []
+        point_attributes_sizes = []
+        point_attributes_types = []
+        position = None
+
+        byte = file.read(4)
+        header['nPoints'] = int.from_bytes(byte, byteorder="big")
+
+        byte = file.read(4)
+        header['nPrims'] = int.from_bytes(byte, byteorder="big")
+
+        byte = file.read(4)
+        header['nPointGroups'] = int.from_bytes(byte, byteorder="big")
+
+        byte = file.read(4)
+        header['nPrimGroups'] = int.from_bytes(byte, byteorder="big")
+
+        byte = file.read(4)
+        header['nPointAttrib'] = int.from_bytes(byte, byteorder="big")
+
+        byte = file.read(4)
+        header['nVertexAttrib'] = int.from_bytes(byte, byteorder="big")
+
+        byte = file.read(4)
+        header['nPrimAttrib'] = int.from_bytes(byte, byteorder="big")
+
+        byte = file.read(4)
+        header['nAttrib'] = int.from_bytes(byte, byteorder="big")
+
+        particle_size = 4
+
+        for _ in range(header['nPointAttrib']):
+            byte = file.read(2)
+            namelength = int.from_bytes(byte, byteorder="big")
+            name_binary = file.read(namelength)
+            name = name_binary.decode('utf-8')
+            point_attributes_names.append(name)
+
+            byte = file.read(2)
+            size = int.from_bytes(byte, byteorder="big")
+            point_attributes_sizes.append(size)
+            particle_size += size
+
+            byte = file.read(4)
+            houdni_type = int.from_bytes(byte, byteorder="big")
+            if houdni_type == 0:
+                point_attributes_types.append('FLOAT')
+                # read default value
+                # not going to do anything about it
+                byte = file.read(size * 4)
+            elif houdni_type == 1:
+                point_attributes_types.append('INT')
+                # read default value
+                # not going to do anything about it
+                byte = file.read(size * 4)
+            elif houdni_type == 5:
+                point_attributes_types.append('VECTOR')
+                # read default value
+                # not going to do anything about it
+                byte = file.read(size * 4)
+            else:
+                raise Exception('houdni_type unknown/ unsupported')
+        byte = file.read(particle_size * header['nPoints'] * 4)
+        # > means big endian
+        attribute_data = np.frombuffer(byte, dtype='>f')
+        attribute_data = np.reshape(attribute_data, (header['nPoints'], particle_size))
+        # the first 3 column is its position data
+        position = attribute_data[:, :3]
+        # the 4th column is homogeneous coordiante, which is all 1, and will be ignored
+
+        current_attribute_start_point = 4
+        for i in range(header['nPointAttrib']):
+            if point_attributes_types[i] == 'FLOAT':
+                point_attributes[point_attributes_names[i]] = attribute_data[:, current_attribute_start_point]
+                current_attribute_start_point += 1
+            elif point_attributes_types[i] == 'VECTOR':
+                point_attributes[
+                    point_attributes_names[i]] = attribute_data[:,
+                                                                current_attribute_start_point:current_attribute_start_point + 3]
+                current_attribute_start_point += 3
+            elif point_attributes_types[i] == 'INT':
+                data = (attribute_data[:, current_attribute_start_point]).tobytes()
+                # > means big endian
+                point_attributes[point_attributes_names[i]] = np.frombuffer(data, dtype='>i')
+                current_attribute_start_point += 1
+        remaining = file.read()
+        if not remaining == b'\x00\xff':
+            raise Exception("file didn't end")
+        return meshio.Mesh(position, [('vertex', [])], point_data=point_attributes)
+

simloader/importer.py

@@ -104,7 +104,10 @@ def create_obj(fileseq, use_relaitve, transform_matrix=Matrix([[1, 0, 0, 0], [0,
     enabled = True
 
     try:
-        meshio_mesh = meshio.read(filepath)
+        if filepath.endswith('.bgeo'):
+            meshio_mesh = mzd.readbgeo_to_meshio(filepath)
+        else:
+            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),
@@ -174,7 +177,10 @@ def update_obj(scene, depsgraph=None):
         else:
             filepath = fs[current_frame % len(fs)]
             try:
-                meshio_mesh = meshio.read(filepath)
+                if filepath.endswith('.bgeo'):
+                    meshio_mesh = mzd.readbgeo_to_meshio(filepath)
+                else:
+                    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),