Source code for viiapackage.tools.start_tool.viia_start_gui

### ===================================================================================================================
###   START TOOL - GUI
### ===================================================================================================================
# Copyright ©VIIA 2025

### ===================================================================================================================
###    1. Import modules
### ===================================================================================================================

# General imports
import os
import sys
from pathlib import Path

# References for functions and classes in the fem-client
from rhdhv_fem.fem_tools import fem_create_folder

# References for functions and classes in the viiaPackage
from viiapackage.tools.start_tool.gui_files.start_mainwindow import Ui_MainWindow
from viiapackage.tools.start_tool.gui_files.start_selectwindow import Ui_SelectWindow
from viiapackage.tools.start_tool.viia_start_files import viia_create_user_config, \
    viia_copy_and_update_model_script_template, viia_copy_and_update_main_script_template, \
    viia_copy_and_update_result_script_template, viia_copy_and_update_report_script_template, \
    viia_copy_and_update_myviia_dashboard_tool_template, viia_create_box_shortcut, viia_create_protocol_shortcut, \
    viia_create_bod_shortcut, viia_download_cheat_sheet
from viiapackage.database.viia_myviia import myviia_login
from viiapackage.database.viia_myviia_get_info import myviia_get_all_objects, myviia_get_object, myviia_get_user_info, \
    myviia_get_all_objects_for_user
from viiapackage import installed_modules
from viiapackage.viiaSettings import ViiaSettings

# Import module PyQt5, check if module is installed
# This module is used for the GUI for the NLKA assessment tool
if 'PyQt5' in installed_modules:
    if 'diana' in installed_modules:
        import warnings
        warnings.warn(
            "PyQt5 modules is installed in DIANA but shouldn't be used in DIANA. Please uninstall", ImportWarning)
    else:
        from PyQt5 import QtWidgets, QtGui


### ===================================================================================================================
###    2. Start tool GUI main-window
### ===================================================================================================================

[docs]class StartGUI(Ui_MainWindow):
    """ This is the first window user enters, in the start tool and collects the user credentials and stores the in the
    user-config file that is created."""

[docs]    @staticmethod
    def start():
        """ Static method to initiate and run the GUI."""
        # Initiate the application
        app = QtWidgets.QApplication(sys.argv)
        main_window = QtWidgets.QMainWindow()

        # Initiate the GUI
        ui = StartGUI(main_window)
        main_window.show()

        # Exit the application
        sys.exit(app.exec_())

[docs]    def start_select_gui(self, token: str):
        # This is temporarily changed due to New NPR transition, will be restored
        """ Static method to initiate and run the select window for the start tool."""
        self.window = QtWidgets.QMainWindow()
        self.ui = StartSelectGUI(self.window, token)
        self.main_window.hide()
        self.window.show()

    def __init__(self, main_window):
        """ Initiate all the relevant setups."""
        self.window = None
        self.ui = None
        self.main_window = main_window
        self.setupUi(main_window)

        # Add icons to the gui window
        icon_path = Path(__file__).parent / 'gui_files/gui_images/RHDHV.png'
        app_icon = QtGui.QIcon(icon_path.as_posix())
        app_icon.addPixmap(QtGui.QPixmap(icon_path.as_posix()), QtGui.QIcon.Normal, QtGui.QIcon.Off)
        viia_logo = Path(__file__).parent / 'gui_files/gui_images/logoVIIA.png'
        self.viia_logo_label.setPixmap(QtGui.QPixmap(viia_logo.as_posix()))

        main_window.setWindowIcon(app_icon)

        # Set the Okay button action
        self.push_button_ok.clicked.connect(self.slot_for_ok_button)

[docs]    def slot_for_ok_button(self):
        """The action slot for the ok button"""
        self.label_warning.clear()

        # Set the username and password for MYVIIA
        email = f"{self.input_for_email.text()}@viiagroningen.nl"
        pword = self.input_for_pw.text()
        os.environ['USERNAME_MYVIIA'] = email
        os.environ['PASSWORD_MYVIIA'] = pword

        # Try to log in MYVIIA
        try:
            token = myviia_login()
        except Exception as e:
            self.label_warning.setText(str(e))
            self.label_warning.setStyleSheet('color: red')
            return None

        # Create the user-config file
        message = viia_create_user_config(folder=Path.cwd(), email=email, password=pword)
        self.label_warning.setStyleSheet('color: black')
        self.label_warning.setText(message)

        # Start the selection of the VIIA object
        self.start_select_gui(token=token)


### ===================================================================================================================
###    3. Start tool GUI select-window
### ===================================================================================================================

[docs]class StartSelectGUI(Ui_SelectWindow):
    """ This is the second window user enters, in the start tool and collects the required object and object-part and
    creates the model script, main script and result script. General info is entered in the scripts."""

    def __init__(self, window, token):
        """ Initiate all the relevant setups."""
        self.window = window
        self.setupUi(window)
        self.token = token

        # Add icons to the gui window
        icon_path = Path(__file__).parent / 'gui_files/gui_images/RHDHV.png'
        app_icon = QtGui.QIcon(icon_path.as_posix())
        app_icon.addPixmap(QtGui.QPixmap(icon_path.as_posix()), QtGui.QIcon.Normal, QtGui.QIcon.Off)
        viia_logo = Path(__file__).parent / 'gui_files/gui_images/logoVIIA.png'
        self.viia_logo_label.setPixmap(QtGui.QPixmap(viia_logo.as_posix()))
        self.window.setWindowIcon(app_icon)

        # Get list of viia-objects and the assigned objects
        self.viia_objects = myviia_get_all_objects(token=token)
        self.assigned_viia_objects = myviia_get_all_objects_for_user(token=token)
        if not self.assigned_viia_objects:
            self.radio_button_assigned.setChecked(False)
            self.radio_button_all.setChecked(True)
            self.set_all_objects()
        else:
            self.set_assigned_objects()
        self.selected_viia_object = None

        # Actions for the radio-buttons
        self.radio_button_assigned.clicked.connect(self.set_assigned_objects)
        self.radio_button_all.clicked.connect(self.set_all_objects)

        # Action for update combo-box
        self.comboBox.currentIndexChanged.connect(self.combobox_to_select_conditions)

        # Set the Okay button action
        self.push_button_ok.clicked.connect(self.slot_for_ok_button)

[docs]    def objecten_combo_box(self, viia_objects):
        """ This is the slot for combobox VIIA objects."""
        self.comboBox.clear()
        self.comboBox.addItems(viia_objects)

[docs]    def set_assigned_objects(self):
        """ Show the list of assigned objects only."""
        self.objecten_combo_box(
            viia_objects=['None'] + [item['viiaobject_nummer'] for item in self.assigned_viia_objects])

[docs]    def set_all_objects(self):
        """ Show list of all objects from MYVIIA."""
        self.objecten_combo_box(
            viia_objects=sorted(['None'] + [item['objectnummer_viia'] for item in self.viia_objects]))

[docs]    def combobox_to_select_conditions(self):
        """ This is the slot for combobox behavior factor."""
        objectname = self.comboBox.currentText()
        if objectname != 'None':
            for viia_object in self.viia_objects:
                if viia_object['objectnummer_viia'] == objectname:
                    self.objectparts_combo_box(viia_object_id=viia_object['id'])

[docs]    def objectparts_combo_box(self, viia_object_id: int = None):
        """ This is the slot for combobox object-parts, based on the selection of the VIIA object."""
        self.comboBox_2.clear()
        self.selected_viia_object = None
        if viia_object_id is not None:
            self.selected_viia_object = myviia_get_object(object_id=viia_object_id, token=self.token)
            parts = [part['naam'] for part in self.selected_viia_object['objectdelen'] if part['rekenmethodiek_id']]
            self.comboBox_2.addItems(parts)

[docs]    def slot_for_ok_button(self):
        """ The action slot for the ok button"""
        self.label_warning.clear()
        viia_object_name = self.comboBox.currentText()

        if viia_object_name == 'None':
            self.window.close()
            return

        if self.selected_viia_object is None or self.selected_viia_object['objectnummer_viia'] != viia_object_name:
            self.label_warning.setText("ERROR: An error occurred, please try again.")
            self.label_warning.setStyleSheet('color: red')
            return

        # Collect the rekenmethodiek
        objectpart_name = self.comboBox_2.currentText()
        analyse_type = None
        upr_versie = None
        box_folder = None
        for objectpart in self.selected_viia_object['objectdelen']:
            if objectpart['naam'] == objectpart_name:
                if objectpart['rekenmethodiek_id'] is not None:
                    analyse_type = objectpart['rekenmethodiek_id']['rekenmethodiek']
                if 'upr_versie' in objectpart and objectpart['upr_versie']:
                    upr_versie = objectpart['upr_versie']['upr_versie']
        if 'boxmap' in self.selected_viia_object:
            box_folder = self.selected_viia_object['boxmap']
        if analyse_type is None or analyse_type.upper() not in ['NLTH', 'NLPO', 'SBS']:
            self.label_warning.setText("ERROR: The analysis type for this model is not NLTH, NLPO, or SBS.")
            self.label_warning.setStyleSheet('color: red')
            return

        # Get initials from MYVIIA
        initials = 'XXX'
        user = myviia_get_user_info()
        if user and user['initialen']:
            initials = user['initialen']

        # Foundation type
        foundation_type = 'shallow'
        if self.radio_button_shallow.isChecked():
            foundation_type = 'shallow'
        elif self.radio_button_piles.isChecked():
            foundation_type = 'piles'
        if self.radio_button_mixed.isChecked():
            foundation_type = 'mixed'

        # Create the files
        if analyse_type and objectpart_name:
            viia_copy_and_update_model_script_template(
                folder=Path.cwd(), object_nr=viia_object_name, initials=initials)
            viia_copy_and_update_main_script_template(
                folder=Path.cwd(), object_nr=viia_object_name, initials=initials, analysis_type=analyse_type,
                object_part=objectpart_name, foundation_type=foundation_type)
            viia_copy_and_update_result_script_template(
                folder=Path.cwd(), object_nr=viia_object_name, initials=initials, analysis_type=analyse_type,
                object_part=objectpart_name, abaqus=self.radio_button_abaqus.isChecked())
            viia_copy_and_update_report_script_template(
                folder=Path.cwd(), object_nr=viia_object_name, initials=initials, analysis_type=analyse_type,
                object_part=objectpart_name)
            viia_copy_and_update_myviia_dashboard_tool_template(folder=Path.cwd())
            viia_create_box_shortcut(object_nr=viia_object_name, box_folder_link=box_folder, folder=Path.cwd())
        viia_create_protocol_shortcut(folder=Path.cwd())
        if upr_versie and upr_versie in ViiaSettings.upr_versions:
            viia_create_bod_shortcut(
                bod_version=upr_versie, bod_folder_link=ViiaSettings.upr_versions[upr_versie], folder=Path.cwd())
        viia_download_cheat_sheet(folder=Path.cwd())

        # Create the folder for the model
        fem_create_folder('model')

        # Close the window
        self.window.close()

### ===================================================================================================================
###    3. End of script
### ===================================================================================================================