forked from Zolko-123/FreeCAD_Assembly4
-
Notifications
You must be signed in to change notification settings - Fork 0
/
newModelCmd.py
106 lines (81 loc) · 3.87 KB
/
newModelCmd.py
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
#
# newModelCmd.py
import math, re, os
from PySide import QtGui, QtCore
import FreeCADGui as Gui
import FreeCAD as App
import Part
import Asm4_libs as Asm4
class newModel:
"""
+-----------------------------------------------+
| creates the Assembly4 Model |
| which is an App::Part |
| with some extra features and properties |
+-----------------------------------------------+
def makeAssembly():
assembly = App.ActiveDocument.addObject('App::Part','Assembly')
assembly.Type='Assembly'
assembly.addProperty( 'App::PropertyString', 'AssemblyType', 'Assembly' )
assembly.AssemblyType = 'Part::Link'
assembly.newObject('App::DocumentObjectGroup','Constraints')
return assembly
"""
def GetResources(self):
tooltip = "<p>Create a new assembly Model</p>"
tooltip += "<p>Use the command \"Insert Part\" to import a part into the assembly. "
tooltip += "Parts can be in this document in the \"Parts\" folder, "
tooltip += "or in another document which must be open in this session</p>"
iconFile = os.path.join( Asm4.iconPath , 'Asm4_Model.svg')
return {"MenuText": "New Model", "ToolTip": tooltip, "Pixmap" : iconFile }
def IsActive(self):
if App.ActiveDocument:
return(True)
else:
return(False)
# the real stuff
def Activated(self):
# get the current active document to avoid errors if user changes tab
self.activeDoc = App.activeDocument()
# check whether there is already Model in the document
if Asm4.checkModel():
Asm4.warningBox("This document already contains a valid Model, please use it")
return
# create a group 'Parts' to hold all parts in the assembly document (if any)
# must be done before creating the Asm4 Model
# because of order of visibilities
partsGroup = self.activeDoc.getObject('Parts')
if partsGroup is None:
partsGroup = self.activeDoc.addObject( 'App::DocumentObjectGroup', 'Parts' )
# create a new App::Part called 'Model'
model = self.activeDoc.addObject('App::Part','Model')
# set the type as a "proof" that it's an Assembly4 Model
model.Type='Assembly'
model.addProperty( 'App::PropertyString', 'AssemblyType', 'Assembly' )
model.AssemblyType = 'Part::Link'
# add an LCS at the root of the Model, and attach it to the 'Origin'
lcs0 = model.newObject('PartDesign::CoordinateSystem','LCS_Origin')
lcs0.Support = [(model.Origin.OriginFeatures[0],'')]
lcs0.MapMode = 'ObjectXY'
lcs0.MapReversed = False
# create a group Constraints to store future solver constraints there
model.newObject('App::DocumentObjectGroup','Constraints')
# create an object Variables to hold variables to be used in this document
model.addObject(Asm4.createVariables())
# create a group Configurations to store future solver constraints there
model.newObject('App::DocumentObjectGroup','Configurations')
# move existing parts and bodies at the document root to the Parts group
# not nested inside other parts, to keep hierarchy
if partsGroup.TypeId=='App::DocumentObjectGroup':
for obj in self.activeDoc.Objects:
if obj.TypeId in Asm4.containerTypes and obj.Name!='Model' and obj.getParentGeoFeatureGroup() is None:
partsGroup.addObject(obj)
else:
Asm4.warningBox( 'There seems to already be a \"Parts\" object, you might get unexpected behaviour' )
# recompute to get rid of the small overlays
model.recompute()
self.activeDoc.recompute()
# add the command to the workbench
Gui.addCommand( 'Asm4_newModel', newModel() )