8234b62ab7
Cython is used to compile some hot paths into native Python extensions. These hot paths were identified through running ufocompile with the hotshot profiler and then converting file by file to Cython, starting with the "hottest" paths and continuing until returns were deminishing. This means that only a few Python files were converted to Cython. Closes #23 Closes #20 (really this time)
137 lines
5.2 KiB
Cython
137 lines
5.2 KiB
Cython
from __future__ import print_function, division, absolute_import
|
|
from .flatten import InputContour, OutputContour
|
|
from .exceptions import (
|
|
InvalidSubjectContourError, InvalidClippingContourError, ExecutionError)
|
|
import pyclipper
|
|
|
|
|
|
"""
|
|
General Suggestions:
|
|
- Contours should only be sent here if they actually overlap.
|
|
This can be checked easily using contour bounds.
|
|
- Only perform operations on closed contours.
|
|
- contours must have an on curve point
|
|
- some kind of a log
|
|
"""
|
|
|
|
|
|
_operationMap = {
|
|
"union": pyclipper.CT_UNION,
|
|
"intersection": pyclipper.CT_INTERSECTION,
|
|
"difference": pyclipper.CT_DIFFERENCE,
|
|
"xor": pyclipper.CT_XOR,
|
|
}
|
|
|
|
_fillTypeMap = {
|
|
"evenOdd": pyclipper.PFT_EVENODD,
|
|
"nonZero": pyclipper.PFT_NONZERO,
|
|
# we keep the misspelling for compatibility with earlier versions
|
|
"noneZero": pyclipper.PFT_NONZERO,
|
|
}
|
|
|
|
|
|
def clipExecute(subjectContours, clipContours, operation, subjectFillType="nonZero",
|
|
clipFillType="nonZero"):
|
|
pc = pyclipper.Pyclipper()
|
|
|
|
for i, subjectContour in enumerate(subjectContours):
|
|
# ignore paths with no area
|
|
if pyclipper.Area(subjectContour):
|
|
try:
|
|
pc.AddPath(subjectContour, pyclipper.PT_SUBJECT)
|
|
except pyclipper.ClipperException:
|
|
raise InvalidSubjectContourError("contour %d is invalid for clipping" % i)
|
|
for j, clipContour in enumerate(clipContours):
|
|
# ignore paths with no area
|
|
if pyclipper.Area(clipContour):
|
|
try:
|
|
pc.AddPath(clipContour, pyclipper.PT_CLIP)
|
|
except pyclipper.ClipperException:
|
|
raise InvalidClippingContourError("contour %d is invalid for clipping" % j)
|
|
|
|
try:
|
|
solution = pc.Execute(_operationMap[operation],
|
|
_fillTypeMap[subjectFillType],
|
|
_fillTypeMap[clipFillType])
|
|
except pyclipper.ClipperException as exc:
|
|
raise ExecutionError(exc)
|
|
|
|
return [[tuple(p) for p in path] for path in solution]
|
|
|
|
|
|
def _performOperation(operation, subjectContours, clipContours, outPen):
|
|
# prep the contours
|
|
subjectInputContours = [InputContour(contour) for contour in subjectContours if contour and len(contour) > 1]
|
|
clipInputContours = [InputContour(contour) for contour in clipContours if contour and len(contour) > 1]
|
|
inputContours = subjectInputContours + clipInputContours
|
|
|
|
resultContours = clipExecute([subjectInputContour.originalFlat for subjectInputContour in subjectInputContours],
|
|
[clipInputContour.originalFlat for clipInputContour in clipInputContours],
|
|
operation, subjectFillType="nonZero", clipFillType="nonZero")
|
|
# convert to output contours
|
|
outputContours = [OutputContour(contour) for contour in resultContours]
|
|
# re-curve entire contour
|
|
for inputContour in inputContours:
|
|
for outputContour in outputContours:
|
|
if outputContour.final:
|
|
continue
|
|
if outputContour.reCurveFromEntireInputContour(inputContour):
|
|
# the input is expired if a match was made,
|
|
# so stop passing it to the outputs
|
|
break
|
|
# re-curve segments
|
|
for inputContour in inputContours:
|
|
# skip contours that were comppletely used in the previous step
|
|
if inputContour.used:
|
|
continue
|
|
# XXX this could be expensive if an input becomes completely used
|
|
# it doesn't stop from being passed to the output
|
|
for outputContour in outputContours:
|
|
outputContour.reCurveFromInputContourSegments(inputContour)
|
|
# curve fit
|
|
for outputContour in outputContours:
|
|
outputContour.reCurveSubSegments(inputContours)
|
|
# output the results
|
|
for outputContour in outputContours:
|
|
outputContour.drawPoints(outPen)
|
|
return outputContours
|
|
|
|
|
|
class BooleanOperationManager(object):
|
|
|
|
@staticmethod
|
|
def union(contours, outPen):
|
|
return _performOperation("union", contours, [], outPen)
|
|
|
|
@staticmethod
|
|
def difference(subjectContours, clipContours, outPen):
|
|
return _performOperation("difference", subjectContours, clipContours, outPen)
|
|
|
|
@staticmethod
|
|
def intersection(subjectContours, clipContours, outPen):
|
|
return _performOperation("intersection", subjectContours, clipContours, outPen)
|
|
|
|
@staticmethod
|
|
def xor(subjectContours, clipContours, outPen):
|
|
return _performOperation("xor", subjectContours, clipContours, outPen)
|
|
|
|
@staticmethod
|
|
def getIntersections(contours):
|
|
from .flatten import _scalePoints, inverseClipperScale
|
|
# prep the contours
|
|
inputContours = [InputContour(contour) for contour in contours if contour and len(contour) > 1]
|
|
|
|
inputFlatPoints = set()
|
|
for contour in inputContours:
|
|
inputFlatPoints.update(contour.originalFlat)
|
|
|
|
resultContours = clipExecute(
|
|
[inputContour.originalFlat for inputContour in inputContours], [],
|
|
"union", subjectFillType="nonZero", clipFillType="nonZero")
|
|
|
|
resultFlatPoints = set()
|
|
for contour in resultContours:
|
|
resultFlatPoints.update(contour)
|
|
|
|
intersections = resultFlatPoints - inputFlatPoints
|
|
return _scalePoints(intersections, inverseClipperScale)
|