RCAIDE-GUI 1.0.0__py3-none-any.whl

tabs/geometry/frames/powertrain/powertrain_frame.py

@@ -0,0 +1,259 @@
+# RCAIDE_GUI/tabs/geometry/frames/powertrain/powertrain_frame.py
+#
+# Created:  Dec 2025, M. Clarke
+
+# ----------------------------------------------------------------------------------------------------------------------
+#  IMPORT
+# ----------------------------------------------------------------------------------------------------------------------
+import RCAIDE
+from PyQt6.QtCore import Qt
+from PyQt6.QtWidgets import QWidget, QVBoxLayout, QHBoxLayout, QLabel, QFrame, QScrollArea, QSpacerItem, QSizePolicy, \
+    QPushButton, QLineEdit, QComboBox
+
+from tabs.geometry.frames import GeometryFrame
+from tabs.geometry.widgets.powertrain.powertrain_widget import PowertrainWidget
+from utilities import show_popup, clear_layout
+
+
+class PowertrainFrame(GeometryFrame):
+    def __init__(self):
+        super(PowertrainFrame, self).__init__()
+
+        self.data_fields = {}
+        self.save_function = None
+        self.tab_index = -1
+        self.index = -1
+
+        # Create a scroll area
+        scroll_area = QScrollArea()
+        # Allow the widget inside to resize with the scroll area
+        scroll_area.setWidgetResizable(True)
+
+        # Create a widget to contain the layout
+        scroll_content = QWidget()
+        # Set the main layout inside the scroll content
+        layout = QVBoxLayout(scroll_content)
+        self.powertrain_layout = QVBoxLayout()
+
+        # Create a horizontal layout for the label and buttons
+        header_layout = QHBoxLayout()
+        header_layout.addWidget(QLabel("<b>Powertrain</b>"))
+
+        layout.addLayout(header_layout)
+        # Create a horizontal line
+        line_bar = QFrame()
+        line_bar.setFrameShape(QFrame.Shape.HLine)
+        line_bar.setFrameShadow(QFrame.Shadow.Sunken)
+
+        # Add the line bar to the main layout
+        layout.addWidget(line_bar)
+
+        self.main_powertrain_widget = self.make_powertrain_widget()
+        # Add the grid layout to the home layout
+        layout.addWidget(self.main_powertrain_widget)
+
+        layout.addWidget(line_bar)
+        layout.addLayout(self.powertrain_layout)
+
+        self.add_buttons_layout()
+
+        # Adds scroll function
+        layout.addItem(QSpacerItem(
+            20, 40, QSizePolicy.Policy.MinimumExpanding, QSizePolicy.Policy.Expanding))
+
+        # Set the scroll content as the widget for the scroll area
+        scroll_area.setWidget(scroll_content)
+
+        # Set the main layout of the scroll area
+        layout_scroll = QVBoxLayout(self)
+        layout_scroll.addWidget(scroll_area)
+
+        # Set the layout to the main window/widget
+        self.setLayout(layout_scroll)
+
+    # noinspection PyUnresolvedReferences
+    def add_buttons_layout(self):
+        # define buttons
+        save_button = QPushButton("Save Energy Network Data", self)
+        # save_button.setStyleSheet("color:#dbe7ff; font-weight:500; margin:0; padding:0;")
+        delete_button = QPushButton("Clear Energy Network", self)
+        delete_button.setStyleSheet("color:#dbe7ff; font-weight:500; margin:0; padding:0;")
+
+        # define action of buttons
+        save_button.clicked.connect(self.save_data)
+        delete_button.clicked.connect(self.create_new_structure)
+
+        # Create a QHBoxLayout to contain the buttons
+        buttons_layout = QHBoxLayout()
+        buttons_layout.addWidget(save_button)
+        buttons_layout.addWidget(delete_button)
+
+        self.powertrain_layout.addLayout(buttons_layout)
+
+    def make_powertrain_widget(self):
+        """Create a widget for the powertrain section.
+
+        Returns:
+            QWidget: The main powertrain widget."""
+        main_powertrain_widget = QWidget()
+        main_layout = QVBoxLayout()
+        name_layout = QHBoxLayout()
+
+        # add spacing
+        spacer_left = QSpacerItem(
+            50, 5, QSizePolicy.Policy.Fixed, QSizePolicy.Policy.Minimum)
+        spacer_right = QSpacerItem(
+            200, 5, QSizePolicy.Policy.Fixed, QSizePolicy.Policy.Minimum)
+        name_layout.addItem(spacer_left)
+        name_layout.addWidget(QLabel("Name: "))
+        self.name_line_edit = QLineEdit(self)
+        name_layout.addWidget(self.name_line_edit)
+        name_layout.addItem(spacer_right)
+
+        # Energy Network
+        energy_label = QLabel("Energy Network Type:")
+        energy_label.setFixedWidth(100)  # Adjust width of label
+        name_layout.addWidget(energy_label)
+        self.powertrain_combo = QComboBox()
+        self.powertrain_combo.addItems(["Select Network Type", "Fuel", "Electric", "Hybrid", "Hydrogen"])
+        self.powertrain_combo.setFixedWidth(400)  # Fix width of combo box
+        name_layout.addWidget(self.powertrain_combo, alignment=Qt.AlignmentFlag.AlignLeft)
+
+        # Connect signal
+        self.powertrain_combo.currentIndexChanged.connect(self.display_selected_network)
+
+        main_layout.addLayout(name_layout)
+        main_powertrain_widget.setLayout(main_layout)
+        return main_powertrain_widget
+
+    def set_save_function(self, function):
+        self.save_function = function
+
+    def set_tab_index(self, index):
+        self.tab_index = index
+
+    def get_data_values(self):
+        """Retrieve the entered data values from the widgets."""
+        selected_network = self.powertrain_combo.currentText()
+        data = {"energy network selected": selected_network}
+
+        assert self.name_line_edit is not None
+        data["name"] = self.name_line_edit.text()
+
+        if selected_network == "Fuel":
+            # Add the data values from each fuel line widget to an array
+            item = self.powertrain_layout.itemAt(0)
+            assert item is not None
+            widget = item.widget()
+            assert widget is not None and isinstance(widget,PowertrainWidget)
+            data_values, net = widget.get_data_values()
+
+            if isinstance(data_values, bool):
+                return False, False
+
+            # add the fuel line data to the main data
+            data["powertrain"] = data_values
+
+        return data, net
+
+    def create_rcaide_structure(self):
+        selected_network = self.powertrain_combo.currentText()
+        if selected_network == "Fuel":
+            item = self.powertrain_layout.itemAt(0)
+            assert item is not None
+            widget = item.widget()
+            assert widget is not None and isinstance(widget, PowertrainWidget)
+            _, net = widget.get_data_values()
+        else:
+            return None
+
+        return net
+
+    def save_data(self):
+        """Call the save function and pass the entered data to it."""
+        entered_data, component = self.get_data_values()
+        if isinstance(entered_data, bool):
+            return
+
+        if self.save_function:
+            if self.index >= 0:
+                self.index = self.save_function(
+                    tab_index=self.tab_index, index=self.index, data=entered_data)
+                return
+            else:
+                self.index = self.save_function(
+                    tab_index=self.tab_index, vehicle_component=component, data=entered_data, new=True)
+
+        show_popup("Data Saved!", self)
+
+    def load_data(self, data, index):
+        """Load the data into the widgets.
+
+        Args:
+            data: The data to be loaded into the widgets.
+            index: The index of the data in the list.
+        """
+
+        # Load the name into the name line edit
+        self.index = index
+        assert self.name_line_edit is not None  
+        self.name_line_edit.setText(data["name"])
+
+        # Load the selected network into the combo box
+        selected_network = data.get("energy network selected", "")
+        network_index = self.powertrain_combo.findText(selected_network)
+        if network_index != -1:
+            self.powertrain_combo.setCurrentIndex(network_index)
+
+        # Clear existing sections before loading new ones
+        clear_layout(self.powertrain_layout)
+
+        # Load sections based on the selected network
+        if selected_network == "Fuel":
+            powertrain_widget = PowertrainWidget()
+            powertrain_widget.load_data_values(data["powertrain"])
+            self.powertrain_layout.addWidget(powertrain_widget)
+        
+        self.add_buttons_layout()
+
+    def create_new_structure(self):
+        """Create a new powertrain structure."""
+
+        # Clear the main data values
+        for data_field in self.data_fields.values():
+            line_edit, unit_picker = data_field
+            line_edit.clear()
+            unit_picker.set_index(0)
+
+        # Clear the name line edit
+        while self.powertrain_layout.count():
+            item = self.powertrain_layout.takeAt(0)
+            assert item is not None
+
+            widget = item.widget()
+            assert widget is not None
+            widget.deleteLater()
+
+        assert self.name_line_edit is not None
+        self.name_line_edit.clear()
+        self.index = -1
+
+    def delete_data(self):
+        pass
+
+    def display_selected_network(self, index):
+        selected_network = self.powertrain_combo.currentText()
+        # Clear the layout first
+        clear_layout(self.powertrain_layout)
+
+        if selected_network == "Fuel":
+            self.main_powertrain_widget = PowertrainWidget()
+            self.powertrain_layout.addWidget(self.main_powertrain_widget)
+        elif selected_network == "None Selected":
+            # Do nothing or add blank widget
+            pass
+        else:
+            # Handle other energy network options here
+            pass
+
+        self.add_buttons_layout()

tabs/geometry/frames/powertrain/propulsors/__init__.py

@@ -0,0 +1,5 @@
+# RCAIDE_GUI/tabs/geometry/frames/powertrain/propulsors/__init__.py
+# ----------------------------------------------------------------------------------------------------------------------
+#  IMPORT
+# ----------------------------------------------------------------------------------------------------------------------  
+from tabs.geometry.frames.powertrain.propulsors.propulsor_frame    import  PropulsorFrame 

tabs/geometry/frames/powertrain/propulsors/propulsor_frame.py

@@ -0,0 +1,146 @@
+# RCAIDE_GUI/tabs/geometry/frames/powertrain/propulsors/propulsor_frame.py
+# 
+# Created:  Dec 2025, M. Clarke 
+
+# ----------------------------------------------------------------------------------------------------------------------
+#  IMPORT
+# ---------------------------------------------------------------------------------------------------------------------- 
+# RCAIDE imports   
+import RCAIDE
+
+# PyQT Imports
+from PyQt6.QtWidgets import QWidget, QLabel, QVBoxLayout, QPushButton, QLineEdit, QHBoxLayout, \
+    QSpacerItem, QSizePolicy, QScrollArea, QFrame
+
+# RCAIDE GUI imports  
+from tabs.geometry.widgets.powertrain.propulsors.turbofan_widget import TurbofanWidget
+from common_widgets import DataEntryWidget
+from utilities import show_popup, create_line_bar, set_data, Units, create_scroll_area, clear_layout
+import rcaide_io
+
+# --------------------------------------------------------------------------------------------------------------------- 
+#  Propulsor Frame 
+# ----------------------------------------------------------------------------------------------------------------------
+class PropulsorFrame(QWidget):
+    def __init__(self):
+        super(PropulsorFrame, self).__init__()
+        self.data_entry_widget: DataEntryWidget | None = None
+
+        # List to store data values propulsor_ sections
+        self.propulsor_sections_layout = QVBoxLayout()
+
+        # Create a horizontal layout for the label and buttons
+        header_layout = QVBoxLayout()
+        # label = QLabel("<u><b>Propulsor Frame</b></u>")
+
+        layout = self.create_scroll_layout()
+
+        # header_layout.addWidget(label)
+
+        # Add propulsor_ Section Button
+        add_turbofan_button = QPushButton("Add Turbofan", self)
+        add_turbofan_button.setStyleSheet("color:#dbe7ff; font-weight:500; margin:0; padding:0;")
+        add_turbofan_button.setMaximumWidth(200)
+        add_turbofan_button.clicked.connect(self.add_turbofan)
+        header_layout.addWidget(add_turbofan_button)
+
+        layout.addLayout(header_layout)
+
+        # Create a horizontal line
+        line_bar = QFrame()
+        line_bar.setFrameShape(QFrame.Shape.HLine)
+        line_bar.setFrameShadow(QFrame.Shadow.Sunken)
+        line_bar.setStyleSheet("background-color: light grey;")
+
+        # Add the line bar to the main layout
+        layout.addWidget(line_bar)
+
+        # Add the layout for additional propulsor_ sections to the main layout
+        layout.addLayout(self.propulsor_sections_layout)
+
+        # Create a QHBoxLayout to contain the buttons
+        button_layout = QHBoxLayout()
+
+        # Add the button layout to the main layout
+        layout.addLayout(button_layout)
+
+        # Adds scroll function
+        layout.addItem(QSpacerItem(20, 40, QSizePolicy.Policy.MinimumExpanding, QSizePolicy.Policy.Expanding))
+
+    def get_data_values(self):
+        """Retrieve the entered data values from the dictionary for the propulsor sections."""
+
+        # Collect data from additional fuselage_widget
+        data = []
+        propulsors = []
+        rcaide_io.propulsor_names = [[]]
+        for index in range(self.propulsor_sections_layout.count()):
+            item = self.propulsor_sections_layout.itemAt(index)
+            assert item is not None
+            widget = item.widget()
+            assert widget is not None and isinstance(widget, TurbofanWidget)
+
+            propulsor_data, propulsor = widget.get_data_values()
+            data.append(propulsor_data)
+            propulsors.append(propulsor)
+
+        return data, propulsors
+
+    def load_data(self, data):
+        while self.propulsor_sections_layout.count():
+            item = self.propulsor_sections_layout.takeAt(0)
+            assert item is not None
+            widget = item.widget()
+            assert widget is not None
+
+            self.propulsor_sections_layout.removeWidget(widget)
+            widget.deleteLater()
+
+        for propulsor_data in data:
+            self.propulsor_sections_layout.addWidget(TurbofanWidget(
+                self.propulsor_sections_layout.count(), self.on_delete_button_pressed, propulsor_data))
+
+    def delete_data(self):
+        """Delete the entered data or perform any other action."""
+        # TODO: Implement proper deletion of data
+
+    def add_turbofan(self):
+        self.propulsor_sections_layout.addWidget(
+            TurbofanWidget(self.propulsor_sections_layout.count(), self.on_delete_button_pressed))
+
+    def on_delete_button_pressed(self, index):
+        propulsor = self.propulsor_sections_layout.itemAt(index)
+        if propulsor is None:
+            return
+
+        widget = propulsor.widget()
+        if widget is None:
+            return
+
+        widget.deleteLater()
+        self.propulsor_sections_layout.removeWidget(widget)
+        self.propulsor_sections_layout.update() 
+
+        for i in range(index, self.propulsor_sections_layout.count()):
+            propulsor = self.propulsor_sections_layout.itemAt(i)
+            if propulsor is None:
+                continue
+
+            widget = propulsor.widget()
+            if widget is None or not isinstance(widget, TurbofanWidget):
+                continue
+
+            widget.index = i 
+
+    def update_units(self, line_edit, unit_combobox):
+        pass
+
+    def create_scroll_layout(self):
+        # Create a widget to contain the layout
+        scroll_content = QWidget()
+        layout = QVBoxLayout(scroll_content)  # Set the main layout inside the scroll content
+
+        # Set the main layout of the widget
+        self.setLayout(layout)
+
+        return layout

tabs/geometry/frames/powertrain/sources/__init__.py

@@ -0,0 +1,6 @@
+# RCAIDE_GUI/tabs/geometry/frames/powertrain/sources/__init__.py
+# ----------------------------------------------------------------------------------------------------------------------
+#  IMPORT
+# ---------------------------------------------------------------------------------------------------------------------- 
+
+from tabs.geometry.frames.powertrain.sources.energy_source_frame import EnergySourceFrame

tabs/geometry/frames/powertrain/sources/energy_source_frame.py

@@ -0,0 +1,147 @@
+# RCAIDE_GUI/tabs/geometry/frames/powertrain/sources/energy_source_frame.py
+# 
+# Created:  Dec 2025, M. Clarke 
+
+# ----------------------------------------------------------------------------------------------------------------------
+#  IMPORT
+# ---------------------------------------------------------------------------------------------------------------------- 
+# RCAIDE imports
+import RCAIDE
+
+# PyQT Imports
+from PyQt6.QtWidgets import QWidget, QVBoxLayout, QPushButton, QHBoxLayout, \
+    QSpacerItem, QSizePolicy, QFrame
+
+# RCAIDE GUI imports
+from tabs.geometry.widgets.powertrain.sources.fuel_tank_widget import FuelTankWidget
+from common_widgets import DataEntryWidget
+
+
+# --------------------------------------------------------------------------------------------------------------------- 
+#  Energy Source Frame 
+# ----------------------------------------------------------------------------------------------------------------------
+class EnergySourceFrame(QWidget):
+    def __init__(self):
+        super(EnergySourceFrame, self).__init__()
+
+        self.save_function = None
+        self.data_entry_widget: DataEntryWidget | None = None
+
+        # List to store data values source_ sections
+        self.source_sections_layout = QVBoxLayout()
+
+        # Create a horizontal layout for the label and buttons
+        header_layout = QVBoxLayout()
+        # label = QLabel("<u><b>source Frame</b></u>")
+
+        layout = self.create_scroll_layout()
+
+        # header_layout.addWidget(label)
+
+        # Add source_ Section Button
+        add_fuel_tank_button = QPushButton("Add Fuel Tank", self)
+        add_fuel_tank_button.setStyleSheet("color:#dbe7ff; font-weight:500; margin:0; padding:0;")
+        add_fuel_tank_button.setMaximumWidth(200)
+        add_fuel_tank_button.clicked.connect(self.add_fuel_tank)
+
+        header_layout.addWidget(add_fuel_tank_button)
+
+        layout.addLayout(header_layout)
+
+        # Create a horizontal line
+        line_bar = QFrame()
+        line_bar.setFrameShape(QFrame.Shape.HLine)
+        line_bar.setFrameShadow(QFrame.Shadow.Sunken)
+        line_bar.setStyleSheet("background-color: light grey;")
+
+        # Add the line bar to the main layout
+        layout.addWidget(line_bar)
+
+        # Add the layout for additional source_ sections to the main layout
+        layout.addLayout(self.source_sections_layout)
+
+        # Create a QHBoxLayout to contain the buttons
+        button_layout = QHBoxLayout()
+
+        # Add the button layout to the main layout
+        layout.addLayout(button_layout)
+
+        # Adds scroll function
+        layout.addItem(QSpacerItem(
+            20, 40, QSizePolicy.Policy.MinimumExpanding, QSizePolicy.Policy.Expanding))
+
+    def get_data_values(self):
+        data = []
+        sources = []
+        for index in range(self.source_sections_layout.count()):
+            item = self.source_sections_layout.itemAt(index)
+            assert item is not None
+            widget = item.widget()
+            assert widget is not None and isinstance(widget, FuelTankWidget)
+
+            source_data, fuel_tank = widget.get_data_values()
+            data.append(source_data)
+            sources.append(fuel_tank)
+
+        return data, sources
+
+    def load_data(self, data):
+        while self.source_sections_layout.count():
+            widget_item = self.source_sections_layout.itemAt(0)
+            assert widget_item is not None
+            widget = widget_item.widget()
+            assert widget is not None
+
+            self.source_sections_layout.removeWidget(widget)
+            widget.deleteLater()
+
+        for section_data in data:
+            self.source_sections_layout.addWidget(FuelTankWidget(
+                self.source_sections_layout.count(), self.on_delete_button_pressed, section_data))
+
+    def delete_data(self):
+        # TODO Implement proper deletion of data
+        pass
+
+    def add_fuel_tank(self):
+        self.source_sections_layout.addWidget(
+            FuelTankWidget(self.source_sections_layout.count(), self.on_delete_button_pressed))
+
+    def on_delete_button_pressed(self, index):
+        tank = self.source_sections_layout.itemAt(index)
+        if tank is None:
+            return
+
+        widget = tank.widget()
+        if widget is None:
+            return
+
+        widget.deleteLater()
+        self.source_sections_layout.removeWidget(widget)
+        self.source_sections_layout.update()
+
+        for i in range(index, self.source_sections_layout.count()):
+            tank = self.source_sections_layout.itemAt(i)
+            if tank is None:
+                continue
+
+            widget = tank.widget()
+            if widget is None or not isinstance(widget, FuelTankWidget):
+                continue
+
+            widget.index = i
+
+    def set_save_function(self, function):
+        self.save_function = function
+
+    def create_scroll_layout(self):
+        # Create a widget to contain the layout
+        scroll_content = QWidget()
+
+        # Set the main layout inside the scroll content
+        layout = QVBoxLayout(scroll_content)
+
+        # Set the main layout of the widget
+        self.setLayout(layout)
+
+        return layout