foxes 1.2.1__py3-none-any.whl → 1.2.2__py3-none-any.whl

foxes/input/yaml/__init__.py

@@ -1,3 +1,3 @@
-from .dict import read_dict, run_dict, run_outputs
+from .dict import read_dict, run_dict, run_outputs, run_obj_function, get_output_obj
 from .yaml import foxes_yaml
 from .windio.windio import read_windio, foxes_windio

foxes/input/yaml/dict.py

@@ -141,11 +141,176 @@ def read_dict(
     return algo, engine
 
 
+def get_output_obj(
+    ocls,
+    odict,
+    algo,
+    farm_results=None,
+    point_results=None,
+    base_class=Output,
+    extra_sig={},
+):
+    """
+    Create the output object
+
+    Parameters
+    ----------
+    ocls: str
+        Name of the output class
+    odict: dict
+        The output dict
+    algo: foxes.core.Algorithm
+        The algorithm
+    farm_results: xarray.Dataset, optional
+        The farm results
+    point_results: xarray.Dataset, optional
+        The point results
+    base_class: object
+        The output's base class
+    extra_sig: dict
+        Extra function signature check, sets
+        arguments (key) with data (value)
+
+    Returns
+    -------
+    obj: object or None
+        The output object
+
+    :group: input.yaml
+
+    """
+    cls = new_cls(base_class, ocls)
+    prs = list(signature(cls.__init__).parameters.keys())
+    if "algo" in prs:
+        assert algo is not None, f"Output of type '{ocls}' requires algo"
+        odict["algo"] = algo
+    if "farm" in prs:
+        odict["farm"] = algo.farm
+    if "farm_results" in prs:
+        if farm_results is None:
+            print(f"No farm results; skipping output {ocls}")
+            return None
+        odict["farm_results"] = farm_results
+    if "point_results" in prs:
+        odict["point_results"] = point_results
+    for k, v in extra_sig.items():
+        if k in prs:
+            odict[k] = v
+
+    return cls(**odict)
+
+
+def _get_object(rlabels, d):
+    """Helper function for object extraction"""
+    d = d.replace("]", "")
+    i0 = d.find("[")
+    if i0 > 0:
+        inds = tuple([int(x) for x in d[i0 + 1 :].split(",")])
+        return rlabels[d[:i0]][inds]
+    else:
+        return rlabels[d]
+
+
+def run_obj_function(
+    obj,
+    fdict,
+    algo,
+    rlabels,
+    verbosity=None,
+):
+    """
+    Runs a function of an object
+
+    Parameters
+    ----------
+    obj: object
+        The object
+    fdict: dict
+        The function call dict
+    algo: foxes.core.Algorithm
+        The algorithm
+    rlabels: dict
+        Storage for result variables
+    verbosity: int, optional
+        The verbosity level, 0 = silent
+
+    Returns
+    -------
+    results: object
+        The returns of the function
+
+    :group: input.yaml
+
+    """
+
+    def _print(*args, level=1, **kwargs):
+        if verbosity is None or verbosity >= level:
+            print(*args, **kwargs)
+
+    fname = fdict.pop_item("function")
+    _print(f"  - {fname}")
+    plt_show = fdict.pop("plt_show", False)
+    plt_close = fdict.pop("plt_close", False)
+    rlbs = fdict.pop("result_labels", None)
+
+    # grab function:
+    ocls = type(obj).__name__
+    assert hasattr(obj, fname), f"Output of type '{ocls}': Function '{fname}' not found"
+    f = getattr(obj, fname)
+
+    # add required input data objects:
+    prs = list(signature(f).parameters.keys())
+    if "algo" in prs:
+        fdict["algo"] = algo
+    if "farm" in prs:
+        fdict["farm"] = algo.farm
+
+    # replace result labels by objects:
+    for k, d in fdict.items():
+        if isinstance(d, str) and d[0] == "$":
+            fdict[k] = _get_object(rlabels, d)
+
+    # run function:
+    args = fdict.pop("args", tuple())
+    results = f(*args, **fdict)
+
+    # pyplot shortcuts:
+    if plt_show:
+        plt.show()
+    if plt_close:
+        results = None
+        plt.close()
+
+    # store results under result labels:
+    if rlbs is not None:
+
+        def _set_label(rlabels, k, r):
+            if k not in ["", "none", "None", "_", "__"]:
+                assert (
+                    k[0] == "$"
+                ), f"Output {i} of type '{ocls}', function '{fname}': result labels must start with '$', got '{k}'"
+                assert (
+                    "[" not in k and "]" not in k and "," not in k
+                ), f"Output {i} of type '{ocls}', function '{fname}': result labels cannot contain '[' or ']' or comma, got '{k}'"
+                _print(f"    result label {k}: {type(r).__name__}")
+                rlabels[k] = r
+
+        if isinstance(rlbs, (list, tuple)):
+            for i, k in enumerate(rlbs):
+                _set_label(rlabels, k, results[i])
+        else:
+            _set_label(rlabels, rlbs, results)
+
+    return results
+
+
 def run_outputs(
     idict,
     algo=None,
     farm_results=None,
     point_results=None,
+    extra_sig={},
+    ret_rlabels=False,
     verbosity=None,
 ):
     """
@@ -161,9 +326,13 @@ def run_outputs(
         The farm results
     point_results: xarray.Dataset, optional
         The point results
+    extra_sig: dict
+        Extra function signature check, sets
+        arguments (key) with data (value)
+    ret_rlabels: bool
+        Flag for returning results variables
     verbosity: int, optional
-        Force a verbosity level, 0 = silent, overrules
-        settings from idict
+        The verbosity level, 0 = silent
 
     Returns
     -------
@@ -171,6 +340,8 @@ def run_outputs(
         For each output enty, a tuple (dict, results),
         where results is a tuple that represents one
         entry per function call
+    rlabels: dict, optional
+        The results variables
 
     :group: input.yaml
 
@@ -181,6 +352,7 @@ def run_outputs(
             print(*args, **kwargs)
 
     out = []
+    rlabels = Dict(name="result_labels")
     if "outputs" in idict:
         _print("Running outputs")
         odicts = [
@@ -188,124 +360,48 @@ def run_outputs(
             for i, odict in enumerate(idict["outputs"])
         ]
 
-        rlabels = Dict(name="result_labels")
-
-        def _get_object(rlabels, d):
-            d = d.replace("]", "")
-            i0 = d.find("[")
-            if i0 > 0:
-                inds = tuple([int(x) for x in d[i0 + 1 :].split(",")])
-                return rlabels[d[:i0]][inds]
-            else:
-                return rlabels[d]
-
         for i, d in enumerate(odicts):
-
             if "output_type" in d:
                 ocls = d.pop_item("output_type")
                 _print(f"  Output {i}: {ocls}")
                 d0 = dict(output_type=ocls)
                 d0.update(d)
+
                 flist = [
-                    Dict(f, name=f"{d.name}.function{i}")
-                    for i, f in enumerate(d.pop_item("functions"))
+                    Dict(f, name=f"{d.name}.function{j}")
+                    for j, f in enumerate(d.pop_item("functions"))
                 ]
-                cls = new_cls(Output, ocls)
-                prs = list(signature(cls.__init__).parameters.keys())
-                if "algo" in prs:
-                    assert (
-                        algo is not None
-                    ), f"Output {i} of type '{ocls}' requires algo"
-                    d["algo"] = algo
-                if "farm" in prs:
-                    d["farm"] = algo.farm
-                if "farm_results" in prs:
-                    if farm_results is None:
-                        print(f"No farm results; skipping output {ocls}")
-                        for fdict in flist:
-                            out += (d0, None)
-                        continue
-                    d["farm_results"] = farm_results
-                if "point_results" in prs:
-                    d["point_results"] = point_results
-                o = cls(**d)
+
+                o = get_output_obj(
+                    ocls, d, algo, farm_results, point_results, extra_sig=extra_sig
+                )
+                if o is None:
+                    out.append((d0, None))
+                    continue
 
             elif "object" in d:
                 ocls = d.pop("object")
+                _print(f"  Output {i}: Object {ocls}")
                 o = _get_object(rlabels, ocls)
                 d0 = dict(object=ocls)
                 d0.update(d)
                 flist = [
-                    Dict(f, name=f"{d.name}.function{i}")
-                    for i, f in enumerate(d.pop_item("functions"))
+                    Dict(f, name=f"{d.name}.function{j}")
+                    for j, f in enumerate(d.pop_item("functions"))
                 ]
 
             else:
                 raise KeyError(
-                    f"Output {i} of type '{ocls}': Please specify either 'output_type' or 'object'"
+                    f"Output {i}: Please specify either 'output_type' or 'object'"
                 )
 
             fres = []
             for fdict in flist:
-                fname = fdict.pop_item("function")
-                _print(f"  - {fname}")
-                plt_show = fdict.pop("plt_show", False)
-                plt_close = fdict.pop("plt_close", False)
-                rlbs = fdict.pop("result_labels", None)
-
-                # grab function:
-                assert hasattr(
-                    o, fname
-                ), f"Output {i} of type '{ocls}': Function '{fname}' not found"
-                f = getattr(o, fname)
-
-                # add required input data objects:
-                prs = list(signature(f).parameters.keys())
-                if "algo" in prs:
-                    fdict["algo"] = algo
-                if "farm" in prs:
-                    fdict["farm"] = algo.farm
-
-                # replace result labels by objects:
-                for k, d in fdict.items():
-                    if isinstance(d, str) and d[0] == "$":
-                        fdict[k] = _get_object(rlabels, d)
-
-                # run function:
-                args = fdict.pop("args", tuple())
-                results = f(*args, **fdict)
-
-                # pyplot shortcuts:
-                if plt_show:
-                    plt.show()
-                if plt_close:
-                    results = None
-                    plt.close()
-
-                # store results under result labels:
-                if rlbs is not None:
-
-                    def _set_label(rlabels, k, r):
-                        if k not in ["", "none", "None", "_", "__"]:
-                            assert (
-                                k[0] == "$"
-                            ), f"Output {i} of type '{ocls}', function '{fname}': result labels must start with '$', got '{k}'"
-                            assert (
-                                "[" not in k and "]" not in k and "," not in k
-                            ), f"Output {i} of type '{ocls}', function '{fname}': result labels cannot contain '[' or ']' or comma, got '{k}'"
-                            _print(f"    result label {k}: {type(r).__name__}")
-                            rlabels[k] = r
-
-                    if isinstance(rlbs, (list, tuple)):
-                        for i, k in enumerate(rlbs):
-                            _set_label(rlabels, k, results[i])
-                    else:
-                        _set_label(rlabels, rlbs, results)
-
+                results = run_obj_function(o, fdict, algo, rlabels, verbosity)
                 fres.append(results)
             out.append((d0, fres))
 
-    return out
+    return out if not ret_rlabels else out, rlabels
 
 
 def run_dict(idict, *args, verbosity=None, **kwargs):
@@ -371,7 +467,7 @@ def run_dict(idict, *args, verbosity=None, **kwargs):
         out += (point_results,)
 
     # run outputs:
-    out += (run_outputs(idict, algo, farm_results, point_results, verbosity),)
+    out += (run_outputs(idict, algo, farm_results, point_results, verbosity=verbosity),)
 
     # shutdown engine, if created above:
     if engine is not None:

{foxes-1.2.1.dist-info → foxes-1.2.2.dist-info}/METADATA

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

{foxes-1.2.1.dist-info → foxes-1.2.2.dist-info}/RECORD

@@ -120,8 +120,8 @@ foxes/input/states/states_table.py,sha256=-y0vYGQ_OVNQUPMgU99da2FVBDEUvNUMzHN0HH
 foxes/input/states/create/__init__.py,sha256=Ocqzmr9SOefeAvGX5DgawsPXfNdGzWOZ2ssvtqEmtAo,134
 foxes/input/states/create/random_abl_states.py,sha256=7KD1q8pF0IQcv-AYq6tqw7jEd3D3Q3xSqOoF5WxMReU,3411
 foxes/input/states/create/random_timeseries.py,sha256=gJpaQ4nEXxOjI4hp3xjNcVbCsmZUm-brXUxoqDb63WE,1266
-foxes/input/yaml/__init__.py,sha256=gxYWznHt2-zf1d8m9WQP2w6RAzxO0JfQhTV-2TgbJGY,133
-foxes/input/yaml/dict.py,sha256=YczHlpmPz6iveMmJeoAtvFGixqGw-s2LEuY7jrPE65M,12293
+foxes/input/yaml/__init__.py,sha256=1EIaXSaykZ3VnhXN_SgqnWkRLVkXsL_608_-f19Xk14,167
+foxes/input/yaml/dict.py,sha256=yHkw_x81ZKQ7BlCaTchoTDDAeakfr_zNdtXZgLGZ-B8,13607
 foxes/input/yaml/yaml.py,sha256=wVn6v-4g_Cz-MuNBJhVku9UXCIJrwd9fviLbvsokUcA,2704
 foxes/input/yaml/windio/__init__.py,sha256=h5zp7VDWYq7K4GVX91CkoEABhpYSTisFWiw-0YAq3eU,196
 foxes/input/yaml/windio/get_states.py,sha256=cl4ZMurwbUyZflaoC1TPdBc5VjS7s7zp4_FYp80hkX0,6081
@@ -301,9 +301,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.1.dist-info/LICENSE,sha256=bBCH6mYTPzSepk2s2UUZ3II_ZYXrn1bnSqB85-aZHxU,1071
-foxes-1.2.1.dist-info/METADATA,sha256=xI_XIIGCyC8pOw7prX-3rqt6Te_pa9zp9KQ8jS4WcgI,8629
-foxes-1.2.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-foxes-1.2.1.dist-info/entry_points.txt,sha256=KuS44FRH5NnMw201A8Btr76eNRKr2UOoKHjejAsqKwE,123
-foxes-1.2.1.dist-info/top_level.txt,sha256=G7oHApEz5nc-iP__XsPcvjYe_NyXGmKMUMPHi3C3x6I,26
-foxes-1.2.1.dist-info/RECORD,,
+foxes-1.2.2.dist-info/LICENSE,sha256=bBCH6mYTPzSepk2s2UUZ3II_ZYXrn1bnSqB85-aZHxU,1071
+foxes-1.2.2.dist-info/METADATA,sha256=7dg2z1b6a9XZFlQCJRKBJ4sSfOkDAP4GJSGR8RLp7MA,8629
+foxes-1.2.2.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+foxes-1.2.2.dist-info/entry_points.txt,sha256=KuS44FRH5NnMw201A8Btr76eNRKr2UOoKHjejAsqKwE,123
+foxes-1.2.2.dist-info/top_level.txt,sha256=G7oHApEz5nc-iP__XsPcvjYe_NyXGmKMUMPHi3C3x6I,26
+foxes-1.2.2.dist-info/RECORD,,