-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
68 lines (57 loc) · 2.05 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
import sys
import os
import cv2
import time
import numpy as np
from PIL import Image,ImageQt
from ui import Ui_ui
from mouse_event import GraphicsScene
from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from PyQt5.QtPrintSupport import QPrintDialog, QPrinter
class ui(QWidget, Ui_ui):
def __init__(self):
super(ui, self).__init__()
self.setupUi(self)
self.show()
self.mode = 0
# self.size = self.sld.value()
self.mouse_clicked = False
self.GotDetails = False
self.scene = GraphicsScene(self.mode)
self.scene.setSceneRect(120,20,512,512)
self.scene.graphicsView = self.graphicsView
self.graphicsView.setScene(self.scene)
self.graphicsView.setAlignment(Qt.AlignTop | Qt.AlignLeft)
self.graphicsView.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.graphicsView.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
def save_img(self):
if False:
if type(self.output_img):
fileName, _ = QFileDialog.getSaveFileName(self, "Save File",
QDir.currentPath())
cv2.imwrite(fileName+'.jpg',self.output_img)
else:
self.scene.strokes_save()
def curvature(self):
if False:
if type(self.output_img):
fileName, _ = QFileDialog.getSaveFileName(self, "Save File",
QDir.currentPath())
cv2.imwrite(fileName+'.jpg',self.output_img)
else:
self.scene.strokes_load1()
def derivation(self):
if False:
if type(self.output_img):
fileName, _ = QFileDialog.getSaveFileName(self, "Save File",
QDir.currentPath())
cv2.imwrite(fileName+'.jpg',self.output_img)
else:
self.scene.strokes_load()
if __name__ == "__main__":
app = QApplication([])
widget = ui()
widget.show()
sys.exit(app.exec_())