def checkSelection(self):
selectedObj = None
# check that there is an App::Part called 'Model'
# a standard App::Part would also do, but then more error checks are necessary
# if App.ActiveDocument.getObject('Model') and App.ActiveDocument.getObject('Model').TypeId=='App::Part' :
if Asm4.getAssembly() and len(Gui.Selection.getSelection()) == 1:
# set the (first) selected object as global variable
selection = Gui.Selection.getSelection()[0]
if Asm4.isAsm4EE(selection) and selection.SolverId != '':
selectedObj = selection
# now we should be safe
return selectedObj
def restorePlacementEE(self, obj):
# only attempt this on fully restored objects
if self.isLoaded(obj) and obj.LinkedObject.isValid():
# if it's indeed an Asm4 object
# LCS_Origin.Placement * AttachmentOffset * varTmpDoc_3#LCS_Origin.Placement ^ -1
if Asm4.isAsm4EE(obj) and (obj.SolverId == 'Asm4EE' or obj.SolverId
== 'Placement::ExpressionEngine'):
# retrieve the info from the object's properties
(a_Link, sep, a_LCS) = obj.AttachedTo.partition('#')
if a_Link == 'Parent Assembly':
a_Part = None
else:
a_Part = obj.Document.getObject(
a_Link).LinkedObject.Document.Name
l_Part = obj.LinkedObject.Document.Name
l_LCS = obj.AttachedBy[1:]
# build the expression
expr = Asm4.makeExpressionPart(a_Link, a_Part, a_LCS, l_Part,
l_LCS)
# set the expression
obj.setExpression('Placement', expr)
def __init__(self):
# remove selectionFilter
self.selectionFilterStatus = selectionFilter.observerStatus()
selectionFilter.observerDisable()
# get the current active document to avoid errors if user changes tab
self.activeDoc = App.ActiveDocument
# we have checked before that all this is correct
selection = Asm4.getSelectedLink()
if selection is None:
selection = Asm4.getSelectedVarLink()
self.selectedObj = selection
#self.rootAssembly = self.selectedObj.getParentGeoFeatureGroup()
self.rootAssembly = Asm4.getAssembly()
# has been checked before, this is for security only
if Asm4.isAsm4EE(self.selectedObj):
# get the old values
self.old_AO = self.selectedObj.AttachmentOffset
self.old_linkLCS = self.selectedObj.AttachedBy[1:]
else:
# this shouldn't happen
FCC.PrintWarning("WARNING : unsupported Assembly/Solver/Part combination, you shouldn't be seeing this\n")
Asm4.makeAsmProperties(self.selectedObj)
self.old_AO = []
self.old_linkLCS = ''
# define the GUI
# draw the GUI, objects are defined later down
# self.UI = QtGui.QWidget()
# self.form = self.UI
self.form = QtGui.QWidget()
iconFile = os.path.join( Asm4.iconPath , 'Place_Link.svg')
self.form.setWindowIcon(QtGui.QIcon( iconFile ))
self.form.setWindowTitle('Place linked Part')
self.drawUI(self.form)
#save original AttachmentOffset of linked part
self.old_LinkAttachmentOffset = self.selectedObj.AttachmentOffset
self.old_LinkRotation = self.selectedObj.AttachmentOffset.Rotation
self.old_LinkPosition = self.selectedObj.AttachmentOffset.Base
# default values correspond to original AttachmentOffset of linked part
self.Xtranslation = self.old_LinkPosition[0]
self.Ytranslation = self.old_LinkPosition[1]
self.Ztranslation = self.old_LinkPosition[2]
self.XrotationAngle = self.old_LinkRotation.toEuler()[0]
self.YrotationAngle = self.old_LinkRotation.toEuler()[1]
self.ZrotationAngle = self.old_LinkRotation.toEuler()[2]
# save previous view properties
self.old_OverrideMaterial = self.selectedObj.ViewObject.OverrideMaterial
self.old_DrawStyle = self.selectedObj.ViewObject.DrawStyle
self.old_LineWidth = self.selectedObj.ViewObject.LineWidth
self.old_DiffuseColor = self.selectedObj.ViewObject.ShapeMaterial.DiffuseColor
self.old_Transparency = self.selectedObj.ViewObject.ShapeMaterial.Transparency
# set new view properties
self.selectedObj.ViewObject.OverrideMaterial = True
self.selectedObj.ViewObject.DrawStyle = DrawStyle
self.selectedObj.ViewObject.LineWidth = LineWidth
self.selectedObj.ViewObject.ShapeMaterial.DiffuseColor = DiffuseColor
self.selectedObj.ViewObject.ShapeMaterial.Transparency = Transparency
# get the old values
self.old_EE = ''
old_Parent = ''
old_ParentPart = ''
old_attLCS = ''
constrName = ''
linkedDoc = ''
old_linkLCS = ''
# get and store the current expression engine:
self.old_EE = Asm4.placementEE(self.selectedObj.ExpressionEngine)
# decode the old ExpressionEngine
# if the decode is unsuccessful, old_Expression is set to False and the other things are set to 'None'
(self.old_Parent, separator, self.old_parentLCS) = self.selectedObj.AttachedTo.partition('#')
( old_Parent, old_attLCS, old_linkLCS ) = self.splitExpressionLink( self.old_EE, self.old_Parent )
# sometimes, the object is in << >> which is an error by FreeCAD,
# because that's reserved for labels, but we still look for it
if len(old_attLCS)>4 and old_attLCS[:2]=='<<' and old_attLCS[-2:]=='>>':
old_attLCS = old_attLCS[2:-2]
if len(old_linkLCS)>4 and old_linkLCS[:2]=='<<' and old_linkLCS[-2:]=='>>':
old_linkLCS = old_linkLCS[2:-2]
# initialize the UI with the current data
self.attLCStable = []
self.initUI()
# now self.parentList and self.parentTable are available
# find all the linked parts in the assembly
for obj in self.activeDoc.findObjects("App::Link"):
if self.rootAssembly.getObject(obj.Name) is not None and hasattr(obj.LinkedObject,'isDerivedFrom'):
linkedObj = obj.LinkedObject
if linkedObj.isDerivedFrom('App::Part') or linkedObj.isDerivedFrom('PartDesign::Body'):
# ... except if it's the selected link itself
if obj != self.selectedObj:
self.parentTable.append( obj )
# add to the drop-down combo box with the assembly tree's parts
objIcon = linkedObj.ViewObject.Icon
objText = Asm4.labelName(obj)
self.parentList.addItem( objIcon, objText, obj)
# find all the LCS in the selected link
self.partLCStable = Asm4.getPartLCS( self.selectedObj.LinkedObject )
# build the list
self.partLCSlist.clear()
for lcs in self.partLCStable:
newItem = QtGui.QListWidgetItem()
newItem.setText(Asm4.labelName(lcs))
newItem.setIcon( lcs.ViewObject.Icon )
self.partLCSlist.addItem(newItem)
# find the old LCS in the list of LCS of the linked part...
# MatchExactly, MatchContains, MatchEndsWith ...
# find with Name ...
lcs_found = self.partLCSlist.findItems( old_linkLCS, QtCore.Qt.MatchExactly )
# ... or with (Name)
if not lcs_found:
lcs_found = self.partLCSlist.findItems( '('+old_linkLCS+')', QtCore.Qt.MatchEndsWith )
if lcs_found:
# ... and select it
self.partLCSlist.setCurrentItem( lcs_found[0] )
# find the oldPart in the part list...
if old_Parent == 'Parent Assembly':
parent_found = True
parent_index = 1
else:
parent_found = False
parent_index = 1
for item in self.parentTable[1:]:
if item.Name == old_Parent:
parent_found = True
break
else:
parent_index = parent_index +1
if not parent_found:
parent_index = 0
self.parentList.setCurrentIndex( parent_index )
# this should have triggered self.getPartLCS() to fill the LCS list
# find the old attachment Datum in the list of the Datums in the linked part...
lcs_found = self.attLCSlist.findItems( old_attLCS, QtCore.Qt.MatchExactly )
if not lcs_found:
lcs_found = self.attLCSlist.findItems( '('+old_attLCS+')', QtCore.Qt.MatchEndsWith )
if lcs_found:
# ... and select it
self.attLCSlist.setCurrentItem( lcs_found[0] )
# selection observer to detect selection of LCS in the 3D window and tree
Gui.Selection.addObserver(self, 0)
def Activated(self):
# try with a regular App::Link
selection = Asm4.getSelectedLink()
# may-be an Asm4::VariantLink ?
if selection is None:
selection = Asm4.getSelectedVarLink()
# if we found a valid link
if selection is not None:
# check that it's in the root assembly
parent = selection.getParentGeoFeatureGroup()
if parent and parent == Asm4.getAssembly():
# if it's a valid assembly and part
if Asm4.isAsm4EE(selection):
# BUGFIX: if the part was corrupted by Assembly4 v0.11.5:
if hasattr(selection, 'MapMode'):
Asm4.warningBox(
"This Part has the Attachment extension, it can only be placed manually"
)
else:
# launch the UI in the task panel
ui = placeLinkUI()
Gui.Control.showDialog(ui)
# else try to convert it
else:
convert = Asm4.confirmBox(
"This Part wasn't assembled with this Assembly4 WorkBench, but I can convert it."
)
if convert:
Asm4.makeAsmProperties(selection, reset=True)
# launch the UI in the task panel
ui = placeLinkUI()
Gui.Control.showDialog(ui)
else:
Asm4.warningBox('Please select a link in the assembly Model.')
else:
# or any part that has a Placement ?
if len(Gui.Selection.getSelection()) == 1:
selection = Gui.Selection.getSelection()[0]
# object has a Placement property
if hasattr(selection,
'Placement') and selection.getTypeIdOfProperty(
'Placement') == 'App::PropertyPlacement':
# we don't want to mess with obects that are attached with the Attacher (MapMode)
if hasattr(selection, 'MapMode'):
FCC.PrintMessage(
'Object has MapMode property, you should use that\n'
)
else:
# check that it's in the root assembly
parent = selection.getParentGeoFeatureGroup()
if parent and parent == Asm4.getAssembly():
# is it's a virgin object, give it Asm4 properties
if not hasattr(selection, 'SolverId'):
Asm4.makeAsmProperties(selection)
# if it's a valid assembly and part
if Asm4.isAsm4EE(selection):
# launch the UI in the task panel
ui = placePartUI()
Gui.Control.showDialog(ui)
# else try to convert it
else:
convert = Asm4.confirmBox(
"This Part wasn't assembled with this Assembly4 WorkBench, but I can convert it."
)
if convert:
Asm4.makeAsmProperties(selection,
reset=True)
# launch the UI in the task panel
ui = placePartUI()
Gui.Control.showDialog(ui)
# the selected object doesn't belong to the root assembly
else:
Asm4.warningBox(
'Please select an object in the assembly Model.'
)
return