initial template, untested

blaylockbk · Sep 8, 2023 · 0f0dfd6 · 0f0dfd6
1 parent 8048885
commit 0f0dfd6
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 0 deletions.
diff --git a/herbie/models/__init__.py b/herbie/models/__init__.py
@@ -27,6 +27,7 @@
 from .rtma import *
 from .urma import *
 from .hrdps import *
+from .hafs import *
 
 # ======================================================================
 #                     Import Private Model Templates

diff --git a/herbie/models/hafs.py b/herbie/models/hafs.py
@@ -0,0 +1,45 @@
+"""
+A Herbie template for the HAFS model.
+"""
+
+
+class hafsa:
+    def template(self):
+        self.DESCRIPTION = "Hurricane Analysis and Forecast System"
+        self.DETAILS = {
+            "Homepage": "https://wpo.noaa.gov/the-hurricane-analysis-and-forecast-system-hafs/",
+            "Hurricane Forecast Improvement Program": "https://hfip.org/hafs",
+        }
+        self.PRODUCTS = {
+            "parent.atm": "?",
+            "parent.sat": "?",
+            "parent.swath": "?",
+            "storm.atm": "?",
+            "ww3": "?",
+        }
+        self.SOURCES = {
+           "nomads": f"https://nomads.ncep.noaa.gov/pub/data/nccf/com/hafs/prod/{self.model}.{self.date:%Y%m%d/%H}/{self.label}.{self.date:%Y%m%d%H}.{self.model}.{self.product}.f{self.fxx:02d}.grb2
+        }
+        self.EXPECT_IDX_FILE = "remote"
+        self.LOCALFILE = f"{self.get_remoteFileName}"
+
+
+class hafsb:
+    def template(self):
+        self.DESCRIPTION = "Hurricane Analysis and Forecast System"
+        self.DETAILS = {
+            "Homepage": "https://wpo.noaa.gov/the-hurricane-analysis-and-forecast-system-hafs/",
+            "Hurricane Forecast Improvement Program": "https://hfip.org/hafs",
+        }
+        self.PRODUCTS = {
+            "parent.atm": "?",
+            "parent.sat": "?",
+            "parent.swath": "?",
+            "storm.atm": "?",
+            "ww3": "?",
+        }
+        self.SOURCES = {
+           "nomads": f"https://nomads.ncep.noaa.gov/pub/data/nccf/com/hafs/prod/{self.model}.{self.date:%Y%m%d/%H}/{self.label}.{self.date:%Y%m%d%H}.{self.model}.{self.product}.f{self.fxx:02d}.grb2
+        }
+        self.EXPECT_IDX_FILE = "remote"
+        self.LOCALFILE = f"{self.get_remoteFileName}"