forked from hamish2014/FreeCAD_drawing_dimensioning
/
dimensionSvgConstructor_testLinear.py
148 lines (128 loc) · 6.16 KB
/
dimensionSvgConstructor_testLinear.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
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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
'''
library for constructing dimension SVGs
'''
from dimensionSvgConstructor import *
from selectionOverlay import generateSelectionGraphicsItems
print('Testing dimensionSvgConstructor.py')
import sys
from PySide import QtGui, QtCore, QtSvg
app = QtGui.QApplication(sys.argv)
width = 640
height = 480
textRenderer = SvgTextRenderer(font_family='Verdana', font_size='16pt', fill="red")
graphicsScene = QtGui.QGraphicsScene(0,0,width,height)
graphicsScene.addText("Linear dimensioning testing app.\nEsc to Exit")
dimensions = []
class DimensioningRect(QtGui.QGraphicsRectItem):
def __init__(self,*args):
super(DimensioningRect, self).__init__(*args)
svgRenderer = QtSvg.QSvgRenderer()
self.action_ind = 0
self.actions = ['selectPoint1','selectPoint2','placeDimensionBaseLine','placeDimensionText']
self.dimPreview = QtSvg.QGraphicsSvgItem()
self.dimSVGRenderer = QtSvg.QSvgRenderer()
self.dimSVGRenderer.load( QtCore.QByteArray( '''<svg width="%i" height="%i"></svg>''' % (args[2],args[3])))
self.dimPreview.setSharedRenderer( self.dimSVGRenderer )
graphicsScene.addItem( self.dimPreview )
self.dim_svg_KWs = dict(
svgTag='svg', svgParms='width="%i" height="%i"' % (args[2],args[3]),
strokeWidth=1.0, arrowL1=10, arrowL2=4, arrowW=6, textRenderer=textRenderer,
scale= 1.0, gap_datum_points = 8, dimension_line_overshoot=4,
)
def selectDimensioningPoint( self, event, referer, elementXML, elementParms, elementViewObject ):
print( elementXML )
if self.action_ind == 0: #then selectPoint1
self.point1 = elementParms['x'], elementParms['y']
self.action_ind = self.action_ind + 1
referer.lockSelection()
self.referer1 = referer
elif self.action_ind == 1: #then selectPoint2
self.point2 = elementParms['x'], elementParms['y']
self.action_ind = self.action_ind + 1
referer.lockSelection()
self.referer2 = referer
def mousePressEvent( self, event ):
if event.button() == QtCore.Qt.MouseButton.LeftButton:
pos = event.scenePos()
x, y = pos.x(), pos.y()
if self.action_ind == 2: #then 'placeDimensionBaseLine':
self.point3 = x, y
print('point3 set to x=%3.1f y=%3.1f' % (x,y))
self.action_ind = self.action_ind + 1
elif self.action_ind == 3: # then placeDimensionText
self.point4 = x, y
self.action_ind = 0
XML = linearDimensionSVG( self.point1[0], self.point1[1],
self.point2[0], self.point2[1],
self.point3[0], self.point3[1],
x, y, **self.dim_svg_KWs )
if XML <> None:
print(XML)
newSvg = QtSvg.QGraphicsSvgItem( )
svgRenderer = QtSvg.QSvgRenderer()
svgRenderer.load( QtCore.QByteArray( XML ))
newSvg.setSharedRenderer( svgRenderer )
dimensions.append([ newSvg, svgRenderer]) #as to prevent the garbage collector from freeing these resources (which causes a crash)
self.scene().addItem( newSvg )
self.referer1.unlockSelection()
self.referer2.unlockSelection()
def hoverMoveEvent(self, event):
pos = event.scenePos()
x, y = pos.x(), pos.y()
XML = None
if self.action_ind == 2: # then placeDimensionBaseLine action
XML = linearDimensionSVG( self.point1[0], self.point1[1],
self.point2[0], self.point2[1], x, y,
**self.dim_svg_KWs )
elif self.action_ind == 3: # then placeDimensionText
XML = linearDimensionSVG( self.point1[0], self.point1[1],
self.point2[0], self.point2[1],
self.point3[0], self.point3[1],
x, y, **self.dim_svg_KWs )
if XML <> None:
self.dimSVGRenderer.load( QtCore.QByteArray( XML ) )
self.dimPreview.update()
self.dimPreview.show()
else:
self.dimPreview.hide()
def wheelEvent( self, event):
if event.delta() > 0:
view.scale(1.1, 1.1)
else:
view.scale(0.9, 0.9)
def keyPressEvent(self, event):
if len(event.text()) == 1:
print('key pressed: event.text %s (ord %i)' % (event.text(), ord(event.text())))
if event.text() == chr(27): #escape key
sys.exit(2)
dimensioningRect = DimensioningRect(0,0,width,height)
dimensioningRect.setAcceptHoverEvents(True)
dimensioningRect.setFlag( QtGui.QGraphicsItem.GraphicsItemFlag.ItemIsFocusable, True )
dimensioningRect.setZValue(-1000)
#creating points to dimension.
circlesXML = []
noPoints = 12
X = [ numpy.random.rand()*width*0.8 for i in range(noPoints) ]
Y = [ numpy.random.rand()*height*0.8 for i in range(noPoints) ]
for i in range(noPoints):
circlesXML.append( '<circle id="circ%03i" cx ="%f" cy ="%f" r ="6" />' % (i,X[i],Y[i]) )
XML = '''<svg id="Ortho_0_1" width="%i" height="%i"> <g stroke-width="0.2" >
%s
</g> </svg>''' % (width, height, '\n'.join(circlesXML) )
class dummyViewObject:
def __init__(self, XML):
self.ViewResult = XML
viewObject = dummyViewObject( XML )
hoverPen = QtGui.QPen( QtGui.QColor(255,0,0) )
hoverPen.setWidth(3.0)
selectGraphicsItems = generateSelectionGraphicsItems( [viewObject], dimensioningRect.selectDimensioningPoint , doCircles=True,
maskPen= QtGui.QPen( QtGui.QColor(0,0,0) ) ,
maskBrush=QtGui.QBrush( QtGui.QColor(0,0,255) ),
maskHoverPen= hoverPen )
graphicsScene.addItem(dimensioningRect)
for g in selectGraphicsItems:
graphicsScene.addItem(g)
view = QtGui.QGraphicsView(graphicsScene)
#view.scale(2, 2)
view.show()
sys.exit(app.exec_())