help

Author: jagoosw

campyros/new_wind.py

@@ -1,6 +1,7 @@
-import getgfs, dateutil.parser, pickle
-from datetime import datetime
+import getgfs, dateutil.parser, pickle, warnings
+from datetime import datetime, date
 from pathlib import Path 
+import numpy as np
 
 __copyright__ = """
 
@@ -24,7 +25,7 @@
 Path("data/wind").mkdir(parents=True,exist_ok=True)
 
 class Wind:
-    def __init__(self,datetime,cache=False):
+    def __init__(self,datetime=date.today().strftime("%Y%m%d %H:%M"),default=np.array([0,0]),variable=True,cache=False):
         """Initites wind object so wind for some position can be searched
 
         Note
@@ -37,33 +38,59 @@ def __init__(self,datetime,cache=False):
         """
         self.launch_time=dateutil.parser.parse(datetime).timestamp()
         self.cache=cache
+        self.default=default
+        self.variable=variable
 
-        self.forecast=getgfs.Forecast("0p25")
+        self.forecast=getgfs.Forecast("0p25","1hr")
         self.profiles={}#(lat,long,datetime):interp profile
         self.points=[]#tuples (lat,long,forecast)
 
 
-    def get(self,lat,long,alt,flight_time):
-        lat=[float(self.forecast.coords["lat"]["resolution"])*n+float(self.forecast.coords["lat"]["minimum"]) for n in range(0,int(self.forecast.coords["lat"]["grads_size"]))][self.forecast.value_to_index("lat",lat)]
-        long=[float(self.forecast.coords["lon"]["resolution"])*n+float(self.forecast.coords["lon"]["minimum"]) for n in range(0,int(self.forecast.coords["lon"]["grads_size"]))][self.forecast.value_to_index("lon",long)]
-        request_timestamp=self.launch_time+flight_time
-        request_datetime=datetime.fromtimestamp(request_timestamp).strftime("%Y-%m-%d %H:%M")
-        forecast_to_use=self.forecast.datetime_to_forecast(request_datetime)
-        if (lat,long, forecast_to_use) not in self.points:
-            if self.cache==False or not Path("data/wind/%s_%s_%s.pkl"%(lat,long,forecast_to_use)).is_file():
-                self.profiles[(lat,long,forecast_to_use)]=self.forecast.get_windprofile(request_datetime,lat,long)
-                self.points.append((lat,long,forecast_to_use))
-                
-                if self.cache==True:
-                    with open("data/wind/%s_%s_%s.pkl"%(lat,long,forecast_to_use),"wb") as dump_file:
-                        pickle.dump(self.profiles[(lat,long,forecast_to_use)],dump_file)
-            else:
-                with open("data/wind/%s_%s_%s.pkl"%(lat,long,forecast_to_use),"rb") as dump_file:
-                    self.profiles[(lat,long,forecast_to_use)]=pickle.load(dump_file)
-                self.points.append((lat,long,forecast_to_use))
-        
-        return self.profiles[(lat,long,forecast_to_use)][0](alt),self.profiles[(lat,long,forecast_to_use)][2](alt)
+    def get(self,lat,long,alt,flight_time=0):
+        """[summary]
 
+        Args:
+            lat (float): latitude
+            long (float): longitude
+            alt (float): altitude
+            flight_time (float, optional): time into flight. Defaults to 0.
+
+        Returns:
+            tuple: returns u and v components of wind
+        """
+        if self.variable==True:
+            lat=[float(self.forecast.coords["lat"]["resolution"])*n+float(self.forecast.coords["lat"]["minimum"]) for n in range(0,int(self.forecast.coords["lat"]["grads_size"]))][self.forecast.value_to_index("lat",lat)]
+            long=[float(self.forecast.coords["lon"]["resolution"])*n+float(self.forecast.coords["lon"]["minimum"]) for n in range(0,int(self.forecast.coords["lon"]["grads_size"]))][self.forecast.value_to_index("lon",long)]
+            request_timestamp=self.launch_time+flight_time
+            request_datetime=datetime.fromtimestamp(request_timestamp).strftime("%Y-%m-%d %H:%M")
+            forecast_to_use=self.forecast.datetime_to_forecast(request_datetime)
+            if (lat,long, forecast_to_use) not in self.points:
+                if self.cache==False or not Path("data/wind/%s_%s_%s.pkl"%(lat,long,forecast_to_use)).is_file():
+                    try:
+                        self.profiles[(lat,long,forecast_to_use)]=self.forecast.get_windprofile(request_datetime,lat,long)
+                        self.points.append((lat,long,forecast_to_use))
+                    except Exception as e:
+                        warnings.warn("Wind could not be found - defaults used, gfspy gave the message %s"%e)
+                        return self.default
+                    
+                    if self.cache==True:
+                        with open("data/wind/%s_%s_%s.pkl"%(lat,long,forecast_to_use),"wb") as dump_file:
+                            pickle.dump(self.profiles[(lat,long,forecast_to_use)],dump_file)
+                else:
+                    with open("data/wind/%s_%s_%s.pkl"%(lat,long,forecast_to_use),"rb") as dump_file:
+                        self.profiles[(lat,long,forecast_to_use)]=pickle.load(dump_file)
+                    self.points.append((lat,long,forecast_to_use))
+            
+            return self.profiles[(lat,long,forecast_to_use)][0](alt),self.profiles[(lat,long,forecast_to_use)][1](alt)
+        else:
+            return self.default
 if __name__=="__main__":
-    w=Wind("20210321 15:00",cache=True)
-    print(w.get(0,0,0,0))
+    w=Wind("20210325 15:00",cache=True)
+    print(w.get(2938,29328,20866566))
+    """times=np.linspace(0,60*60*12,1000)
+    x=[]
+    for t in times:
+        x.append(w.get(37,20,1000,t)[0])
+    import matplotlib.pyplot as plt
+    plt.plot(times,x)
+    plt.show()"""

campyros/tests/test.py

@@ -8,10 +8,12 @@
 )
 import campyros as pyro
 from campyros import statistical as stats
+from campyros import new_wind as wind
 import csv
 import time
 import numpy as np
 import pandas as pd
+from datetime import date, timedelta
 
 __copyright__ = """
 
@@ -136,7 +138,7 @@
     alt_poll_interval=1,
     parachute=parachute,
 )
-
+"""
 run = martlet4.run(debug=False)
 
 test_output = pyro.from_json("campyros/tests/test.json")
@@ -166,10 +168,10 @@
         parachute_ind_run = ind
     if "Cleared rail" in ev:
         rail_ind_run = ind
-
+"""
 
 class ExampleTest(unittest.TestCase):
-    def test_time(self):
+    """def test_time(self):
         self.assertAlmostEqual(run_time // 5, run.time.max() // 5, places=0)
 
     def test_apogee(self):
@@ -214,8 +216,18 @@ def test_stats(self):
             True,
             ran,
             msg="Statistical model run failed, no further information automatically available",
-        )
-
+        )"""
+
+    def test_wind(self):
+        wind1=wind.Wind((date.today()-timedelta(days=1)).strftime("%Y%m%d %H:%M"))
+        wind1_val=wind1.get(239,923,292728238)
+        self.assertIsInstance(wind1_val,type((0.0,0.0)))
+        self.assertEqual(2,len(wind1_val))
+
+        wind2=wind.Wind(variable=False)
+        wind2_val=wind2.get(-239,-923,-292728238)
+        self.assertEqual(0,wind2_val[0])
+        self.assertEqual(0,wind2_val[1])
 
 if __name__ == "__main__":
     unittest.main()