IfcOpenShell/src/ifcbimtester/bimtester/guiwidget.py

# TODO: improve layout, start with feature file path and beside button !!!!!
# TODO: if browse widgets will be canceled, last QLineEdit should be restored

import os

from PySide2 import QtCore
from PySide2 import QtGui
from PySide2 import QtWidgets

from .run import run_all


class GuiWidgetBimTester(QtWidgets.QWidget):

    # get some initial values
    this_path = os.path.join(os.path.dirname(__file__))
    desktop_path = os.path.join(os.path.expanduser("~"), "Desktop")

    initial_ifcfile = "/home/hugo/Documents/zeug_sort/z_some_ifc/example_model.ifc"
    if not os.path.isfile(initial_ifcfile):
        initial_ifcfile = desktop_path
    initial_featurespath = os.path.join(this_path, "..", "..", "features_bimtester", "fea_min")
    if not os.path.isdir(initial_featurespath):
        initial_featurespath = desktop_path

    def __init__(self):
        super(GuiWidgetBimTester, self).__init__()
        self._setup_ui()

    def __del__(self,):
        # need as fix for qt event error
        # http://forum.freecadweb.org/viewtopic.php?f=18&t=10732&start=10#p86493
        return

    def _setup_ui(self):

        # a lot code is taken from FreeCAD FEM solver frame work task panel
        # https://forum.freecadweb.org/viewtopic.php?f=10&t=51419
        # use a browse button, and a line edit
        # the browse button opens a file dialog, which will set the line edit

        # icon
        # print(__file__)
        package_path = os.path.dirname(os.path.realpath(__file__))
        iconpath = os.path.join(
            package_path, "resources", "icons", "bimtester.ico"
        )
        """
        # as svg
        # https://stackoverflow.com/a/35138314
        theicon = QtSvg.QSvgWidget(iconpath)
        # none works ...
        #theicon.setGeometry(20,20,200,200)
        #theicon.setSizePolicy(QtGui.QSizePolicy.Policy.Maximum, QtGui.QSizePolicy.Policy.Maximum)
        #theicon.sizeHint()
        """
        # as pixmap
        theicon = QtWidgets.QLabel(self)
        iconpixmap = QtGui.QPixmap(iconpath)
        iconpixmap = iconpixmap.scaled(100, 100, QtCore.Qt.KeepAspectRatio)
        theicon.setPixmap(iconpixmap)

        # ifc file
        _ifcfile_label = QtWidgets.QLabel("IFC file", self)
        self.ifcfile_text = QtWidgets.QLineEdit()
        self.set_ifcfile(self.initial_ifcfile)
        _ifcfile_browse_btn = QtWidgets.QToolButton()
        _ifcfile_browse_btn.setText("...")
        _ifcfile_browse_btn.clicked.connect(self.select_ifcfile)

        # feature files path
        # use a layout with a frame and a title, see solver framework tp
        # beside button
        ffifc_str = (
            "Feature files beside IFC file. "
            "Feature files in directory features."
        )
        featuredirfromifc_label = QtWidgets.QLabel(ffifc_str, self)
        self.featuredirfromifc_cb = QtWidgets.QCheckBox(self)
        self.featuredirfromifc_cb.stateChanged.connect(
            self.featuredirfromifc_clicked
        )

        # path browser and line edit
        _ffdir_str = (
            "Feature files directory. "
            "Feature files in directory features."
        )
        _featurefilesdir_label = QtWidgets.QLabel(_ffdir_str, self)
        self.featurefilesdir_text = QtWidgets.QLineEdit()
        self.set_featurefilesdir(self.initial_featurespath)
        self.feafilesdir_browse_btn = QtWidgets.QToolButton()
        self.feafilesdir_browse_btn.setText("...")
        self.feafilesdir_browse_btn.clicked.connect(
            self.select_featurefilesdir
        )

        # buttons
        self.run_button = QtWidgets.QPushButton(
            QtGui.QIcon.fromTheme("document-new"), "Run"
        )
        self.close_button = QtWidgets.QPushButton(
            QtGui.QIcon.fromTheme("window-close"), "Close"
        )
        self.run_button.clicked.connect(self.run_bimtester)
        self.close_button.clicked.connect(self.close_widget)
        _buttons = QtWidgets.QHBoxLayout()
        _buttons.addWidget(self.run_button)
        _buttons.addWidget(self.close_button)

        # Layout:
        layout = QtWidgets.QGridLayout()
        layout.addWidget(theicon, 1, 0, alignment=QtCore.Qt.AlignRight)
        layout.addWidget(_ifcfile_label, 2, 0)
        layout.addWidget(self.ifcfile_text, 3, 0)
        layout.addWidget(_ifcfile_browse_btn, 3, 1)
        layout.addWidget(_featurefilesdir_label, 4, 0)
        layout.addWidget(self.featurefilesdir_text, 5, 0)
        layout.addWidget(self.feafilesdir_browse_btn, 5, 1)
        layout.addWidget(featuredirfromifc_label, 6, 0)
        layout.addWidget(self.featuredirfromifc_cb, 6, 1)
        layout.addLayout(_buttons, 7, 0)
        # row stretches by 10 compared to the others, std is 0
        # first parameter is the row number
        # second is the stretch factor.
        layout.setRowStretch(0, 10)
        self.setLayout(layout)

    # **********************************************************
    def select_ifcfile(self):
        # print(self.get_ifcfile())
        # print(os.path.isfile(self.get_ifcfile()))
        ifcfile = QtWidgets.QFileDialog.getOpenFileName(
            self,
            dir=self.get_ifcfile()
        )[0]
        self.set_ifcfile(ifcfile)

    def set_ifcfile(self, a_file):
        self.ifcfile_text.setText(a_file)

    def get_ifcfile(self):
        return self.ifcfile_text.text()

    def featuredirfromifc_clicked(self):
        if self.featuredirfromifc_cb.isChecked() is True:
            self.set_featurefilesdir("")
            # TODO
            self.featurefilesdir_text.setEnabled(False)
            self.feafilesdir_browse_btn.setEnabled(False)
            # deactivate feature path browser button
            # deactivate lineedit text
        else:
            self.set_featurefilesdir(self.initial_featurespath)
            self.featurefilesdir_text.setEnabled(True)
            self.feafilesdir_browse_btn.setEnabled(True)

    def select_featurefilesdir(self):
        thedir = self.featurefilesdir_text.text()
        # print(thedir)
        # print(os.path.isdir(thedir))
        # hidden directories are only shown if the option is set
        features_path = QtWidgets.QFileDialog.getExistingDirectory(
            self,
            caption="Choose features directory ...",
            dir=thedir,
            options=QtWidgets.QFileDialog.HideNameFilterDetails
        )
        self.set_featurefilesdir(features_path)

    def set_featurefilesdir(self, a_directory):
        self.featurefilesdir_text.setText(a_directory)

    def get_featurefilesdir(self):
        return self.featurefilesdir_text.text()

    # **********************************************************
    def run_bimtester(self):
        print("Run BIMTester")
        QtWidgets.QApplication.setOverrideCursor(QtCore.Qt.WaitCursor)

        # get input values
        splitifcpath = os.path.split(self.get_ifcfile())
        the_ifcfile_path, the_ifcfile_name = splitifcpath[0], splitifcpath[1]
        if self.featuredirfromifc_cb.isChecked() is True:
            the_features_path = the_ifcfile_path
            print(
                "Make sure the feature files are beside "
                "the ifc file in a directory named 'features'."
            )
        else:
            the_features_path = self.get_featurefilesdir()
        print(the_features_path)
        print(the_ifcfile_path)
        print(the_ifcfile_name)

        # run bimtester
        status = run_all(
            the_features_path,
            the_ifcfile_path,
            the_ifcfile_name
        )
        print(status)

        QtWidgets.QApplication.restoreOverrideCursor()

    def close_widget(self):
        print("Close BIMTester Gui")
        self.close()

    def closeEvent(self, ev):
        pw = self.parentWidget()
        if pw and pw.inherits("QDockWidget"):
            pw.deleteLater()