add units to curves

Author: hiaselhans

openglider/glider/cell/attachment_point.py

@@ -7,7 +7,7 @@
 from openglider.glider.shape import Shape
 from openglider.lines.node import Node, NODE_TYPE_ENUM
 from openglider.utils import table
-from openglider.vector.unit import Percentage
+from openglider.vector.unit import Length, Percentage
 
 if TYPE_CHECKING:
     from openglider.glider.cell.cell import Cell
@@ -21,7 +21,7 @@ class CellAttachmentPoint(Node):
     rib_pos: Percentage
     node_type: NODE_TYPE_ENUM = Node.NODE_TYPE.UPPER
     ballooned: bool=False
-    offset: float = 0.
+    offset: Length = 0.
 
     def __repr__(self) -> str:
         return f"<Attachment point '{self.name}' ({self.rib_pos})>"

openglider/glider/curve.py

@@ -5,8 +5,25 @@
 
 from openglider.glider.shape import Shape
 from openglider.utils.cache import cached_property
+from openglider.vector.unit import Angle, Length, Percentage, Quantity
 
-class FreeCurve:
+class CurveBase:
+    unit: str | None
+
+    def to_unit(self, value: float):
+        if self.unit is None:
+            return value
+        
+        if self.unit in Angle.unit_variants or self.unit == Angle.unit:
+            return Angle(value, unit=self.unit)
+        if self.unit in Length.unit_variants or self.unit == Length.unit:
+            return Length(value, unit=self.unit)
+        if self.unit in Percentage.unit_variants:
+            return Percentage(value, self.unit)
+
+        raise ValueError()
+
+class FreeCurve(CurveBase):
     def __init__(self, points: list[euklid.vector.Vector2D], shape: Shape):
         self.shape = shape
         self.interpolation = euklid.vector.Interpolation(points)
@@ -67,11 +84,12 @@ def to_controlpoints(self, points: list[euklid.vector.Vector2D]) -> list[euklid.
     def points_2d(self) -> list[euklid.vector.Vector2D]:
         return self.to_2d(self.interpolation.nodes)
 
-    def get(self, rib_no: int) -> float:
+    def get(self, rib_no: int) -> float | Quantity:
         if rib_no == 0 and self.shape.has_center_cell:
             rib_no = 1
 
-        return self.interpolation.get_value(rib_no)
+        value = self.interpolation.get_value(rib_no)
+        return self.to_unit(value)
 
     def draw(self) -> euklid.vector.PolyLine2D:
         x_values = [p[0] for p in self.controlpoints]
@@ -92,8 +110,8 @@ def draw(self) -> euklid.vector.PolyLine2D:
         return euklid.vector.PolyLine2D(self.to_2d([euklid.vector.Vector2D([x, self.interpolation.get_value(x)]) for x in x_values_lst]))
 
 
-class Curve:
-    upper = False
+class Curve(CurveBase):
+    unit: str
     def __init__(self, points: list[euklid.vector.Vector2D], shape: Shape):
         self.interpolation = euklid.vector.Interpolation(points)
         self.shape = shape
@@ -154,16 +172,13 @@ def points_2d(self) -> euklid.vector.PolyLine2D:
             self.shape.get_point(*p) for p in self.interpolation.nodes
         ])
 
-    def get(self, rib_no: int) -> float:
+    def get(self, rib_no: int) -> float | Quantity:
         if rib_no == 0 and self.shape.has_center_cell:
             rib_no = 1
 
         y = self.interpolation.get_value(rib_no)
 
-        if self.upper:
-            y = -y
-
-        return y
+        return self.to_unit(y)
 
     def draw(self) -> euklid.vector.PolyLine2D:
         x_values = [p[0] for p in self.controlpoints]
@@ -191,8 +206,7 @@ def draw(self) -> euklid.vector.PolyLine2D:
 
 
 class ShapeCurve(Curve):
-    
-    def get(self, rib_no: int) -> float:
+    def get(self, rib_no: int) -> float | Quantity:
         if rib_no == 0 and self.shape.has_center_cell:
             rib_no = 1
 
@@ -203,7 +217,7 @@ def get(self, rib_no: int) -> float:
         if len(results) != 1:
             raise Exception(f"wrong number of cut results: {len(results)}")
 
-        return results[0][1]
+        return self.to_unit(results[0][1])
 
 
 class ShapeBSplineCurve(ShapeCurve):

openglider/glider/parametric/config.py

@@ -1,6 +1,7 @@
 import re
 from typing import Any, ClassVar, Self
 
+from packaging.version import Version
 import euklid
 import pydantic
 
@@ -121,7 +122,7 @@ class ParametricGliderConfig(ConfigTable):
 
     use_sag: bool = True
 
-    version: str = __version__
+    version: Version = Version(__version__)
 
     @classmethod
     def __from_json__(cls, **data: Any) -> Self:

openglider/glider/parametric/export_ods.py

@@ -185,7 +185,6 @@ def add_curve(name: str, curve: CurveType, column_no: int) -> None:
     add_curve("rib_distribution", glider.shape.rib_distribution, 4)
     add_curve("arc", glider.arc.curve, 6)
     add_curve("aoa", glider.aoa, 8)
-    add_curve("zrot", glider.zrot, 10)
     add_curve("ballooning_merge_curve", glider.ballooning_merge_curve, 12)
     add_curve("profile_merge_curve", glider.profile_merge_curve, 14)
 

openglider/glider/parametric/glider.py

@@ -28,7 +28,7 @@
 from openglider.utils.distribution import Distribution
 from openglider.utils.table import Table
 from openglider.utils.types import CurveType, SymmetricCurveType
-from openglider.vector.unit import Percentage
+from openglider.vector.unit import Percentage, Quantity
 
 if TYPE_CHECKING:
     from openglider.glider.curve import GliderCurveType
@@ -53,7 +53,6 @@ class ParametricGlider:
     speed: float
     glide: float
     tables: GliderTables = Field(default_factory=lambda: GliderTables())
-    zrot: SymmetricCurveType = Field(default_factory=lambda: euklid.spline.SymmetricBSplineCurve([[0,0],[1,0]]))
 
     num_interpolate: int=30
     num_profile: int | None=None
@@ -278,7 +277,7 @@ def resolvers(self) -> list[Parser]:
         parsers = []
         curves=self.get_curves()
 
-        def resolver_factory(rib_no: int) -> Callable[[str], float]:
+        def resolver_factory(rib_no: int) -> Callable[[str], float | Quantity]:
             def resolve(name: str) -> float:
                 if name not in curves:
                     raise ValueError(
@@ -386,7 +385,6 @@ def get_glider_3d(self, glider: Glider=None, num: int=50, num_profile: int | Non
         shape_ribs = self.shape.ribs
 
         aoa_values = self.get_aoa()
-        zrot_int = euklid.vector.Interpolation(self.zrot.get_sequence(num).nodes)
 
         arc_pos = self.arc.get_arc_positions(x_values).tolist()
         rib_angles = self.arc.get_rib_angles(x_values)
@@ -421,16 +419,18 @@ def get_glider_3d(self, glider: Glider=None, num: int=50, num_profile: int | Non
             }
             data.update(self.tables.rib_modifiers.get_rib_args(rib_no, resolvers=resolvers))
 
+            rotation = self.tables.rib_modifiers.get_rotation(rib_no, resolvers=resolvers)
+
             rib = Rib(
                 profile_2d=profile,
                 pos=startpoint,
                 chord=chord,
                 arcang=rib_angles[rib_no],
-                xrot=self.tables.rib_modifiers.get_xrot(rib_no),
+                xrot=rotation.x or self.tables.rib_modifiers.get_xrot(rib_no),
                 offset=self.tables.rib_modifiers.get_offset(rib_no, resolvers=resolvers),
                 glide=self.glide,
                 aoa_absolute=aoa_values[rib_no],
-                zrot=zrot_int.get_value(abs(x_value)),
+                zrot=rotation.z,
                 holes=this_rib_holes,
                 rigidfoils=this_rigid_foils,
                 name=f"rib{rib_no}",
@@ -568,5 +568,4 @@ def rescale(curve: CurveType) -> None:
         rescale(self.ballooning_merge_curve)
         rescale(self.profile_merge_curve)
         rescale(self.aoa)
-        rescale(self.zrot)
         self.arc.rescale(self.shape.rib_x_values)

openglider/glider/parametric/import_ods.py

@@ -4,6 +4,7 @@
 import math
 import numbers
 from typing import TYPE_CHECKING
+from packaging.version import Version
 
 import euklid
 import pyfoil
@@ -72,7 +73,7 @@ def import_ods_glider(cls: type[ParametricGlider], tables: list[Table]) -> Param
     # profiles = [BezierProfile2D(profile) for profile in transpose_columns(sheets[3])]
     profiles = [pyfoil.Airfoil(profile, name).normalized() for name, profile in transpose_columns(tables[3])]
 
-    if config.version > "0.0.1":
+    if config.version > Version("0.0.1"):
         has_center_cell = not tables[0]["C2"] == 0
         cell_no = (tables[0].num_rows - 2) * 2 + has_center_cell
         geometry = get_geometry_parametric(table_dct[TableNames.parametric_data], cell_no, config)
@@ -88,7 +89,8 @@ def import_ods_glider(cls: type[ParametricGlider], tables: list[Table]) -> Param
     migrate_header = cell_sheet[0, 0] is not None and cell_sheet[0, 0] < "V4"
 
     glider_tables = GliderTables()
-    glider_tables.curves = CurveTable(table_dct.get("Curves", None))
+    glider_tables.curves = CurveTable(table_dct.get("Curves", None), config.version)
+
     glider_tables.cuts = CutTable(cell_sheet, migrate_header=migrate_header)
     glider_tables.ballooning_modifiers = BallooningModifierTable(cell_sheet, migrate_header=migrate_header)
     glider_tables.holes = HolesTable(rib_sheet, migrate_header=migrate_header)
@@ -121,7 +123,6 @@ class Geometry(BaseModel):
     shape: ParametricShape
     arc: ArcCurve
     aoa: SymmetricCurveType
-    zrot: SymmetricCurveType
     profile_merge_curve: SymmetricCurveType
     ballooning_merge_curve: SymmetricCurveType
 
@@ -135,7 +136,6 @@ def get_geometry_explicit(sheet: Table, config: ParametricGliderConfig) -> Geome
     arc = []
     profile_merge = []
     ballooning_merge = []
-    zrot = []
 
     y = z = span_last = alpha = 0.
     for i in range(1, sheet.num_rows):
@@ -162,8 +162,6 @@ def get_geometry_explicit(sheet: Table, config: ParametricGliderConfig) -> Geome
         profile_merge.append([span, line[8]])
         ballooning_merge.append([span, line[9]])
 
-        zrot.append([span, line[7] * math.pi / 180])
-
         span_last = span
 
     def symmetric_fit(data: list[list[float]], bspline: bool=True) -> SymmetricCurveType:
@@ -200,7 +198,6 @@ def symmetric_fit(data: list[list[float]], bspline: bool=True) -> SymmetricCurve
         shape=parametric_shape,
         arc=arc_curve,
         aoa=symmetric_fit(aoa),
-        zrot=symmetric_fit(zrot),
         profile_merge_curve=symmetric_fit(profile_merge, bspline=True),
         ballooning_merge_curve=symmetric_fit(ballooning_merge, bspline=True)
     )
@@ -214,13 +211,16 @@ def get_geometry_parametric(table: Table, cell_num: int, config: ParametricGlide
         "rib_distribution": euklid.spline.BezierCurve,
         "arc": euklid.spline.SymmetricBSplineCurve,
         "aoa": euklid.spline.SymmetricBSplineCurve,
-        "zrot": euklid.spline.SymmetricBSplineCurve,
         "profile_merge_curve": euklid.spline.SymmetricBSplineCurve,
         "ballooning_merge_curve": euklid.spline.SymmetricBSplineCurve
     }
 
     for column in range(0, table.num_columns, 2):
         key = table[0, column]
+        if key not in curve_types:
+            if key == "zrot":
+                continue
+            raise ValueError("Invalid curve: {key}")
         points = []
 
         if table[0, column+1] is not None:

openglider/glider/parametric/table/base/parser.py

@@ -30,7 +30,7 @@ def default_resolver(key: str) -> float:
 
 class Parser(BaseModel):
     units: list[type[Quantity]] = Field(default_factory=default_units.copy)
-    variable_resolver: Callable[[str], float] = default_resolver
+    variable_resolver: Callable[[str], float | Quantity] = default_resolver
 
     _parser: Forward | None =  pydantic.PrivateAttr(default=None)
     _units: dict[str, type[Quantity]] | None = pydantic.PrivateAttr(default=None)

openglider/glider/parametric/table/base/table.py

@@ -9,6 +9,7 @@
 from openglider.glider.parametric.table.base.dto import DTO
 from openglider.glider.parametric.table.base.parser import Parser
 from openglider.utils.table import Table
+from openglider.vector.unit import Quantity
 
 from .keyword import Keyword
 
@@ -110,7 +111,7 @@ def get(self, row_no: int, keywords: list[str] | None=None, **kwargs: Any) -> li
         return elements
 
     @staticmethod
-    def get_curve_value(curves: dict[str, GliderCurveType] | None, curve_name: str | float, rib_no: int) -> float:
+    def get_curve_value(curves: dict[str, GliderCurveType] | None, curve_name: str | float, rib_no: int) -> float | Quantity:
         if curves is None:
             raise ValueError("No curves specified")
 

openglider/glider/parametric/table/curve.py

@@ -1,8 +1,10 @@
 from __future__ import annotations
 
-from typing import Any
+from typing import Any, Type
+from packaging.version import Version
 from openglider.glider.parametric.table.base import TableType
 
+from openglider.version import version
 from openglider.utils.table import Table
 from openglider.glider.shape import Shape
 import openglider.glider.curve
@@ -12,8 +14,15 @@ class CurveTable:
     table_type = TableType.general
     table_name = "Curves"
 
-    def __init__(self, table: Table=None):
-        self.table = table or Table()
+    def __init__(self, table: Table | None=None, openglider_version: Version=version):
+        if table is None:
+            self.table = Table()
+        else:
+            self.table = table or Table()
+
+            if openglider_version < Version("0.1.3"):
+                self.table = migrate_0_1_2(self.table)
+
         self.table.name = self.table_name
 
     def __json__(self) -> dict[str, Any]:
@@ -27,11 +36,13 @@ def get_curves(self, shape: Shape) -> dict[str, openglider.glider.curve.GliderCu
         curve_columns = 2
 
         while column < self.table.num_columns:
-            name = self.table[0, column]
-            curve_type = self.table[0, column + 1] or "Curve"
+            name = self.table[0, column+1]
+            curve_type = self.table[1, column+1] or "Curve"
+            curve_unit = self.table[2, column+1]
+
             points = []
 
-            for row in range(1, self.table.num_rows):
+            for row in range(3, self.table.num_rows):
                 coords = [self.table[row, column+i] for i in range(curve_columns)]
 
                 if any([c is None for c in coords]):
@@ -40,10 +51,13 @@ def get_curves(self, shape: Shape) -> dict[str, openglider.glider.curve.GliderCu
                 points.append(coords)
 
             try:
-                curve_cls = getattr(openglider.glider.curve, curve_type)
+                curve_cls: Type[openglider.glider.curve.Curve] = getattr(openglider.glider.curve, curve_type)
             except Exception:
                 raise Exception(f"invalid curve type: {curve_type}")
-            curves[name] = curve_cls(points, shape)
+            curve_instance = curve_cls(points, shape)
+            curve_instance.unit = curve_unit
+
+            curves[name] = curve_instance 
 
             column += curve_columns
 
@@ -67,3 +81,20 @@ def apply_curves(self, curves: dict[str, openglider.glider.curve.GliderCurveType
 
 
 
+def migrate_0_1_2(table: Table) -> Table:
+    new_table = Table()
+    
+    column = 0
+    while column < table.num_columns:
+        new_column = Table()
+        new_column.insert_row(["Name", table[0, column]])
+        new_column.insert_row(["Type", table[0,  column+1]])
+        new_column.insert_row(["Unit", None])
+        old_column = table.get_columns(column, column+2)
+
+        new_column.append_bottom(old_column.get_rows(1, old_column.num_rows))
+
+        new_table.append_right(new_column)
+        column += 2
+    
+    return new_table