HYTOS

import sys

import os

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

from AppDocData import AppDocData, Source

import TrainingEditor_UI

import QtImageViewer

from GraphicsBoundingBoxItem import QGraphicsBoundingBoxItem

from TrainingBoxItem import QTrainingBoxItem

import cv2

from PIL import Image

import AreaZoomCommand

import PlaceLineCommand

class QTrainingEditorDialog(QDialog):

    def __init__(self, parent):

        self.spinBoxFlag = False

        QDialog.__init__(self, parent)

        appDocData = AppDocData.instance()

        project = appDocData.getCurrentProject()

        self.ui = TrainingEditor_UI.Ui_TrainingEditorDialog()

        self.ui.setupUi(self)

        self.graphicsViewTrainingDrawing = QtImageViewer.QtImageViewer(self)

        self.graphicsViewTrainingDrawing.setParent(self.ui.centralWidget)

        self.graphicsViewTrainingDrawing.useDefaultCommand()

        self.ui.verticalLayoutTrainingDrawing.addWidget(self.graphicsViewTrainingDrawing)

        self.graphicsViewZoomDrawing = QGraphicsView(self)

        self.graphicsViewZoomDrawing.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)

        self.graphicsViewZoomDrawing.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)

        self.graphicsViewZoomDrawing.setParent(self.ui.leftSideWidget)

        self.ui.horizontalLayoutZoomDrawing.addWidget(self.graphicsViewZoomDrawing)

        # 학습 이미지 읽어서 메인 뷰에 그림, 사이드 뷰에 추가

        try:

            trainingImgPath = os.path.join(project.getTrainingFilePath(), 'seed.seedF.exp0.tif')

            cvImg = cv2.cvtColor(cv2.imread(trainingImgPath), cv2.COLOR_BGR2GRAY)

            blur = cv2.GaussianBlur(cvImg, (5,5),0)

            cvImg = cv2.threshold(cvImg, 0, 255, cv2.THRESH_BINARY+cv2.THRESH_OTSU)[1]

            bytesPerLine = cvImg.shape[1]

            image = QImage(cvImg.data, cvImg.shape[1], cvImg.shape[0], bytesPerLine, QImage.Format_Indexed8)

            #img = cv2.imread(trainingImgPath)

            #image = QImage(img.data, img.shape[1], img.shape[0], img.shape[1], QImage.Format_Indexed8)

            self.graphicsViewTrainingDrawing.setImage(image)

            # 사이드 뷰

            if type(image) is QPixmap:

                pixmap = image

            elif type(image) is QImage:

                pixmap = QPixmap.fromImage(image)

            scene = QGraphicsScene()

            self.graphicsViewZoomDrawing.setScene(scene)

            self._pixmapHandle = self.graphicsViewZoomDrawing.scene().addPixmap(pixmap)

            rect = QGraphicsRectItem(0, 0, 0, 0)

            pen = QPen(Qt.SolidLine)

            pen.setColor(Qt.green)

            pen.setWidthF(1)

            pen.setJoinStyle(Qt.MiterJoin)

            rect.setPen(pen)

            self.graphicsViewZoomDrawing.scene().addItem(rect)

        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))

        # 박스 읽어서 메인 뷰에 그림

        try:

            trainingBoxPath = os.path.join(project.getTrainingFilePath(), 'seed.seedF.exp0.box')

            fBox = open(trainingBoxPath, 'r', encoding='utf8')

            SBox = fBox.read()

            fBox.close()

            boxList = []

            boxList = SBox.split('\n')

            for box in boxList:

                boxComponent = box.split(' ')

                singleBox = QTrainingBoxItem(int(boxComponent[1]), cvImg.shape[0] - int(boxComponent[4]), int(boxComponent[3]) - int(boxComponent[1]), int(boxComponent[4]) - int(boxComponent[2]))

                singleBox.loc = [int(boxComponent[1]), cvImg.shape[0] - int(boxComponent[2])]

                singleBox.size = [int(boxComponent[3]) - int(boxComponent[1]), int(boxComponent[2]) - int(boxComponent[4])]

                singleBox.angle = 0

                #singleBox.setPlainText(boxComponent[0])

                singleBox.transfer.onRemoved.connect(self.itemRemoved)

                singleBox.addTextItemToScene(self.ui, self.graphicsViewTrainingDrawing, self.graphicsViewZoomDrawing, self.spinBoxFlag)

        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))

        #'''test'''

        #item = QGraphicsBoundingBoxItem(0, 0, 30, 30)

        #item.isSymbol = True

        #item.angle = 0

        #item.setPen(QPen(Qt.red, 1, Qt.SolidLine))

        #self.graphicsViewTrainingDrawing.scene.addItem(item)

        self.removedItems = []

        self.ui.pushButtonZoom.clicked.connect(self.onAreaZoom)

        self.ui.spinBoxLeft.valueChanged.connect(self.spinBoxChangedEvent)

        self.ui.spinBoxTop.valueChanged.connect(self.spinBoxChangedEvent)

        self.ui.spinBoxWidth.valueChanged.connect(self.spinBoxChangedEvent)

        self.ui.spinBoxHeight.valueChanged.connect(self.spinBoxChangedEvent)

    def spinBoxChangedEvent(self, event):

        items = self.graphicsViewTrainingDrawing.scene.selectedItems()

        if(len(items) is not 1) or self.spinBoxFlag:

            return

        spinBoxName = self.sender().objectName()

        rect = items[0].rect()

        bound = self.graphicsViewZoomDrawing.scene().items()[0]

        if spinBoxName == 'spinBoxLeft':

            spinBoxValue = self.ui.spinBoxLeft.value()

            items[0].setRect(QRectF(spinBoxValue, rect.y(), rect.width(), rect.height()))

        elif spinBoxName == 'spinBoxTop':

            spinBoxValue = self.ui.spinBoxTop.value()

            items[0].setRect(QRectF(rect.x(), spinBoxValue, rect.width(), rect.height()))

        elif spinBoxName == 'spinBoxWidth':

            spinBoxValue = self.ui.spinBoxWidth.value()

            items[0].setRect(QRectF(rect.x(),  rect.y(), spinBoxValue, rect.height()))

        elif spinBoxName == 'spinBoxHeight':

            spinBoxValue = self.ui.spinBoxHeight.value()

            items[0].setRect(QRectF(rect.x(), rect.y(), rect.width(), spinBoxValue))

        rect = items[0].rect()        

        bound.setRect(rect)

        rectSide = QRectF(rect.x() -3, rect.y() -3, rect.width() +6,  rect.height() +6)

        self.graphicsViewZoomDrawing.fitInView(rectSide)

        self.graphicsViewTrainingDrawing.scene.update()

    def onAreaZoom(self, action):

        if self.ui.pushButtonZoom.isChecked():

            cmd = AreaZoomCommand.AreaZoomCommand(self.graphicsViewTrainingDrawing)

            cmd.onRejected.connect(self.onCommandRejected)

            self.graphicsViewTrainingDrawing.command = cmd

    def onCommandRejected(self, cmd):

        try:

            if type(cmd) is AreaZoomCommand.AreaZoomCommand:

                self.ui.pushButtonZoom.setChecked(False)

        finally:

            self.graphicsViewTrainingDrawing.useDefaultCommand()

    def itemRemoved(self, item):

        try:

            if type(item) is QTrainingBoxItem:

                self.removedItems.append(str(item.uid))

            if item.scene() is not None: item.scene().removeItem(item)

        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))