Merge branch 'master' into NthSelector

This commit is contained in:
Adam Urbańczyk
2021-01-03 15:48:02 +01:00
committed by GitHub
8 changed files with 120 additions and 17 deletions

View File

@ -68,4 +68,4 @@ __all__ = [
"plugins",
]
__version__ = "2.1dev"
__version__ = "2.1RC1"

View File

@ -21,15 +21,26 @@ AssemblyObjects = Union[Shape, Workplane, None]
ConstraintKinds = Literal["Plane", "Point", "Axis"]
ExportLiterals = Literal["STEP", "XML"]
PATH_DELIM = "/"
# enitity selector grammar definiiton
def _define_grammar():
from pyparsing import Literal as Literal, Word, Optional, alphas, alphanums
from pyparsing import (
Literal as Literal,
Word,
Optional,
alphas,
alphanums,
delimitedList,
)
Separator = Literal("@").suppress()
TagSeparator = Literal("?").suppress()
Name = Word(alphas, alphanums + "_").setResultsName("name")
Name = delimitedList(
Word(alphas, alphanums + "_"), PATH_DELIM, combine=True
).setResultsName("name")
Tag = Word(alphas, alphanums + "_").setResultsName("tag")
Selector = _selector_grammar.setResultsName("selector")
@ -234,6 +245,11 @@ class Assembly(object):
if isinstance(arg, Assembly):
# enforce unique names
name = kwargs["name"] if kwargs.get("name") else arg.name
if name in self.objects:
raise ValueError("Unique name is required")
subassy = arg._copy()
subassy.loc = kwargs["loc"] if kwargs.get("loc") else arg.loc
@ -242,11 +258,10 @@ class Assembly(object):
subassy.parent = self
self.children.append(subassy)
self.objects[subassy.name] = subassy
self.objects.update(subassy.objects)
self.objects.update(subassy._flatten())
else:
assy = Assembly(arg, **kwargs)
assy = self.__class__(arg, **kwargs)
assy.parent = self
self.add(assy)
@ -454,3 +469,17 @@ class Assembly(object):
yield el
yield (self.name, self)
def _flatten(self, parents=[]):
"""
Generate a dict with all ancestors with keys indicating parent-child relations.
"""
rv = {}
for ch in self.children:
rv.update(ch._flatten(parents=parents + [self.name]))
rv[PATH_DELIM.join(parents + [self.name])] = self
return rv

View File

@ -1746,6 +1746,7 @@ class Workplane(object):
N: int = 400,
start: float = 0,
stop: float = 1,
makeWire: bool = True,
) -> "Workplane":
"""
Create a spline interpolated through the provided points.
@ -1755,6 +1756,7 @@ class Workplane(object):
:param N: number of points for discretization
:param start: starting value of the parameter t
:param stop: final value of the parameter t
:param makeWire: convert the resulting spline edge to a wire
:return: a Workplane object with the current point unchanged
"""
@ -1762,7 +1764,7 @@ class Workplane(object):
diff = stop - start
allPoints = [func(start + diff * t / N) for t in range(N + 1)]
return self.spline(allPoints, includeCurrent=False, makeWire=True)
return self.spline(allPoints, includeCurrent=False, makeWire=makeWire)
def ellipseArc(
self,