RCAIDE-GUI 1.0.0__py3-none-any.whl

tabs/mission/widgets/__init__.py

@@ -0,0 +1,3 @@
+from tabs.mission.widgets.mission_segment_widget import MissionSegmentWidget
+from tabs.mission.widgets.flight_controls_widget import FlightControlsWidget
+from tabs.mission.widgets.mission_analysis_widget import MissionAnalysisWidget

tabs/mission/widgets/flight_controls_widget.py

@@ -0,0 +1,186 @@
+# RCAIDE_GUI/tabs/mission/widgets/flight_controls_widget.py
+
+# Created: May 2023, M. Clarke
+# ------------------------------------------------------------------------------
+# Imports
+# ------------------------------------------------------------------------------
+from PyQt6.QtWidgets import (
+    QHBoxLayout,
+    QLabel,
+    QVBoxLayout,
+    QWidget,
+    QCheckBox,
+    QSizePolicy,
+)
+
+from utilities import Units, create_line_bar, set_data
+from common_widgets import DataEntryWidget
+import rcaide_io
+
+# ------------------------------------------------------------------------------
+# Flight Controls Widget
+# ------------------------------------------------------------------------------
+class FlightControlsWidget(QWidget):
+    def __init__(self):
+        super().__init__()
+
+        base_layout = QVBoxLayout()
+        toggles_layout = QHBoxLayout()
+
+        self.data_entry_widgets = []
+
+        for key, toggles in self.fields.items():
+            sub_layout = QVBoxLayout()
+
+            header = QLabel("<b>" + key + "</b>")
+            header.setSizePolicy(
+                QSizePolicy.Policy.Minimum,
+                QSizePolicy.Policy.Minimum,
+            )
+            sub_layout.addWidget(header)
+            sub_layout.addWidget(create_line_bar())
+
+            self.data_entry_widgets.append(DataEntryWidget(toggles, 1))
+            sub_layout.addWidget(self.data_entry_widgets[-1])
+
+            toggles_layout.addLayout(sub_layout, 1)
+
+        base_layout.addLayout(toggles_layout)
+
+        # Propulsor assignment checkboxes
+        base_layout.addWidget(QLabel("<b>Assigned Propulsors</b>"))
+        base_layout.addWidget(create_line_bar())
+
+        self.propulsor_checkboxes = {}
+        self._prop_row = QHBoxLayout()
+        self._build_propulsor_checkboxes()
+        base_layout.addLayout(self._prop_row)
+
+        self.setLayout(base_layout)
+
+    # -------------------------------------------------------------------------
+    # Propulsor checkboxes
+    # -------------------------------------------------------------------------
+    def _build_propulsor_checkboxes(self):
+        while self._prop_row.count():
+            item = self._prop_row.takeAt(0)
+            if item.widget():
+                item.widget().deleteLater()
+        self.propulsor_checkboxes.clear()
+
+        names = []
+        try:
+            names = rcaide_io.propulsor_names[0] or []
+        except (AttributeError, IndexError, TypeError):
+            pass
+
+        for name in names:
+            cb = QCheckBox(name)
+            cb.setChecked(True)
+            self.propulsor_checkboxes[name] = cb
+            self._prop_row.addWidget(cb)
+
+        self._prop_row.addStretch(1)
+
+    def refresh_propulsors(self):
+        """Rebuild propulsor checkboxes from the current rcaide_io.propulsor_names."""
+        self._build_propulsor_checkboxes()
+
+    # -------------------------------------------------------------------------
+    # Data access
+    # -------------------------------------------------------------------------
+    def get_data(self):
+        data = {}
+        for widget in self.data_entry_widgets:
+            assert isinstance(widget, DataEntryWidget)
+            data.update(widget.get_values())
+        data["assigned_propulsors"] = [
+            name for name, cb in self.propulsor_checkboxes.items() if cb.isChecked()
+        ]
+        return data
+
+    def load_data(self, data):
+        for widget in self.data_entry_widgets:
+            assert isinstance(widget, DataEntryWidget)
+            widget.load_data(data)
+
+        selected = data.get("assigned_propulsors", [])
+        # Handle [value, 0] unit-arg format that may appear when loading from RCAIDE JSON
+        if (isinstance(selected, list) and len(selected) == 2
+                and isinstance(selected[1], int) and not isinstance(selected[1], bool)):
+            selected = selected[0] if isinstance(selected[0], list) else []
+
+        for name, cb in self.propulsor_checkboxes.items():
+            cb.setChecked(not selected or name in selected)
+
+    def set_control_variables(self, segment):
+        for widget in self.data_entry_widgets:
+            assert isinstance(widget, DataEntryWidget)
+            data_units_labels = widget.data_units_labels
+            data = widget.get_values()
+            for data_unit_label in data_units_labels:
+                rcaide_label = data_unit_label[-1]
+                user_label = data_unit_label[0]
+                set_data(segment, rcaide_label, data[user_label][0])
+
+        # Use the propulsors the user checked; fall back to all available propulsors
+        selected = [name for name, cb in self.propulsor_checkboxes.items() if cb.isChecked()]
+        if not selected:
+            selected = list(self.propulsor_checkboxes.keys())
+        if not selected:
+            # Last-resort fallback: derive names from vehicle networks
+            try:
+                for network in rcaide_io.rcaide_vehicle.networks:
+                    for prop in network.propulsors:
+                        if prop.tag not in selected:
+                            selected.append(prop.tag)
+            except Exception:
+                pass
+
+        segment.assigned_control_variables.throttle.assigned_propulsors = [selected]
+
+        if getattr(segment.assigned_control_variables.throttle, "active", False):
+            if hasattr(segment.assigned_control_variables.throttle, "initial_guess_values"):
+                if not segment.assigned_control_variables.throttle.initial_guess_values:
+                    segment.assigned_control_variables.throttle.initial_guess_values = [[0.7]]
+
+        if getattr(segment.assigned_control_variables.body_angle, "active", False):
+            if hasattr(segment.assigned_control_variables.body_angle, "initial_guess_values"):
+                if not segment.assigned_control_variables.body_angle.initial_guess_values:
+                    segment.assigned_control_variables.body_angle.initial_guess_values = [[0.0]]
+
+    def set_defaults(self, throttle=False, body_angle=False):
+        for widget in self.data_entry_widgets:
+            assert isinstance(widget, DataEntryWidget)
+            defaults = {}
+            for label, _, rcaide_label in widget.data_units_labels:
+                label_lower = rcaide_label.lower()
+                value = False
+                if "throttle.active" in label_lower:
+                    value = throttle
+                elif "body_angle.active" in label_lower:
+                    value = body_angle
+                defaults[label] = (value, 0)
+            widget.load_data(defaults)
+
+    fields = {
+        "Kinematics": [
+            ("Body Angle", Units.Boolean, "assigned_control_variables.body_angle.active"),
+            ("Bank Angle", Units.Boolean, "assigned_control_variables.bank_angle.active"),
+            ("Wind Angle", Units.Boolean, "assigned_control_variables.wind_angle.active"),
+            ("Velocity", Units.Boolean, "assigned_control_variables.velocity.active"),
+            ("Acceleration", Units.Boolean, "assigned_control_variables.acceleration.active"),
+            ("Altitude", Units.Boolean, "assigned_control_variables.altitude.active"),
+        ],
+        "Control Surfaces": [
+            ("Elevator Deflection", Units.Boolean, "assigned_control_variables.elevator_deflection.active"),
+            ("Rudder Deflection", Units.Boolean, "assigned_control_variables.rudder_deflection.active"),
+            ("Flap Deflection", Units.Boolean, "assigned_control_variables.flap_deflection.active"),
+            ("Slat Deflection", Units.Boolean, "assigned_control_variables.slat_deflection.active"),
+            ("Aileron Deflection", Units.Boolean, "assigned_control_variables.aileron_deflection.active"),
+        ],
+        "Propulsion": [
+            ("Throttle", Units.Boolean, "assigned_control_variables.throttle.active"),
+            ("Thrust Vector Angle", Units.Boolean, "assigned_control_variables.thrust_vector_angle.active"),
+        ],
+    }

tabs/mission/widgets/mission_analysis_widget.py

@@ -0,0 +1,166 @@
+# RCAIDE_GUI/tabs/mission/widgets/mission_analysis_widget.py
+
+# Created: May 2023, M. Clarke
+# ------------------------------------------------------------------------------
+# Imports
+# ------------------------------------------------------------------------------
+from PyQt6.QtCore import Qt
+from PyQt6.QtWidgets import QWidget, QHBoxLayout, QVBoxLayout, QTreeWidget, \
+    QTreeWidgetItem, QHeaderView, QPushButton, QLabel
+
+from tabs.analysis.widgets import *
+from tabs import TabWidget
+from utilities import create_scroll_area
+import rcaide_io
+import RCAIDE
+
+# ============================================================
+#  Full Functional Analysis Widget (used by Mission tab)
+# ============================================================
+class MissionAnalysisWidget(TabWidget):
+    def __init__(self):
+        super(MissionAnalysisWidget, self).__init__()
+
+        options = [
+            "Aerodynamics",
+            "Atmospheric",
+            "Planets",
+            "Weights",
+            "Geometry",
+            "Energy",
+            "Propulsion",
+            "Costs",
+            "Aeroacoustics",
+            "Stability",
+        ]
+
+        self.tree_frame_layout = QVBoxLayout()
+        self.tree_widget = QTreeWidget()
+
+        save_analysis_button = QPushButton("Save Analyses")
+        save_analysis_button.clicked.connect(self.save_analyses)
+
+        self.tree_frame_layout.addWidget(save_analysis_button)
+        self.tree_frame_layout.addWidget(self.tree_widget)
+
+        self.tree_widget.setColumnCount(2)
+        self.tree_widget.setHeaderLabels(["Analysis", "Enabled"])
+        header = self.tree_widget.header()
+        assert header is not None
+        header.setSectionResizeMode(0, QHeaderView.ResizeMode.ResizeToContents)
+
+        for index, option in enumerate(options):
+            item = QTreeWidgetItem([option])
+            if index > 5:
+                item.setCheckState(1, Qt.CheckState.Unchecked)
+            else:
+                item.setData(1, Qt.ItemDataRole.CheckStateRole, Qt.CheckState.Checked)
+                item.setFlags(item.flags() & ~Qt.ItemFlag.ItemIsUserCheckable & ~Qt.ItemFlag.ItemIsEnabled)
+
+            self.tree_widget.addTopLevelItem(item)
+
+        self.tree_widget.itemChanged.connect(self.handleItemChanged)
+
+        self.base_layout = QHBoxLayout()
+        self.base_layout.addLayout(self.tree_frame_layout, 1)
+
+        self.main_layout = None
+        layout_scroll = create_scroll_area(self, False)
+        self.base_layout.addLayout(layout_scroll, 4)
+
+        assert self.main_layout is not None and isinstance(self.main_layout, QVBoxLayout)
+
+        self.analysis_widgets = [
+            AerodynamicsWidget,
+            AtmosphereWidget,
+            PlanetsWidget,
+            WeightsWidget,
+            GeometryWidget,
+            EnergyWidget,
+            PropulsionWidget,
+            CostsWidget,
+            AeroacousticsWidget,
+            StabilityWidget,
+        ]
+        self.widgets = []
+
+        for index, analysis_widget in enumerate(self.analysis_widgets):
+            widget = analysis_widget()
+            assert isinstance(widget, AnalysisDataWidget)
+            if index >= 5:
+                widget.setVisible(False)
+            else:
+                widget.setVisible(True)
+
+            self.widgets.append(widget)
+            self.main_layout.addWidget(widget)
+
+        self.main_layout.addStretch()
+        self.main_layout.setSpacing(3)
+        self.base_layout.setSpacing(3)
+        self.setLayout(self.base_layout)
+
+    # ============================================================
+    #  Logic for enabling/disabling sub-analyses
+    # ============================================================
+    def handleItemChanged(self, item, column):
+        if column != 1:
+            return
+
+        index = self.tree_widget.indexOfTopLevelItem(item)
+        layout_item = self.main_layout.itemAt(index)
+        assert layout_item is not None
+        widget = layout_item.widget()
+        assert widget is not None
+        widget.setVisible(item.checkState(1) == Qt.CheckState.Checked)
+
+    def load_from_values(self):
+        if not rcaide_io.analysis_data:
+            self.save_analyses()
+            return
+
+        for index, widget in enumerate(self.widgets):
+            assert isinstance(widget, AnalysisDataWidget)
+            widget.load_values(rcaide_io.analysis_data[index])
+
+        self.save_analyses()
+
+    def get_check_state(self, index) -> bool:
+        top_level_item = self.tree_widget.topLevelItem(index)
+        assert top_level_item is not None
+        return top_level_item.checkState(1) == Qt.CheckState.Checked
+
+    def save_analyses(self):
+        if (not getattr(values, "config_data", None) and
+                not getattr(values, "rcaide_configs", None)):
+            raise RuntimeError(
+                "No aircraft configuration data found. "
+                "Create a configuration in the Aircraft Configs tab."
+            )
+        if (not getattr(values, "rcaide_configs", None) or
+                not rcaide_io.rcaide_configs):
+            from tabs.aircraft_configs.aircraft_configs import build_rcaide_configs_from_geometry
+            try:
+                rcaide_io.rcaide_configs = build_rcaide_configs_from_geometry()
+            except Exception as exc:
+                raise RuntimeError(
+                    "No RCAIDE configurations found. "
+                    "Save a configuration in the Aircraft Configs tab."
+                ) from exc
+        rcaide_io.analysis_data = []
+        rcaide_io.rcaide_analyses = {}
+
+        # Build and save one RCAIDE analysis container per aircraft configuration.   
+        for tag, config in rcaide_io.rcaide_configs.items():
+
+            analyses = RCAIDE.Framework.Analyses.Vehicle()
+            analyses.vehicle = config  
+            
+            for index, widget in enumerate(self.widgets):
+                assert isinstance(widget, AnalysisDataWidget)
+
+                if self.get_check_state(index):
+                    analyses.append(widget.create_analysis(config)) 
+
+            rcaide_io.rcaide_analyses[tag] = analyses
+            rcaide_io.analysis_data.append(analyses)