foxes 0.8.2__py3-none-any.whl → 1.1.0.2__py3-none-any.whl

examples/dyn_wakes/run.py

@@ -0,0 +1,311 @@
+import time
+import argparse
+import numpy as np
+import pandas as pd
+import matplotlib.pyplot as plt
+from pathlib import Path
+
+import foxes
+import foxes.variables as FV
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "-a", "--animation", help="Write flow animation file", action="store_true"
+    )
+    parser.add_argument(
+        "-A",
+        "--ani_file",
+        help="Path to the animation file to be written",
+        default="ani.gif",
+    )
+    parser.add_argument(
+        "-F",
+        "--fps",
+        help="The frames per second value for the animation",
+        type=int,
+        default=4,
+    )
+    parser.add_argument(
+        "-b0",
+        "--background0",
+        help="Switch off dynamic background interpretation",
+        action="store_true",
+    )
+    parser.add_argument(
+        "-ws0", "--fix_ws", help="Switch off dynamic wind speed", action="store_true"
+    )
+    parser.add_argument(
+        "-S",
+        "--max_state",
+        help="States subset to the first n states",
+        type=int,
+        default=None,
+    )
+    parser.add_argument(
+        "-R",
+        "--ref_xy",
+        help="The reference point x y",
+        type=float,
+        default=[2500, 2500],
+        nargs="+",
+    )
+    parser.add_argument(
+        "-nt", "--n_turbines", help="The number of turbines", default=9, type=int
+    )
+    parser.add_argument(
+        "-mit", "--max_it", help="Run until maximal iteration", default=None, type=int
+    )
+    parser.add_argument(
+        "-s",
+        "--states",
+        help="The timeseries input file (path or static)",
+        default="timeseries_100.csv.gz",
+    )
+    parser.add_argument(
+        "-t",
+        "--turbine_file",
+        help="The P-ct-curve csv file (path or static)",
+        default="NREL-5MW-D126-H90.csv",
+    )
+    parser.add_argument("-r", "--rotor", help="The rotor model", default="centre")
+    parser.add_argument(
+        "-p", "--pwakes", help="The partial wakes models", default=None, nargs="+"
+    )
+    parser.add_argument("-sc", "--scheduler", help="The scheduler choice", default=None)
+    parser.add_argument(
+        "-w",
+        "--wakes",
+        help="The wake models",
+        default=["Bastankhah2014_linear_loc_k004"],
+        nargs="+",
+    )
+    parser.add_argument("-f", "--frame", help="The wake frame", default="dyn_wakes")
+    parser.add_argument(
+        "-m", "--tmodels", help="The turbine models", default=[], nargs="+"
+    )
+    parser.add_argument(
+        "-sl",
+        "--show_layout",
+        help="Flag for showing layout figure",
+        action="store_true",
+    )
+    parser.add_argument("-e", "--engine", help="The engine", default="process")
+    parser.add_argument(
+        "-n", "--n_cpus", help="The number of cpus", default=None, type=int
+    )
+    parser.add_argument(
+        "-c",
+        "--chunksize_states",
+        help="The chunk size for states",
+        default=None,
+        type=int,
+    )
+    parser.add_argument(
+        "-C",
+        "--chunksize_points",
+        help="The chunk size for points",
+        default=5000,
+        type=int,
+    )
+    parser.add_argument(
+        "-nf", "--nofig", help="Do not show figures", action="store_true"
+    )
+    args = parser.parse_args()
+
+    mbook = foxes.models.ModelBook()
+    ttype = foxes.models.turbine_types.PCtFile(args.turbine_file)
+    mbook.turbine_types[ttype.name] = ttype
+    mbook.wake_frames["dyn_wakes"] = foxes.models.wake_frames.DynamicWakes(
+        max_length_km=8
+    )
+
+    if args.background0:
+        States = foxes.input.states.Timeseries
+        kwargs = {}
+    else:
+        States = foxes.input.states.OnePointFlowTimeseries
+        kwargs = {"ref_xy": args.ref_xy}
+
+    sdata = pd.read_csv(
+        foxes.StaticData().get_file_path(foxes.STATES, args.states),
+        index_col=0,
+        parse_dates=[0],
+    )
+    if not args.fix_ws:
+        n_times = len(sdata.index)
+        sdata["ws"] = 5 + 0.3 * np.sin(np.arange(n_times) * 2 * np.pi / 20)
+
+    states = States(
+        data_source=sdata,
+        output_vars=[FV.WS, FV.WD, FV.TI, FV.RHO],
+        var2col={FV.WS: "ws", FV.WD: "wd", FV.TI: "ti"},
+        fixed_vars={FV.RHO: 1.225, FV.TI: 0.07},
+        states_sel=range(args.max_state) if args.max_state is not None else None,
+        **kwargs,
+    )
+
+    farm = foxes.WindFarm()
+    N = int(args.n_turbines**0.5)
+    foxes.input.farm_layout.add_grid(
+        farm,
+        xy_base=np.array([0.0, 0.0]),
+        step_vectors=np.array([[1000.0, 0], [0, 800.0]]),
+        steps=(N, N),
+        turbine_models=args.tmodels + [ttype.name],
+    )
+
+    if not args.nofig and args.show_layout:
+        ax = foxes.output.FarmLayoutOutput(farm).get_figure()
+        plt.show()
+        plt.close(ax.get_figure())
+
+    engine = foxes.Engine.new(
+        engine_type=args.engine,
+        n_procs=args.n_cpus,
+        chunk_size_states=args.chunksize_states,
+        chunk_size_points=args.chunksize_points,
+    )
+    engine.initialize()
+
+    algo = foxes.algorithms.Iterative(
+        farm,
+        states=states,
+        rotor_model=args.rotor,
+        wake_models=args.wakes,
+        wake_frame=args.frame,
+        partial_wakes=args.pwakes,
+        mbook=mbook,
+        max_it=args.max_it,
+        conv_crit="default" if args.max_it is None else None,
+        verbosity=1,
+    )
+
+    time0 = time.time()
+    farm_results = algo.calc_farm()
+    time1 = time.time()
+
+    print("\nCalc time =", time1 - time0, "\n")
+
+    o = foxes.output.FarmResultsEval(farm_results)
+    o.add_capacity(algo)
+    o.add_capacity(algo, ambient=True)
+    o.add_efficiency()
+
+    print("\nFarm results:\n")
+    print(farm_results)
+
+    # state-turbine results
+    farm_df = farm_results.to_dataframe()
+    print("\nFarm results data:\n")
+    print(
+        farm_df[
+            [
+                FV.X,
+                FV.Y,
+                FV.WD,
+                FV.AMB_REWS,
+                FV.REWS,
+                FV.AMB_TI,
+                FV.TI,
+                FV.AMB_P,
+                FV.P,
+                FV.EFF,
+            ]
+        ]
+    )
+    print()
+    print(farm_df[[FV.AMB_REWS, FV.REWS, FV.CT, FV.EFF]].describe())
+
+    # 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} %")
+
+    engine.finalize()
+
+    if not args.nofig:
+        sts = np.arange(farm_results.sizes["state"])
+        plt.plot(sts, farm_results.REWS[:, 1], label="Turbine 1")
+        plt.plot(sts, farm_results.REWS[:, 4], label="Turbine 4")
+        plt.plot(sts, farm_results.REWS[:, 7], label="Turbine 7")
+        plt.legend()
+        plt.xlabel("State")
+        plt.ylabel("REWS [m/s]")
+        plt.show()
+        plt.close()
+
+    if not args.nofig and args.animation:
+        print("\nCalculating animation")
+
+        fig, axs = plt.subplots(
+            2, 1, figsize=(5.2, 7), gridspec_kw={"height_ratios": [3, 1]}
+        )
+
+        engine.initialize()
+
+        of = foxes.output.FlowPlots2D(algo, farm_results)
+        ofg = of.gen_states_fig_xy(
+            FV.WS,
+            resolution=30,
+            quiver_pars=dict(angles="xy", scale_units="xy", scale=0.013),
+            quiver_n=35,
+            xmax=5000,
+            ymax=5000,
+            vmin=0,
+            fig=fig,
+            ax=axs[0],
+            ret_im=True,
+            title=None,
+            animated=True,
+            precalc=True,
+        )
+        next(ofg)
+
+        engine.finalize()
+
+        anim = foxes.output.Animator(fig)
+        anim.add_generator(ofg)
+        anim.add_generator(
+            o.gen_stdata(
+                turbines=[1, 4, 7],
+                variable=FV.REWS,
+                fig=fig,
+                ax=axs[1],
+                ret_im=True,
+                legloc="upper right",
+                animated=True,
+            )
+        )
+
+        ani = anim.animate()
+
+        lo = foxes.output.FarmLayoutOutput(farm)
+        lo.get_figure(
+            fig=fig,
+            ax=axs[0],
+            title="",
+            annotate=1,
+            anno_delx=-120,
+            anno_dely=-60,
+            alpha=0,
+        )
+
+        axs[0].scatter(
+            [args.ref_xy[0]],
+            [args.ref_xy[1]],
+            marker="x",
+            color="red",
+            s=80,
+            animated=True,
+        )
+
+        fpath = Path(args.ani_file)
+        print("Writing file", fpath)
+        if fpath.suffix == ".gif":
+            ani.save(filename=fpath, writer="pillow", fps=args.fps)
+        else:
+            ani.save(filename=fpath, writer="ffmpeg", fps=args.fps)

examples/field_data_nc/run.py

@@ -0,0 +1,121 @@
+import time
+import argparse
+import matplotlib.pyplot as plt
+import numpy as np
+
+import foxes
+import foxes.variables as FV
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "-f",
+        "--file_pattern",
+        help="The search pattern for input *.nc files",
+        default="data/data_*.nc",
+    )
+    parser.add_argument(
+        "-t",
+        "--turbine_file",
+        help="The P-ct-curve csv file (path or static)",
+        default="NREL-5MW-D126-H90.csv",
+    )
+    parser.add_argument("-r", "--rotor", help="The rotor model", default="centre")
+    parser.add_argument(
+        "-p", "--pwakes", help="The partial wakes models", default="centre", nargs="+"
+    )
+    parser.add_argument(
+        "-w",
+        "--wakes",
+        help="The wake models",
+        default=["Jensen_linear_k007"],
+        nargs="+",
+    )
+    parser.add_argument(
+        "-wf", "--wake_frame", help="The wake frame choice", default="rotor_wd"
+    )
+    parser.add_argument(
+        "-m", "--tmodels", help="The turbine models", default=[], nargs="+"
+    )
+    parser.add_argument(
+        "-nt", "--n_turbines", help="The number of turbines", default=9, type=int
+    )
+    parser.add_argument("-e", "--engine", help="The engine", default="ProcessEngine")
+    parser.add_argument(
+        "-n", "--n_cpus", help="The number of cpus", default=None, type=int
+    )
+    parser.add_argument(
+        "-c",
+        "--chunksize_states",
+        help="The chunk size for states",
+        default=None,
+        type=int,
+    )
+    parser.add_argument(
+        "-C",
+        "--chunksize_points",
+        help="The chunk size for points",
+        default=None,
+        type=int,
+    )
+    parser.add_argument(
+        "-npl", "--no_pre_load", help="Do not pre-load data", action="store_true"
+    )
+    parser.add_argument(
+        "-nf", "--nofig", help="Do not show figures", action="store_true"
+    )
+    args = parser.parse_args()
+
+    states = foxes.input.states.FieldDataNC(
+        args.file_pattern,
+        output_vars=[FV.WS, FV.WD, FV.TI, FV.RHO],
+        # var2ncvar={FV.WS: "ws", FV.WD: "wd", FV.TI: "ti"},
+        fixed_vars={FV.RHO: 1.225},
+        pre_load=not args.no_pre_load,
+    )
+
+    mbook = foxes.models.ModelBook()
+    ttype = foxes.models.turbine_types.PCtFile(args.turbine_file)
+    mbook.turbine_types[ttype.name] = ttype
+
+    farm = foxes.WindFarm()
+    N = int(args.n_turbines**0.5)
+    foxes.input.farm_layout.add_grid(
+        farm,
+        xy_base=np.array([500.0, 500.0]),
+        step_vectors=np.array([[500.0, 0], [0, 500.0]]),
+        steps=(N, N),
+        turbine_models=args.tmodels + [ttype.name],
+    )
+
+    algo = foxes.algorithms.Downwind(
+        farm,
+        states,
+        wake_models=args.wakes,
+        rotor_model=args.rotor,
+        wake_frame=args.wake_frame,
+        partial_wakes=args.pwakes,
+        mbook=mbook,
+        engine=args.engine,
+        n_procs=args.n_cpus,
+        chunk_size_states=args.chunksize_states,
+        chunk_size_points=args.chunksize_points,
+        verbosity=1,
+    )
+
+    time0 = time.time()
+    farm_results = algo.calc_farm()
+    time1 = time.time()
+
+    print("\nCalc time =", time1 - time0, "\n")
+
+    print(farm_results)
+
+    fr = farm_results.to_dataframe()
+    print(fr[[FV.WD, FV.AMB_REWS, FV.REWS, FV.AMB_P, FV.P]])
+
+    if not args.nofig:
+        o = foxes.output.FlowPlots2D(algo, farm_results)
+        o.get_mean_fig_xy(FV.WS, resolution=10)
+        plt.show()

examples/induction/run.py

@@ -0,0 +1,201 @@
+import numpy as np
+import argparse
+import matplotlib.pyplot as plt
+
+import foxes
+import foxes.variables as FV
+
+if __name__ == "__main__":
+    # define arguments and options:
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--ws", help="The wind speed", type=float, default=9.0)
+    parser.add_argument("--wd", help="The wind direction", type=float, default=270.0)
+    parser.add_argument("--ti", help="The TI value", type=float, default=0.08)
+    parser.add_argument("--rho", help="The air density", type=float, default=1.225)
+    parser.add_argument(
+        "-dx", help="The turbine spacing in x", type=float, default=800.0
+    )
+    parser.add_argument(
+        "-dy", help="The turbine spacing in y", type=float, default=400.0
+    )
+    parser.add_argument(
+        "-nx", help="The number of turbines in x direction", type=int, default=6
+    )
+    parser.add_argument(
+        "-ny", help="The number of turbines in y direction", type=int, default=10
+    )
+    parser.add_argument(
+        "-t",
+        "--turbine_file",
+        help="The P-ct-curve csv file (path or static)",
+        default="NREL-5MW-D126-H90.csv",
+    )
+    parser.add_argument(
+        "-m", "--tmodels", help="The turbine models", default=[], nargs="+"
+    )
+    parser.add_argument(
+        "-w",
+        "--wakes",
+        help="The wake models",
+        default=["RHB", "Bastankhah2014_linear_lim_k004"],
+        nargs="+",
+    )
+    parser.add_argument("-r", "--rotor", help="The rotor model", default="centre")
+    parser.add_argument(
+        "-p", "--pwakes", help="The partial wakes models", default=None, nargs="+"
+    )
+    parser.add_argument("-f", "--frame", help="The wake frame", default="rotor_wd")
+    parser.add_argument("-v", "--var", help="The plot variable", default=FV.WS)
+    parser.add_argument("-e", "--engine", help="The engine", default=None)
+    parser.add_argument(
+        "-nit",
+        "--not_iterative",
+        help="Don't use the iterative algorithm",
+        action="store_true",
+    )
+    parser.add_argument(
+        "-nf", "--nofig", help="Do not show figures", action="store_true"
+    )
+    args = parser.parse_args()
+
+    # create model book
+    mbook = foxes.ModelBook()
+    ttype = foxes.models.turbine_types.PCtFile(args.turbine_file)
+    mbook.turbine_types[ttype.name] = ttype
+
+    # create states
+    states = foxes.input.states.SingleStateStates(
+        ws=args.ws, wd=args.wd, ti=args.ti, rho=args.rho
+    )
+
+    # create wind farm
+    print("\nCreating wind farm")
+    farm = foxes.WindFarm()
+    foxes.input.farm_layout.add_grid(
+        farm=farm,
+        xy_base=[0.0, 0.0],
+        step_vectors=[[args.dx, 0.0], [0.0, args.dy]],
+        steps=[args.nx, args.ny],
+        turbine_models=["DTU10MW"],
+        verbosity=0,
+    )
+
+    # create algorithm
+    Algo = (
+        foxes.algorithms.Downwind if args.not_iterative else foxes.algorithms.Iterative
+    )
+    algo = Algo(
+        farm,
+        states,
+        rotor_model=args.rotor,
+        wake_models=args.wakes,
+        wake_frame=args.frame,
+        partial_wakes=args.pwakes,
+        mbook=mbook,
+        engine=args.engine,
+        verbosity=1,
+    )
+
+    # calculate farm results
+    farm_results = algo.calc_farm()
+    print("\nResults data:\n", farm_results)
+
+    # 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.WD,
+                FV.AMB_REWS,
+                FV.REWS,
+                FV.AMB_TI,
+                FV.TI,
+                FV.AMB_P,
+                FV.P,
+                FV.CT,
+                FV.EFF,
+            ]
+        ]
+    )
+    print()
+
+    # results by turbine
+    turbine_results = o.reduce_states(
+        {
+            FV.AMB_P: "mean",
+            FV.P: "mean",
+            FV.AMB_CAP: "mean",
+            FV.CAP: "mean",
+            FV.EFF: "mean",
+        }
+    )
+    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)
+    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.")
+
+    # horizontal flow plot
+    if not args.nofig:
+        o = foxes.output.FlowPlots2D(algo, farm_results)
+        xmin = -2000
+        xmax = (args.nx - 1) * args.dx + 2000
+        g = o.gen_states_fig_xy(
+            args.var,
+            figsize=(7, 5),
+            resolution=20,
+            xmin=xmin,
+            xmax=xmax,
+            yspace=1000.0,
+            levels=40,
+            rotor_color="black",
+        )
+        fig = next(g)
+        plt.show()
+        plt.close(fig)
+
+        # center line plot:
+        H = mbook.turbine_types[ttype.name].H
+        n_points = 10000
+        points = np.zeros((1, n_points, 3))
+        points[:, :, 0] = np.linspace(xmin, xmax, n_points)[None, :]
+        points[:, :, 2] = H
+        point_results = algo.calc_points(farm_results, points)
+        fig, ax = plt.subplots(figsize=(10, 5))
+        ax.plot(points[0, :, 0], point_results[FV.WS][0, :])
+        ax.set_xlabel("x [m]")
+        ax.set_ylabel("Wind speed [m/s]")
+        plt.show()
+        plt.close(fig)
+
+        # front line plot:
+        points = np.zeros((1, n_points, 3))
+        points[:, :, 0] = -200
+        points[:, :, 1] = np.linspace(-500.0, args.ny * args.dy + 500, n_points)[
+            None, :
+        ]
+        points[:, :, 2] = H
+        point_results = algo.calc_points(farm_results, points)
+        fig, ax = plt.subplots(figsize=(10, 5))
+        ax.plot(point_results[FV.WS][0, :], points[0, :, 1])
+        ax.set_ylabel("y [m]")
+        ax.set_xlabel("Wind speed [m/s]")
+        plt.show()
+        plt.close(fig)