HYTOS

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

from PyQt5 import QtCore, QtGui, QtWidgets

from PyQt5.QtWidgets import *

import os, sys

from AppDocData import AppDocData

import SymbolThickness_UI

import cv2

import numpy as np

class QSymbolThicknessDialog(QDialog):

    def __init__(self, parent):

        QDialog.__init__(self, parent)

        self.ui = SymbolThickness_UI.Ui_SymbolThicknessDialog()

        self.ui.setupUi(self)

        self.isAccepted = False

        configs = AppDocData.instance().getConfigs('Filter', 'DilateSize')

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

    def showDialog(self):

        self.setWindowFlags(self.windowFlags() & ~QtCore.Qt.WindowContextHelpButtonHint)

        self.exec_()

        return self.isAccepted, self.angle, self.sx, self.sy, self.sz

    def accept(self):

        size = self.ui.spinBox.value()

        if size:

            reply = QMessageBox.question(self, self.tr('Continue?'), self.tr('Are you sure to continue?'), QMessageBox.Yes, QMessageBox.Cancel)

            if reply == QMessageBox.Yes:

                try:

                    project = AppDocData.instance().getCurrentProject()

                    rootPath = project.getImageFilePath()

                    rootList = os.listdir(rootPath)

                    for folder in rootList:

                        path = os.path.join(rootPath, folder)

                        imgList = os.listdir(path)

                        for imgPath in imgList:

                            fullPath = os.path.join(path, imgPath)

                            img = cv2.imread(fullPath, cv2.IMREAD_GRAYSCALE)

                            kernel = np.ones((size, size), np.uint8)

                            height, width = img.shape[0], img.shape[1]

                            new = np.ones((height + 2 * size, width + 2 * size), np.uint8) * 255

                            new[size:size+height, size:size+width] = img

                            e = cv2.erode(new, kernel, iterations=1)

                            cv2.imwrite(fullPath, e)

                    QMessageBox.information(self, self.tr('Information'), self.tr('Succeeded'))

                except Exception as ex:

                    from App import App

                    from AppDocData import MessageType

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

                    App.mainWnd().addMessage.emit(MessageType.Error, message)

            QDialog.accept(self)

    def reject(self):

        QDialog.reject(self)