Difficult-Rocket/Difficult_Rocket/client/fps/fps_log.py
2023-02-25 09:18:41 +08:00

66 lines
1.9 KiB
Python

# -------------------------------
# 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