ByoungSooPark/cadquery
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ProjectedOrigin with origin point option

Browse Source 
Removed ProjectedGlobalOrigin option for centerOption for the workplane cunstructor. Added origin keyword parameter that defaults to (0,0,0).
This commit is contained in:
Michael Greminger
2019-06-02 16:16:27 -05:00
parent ae65dc3579
commit 0179b674e0
2 changed files with 18 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
cadquery/cq.py
View File

@ -261,16 +261,19 @@ class CQ(object):
return self.objects[0].wrapped
def workplane(self, offset=0.0, invert=False, centerOption='CenterOfMass'):
def workplane(self, offset=0.0, invert=False, centerOption='CenterOfMass',
origin=(0, 0, 0)):
"""
Creates a new 2-D workplane, located relative to the first face on the stack.
:param offset: offset for the work plane in the Z direction. Default
:param invert: invert the Z direction from that of the face.
:param centerOption: how local origin of workplane is determined
:param centerOption: how local origin of workplane is determined.
:param origin: origin for plane center, requires 'ProjectedOrigin' centerOption.
:type offset: float or None=0.0
:type invert: boolean or None=False
:type centerOption: string or None='CenterOfMass'
:type origin: tuple
:rtype: Workplane object ( which is a subclass of CQ )
The first element on the stack must be a face, a set of
@ -285,11 +288,10 @@ class CQ(object):
face/faces, if a face/faces was selected. If a vertex was
selected, the origin will be at the vertex, and located
on the face. The centerOption paramter sets how the center is defined.
Options are 'CenterOfMass', 'CenterOfBoundBox', 'ProjectedOrigin', or
'ProjectedGlobalOrigin'. 'CenterOfMass' and 'CenterOfBoundBox' are
in relation to the selected face/faces. 'ProjectedOrigin' uses the
current origin that may have been updated by calls to transformed or
center. 'ProjectedGlobalOrigin' projects the global (0,0,0) origin.
Options are 'CenterOfMass', 'CenterOfBoundBox', or 'ProjectedOrigin'.
'CenterOfMass' and 'CenterOfBoundBox' are in relation to the selected
face/faces. 'ProjectedOrigin' uses the supplied origin, which defaults
to (0, 0, 0).
* The Z direction will be normal to the plane of the face,computed
at the center point.
* The X direction will be parallel to the x-y plane. If the workplane is parallel to
@ -338,6 +340,9 @@ class CQ(object):
xd = Vector(1, 0, 0)
return xd
if centerOption not in {'CenterOfMass', 'ProjectedOrigin', 'CenterOfBoundBox'}:
raise ValueError('Undefined centerOption value provided.')
if len(self.objects) > 1:
# are all objects 'PLANE'?
if not all(o.geomType() in ('PLANE', 'CIRCLE') for o in self.objects):
@ -348,12 +353,10 @@ class CQ(object):
if not all(_isCoPlanar(self.objects[0], f) for f in self.objects[1:]):
raise ValueError("Selected faces must be co-planar.")
if centerOption in {'CenterOfMass', 'ProjectedOrigin', 'ProjectedGlobalOrigin'}:
if centerOption in {'CenterOfMass', 'ProjectedOrigin'}:
center = Shape.CombinedCenter(self.objects)
elif centerOption == 'CenterOfBoundBox':
center = Shape.CombinedCenterOfBoundBox(self.objects)
else:
raise ValueError('Undefined value passed to centerOption')
normal = self.objects[0].normalAt()
xDir = _computeXdir(normal)
@ -362,7 +365,7 @@ class CQ(object):
obj = self.objects[0]
if isinstance(obj, Face):
if centerOption in {'CenterOfMass', 'ProjectedOrigin', 'ProjectedGlobalOrigin'}:
if centerOption in {'CenterOfMass', 'ProjectedOrigin'}:
center = obj.Center()
elif centerOption == 'CenterOfBoundBox':
center = obj.CenterOfBoundBox()
@ -370,7 +373,7 @@ class CQ(object):
xDir = _computeXdir(normal)
else:
if hasattr(obj, 'Center'):
if centerOption in {'CenterOfMass', 'ProjectedOrigin', 'ProjectedGlobalOrigin'}:
if centerOption in {'CenterOfMass', 'ProjectedOrigin'}:
center = obj.Center()
elif centerOption == 'CenterOfBoundBox':
center = obj.CenterOfBoundBox()
@ -382,9 +385,7 @@ class CQ(object):
# update center to projected origin if desired
if centerOption == 'ProjectedOrigin':
center = self.plane.origin.projectToPlane(Plane(center, xDir, normal))
elif centerOption == 'ProjectedGlobalOrigin':
center = Vector(0,0,0).projectToPlane(Plane(center, xDir, normal))
center = Vector(origin).projectToPlane(Plane(center, xDir, normal))
# invert if requested
if invert:

12
tests/TestCadQuery.py
View File

@ -1924,17 +1924,9 @@ class TestCadQuery(BaseTest):
.plane.origin.toTuple()
self.assertTupleAlmostEquals(origin, (45.0, 30.0, 10.0), decimal_places)
# test case where plane origin is shifted with center call
r = r.faces(">Z").workplane(centerOption='ProjectedOrigin').center(30,0) \
.hole(90)
origin = r.faces(">Z").workplane(centerOption='ProjectedOrigin') \
origin = r.faces(">Z").workplane(centerOption='ProjectedOrigin',origin=(30,10,20)) \
.plane.origin.toTuple()
self.assertTupleAlmostEquals(origin, (30.0, 0.0, 10.0), decimal_places)
origin = r.faces(">Z").workplane(centerOption='ProjectedGlobalOrigin') \
.plane.origin.toTuple()
self.assertTupleAlmostEquals(origin, (0.0, 0.0, 10.0), decimal_places)
self.assertTupleAlmostEquals(origin, (30.0, 10.0, 10.0), decimal_places)
with self.assertRaises(ValueError):
origin = r.faces(">Z").workplane(centerOption='undefined')