"""Utility functions for rocketPy"""
from . import ureg, Q_
import numpy as np
[docs]def si(v):
"""Utility function to convert a Pint Quantity into a float in SI units.
Args:
v (Pint.Quantity or list/numpy.Array of Pint.Quantity): quantity or list of quantities to convert.
Returns:
float or list of floats: magnitudes in SI units.
Examples
Examples should be written in doctest format, and
should illustrate how to use the function/class.
>>> si(5.0*ureg.meter)
5.0
>> si(6.0*ureg.inch)
0.1524
"""
try:
return v.to_base_units().magnitude
except:
try:
return [si(vv) for vv in v]
except:
raise RuntimeError('Conversion to SI has failed')
[docs]def mach_correction(Ma=0.0, method='default'):
"""Performs the Prandtl-Glauert compressibility correction, extended for supersonic region.
Args:
Ma (dimensionless float): Mach Number. Defaults to 0.0.
method (str): Choose the correction method ('default' or 'Cambridge'). Defaults to 'default'.
Returns:
dimensionless float: Mach correction multiplier (float)
Examples
Examples should be written in doctest format, and
should illustrate how to use the function/class.
>>> mach_correction(0.5)
1.1547005383792517
>> mach_correction(1.5)
0.8944271909999159
"""
if method == 'default':
# my own correction:
beta = max(np.sqrt(abs(1-Ma**2)), np.sqrt(1-0.8**2))
return 1/beta
if method == 'Cambridge':
# Follows the formulation of eqn. 55-57 of Box:
if Ma < 0.8:
return 1/np.sqrt(1-Ma**2)
elif Ma > 1.1:
return 1/np.sqrt(Ma**2-1)
else:
return 1/np.sqrt(1-0.8**2)