kmk_firmware/kmk/kmktime.py

from micropython import const
from supervisor import ticks_ms

_TICKS_PERIOD = const(1 << 29)
_TICKS_MAX = const(_TICKS_PERIOD - 1)
_TICKS_HALFPERIOD = const(_TICKS_PERIOD // 2)


def ticks_diff(new: int, start: int) -> int:
    diff = (new - start) & _TICKS_MAX
    diff = ((diff + _TICKS_HALFPERIOD) & _TICKS_MAX) - _TICKS_HALFPERIOD
    return diff


def ticks_add(ticks: int, delta: int) -> int:
    return (ticks + delta) % _TICKS_PERIOD


def check_deadline(new: int, start: int, ms: int) -> int:
    return ticks_diff(new, start) < ms


class PeriodicTimer:
    def __init__(self, period: int):
        self.period = period
        self.last_tick = ticks_ms()

    def tick(self) -> bool:
        now = ticks_ms()
        if ticks_diff(now, self.last_tick) >= self.period:
            self.last_tick = now
            return True
        else:
            return False