Source code for fastoad.models.performances.mission.openmdao.mission_component

OpenMDAO component for time-step computation of missions.
#  This file is part of FAST-OAD : A framework for rapid Overall Aircraft Design
#  Copyright (C) 2022 ONERA & ISAE-SUPAERO
#  FAST is free software: you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation, either version 3 of the License, or
#  (at your option) any later version.
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  GNU General Public License for more details.
#  You should have received a copy of the GNU General Public License
#  along with this program.  If not, see <>.

import logging
from enum import Enum
from os import makedirs, path as pth

import numpy as np
from openmdao import api as om

from fastoad.model_base import FlightPoint
from fastoad.model_base.propulsion import IOMPropulsionWrapper
from fastoad.models.performances.mission.openmdao.mission_wrapper import MissionWrapper
from fastoad.models.performances.mission.polar import Polar
from import BreguetCruiseSegment
from fastoad.module_management.service_registry import RegisterPropulsion

_LOGGER = logging.getLogger(__name__)  # Logger for this module

[docs]class MissionComponent(om.ExplicitComponent): """ Computes a mission as specified in mission input file Options: - propulsion_id: (mandatory) the identifier of the propulsion wrapper. - out_file: if provided, a csv file will be written at provided path with all computed flight points. - mission_wrapper: the MissionWrapper instance that defines the mission. - use_initializer_iteration: During first solver loop, a complete mission computation can fail or consume useless CPU-time. When activated, this option ensures the first iteration is done using a simple, dummy, formula instead of the specified mission. Set this option to False if you do expect this model to be computed only once. - is_sizing: if True, TOW will be considered equal to MTOW and mission payload will be considered equal to design payload. - reference_area_variable: Defines the name of the variable for providing aircraft reference surface area. """ def __init__(self, **kwargs): super().__init__(**kwargs) self.flight_points = None self._engine_wrapper = None self._mission_wrapper: MissionWrapper = None self._name_provider = None self._input_weight_variable_name = ""
[docs] def initialize(self): self.options.declare("propulsion_id", default="", types=str) self.options.declare("out_file", default="", types=str) self.options.declare("use_initializer_iteration", default=True, types=bool) self.options.declare("mission_wrapper", types=MissionWrapper) self.options.declare("mission_name", types=str) self.options.declare("is_sizing", default=False, types=bool) self.options.declare( "reference_area_variable", default="data:geometry:wing:area", types=str )
[docs] def setup(self): self._engine_wrapper = self._get_engine_wrapper() self._engine_wrapper.setup(self) self._mission_wrapper = self.options["mission_wrapper"] self._mission_wrapper.setup(self, self.options["mission_name"]) mission_name = self.options["mission_name"] self._name_provider = _get_variable_name_provider(mission_name) self._input_weight_variable_name = self._mission_wrapper.get_input_weight_variable_name( self.options["mission_name"] ) try: self.add_input(self.options["reference_area_variable"], np.nan, units="m**2") except ValueError: pass # Global mission outputs self.add_output( self._name_provider.NEEDED_BLOCK_FUEL.value, units="kg", desc=f'Needed fuel to complete mission "{mission_name}", including reserve fuel', ) self.add_output( self._name_provider.CONSUMED_FUEL_BEFORE_INPUT_WEIGHT.value, units="kg", desc=f'consumed fuel quantity before target mass defined for "{mission_name}",' f" if any (e.g. TakeOff Weight)", ) if self.options["is_sizing"]: self.add_output("data:weight:aircraft:sizing_block_fuel", units="kg") self.add_output("data:weight:aircraft:sizing_onboard_fuel_at_input_weight", units="kg")
[docs] def setup_partials(self): self.declare_partials(["*"], ["*"], method="fd")
[docs] def compute(self, inputs, outputs, discrete_inputs=None, discrete_outputs=None): iter_count = self.iter_count_without_approx message_prefix = f"Mission computation - iteration {iter_count:d} : " if iter_count == 0 and self.options["use_initializer_iteration"]: "%sUsing initializer computation. OTHER ITERATIONS NEEDED.", message_prefix ) self._compute_breguet(inputs, outputs) else:"%sUsing mission definition.", message_prefix) self._compute_mission(inputs, outputs)
def _compute_breguet(self, inputs, outputs): """ Computes mission using simple Breguet formula at altitude==100m and Mach 0.1 (but max L/D ratio is assumed anyway) Useful only for initiating the computation. :param inputs: OpenMDAO input vector :param outputs: OpenMDAO output vector """ propulsion_model = self._engine_wrapper.get_model(inputs) high_speed_polar = self._get_initial_polar(inputs) distance = np.sum( self._mission_wrapper.get_route_ranges(inputs, self.options["mission_name"]) ).item() altitude = 100.0 cruise_mach = 0.1 breguet = BreguetCruiseSegment( target=FlightPoint(ground_distance=distance), propulsion=propulsion_model, polar=high_speed_polar, use_max_lift_drag_ratio=True, ) start_point = FlightPoint( mass=inputs[self._input_weight_variable_name], altitude=altitude, mach=cruise_mach, ) flight_points = breguet.compute_from(start_point) end_point = FlightPoint.create(flight_points.iloc[-1]) outputs[self._name_provider.NEEDED_BLOCK_FUEL.value] = start_point.mass - end_point.mass @staticmethod def _get_initial_polar(inputs) -> Polar: """ At computation start, polar may be irrelevant and give a very low lift/drag ratio. In that case, this method returns a fake polar that has 10.0 as max lift drag ratio. Otherwise, the actual cruise polar is returned. """ high_speed_polar = Polar( inputs["data:aerodynamics:aircraft:cruise:CL"], inputs["data:aerodynamics:aircraft:cruise:CD"], ) use_minimum_l_d_ratio = False try: if ( high_speed_polar.optimal_cl / < 10.0 ): use_minimum_l_d_ratio = True except ZeroDivisionError: use_minimum_l_d_ratio = True if use_minimum_l_d_ratio: # We replace by a polar that has at least 10.0 as max L/D ratio high_speed_polar = Polar(np.array([0.0, 0.5, 1.0]), np.array([0.1, 0.05, 1.0])) return high_speed_polar def _compute_mission(self, inputs, outputs): """ Computes mission using time-step integration. :param inputs: OpenMDAO input vector :param outputs: OpenMDAO output vector """ propulsion_model = self._engine_wrapper.get_model(inputs) reference_area = inputs[self.options["reference_area_variable"]] self._mission_wrapper.propulsion = propulsion_model self._mission_wrapper.reference_area = reference_area # This is the default start point, that can be overridden by using the "start" # segment in the mission definition. start_flight_point = FlightPoint( altitude=0.0, mass=inputs[self._input_weight_variable_name], true_airspeed=0.0 ) self.flight_points = self._mission_wrapper.compute(start_flight_point, inputs, outputs) # Final ================================================================ start_of_mission = FlightPoint.create(self.flight_points.iloc[0]) end_of_mission = FlightPoint.create(self.flight_points.iloc[-1]) reserve = self._mission_wrapper.get_reserve( self.flight_points, self.options["mission_name"] ) reserve_name = self._mission_wrapper.get_reserve_variable_name() if reserve_name in outputs: outputs[reserve_name] = reserve outputs[self._name_provider.NEEDED_BLOCK_FUEL.value] = ( start_of_mission.mass - end_of_mission.mass + reserve ) outputs[ self._name_provider.CONSUMED_FUEL_BEFORE_INPUT_WEIGHT.value ] = self._mission_wrapper.consumed_fuel_before_input_weight if self.options["is_sizing"]: outputs["data:weight:aircraft:sizing_block_fuel"] = outputs[ self._name_provider.NEEDED_BLOCK_FUEL.value ] outputs["data:weight:aircraft:sizing_onboard_fuel_at_input_weight"] = ( outputs[self._name_provider.NEEDED_BLOCK_FUEL.value] - self._mission_wrapper.consumed_fuel_before_input_weight ) def as_scalar(value): if isinstance(value, np.ndarray): return value.item() return value self.flight_points = self.flight_points.applymap(as_scalar) rename_dict = { field_name: f"{field_name} [{unit}]" for field_name, unit in FlightPoint.get_units().items() } self.flight_points.rename(columns=rename_dict, inplace=True) if self.options["out_file"]: makedirs(pth.dirname(self.options["out_file"]), exist_ok=True) self.flight_points.to_csv(self.options["out_file"]) def _get_engine_wrapper(self) -> IOMPropulsionWrapper: """ Overloading this method allows to define the engine without relying on the propulsion option. (useful for tests) :return: the engine wrapper instance """ return RegisterPropulsion.get_provider(self.options["propulsion_id"])
def _get_variable_name_provider(mission_name): """Factory for enum class that provide mission variable names.""" def get_variable_name(suffix): return f"data:mission:{mission_name}:{suffix}" class VariableNames(Enum): """Enum with mission-related variable names.""" ZFW = get_variable_name("ZFW") PAYLOAD = get_variable_name("payload") BLOCK_FUEL = get_variable_name("block_fuel") NEEDED_BLOCK_FUEL = get_variable_name("needed_block_fuel") CONSUMED_FUEL_BEFORE_INPUT_WEIGHT = get_variable_name("consumed_fuel_before_input_weight") return VariableNames