Difficult-Rocket/Difficult_Rocket/client/fps/fps_log.py

#  -------------------------------
#  Difficult Rocket
#  Copyright © 2020-2023 by shenjackyuanjie 3695888@qq.com
#  All rights reserved
#  -------------------------------

"""
writen by shenjackyuanjie
mail:   3695888@qq.com
github: @shenjackyuanjie
gitee:  @shenjackyuanjie
"""

import statistics

from typing import Union, List
from decimal import Decimal


class FpsLogger:
    def __init__(self,
                 stable_fps: int = 60,
                 count: int = 700):
        self.stable_fps = stable_fps
        self.count = count
        self._fps = stable_fps
        self.middle_fps = stable_fps
        self.fps_list: List[Union[int, float]] = [stable_fps]
        self.get_fps_list: List[Union[int, float]] = [stable_fps]
        self._max_fps = stable_fps
        self._min_fps = stable_fps

    def update_tick(self,
                    pyglet_fps: float,
                    tick: Decimal):
        if pyglet_fps != 0:
            self.fps_list.append(pyglet_fps)
        elif tick == 0:
            self.fps_list.append(1)
        else:
            self.fps_list.append(float(1 / tick))
        if len(self.fps_list) > self.count:
            self.fps_list = self.fps_list[-self.count + 1:]  # 整个列表往前挪一位
        if len(self.get_fps_list) > self.count:
            self.get_fps_list = self.get_fps_list[-self.count + 1:]  # 整个列表往前挪一位
        try:
            self._fps = statistics.geometric_mean(self.fps_list[-100:])  # 取最后100个值的平均值
            self.middle_fps = statistics.median(self.fps_list)  # 取中间值
        except Exception:
            print(self.fps_list)
            raise
        self._max_fps = max(self.fps_list)
        self._min_fps = min(self.fps_list)

    @property
    def max_fps(self):
        return self._max_fps

    @property
    def min_fps(self):
        return self._min_fps

    @property
    def fps(self):
        return self._fps