view pytouhou/utils/interpolator.pyx @ 261:2876c267be00

Add heuristics to extract character data from the demo .exe.
author Thibaut Girka <thib@sitedethib.com>
date Mon, 23 Jan 2012 00:03:06 +0100
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)]