mirror-github-bspeice/cuburn
1
0
Fork 0
mirror of https://github.com/stevenrobertson/cuburn.git synced 2025-11-03 18:00:55 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add velocity-matching to SplEval

Browse Source
This commit is contained in:
Steven Robertson
2011-12-19 16:37:55 -05:00
parent 39e8ad2447
commit 644ca7f62d
1 changed files with 8 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
cuburn/genome.py
View File

@ -13,8 +13,7 @@ class SplEval(object):
[1,-1, 0, 0], [-2, 3, 0, 0]]) [1,-1, 0, 0], [-2, 3, 0, 0]])
_deriv = np.matrix(np.diag([3,2,1], 1)) _deriv = np.matrix(np.diag([3,2,1], 1))
def __init__(self, knots): def __init__(self, knots, v0=None, v1=None):
# If a single value is passed, normalize to a constant set of knots
if isinstance(knots, (int, float)): if isinstance(knots, (int, float)):
knots = [-0.1, knots, 0.0, knots, 1.0, knots, 1.1, knots] knots = [-0.1, knots, 0.0, knots, 1.0, knots, 1.1, knots]
elif not np.all(np.diff(np.float32(np.asarray(knots))[::2]) > 0): elif not np.all(np.diff(np.float32(np.asarray(knots))[::2]) > 0):
@ -22,16 +21,15 @@ class SplEval(object):
"nextafterf()-spaced times to anchor tangents.)") "nextafterf()-spaced times to anchor tangents.)")
# If stabilizing knots are missing before or after the edges of the # If stabilizing knots are missing before or after the edges of the
# [0,1] interval, add them. We choose knots that preserve the first # [0,1] interval, add them.
# differential, which is probably what is expected when two knots are
# present but may be less accurate for three. Of course, one can just
# add stabilizing knots to the genome to change this.
if knots[0] >= 0: if knots[0] >= 0:
m = (knots[3] - knots[1]) / (knots[2] - knots[0]) if v0 is None:
knots = [-0.1, knots[1] + -0.1 * m] + knots v0 = (knots[3] - knots[1]) / (knots[2] - knots[0])
knots = [-2, knots[3] + (knots[2] - 2) * v0] + knots
if knots[-2] <= 1: if knots[-2] <= 1:
m = (knots[-1] - knots[-3]) / (knots[-2] - knots[-4]) if v1 is None:
knots.extend([1.1, knots[-1] + 0.1 * m]) v1 = (knots[-1] - knots[-3]) / (knots[-2] - knots[-4])
knots.extend([3, knots[-3] + (3 - knots[-4]) * v1])
self.knots = np.zeros((2, len(knots)/2)) self.knots = np.zeros((2, len(knots)/2))
self.knots.T.flat[:] = knots self.knots.T.flat[:] = knots