suzanne.soy/cadquery-freecad-module
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updated CadQuery library.

Browse Source
This commit is contained in:
Jeremy Mack Wright 2015-10-12 17:08:00 -04:00
parent 43c593e76f
commit 3991aee2d1
1 changed files with 19 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
CadQuery/Libs/cadquery/CQ.py
View File

@ -17,7 +17,8 @@
License along with this library; If not, see <http://www.gnu.org/licenses/> License along with this library; If not, see <http://www.gnu.org/licenses/>
""" """
import time, math import time
import math
from cadquery import * from cadquery import *
from cadquery import selectors from cadquery import selectors
from cadquery import exporters from cadquery import exporters
@ -38,6 +39,7 @@ class CQContext(object):
self.firstPoint = None self.firstPoint = None
self.tolerance = 0.0001 # user specified tolerance self.tolerance = 0.0001 # user specified tolerance
class CQ(object): class CQ(object):
""" """
Provides enhanced functionality for a wrapped CAD primitive. Provides enhanced functionality for a wrapped CAD primitive.
@ -94,9 +96,9 @@ class CQ(object):
all = {} all = {}
for o in self.objects: for o in self.objects:
#tricky-- if an object is a compound of solids, # tricky-- if an object is a compound of solids,
#do not return all of the solids underneath-- typically # do not return all of the solids underneath-- typically
#then we'll keep joining to ourself # then we'll keep joining to ourself
if propName == 'Solids' and isinstance(o, Solid) and o.ShapeType() == 'Compound': if propName == 'Solids' and isinstance(o, Solid) and o.ShapeType() == 'Compound':
for i in getattr(o, 'Compounds')(): for i in getattr(o, 'Compounds')():
all[i.hashCode()] = i all[i.hashCode()] = i
@ -139,11 +141,11 @@ class CQ(object):
bottom = solid.cut(topCutBox) bottom = solid.cut(topCutBox)
if keepTop and keepBottom: if keepTop and keepBottom:
#put both on the stack, leave original unchanged # Put both on the stack, leave original unchanged.
return self.newObject([top, bottom]) return self.newObject([top, bottom])
else: else:
# put the one we are keeping on the stack, and also update the context solid # Put the one we are keeping on the stack, and also update the
#to the one we kept # context solidto the one we kept.
if keepTop: if keepTop:
solid.wrapped = top.wrapped solid.wrapped = top.wrapped
return self.newObject([top]) return self.newObject([top])
@ -467,7 +469,7 @@ class CQ(object):
toReturn = self._collectProperty(objType) toReturn = self._collectProperty(objType)
if selector is not None: if selector is not None:
if type(selector) == str: if isinstance(selector, str) or isinstance(selector, unicode):
selectorObj = selectors.StringSyntaxSelector(selector) selectorObj = selectors.StringSyntaxSelector(selector)
else: else:
selectorObj = selector selectorObj = selector
@ -818,7 +820,7 @@ class CQ(object):
solid.wrapped = s.wrapped solid.wrapped = s.wrapped
return self.newObject([s]) return self.newObject([s])
def chamfer(self, length, length2 = None): def chamfer(self, length, length2=None):
""" """
Chamfers a solid on the selected edges. Chamfers a solid on the selected edges.
@ -857,6 +859,7 @@ class CQ(object):
solid.wrapped = s.wrapped solid.wrapped = s.wrapped
return self.newObject([s]) return self.newObject([s])
class Workplane(CQ): class Workplane(CQ):
""" """
Defines a coordinate system in space, in which 2-d coordinates can be used. Defines a coordinate system in space, in which 2-d coordinates can be used.
@ -909,13 +912,14 @@ class Workplane(CQ):
if inPlane.__class__.__name__ == 'Plane': if inPlane.__class__.__name__ == 'Plane':
tmpPlane = inPlane tmpPlane = inPlane
elif type(inPlane) == str: elif isinstance(inPlane, str) or isinstance(inPlane, unicode):
tmpPlane = Plane.named(inPlane, origin) tmpPlane = Plane.named(inPlane, origin)
else: else:
tmpPlane = None tmpPlane = None
if tmpPlane is None: if tmpPlane is None:
raise ValueError(" Provided value %s is not a valid work plane." % str(inPlane)) raise ValueError(
'Provided value {} is not a valid work plane'.format(inPlane))
self.obj = obj self.obj = obj
self.plane = tmpPlane self.plane = tmpPlane
@ -1118,7 +1122,7 @@ class Workplane(CQ):
return self.newObject([p]) return self.newObject([p])
#line a specified incremental amount from current point # line a specified incremental amount from current point
def line(self, xDist, yDist, forConstruction=False): def line(self, xDist, yDist, forConstruction=False):
""" """
Make a line from the current point to the provided point, using Make a line from the current point to the provided point, using
@ -2077,7 +2081,7 @@ class Workplane(CQ):
elif type(toUnion) == Solid: elif type(toUnion) == Solid:
newS = toUnion newS = toUnion
else: else:
raise ValueError("Cannot union Type '%s' " % str(type(toUnion))) raise ValueError("Cannot union type '{}'".format(type(toUnion)))
#now combine with existing solid, if there is one #now combine with existing solid, if there is one
# look for parents to cut from # look for parents to cut from
@ -2110,14 +2114,14 @@ class Workplane(CQ):
solidRef = self.findSolid(searchStack=True, searchParents=True) solidRef = self.findSolid(searchStack=True, searchParents=True)
if solidRef is None: if solidRef is None:
raise ValueError("Cannot find solid to cut from!!!") raise ValueError("Cannot find solid to cut from")
solidToCut = None solidToCut = None
if type(toCut) == CQ or type(toCut) == Workplane: if type(toCut) == CQ or type(toCut) == Workplane:
solidToCut = toCut.val() solidToCut = toCut.val()
elif type(toCut) == Solid: elif type(toCut) == Solid:
solidToCut = toCut solidToCut = toCut
else: else:
raise ValueError("Cannot cut Type '%s' " % str(type(toCut))) raise ValueError("Cannot cut type '{}'".format(type(toCut)))
newS = solidRef.cut(solidToCut) newS = solidRef.cut(solidToCut)