foxes 1.2.3__tar.gz → 1.2.4__tar.gz

{foxes-1.2.3/foxes.egg-info → foxes-1.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: foxes
-Version: 1.2.3
+Version: 1.2.4
 Summary: Farm Optimization and eXtended yield Evaluation Software
 Author: Jonas Schulte
 Maintainer: Jonas Schulte

{foxes-1.2.3 → foxes-1.2.4}/foxes/config/config.py

@@ -1,5 +1,6 @@
 import numpy as np
 from pathlib import Path
+from sys import version_info
 
 from foxes.utils import Dict, import_module
 import foxes.constants as FC
@@ -26,6 +27,10 @@ class Config(Dict):
             name="config",
         )
 
+        # special treat for Python 3.8:
+        if version_info[0] == 3 and version_info[1] == 8:
+            self["nc_engine"] = None
+
     @property
     def dtype_double(self):
         """
@@ -119,8 +124,10 @@ class Config(Dict):
             The NetCDF engine
 
         """
-        import_module(self[FC.NC_ENGINE])
-        return self[FC.NC_ENGINE]
+        nce = self[FC.NC_ENGINE]
+        if nce is not None:
+            import_module(nce)
+        return nce
 
 
 config = Config()

{foxes-1.2.3 → foxes-1.2.4}/foxes/utils/__init__.py

@@ -17,6 +17,7 @@ from .regularize import sqrt_reg
 from .tab_files import read_tab_file
 from .random_xy import random_xy_square
 from .dev_utils import print_mem
+from .wrg_utils import ReaderWRG
 
 from . import two_circles
 from . import abl

foxes-1.2.4/foxes/utils/wrg_utils.py

@@ -0,0 +1,79 @@
+import pandas as pd
+from pathlib import Path
+
+
+class ReaderWRG:
+    """
+    A reader for WRG files
+
+    Attributes
+    ----------
+    fpath: pathlib.Path
+        Path to the wrg file
+
+    """
+
+    def __init__(self, fpath):
+        """
+        Constructor
+
+        Parameters
+        ----------
+        fpath: str
+            Path to the wrg file
+        """
+        self.fpath = Path(fpath)
+        self._prepare()
+
+    def _prepare(self):
+        # read the first two lines
+        with open(self.fpath, "r") as fstream:
+            self._nx, self._ny, self._utmx0, self._utmy0, self._res = (
+                fstream.readline().split()
+            )
+            second_line = fstream.readline().split()
+            n_cols = len(second_line)
+            self._n_sectors = int(second_line[8])
+            self._nx, self._ny, self._utmx0, self._utmy0, self._res = (
+                int(self._nx),
+                int(self._ny),
+                float(self._utmx0),
+                float(self._utmy0),
+                int(self._res),
+            )
+
+        cols_sel = lambda name, secs: [f"{name}_{i}" for i in range(secs)]
+
+        cols = [0] * (8 + 3 * self._n_sectors)
+        cols[0] = "utmx"
+        cols[1] = "utmy"
+        cols[2] = "z"
+        cols[3] = "h"
+        cols[4] = "A"
+        cols[5] = "K"
+        cols[6] = "pw"
+        cols[7] = "n_sectors"
+        cols[8::3] = cols_sel("fs", self._n_sectors)
+        cols[9::3] = cols_sel("As", self._n_sectors)
+        cols[10::3] = cols_sel("Ks", self._n_sectors)
+
+        self._data = pd.read_csv(
+            self.fpath, names=cols, skiprows=1, sep="\s+", usecols=range(1, n_cols)
+        )
+
+        self._data[cols_sel("fs", self._n_sectors)] /= 10
+        self._data[cols_sel("As", self._n_sectors)] /= 10
+        self._data[cols_sel("Ks", self._n_sectors)] /= 100
+
+    @property
+    def data(self):
+        """
+        The WRG data
+
+        Returns
+        -------
+        df: pandas.DataFrame
+            The WRG data
+
+        """
+        return self._data

{foxes-1.2.3 → foxes-1.2.4/foxes.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: foxes
-Version: 1.2.3
+Version: 1.2.4
 Summary: Farm Optimization and eXtended yield Evaluation Software
 Author: Jonas Schulte
 Maintainer: Jonas Schulte

{foxes-1.2.3 → foxes-1.2.4}/foxes.egg-info/SOURCES.txt

@@ -275,6 +275,7 @@ foxes/utils/subclasses.py
 foxes/utils/tab_files.py
 foxes/utils/two_circles.py
 foxes/utils/wind_dir.py
+foxes/utils/wrg_utils.py
 foxes/utils/xarray_utils.py
 foxes/utils/abl/__init__.py
 foxes/utils/abl/neutral.py

{foxes-1.2.3 → foxes-1.2.4}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "foxes"
-version = "1.2.3"
+version = "1.2.4"
 authors = [
   {name = "Jonas Schulte"},
 ]