AndreasL
/
assembly_handbook
forked from youen/assembly_handbook
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

Added cache system to reduce part highlighting time

master
Youen 2 years ago
parent
ed83fdb0b4
commit
4bf1b9245f
4 changed files with 142 additions and 107 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 12
      InitGui.py
  2. 139
      ahb_cmd_view_annotate.py
  3. 87
      ahb_parts_cache.py
  4. 9
      resources/A4_Landscape_blank.svg

12
InitGui.py
Unescape View File

@ -27,6 +27,8 @@ class AssemblyHandbookWorkbench(Gui.Workbench):
docObserver = None docObserver = None
partsCache = None
def GetClassName(self): def GetClassName(self):
return "Gui::PythonWorkbench" return "Gui::PythonWorkbench"
@ -43,11 +45,15 @@ class AssemblyHandbookWorkbench(Gui.Workbench):
if self.context is None: if self.context is None:
self.initializeContext() self.initializeContext()
import ahb_document_observer
if self.docObserver is None: if self.docObserver is None:
import ahb_document_observer
self.docObserver = ahb_document_observer.DocObserver() self.docObserver = ahb_document_observer.DocObserver()
App.addDocumentObserver(self.docObserver) App.addDocumentObserver(self.docObserver)
if self.partsCache is None:
import ahb_parts_cache
self.partsCache = ahb_parts_cache.PartsCache()
def Deactivated(self): def Deactivated(self):
""" """
code which should be computed when this workbench is deactivated code which should be computed when this workbench is deactivated
@ -100,6 +106,10 @@ class AssemblyHandbookWorkbench(Gui.Workbench):
import ahb_context import ahb_context
importlib.reload(ahb_context) importlib.reload(ahb_context)
import ahb_parts_cache
importlib.reload(ahb_parts_cache)
self.partsCache = ahb_parts_cache.PartsCache()
self.initializeContext() self.initializeContext()
self.registerCommands() self.registerCommands()
pass pass

139
ahb_cmd_view_annotate.py
Unescape View File

@ -1,7 +1,5 @@
import FreeCADGui as Gui import FreeCADGui as Gui
import FreeCAD as App import FreeCAD as App
import numpy as np
from pickle import NONE
class AHB_View_Annotate: class AHB_View_Annotate:
updating_balloon = False updating_balloon = False
@ -85,8 +83,9 @@ class AHB_View_Annotate:
#print(operation, obj.Name, sub, point) #print(operation, obj.Name, sub, point)
if "AssemblyHandbook_PartName" in balloon.PropertiesList: if "AssemblyHandbook_PartName" in balloon.PropertiesList:
if operation == 'added': if operation == 'added':
print("Selected balloon of " + balloon.AssemblyHandbook_PartName) #print("Selected balloon of " + balloon.AssemblyHandbook_PartName)
workbench = Gui.getWorkbench("AssemblyHandbookWorkbench") #: :type workbench: AssemblyHandbookWorkbench
doc = balloon.Document doc = balloon.Document
view = balloon.SourceView view = balloon.SourceView
page = self.getViewPage(view) page = self.getViewPage(view)
@ -95,85 +94,15 @@ class AHB_View_Annotate:
cache_edges_start_time = time.perf_counter() cache_edges_start_time = time.perf_counter()
# create temporary view of part part_view = workbench.partsCache.getPart2DView(view, partLink)
tmpView = None
for v in page.Views:
if v.TypeId == 'TechDraw::DrawViewPart' and len(v.XSource) == 1 and v.XSource[0] == partLink:
tmpView = v
if tmpView is None:
keepUpdated = page.KeepUpdated
page.KeepUpdated = True
tmpView = doc.addObject("TechDraw::DrawViewPart", "View")
tmpView.Direction = view.Direction
tmpView.XDirection = view.XDirection
tmpView.ScaleType = 'Custom'
tmpView.Scale = 1.0
tmpView.XSource = partLink
page.addView(tmpView)
tmpView.recompute()
page.KeepUpdated = keepUpdated
cache_edges_end_render_time = time.perf_counter()
#tmpView.clearCosmeticEdges()
#tmpView.recompute()
# copy edges relative to center
# TODO: in-memory cache
tmpCenter = self.computePartCenter(tmpView, partLink)
#print('tmpCenter', tmpCenter)
tmpEdges = tmpView.getVisibleEdges()
# count lines
numLines = 0
for edge in tmpEdges:
if not hasattr(edge.Curve, 'Degree') or edge.Curve.Degree == 1: numLines += 1
# store all lines in a packed array of floats (for each line: X1, Y1, X2, Y2)
cachedLines = np.empty([numLines, 4])
lineIdx = 0
for edge in tmpEdges:
degree = 1
try:
degree = edge.Curve.Degree
except:
pass
if degree == 1:
#print('tmpEdge', edge.Vertexes[0].Point.x, edge.Vertexes[0].Point.y, edge.Vertexes[1].Point.x, edge.Vertexes[1].Point.y)
sx = 1.0
sy = -1.0
cachedLines[lineIdx] = [
edge.Vertexes[0].Point.x*sx - tmpCenter.x,
edge.Vertexes[0].Point.y*sy - tmpCenter.y,
edge.Vertexes[1].Point.x*sx - tmpCenter.x,
edge.Vertexes[1].Point.y*sy - tmpCenter.y
]
#print('tmpLine', cachedLines[lineIdx])
lineIdx = lineIdx + 1
cache_edges_end_time = time.perf_counter() cache_edges_end_time = time.perf_counter()
'''tmpView.makeCosmeticLine(tmpCenter - App.Vector(0,3), tmpCenter + App.Vector(0,3))
tmpView.makeCosmeticLine(tmpCenter - App.Vector(3,0), tmpCenter + App.Vector(3,0))
drawCenter = tmpCenter + App.Vector(-1,0)
for line in cachedLines:
p0 = drawCenter + App.Vector(line[0], line[1])
p1 = drawCenter + App.Vector(line[2], line[3])
tmpView.makeCosmeticLine(p0, p1)
tmpView.recompute()'''
# delete temporary view
doc.removeObject(tmpView.Name)
#view.clearCosmeticEdges()
#view.recompute()
update_final_edges_start_time = time.perf_counter() update_final_edges_start_time = time.perf_counter()
# iterate edges of actual view and highlight matching edges # iterate edges of actual view and highlight matching edges
center = self.computePartCenter(view, partLink) center = self.computePartCenter(view, partLink)
#print('center', center) for edgeIdx in range(10000):
for edgeIdx in range(1000):
hasEdge = False hasEdge = False
try: try:
edge = view.getEdgeByIndex(edgeIdx) edge = view.getEdgeByIndex(edgeIdx)
@ -183,28 +112,20 @@ class AHB_View_Annotate:
if not hasEdge: if not hasEdge:
break break
view.formatGeometricEdge(edgeIdx,1,0.25,0,True) # reset edge format
#view.formatGeometricEdge(edgeIdx,1,0.25,0,True)
degree = 1 if not hasattr(edge.Curve, 'Degree') or edge.Curve.Degree == 1:
try:
degree = edge.Curve.Degree
except:
pass
if degree == 1:
#print('edge', edge.Vertexes[0].X, edge.Vertexes[0].Y, edge.Vertexes[1].X, edge.Vertexes[1].Y)
sx = 1
sy = 1
edgeData = [ edgeData = [
(edge.Vertexes[0].X - center.x)*sx, edge.Vertexes[0].X - center.x,
(edge.Vertexes[0].Y - center.y)*sy, edge.Vertexes[0].Y - center.y,
(edge.Vertexes[1].X - center.x)*sx, edge.Vertexes[1].X - center.x,
(edge.Vertexes[1].Y - center.y)*sy edge.Vertexes[1].Y - center.y
] ]
v0 = App.Vector(edgeData[0], edgeData[1]) v0 = App.Vector(edgeData[0], edgeData[1])
v1 = App.Vector(edgeData[2], edgeData[3]) v1 = App.Vector(edgeData[2], edgeData[3])
#print(edgeData)
for line in cachedLines: for line in part_view.cached_lines:
l0 = App.Vector(line[0], line[1]) l0 = App.Vector(line[0], line[1])
l1 = App.Vector(line[2], line[3]) 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 = abs(edgeData[0] - line[0]) + abs(edgeData[1] - line[1]) + abs(edgeData[2] - line[2]) + abs(edgeData[3] - line[3])
@ -212,24 +133,32 @@ class AHB_View_Annotate:
if d < 0.001: if d < 0.001:
view.formatGeometricEdge(edgeIdx,1,0.25,(0,0.85,0),True) view.formatGeometricEdge(edgeIdx,1,0.25,(0,0.85,0),True)
view.requestPaint()
update_final_edges_end_time = time.perf_counter() update_final_edges_end_time = time.perf_counter()
print("Part render time: " + str(round((cache_edges_end_render_time - cache_edges_start_time) * 1000) / 1000) + "s") #print("Part render time: " + str(round((cache_edges_end_time - cache_edges_start_time) * 1000) / 1000) + "s")
print("Part read time: " + str(round((cache_edges_end_time - cache_edges_end_render_time) * 1000) / 1000) + "s") #print("Update view time: " + str(round((update_final_edges_end_time - update_final_edges_start_time) * 1000) / 1000) + "s")
print("Update view time: " + str(round((update_final_edges_end_time - update_final_edges_start_time) * 1000) / 1000) + "s")
'''view.makeCosmeticLine(center - App.Vector(0,3), center + App.Vector(0,3)) elif operation == 'removed':
view.makeCosmeticLine(center - App.Vector(3,0), center + App.Vector(3,0)) #print("Deselected balloon of " + balloon.AssemblyHandbook_PartName)
drawCenter = center + App.Vector(1,0)
for line in cachedLines:
p0 = drawCenter + App.Vector(line[0], line[1])
p1 = drawCenter + App.Vector(line[2], line[3])
view.makeCosmeticLine(p0, p1)'''
#view.recompute() view = balloon.SourceView
elif operation == 'removed': # reset edges format
print("Deselected balloon of " + balloon.AssemblyHandbook_PartName) for edgeIdx in range(10000):
hasEdge = False
try:
edge = view.getEdgeByIndex(edgeIdx)
hasEdge = True
except:
pass
if not hasEdge:
break
view.formatGeometricEdge(edgeIdx,1,0.25,0,True)
view.requestPaint()
def updateBalloon(self, balloon): def updateBalloon(self, balloon):
doc = App.activeDocument() doc = App.activeDocument()

87
ahb_parts_cache.py
Unescape View File

@ -0,0 +1,87 @@
import FreeCAD as App
import FreeCADGui as Gui
class PartCachedView:
def __init__(self, direction, x_direction, obj):
self.direction = direction
self.x_direction = x_direction
self.doc_name = obj.Document.Name
self.obj_name = obj.Name
self.cached_lines = None
def render(self):
import numpy as np
import os
doc = App.getDocument(self.doc_name) #: :type doc: App.Document
obj = doc.getObject(self.obj_name) #: :type obj: App.DocumentObject
# create temporary view
page = doc.addObject('TechDraw::DrawPage', 'TmpPage')
if not page.KeepUpdated: page.KeepUpdated = True
template = doc.addObject('TechDraw::DrawSVGTemplate', 'Template')
import ahb_locator
template.Template = os.path.join(os.path.dirname(ahb_locator.__file__), "resources/A4_Landscape_blank.svg")
page.Template = template
tmpView = doc.addObject('TechDraw::DrawViewPart', 'TmpView')
tmpView.Direction = self.direction
tmpView.XDirection = self.x_direction
tmpView.Perspective = False
tmpView.ScaleType = 'Custom'
tmpView.Scale = 1.0
tmpView.XSource = [obj]
page.addView(tmpView)
tmpView.recompute()
# copy edges relative to center
tmpCenter = self.computePartCenter(tmpView, obj)
# count lines
tmpEdges = tmpView.getVisibleEdges()
numLines = 0
for edge in tmpEdges:
if not hasattr(edge.Curve, 'Degree') or edge.Curve.Degree == 1: numLines += 1
# store all lines in a packed array of floats (for each line: X1, Y1, X2, Y2)
self.cached_lines = np.empty([numLines, 4])
lineIdx = 0
for edge in tmpEdges:
if not hasattr(edge.Curve, 'Degree') or edge.Curve.Degree == 1:
sx = 1.0
sy = -1.0
self.cached_lines[lineIdx] = [
edge.Vertexes[0].Point.x*sx - tmpCenter.x,
edge.Vertexes[0].Point.y*sy - tmpCenter.y,
edge.Vertexes[1].Point.x*sx - tmpCenter.x,
edge.Vertexes[1].Point.y*sy - tmpCenter.y
]
lineIdx = lineIdx + 1
# delete temporary view
doc.removeObject(page.Name)
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
class PartsCache:
part_views = {}
def getPart2DView(self, view, obj):
key = (view.Direction.x, view.Direction.y, view.Direction.z, view.XDirection.x, view.XDirection.y, view.XDirection.z, obj.Document.Name, obj.Name)
part_view = self.part_views.get(key, None)
if part_view is None:
part_view = PartCachedView(view.Direction, view.XDirection, obj)
part_view.render()
self.part_views[key] = part_view
return part_view

9
resources/A4_Landscape_blank.svg
Unescape View File

@ -0,0 +1,9 @@
<?xml version="1.0" encoding="UTF-8" standalone="no"?>
<svg
xmlns="http://www.w3.org/2000/svg" version="1.1"
xmlns:freecad="http://www.freecadweb.org/wiki/index.php?title=Svg_Namespace"
width="297mm"
height="210mm"
viewBox="0 0 297 210">
</svg>
Side by Side

After

Width:  |  Height:  |  Size: 251 B

Write Preview
Loading…
Cancel
Save