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

Scale velocities according to duration

Browse Source
This commit is contained in:
Steven Robertson 2012-04-22 15:46:44 -07:00
parent e973dcf9d7
commit 233f9f003e
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
cuburn/genome/use.py
View File

@ -100,38 +100,34 @@ class Wrapper(object):
class RefWrapper(Wrapper): class RefWrapper(Wrapper):
""" """
Wrapper that handles RefScalars, as with profile objects. Wrapper that handles RefScalars, as with profile objects.
Must provide an 'other' object as a kwarg to __init__.
""" """
# Turns out (somewhat intentionally) that every spline parameter used on
# the host has a matching parameter in the profile, so this
def __init__(self, val, other, spec=None, path=()):
super(RefWrapper, self).__init__(val, spec, path)
self.other = other
def wrap_dict(self, path, spec, val):
return type(self)(val or {}, self.other, spec, path)
def wrap_refscalar(self, path, spec, val): def wrap_refscalar(self, path, spec, val):
spev = self.other spev = self._params['other']
for part in spec.ref.split('.'): for part in spec.ref.split('.'):
spev = spev[part] spev = spev[part]
spev *= val if val is not None else spec.default spev *= val if val is not None else spec.default
return spev return spev
class SplineWrapper(Wrapper): class SplineWrapper(Wrapper):
"""
Wrapper that handles splines. Must provide a 'scale' object
(normally, equal to duration) as a kwarg to __init__.
"""
def wrap_spline(self, path, spec, val): def wrap_spline(self, path, spec, val):
return SplineEval(val if val is not None else spec.default, return SplineEval(val if val is not None else spec.default,
spec.interp) self._params['scale'], spec.interp)
class SplineEval(object): class SplineEval(object):
_mat = np.matrix([[1.,-2, 1, 0], [2,-3, 0, 1], _mat = np.matrix([[1.,-2, 1, 0], [2,-3, 0, 1],
[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, interp='linear'): def __init__(self, knots, scale, interp='linear'):
self.knots, self.interp = self.normalize(knots), interp self.knots, self.interp = self.normalize(knots, scale), interp
@staticmethod @staticmethod
def normalize(knots): def normalize(knots, scale):
if isinstance(knots, (int, float)): if isinstance(knots, (int, float)):
v0, v1 = 0, 0 v0, v1 = 0, 0
knots = [(0, knots), (1, knots)] knots = [(0, knots), (1, knots)]
@ -143,6 +139,8 @@ class SplineEval(object):
else: else:
p0, v0, p1, v1 = knots[:4] p0, v0, p1, v1 = knots[:4]
knots = [(0, p0), (1, p1)] + zip(knots[4::2], knots[5::2]) knots = [(0, p0), (1, p1)] + zip(knots[4::2], knots[5::2])
v0 *= scale
v1 *= scale
knots = sorted(knots) knots = sorted(knots)
@ -205,7 +203,9 @@ class SplineEval(object):
def wrap_genome(prof, gnm): def wrap_genome(prof, gnm):
# It's not obvious that this is what needs to happen, so we wrap. The # It's not obvious that this is what needs to happen, so we wrap. The
# timing is simplistic, and may get expanded or moved later. # timing is simplistic, and may get expanded or moved later.
gprof = RefWrapper(prof, SplineWrapper(gnm), toplevels['profile']) scale = gnm.get('time', {}).get('duration', 1)
gprof = RefWrapper(prof, toplevels['profile'],
other=SplineWrapper(gnm, scale=scale))
nframes = round(gprof.fps * gprof.duration) nframes = round(gprof.fps * gprof.duration)
times = np.linspace(0, 1, nframes + 1) times = np.linspace(0, 1, nframes + 1)
times = times[:-1] + 0.5 * (times[1] - times[0]) times = times[:-1] + 0.5 * (times[1] - times[0])