UPDATE: add number_joints and total_number_joints to DFBeam and DFAssembly resp.

Author: DamienGilliard

src/gh/components/DF_joint_segmentator/code.py

@@ -40,17 +40,13 @@ def RunScript(self,
 
         if not isinstance(i_clusters[0], Rhino.Geometry.PointCloud):
             raise ValueError("The input clusters must be PointClouds.")
-        
-        all_joints = i_assembly.all_joints
-        if not isinstance(all_joints[0].faces[0].to_mesh(), Rhino.Geometry.Mesh):
-            raise ValueError("The input joints must be convertible to Meshes.")
 
         # get number of joints
-        n_joints = max([joint.id for joint in all_joints]) + 1
+        n_joints = i_assembly.total_number_joints
 
         # prepping the reference meshes
         joints = [[] for _ in range(n_joints)]
-        for joint in all_joints:
+        for joint in i_assembly.all_joints:
             for face in joint.faces:
                 face = face.to_mesh()
                 face.Subdivide()

src/gh/diffCheck/diffCheck/df_geometries.py

@@ -313,6 +313,10 @@ def __repr__(self):
     def id(self):
         return self.__id
 
+    @property
+    def number_joints(self):
+        return max([joint.id for joint in self.joints]) + 1
+
     @property
     def joint_faces(self):
         return [face for face in self.faces if face.is_joint]
@@ -420,6 +424,10 @@ def dump_xml(self, pretty_xml: str, dir: str):
         with open(file_path, "w") as f:
             f.write(pretty_xml)
 
+    @property
+    def total_number_joints(self):
+        return max([joint.id for joint in self.all_joints]) + 1
+
     @property
     def all_joint_faces(self):
         if len(self._all_jointfaces) == 0: