HYTOS

# coding: utf-8

import os

import sys

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

import sqlite3

from AppDocData import AppDocData

from AppDocData import Config

from AppDocData import Color

import Configuration_UI

class ListView(QListView):

    def __init__(self, *args, **kwargs):

        super(ListView, self).__init__(*args, **kwargs)

    '''

        @brief  deleted selected item

        @author humkyung

        @date   2018.04.09

    '''

    def removeSelectedItem(self):

        selectedIndexes = self.selectedIndexes()

        selectedRows = [item.row() for item in selectedIndexes]

        model = self.model()

        # iterates each selected row in descending order

        for row in sorted(selectedRows, reverse=True):

            # remove the row from the model

            model.removeRow(row)

    '''

        @brief  key press event

        @author humkyung

        @date   2018.04.09

    '''

    def keyPressEvent(self, event):

        if event.key() == Qt.Key_Delete:

            self.removeSelectedItem()

            return

        super(ListView, self).keyPressEvent(event)

class QConfigurationDialog(QDialog):

    '''

        @history    humkyung 2018.05.05 read configuration for instrument and opc tag no rule

                    humkyung 2018.05.09 read line no tag rule configuration

                    Jeongwoo 2018.05.18 read Small Line Minimum Length

                    Jeongwoo 2018.06.04 read Min/Max Text Size

                    Jeongwoo 2018.06.05 read Text Area Detection Method

                    humkyung 2018.06.20 add expand,shrink and merge size for recognizing text

                    humkyung 2018.06.29 add line type table

                    kyouho 2018.07.04 add self.delimiter = '"'

    '''

    def __init__(self, parent):

        QDialog.__init__(self, parent)

        self.ui = Configuration_UI.Ui_ConfigurationDialog()

        self.ui.setupUi(self)

        self.ui.listViewProperties = ListView(self.ui.groupBoxLineNo)

        self.ui.listViewProperties.setObjectName('listViewProperties')

        self.ui.horizontalLayout_3.addWidget(self.ui.listViewProperties)

        self.isAccepted = False

        self.itemModel = QStandardItemModel()

        self.delimiter = '"'

        self.defaultColor = Color(0, 0, 0, 255)

        self.currentIndex = 0

        docData = AppDocData.instance()

        configs = docData.getConfigs('Text Area', 'Text Area')

        value = int(configs[0].value) if 1 == len(configs) else 0

        if value == 0:

            self.ui.textAreaTypeARadioButton.setChecked(True)

        else:

            self.ui.textAreaTypeBRadioButton.setChecked(True)

        configs = docData.getConfigs('Text Recognition', 'Expand Size')

        self.ui.spinBoxExpandSize.setValue(int(configs[0].value)) if 1 == len(configs) else self.ui.spinBoxExpandSize.setValue(10)

        configs = docData.getConfigs('Text Recognition', 'Shrink Size')

        self.ui.spinBoxShrinkSize.setValue(int(configs[0].value)) if 1 == len(configs) else self.ui.spinBoxShrinkSize.setValue(0)

        configs = docData.getConfigs('Text Recognition', 'Merge Size')

        self.ui.spinBoxMergeSize.setValue(int(configs[0].value)) if 1 == len(configs) else self.ui.spinBoxMergeSize.setValue(10)

        # text replace options - 2018.07.03 added by humkyung

        configs = docData.getConfigs('Text Recognition', 'Remove White Space')

        self.ui.checkBoxRemoveWhiteSpace.setChecked(('True' == configs[0].value) if 1 == len(configs) else False)

        configs = docData.getConfigs('Text Recognition', 'OldStr')

        self.ui.lineEditOldStr.setText(configs[0].value if 1 == len(configs) else '')

        configs = docData.getConfigs('Text Recognition', 'NewStr')

        self.ui.lineEditNewStr.setText(configs[0].value if 1 == len(configs) else '')

        # up to here

        configs = docData.getConfigs('Text Size', 'Min Text Size')

        self.ui.minTextSizeSpinBox.setValue(int(configs[0].value)) if 1 == len(configs) else self.ui.minTextSizeSpinBox.setValue(30)

        configs = docData.getConfigs('Text Size', 'Max Text Size')

        self.ui.maxTextSizeSpinBox.setValue(int(configs[0].value)) if 1 == len(configs) else self.ui.maxTextSizeSpinBox.setValue(60)

        configs = docData.getConfigs('Size', 'Delimiter')

        self.ui.lineEditSizeDelimiter.setText(configs[0].value if 1 == len(configs) else 'X')

        # set min,max area for small object

        configs = docData.getConfigs('Small Object Size', 'Min Area')

        self.ui.spinBoxMinArea.setValue(int(configs[0].value)) if 1 == len(configs) else self.ui.spinBoxMinArea.setValue(20)

        configs = docData.getConfigs('Small Object Size', 'Max Area')

        self.ui.spinBoxMaxArea.setValue(int(configs[0].value)) if 1 == len(configs) else self.ui.spinBoxMaxArea.setValue(50)

        # up to here

        windowSize = docData.getSlidingWindowSize()

        self.ui.spinBoxWidth.setValue(windowSize[0])

        self.ui.spinBoxHeight.setValue(windowSize[1])

        configs = docData.getConfigs('Small Line Minimum Length', 'Min Length')

        self.ui.smallLineMinLengthSpinBox.setValue(int(configs[0].value)) if 1 == len(configs) else self.ui.smallLineMinLengthSpinBox.setValue(10)

        configs = docData.getConfigs('Line No', 'Size Unit')

        if 1 == len(configs):

            self.ui.radioButtonMetric.setChecked(True) if 'Metric' == configs[0].value else self.ui.radioButtonInch.setChecked(True)

        else:

            self.ui.radioButtonMetric.setChecked(True)

        configs = docData.getConfigs('Line No', 'Delimiter')

        if 1 == len(configs):

            self.ui.lineEdit.setText(configs[0].value)

        properties = docData.getLineProperties()

        for prop in properties:

            self.ui.comboBoxProperties.addItem(prop)

            if prop != "Tag Seq No":

                self.ui.comboBoxColorOption.addItem(prop)

        configs = docData.getConfigs('Line No', 'Configuration')

        if len(configs) == 1 and configs[0].value is not None:

            for value in configs[0].value.split(self.delimiter):

            #for value in configs[0].value.split(self.ui.lineEdit.text()):

                self.itemModel.appendRow(QStandardItem(value))

        self.ui.listViewProperties.setModel(self.itemModel)

        configs = docData.getConfigs('Line No Tag Rule', 'Tag Seq No')

        self.ui.lineEditLineNoTagSeqNo.setText(configs[0].value if configs else '')

        configs = docData.getConfigs('Instrument Tag No Rule', 'Measured Variable Code')

        self.ui.lineEditMeasuredVariableCode.setText(configs[0].value if configs else '')

        configs = docData.getConfigs('Instrument Tag No Rule', 'Type Modifier')

        self.ui.lineEditTypeModifier.setText(configs[0].value if configs else '')

        configs = docData.getConfigs('Instrument Tag No Rule', 'Tag Seq No')

        self.ui.lineEditTagSeqNo.setText(configs[0].value if configs else '')

        configs = docData.getConfigs('Instrument Tag No Rule', 'Tag Suffix')

        self.ui.lineEditTagSuffix.setText(configs[0].value if configs else '')

        configs = docData.getConfigs('OPC Tag No Rule', 'Description')

        self.ui.lineEditDescription.setText(configs[0].value if configs else '')

        configs = docData.getConfigs('OPC Tag No Rule', 'OPC Tag')

        self.ui.lineEditOPCTag.setText(configs[0].value if configs else '')

        configs = docData.getConfigs('LineTypes')

        lineTypes = docData.getLineTypes()

        self.ui.tableWidgetLineTypes.setColumnCount(3)

        self.ui.tableWidgetLineTypes.setHorizontalHeaderLabels(['Name', 'Width', 'Style'])

        self.ui.tableWidgetLineTypes.setRowCount(len(lineTypes))

        row = 0

        for lineType in lineTypes:

            item = QTableWidgetItem(lineType)

            item.setFlags(Qt.ItemIsEnabled)

            self.ui.tableWidgetLineTypes.setItem(row, 0, item)

            ### line width

            lineWidthSpinBox = QSpinBox()

            lineWidthSpinBox.setRange(1, 10)

            lineWidthSpinBox.setSingleStep(1)

            matches = [config for config in configs if config.key == lineType]

            lineWidthSpinBox.setValue(int(matches[0].value.split(',')[0])) if matches else lineWidthSpinBox.setValue(1)

            self.ui.tableWidgetLineTypes.setCellWidget(row, 1, lineWidthSpinBox)

            ## line style

            lineStyleComboBox = QComboBox()

            lineStyleComboBox.addItems(['SolidLine', 'DashLine', 'DotLine', 'DashDotLine', 'DashDotDotLine', 'CustomDashLine'])

            lineStyleComboBox.setCurrentText(matches[0].value.split(',')[1]) if matches else lineStyleComboBox.setCurrentText('SolidLine')

            self.ui.tableWidgetLineTypes.setCellWidget(row, 2, lineStyleComboBox)

            row += 1

        self.ui.tableWidgetLineTypes.horizontalHeaderItem(0).setSizeHint(QSize(30, 30))

        self.ui.labelFontName.setBuddy(self.ui.fontComboBox)

        configs = docData.getConfigs('Text Style', 'Font Name')

        if configs:

            self.ui.fontComboBox.setCurrentFont(QFont(configs[0].value, 10))

        configs = docData.getConfigs('Text Style', 'Font Size')

        if configs:

            size = int(configs[0].value)

            self.ui.radioButtonAutoSize.setChecked(True if size == -1 else False)

            self.ui.radioButtonFixedSize.setChecked(True if size != -1 else False)

            self.ui.spinBoxFontSize.setValue(size if size != -1 else 10)

            self.ui.spinBoxFontSize.setEnabled(self.ui.radioButtonFixedSize.isChecked())

        else:

            self.ui.radioButtonAutoSize.setChecked(True)

            self.ui.radioButtonFixedSize.setChecked(False)

            self.ui.spinBoxFontSize.setValue(10)

            self.ui.spinBoxFontSize.setEnabled(self.ui.radioButtonFixedSize.isChecked())

        # Line Color Visible Option 가져옴

        configs = docData.getConfigs('Line Color', 'Visible Option')

        if configs:

            data = configs[0].value

            self.ui.radioButtonRandom.setChecked(True if data == 'Random' else False)

            self.ui.radioButtonProperty.setChecked(True if data == 'Property' else False)

        else:

            self.ui.radioButtonRandom.setChecked(True)

            self.ui.radioButtonProperty.setChecked(False)

        # Color Property 선택값 가져옴

        table = self.ui.tableWidgetColorProperty

        configs = docData.getConfigs('Color Property', 'State')

        selectedOption = configs[0].value if configs and self.ui.comboBoxColorOption.findText(configs[0].value) >= 0 else 'Nominal Diameter'

        self.ui.tableWidgetColorProperty.setHorizontalHeaderLabels(['Value', 'Color', 'ref', 'colorStr'])

        index = self.ui.comboBoxColorOption.findText(selectedOption)

        self.ui.comboBoxColorOption.setCurrentIndex(index)

        self.currentIndex = index

        table.hideColumn(2)

        table.hideColumn(3)

        #Column Header Size

        self.ui.tableWidgetColorProperty.horizontalHeaderItem(0).setSizeHint(QSize(30, 30))

        self.setPropertyToggle(self.ui.radioButtonProperty.isChecked())

        # connect signals and slots

        self.ui.pushButtonAddProperty.clicked.connect(self.addLineProperty)

        self.ui.radioButtonFixedSize.toggled.connect(self.onFixedSizeToggled)

        self.ui.pushButtonAddDelimiter.clicked.connect(self.addDelimiter)

        self.ui.tableWidgetColorProperty.cellDoubleClicked.connect(self.cellDoubleClick)

        self.ui.comboBoxColorOption.currentIndexChanged.connect(self.currentIndexChanged)

        self.ui.radioButtonRandom.toggled.connect(self.onPropertyToggled)

    '''

        @brief  Clear Table

        @author kyouho

        @date   2018.07.10

    '''

    def clearColorPropertyTable(self):

        table = self.ui.tableWidgetColorProperty

        model = table.model()

        while table.rowCount():

            model.removeRow(table.rowCount() - 1)

    '''

        @brief  setting default table

        @author kyouho

        @date   2018.07.09

    '''

    def settingDefaultColorTable(self, property):

        docData = AppDocData.instance()

        table = self.ui.tableWidgetColorProperty

        self.clearColorPropertyTable()

        dic = {}

        if property == "Nominal Diameter":

            #중복 체크 배열

            checkRepeat = []

            # Size 표기 가져옴

            configs = docData.getConfigs('Line No', 'Size Unit')

            # Size 관련 Table 가져옴

            result = docData.getNomialPipeSizeData(False, "Metric")

            for pipeSize in result:

                if not pipeSize.inchStr or not pipeSize.metricStr:

                    continue

                else:

                    replaceInchStr = pipeSize.inchStr.replace("'", '"')

                    if checkRepeat.count(replaceInchStr):

                        continue

                    else:

                        checkRepeat.append(replaceInchStr)

                if configs[0].value == 'Inch':

                    dic[replaceInchStr] = pipeSize.metricStr

                else:

                    dic[pipeSize.metricStr] = pipeSize.metricStr

        elif property != "Tag Seq No":

            result = docData.getCodeTable(property.replace(' ','').replace('&', 'n'), True)

            for fluidCode in result:

                dic[fluidCode] = fluidCode

        if dic:

            table.setRowCount(len(dic))

            row = 0

            for i in dic.keys():

                table.setItem(row, 0, QTableWidgetItem(i))

                table.setItem(row, 1, QTableWidgetItem(''))

                table.setItem(row, 2, QTableWidgetItem(dic[i]))

                table.setItem(row, 3, QTableWidgetItem(''))

                row += 1

    '''

        @brief  setting Color String Cell

        @author kyouho

        @date   2018.07.09

    '''

    def settingColorStringCell(self, property):

        docData = AppDocData.instance()

        table = self.ui.tableWidgetColorProperty

        rowCount = table.rowCount()

        for j in range(rowCount):

            ref = table.item(j, 2)

            configs = docData.getConfigs(property, ref.text())

            if configs:

                colorStr = table.item(j, 3)

                colorStr.setText(configs[0].value)

    '''

        @brief  setting color cell

        @author kyouho

        @date   2018.07.09

    '''

    def settingColorCell(self):

        table = self.ui.tableWidgetColorProperty

        rowCount = table.rowCount()

        for i in range(rowCount):

            colorCell = table.item(i, 1)

            colorDataCell = table.item(i, 3)

            colorStr = colorDataCell.text()

            if colorStr:

                split = colorStr.split(',')

                r = split[0]

                g = split[1]

                b = split[2]

                colorCell.setBackground(QColor(int(r), int(g), int(b)))

            else:

                colorCell.setBackground(QColor(self.defaultColor.red, self.defaultColor.green, self.defaultColor.blue))

    '''

        @brief  Cell Double Click Event 

        @author kyouho

        @date   2018.07.09

    '''

    def currentIndexChanged(self, index):

        if self.currentIndex != index:

            self.currentIndex = index

            selectedOption = self.ui.comboBoxColorOption.currentText()

            # 기본 테이블 셋팅

            self.settingDefaultColorTable(selectedOption)

            # 설정된 색상 가져옴

            self.settingColorStringCell(selectedOption)

            #Table Color Setting

            self.settingColorCell()

    '''

        @brief  Cell Double Click Event 

        @author kyouho

        @date   2018.07.09

    '''

    def cellDoubleClick(self, row, column):

        if column == 1:

            color = QColorDialog.getColor()

            if color.isValid():

                self.ui.tableWidgetColorProperty.setItem(row, 3, QTableWidgetItem(str(color.red()) + ',' + str(color.green()) + ',' + str(color.blue())))

                #Table Color Setting

                self.settingColorCell()

    '''

        @brief  add delimiter 

        @author kyouho

        @date   2018.07.04

    '''

    def addDelimiter(self):

        text = self.ui.lineEdit.text()

        if text != self.delimiter:

            self.itemModel.appendRow(QStandardItem(text))

        else:

            pass

    '''

        @brief  add line property

        @author humkyung

        @date   2018.04.09

    '''

    def addLineProperty(self):

        index = self.ui.comboBoxProperties.currentIndex()

        prop = self.ui.comboBoxProperties.itemText(index)

        self.itemModel.appendRow(QStandardItem(prop))

    '''

        @brief  enable/disable font size spinbox

        @author humkyung

        @date   2018.06.30

    '''

    def onFixedSizeToggled(self, radioButton):

        self.ui.spinBoxFontSize.setEnabled(self.ui.radioButtonFixedSize.isChecked())

    '''

        @brief  property radio button toggle event

        @author kyouho

        @date   2018.07.10

    '''

    def onPropertyToggled(self, radioButton):

        self.setPropertyToggle(self.ui.radioButtonProperty.isChecked())

    '''

        @brief  enable/disable font size spinbox

        @author kyouho

        @date   2018.07.10

    '''

    def setPropertyToggle(self, enable):

        if enable:

            selectedOption = self.ui.comboBoxColorOption.currentText()

            # 기본 테이블 셋팅

            self.settingDefaultColorTable(selectedOption)

            # 설정된 색상 가져옴

            self.settingColorStringCell(selectedOption)

            #Table Color Setting

            self.settingColorCell()

        else:

            self.clearColorPropertyTable()

        self.ui.comboBoxColorOption.setEnabled(enable)

    '''

        @brief      insert or update configurations modified by user

        @author     humkyung

        @date       2018.??.??

        @history    humkyung 2018.04.24 save size unit of line no

                    humkyung 2018.04.26 save min,max area for small object

                    humkyung 2018.05.02 save size delimiter

                    humkyung 2018.05.05 save instrument and opc tag no rule

                    humkyung 2018.05.09 save line no tag rule

                    Jeongwoo 2018.05.18 save Small Line Minimum Length

                    Jeongwoo 2018.06.04 save Min/Max Text Size

                    Jeongwoo 2018.06.05 save Text Area Detection Method

                    humkyung 2018.06.20 save Expand and Shrink size for recognizing text

                    kyouho 2018.07.04 edit cofiguration new delimiter (self.delimiter)

    '''

    def accept(self):

        try:

            docData = AppDocData.instance()

            self.isAccepted = True

            configs = []

            configs.append(Config('Text Area', 'Text Area', 0 if self.ui.textAreaTypeARadioButton.isChecked() else 1))

            configs.append(Config('Text Recognition', 'Expand Size', self.ui.spinBoxExpandSize.value()))

            configs.append(Config('Text Recognition', 'Shrink Size', self.ui.spinBoxShrinkSize.value()))

            configs.append(Config('Text Recognition', 'Merge Size', self.ui.spinBoxMergeSize.value()))

            # save text replace options - 2018.07.03 added by humkyung

            configs.append(Config('Text Recognition', 'Remove White Space', 'True' if self.ui.checkBoxRemoveWhiteSpace.isChecked() else 'False'))

            configs.append(Config('Text Recognition', 'OldStr', self.ui.lineEditOldStr.text()))

            configs.append(Config('Text Recognition', 'NewStr', self.ui.lineEditNewStr.text()))

            # up to here

            configs.append(Config('Text Size', 'Min Text Size', self.ui.minTextSizeSpinBox.value()))

            configs.append(Config('Text Size', 'Max Text Size', self.ui.maxTextSizeSpinBox.value()))

            configs.append(Config('Size', 'Delimiter', self.ui.lineEditSizeDelimiter.text()))

            configs.append(Config('Small Object Size', 'Min Area', self.ui.spinBoxMinArea.value()))

            configs.append(Config('Small Object Size', 'Max Area', self.ui.spinBoxMaxArea.value()))

            configs.append(Config('Sliding Window', 'Width', self.ui.spinBoxWidth.value()))

            configs.append(Config('Sliding Window', 'Height', self.ui.spinBoxHeight.value()))

            configs.append(Config('Small Line Minimum Length', 'Min Length', self.ui.smallLineMinLengthSpinBox.value()))

            configs.append(Config('Line No', 'Size Unit', 'Metric' if self.ui.radioButtonMetric.isChecked() else 'Inch'))

            configs.append(Config('Line No', 'Delimiter', self.ui.lineEdit.text()))

            configs.append(Config('Line No Tag Rule', 'Tag Seq No', self.ui.lineEditLineNoTagSeqNo.text()))

            configs.append(Config('Instrument Tag No Rule', 'Measured Variable Code', self.ui.lineEditMeasuredVariableCode.text()))

            configs.append(Config('Instrument Tag No Rule', 'Type Modifier', self.ui.lineEditTypeModifier.text()))

            configs.append(Config('Instrument Tag No Rule', 'Tag Seq No', self.ui.lineEditTagSeqNo.text()))

            configs.append(Config('Instrument Tag No Rule', 'Tag Suffix', self.ui.lineEditTagSuffix.text()))

            configs.append(Config('OPC Tag No Rule', 'Description', self.ui.lineEditDescription.text()))

            configs.append(Config('OPC Tag No Rule', 'OPC Tag', self.ui.lineEditOPCTag.text()))

            # Add Line Color Option - 2018.07.06 by kyouho

            rbRandomValue = self.ui.radioButtonRandom.isChecked()

            cbColorOptionValue = self.ui.comboBoxColorOption.currentText()

            configs.append(Config('Line Color', 'Visible Option', 'Random' if rbRandomValue else 'Property'))

            configs.append(Config('Color Property', 'State', cbColorOptionValue))

            for row in range(self.ui.tableWidgetLineTypes.rowCount()):

                lineType = self.ui.tableWidgetLineTypes.item(row, 0).text()

                width = self.ui.tableWidgetLineTypes.cellWidget(row, 1).text()

                style = self.ui.tableWidgetLineTypes.cellWidget(row, 2).currentText()

                configs.append(Config('LineTypes', lineType, '{},{}'.format(width, style)))

            if not rbRandomValue:

                table = self.ui.tableWidgetColorProperty

                rowCount = self.ui.tableWidgetColorProperty.rowCount()

                for i in range(rowCount):

                    refStr = table.item(i, 2).text()

                    colorStr = table.item(i, 3).text()

                    if colorStr:

                        configs.append(Config(cbColorOptionValue, refStr, colorStr))

            #Configuration

            configuration = None

            self.ui.listViewProperties.model()

            for i in range(self.itemModel.rowCount()):

                item = self.ui.listViewProperties.model().item(i).text()

                if configuration is None:

                    configuration = item

                else:

                    configuration += self.delimiter + item

                    #configuration += self.ui.lineEdit.text() + item

            configs.append(Config('Line No', 'Configuration', configuration))

            font = self.ui.fontComboBox.currentFont()

            configs.append(Config('Text Style', 'Font Name', font.family()))

            configs.append(Config('Text Style', 'Font Size', str(self.ui.spinBoxFontSize.value()) if self.ui.radioButtonFixedSize.isChecked() else '-1'))

            docData.saveConfigs(configs)

            docData.lineTypeConfigs = None  # clear line type configurations

        except Exception as ex:

            print('error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno))

        QDialog.accept(self)