Initial implementation of lattice base object

Without markers yet. Just to get some functional stuff, to make testing
of markers possible.
This commit is contained in:
DeepSOIC 2015-10-24 00:41:43 +03:00
parent 7effde8381
commit e74072d1c9
2 changed files with 120 additions and 0 deletions

108
latticeBaseFeature.py Normal file
View File

@ -0,0 +1,108 @@
#***************************************************************************
#* *
#* Copyright (c) 2015 - Victor Titov (DeepSOIC) *
#* <vv.titov@gmail.com> *
#* *
#* This program is free software; you can redistribute it and/or modify *
#* it under the terms of the GNU Lesser General Public License (LGPL) *
#* as published by the Free Software Foundation; either version 2 of *
#* the License, or (at your option) any later version. *
#* for detail see the LICENCE text file. *
#* *
#* This program is distributed in the hope that it will be useful, *
#* but WITHOUT ANY WARRANTY; without even the implied warranty of *
#* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
#* GNU Library General Public License for more details. *
#* *
#* You should have received a copy of the GNU Library General Public *
#* License along with this program; if not, write to the Free Software *
#* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
#* USA *
#* *
#***************************************************************************
__title__="Base feature module for lattice object of lattice workbench for FreeCAD"
__author__ = "DeepSOIC"
__url__ = ""
import Part
from pivy import coin
from latticeCommon import *
import latticeCompoundExplorer as LCE
def makeLatticeFeature(name, AppClass, icon, ViewClass = None):
'''makeLatticeFeature(name, AppClass, ViewClass = None): makes a document object for a LatticeFeature-derived object.'''
obj = FreeCAD.ActiveDocument.addObject("Part::FeaturePython",name)
AppClass(obj)
if ViewClass:
ViewClass(obj.ViewObject)
else:
vp = ViewProviderLatticeFeature(obj.ViewObject)
vp.icon = icon
return obj
class LatticeFeature():
"Base object for lattice objects (arrays of placements)"
def __init__(self,obj):
self.Type = "latticeFeature"
prop = "NumElements"
obj.addProperty("App::PropertyInteger",prop,"Info","Number of placements in the array")
obj.setEditorMode(prop, 1) # set read-only
self.derivedInit(obj)
obj.Proxy = self
def derivedInit(self, obj):
'''for overriding by derived classes'''
pass
def execute(self,obj):
derivedExecute(self, obj)
obj.NumElements = LCE.CalculateNumberOfLeaves(obj.Shape)
return
def derivedExecute():
'''For overriding by derived class'''
pass
class ViewProviderLatticeFeature:
"A View Provider for base lattice object"
def __init__(self,vobj):
vobj.Proxy = self
# vobj.DisplayMode = "Markers"
vobj.PointSize = 4
vobj.PointColor = (1.0, 0.7019608020782471, 0.0, 0.0) #orange
def getIcon(self):
if hasattr(self, "icon"):
if self.icon:
return getIconPath(self.icon)
return getIconPath("Lattice.svg")
def attach(self, vobj):
self.ViewObject = vobj
self.Object = vobj.Object
def setEdit(self,vobj,mode):
return False
def unsetEdit(self,vobj,mode):
return
def __getstate__(self):
return None
def __setstate__(self,state):
return None

View File

@ -130,3 +130,15 @@ class CompoundExplorer:
assert(msg != 0) assert(msg != 0)
self.lastMsg = msg self.lastMsg = msg
return (sh, msg, self) return (sh, msg, self)
def CalculateNumberOfLeaves(compound):
'''CalculateNumberOfLeaves(compound): calculates the number of non-compound shapes (leaves) in the compound tree. Slow; good candidate for transferring into C++.'''
if compound.ShapeType != 'Compound':
return 1
else:
children = compound.childShapes(False,False)
cnt = 0
for ch in children:
cnt += CalculateNumberOfLeaves(ch)
return cnt