1 # Copyright (C) 2006-2019 CEA/DEN, EDF R&D
3 # This library is free software; you can redistribute it and/or
4 # modify it under the terms of the GNU Lesser General Public
5 # License as published by the Free Software Foundation; either
6 # version 2.1 of the License, or (at your option) any later version.
8 # This library is distributed in the hope that it will be useful,
9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 # Lesser General Public License for more details.
13 # You should have received a copy of the GNU Lesser General Public
14 # License along with this library; if not, write to the Free Software
15 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
22 from qtcanvas import *
25 class DynamicTip( QToolTip ):
26 def __init__( self, parent ):
27 QToolTip.__init__( self, parent )
29 def maybeTip( self, pos ):
30 pos2=self.parentWidget().viewportToContents(pos)
31 point = self.parentWidget().inverseWorldMatrix().map(pos2)
32 ilist = self.parentWidget().canvas().collisions(point) #QCanvasItemList ilist
33 for each_item in ilist:
34 if hasattr(each_item,"tooltip"):
35 each_item.tooltip(self,pos)
37 elif hasattr(each_item,"getObj"):
38 each_item.getObj().tooltip(self,pos)
41 class GraphViewer(QCanvasView):
42 def __init__(self,c,parent,name,f):
43 QCanvasView.__init__(self,c,parent,name,f)
46 self.__moving_start= 0
47 # for highlighting selections
48 self.selectPen=QPen(QColor(255,255,0),2,Qt.DashLine)
49 self.selectBrush=QBrush(Qt.red)
50 self.selectStyle = Qt.Dense5Pattern
52 self.tooltip = DynamicTip( self )
54 def contentsMouseDoubleClickEvent(self,e): # QMouseEvent e
55 point = self.inverseWorldMatrix().map(e.pos())
56 ilist = self.canvas().collisions(point) #QCanvasItemList ilist
57 for each_item in ilist:
58 if each_item.rtti()==984376:
59 if not each_item.hit(point):
61 if e.button()== Qt.LeftButton:
62 if hasattr(each_item,"handleDoubleClick"):
63 each_item.handleDoubleClick(point)
64 self.canvas().update()
68 def contentsMousePressEvent(self,e): # QMouseEvent e
70 point = self.inverseWorldMatrix().map(e.pos())
71 ilist = self.canvas().collisions(point) #QCanvasItemList ilist
72 for each_item in ilist:
73 if each_item.rtti()==984376:
74 if not each_item.hit(point):
76 if e.button()== Qt.RightButton:
80 if hasattr(each_item,"popup"):
81 menu=each_item.popup(self)
83 menu.exec_loop( QCursor.pos() )
84 self.canvas().update()
85 elif hasattr(each_item,"getObj"):
86 menu=each_item.getObj().popup(self)
88 menu.exec_loop( QCursor.pos() )
89 self.canvas().update()
91 elif e.button()== Qt.LeftButton:
94 #We are linking objects
95 if hasattr(each_item,"getObj"):
96 #a connection is ending
97 self.__connecting.link(each_item.getObj().item)
98 #self.__connecting.link(each_item.getObj())
99 self.canvas().update()
102 #We are moving or selecting a composite object
104 self.__moving=each_item
105 self.__moving_start=point
106 self.canvas().update()
108 if e.button()== Qt.RightButton:
111 menu.exec_loop( QCursor.pos() )
112 self.canvas().update()
115 QCanvasView.contentsMousePressEvent(self,e)
117 def selectItem(self,item):
118 #print "selectItem",item
121 self.deselectObj(self.selected)
125 #need to find equivalent canvas item
126 for citem in self.canvas().allItems():
127 if hasattr(citem,"item") and citem.item is item:
129 self.selectObj(self.selected)
131 self.canvas().update()
133 def selectObj(self,obj):
135 obj._origPen = obj.pen()
136 obj._origBrush = obj.brush()
137 obj._origStyle = obj.brush().style()
138 obj.setPen(self.selectPen)
139 #obj.setBrush(self.selectBrush)
141 def deselectObj(self,obj):
143 obj.setPen(obj._origPen)
144 #obj.setBrush(obj._origBrush)
148 caption = QLabel( "<font color=darkblue><u><b>View Menu</b></u></font>", self )
149 caption.setAlignment( Qt.AlignCenter )
150 menu.insertItem( caption )
151 menu.insertItem("add Node", self.addNode)
154 #def layout(self,rankdir):
157 def updateCanvas(self):
158 #Par defaut, Qt n'efface pas le background. Seul repaintContents
159 #semble le faire. Utile apres un popup ou un resize avec scrollbars
160 #Peut-on l'utiliser partout ? Pb de performance ?
161 self.repaintContents(True)
162 #self.canvas().update()
168 m = self.worldMatrix()
170 self.setWorldMatrix( m )
173 m = self.worldMatrix()
175 self.setWorldMatrix( m )
178 ilist = self.canvas().allItems()
179 for each_item in ilist:
181 each_item.setCanvas(None)
183 self.canvas().update()
185 def connecting(self,obj):
186 """Method called by an item to notify canvasView a connection has begun"""
187 print("connecting",obj)
188 self.__connecting=obj
190 def contentsMouseMoveEvent(self,e):
192 point = self.inverseWorldMatrix().map(e.pos())
193 self.__moving.moveBy(point.x()-self.__moving_start.x(),point.y()-self.__moving_start.y())
194 self.__moving_start = point
195 self.canvas().update()
197 #self.tooltip.maybeTip(point)
198 QCanvasView.contentsMouseMoveEvent(self,e)
200 class ImageItem(QCanvasRectangle):
201 def __init__(self,img,canvas):
202 QCanvasRectangle.__init__(self,canvas)
204 self.imageRTTI=984376
206 self.pixmap=QPixmap()
207 self.setSize(self.image.width(), self.image.height())
208 self.pixmap.convertFromImage(self.image, Qt.OrderedAlphaDither);
211 return self.imageRTTI
216 if not self.image.valid( ix , iy ):
218 self.pixel = self.image.pixel( ix, iy )
219 return (qAlpha( self.pixel ) != 0)
221 def drawShape(self,p):
222 p.drawPixmap( self.x(), self.y(), self.pixmap )
224 class LinkItem(QCanvasLine):
225 def __init__(self,fromPort, toPort,canvas):
226 QCanvasLine.__init__(self,canvas)
227 self.setPen(QPen(Qt.black))
228 self.setBrush(QBrush(Qt.red))
229 fromPort.addOutLink(self)
230 toPort.addInLink(self)
231 self.setPoints(int(fromPort.x()),int(fromPort.y()), int(toPort.x()), int(toPort.y()))
232 self.setZ(min(fromPort.z(),toPort.z())-1)
233 self.setVisible(True)
234 self.arrow = QCanvasPolygon(self.canvas())
235 self.arrow.setBrush(QBrush(Qt.black))
239 def setFromPoint(self,x,y):
240 self.setPoints(x,y,self.endPoint().x(),self.endPoint().y())
243 def setToPoint(self,x,y):
244 self.setPoints(self.startPoint().x(), self.startPoint().y(),x,y)
247 def moveBy(self,dx,dy):
251 x1,y1=self.startPoint().x(),self.startPoint().y()
252 x2,y2=self.endPoint().x(),self.endPoint().y()
253 d=math.hypot(x2-x1,y2-y1)
259 pa.setPoint(0, QPoint(x+10*cosa,y+10*sina))
260 pa.setPoint(1, QPoint(x-5*sina,y+5*cosa))
261 pa.setPoint(2, QPoint(x+5*sina,y-5*cosa))
262 self.arrow.setPoints(pa)
264 def popup(self,canvasView):
266 caption = QLabel( "<font color=darkblue><u><b>Node Menu</b></u></font>",menu )
267 caption.setAlignment( Qt.AlignCenter )
268 menu.insertItem( caption )
269 menu.insertItem("Delete", self.delete)
275 def tooltip(self,view,pos):
276 r = QRect(pos.x(), pos.y(), pos.x()+10, pos.y()+10)
277 s = QString( "link: %d,%d" % (r.center().x(), r.center().y()) )
280 class PortItem(QCanvasEllipse):
281 def __init__(self,node,canvas):
282 QCanvasEllipse.__init__(self,6,6,canvas)
283 self.setPen(QPen(Qt.black))
284 self.setBrush(QBrush(Qt.red))
285 self.setZ(node.z()+1)
288 def moveBy(self,dx,dy):
289 self.node.moveBy(dx,dy)
291 def myMove(self,dx,dy):
292 QCanvasEllipse.moveBy(self,dx,dy)
297 def popup(self,canvasView):
298 self.context=canvasView
300 caption = QLabel( "<font color=darkblue><u><b>Port Menu</b></u></font>",menu )
301 caption.setAlignment( Qt.AlignCenter )
302 menu.insertItem( caption )
303 menu.insertItem("Connect", self.connect)
307 print("connect",self.context)
308 self.context.connecting(self)
313 def tooltip(self,view,pos):
314 r = QRect(self.x(), self.y(), self.width(), self.height())
315 s = QString( "port: %d,%d" % (r.center().x(), r.center().y()) )
318 class InPortItem(PortItem):
319 def __init__(self,node,canvas):
320 PortItem.__init__(self,node,canvas)
323 def myMove(self,dx,dy):
324 PortItem.myMove(self,dx,dy)
325 for link in self.__inList:
326 link.setToPoint( int(self.x()), int(self.y()) )
330 if isinstance(obj,OutPortItem):
332 l=LinkItem(obj,self,self.canvas())
334 def addInLink(self,link):
335 self.__inList.append(link)
337 class OutPortItem(PortItem):
338 def __init__(self,node,canvas):
339 PortItem.__init__(self,node,canvas)
342 def myMove(self,dx,dy):
343 PortItem.myMove(self,dx,dy)
344 for link in self.__outList:
345 link.setFromPoint( int(self.x()), int(self.y()) )
349 if isinstance(obj,InPortItem):
351 l=LinkItem(self,obj,self.canvas())
353 def addOutLink(self,link):
354 self.__outList.append(link)
356 class Cell(QCanvasRectangle):
357 def __init__(self,canvas):
358 QCanvasRectangle.__init__(self,canvas)
364 p=InPortItem(self,canvas)
366 self.inports.append(p)
367 p=OutPortItem(self,canvas)
369 self.outports.append(p)
371 def moveBy(self,dx,dy):
372 QCanvasRectangle.moveBy(self,dx,dy)
373 for p in self.inports:
375 for p in self.outports:
379 QCanvasRectangle.show(self)
380 for p in self.inports:
382 for p in self.outports:
388 def popup(self,canvasView):
390 caption = QLabel( "<font color=darkblue><u><b>Node Menu</b></u></font>",menu )
391 caption.setAlignment( Qt.AlignCenter )
392 menu.insertItem( caption )
393 menu.insertItem("Browse", self.browse)
396 def tooltip(self,view,pos):
397 r = QRect(self.x(), self.y(), self.width(), self.height())
398 s = QString( "node: %d,%d" % (r.center().x(), r.center().y()) )
407 if __name__=='__main__':
408 app=QApplication(sys.argv)
410 QToolBar(box,"toolbar")
411 canvas=QCanvas(800,600)
412 canvas.setAdvancePeriod(30)
413 m=GraphViewer(canvas,box,"example",0)
414 box.setCentralWidget(m)
421 c1.outports[0].link(c2.inports[0])
423 qApp.setMainWidget(box)