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:

            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)