Example: Visualize rotor velocities

"""Example: Visualize rotor velocities

Demonstrate visualizing the flow velocities at the rotor using plot_rotor_values

"""

import matplotlib.pyplot as plt

import floris.flow_visualization as flowviz
from floris import FlorisModel


fmodel = FlorisModel("../inputs/gch.yaml")

# Set a 2 turbine layout
fmodel.set(
    layout_x=[0, 500],
    layout_y=[0, 0],
    wind_directions=[270],
    wind_speeds=[8],
    turbulence_intensities=[0.06],
)

# Run the model
fmodel.run()

# Plot the values at each rotor
fig, axes, _, _ = flowviz.plot_rotor_values(
    fmodel.core.flow_field.u, findex=0, n_rows=1, n_cols=2, return_fig_objects=True
)
fig.suptitle("Rotor Plane Visualization, Original Resolution")

plt.show()
import warnings
warnings.filterwarnings('ignore')