adjusts airfoil signature accordingly

RocketPy-Team · Oct 2, 2024 · 1a8e739 · 1a8e739
1 parent 585a98e
commit 1a8e739
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/lib/models/aerosurfaces.py b/lib/models/aerosurfaces.py
@@ -1,5 +1,5 @@
 from enum import Enum
-from typing import Optional, Tuple
+from typing import Optional, Tuple, List
 from pydantic import BaseModel
 
 
@@ -24,6 +24,11 @@ class FinsKinds(str, Enum):
     ELLIPTICAL: str = "ELLIPTICAL"
 
 
+class AngleUnit(str, Enum):
+    RADIANS: str = "RADIANS"
+    DEGREES: str = "DEGREES"
+
+
 class Fins(BaseModel):
     fins_kind: FinsKinds
     name: str
@@ -34,7 +39,7 @@ class Fins(BaseModel):
     position: float
     cant_angle: float
     radius: float
-    airfoil: Tuple[float, float]
+    airfoil: Tuple[List[Tuple[float, float]], AngleUnit]
 
 
 # TODO: implement airbrakes

diff --git a/lib/models/rocket.py b/lib/models/rocket.py
@@ -71,7 +71,7 @@ class Rocket(BaseModel):
             position=-1.04956,
             cant_angle=0,
             radius=0.0635,
-            airfoil=(0.0, 0.0),
+            airfoil=([(0.0, 0.0), (0.1, 0.1), (0.2, 0.2)], "RADIANS"),
         )
     ]
     tail: Optional[Tail] = Tail(