Added start position args

paramak/parametric_components/toroidal_field_coil_coat_hanger.py

@@ -41,6 +41,7 @@ def __init__(
         number_of_coils: int,
         with_inner_leg: bool = True,
         color: Tuple[float, float, float, Optional[float]] = (0.0, 0.0, 1.0),
+        azimuth_start_angle: float = 0,
         **kwargs
     ) -> None:
 
@@ -54,6 +55,7 @@ def __init__(
         self.distance = distance
         self.number_of_coils = number_of_coils
         self.with_inner_leg = with_inner_leg
+        self.azimuth_start_angle = azimuth_start_angle
 
     @property
     def azimuth_placement_angle(self):
@@ -205,7 +207,14 @@ def find_azimuth_placement_angle(self):
         """Calculates the azimuth placement angles based on the number of
         toroidal field coils"""
 
-        angles = list(np.linspace(0, 360, self.number_of_coils, endpoint=False))
+        angles = list(
+            np.linspace(
+                self.azimuth_start_angle,
+                360 + self.azimuth_start_angle,
+                self.number_of_coils,
+                endpoint=False
+            )
+        )
 
         self.azimuth_placement_angle = angles
 

paramak/parametric_components/toroidal_field_coil_princeton_d.py

@@ -33,6 +33,7 @@ def __init__(
         vertical_displacement: float = 0.0,
         with_inner_leg: bool = True,
         color: Tuple[float, float, float, Optional[float]] = (0.0, 0.0, 1.0),
+        azimuth_start_angle: float = 0,
         **kwargs
     ) -> None:
 
@@ -45,6 +46,7 @@ def __init__(
         self.number_of_coils = number_of_coils
         self.vertical_displacement = vertical_displacement
         self.with_inner_leg = with_inner_leg
+        self.azimuth_start_angle = azimuth_start_angle
 
     @property
     def inner_points(self):
@@ -180,6 +182,13 @@ def find_azimuth_placement_angle(self):
         """Calculates the azimuth placement angles based on the number of tf
         coils"""
 
-        angles = list(np.linspace(0, 360, self.number_of_coils, endpoint=False))
+        angles = list(
+            np.linspace(
+                self.azimuth_start_angle,
+                360 + self.azimuth_start_angle,
+                self.number_of_coils,
+                endpoint=False
+            )
+        )
 
         self.azimuth_placement_angle = angles

paramak/parametric_components/toroidal_field_coil_rectangle.py

@@ -34,6 +34,7 @@ def __init__(
         number_of_coils: int,
         with_inner_leg: bool = True,
         color: Tuple[float, float, float, Optional[float]] = (0.0, 0.0, 1.0),
+        azimuth_start_angle: float = 0,
         **kwargs
     ) -> None:
 
@@ -45,6 +46,7 @@ def __init__(
         self.distance = distance
         self.number_of_coils = number_of_coils
         self.with_inner_leg = with_inner_leg
+        self.azimuth_start_angle = azimuth_start_angle
 
     @property
     def azimuth_placement_angle(self):
@@ -112,7 +114,14 @@ def find_azimuth_placement_angle(self):
         """Calculates the azimuth placement angles based on the number of tf
         coils"""
 
-        angles = list(np.linspace(0, 360, self.number_of_coils, endpoint=False))
+        angles = list(
+            np.linspace(
+                self.azimuth_start_angle,
+                360 + self.azimuth_start_angle,
+                self.number_of_coils,
+                endpoint=False
+            )
+        )
 
         self.azimuth_placement_angle = angles
 

paramak/parametric_components/toroidal_field_coil_round_corners.py

@@ -39,6 +39,7 @@ def __init__(
         distance: float,
         number_of_coils: int = 1,
         with_inner_leg: Optional[bool] = False,
+        azimuth_start_angle: float = 0,
         **kwargs
     ) -> None:
 
@@ -56,6 +57,7 @@ def __init__(
         self._height = 0
         self._inner_curve_radius = 0
         self._outter_curve_radius = 0
+        self.azimuth_start_angle = azimuth_start_angle
 
         if len(lower_inner_coordinates) != 2 or len(mid_point_coordinates) != 2:
             msg = (
@@ -329,7 +331,15 @@ def find_azimuth_placement_angle(self):
         """Finds the placement angles from the number of coils
         given in a 360 degree"""
 
-        angles = list(np.linspace(0, 360, self._number_of_coils, endpoint=False))
+        angles = list(
+            np.linspace(
+                self.azimuth_start_angle,
+                360 + self.azimuth_start_angle,
+                self.number_of_coils,
+                endpoint=False
+            )
+        )
+
         self.azimuth_placement_angle = angles
 
     def create_solid(self):

paramak/parametric_components/toroidal_field_coil_triple_arc.py

@@ -34,6 +34,7 @@ def __init__(
         vertical_displacement: float = 0.0,
         with_inner_leg: bool = True,
         color: Tuple[float, float, float, Optional[float]] = (0.0, 0.0, 1.0),
+        azimuth_start_angle: float = 0,
         **kwargs
     ) -> None:
 
@@ -48,6 +49,7 @@ def __init__(
         self.number_of_coils = number_of_coils
         self.vertical_displacement = vertical_displacement
         self.with_inner_leg = with_inner_leg
+        self.azimuth_start_angle = azimuth_start_angle
 
     @property
     def azimuth_placement_angle(self):
@@ -164,6 +166,13 @@ def find_azimuth_placement_angle(self):
         """Calculates the azimuth placement angles based on the number of tf
         coils"""
 
-        angles = list(np.linspace(0, 360, self.number_of_coils, endpoint=False))
+        angles = list(
+            np.linspace(
+                self.azimuth_start_angle,
+                360 + self.azimuth_start_angle,
+                self.number_of_coils,
+                endpoint=False
+            )
+        )
 
         self.azimuth_placement_angle = angles