Create a FeaturePython object part I

From FreeCAD Documentation
PySide
Embedding FreeCAD

Introduction

FeaturePython objects (also often referred to as 'Scripted Objects') provide users the ability to extend FreeCAD's with objects that integrate seamlessly into the FreeCAD framework.


This encourages:

  • Rapid prototyping of new objects and tools with custom Python classes.
  • Serialization through 'App::Property' objects, without embedding any script in the FreeCAD document file.
  • Creative freedom to adapt FreeCAD for any task!


How Does It Work?

FreeCAD comes with a number of default object types for managing different kinds of geometry. Some of them have 'FeaturePython' alternatives that allow for user customization with a custom python class.

The custom python class simply takes a reference to one of these objects and modifies it in any number of ways. For example, the python class may add properties directly to the object, modifying other properties when it's recomputed, or linking it to other objects. In addition the python class implements certain methods to enable it to respond to document events, making it possible to trap object property changes and document recomputes.

It's important to remember, however, that for as much as one can accomplish with custom classes and FeaturePython objects, when it comes time to save the document, only the FeaturePython object itself is serialized. The custom class and it's state are not retained between document reloading. Doing so would require embedding script in the FreeCAD document file, which poses a significant security risk, much like the risks posed by embedding VBA macros in Microsoft Office documents.

Thus, a FeaturePython object ultimately exists entirely apart from it's script. The inconvenience posed by not packing the script with the object in the document file is far less than the risk posed by running a file embedded with an unknown script. However, the script module path is stored in the document file. Therefore, a user need only install the custom python class code as an importable module following the same directory structure to regain the lost functionality.

The Complete Beginner's Guide to FeaturePython Objects

Let's start from the very beginning (it's a very good place to start. ;) )


We're going to construct a complete, working example of a FeaturePython custom class, identifying all of the major components and gaining an intimate understanding of how everything works as we go. But before we do, there's a few details to get out of the way.

Setting up your development environment

To begin, FeaturePython Object classes need to act as importable modules in FreeCAD. That means you need to place them in a path that exists in your Python environment (or add it specifically). For the purposes of this tutorial, we're going to use the FreeCAD user Macro folder, though if you have another idea in mind, feel free to use that instead!


Anyway, if you don't know where the FreeCAD Macro folder is:

  • Windows: Type '%APPDATA%/FreeCAD/Macro' in the filepath bar at the top of Explorer
  • Linux: Navigate to /home/USERNAME/.FreeCAD/Macro
  • Mac: Navigate to /Users/USERNAME/Library/Preferences/FreeCAD/Macro


Now we need to create some files.

  • In the Macro folder create a new folder called fpo.
  • In the fpo folder, create a new folder called box.
  • In the box folder create two files: __init__.py and box.py (leave both empty for now)


The fpo folder provides a nice spot to play with new FeaturePython objects and the box folder is the module we will be working in.

__init__.py tells Python that in the folder is an importable module, and box.py will be the class file for our new FeaturePython Object


Your directory structure should look like this:

.FreeCAD
  |--> Macro
       |--> fpo
            |--> box
                 |--> __init__.py
                 |--> box.py


With our module paths and files created, let's make sure FreeCAD is set up properly. If FreeCAD isn't loaded, now is a good time to boot it up.

Make sure the Python Console and Report View are enabled by selecting View -> Panels -> Report view and Python console

If you haven't been introduced to the Python Console in FreeCAD, you can learn more about it here


Now that FreeCAD is set up, switch over to your favorite code editor, navigate to the Macro/fpo/box folder, and open box.py. It's time to write some code!


A Very Basic FeaturePython Object

We're going to start really simply. It won't be impressive, but it will provide a clear idea of how FeaturePython objects work with custom python classes.


So, let's get started by writing our class and it's constructor:

 class box():
   
     def __init__(self, obj):
         """
         Default Constructor
         """
   
         self.Type = 'box'
   
         obj.Proxy = self
         self.Object = obj
The __init__() method breakdown
def __init__(self, obj): Parameters refer to the Python class itself and the FeaturePython object that it is attached to.
self.Type = 'box' String definition of the custom python type
obj.Proxy = self Stores a reference to the Python class in the FeaturePython object
self.Object = obj Stores a reference to the FeaturePython object in the Python class.


This small piece of code illustrates just how FeaturePython objects interact with Python classes.

We'll examine that in more detail momentarily, but first we need to add a little more code to manage object creation.

In the box.py file at the top, add the following code:

 import FreeCAD as App

 def create(obj_name):
     """
     Object creation method
     """

     obj = App.ActiveDocument.addObject('App::FeaturePython', obj_name)

     fpo = box(obj)

     return fpo


The create() method breakdown
import FreeCAD as App Standard import for most python scripts. The App alias is not required, but it is consistent with the FreeCAD Python console.
obj = ... addObject(...) Creates a new FreeCAD FeaturePython object with the name passed to the method.
fpo = box(obj) Create our custom class instance and link it to the FeaturePython object.



The create() method is not required, but it provides a nice way to encapsulate the object creation code.


Testing the Code

Now we can try our new object. Save your code and return to FreeCAD


In the Python Console, type the following:

>>> from fpo.box import box


Now, we need to create our object:

>>> box.create('my_box')


You should see a new object appear in the tree view at the top left labelled my_box. Note that the icon is gray. FreeCAD is simply telling us that the object is not able to display anything in the 3D view... yet.

That's ok. There's still something to learn here.


Click on the object and note what appears in the property panel under it. Not very much - just the name of the object. We'll need to add some properties in a bit.

Now, back at the console, let's see just what that code we wrote in the __init__() function does for us.


First, let's make referencing our new object a little more convenient:

>>> mybox = App.ActiveDocument.my_box


Now, let's list the attributes of the object:

>>> dir(mybox)
['Content', 'Document', 'ExpressionEngine', 'InList', 'InListRecursive', 'Label', 'MemSize', 'Module', 'Name', 'OutList', 'OutListRecursive', 'PropertiesList', 'Proxy', 'State', 'TypeId', 'ViewObject', '__class__', 
 ...
 'setEditorMode', 'setExpression', 'supportedProperties', 'touch']


There's a lot of attributes there! That's because we're accessing the native FreeCAD FeaturePyton object that we created in the first line of our create() method.


Remember the Proxy property we added in our class __init__() method? Did you see it in the attribute list?


Let's try looking at it's attributes:

>>> dir(mybox.Proxy)
['Object', 'Type', '__class__', '__delattr__', '__dict__', '__dir__', 
 ...
 '__str__', '__subclasshook__', '__weakref__']

Among the list of protected methods and properties, we see our Object and Type properties!


Call the Type property and look at the result:

>>> mybox.Proxy.Type
'box'


Sure enough, it returns the value we assigned, so we know we're accessing the custom class itself through the FeaturePython object. likewise, if you call:

>>> mybox.Proxy.Object


You'll get back the FeaturePython Object, the same thing that mybox refers to.


That was fun! But now let's see if we can make our class a little more interesting... and maybe more useful.