SubLink: remove dead properties if shape type changes

This commit is contained in:
DeepSOIC 2016-01-08 18:36:24 +03:00
parent 3365f06155
commit 045b7133f1

View File

@ -107,69 +107,79 @@ class LatticeSubLink:
App.Console.PrintError(selfobj.Name+": analyzeShape: "+err.message+"\n") App.Console.PrintError(selfobj.Name+": analyzeShape: "+err.message+"\n")
def analyzeShape(self,selfobj): def analyzeShape(self,selfobj):
sh = selfobj.Shape
self.assignProp(selfobj,"App::PropertyString","ShapeType",sh.ShapeType) self.updatedProperties = set()
if sh.ShapeType == "Compound": try:
self.assignProp(selfobj,"App::PropertyInteger","NumChildren",len(sh.childShapes(False,False))) sh = selfobj.Shape
elif sh.ShapeType == "Face":
self.assignProp(selfobj,"App::PropertyFloat","Area",sh.Area)
typelist = ["BSplineSurface", self.assignProp(selfobj,"App::PropertyString","ShapeType",sh.ShapeType)
"BezierSurface", if sh.ShapeType == "Compound":
"Cone", self.assignProp(selfobj,"App::PropertyInteger","NumChildren",len(sh.childShapes(False,False)))
"Cylinder", elif sh.ShapeType == "Face":
"OffsetSurface", self.assignProp(selfobj,"App::PropertyFloat","Area",sh.Area)
"Plane",
"PlateSurface",
"RectangularTrimmedSurface",
"Sphere",
"SurfaceOfExtrusion",
"SurfaceOfRevolution",
"Toroid",
]
surf = sh.Surface
for typename in typelist:
if type(surf) is getattr(Part, typename):
break
typename = None
self.assignProp(selfobj,"App::PropertyString","FaceType",typename)
self.transplant_all_attributes(selfobj,surf,"Face") typelist = ["BSplineSurface",
elif sh.ShapeType == "Edge": "BezierSurface",
self.assignProp(selfobj,"App::PropertyFloat","Length",sh.Length) "Cone",
"Cylinder",
"OffsetSurface",
"Plane",
"PlateSurface",
"RectangularTrimmedSurface",
"Sphere",
"SurfaceOfExtrusion",
"SurfaceOfRevolution",
"Toroid",
]
surf = sh.Surface
for typename in typelist:
if type(surf) is getattr(Part, typename):
break
typename = None
self.assignProp(selfobj,"App::PropertyString","FaceType",typename)
typelist = ["Arc", self.transplant_all_attributes(selfobj,surf,"Face")
"ArcOfCircle", elif sh.ShapeType == "Edge":
"ArcOfEllipse", self.assignProp(selfobj,"App::PropertyFloat","Length",sh.Length)
"ArcOfHyperbola",
"ArcOfParabola",
"BSplineCurve",
"BezierCurve",
"Circle",
"Ellipse",
"Hyperbola",
"Line",
"OffsetCurve",
"Parabola",
]
crv = sh.Curve
for typename in typelist:
if type(crv) is getattr(Part, typename):
break
typename = None
self.assignProp(selfobj,"App::PropertyString","EdgeType",typename)
self.transplant_all_attributes(selfobj,crv,"Edge") typelist = ["Arc",
"ArcOfCircle",
"ArcOfEllipse",
"ArcOfHyperbola",
"ArcOfParabola",
"BSplineCurve",
"BezierCurve",
"Circle",
"Ellipse",
"Hyperbola",
"Line",
"OffsetCurve",
"Parabola",
]
crv = sh.Curve
for typename in typelist:
if type(crv) is getattr(Part, typename):
break
typename = None
self.assignProp(selfobj,"App::PropertyString","EdgeType",typename)
elif sh.ShapeType == "Vertex": self.transplant_all_attributes(selfobj,crv,"Edge")
self.assignProp(selfobj,"App::PropertyVector","Position",sh.Point)
elif sh.ShapeType == "Vertex":
self.assignProp(selfobj,"App::PropertyVector","Position",sh.Point)
finally:
#remove properties that haven't been updated
for propname in selfobj.PropertiesList:
if selfobj.getGroupOfProperty(propname) == "info":
if not (propname in self.updatedProperties):
selfobj.removeProperty(propname)
def assignProp(self, selfobj, proptype, propname, propvalue): def assignProp(self, selfobj, proptype, propname, propvalue):
if not hasattr(selfobj,propname): if not hasattr(selfobj,propname):
selfobj.addProperty(proptype, propname,"info") selfobj.addProperty(proptype, propname,"info")
selfobj.setEditorMode(propname,1) #set read-only selfobj.setEditorMode(propname,1) #set read-only
setattr(selfobj,propname,propvalue) setattr(selfobj,propname,propvalue)
self.updatedProperties.add(propname)
def transplant_all_attributes(self, selfobj, source, prefix): def transplant_all_attributes(self, selfobj, source, prefix):
for attrname in dir(source): for attrname in dir(source):