treeSelectionOverride.py

#!/usr/bin/env python3
# coding: utf-8
#
# treeSelectionOverride.py
#
# The code tracks the selections made

import os
from PySide import QtGui, QtCore
import FreeCADGui as Gui
import FreeCAD as App
from FreeCAD import Console as FCC

import Asm4_libs as Asm4



observer = None

"""
    +-----------------------------------------------+
    |                  main class                   |
    +-----------------------------------------------+
"""
class treeSelectionOverrideCmd( QtGui.QDialog):
    def __init__(self):
        super(treeSelectionOverrideCmd,self).__init__()

    def GetResources(self):
        return {"MenuText": "Enable/Disable 3D View selection mode",
                "ToolTip": "Enable/Disable 3D View selection mode\n\n"    + \
                "Allows to select a Link object in the 3D view\n" + \
                "window instead of the Model tree",
                "Pixmap" : os.path.join( Asm4.iconPath , 'Asm4_enableLinkSelection.svg')
                }

    def IsActive(self):
        return True

    def Activated(self):
        global observer
        # This function is executed when the command is activated
        if observer is None:
            Enable()
        else:
            Disable()


"""
    +-----------------------------------------------+
    |               observer class                  |
    +-----------------------------------------------+
"""
class selObserver3DView:
    # selects the App::Link object when clicked on in the 3D view
    def addSelection(self,doc,obj,sub,pnt):               # Selection object
        # 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 pnt != (0,0,0):
            # 3D view click
            # Get linked object name that handles sub-sub-assembly
            #subObjName = getLinkedObjectName(doc, obj, sub)
            objList = App.getDocument(doc).getObject(obj).getSubObjectList(sub)
            # Build the name of the selected sub-object for multiple sub-assembly levels
            subObjName = ''
            for subObj in objList:
                if subObj.TypeId == 'App::Link':
                    subObjName = subObjName + subObj.Name + '.'
            if subObjName != '':
                Gui.Selection.clearSelection()
                Gui.Selection.addSelection(doc, obj, subObjName)


def Enable():
    global observer
    observer = selObserver3DView();
    # add the listener, 0 forces to resolve the links
    Gui.Selection.addObserver(observer, 0)
    #print("3D view link selection mode is now enabled")
    FCC.PrintMessage("3D view link selection mode is now ENABLED\n")


def Disable():
    global observer
    # only print to Console if the observer was there
    if observer:
        print_disable = True
    else:
        print_disable = False
    Gui.Selection.removeObserver(observer) 
    observer = None
    #print("3D view link selection mode is now disabled")
    if print_disable:
        FCC.PrintMessage("3D view link selection mode is now DISABLED\n")



"""
    +-----------------------------------------------+
    |       add the command to the workbench        |
    +-----------------------------------------------+
"""
Gui.addCommand( 'Asm4_treeSelectionOverrideCmd', treeSelectionOverrideCmd() )