import FreeCAD as App import FreeCADGui as Gui from PySide.QtCore import QTimer import TechDraw, TechDrawGui from PySide import QtGui, QtCore TDG = TechDrawGui class CursorItem(QtGui.QGraphicsItem): def __init__(self, parent = None, view = None): super().__init__(parent) self.Type = QtGui.QGraphicsItem.UserType + 501 self.setZValue(500) self.margin = 10.0 self.size = 100.0 self.view = view def onViewPosChange(self, callback): self.viewPosChangeCallback = callback def boundingRect(self): return QtCore.QRectF(-self.size/2 - self.margin, -self.size/2 - self.margin, self.size + 2.0*self.margin, self.size + 2.0*self.margin) def paint(self, painter, option, widget): #print("paint") painter.setBrush(QtCore.Qt.darkRed) #painter.drawRoundedRect(-100, -100, 200, 200, 50, 50) h = self.size/2.0 painter.drawLine(-h, 0, h, 0) painter.drawLine(0, -h, 0, h) def mousePressEvent(self, event): #print('mouse press', event) self.startMovePos = event.pos() def mouseMoveEvent(self, event): #print('mouse move', event) offset = event.pos() - self.startMovePos self.moveBy(offset.x(), offset.y()) def mouseReleaseEvent(self, event): #print('mouse release', event) #print('new pos', self.x(), self.y()) if self.viewPosChangeCallback is not None: self.viewPosChangeCallback(self.getViewPos()) def getViewPos(self): scale = self.view.Scale * 10.0 return App.Vector(self.x() / scale, -self.y() / scale) def setViewPos(self, p): scale = self.view.Scale * 10.0 self.setPos(p.x * scale, -p.y * scale) class TechDrawExtensions: views_to_repaint = {} view_cursors = {} updating_balloon = False def __init__(self): workbench = Gui.getWorkbench("AssemblyHandbookWorkbench") #: :type workbench: AssemblyHandbookWorkbench workbench.docObserver.onObjectTypeChanged('balloon_changed', 'TechDraw::DrawViewBalloon', lambda obj, prop: self.onBalloonChanged(obj, prop)) workbench.docObserver.onObjectTypeSelected('balloon_selected', 'TechDraw::DrawViewBalloon', lambda operation, obj, sub, point: self.onBalloonSelected(operation, obj, sub, point)) def repaint(self, view): self.views_to_repaint[view] = True QTimer.singleShot(100, self._do_repaint) def _do_repaint(self): import time workbench = Gui.getWorkbench("AssemblyHandbookWorkbench") #: :type workbench: AssemblyHandbookWorkbench to_repaint = self.views_to_repaint.keys() self.views_to_repaint = {} for view in to_repaint: #print("Repainting " + view.Name) selected_balloons = [] for obj in Gui.Selection.getSelection(): if obj.TypeId == 'TechDraw::DrawViewBalloon' and obj.SourceView == view and 'AssemblyHandbook_PartName' in obj.PropertiesList: selected_balloons.append(obj) is_first_part = True if len(selected_balloons) == 0: selected_balloons.append(None) for balloon in selected_balloons: if balloon is not None: doc = balloon.Document partLink = doc.getObject(balloon.AssemblyHandbook_PartName) part_view = workbench.partsCache.getPart2DView(view, partLink) center = self.computePartCenter(view, partLink) # iterate edges of actual view and highlight matching edges for edgeIdx in range(10000): hasEdge = False try: edge = view.getEdgeByIndex(edgeIdx) hasEdge = True except: pass if not hasEdge: break # reset edge format if is_first_part: view.formatGeometricEdge(edgeIdx,1,0.25,0,True) if balloon is not None and (not hasattr(edge.Curve, 'Degree') or edge.Curve.Degree == 1) and len(edge.Vertexes) == 2: edgeData = [ edge.Vertexes[0].X - center.x, edge.Vertexes[0].Y - center.y, edge.Vertexes[1].X - center.x, edge.Vertexes[1].Y - center.y ] v0 = App.Vector(edgeData[0], edgeData[1]) v1 = App.Vector(edgeData[2], edgeData[3]) for line in part_view.cached_lines: l0 = App.Vector(line[0], line[1]) l1 = App.Vector(line[2], line[3]) #d = abs(edgeData[0] - line[0]) + abs(edgeData[1] - line[1]) + abs(edgeData[2] - line[2]) + abs(edgeData[3] - line[3]) d = v0.distanceToLineSegment(l0, l1).Length + v1.distanceToLineSegment(l0, l1).Length if d < 0.001: view.formatGeometricEdge(edgeIdx,1,0.25,(0,0.85,0),True) is_first_part = False view.requestPaint() cursor = self.view_cursors.get(view, None) if cursor is None: cursor = CursorItem(view = view) TDG.addQGIToView(view, cursor); cursor.onViewPosChange(lambda new_pos: self.onCursorMoved(view, new_pos)) self.view_cursors[view] = cursor if selected_balloons[0] is None: cursor.setVisible(False) else: cursor.setViewPos(App.Vector(selected_balloons[0].OriginX, selected_balloons[0].OriginY)) cursor.setVisible(True) def onCursorMoved(self, view, new_pos): if len(Gui.Selection.getSelection()) == 0: return balloon = Gui.Selection.getSelection()[0] if balloon.TypeId != 'TechDraw::DrawViewBalloon' or not 'AssemblyHandbook_PartName' in balloon.PropertiesList: return if balloon.SourceView != view: return balloon.OriginX = new_pos.x balloon.OriginY = new_pos.y obj = balloon.Document.getObject(balloon.AssemblyHandbook_PartName) view = balloon.SourceView center = self.computePartCenter(view, obj) balloon.AssemblyHandbook_OriginOffsetX = new_pos.x - center.x balloon.AssemblyHandbook_OriginOffsetY = new_pos.y - center.y def onBalloonSelected(self, operation, balloon, sub, point): import time #print(operation, obj.Name, sub, point) if "AssemblyHandbook_PartName" in balloon.PropertiesList: #print(operation + " " + balloon.Name) view = balloon.SourceView self.repaint(view) def onBalloonChanged(self, obj, prop): # Avoid reentry if self.updating_balloon: return #print('Balloon changed: ' + obj.Name + '.' + prop) if prop == 'Y' and "AssemblyHandbook_PartName" in obj.PropertiesList: self.updating_balloon = True self.updateBalloon(obj) self.updating_balloon = False def updateBalloon(self, balloon): workbench = Gui.getWorkbench("AssemblyHandbookWorkbench") #: :type workbench: AssemblyHandbookWorkbench view = balloon.SourceView doc = view.Document obj = doc.getObject(balloon.AssemblyHandbook_PartName) objectCenterView = workbench.techDrawExtensions.computePartCenter(view, obj) balloon.OriginX = objectCenterView.x + balloon.AssemblyHandbook_OriginOffsetX balloon.OriginY = objectCenterView.y + balloon.AssemblyHandbook_OriginOffsetY balloon.Text = obj.LinkedObject.Document.Name if obj.TypeId == 'App::Link' else obj.Name balloon.ViewObject.Font = 'DejaVu Sans' balloon.ViewObject.Fontsize = 4 balloon.BubbleShape = 'Inspection' balloon.EndTypeScale = 4 def getViewPage(self, view): for obj in view.InList: if obj.TypeId == 'TechDraw::DrawPage': if view in obj.Views: return obj return None def computePartCenter(self, view, obj): if obj.TypeId == 'App::Link': partLink = obj objectCenterWorld = partLink.LinkPlacement.Matrix.multiply(partLink.LinkedObject.Shape.CenterOfGravity) else: objectCenterWorld = obj.Shape.CenterOfGravity vertId = view.makeCosmeticVertex3d(objectCenterWorld) vert = view.getCosmeticVertex(vertId) objectCenterView = vert.Point view.removeCosmeticVertex(vertId) return objectCenterView