Mercurial > touhou
view pytouhou/utils/interpolator.pyx @ 357:26f78fc7acea
Fix Koakuma
author | Thibaut Girka <thib@sitedethib.com> |
---|---|
date | Sun, 08 Jul 2012 01:20:47 +0200 (2012-07-07) |
parents | fab7ad2f0d8b |
children | d778db08190f |
line wrap: on
line source
# -*- encoding: utf-8 -*- ## ## Copyright (C) 2011 Thibaut Girka <thib@sitedethib.com> ## ## This program is free software; you can redistribute it and/or modify ## it under the terms of the GNU General Public License as published ## by the Free Software Foundation; version 3 only. ## ## This program is distributed in the hope that it will be useful, ## but WITHOUT ANY WARRANTY; without even the implied warranty of ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the ## GNU General Public License for more details. ## class Interpolator(object): __slots__ = ('values', 'start_values', 'end_values', 'start_frame', 'end_frame', '_frame', '_formula') def __init__(self, values=(), start_frame=0, end_values=(), end_frame=0, formula=None): self.values = tuple(values) self.start_values = tuple(values) self.end_values = tuple(end_values) self.start_frame = start_frame self.end_frame = end_frame self._frame = 0 self._formula = formula or (lambda x: x) def __nonzero__(self): return self._frame < self.end_frame def set_interpolation_start(self, frame, values): self.start_values = tuple(values) self.start_frame = frame def set_interpolation_end(self, frame, values): self.end_values = tuple(values) self.end_frame = frame def set_interpolation_end_frame(self, end_frame): self.end_frame = end_frame def set_interpolation_end_values(self, values): self.end_values = tuple(values) def update(self, frame): self._frame = frame if frame >= self.end_frame - 1: #XXX: skip the last interpolation step # This bug is replicated from the original game self.values = self.end_values self.start_values = self.end_values self.start_frame = frame else: coeff = self._formula(float(frame - self.start_frame) / float(self.end_frame - self.start_frame)) self.values = [start_value + coeff * (end_value - start_value) for (start_value, end_value) in zip(self.start_values, self.end_values)]