Merged with conflict fix from ICOGS3D_new

schism-dev · Aug 14, 2021 · b9c614e · b9c614e
2 parents cdf44f1 + 87711ce
commit b9c614e
Show file tree

Hide file tree

Showing 12 changed files with 2,103 additions and 1,649 deletions.
diff --git a/.gitignore b/.gitignore
@@ -13,3 +13,4 @@ forecast
 hgrid.*
 fgrid.*
 vgrid.*
+.idea
diff --git a/examples/examples_hindcast/01_nwm.py b/examples/examples_hindcast/01_nwm.py
@@ -0,0 +1,13 @@
+from datetime import datetime
+
+from pyschism.mesh.hgrid import Hgrid
+from pyschism.forcing.hydrology.nwmhindcast import NationalWaterModel
+
+startdate=datetime(2018,7,3)
+rnday=10
+
+hgrid=Hgrid.open('hgrid.gr3', crs='EPSG:4326')
+
+nwm=NationalWaterModel()
+nwm.fetch_data(hgrid, startdate, rnday)
+nwm.write('./', overwrite=True)
diff --git a/examples/examples_hindcast/02_era5.py b/examples/examples_hindcast/02_era5.py
@@ -0,0 +1,18 @@
+from datetime import datetime
+from time import time
+import pathlib
+
+from pyschism.forcing.atmosphere.era5 import ERA5
+
+from pyschism.mesh.hgrid import Hgrid
+
+startdate=datetime(2018,7,1)
+
+t0=time()
+hgrid=Hgrid.open('hgrid.gr3',crs='EPSG:4326')
+bbox = hgrid.get_bbox('EPSG:4326', output_type='bbox')
+
+er=ERA5()
+outdir = pathlib.Path('./ERA5')
+er.gen_sflux(startdate, rnday=10, bbox=bbox, outdir=outdir)
+print(f'It took {(time()-t0)/60} minutes to generate 10 days')
diff --git a/pyschism/enums.py b/pyschism/enums.py
@@ -22,6 +22,21 @@ class Coriolis(Enum):
     CORICOEFF = 0
     RLATITUDE = -1
 
+class IofWetdryVariables(Enum):
+    wetdry_node = "wetdry_node"
+    wetdry_elem = "wetdry_elem"
+    wetdry_side = "wetdry_side"
+
+    @classmethod
+    def _missing_(cls, name):
+        raise ValueError(f"{name} is not a valid WETDRY output variable.")
+
+class IofZcorVariables(Enum):
+    zcor = "zcor"
+
+    @classmethod
+    def _missing_(cls, name):
+        raise ValueError(f"{name} is not a valid ZCOR output variable.")
 
 class IofHydroVariables(Enum):
     elev = "elev"
-Original file line number
+Diff line change
@@ Expand Up / @@ -13,3 +13,4 @@ forecast @@
     hgrid.*
     fgrid.*
     vgrid.*
+    .idea