HYTOS

# -*- coding: utf-8 -*-

# Form implementation generated from reading ui file './UI/LineTypeRegistration.ui'

#

# Created by: PyQt5 UI code generator 5.11.3

#

# WARNING! All changes made in this file will be lost!

from PyQt5 import QtCore, QtGui, QtWidgets

class Ui_Dialog(object):

    def setupUi(self, Dialog):

        Dialog.setObjectName("Dialog")

        Dialog.resize(513, 273)

        self.gridLayout = QtWidgets.QGridLayout(Dialog)

        self.gridLayout.setObjectName("gridLayout")

        self.verticalLayout = QtWidgets.QVBoxLayout()

        self.verticalLayout.setObjectName("verticalLayout")

        self.lineEdit = QtWidgets.QLineEdit(Dialog)

        self.lineEdit.setEnabled(False)

        self.lineEdit.setReadOnly(False)

        self.lineEdit.setObjectName("lineEdit")

        self.verticalLayout.addWidget(self.lineEdit)

        self.horizontalLayout = QtWidgets.QHBoxLayout()

        self.horizontalLayout.setObjectName("horizontalLayout")

        self.verticalLayout.addLayout(self.horizontalLayout)

        self.buttonBox = QtWidgets.QDialogButtonBox(Dialog)

        self.buttonBox.setOrientation(QtCore.Qt.Horizontal)

        self.buttonBox.setStandardButtons(QtWidgets.QDialogButtonBox.Cancel|QtWidgets.QDialogButtonBox.Save)

        self.buttonBox.setObjectName("buttonBox")

        self.verticalLayout.addWidget(self.buttonBox)

        self.gridLayout.addLayout(self.verticalLayout, 0, 0, 1, 1)

        self.retranslateUi(Dialog)

        self.buttonBox.accepted.connect(Dialog.accept)

        self.buttonBox.rejected.connect(Dialog.reject)

        QtCore.QMetaObject.connectSlotsByName(Dialog)

    def retranslateUi(self, Dialog):

        _translate = QtCore.QCoreApplication.translate

        Dialog.setWindowTitle(_translate("Dialog", "Line Type Registration"))

if __name__ == "__main__":

    import sys

    app = QtWidgets.QApplication(sys.argv)

    Dialog = QtWidgets.QDialog()

    ui = Ui_Dialog()

    ui.setupUi(Dialog)

    Dialog.show()

    sys.exit(app.exec_())

        self.actionGroup.addAction(self.actionLineRegistration)

        self.actionLineRegistration.triggered.connect(self.onLineTypeRegistration)

    def onLineTypeRegistration(self):

        ''' register line type '''

        if not self.graphicsView.hasImage():

            self.showImageSelectionMessageBox()

            return

        cmd = FenceCommand.FenceCommand(self.graphicsView)

        cmd.onSuccess.connect(self.onLineType)

        self.graphicsView.command = cmd

        QApplication.setOverrideCursor(Qt.CrossCursor)

    def onLineType(self):

        pass

        self.actionLineRegistration = QtWidgets.QAction(MainWindow)

        icon27 = QtGui.QIcon()

        icon27.addPixmap(QtGui.QPixmap(":/newPrefix/crop.png"), QtGui.QIcon.Normal, QtGui.QIcon.Off)

        self.actionLineRegistration.setIcon(icon27)

        self.actionLineRegistration.setObjectName("actionLineRegistration")

        self.toolBar.addSeparator()

        self.toolBar.addAction(self.actionLineRegistration)

        self.actionLineRegistration.setText(_translate("MainWindow", "Line Registration"))

        self.actionLineRegistration.setToolTip(_translate("MainWindow", "Line Registration"))

                    # change line type using visual pattern

                    try:

                        Worker.changeVisualLineType(app_doc_data.lines)

                    except Exception as ex:

                        message = f"error occurred({repr(ex)}) in {sys.exc_info()[-1].tb_frame.f_code.co_filename}:" \

                                  f"{sys.exc_info()[-1].tb_lineno}"

                        worker.displayLog.emit(MessageType.Error, message)

                    # up to here

    def changeVisualLineType(lines):

        from LineDetector import LineDetector

        app_doc_data = AppDocData.instance()

        image = app_doc_data.activeDrawing.image_origin

        imgNot = np.ones(image.shape, np.uint8) * 255

        image = cv2.bitwise_xor(image, imgNot)

        electric = [137, [1,1,1,1,1,1,1,1,1], sys.maxsize]

        software = [187, [0.948,1.081,0.932,1.081,0.932,1.068,0.932,1.081,0.929], sys.maxsize]

        line_patterns = {'Electric':electric, 'Software':software }

        lines_found = []

        for line in lines:

            vertical = LineDetector.is_vertical([line.start_point()[0], line.start_point()[1], line.end_point()[0], line.end_point()[1]])

            rect = line.boundingRect()

            image_line = image[round(rect.y()):round(rect.y() + rect.height()), round(rect.x()):round(rect.x() + rect.width())]

            contours, _ = cv2.findContours(image_line, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

            # skip piping line

            if len(contours) < 3:

                continue

            i = 1 if vertical else 0

            boundingBoxes = [cv2.boundingRect(contour) for contour in contours]

            (contours, boundingBoxes) = zip(*sorted(zip(contours, boundingBoxes), key=lambda b:b[1][i], reverse=False))

            avg_area = sum([cv2.contourArea(contour) for contour in contours]) / len(contours)

            ratio_area = [cv2.contourArea(contour) / avg_area for contour in contours]

            for line_type, line_pattern in line_patterns.items():

                line_pattern[2] = sys.maxsize

                ratio_area_cal = [ratio * avg_area / line_pattern[0] for ratio in ratio_area]

                long_line = ratio_area_cal if len(ratio_area_cal) > len(line_pattern[1]) else line_pattern[1]

                short_line = line_pattern[1] if len(ratio_area_cal) > len(line_pattern[1]) else ratio_area_cal

                min_error = sys.maxsize

                for offset in range(len(long_line) - len(short_line) + 1):

                    error = 0

                    for index in range(len(short_line)):

                        error += abs(short_line[index] - long_line[index + offset])

                    error = error / len(short_line)

                    if error < min_error:

                        min_error = error

                line_pattern[2] = min_error

            line_type_found = sorted([(line_type, line_pattern[2]) for line_type, line_pattern in line_patterns.items()], key=lambda error:error[1])[0]

            if line_type_found[1] < 0.4:

                lines_found.append([line, line_type_found])

            '''

            orb = cv2.ORB_create()

            kp1, des1 = orb.detectAndCompute(image_line, None)

            kp2, des2 = orb.detectAndCompute(image_line, None)

            bf = cv2.BFMatcher(cv2.NORM_HAMMING, crossCheck=True)

            matches = bf.match(des1, des2)

            matches = sorted(matches, key=lambda x: x.distance)

            good = []

            for m, n in matches:

                if m.distance < 0.75 * n.distance:

                    good.append([m])

            sift = cv2.xfeatures2d.SIFT_create()

            kp1, des1 = sift.detectAndCompute(image_line, None)

            kp2, des2 = sift.detectAndCompute(image_line, None)

            bf = cv2.BFMatcher(cv2.NORM_L2, crossCheck=True)

            matches = bf.match(des1, des2)

            matches = sorted(matches, key=lambda x: x.distance)

            good = []

            for m, n in matches:

                if m.distance < 0.75 * n.distance:

                    good.append([m])

            '''

        line_runs = []

        for line_found in lines_found:

            inserted = False

            for line_run in line_runs:

                if line_found[0] in line_run:

                    inserted = True

                    break

            if inserted:

                continue

            else:

                line_runs.append(Worker.find_connected_line(line_found[0]))

        for line_run in line_runs:

            _lines_found = []

            for _line in line_run:

                _lines = [line_found[0] for line_found in lines_found]

                if _line in _lines:

                    index = _lines.index(_line)

                    _lines_found.append(lines_found[index])

            _line_found = sorted(_lines_found, key=lambda param:param[1][1])[0]

            Worker.changeConnectedLineType(_line_found[0], _line_found[1][0])

    @staticmethod

    def find_connected_line(line):

        lines = [line]

        current_line = line

        while True:

            if type(current_line.connectors[0].connectedItem) is QEngineeringLineItem:

                current_line = current_line.connectors[0].connectedItem

                lines.append(current_line)

            else:

                break

        current_line = line

        while True:

            if type(current_line.connectors[1].connectedItem) is QEngineeringLineItem:

                current_line = current_line.connectors[1].connectedItem

                lines.append(current_line)

            else:

                break

        return lines

    @staticmethod

<?xml version="1.0" encoding="UTF-8"?>

<ui version="4.0">

 <class>Dialog</class>

 <widget class="QDialog" name="Dialog">

  <property name="geometry">

   <rect>

    <x>0</x>

    <y>0</y>

    <width>513</width>

    <height>273</height>

   </rect>

  </property>

  <property name="windowTitle">

   <string>Line Type Registration</string>

  </property>

  <layout class="QGridLayout" name="gridLayout">

   <item row="0" column="0">

    <layout class="QVBoxLayout" name="verticalLayout">

     <item>

      <widget class="QLineEdit" name="lineEdit">

       <property name="enabled">

        <bool>false</bool>

       </property>

       <property name="readOnly">

        <bool>false</bool>

       </property>

      </widget>

     </item>

     <item>

      <layout class="QHBoxLayout" name="horizontalLayout"/>

     </item>

     <item>

      <widget class="QDialogButtonBox" name="buttonBox">

       <property name="orientation">

        <enum>Qt::Horizontal</enum>

       </property>

       <property name="standardButtons">

        <set>QDialogButtonBox::Cancel|QDialogButtonBox::Save</set>

       </property>

      </widget>

     </item>

    </layout>

   </item>

  </layout>

 </widget>

 <resources/>

 <connections>

  <connection>

   <sender>buttonBox</sender>

   <signal>accepted()</signal>

   <receiver>Dialog</receiver>

   <slot>accept()</slot>

   <hints>

    <hint type="sourcelabel">

     <x>248</x>

     <y>254</y>

    </hint>

    <hint type="destinationlabel">

     <x>157</x>

     <y>274</y>

    </hint>

   </hints>

  </connection>

  <connection>

   <sender>buttonBox</sender>

   <signal>rejected()</signal>

   <receiver>Dialog</receiver>

   <slot>reject()</slot>

   <hints>

    <hint type="sourcelabel">

     <x>316</x>

     <y>260</y>

    </hint>

    <hint type="destinationlabel">

     <x>286</x>

     <y>274</y>

    </hint>

   </hints>

  </connection>

 </connections>

</ui>

   <addaction name="separator"/>

   <addaction name="actionLineRegistration"/>

  <action name="actionLineRegistration">

   <property name="icon">

    <iconset resource="../res/MainWindow.qrc">

     <normaloff>:/newPrefix/crop.png</normaloff>:/newPrefix/crop.png</iconset>

   </property>

   <property name="text">

    <string>Line Registration</string>

   </property>

   <property name="toolTip">

    <string>Line Registration</string>

   </property>

  </action>