@sh.add_function(dsp, outputs=['planetary_speeds_in'])
def calculate_planetary_speeds_in(
engine_speeds_out, final_drive_speeds_in, planetary_ratio):
"""
Calculates the planetary speed [RPM].
:param engine_speeds_out:
Engine speed vector [RPM].
:type engine_speeds_out: numpy.array
:param final_drive_speeds_in:
Final drive speed in [RPM].
:type final_drive_speeds_in: numpy.array
:param planetary_ratio:
Fundamental planetary speed ratio [-].
:type planetary_ratio: float
:return:
Planetary speed vector [RPM].
:rtype: numpy.array
"""
r = planetary_ratio
return engine_speeds_out * (1 + r) - final_drive_speeds_in * r