import matplotlib.pyplot as plt
from .motor_plots import _MotorPlots
from .plot_helpers import show_or_save_plot
[docs]
class _SolidMotorPlots(_MotorPlots):
"""Class that holds plot methods for SolidMotor class.
Attributes
----------
_SolidMotorPlots.solid_motor : SolidMotor
SolidMotor object that will be used for the plots.
"""
[docs]
def grain_inner_radius(self, lower_limit=None, upper_limit=None):
"""Plots grain_inner_radius of the solid_motor as a function of time.
Parameters
----------
lower_limit : float
Lower limit of the plot. Default is none, which means that the plot
limits will be automatically calculated.
upper_limit : float
Upper limit of the plot. Default is none, which means that the plot
limits will be automatically calculated.
Return
------
None
"""
self.motor.grain_inner_radius.plot(lower=lower_limit, upper=upper_limit)
[docs]
def grain_height(self, lower_limit=None, upper_limit=None):
"""Plots grain_height of the solid_motor as a function of time.
Parameters
----------
lower_limit : float
Lower limit of the plot. Default is none, which means that the plot
limits will be automatically calculated.
upper_limit : float
Upper limit of the plot. Default is none, which means that the plot
limits will be automatically calculated.
Return
------
None
"""
self.motor.grain_height.plot(lower=lower_limit, upper=upper_limit)
[docs]
def burn_rate(self, lower_limit=None, upper_limit=None):
"""Plots burn_rate of the solid_motor as a function of time.
Parameters
----------
lower_limit : float
Lower limit of the plot. Default is none, which means that the plot
limits will be automatically calculated.
upper_limit : float
Upper limit of the plot. Default is none, which means that the plot
limits will be automatically calculated.
Return
------
None
"""
self.motor.burn_rate.plot(lower=lower_limit, upper=upper_limit)
[docs]
def burn_area(self, lower_limit=None, upper_limit=None):
"""Plots burn_area of the solid_motor as a function of time.
Parameters
----------
lower_limit : float
Lower limit of the plot. Default is none, which means that the plot
limits will be automatically calculated.
upper_limit : float
Upper limit of the plot. Default is none, which means that the plot
limits will be automatically calculated.
Return
------
None
"""
self.motor.burn_area.plot(lower=lower_limit, upper=upper_limit)
[docs]
def Kn(self, lower_limit=None, upper_limit=None):
"""Plots Kn of the solid_motor as a function of time.
Parameters
----------
lower_limit : float
Lower limit of the plot. Default is none, which means that the plot
limits will be automatically calculated.
upper_limit : float
Upper limit of the plot. Default is none, which means that the plot
limits will be automatically calculated.
filename : str | None, optional
The path the plot should be saved to. By default None, in which case
the plot will be shown instead of saved. Supported file endings are:
eps, jpg, jpeg, pdf, pgf, png, ps, raw, rgba, svg, svgz, tif, tiff
and webp (these are the formats supported by matplotlib).
Return
------
None
"""
self.motor.Kn.plot(lower=lower_limit, upper=upper_limit)
[docs]
def draw(self, *, filename=None):
"""Draw a representation of the SolidMotor.
Parameters
----------
filename : str | None, optional
The path the plot should be saved to. By default None, in which case
the plot will be shown instead of saved. Supported file endings are:
eps, jpg, jpeg, pdf, pgf, png, ps, raw, rgba, svg, svgz, tif, tiff
and webp (these are the formats supported by matplotlib).
Returns
-------
None
"""
_, ax = plt.subplots(figsize=(8, 6), facecolor="#EEEEEE")
nozzle = self._generate_nozzle(
translate=(self.motor.nozzle_position, 0), csys=self.motor._csys
)
chamber = self._generate_combustion_chamber(
translate=(self.motor.grains_center_of_mass_position, 0)
)
grains = self._generate_grains(
translate=(self.motor.grains_center_of_mass_position, 0)
)
outline = self._generate_motor_region(
list_of_patches=[nozzle, chamber, *grains]
)
ax.add_patch(outline)
ax.add_patch(chamber)
for grain in grains:
ax.add_patch(grain)
ax.add_patch(nozzle)
ax.set_title("Solid Motor Representation")
self._draw_center_of_mass(ax)
self._set_plot_properties(ax)
show_or_save_plot(filename)
[docs]
def all(self):
"""Prints out all graphs available about the SolidMotor. It simply calls
all the other plotter methods in this class.
Returns
-------
None
"""
self.draw()
self.thrust(*self.motor.burn_time)
self.mass_flow_rate(*self.motor.burn_time)
self.exhaust_velocity(*self.motor.burn_time)
self.total_mass(*self.motor.burn_time)
self.propellant_mass(*self.motor.burn_time)
self.center_of_mass(*self.motor.burn_time)
self.grain_inner_radius(*self.motor.burn_time)
self.grain_height(*self.motor.burn_time)
self.burn_rate(self.motor.burn_time[0], self.motor.grain_burn_out)
self.burn_area(*self.motor.burn_time)
self.Kn()
self.inertia_tensor(*self.motor.burn_time)
