add rotation and scale for blade iges faces

Author: ndem0

bladex/blade.py

@@ -429,6 +429,39 @@ def rotate(self, deg_angle=None, rad_angle=None, axis='x'):
         self.blade_coordinates_down = np.einsum('ij, kjl->kil',
             rot_matrix, self.blade_coordinates_down)
 
+        # TODO: working but ugly
+        for id, face in enumerate([self.upper_face, self.lower_face,
+                     self.tip_face, self.root_face]):
+            from OCC.Core.BRepBuilderAPI import BRepBuilderAPI_Transform
+            from OCC.Core.gp import gp_Pnt
+            from OCC.Core.gp import gp_Dir
+            from OCC.Core.gp import gp_Ax1
+            from OCC.Core.gp import gp_Trsf
+            
+            origin = gp_Pnt(0, 0, 0)
+            if axis == 'y':
+                direction = gp_Dir(0, 1, 0)
+            elif axis == 'z':
+                direction = gp_Dir(0, 0, 1)
+            elif axis == 'x':
+                direction = gp_Dir(1, 0, 0)
+            else:
+                raise ValueError('Axis must be either x, y, or z.')
+            ax1 = gp_Ax1(origin, direction)
+            trsf = gp_Trsf()
+            trsf.SetRotation(ax1, rad_angle)
+
+            brep_tr = BRepBuilderAPI_Transform(face, trsf, True, True)
+            face = brep_tr.Shape()
+            if id == 0:
+                self.upper_face = face
+            elif id == 1:
+                self.lower_face = face
+            elif id == 2:
+                self.tip_face = face
+            elif id == 3:
+                self.root_face = face
+
     def scale(self, factor):
         """
         Scale the blade coordinates by a specified factor.
@@ -438,11 +471,29 @@ def scale(self, factor):
         from OCC.Core.BRepBuilderAPI import BRepBuilderAPI_Transform
         self.blade_coordinates_up *= factor
         self.blade_coordinates_down *= factor
-        # for face in [self.upper_face, self.lower_face, self.tip_face, self.root_face]:
-        #     brepgp_Trsf = BRepBuilderAPI_Transform()
-        #     brepgp_Trsf.SetScale(gp_Pnt(0, 0, 0), factor)
-        #     brepgp_Trsf.Perform(face, True)
-        #     face = brepgp_Trsf.Shape()
+
+        for id, face in enumerate([self.upper_face, self.lower_face,
+                     self.tip_face, self.root_face]):
+            from OCC.Core.BRepBuilderAPI import BRepBuilderAPI_Transform
+            from OCC.Core.gp import gp_Pnt
+            from OCC.Core.gp import gp_Dir
+            from OCC.Core.gp import gp_Ax1
+            from OCC.Core.gp import gp_Trsf
+            
+            origin = gp_Pnt(0, 0, 0)
+            trsf = gp_Trsf()
+            trsf.SetScale(origin, factor)
+
+            brep_tr = BRepBuilderAPI_Transform(face, trsf, True, True)
+            face = brep_tr.Shape()
+            if id == 0:
+                self.upper_face = face
+            elif id == 1:
+                self.lower_face = face
+            elif id == 2:
+                self.tip_face = face
+            elif id == 3:
+                self.root_face = face
 
     def plot(self, elev=None, azim=None, ax=None, outfile=None):
         """
@@ -944,3 +995,15 @@ def __str__(self):
         string += '\nInduced rake from skew (in unit length)'\
                   ' for the sections = {}'.format(self.induced_rake)
         return string
+
+    def display(self):
+        """
+        Display the propeller with shaft.
+        """
+        from OCC.Display.SimpleGui import init_display
+        display, start_display = init_display()[:2]
+        display.DisplayShape(self.upper_face, update=True)
+        display.DisplayShape(self.lower_face, update=True)
+        display.DisplayShape(self.root_face, update=True)
+        display.DisplayShape(self.tip_face, update=True)
+        start_display()

bladex/propeller.py

@@ -27,17 +27,27 @@ def __init__(self, shaft, blade, n_blades):
         self.shaft_solid = shaft.generate_solid()
 
         blade.build(reflect=True)
+        import copy
+
+        self.blades = [copy.deepcopy(blade) for _ in range(n_blades)]
+        [
+            blade.rotate(rad_angle=i * 2.0 * np.pi / float(n_blades))
+            for i, blade in enumerate(self.blades)
+        ]
+
         blade_solid = blade.generate_solid()
         blades = []
         blades.append(blade_solid)
         for i in range(n_blades - 1):
             blade.rotate(rad_angle=1.0 * 2.0 * np.pi / float(n_blades))
             blade_solid = blade.generate_solid()
             blades.append(blade_solid)
-        blades_combined = blades[0]
+
+        blade_solids = [blade_.generate_solid() for blade_ in self.blades]
+        blades_combined = blade_solids[0]
         for i in range(len(blades) - 1):
             print(i)
-            boolean_union = BRepAlgoAPI_Fuse(blades_combined, blades[i + 1])
+            boolean_union = BRepAlgoAPI_Fuse(blades_combined, blade_solids[i + 1])
             boolean_union.Build()
             if not boolean_union.IsDone():
                 raise RuntimeError("Unsuccessful assembling of blade")
@@ -245,4 +255,4 @@ def display(self):
         """
         display, start_display = init_display()[:2]
         display.DisplayShape(self.sewed_full_body, update=True)
-        start_display()
+        start_display()