timer.py

# Imports, sorted alphabetically.

# Python packages
# Nothing for now...

# Third-party packages
# Nothing for now...

# Modules from this project
# Nothing for now...

import threading
import time

__all__ = (
    'TimerTask', 'Timer',
)


class TimerTask:
    def __init__(self, ticks, callback, speed):
        self.ticks = self.expire = ticks
        self.callback = callback
        self.speed = speed

    def progress(self):
        return self.ticks / self.expire


# Timer used in furnace and redstone circuits
class Timer(threading.Thread):
    def __init__(self, interval, name=None):
        super(Timer, self).__init__(name=name)
        self.queue = [None]
        self.interval = interval
        self._stop = threading.Event()

    def add_task(self, ticks, callback, speed=1):
        if ticks == 0 and callback is not None:
            callback()

        task = TimerTask(ticks, callback, speed)
        for index, _task in enumerate(self.queue):
            if _task is None:
                self.queue[index] = task
                return index

        self.queue.append(task)
        return len(self.queue) - 1

    def remove_task(self, index):
        if index >= len(self.queue):
            return False

        self.queue[index] = None
        return True

    def run(self):
        while True:
            time.sleep(self.interval)
            if self._stop.is_set(): return

            for index, _task in enumerate(self.queue):
                if _task is None:
                    continue
                self.queue[index].ticks -= self.interval * self.queue[index].speed
                if self.queue[index].ticks <= 0:
                    self.queue[index].callback()
                    self.queue[index] = None

    def progress(self, index):
        if index >= len(self.queue):
            return 0

        if self.queue[index] is not None:
            return self.queue[index].progress()
        else:
            return 0

    def stop(self):
        self._stop.set()