foxes 1.5__py3-none-any.whl → 1.5.1__py3-none-any.whl

examples/yawed_wake/run.py

@@ -114,93 +114,95 @@ if __name__ == "__main__":
         wake_deflection=args.deflection,
         partial_wakes=args.pwakes,
         mbook=mbook,
-        engine=args.engine,
+    )
+
+    with foxes.Engine.new(
+        engine_type=args.engine,
         n_procs=args.n_cpus,
         chunk_size_states=args.chunksize_states,
         chunk_size_points=args.chunksize_points,
-    )
+    ):
+        # calculate farm results
+        farm_results = algo.calc_farm()
+        print("\nResults data:\n", farm_results)
 
-    # calculate farm results
-    farm_results = algo.calc_farm()
-    print("\nResults data:\n", farm_results)
-
-    if not args.nofig:
-        # xy horizontal flow plot
-        print("\nHorizontal flow figure output:")
-        o = foxes.output.FlowPlots2D(algo, farm_results)
-        g = o.gen_states_fig_xy(
-            args.var, resolution=10, xmin=-500, xmax=3000, rotor_color="red"
-        )
-        fig = next(g)
-        plt.show()
-        plt.close(fig)
+        if not args.nofig:
+            # xy horizontal flow plot
+            print("\nHorizontal flow figure output:")
+            o = foxes.output.FlowPlots2D(algo, farm_results)
+            g = o.gen_states_fig_xy(
+                args.var, resolution=10, xmin=-500, xmax=3000, rotor_color="red"
+            )
+            fig = next(g)
+            plt.show()
+            plt.close(fig)
 
-        # yz flow plot
-        print("\nVertical flow figure output:")
-        o = foxes.output.FlowPlots2D(algo, farm_results)
-        g = o.gen_states_fig_yz(
-            args.var,
-            resolution=5,
-            x=750,
-            ymin=-200,
-            ymax=200,
-            zmin=0,
-            zmax=250,
-            rotor_color="red",
-            verbosity=0,
-        )
-        fig = next(g)
-        plt.show()
-        plt.close(fig)
+            # yz flow plot
+            print("\nVertical flow figure output:")
+            o = foxes.output.FlowPlots2D(algo, farm_results)
+            g = o.gen_states_fig_yz(
+                args.var,
+                resolution=5,
+                x=750,
+                ymin=-200,
+                ymax=200,
+                zmin=0,
+                zmax=250,
+                rotor_color="red",
+                verbosity=0,
+            )
+            fig = next(g)
+            plt.show()
+            plt.close(fig)
 
-    # add capacity and efficiency to farm results
-    o = foxes.output.FarmResultsEval(farm_results)
-    o.add_capacity(algo)
-    o.add_capacity(algo, ambient=True)
-    o.add_efficiency()
+        # add capacity and efficiency to farm results
+        o = foxes.output.FarmResultsEval(farm_results)
+        o.add_capacity(algo)
+        o.add_capacity(algo, ambient=True)
+        o.add_efficiency()
 
-    # state-turbine results
-    farm_df = farm_results.to_dataframe()
-    print("\nFarm results data:\n")
-    print(
-        farm_df[
-            [
-                FV.X,
-                FV.AMB_REWS,
-                FV.REWS,
-                FV.AMB_TI,
-                FV.TI,
-                FV.AMB_P,
-                FV.P,
-                FV.WD,
-                FV.YAW,
-                FV.YAWM,
+        # state-turbine results
+        farm_df = farm_results.to_dataframe()
+        print("\nFarm results data:\n")
+        print(
+            farm_df[
+                [
+                    FV.X,
+                    FV.AMB_REWS,
+                    FV.REWS,
+                    FV.AMB_TI,
+                    FV.TI,
+                    FV.AMB_P,
+                    FV.P,
+                    FV.WD,
+                    FV.YAW,
+                    FV.YAWM,
+                ]
             ]
-        ]
-    )
-    print()
+        )
+        print()
 
-    # results by turbine
-    turbine_results = o.reduce_states(
-        {
-            FV.AMB_P: "weights",
-            FV.P: "weights",
-            FV.AMB_CAP: "weights",
-            FV.CAP: "weights",
-        }
-    )
-    turbine_results[FV.AMB_YLD] = o.calc_turbine_yield(
-        algo=algo, annual=True, ambient=True
-    )
-    turbine_results[FV.YLD] = o.calc_turbine_yield(algo=algo, annual=True)
-    turbine_results[FV.EFF] = turbine_results[FV.P] / turbine_results[FV.AMB_P]
-    print("\nResults by turbine:\n")
-    print(turbine_results)
+        # results by turbine
+        turbine_results = o.reduce_states(
+            {
+                FV.AMB_P: "weights",
+                FV.P: "weights",
+                FV.AMB_CAP: "weights",
+                FV.CAP: "weights",
+            }
+        )
+        turbine_results[FV.AMB_YLD] = o.calc_turbine_yield(
+            algo=algo, annual=True, ambient=True
+        )
+        turbine_results[FV.YLD] = o.calc_turbine_yield(algo=algo, annual=True)
+        turbine_results[FV.EFF] = turbine_results[FV.P] / turbine_results[FV.AMB_P]
+        print("\nResults by turbine:\n")
+        print(turbine_results)
 
-    # power results
-    P0 = o.calc_mean_farm_power(ambient=True)
-    P = o.calc_mean_farm_power()
-    print(f"\nFarm power        : {P / 1000:.1f} MW")
-    print(f"Farm ambient power: {P0 / 1000:.1f} MW")
-    print(f"Farm efficiency   : {o.calc_farm_efficiency() * 100:.2f} %")
-    print(f"Annual farm yield : {turbine_results[FV.YLD].sum():.2f} GWh.")
+        # power results
+        P0 = o.calc_mean_farm_power(ambient=True)
+        P = o.calc_mean_farm_power()
+        print(f"\nFarm power        : {P / 1000:.1f} MW")
+        print(f"Farm ambient power: {P0 / 1000:.1f} MW")
+        print(f"Farm efficiency   : {o.calc_farm_efficiency() * 100:.2f} %")
+        print(f"Annual farm yield : {turbine_results[FV.YLD].sum():.2f} GWh.")

foxes/core/algorithm.py

@@ -912,6 +912,7 @@ class Algorithm(Model):
             isel=isel,
             **kwargs,
         )
+        self.reset_chunk_store(chunk_store)
 
         # reset to not running:
         self.unset_running(

foxes/engines/pool.py

@@ -187,6 +187,7 @@ class PoolEngine(Engine):
             The model results
 
         """
+
         # subset selection:
         model_data, farm_data, point_data = self.select_subsets(
             model_data, farm_data, point_data, sel=sel, isel=isel

foxes/utils/xarray_utils.py

@@ -31,24 +31,32 @@ def write_nc(
         Additional parameters for xarray.to_netcdf
 
     """
+
+    def _round(x, v, d):
+        """Helper function to round values"""
+        if d is not None:
+            if verbosity > 1:
+                print(f"File {fpath.name}: Rounding {v} to {d} decimals")
+            try:
+                x = x.astype(np.float32)
+            except ValueError:
+                pass
+            try:
+                return np.round(x, d)
+            except Exception:
+                pass
+        return x
+
     fpath = Path(fpath)
     if round is not None:
         crds = {}
-        for v in ds.coords.keys():
+        for v, x in ds.coords.items():
             d = round.get(v, get_default_digits(v))
-            if d is not None:
-                if verbosity > 1:
-                    print(f"File {fpath.name}: Rounding {v} to {d} decimals")
-                crds[v] = np.round(ds[v].to_numpy(), d)
-            else:
-                crds[v] = ds[v].to_numpy()
+            crds[v] = _round(x.to_numpy(), v, d)
         dvrs = {}
-        for v in ds.data_vars.keys():
+        for v, x in ds.data_vars.items():
             d = round.get(v, get_default_digits(v))
-            if d is not None:
-                if verbosity > 1:
-                    print(f"File {fpath.name}: Rounding {v} to {d} decimals")
-                dvrs[v] = (ds[v].dims, np.round(ds[v].to_numpy(), d))
+            dvrs[v] = (x.dims, _round(x.to_numpy(), v, d))
         ds = Dataset(coords=crds, data_vars=dvrs)
 
     enc = None

{foxes-1.5.dist-info → foxes-1.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: foxes
-Version: 1.5
+Version: 1.5.1
 Summary: Farm Optimization and eXtended yield Evaluation Software
 Author: Jonas Schulte
 Maintainer: Jonas Schulte

{foxes-1.5.dist-info → foxes-1.5.1.dist-info}/RECORD

@@ -1,26 +1,26 @@
 docs/source/conf.py,sha256=vJI7sGh52LTnDRfKhQFWOMxheU4Vnq7_lWqL7G4OQAM,12351
-examples/abl_states/run.py,sha256=NUyNTRzPb2mW_oiWj3NWgHb2lx-6F_CZ27g9Y9_a_to,4598
+examples/abl_states/run.py,sha256=DhFge87Sx6PK0n9BRU7vsGN-0F5V0I8ZqVTOWxwHyrM,4853
 examples/compare_rotors_pwakes/run.py,sha256=yGvSGguLcgJmYNwTUGKJDRBxRRFFRYKF4v4VLUo9Wfw,6642
 examples/compare_wakes/run.py,sha256=uS3svL5UYcttvtSW6jCUyS0_qW0-ArBkY-msYfvJSkE,7353
-examples/dyn_wakes/run.py,sha256=7L9Ao-U2TIxE8VT1ZT1rFCHrOBlsBeV1Vck0QDzH9Ak,9364
-examples/field_data_nc/run.py,sha256=fWD8h905psa1ijSuNdVk9N_oJxBDtPKtiVPCvW9BlSM,3568
+examples/dyn_wakes/run.py,sha256=OdrRgR1MXeAfwKErhGLN7bCAtd6aCAVeSSQWltQYSJ0,9689
+examples/field_data_nc/run.py,sha256=RIVoNtPM3ewhfCJLBFZ1m8F9sKbFAwS7m2JUf4sBQ1s,3671
 examples/induction/run.py,sha256=Vqrjuw1_inqseihl-8_qgp4rq-7kc2NawsKa39Ux7V4,6340
-examples/multi_height/run.py,sha256=9Gbx-FiMro1pxiMHMO8pomP8tgBbuW6w-jaJW_S5zMA,3294
+examples/multi_height/run.py,sha256=qnx2mAJZ1aY7N6wSmczjkSNcAsvQibzHh4e0Bf_mQ3o,3345
 examples/power_mask/run.py,sha256=BXR0V_9y1aebzLnj2bXIFXMHQ1veaCMAPLRAH8EMOhY,7596
 examples/quickstart/run.py,sha256=fo8A_oioPyUtNs4EiHZE1fCHH6DRuXFQzWJyHHd_oiI,427
 examples/random_timeseries/run.py,sha256=ASpLTZkG9tLVbG34vyiBK1zJgvvAfl45EhA6r6jHDxM,6111
-examples/scan_row/run.py,sha256=ti3XCL4MSyL5b0qMsbJOZfpLNOm6OiMwHd_sN5939tg,5850
-examples/sector_management/run.py,sha256=WQvrHqtC7_u6JvZfbRsVSb3SPYDjCp-ZJpC-nZsQf5M,4451
+examples/scan_row/run.py,sha256=N9UX_Fb5ZGTLEzKZsv_paHt2Wk6RAanHKfEL6liuMj8,6197
+examples/sector_management/run.py,sha256=aNT22eIsRlgOy0aN3TVWsVNc3wCfNa6keg7gfwD6hl8,4626
 examples/sequential/run.py,sha256=pd7L3obYyqNZXzDxntRKsfhqunG0SIjZslbD42WcveI,6758
 examples/single_state/run.py,sha256=VfZD5Ev4jLo6LQ7antKmZFU8SDfezeI9ERSf4ASvKsU,6162
-examples/states_lookup_table/run.py,sha256=XZkntAbHfZwozQ0GFOQiKl4yjm7v4TjC4lwSsF1Ixgs,3922
-examples/streamline_wakes/run.py,sha256=weBGjuHxZJQFOFK9Anct0UK0gwd-4dnrZIHYjiVNlqU,4698
+examples/states_lookup_table/run.py,sha256=IJu6SJOq1S9VvJIv785hJVXn7zpjDRp41Cj-mN7ZQZ4,3965
+examples/streamline_wakes/run.py,sha256=mzuxLGzajazIGfzWDCzDU2zRUODQmstYUlqzK46aA-U,4749
 examples/tab_file/run.py,sha256=xe4VUwyl3eYMUGtyS7kuieGeoYo9mEq_ZvQ6E1rZ1uE,4370
-examples/timelines/run.py,sha256=ca1koiA6qX8LVrAGrYMJRJY7xChwwjHndP6AB5LNlqA,8261
-examples/timeseries/run.py,sha256=QN6xXXUOZFh12FBZ6to_WC1rnLkCzebay2ORiLp1X10,5399
+examples/timelines/run.py,sha256=Tc37gtvV9q5ALoCuT5p1N8m2X7J8UK0E7qXn0YZunpA,8688
+examples/timeseries/run.py,sha256=3aAf_qC4IlEr_QhMUoZIV5A4K_jnnizy5H9-HBFTdjE,5571
 examples/timeseries_slurm/run.py,sha256=4yHi4SXr7rff4tbJNQEUa9Dd7Dvli2XrJExu-xdE2Us,5587
-examples/wind_rose/run.py,sha256=QeV6tbJcJcDD8vC1kQqRgyyeRjACzTmYkedLQ9eqYe0,4253
-examples/yawed_wake/run.py,sha256=msw7_xkxH45gegvUn9_0-lAB7D1gpmzvtMDd6kGKfVo,6531
+examples/wind_rose/run.py,sha256=CU7pzOSlWHifFvdtKl_t9gufaAKw2TkjwvklDZRaawM,4380
+examples/yawed_wake/run.py,sha256=X-cxmhC0rCRHyB00kXrsnqQ9fufSY7L35lozls7Nxuw,6882
 foxes/__init__.py,sha256=1RPU7w3WJ3jFb2aJdZKRy-6Fje9qAaH85WYcOCiDvwk,1375
 foxes/constants.py,sha256=jQcyeU53V629Tf8qQWc2F4Xw1rV3ywtJiT-g1JWenNg,3842
 foxes/variables.py,sha256=sK58u3HJRsWo-fbE7GPlQUhF0YUhY-qSg6Vl8Pv1Eo8,6144
@@ -48,7 +48,7 @@ foxes/algorithms/sequential/models/seq_state.py,sha256=al7XhLKLQSGhfUZw--Bz4RNNb
 foxes/config/__init__.py,sha256=M2FrI1gO6SZUUPsQe1KL_-JNyUUt_Lk4nOw7cyNpHew,223
 foxes/config/config.py,sha256=e3HhitD2Xleqm3utoXLGAu8Vg16app35Cpx4nBhnZm8,4318
 foxes/core/__init__.py,sha256=iMuUuPsFERvjY2PQa7PCEsxrZYRJ8IORJs3k3OoerFw,1985
-foxes/core/algorithm.py,sha256=xRED8Y7c3xqr3Biurf5jJrZpbpcF-Xq0sEBPikSOFZs,27450
+foxes/core/algorithm.py,sha256=EZPXOXAS-IjhWHatxZO-tvQch4yPMIPkdDHWtpATEmQ,27494
 foxes/core/axial_induction_model.py,sha256=sGbTHFMjEYKiVjElLP_SCRulKatcK0wsEXf30zty2Vc,1023
 foxes/core/data.py,sha256=sYkL-kZnWU9wdKkbSS85KS66BMUXToloCNkWGPE7C3s,25280
 foxes/core/data_calc_model.py,sha256=c96fv7HrLK29WD5SRc4WFXjT24lNImnLflQJvIMACn8,1488
@@ -103,7 +103,7 @@ foxes/engines/futures.py,sha256=_VQXNLb0AxTtTouy96LhBnwPldHiG0xldHjpChKyNxs,1503
 foxes/engines/mpi.py,sha256=XR9xFX6Spw8OKQ7x4qYOSeUlxcFWkwndNRiQ3IomkdM,650
 foxes/engines/multiprocess.py,sha256=X4S-AnhnYuONpfZ3pKWi4VH9mm68ZMNm43ELPoeb-Ek,1328
 foxes/engines/numpy.py,sha256=yljXIiVZgQjw97jmhpz_vKOh1TW3RxSI4SxN7rQnC4M,5806
-foxes/engines/pool.py,sha256=CFKwrmwWXNqKzwRXNYsvChSiKwiIi0pzQvwwGT8ysSY,8708
+foxes/engines/pool.py,sha256=SxJCH45mqFSwvq9YtLDoamu_5AWnVPH-VDZkDI5Tfiw,8709
 foxes/engines/ray.py,sha256=Otlc1ZfoTiNE6sq9rq1y212tVbaVDfxryqAl4T0AM_A,1622
 foxes/engines/single.py,sha256=Z0tun1ZKQD-LMdSK291AUagBDJiHnexJL5ysrkP04tY,4725
 foxes/input/__init__.py,sha256=JodEdc77y5I-CsgfT5HadMK7DH5EKcr9uGLEVcgM-wY,157
@@ -282,7 +282,7 @@ foxes/utils/two_circles.py,sha256=xkj-SA_x-VXY7KtmSU4lcV4gFdplyhU3sBAC9vTdkF4,28
 foxes/utils/weibull.py,sha256=2fUWt-vZVMu55fFrXuo0M3B771xtoWeVVmqE7Nr5yXM,656
 foxes/utils/wind_dir.py,sha256=6W0njWDvnIdOOjwqcMr64MW9ApjdtFA75blVUxirPMo,2823
 foxes/utils/wrg_utils.py,sha256=QzaASvn50Pzegk2NoCJhC9WozO95KiQPIwiWGNMeA4w,3604
-foxes/utils/xarray_utils.py,sha256=pqtBkCzSXJMlTM0VTI-Su4SGxOP57sYVlZORBr7Z2qw,1880
+foxes/utils/xarray_utils.py,sha256=ddnjrZ6HLMqIE1JpP_yCx4tKeSDjNRsSI6c5CL2neJo,1961
 foxes/utils/abl/__init__.py,sha256=yDbyZodXZ_dsmIJ4QH981CZ7xvVQyLneGzjOgYB7ZNE,179
 foxes/utils/abl/neutral.py,sha256=E4DEhvXvw74BPrYr1MjQjeIaoz6ZOTWVlqScKflm-0M,1358
 foxes/utils/abl/sheared.py,sha256=nS6gxa5cR7B7XcZbEk3rwOvbF6m4cinXAIhhn9gGBBs,446
@@ -295,7 +295,7 @@ foxes/utils/geom2d/example_intersection.py,sha256=4e6sjpZEk_bNc462YvwKPzwxdV1B90
 foxes/utils/geom2d/example_union.py,sha256=BKfLt1mtQcSto-qExeMQkq8tQ6kfFXVJ93Cc7DhOal8,1750
 foxes/utils/geom2d/half_plane.py,sha256=kzZD6pkZxZ03MK9WAboWzXb5Ws5dWLQY9GIahD4D9mA,6167
 foxes/utils/geom2d/polygon.py,sha256=Xj7triA5Pe4-48sNSAvGxEXlQGptV161LUpKKCf3YOY,5535
-foxes-1.5.dist-info/licenses/LICENSE,sha256=bBCH6mYTPzSepk2s2UUZ3II_ZYXrn1bnSqB85-aZHxU,1071
+foxes-1.5.1.dist-info/licenses/LICENSE,sha256=bBCH6mYTPzSepk2s2UUZ3II_ZYXrn1bnSqB85-aZHxU,1071
 tests/0_consistency/iterative/test_iterative.py,sha256=lzagmJ1y241l6Szw4Cu80S8S1ATHIyD7ukr1vVBrusY,2637
 tests/0_consistency/partial_wakes/test_partial_wakes.py,sha256=7rdg2lcUYUzxwCdGhue7A4QYCJQGrOO4E0bytgatYj4,2584
 tests/1_verification/flappy_0_6/PCt_files/test_PCt_files.py,sha256=8tIqifsykvxMCd9eD-mZWxRwT56sKjIxrGf3hf8aCFA,2530
@@ -321,8 +321,8 @@ 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=QUULW3AJJq8zvUIdIQHkXiNZpaysVYIALh5HOJ6-428,2595
 tests/1_verification/flappy_0_6_2/row_Bastankhah_linear_centre/flappy/run.py,sha256=1paRjCe71AIM8igK4_f0FFH1xiIETvuBwnZAATiwWUU,2102
 tests/3_examples/test_examples.py,sha256=9_PtvlYA6ELsQGHt-xAnsjCYdGj1ekX237ymWvEJaCk,732
-foxes-1.5.dist-info/METADATA,sha256=3SO55BIVmGwr2wJ-TWDMqSdqAQ9vJgzJOXRZHrQ0C1Y,8121
-foxes-1.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-foxes-1.5.dist-info/entry_points.txt,sha256=KuS44FRH5NnMw201A8Btr76eNRKr2UOoKHjejAsqKwE,123
-foxes-1.5.dist-info/top_level.txt,sha256=G7oHApEz5nc-iP__XsPcvjYe_NyXGmKMUMPHi3C3x6I,26
-foxes-1.5.dist-info/RECORD,,
+foxes-1.5.1.dist-info/METADATA,sha256=zRir9XRuTKME-eYx3K7yLKrD9wqWTah5-PTPWIKvEjw,8123
+foxes-1.5.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+foxes-1.5.1.dist-info/entry_points.txt,sha256=KuS44FRH5NnMw201A8Btr76eNRKr2UOoKHjejAsqKwE,123
+foxes-1.5.1.dist-info/top_level.txt,sha256=G7oHApEz5nc-iP__XsPcvjYe_NyXGmKMUMPHi3C3x6I,26
+foxes-1.5.1.dist-info/RECORD,,