583f2bdd13
BOMs barf on UTF-8 decoding within PyTorch: File "D:\Soft\Miniconda3\lib\site-packages\torch\utils_cpp_extension_versioner.py", line 16, in hash_source_files hash_value = update_hash(hash_value, file.read()) UnicodeDecodeError: 'gbk' codec can't decode byte 0xbf in position 2: illegal multibyte sequence Should fix #10, #14
73 lines
3.5 KiB
Python
73 lines
3.5 KiB
Python
# Copyright (c) 2021, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
#
|
|
# NVIDIA CORPORATION and its licensors retain all intellectual property
|
|
# and proprietary rights in and to this software, related documentation
|
|
# and any modifications thereto. Any use, reproduction, disclosure or
|
|
# distribution of this software and related documentation without an express
|
|
# license agreement from NVIDIA CORPORATION is strictly prohibited.
|
|
|
|
import array
|
|
import numpy as np
|
|
import imgui
|
|
from gui_utils import imgui_utils
|
|
|
|
#----------------------------------------------------------------------------
|
|
|
|
class PerformanceWidget:
|
|
def __init__(self, viz):
|
|
self.viz = viz
|
|
self.gui_times = [float('nan')] * 60
|
|
self.render_times = [float('nan')] * 30
|
|
self.fps_limit = 60
|
|
self.use_vsync = False
|
|
self.is_async = False
|
|
self.force_fp32 = False
|
|
|
|
@imgui_utils.scoped_by_object_id
|
|
def __call__(self, show=True):
|
|
viz = self.viz
|
|
self.gui_times = self.gui_times[1:] + [viz.frame_delta]
|
|
if 'render_time' in viz.result:
|
|
self.render_times = self.render_times[1:] + [viz.result.render_time]
|
|
del viz.result.render_time
|
|
|
|
if show:
|
|
imgui.text('GUI')
|
|
imgui.same_line(viz.label_w)
|
|
with imgui_utils.item_width(viz.font_size * 8):
|
|
imgui.plot_lines('##gui_times', array.array('f', self.gui_times), scale_min=0)
|
|
imgui.same_line(viz.label_w + viz.font_size * 9)
|
|
t = [x for x in self.gui_times if x > 0]
|
|
t = np.mean(t) if len(t) > 0 else 0
|
|
imgui.text(f'{t*1e3:.1f} ms' if t > 0 else 'N/A')
|
|
imgui.same_line(viz.label_w + viz.font_size * 14)
|
|
imgui.text(f'{1/t:.1f} FPS' if t > 0 else 'N/A')
|
|
imgui.same_line(viz.label_w + viz.font_size * 18 + viz.spacing * 3)
|
|
with imgui_utils.item_width(viz.font_size * 6):
|
|
_changed, self.fps_limit = imgui.input_int('FPS limit', self.fps_limit, flags=imgui.INPUT_TEXT_ENTER_RETURNS_TRUE)
|
|
self.fps_limit = min(max(self.fps_limit, 5), 1000)
|
|
imgui.same_line(imgui.get_content_region_max()[0] - 1 - viz.button_w * 2 - viz.spacing)
|
|
_clicked, self.use_vsync = imgui.checkbox('Vertical sync', self.use_vsync)
|
|
|
|
if show:
|
|
imgui.text('Render')
|
|
imgui.same_line(viz.label_w)
|
|
with imgui_utils.item_width(viz.font_size * 8):
|
|
imgui.plot_lines('##render_times', array.array('f', self.render_times), scale_min=0)
|
|
imgui.same_line(viz.label_w + viz.font_size * 9)
|
|
t = [x for x in self.render_times if x > 0]
|
|
t = np.mean(t) if len(t) > 0 else 0
|
|
imgui.text(f'{t*1e3:.1f} ms' if t > 0 else 'N/A')
|
|
imgui.same_line(viz.label_w + viz.font_size * 14)
|
|
imgui.text(f'{1/t:.1f} FPS' if t > 0 else 'N/A')
|
|
imgui.same_line(viz.label_w + viz.font_size * 18 + viz.spacing * 3)
|
|
_clicked, self.is_async = imgui.checkbox('Separate process', self.is_async)
|
|
imgui.same_line(imgui.get_content_region_max()[0] - 1 - viz.button_w * 2 - viz.spacing)
|
|
_clicked, self.force_fp32 = imgui.checkbox('Force FP32', self.force_fp32)
|
|
|
|
viz.set_fps_limit(self.fps_limit)
|
|
viz.set_vsync(self.use_vsync)
|
|
viz.set_async(self.is_async)
|
|
viz.args.force_fp32 = self.force_fp32
|
|
|
|
#----------------------------------------------------------------------------
|