RCAIDE-GUI 1.0.0__py3-none-any.whl

tabs/solve/plots/aerodynamics/plot_aerodynamic_forces.py

@@ -0,0 +1,155 @@
+# RCAIDE_GUI/tabs/solve/plots/aerodynamics/plot_aerodynamic_forces.py
+
+# RCAIDE imports
+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
+
+# ------------------------------------------------------------------------------
+# Plot Aerodynamic Forces
+# ------------------------------------------------------------------------------
+def plot_aerodynamic_forces(results,
+                             save_figure = False,
+                             show_legend = True,
+                             save_filename = "Aerodynamic_Forces",
+                             file_type = ".png",
+                             width = 11, height = 7):
+    """
+    Creates a multi-panel visualization of aerodynamic forces throughout flight.
+
+    Parameters
+    ----------
+    results : Results
+        RCAIDE results data structure containing:
+            - segments[i].conditions.frames
+                Frame data containing:
+                    - body.thrust_force_vector[:,0]
+                        Thrust force in body frame [N]
+                    - wind.force_vector[:,0]
+                        Drag force in wind frame [N]
+                    - wind.force_vector[:,2]
+                        Lift force in wind frame [N]
+            - segments[i].conditions.energy
+                Energy data containing:
+                    - power[:,0]
+                        Total power in watts
+            - segments[i].conditions.frames.inertial.time[:,0]
+                Time history for each segment
+            - segments[i].tag
+                Segment identifier string
+            
+    save_figure : bool, optional
+        Flag for saving the figure (default: False)
+        
+    show_legend : bool, optional
+        Flag to display segment legend (default: True)
+        
+    save_filename : str, optional
+        Name of file for saved figure (default: "Aerodynamic_Forces")
+        
+    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 visualization showing:
+        - Aerodynamic force magnitudes
+        - Power requirements
+        - Force balance evolution
+        - Time history of forces
+    
+    **Definitions**
+    
+    'Lift'
+        Force perpendicular to airflow
+    'Drag'
+        Force parallel to airflow
+    'Thrust'
+        Propulsive force
+    'Power'
+        Rate of energy use
+    
+    See Also
+    --------
+    RCAIDE.Library.Plots.Aerodynamics.plot_aerodynamic_coefficients : Non-dimensional force plots
+    RCAIDE.Library.Plots.Aerodynamics.plot_drag_components : Drag breakdown analysis
+    """ 
+
+    # 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)
+    
+    for i in range(len(results.segments)): 
+        time   = results.segments[i].conditions.frames.inertial.time[:,0] / Units.min
+        Power  = results.segments[i].conditions.energy.power[:,0] 
+        Thrust = results.segments[i].conditions.frames.body.thrust_force_vector[:,0]
+        Lift   = -results.segments[i].conditions.frames.wind.force_vector[:,2]
+        Drag   = -results.segments[i].conditions.frames.wind.force_vector[:,0]
+         
+        segment_tag  =  results.segments[i].tag
+        segment_name = segment_tag.replace('_', ' ')
+        
+        # power 
+        axis_1 = plt.subplot(2,2,1)
+        axis_1.set_ylabel(r'Power (MW)')
+        axis_1.plot(time,Power/1E6, color = line_colors[i], marker = ps.markers[0], linewidth = ps.line_width, label = segment_name) 
+        set_axes(axis_1)                
+        
+        axis_2 = plt.subplot(2,2,2)
+        axis_2.plot(time, Thrust/1000, color = line_colors[i], marker = ps.markers[0], linewidth = ps.line_width) 
+        axis_2.set_ylabel(r'Thrust (kN)')
+        set_axes(axis_2) 
+
+        axis_3 = plt.subplot(2,2,3)
+        axis_3.plot(time, Lift/1000, color = line_colors[i], marker = ps.markers[0], linewidth = ps.line_width)
+        axis_3.set_xlabel('Time (mins)')
+        axis_3.set_ylabel(r'Lift (kN)')
+        set_axes(axis_3) 
+        
+        axis_4 = plt.subplot(2,2,4)
+        axis_4.plot(time,Drag/1000 , color = line_colors[i], marker = ps.markers[0], linewidth = ps.line_width)
+        axis_4.set_xlabel('Time (mins)')
+        axis_4.set_ylabel(r'Drag (kN)')
+        set_axes(axis_4)  
+        
+ 
+    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    = 'Aerodynamic Forces'      
+    fig.suptitle(title_text)
+    
+    if save_figure:
+        plt.savefig(save_filename + file_type)   
+    return fig 

tabs/solve/plots/aerodynamics/plot_aircraft_aerodynamics.py

@@ -0,0 +1,136 @@
+# RCAIDE_GUI/tabs/solve/plots/aerodynamics/plot_aircraft_aerodynamics.py
+
+# RCAIDE imports
+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
+
+# ------------------------------------------------------------------------------
+# Plot Aircraft Aerodynamics
+# ------------------------------------------------------------------------------
+def plot_aircraft_aerodynamics(results,
+                            save_figure = False,
+                            line_color = 'bo-',
+                            line_color2 = 'rs--',
+                            save_filename = "Aircraft_Aerodynamic_Analysis",
+                            file_type = ".png",
+                            width = 11, height = 7):
+    """
+    Creates 3D surface plots of aircraft lift and drag coefficients as functions of 
+    Mach number and angle of attack.
+
+    Parameters
+    ----------
+    results : Results
+        RCAIDE results data structure containing:
+            - Mach : array
+                Array of Mach numbers
+            - alpha : array
+                Array of angles of attack in radians
+            - lift_coefficient : array
+                2D array of lift coefficients [Mach, alpha]
+            - drag_coefficient : array
+                2D array of drag coefficients [Mach, alpha]
+            
+    save_figure : bool, optional
+        Flag for saving the figure (default: False)
+        
+    line_color : str, optional
+        Color and style specification for lift plot (default: 'bo-')
+        
+    line_color2 : str, optional
+        Color and style specification for drag plot (default: 'rs--')
+        
+    save_filename : str, optional
+        Name of file for saved figure (default: "Aircraft_Aerodynamic_Analysis")
+        
+    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
+    -------
+    None
+
+
+    Notes
+    -----
+    Creates visualization showing:
+        - Aerodynamic coefficient variations
+        - Mach number effects
+        - Angle of attack sensitivity
+        - Nonlinear aerodynamic behavior
+        
+    Function displays and optionally saves two 3D surface plots:
+        Left Panel:
+            - Lift coefficient surface
+            - Shows CL variation with Mach and alpha
+            - X-axis: Mach number
+            - Y-axis: Angle of attack [degrees]
+            - Z-axis: Lift coefficient
+            
+        Right Panel:
+            - Drag coefficient surface
+            - Shows CD variation with Mach and alpha
+            - X-axis: Mach number
+            - Y-axis: Angle of attack [degrees]
+            - Z-axis: Drag coefficient
+    
+    **Definitions**
+    
+    'Mach Number'
+        Ratio of airspeed to speed of sound
+    'Angle of Attack'
+        Angle between airflow and reference line
+    'Lift Coefficient'
+        Non-dimensional lift force
+    'Drag Coefficient'
+        Non-dimensional drag force
+    
+    See Also
+    --------
+    plot_aerodynamic_coefficients : Time history of coefficients
+    plot_drag_components : Drag breakdown analysis
+    """
+    
+
+    # 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)
+    
+    #------------------------------------------------------------------------
+    # setup figures
+    #------------------------------------------------------------------------
+    fig = plt.figure()  
+    fig.set_size_inches(12,6) 
+    axis_1 = fig.add_subplot(1, 2, 1)
+    axis_2 = fig.add_subplot(1, 2, 2) 
+  
+    axis_1.plot( results.alpha/Units.degree, results.lift_coefficient) 
+    axis_2.plot( results.alpha/Units.degree, results.drag_coefficient) 
+            
+    axis_1.set_xlabel('AoA') 
+    axis_2.set_xlabel('AoA')  
+    axis_1.set_ylabel('$C_L$') 
+    axis_2.set_ylabel('$C_D$')   
+    
+    plt.tight_layout()    
+    if save_figure:    
+        fig.savefig(save_filename + file_type) 
+    
+    plt.tight_layout()
+    return

tabs/solve/plots/aerodynamics/plot_drag_components.py

@@ -0,0 +1,157 @@
+# RCAIDE_GUI/tabs/solve/plots/aerodynamics/plot_drag_components.py
+
+# RCAIDE imports
+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
+
+# ------------------------------------------------------------------------------
+# Plot Drag Components
+# ------------------------------------------------------------------------------
+def plot_drag_components(results,
+                         save_figure=False,
+                         show_legend= True,
+                         save_filename="Drag_Components",
+                         file_type=".png",
+                         width = 11, height = 7):
+    """
+    Generate plots showing the breakdown of aircraft drag components over time.
+
+    Parameters
+    ----------
+    results : Data
+        Mission results data structure containing:
+        results.segments[i].conditions.aerodynamics.coefficients.drag with fields:
+            - parasite.total : array
+                Total parasite drag coefficient
+            - induced.total : array
+                Total induced drag coefficient
+            - compressible.total : array
+                Total compressibility drag coefficient
+            - miscellaneous.total : array
+                Other drag components coefficient
+            - total : array
+                Total aircraft drag coefficient
+
+    save_figure : bool, optional
+        Save figure to file if True, default False
+
+    show_legend : bool, optional
+        Display component legend if True, default True
+
+    save_filename : str, optional
+        Name for saved figure file, default "Drag_Components"
+
+    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 single plot showing:
+        - Parasite drag coefficient (CDp)
+        - Induced drag coefficient (CDi)
+        - Compressibility drag coefficient (CDc)
+        - Miscellaneous drag coefficient (CDm)
+        - Total drag coefficient (CD)
+
+    Each mission segment uses a different color from the inferno colormap.
+    Components are distinguished by different markers.
+
+    **Definitions**
+
+    'Parasite Drag'
+        Zero-lift drag due to viscous effects
+    
+    'Induced Drag'
+        Drag due to lift production
+    
+    'Compressibility Drag'
+        Additional drag due to shock waves
+    
+    'Miscellaneous Drag'
+        Other sources including interference
+
+    See Also
+    --------
+    RCAIDE.Library.Plots.Common.set_axes : Standardized axis formatting
+    RCAIDE.Library.Plots.Common.plot_style : RCAIDE plot styling
+    """ 
+    # 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)
+    axis_1 = plt.subplot(1,1,1)
+    fig.set_size_inches(width,height)
+    
+    for i in range(len(results.segments)): 
+        time   = results.segments[i].conditions.frames.inertial.time[:,0] / Units.min 
+        drag   = results.segments[i].conditions.aerodynamics.coefficients.drag 
+        cdp    = drag.parasite.total[:,0]
+        cdi    = drag.induced.total[:,0]
+        cdc    = drag.compressible.total[:,0]
+        cdm    = drag.miscellaneous.total[:,0]  
+        cdf    = drag.form.total[:,0] 
+        cdk    = drag.cooling.total[:,0] 
+        cd     = drag.total[:,0]  
+        
+        if i ==  0:
+            axis_1.plot(time, cdp, color = line_colors[i], marker = ps.markers[0], linewidth = ps.line_width, label = r'$C_{Dpar}$') 
+            axis_1.plot(time,cdi, color = line_colors[i], marker = ps.markers[1], linewidth = ps.line_width,  label = r'$C_{Dind}$')  
+            axis_1.plot(time, cdc, color = line_colors[i], marker = ps.markers[2], linewidth = ps.line_width,  label =r'$C_{Dcomp}$')  
+            axis_1.plot(time, cdm, color = line_colors[i], marker = ps.markers[3], linewidth = ps.line_width,  label =r'$C_{Dmisc}$')  
+            axis_1.plot(time,cdw, color = line_colors[i], marker = ps.markers[4], linewidth = ps.line_width,  label = r'$C_{Dwave}$')  
+            axis_1.plot(time, cdk, color = line_colors[i], marker = ps.markers[5], linewidth = ps.line_width,  label =r'$C_{Dcool}$')  
+            axis_1.plot(time, cdf, color = line_colors[i], marker = ps.markers[6], linewidth = ps.line_width,  label =r'$C_{Dform}$')  
+            axis_1.plot(time, cd, color = line_colors[i], marker = ps.markers[7], linewidth = ps.line_width,  label =r'$C_D$')
+        else:
+            axis_1.plot(time, cdp, color = line_colors[i], marker = ps.markers[0], linewidth = ps.line_width,)
+            axis_1.plot(time,cdi, color = line_colors[i], marker = ps.markers[1], linewidth = ps.line_width, )
+            axis_1.plot(time, cdc, color = line_colors[i], marker = ps.markers[2], linewidth = ps.line_width,)
+            axis_1.plot(time, cdm, color = line_colors[i], marker = ps.markers[3], linewidth = ps.line_width,)
+            axis_1.plot(time,cdw, color = line_colors[i], marker = ps.markers[4], linewidth = ps.line_width, )
+            axis_1.plot(time, cdk, color = line_colors[i], marker = ps.markers[5], linewidth = ps.line_width,)
+            axis_1.plot(time, cdf, color = line_colors[i], marker = ps.markers[6], linewidth = ps.line_width,)
+            axis_1.plot(time, cd, color = line_colors[i], marker = ps.markers[7], linewidth = ps.line_width, )
+    
+        set_axes(axis_1)            
+        axis_1.set_xlabel('Time (mins)')
+        axis_1.set_ylabel('Drag Components') 
+        
+    
+    if show_legend:                    
+        leg =  fig.legend(bbox_to_anchor=(0.5, 0.95), loc='upper center', ncol = 5)
+    
+    # Adjusting the sub-plots for legend 
+    fig.tight_layout()
+    fig.subplots_adjust(top=0.8)
+    
+    # set title of plot 
+    title_text    = 'Drag Components'      
+    fig.suptitle(title_text)
+    
+    if save_figure:
+        plt.savefig(save_filename + file_type)   
+    return fig 

tabs/solve/plots/aerodynamics/plot_lift_distribution.py

@@ -0,0 +1,107 @@
+# RCAIDE_GUI/tabs/solve/plots/aerodynamics/plot_lift_distribution.py
+
+# RCAIDE imports
+from RCAIDE.Library.Plots.Common import plot_style
+
+# Python imports
+import matplotlib.pyplot as plt
+import numpy as np
+
+# ------------------------------------------------------------------------------
+# Plot Lift Distribution
+# ------------------------------------------------------------------------------
+
+def plot_lift_distribution(results,
+                           save_figure = False,
+                           save_filename = "Lift_Distribution",
+                           file_type = ".png",
+                           width = 11, height = 7):
+    """
+    Generate plots of spanwise lift distribution for lifting surfaces.
+
+    Parameters
+    ----------
+    results : Data
+        Mission results data structure containing:
+            - results.segments.conditions.aerodynamics.coefficients.lift with fields:
+                - inviscid_wings_sectional : array
+                    Sectional lift coefficients at control points
+
+    save_figure : bool, optional
+        Save figure to file if True, default False
+
+    save_filename : str, optional
+        Base name for saved figure files, default "Lift_Distribution"
+
+    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
+        Figure showing spanwise lift distribution
+
+    Notes
+    -----
+    Creates figures showing:
+        - Sectional lift coefficient (CLy) vs spanwise location
+        - Separate plot for each timestep in each segment
+        - Different wings distinguished by line colors:
+            - Blue: Main wings
+            - Red: Horizontal tails
+            - Black: Other surfaces
+
+    **Definitions**
+
+    'Sectional Lift Coefficient'
+        Non-dimensional lift force per unit span
+    
+    'Control Points'
+        Points where circulation/lift is evaluated
+
+    See Also
+    --------
+    RCAIDE.Library.Plots.Common.set_axes : Standardized axis formatting
+    RCAIDE.Library.Plots.Common.plot_style : RCAIDE plot styling
+    """   
+
+    # 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)
+    
+    VD         = results.segments[0].analyses.aerodynamics.settings.vortex_distribution	 	 
+    img_idx    = 1
+    seg_idx    = 1
+    for segment in results.segments.values():   	
+        num_ctrl_pts = len(segment.conditions.frames.inertial.time)	
+        for ti in range(num_ctrl_pts):   
+            line = ['-b','-b','-r','-r','-k']
+            fig  = plt.figure(save_filename + '_' + str(img_idx))
+            fig.set_size_inches(8,8)  
+            fig.set_size_inches(width,height)      
+            b_sw = np.concatenate(([0],np.cumsum(VD.n_sw[ti])))
+            axes = plt.subplot(1,1,1)
+            for i in range(int(VD.n_w[ti]) ): 
+                y_pts = VD.Y_SW[ti,b_sw[i]:b_sw[i+1]]
+                z_pts = segment.conditions.aerodynamics.coefficients.lift.inviscid.spanwise[ti,b_sw[i]:b_sw[i+1]]
+                axes.plot(y_pts, z_pts, line[i] ) 
+            axes.set_xlabel("Spanwise Location (m)")
+            axes.set_ylabel('$C_{Ly}$')  
+
+            if save_figure: 
+                plt.savefig( save_filename + '_' + str(img_idx) + file_type) 	
+            img_idx += 1
+        seg_idx +=1
+
+    return fig