# -*- 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):
    def __init__(self, values=()):
        self.values = tuple(values)
        self.start_values = tuple(values)
        self.end_values = tuple(values)
        self.start_frame = 0
        self.end_frame = 0
        self._frame = 0


    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:
            self.values = tuple(self.end_values)
            self.start_values = tuple(self.end_values)
            self.start_frame = frame
        else:
            truc = float(frame - self.start_frame) / float(self.end_frame - self.start_frame)
            self.values = tuple(start_value + truc * (end_value - start_value)
                                for (start_value, end_value) in zip(self.start_values, self.end_values))