Merge pull request #73 from leosaffin/witrack

Witrack
Huracan-project · Dec 6, 2024 · f8cae13 · f8cae13
2 parents 93f5638 + 5467bc9
commit f8cae13
Show file tree

Hide file tree

Showing 6 changed files with 3,498 additions and 1 deletion.
diff --git a/huracanpy/__init__.py b/huracanpy/__init__.py
@@ -24,6 +24,7 @@
     "example_TE_file",
     "example_CHAZ_file",
     "example_MIT_file",
+    "example_WiTRACK_file",
     "_accessor",
 ]
 
@@ -39,6 +40,7 @@
     example_TE_file,
     example_CHAZ_file,
     example_MIT_file,
+    example_WiTRACK_file,
 )
 from ._basins import basins
 from ._interp import interp_time

diff --git a/huracanpy/_data/__init__.py b/huracanpy/_data/__init__.py
@@ -12,6 +12,7 @@
     "example_TE_file",
     "example_CHAZ_file",
     "example_MIT_file",
+    "example_WiTRACK_file",
 ]
 
 import pathlib
@@ -43,3 +44,4 @@
 
 example_CHAZ_file = str(testdata_dir / "CHAZ_sample.nc")
 example_MIT_file = str(testdata_dir / "MIT_sample.nc")
+example_WiTRACK_file = str(testdata_dir / "event_tracks_2017_ERA5.dat")
diff --git a/huracanpy/_data/_load.py b/huracanpy/_data/_load.py
@@ -1,7 +1,7 @@
 import pandas as pd
 
 
-from . import _csv, _TRACK, _netcdf, _tempestextremes
+from . import _csv, _TRACK, _netcdf, _tempestextremes, witrack
 from . import ibtracs
 
 
@@ -52,6 +52,7 @@ def load(
         * **track**
         * **track.tilt**
         * **te**, **tempest**, **tempestextremes**, **uz** (For textual format, not csv)
+        * **witrack**
         * **ibtracs**
         * **csv**
         * **netcdf**, **nc** (includes support for CHAZ & MIT-Open file provided appropriate variable_names)
@@ -168,6 +169,8 @@ def load(
                 tempest_extremes_unstructured,
                 tempest_extremes_header_str,
             )
+        elif source == "witrack":
+            data = witrack.load(filename)
         elif source == "ibtracs":
             data = ibtracs.load(ibtracs_subset, filename, **kwargs)
         elif source == "netcdf":