FreeCAD_Assembly4/treeSelectionOverride.py at master · kbwbe/FreeCAD_Assembly4 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#!/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 libAsm4 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() )