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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
#!/usr/bin/env python3
import PySide6.QtWidgets as QtWidgets
import PySide6.QtGui as QtGui
import PySide6.QtCore as QtCore
import numpy as numpy
import xasy2asy as xasy2asy
from xasyqtui.custMatTransform import Ui_Dialog
class CustMatTransform(QtWidgets.QDialog):
def __init__(self):
super().__init__()
self.ui = Ui_Dialog()
self.ui.setupUi(self)
self.ui.btnAccept.clicked.connect(self.accept)
self.ui.btnCancel.clicked.connect(self.reject)
self.ui.btnReset.clicked.connect(self.resetDialog)
self.mainTransformation = QtGui.QTransform()
self.mainTransformation.scale(1, -1)
self.matrixLineInputs = [
self.ui.lineMat00, self.ui.lineMat01, self.ui.lineMatTx,
self.ui.lineMat10, self.ui.lineMat11, self.ui.lineMatTy]
validator = QtGui.QDoubleValidator()
for lineInput in self.matrixLineInputs:
lineInput.setValidator(validator)
lineInput.textChanged.connect(self.handleUpdateText)
def show(self):
super().show()
self.createCanvas()
self.updatePreview()
def createCanvas(self):
self.canvSize = self.ui.imgPreview.size()
self.previewPixmap = QtGui.QPixmap(self.canvSize)
tx, ty = self.canvSize.width() / 2, self.canvSize.height() / 2
self.mainTransformation.translate(tx, -ty)
def handleUpdateText(self, text):
if str(text) not in {'.', '-', '.-', '-.'} and str(text):
self.updatePreview()
self.ui.btnAccept.setEnabled(True)
else:
self.previewPixmap.fill()
self.ui.imgPreview.setPixmap(self.previewPixmap)
self.ui.btnAccept.setEnabled(False)
def updatePreview(self):
self.previewPixmap.fill()
canvas = QtGui.QPainter(self.previewPixmap)
if not canvas.isActive():
return
canvas.setTransform(self.mainTransformation)
canvas.save()
canvas.setPen(QtCore.Qt.GlobalColor.lightGray)
self.drawBasicGrid(canvas)
transform = xasy2asy.asyTransform.fromNumpyMatrix(self.getTransformationMatrix())
canvTransform = transform.toQTransform()
canvas.setTransform(canvTransform, True)
canvas.setPen(QtCore.Qt.GlobalColor.black)
if canvTransform.isInvertible():
self.drawBasicGrid(canvas, False)
if canvTransform.determinant() <= 0:
canvas.setPen(QtCore.Qt.GlobalColor.red)
canvas.drawRect(QtCore.QRect(QtCore.QPoint(0, 0), QtCore.QSize(20, 20)))
self.ui.imgPreview.setPixmap(self.previewPixmap)
def resetDialog(self):
self.ui.lineMatTx.setText('0')
self.ui.lineMatTx.setText('0')
self.ui.lineMat00.setText('1')
self.ui.lineMat01.setText('0')
self.ui.lineMat10.setText('0')
self.ui.lineMat11.setText('1')
def drawBasicGrid(self, canvas, grid=True):
canvas.drawLine(QtCore.QLine(-9999, 0, 9999, 0))
canvas.drawLine(QtCore.QLine(0, -9999, 0, 9999))
fromIter, toIter = -7, 7
gridSize = 20
if grid:
for iterIndex in range(fromIter, toIter + 1):
canvas.drawLine(QtCore.QLine(-9999, iterIndex * gridSize, 9999, iterIndex * gridSize))
canvas.drawLine(QtCore.QLine(iterIndex * gridSize, -9999, iterIndex * gridSize, 9999))
def getTransformationMatrix(self):
rawMatrixNum = [float(lineInput.text()) for lineInput in self.matrixLineInputs]
rawMatrixNum.extend([0, 0, 1])
return numpy.matrix(rawMatrixNum).reshape((3, 3))
|
