from typing import Dict, TYPE_CHECKING
from TidalPy.utilities.performance import njit
from ...inclination_funcs import orderl2, orderl3, orderl4, orderl5, orderl6
if TYPE_CHECKING:
from TidalPy.utilities.types import FloatArray
from ...inclination_funcs import InclinOutput
[docs]
@njit(cacheable=True)
def inclination_off_maxl_6(obliquity: 'FloatArray') -> Dict[int, 'InclinOutput']:
""" Calculates inclination functions (squared) for a given maximum tidal order (going through each l) - Off Mode
Obliquity is assumed to be zero.
Max Supported l = 6
Parameters
----------
obliquity : FloatArray
Planet's obliquity [radians]
Returns
-------
result_by_orderl : Dict[int, InclinOutput]
Inclination function L^2_lmp truncated. Stored by order-l.
"""
result_by_orderl = {
2: orderl2.calc_inclination_off(obliquity),
3: orderl3.calc_inclination_off(obliquity),
4: orderl4.calc_inclination_off(obliquity),
5: orderl5.calc_inclination_off(obliquity),
6: orderl6.calc_inclination_off(obliquity)
}
return result_by_orderl
[docs]
@njit(cacheable=True)
def inclination_on_maxl_6(obliquity: 'FloatArray') -> Dict[int, 'InclinOutput']:
""" Calculates inclination functions (squared) for a given maximum tidal order (going through each l) - On Mode
Obliquity can be arbitrary.
Max Supported l = 6
Parameters
----------
obliquity : FloatArray
Planet's obliquity [radians]
Returns
-------
result_by_orderl : Dict[int, InclinOutput]
Inclination function L^2_lmp truncated. Stored by order-l.
"""
result_by_orderl = {
2: orderl2.calc_inclination(obliquity),
3: orderl3.calc_inclination(obliquity),
4: orderl4.calc_inclination(obliquity),
5: orderl5.calc_inclination(obliquity),
6: orderl6.calc_inclination(obliquity)
}
return result_by_orderl