"""
Defines the analysis and plotting functions for postprocessing regarding the mission
"""
# This file is part of FAST-OAD : A framework for rapid Overall Aircraft Design
# Copyright (C) 2021 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
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# 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 <https://www.gnu.org/licenses/>.
import os.path as pth
from typing import Union
import ipywidgets as widgets
import pandas as pd
import plotly.graph_objects as go
from IPython.display import clear_output, display
[docs]class MissionViewer:
"""
A class for facilitating the post-processing of mission and trajectories
"""
def __init__(self):
# The dataframes containing each mission
self.missions = {}
# The output widget containing the figure to display
self._output_widget = None
# The x selector
self._x_widget = None
# The y selector
self._y_widget = None
[docs] def add_mission(self, mission_data: Union[str, pd.DataFrame], name=None):
"""
Adds the mission to the mission database (self.missions)
:param mission_data: path of the mission file or Dataframe containing the mission data
:param name: name to give to the mission
"""
if (
isinstance(mission_data, str)
and mission_data.endswith(".csv")
and pth.exists(mission_data)
):
self.missions[name] = pd.read_csv(mission_data, index_col=0)
elif isinstance(mission_data, pd.DataFrame):
self.missions[name] = mission_data
else:
raise TypeError("Unknown type for mission data, please use .csv of DataFrame")
[docs] def display(self):
"""
Display the user interface
:return the display object
"""
key = list(self.missions)[0]
keys = self.missions[key].keys()
self._output_widget = widgets.Output()
# By default ground distance
self._x_widget = widgets.Dropdown(value=keys[2], options=keys)
self._x_widget.observe(self._show_plot, "value")
# By default altitude
self._y_widget = widgets.Dropdown(value=keys[1], options=keys)
self._y_widget.observe(self._show_plot, "value")
self._show_plot()
toolbar = widgets.HBox(
[widgets.Label(value="x:"), self._x_widget, widgets.Label(value="y:"), self._y_widget]
)
ui = display(toolbar, self._output_widget)
return ui
# pylint: disable=unused-argument # change has to be there for observe() to work
def _show_plot(self, change=None):
"""
Updates and shows the plots
"""
with self._output_widget:
clear_output(wait=True)
x_name = self._x_widget.value
y_name = self._y_widget.value
fig = None
for mission_name in self.missions:
if fig is None:
fig = go.Figure()
# pylint: disable=invalid-name # that's a common naming
x = self.missions[mission_name][x_name]
# pylint: disable=invalid-name # that's a common naming
y = self.missions[mission_name][y_name]
scatter = go.Scatter(x=x, y=y, mode="lines", name=mission_name)
fig.add_trace(scatter)
fig.update_layout(
title_text="Mission", title_x=0.5, xaxis_title=x_name, yaxis_title=y_name
)
fig = go.FigureWidget(fig)
display(fig)