foxes 1.2.3__py3-none-any.whl → 1.2.4__py3-none-any.whl

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/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/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.dist-info → foxes-1.2.4.dist-info}/METADATA

@@ -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.dist-info → foxes-1.2.4.dist-info}/RECORD

@@ -45,7 +45,7 @@ foxes/algorithms/sequential/models/__init__.py,sha256=OneaRLxMPzVWLKL9cR3JIYHojf
 foxes/algorithms/sequential/models/plugin.py,sha256=HhCeHM_YoJRMXuZKa8rXUJZDhWf5rNl28_UfDLSGWvA,1330
 foxes/algorithms/sequential/models/seq_state.py,sha256=J1PorBVV5xZFWobHvCrGDCD-q7Egz_Op1p4PDiacBy0,3654
 foxes/config/__init__.py,sha256=k3irS7dNsXTzQpz9e6GOqoqGZkyMAW_BGxnV8JGcuRs,78
-foxes/config/config.py,sha256=L3p4p6hhWu-PsBXszBrmVM8YIDl79zVJTit2zpfnvjY,4118
+foxes/config/config.py,sha256=e3HhitD2Xleqm3utoXLGAu8Vg16app35Cpx4nBhnZm8,4318
 foxes/core/__init__.py,sha256=3dOuKrFeSM6fCMxfLxTbzUsBPVTZiRp0sFhnibniBko,1021
 foxes/core/algorithm.py,sha256=ggdvtQT6kbmrgPUma-jCHqW7Q3ISvXHud1xl-t-kcls,27446
 foxes/core/axial_induction_model.py,sha256=sGbTHFMjEYKiVjElLP_SCRulKatcK0wsEXf30zty2Vc,1023
@@ -247,7 +247,7 @@ foxes/output/flow_plots_2d/get_fig.py,sha256=gGMOfzNCukq_l0FZLmi3zSBl0ANkQi2WoKa
 foxes/output/seq_plugins/__init__.py,sha256=d6Tl5NBVV4MaZwXtPgyxBXBnh7uCBFTbNzRU6IXCmQQ,110
 foxes/output/seq_plugins/seq_flow_ani_plugin.py,sha256=yAC-ZA_rkIOCHc1TN3uQYW0Z1CbxLTgP84Lrfr4D8cI,3654
 foxes/output/seq_plugins/seq_wake_debug_plugin.py,sha256=oAijRu9XRjUL9X4lZF-V6jqOfv8omokRp7YMkL1I2_M,3969
-foxes/utils/__init__.py,sha256=cVrVo5P9Ge-AxHAtc3_gMGmj1XL46xbfZY4okvIZNT8,744
+foxes/utils/__init__.py,sha256=TkOej3vujD_5kM39hYlPTAuLtavrKZ16nO5wZloXPNA,777
 foxes/utils/cubic_roots.py,sha256=u2Pf-yHZ6EASpFONMfkv0mvldmd1E1VRZxj69YabDoc,3321
 foxes/utils/data_book.py,sha256=z96QUC9FI7mv1NEFJUkkTcxWqx5Fc9gPeyzPwbFbyoM,5301
 foxes/utils/dev_utils.py,sha256=6vryhcTg8lNXhtZOsQ0F8PTVpmUw250VNE-UyR0G70Y,1241
@@ -264,6 +264,7 @@ foxes/utils/subclasses.py,sha256=wCfhuialpBOQOPMwNbaplkVmf60vR9AIkY_o3tdkgXI,173
 foxes/utils/tab_files.py,sha256=H50IpLaqffJn9A51orCGc4fOhCOzoNUYDUKek4OAayU,1811
 foxes/utils/two_circles.py,sha256=xkj-SA_x-VXY7KtmSU4lcV4gFdplyhU3sBAC9vTdkF4,2810
 foxes/utils/wind_dir.py,sha256=6W0njWDvnIdOOjwqcMr64MW9ApjdtFA75blVUxirPMo,2823
+foxes/utils/wrg_utils.py,sha256=3T6NkznU1_nc5bTaq-rEuBRFeBPjqP4XnEhwhQArKnk,2019
 foxes/utils/xarray_utils.py,sha256=6VLTAitXTK6W5zlBTadWwN05qmq_TlEDpUCx227JxtY,1677
 foxes/utils/abl/__init__.py,sha256=ijn-ubLLlqqH6tTAXFRmBAxJZmVBlTEmtx1cdCCtG4I,135
 foxes/utils/abl/neutral.py,sha256=E4DEhvXvw74BPrYr1MjQjeIaoz6ZOTWVlqScKflm-0M,1358
@@ -302,9 +303,9 @@ tests/1_verification/flappy_0_6_2/row_Bastankhah_Crespo/flappy/run.py,sha256=nwI
 tests/1_verification/flappy_0_6_2/row_Bastankhah_linear_centre/test_row_Bastankhah_linear_centre.py,sha256=a7CQS-_Mnz3NynaTv1OY1CZ10iAqa7E3YxmkbDtoshw,2590
 tests/1_verification/flappy_0_6_2/row_Bastankhah_linear_centre/flappy/run.py,sha256=s6FbEdpiIdHYmdD8S85_NhLH-S3EOinXvw8RHmR2QOU,2122
 tests/3_examples/test_examples.py,sha256=rS2Dz04ktbS6v3TRDr96AkWGypr5u49jihqbEmGFmRU,694
-foxes-1.2.3.dist-info/LICENSE,sha256=bBCH6mYTPzSepk2s2UUZ3II_ZYXrn1bnSqB85-aZHxU,1071
-foxes-1.2.3.dist-info/METADATA,sha256=3GhrNj9vLhSr7s_EZ75rAKRYoBZ9pu5_nH8UoIw6u4w,8811
-foxes-1.2.3.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-foxes-1.2.3.dist-info/entry_points.txt,sha256=KuS44FRH5NnMw201A8Btr76eNRKr2UOoKHjejAsqKwE,123
-foxes-1.2.3.dist-info/top_level.txt,sha256=G7oHApEz5nc-iP__XsPcvjYe_NyXGmKMUMPHi3C3x6I,26
-foxes-1.2.3.dist-info/RECORD,,
+foxes-1.2.4.dist-info/LICENSE,sha256=bBCH6mYTPzSepk2s2UUZ3II_ZYXrn1bnSqB85-aZHxU,1071
+foxes-1.2.4.dist-info/METADATA,sha256=vAXcCQux4WPpc196ZFrNyXhyDNKBVL8PETlRB824oiM,8811
+foxes-1.2.4.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+foxes-1.2.4.dist-info/entry_points.txt,sha256=KuS44FRH5NnMw201A8Btr76eNRKr2UOoKHjejAsqKwE,123
+foxes-1.2.4.dist-info/top_level.txt,sha256=G7oHApEz5nc-iP__XsPcvjYe_NyXGmKMUMPHi3C3x6I,26
+foxes-1.2.4.dist-info/RECORD,,