Merge pull request #54 from NREL/fix/pyi-updates

added `new` and `__new__` methods for Elev

python/altrios/altrios_pyo3.pyi

@@ -968,6 +968,22 @@ class Link(SerdeAPI):
     def default(cls) -> Self: ...
 
 
+class Elev(SerdeAPI):
+    offset_meters: float
+    elev_meters: float
+    @classmethod
+    def default(cls) -> Self: ...
+
+
+class Heading(SerdeAPI):
+    offset_meters: float
+    heading: float
+    lat: Optional[float]
+    lon: Optional[float]
+    @classmethod
+    def default(cls) -> Self: ...
+
+
 def import_locations(filename: str) -> Dict[str, List[Location]]: ...
 def import_rail_vehicles(filename: str) -> Dict[str, RailVehicle]: ...
 

rust/altrios-core/src/track/link/elev.rs

@@ -2,12 +2,26 @@ use crate::imports::*;
 
 /// Struct containing elevation for a particular offset w.r.t. `Link`
 #[derive(Clone, Copy, Default, Debug, PartialEq, PartialOrd, Serialize, Deserialize, SerdeAPI)]
-#[altrios_api]
+#[altrios_api(
+    #[new]
+    fn __new__(
+        offset_meters: f64,
+        elev_meters: f64,
+    ) -> PyResult<Self> {
+        Ok(Self::new(offset_meters * uc::M, elev_meters * uc::M))
+    }
+)]
 pub struct Elev {
     pub offset: si::Length,
     pub elev: si::Length,
 }
 
+impl Elev {
+    pub fn new(offset: si::Length, elev: si::Length) -> Self {
+        Self { offset, elev }
+    }
+}
+
 impl Valid for Elev {}
 
 impl ObjState for Elev {