From cadfb29272ba4fc32765db4e3494fb122c40dbed Mon Sep 17 00:00:00 2001
From: dario <dario.vandenboom@arcor.de>
Date: Sun, 21 Apr 2024 22:16:42 +0200
Subject: [PATCH] Better atmosphere model

---
 spatz/models/pressure.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/spatz/models/pressure.py b/spatz/models/pressure.py
index a7ddb73..893797f 100644
--- a/spatz/models/pressure.py
+++ b/spatz/models/pressure.py
@@ -2,6 +2,7 @@ from typing import Any
 import numpy as np
 
 from enum import Enum
+from ambiance import Atmosphere
 
 
 class PressUnit:
@@ -42,3 +43,11 @@ class AltitudeModel:
             press = to_hpa[unit]
 
         return 44330 * (1 - (press / self.__press_0)**(1 / 5.255))
+
+
+class StandardAtmosphere:
+    def __init__(self) -> None:
+        pass
+
+    def predict(self, press: float, unit: PressUnit = PressUnit.Pa) -> float:
+        return Atmosphere.from_pressure(press).h[0]