FreeCAD workbench to create assembly handbooks
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

92 lines
4.1 KiB

import FreeCAD as App
import FreeCADGui as Gui
# Code copied from Assembly4 to select the top-most link instead of a part or sub-object when clicking in the 3D view
class DocLinkObserver:
select_link_mode = False
def __init__(self):
Gui.Selection.addObserver(self, 0) # 0 forces to resolve the links
def addSelection(self, doc, obj, sub, pnt):
# Since both 3D view clicks and manual tree selection gets into the same callback
# we will determine by clicked coordinates, for manual tree selections the coordinates are (0,0,0)
if self.select_link_mode and pnt != (0,0,0):
# 3D view click
objList = App.getDocument(doc).getObject(obj).getSubObjectList(sub)
# Build the name of the selected sub-object for multiple sub-assembly levels
subObjName = ''
# first look for the linked object of the selected entity:
# Get linked object name that handles sub-sub-assembly
for subObj in objList:
if subObj.TypeId=='App::Link':
subObjName = subObjName + subObj.Name + '.'
# if no App::Link found, let's look for other things:
if subObjName == '':
for subObj in objList:
if subObj.TypeId=='App::Part' or subObj.TypeId=='PartDesign::Body'or subObj.isDerivedFrom('Part::Feature'):
# the objList contains also the top-level object, don't count it twice
if subObj.Name != obj:
subObjName = subObjName + subObj.Name + '.'
# if we found something, make it the selection
if subObjName != '':
Gui.Selection.removeSelection(doc, obj, sub)
Gui.Selection.addSelection(doc, obj, subObjName)
#FCC.PrintMessage("*"+doc+"*"+obj+"*"+subObjName+"*\n")
# The main document observer (also observes selections)
class DocObserver:
changed_object_by_type = {}
selection_by_type = {}
was_selected = []
def __init__(self):
Gui.Selection.addObserver(self)
def slotChangedObject(self, obj, prop):
#print("object changed: " + str(obj).replace('<', '').replace(' object>', '') + " " + obj.Name + " : " + str(prop))
callbacks = self.changed_object_by_type.get(obj.TypeId, None)
if callbacks is not None:
for callback in callbacks.values():
callback(obj, prop)
def addSelection(self, doc, obj_name, sub, pnt):
self.was_selected = Gui.Selection.getSelection()
obj = App.getDocument(doc).getObject(obj_name)
callbacks = self.selection_by_type.get(obj.TypeId, None)
if callbacks is not None:
for callback in callbacks.values():
callback('added', obj, sub, pnt)
def removeSelection(self, doc, obj_name, sub):
self.was_selected = Gui.Selection.getSelection()
obj = App.getDocument(doc).getObject(obj_name)
callbacks = self.selection_by_type.get(obj.TypeId, None)
if callbacks is not None:
for callback in callbacks.values():
callback('removed', obj, sub, None)
def clearSelection(self, p):
was_selected = self.was_selected
self.was_selected = []
for obj in was_selected:
callbacks = self.selection_by_type.get(obj.TypeId, None)
if callbacks is not None:
for callback in callbacks.values():
callback('removed', obj, None, None)
def onObjectTypeChanged(self, callback_id: str, type_id: str, callback):
callbacks = self.changed_object_by_type.get(type_id, None)
if callbacks is None:
callbacks = {}
self.changed_object_by_type[type_id] = callbacks
callbacks[callback_id] = callback
def onObjectTypeSelected(self, callback_id: str, type_id: str, callback):
callbacks = self.selection_by_type.get(type_id, None)
if callbacks is None:
callbacks = {}
self.selection_by_type[type_id] = callbacks
callbacks[callback_id] = callback