newModelCmd.py

#!/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 libAsm4 as Asm4




class newModel:
    """
    +-----------------------------------------------+
    |          creates the Assembly4 Model          |
    |             which is an App::Part             |
    |    with some extra features and properties    |
    +-----------------------------------------------+
    """
    def GetResources(self):
        return {"MenuText": "New Model",
                "Accel": "Ctrl+M",
                "ToolTip": "Create a new Assembly4 Model",
                "Pixmap" : os.path.join( Asm4.iconPath , 'Asm4_Model.svg')
                }


    def IsActive(self):
        if App.ActiveDocument:
            return(True)
        else:
            return(False)


    # checks whether there already is an Asm4 Model in the document
    def checkModel(self):
        if self.activeDoc.getObject('Model'):
            if self.activeDoc.Model.TypeId=='App::Part' and self.activeDoc.Model.Type=='Assembly4 Model':
                Asm4.warningBox("This document already contains an Assembly4 Model.")
            else:
                Asm4.warningBox("This document already contains another FreeCAD object called \"Model\". I cannot create an Assembly4 Model.")
            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 not self.checkModel():
            # create a group 'Parts' to hold all parts in the assembly document (if any)
            if not self.activeDoc.getObject('Parts'):
                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='Assembly4 Model'
            # 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
            variables = model.newObject('App::FeaturePython','Variables')
            #variables.ViewObject.Proxy = Asm4.setCustomIcon(variables,'Asm4_Variables.svg')

            # create a Configuration property
            model.addProperty('App::PropertyEnumeration', 'Configuration', 'Parameters')
            model.Configuration = ['Default']
            # move existing parts at the document root to the Parts group
            # not nested inside other parts, to keep hierarchy
            for obj in self.activeDoc.Objects:
                if obj.TypeId=='App::Part' and obj.Name!='Model' and obj.getParentGeoFeatureGroup()==None:
                    partsGroup.addObject(obj)
            # move existing bodies at the document root to the Parts group
            # not nested inside other parts, to keep hierarchy
            for obj in self.activeDoc.Objects:
                if obj.TypeId=='PartDesign::Body' and obj.getParentGeoFeatureGroup()==None:
                    partsGroup.addObject(obj)

            # recompute to get rid of the small overlays
            model.recompute()
            self.activeDoc.recompute()



# add the command to the workbench
Gui.addCommand( 'Asm4_newModel', newModel() )