stylegan3/viz/performance_widget.py

74 lines
3.5 KiB
Python
Raw Normal View History

# Copyright (c) 2021, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2021-10-07 09:55:26 +00:00
#
# 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
#----------------------------------------------------------------------------