# RCAIDE/Library/Plots/Performance/Aerodynamics/plot_disc_and_power_loading.py
#
#
# Created: Jul 2023, M. Clarke
# ----------------------------------------------------------------------------------------------------------------------
# IMPORT
# ----------------------------------------------------------------------------------------------------------------------
from RCAIDE.Framework.Core import Units
from RCAIDE.Library.Plots.Common import set_axes, plot_style
# python imports
import matplotlib.pyplot as plt
import matplotlib.cm as cm
import numpy as np
# ----------------------------------------------------------------------------------------------------------------------
# PLOTS
# ----------------------------------------------------------------------------------------------------------------------
[docs]
def plot_disc_and_power_loading(results,
save_figure=False,
show_legend = True,
save_filename="Disc_And_Power_Loading",
file_type = ".png",
width = 11, height = 7):
"""
Generate plots of rotor disc and power loading over time.
Parameters
----------
results : Data
Mission results data structure containing:
results.segments[i].conditions with fields:
- energy[network_tag][rotor_tag].disc_loading
- energy[network_tag][rotor_tag].power_loading
- frames.inertial.time
save_figure : bool, optional
Save figure to file if True, default False
show_legend : bool, optional
Display segment legend if True, default True
save_filename : str, optional
Name for saved figure file, default "Disc_And_Power_Loading"
file_type : str, optional
File extension for saved figure, default ".png"
width : float, optional
Figure width in inches, default 11
height : float, optional
Figure height in inches, default 7
Returns
-------
fig : matplotlib.figure.Figure
Notes
-----
Creates a figure with two vertically stacked subplots showing:
- Top: Disc loading (N/m²) vs time (minutes)
- Bottom: Power loading (N/W) vs time (minutes)
Each mission segment is plotted in a different color using the
inferno colormap. Multiple rotors/propellers are distinguished
by different markers.
**Definitions**
'Disc Loading'
Thrust per unit rotor disc area
'Power Loading'
Thrust per unit power (measure of efficiency)
See Also
--------
RCAIDE.Library.Plots.Common.set_axes : Standardized axis formatting
RCAIDE.Library.Plots.Common.plot_style : RCAIDE plot styling
RCAIDE.Library.Analysis.Performance.propulsion : Analysis modules
"""
# get plotting style
ps = plot_style()
parameters = {'axes.labelsize': ps.axis_font_size,
'xtick.labelsize': ps.axis_font_size,
'ytick.labelsize': ps.axis_font_size,
'axes.titlesize': ps.title_font_size}
plt.rcParams.update(parameters)
# get line colors for plots
line_colors = cm.inferno(np.linspace(0,0.9,len(results.segments)))
fig = plt.figure(save_filename)
fig.set_size_inches(width,height)
axis_1 = plt.subplot(2,1,1)
axis_2 = plt.subplot(2,1,2)
pi = 0
for network in results.segments[0].analyses.energy.vehicle.networks:
for p_i, propulsor in enumerate(network.propulsors):
if (p_i == 0) or (network.identical_propulsors == False):
plot_propulsor_data(results,propulsor,axis_1,axis_2,line_colors,ps,pi)
if show_legend:
leg = fig.legend(bbox_to_anchor=(0.5, 0.95), loc='upper center', ncol = 4)
leg.set_title('Flight Segment', prop={'size': ps.legend_font_size, 'weight': 'heavy'})
# Adjusting the sub-plots for legend
fig.tight_layout()
fig.subplots_adjust(top=0.8)
# set title of plot
title_text = 'Disc and Power Loading'
fig.suptitle(title_text)
if save_figure:
plt.savefig(save_filename + file_type)
return fig
[docs]
def plot_propulsor_data(results, propulsor, axis_1, axis_2, line_colors, ps, pi):
"""
Plot disc and power loading data for a single propulsor across mission segments.
Parameters
----------
results : Data
Mission results data structure containing:
results.segments[i].conditions with fields:
- energy[propulsor_tag][thrustor_tag].disc_loading
- energy[propulsor_tag][thrustor_tag].power_loading
- frames.inertial.time
propulsor : Data
Propulsor data structure containing:
- tag : str
Identifier for the propulsor
- rotor/propeller : Data
Thrustor component data
axis_1 : matplotlib.axes.Axes
Axis for disc loading plot
axis_2 : matplotlib.axes.Axes
Axis for power loading plot
line_colors : array
Array of RGB colors for different segments
ps : Data
Plot style data structure with fields:
- markers : list
Marker styles for different propulsors
- line_width : float
Width for plot lines
pi : int
Index of current propulsor for marker selection
Returns
-------
None
Notes
-----
Helper function for plot_disc_and_power_loading that handles plotting
data for a single propulsor (rotor or propeller) across all mission
segments.
The function:
- Identifies thrustor type (rotor vs propeller)
- Extracts time histories for each segment
- Plots disc and power loading vs time
- Applies consistent styling
- Adds legend for first segment only
See Also
--------
plot_disc_and_power_loading : Main plotting function
RCAIDE.Library.Plots.Common.set_axes : Axis formatting
"""
if 'rotor' in propulsor:
thrustor = propulsor.rotor
elif 'propeller' in propulsor:
thrustor = propulsor.propeller
for i in range(len(results.segments)):
time = results.segments[i].conditions.frames.inertial.time[:,0] / Units.min
DL = results.segments[i].conditions.energy.converters[thrustor.tag].disc_loading[:,0]
PL = results.segments[i].conditions.energy.converters[thrustor.tag].power_loading[:,0]
if pi == 0 and i ==0:
axis_1.plot(time,DL, color = line_colors[i], marker = ps.markers[pi], linewidth = ps.line_width, label = thrustor.tag)
else:
axis_1.plot(time,DL, color = line_colors[i], marker = ps.markers[pi], linewidth = ps.line_width)
axis_1.set_ylabel(r'Disc Loading (N/m^2)')
set_axes(axis_1)
axis_2.plot(time,PL, color = line_colors[i], marker = ps.markers[pi], linewidth = ps.line_width)
axis_2.set_xlabel('Time (mins)')
axis_2.set_ylabel(r'Power Loading (N/W)')
set_axes(axis_2)
return
