MH3SP/mh/time_utils.py

#! /usr/bin/env python
# -*- coding: utf-8 -*-
# SPDX-FileCopyrightText: Copyright (C) 2021-2022 MH3SP Server Project
# SPDX-License-Identifier: AGPL-3.0-or-later
"""Monster Hunter time utils module."""

import datetime
import time


EPOCH = datetime.datetime(1970, 1, 1)
TICKS_PER_CYCLE = 23040  # 6.4 hours per cycle, 3.2 hours per daytime/nighttime

SECONDS_PER_DAY = 24*60*60
JHEN_EVENT_OFFSET = 14  # Cycle of 14 (real) days
FOG_START = 0
JHEN_START = 1  # 1 (real) day of fog
JHEN_END = 3  # Followed by 2 (real) days of sandstorm


def datetime_to_int(dt):
    return int((dt - EPOCH).total_seconds())


def current_server_time():
    return datetime_to_int(datetime.datetime.now())


def get_jhen_event_times():
    """
    If the first int is less than the gametime at server login:
      If the second int is greater than the gametime at server login, fog
      Otherwise, if the third int is greater than the gametime at server login,
      sandstorm
    FIRST INT: Start of fog (epoch seconds)
    SECOND INT: Start of Jhen event (epoch seconds)
    THIRD INT: End of Jhen event (epoch seconds)
    """
    current_day = int(current_server_time()//SECONDS_PER_DAY)
    day_in_cycle = current_day % JHEN_EVENT_OFFSET
    if day_in_cycle < JHEN_END:  # Current period
        cycle_start = (current_day - day_in_cycle) * SECONDS_PER_DAY
    else:  # Upcoming period
        cycle_start = (current_day - day_in_cycle + JHEN_EVENT_OFFSET) * \
                      SECONDS_PER_DAY
    return (int(cycle_start + FOG_START*SECONDS_PER_DAY),  # fog start
            int(cycle_start + JHEN_START*SECONDS_PER_DAY),  # sandstorm start
            int(cycle_start + JHEN_END*SECONDS_PER_DAY))  # sandstorm end


def current_event_time_slot():
    """
    There are JHEN_EVENT_OFFSET temporal event slots per data event slot.
        Each one activates on a different day, counting up from 0.
    """
    return int(current_server_time()//SECONDS_PER_DAY) % JHEN_EVENT_OFFSET


def is_jhen_active():
    day_in_cycle = int(
        current_server_time() // SECONDS_PER_DAY
    ) % JHEN_EVENT_OFFSET
    return JHEN_START <= day_in_cycle < JHEN_END


class Timer(object):
    def __init__(self):
        self.__start = time.time()

    def elapsed(self):
        """Elapsed seconds since start"""
        return time.time() - self.__start

    def restart(self):
        """Reset timer and start again"""
        self.__start = time.time()