add numpy/numba support

psychrometrics · Apr 7, 2020 · 1031525 · 1031525
1 parent 73e757a
commit 1031525
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/python/psychrolib.py b/src/python/psychrolib.py
@@ -1459,3 +1459,24 @@ def CalcPsychrometricsFromRelHum(TDryBulb: float, RelHum: float, Pressure: float
     MoistAirVolume = GetMoistAirVolume(TDryBulb, HumRatio, Pressure)
     DegreeOfSaturation = GetDegreeOfSaturation(TDryBulb, HumRatio, Pressure)
     return HumRatio, TWetBulb, TDewPoint, VapPres, MoistAirEnthalpy, MoistAirVolume, DegreeOfSaturation
+
+
+from inspect import isfunction
+
+func_list = []
+for func in list(globals().items()):
+    if isfunction(func[1]) and func[0].startswith(('Get', 'Calc', 'dLnPws_')) and func != 'GetUnitSystem':
+        func_list.append(func)
+
+try:
+    from numba import vectorize, njit
+    isIP = njit(isIP)
+except ImportError:
+    try:
+        from numpy import vectorize
+    except ImportError:
+        vectorize = None
+
+if vectorize:
+    for func in func_list:
+        globals()[func[0]] = vectorize(func[1])