@sh.add_function(dsp, outputs=['gear_box_powers_out'])
def calculate_gear_box_powers_out(
final_drive_powers_in, motor_p3_front_powers, motor_p3_rear_powers):
"""
Calculate gear box power vector [kW].
:param final_drive_powers_in:
Final drive power in [kW].
:type final_drive_powers_in: numpy.array | float
:param motor_p3_front_powers:
Power at motor P3 front [kW].
:type motor_p3_front_powers: numpy.array | float
:param motor_p3_rear_powers:
Power at motor P3 rear [kW].
:type motor_p3_rear_powers: numpy.array | float
:return:
Gear box power vector [kW].
:rtype: numpy.array | float
"""
return final_drive_powers_in - motor_p3_front_powers - motor_p3_rear_powers