File size: 3,897 Bytes
f12ab4c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
# SPDX-FileCopyrightText: Copyright (c) 2021-2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: LicenseRef-NvidiaProprietary
#
# NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
# property and proprietary rights in and to this material, related
# documentation and any modifications thereto. Any use, reproduction,
# disclosure or distribution of this material and related documentation
# without an express license agreement from NVIDIA CORPORATION or
# its affiliates is strictly prohibited.
import imgui
from gui_utils import imgui_utils
#----------------------------------------------------------------------------
class TruncationNoiseWidget:
def __init__(self, viz):
self.viz = viz
self.prev_num_ws = 0
self.trunc_psi = 0.7
self.trunc_cutoff = 7
self.noise_enable = True
self.noise_seed = 0
self.noise_anim = False
@imgui_utils.scoped_by_object_id
def __call__(self, show=True):
viz = self.viz
num_ws = viz.result.get('num_ws', 0)
has_noise = viz.result.get('has_noise', False)
if num_ws > 0 and num_ws != self.prev_num_ws:
if self.trunc_cutoff > num_ws or self.trunc_cutoff == self.prev_num_ws:
self.trunc_cutoff = num_ws
self.prev_num_ws = num_ws
if show:
imgui.text('Truncate')
imgui.same_line(viz.label_w)
with imgui_utils.item_width(viz.font_size * 10), imgui_utils.grayed_out(num_ws == 0):
_changed, self.trunc_psi = imgui.slider_float('##psi', self.trunc_psi, -1, 2, format='Psi %.2f')
imgui.same_line()
if num_ws == 0:
imgui_utils.button('Cutoff 0', width=(viz.font_size * 8 + viz.spacing), enabled=False)
else:
with imgui_utils.item_width(viz.font_size * 8 + viz.spacing):
changed, new_cutoff = imgui.slider_int('##cutoff', self.trunc_cutoff, 0, num_ws, format='Cutoff %d')
if changed:
self.trunc_cutoff = min(max(new_cutoff, 0), num_ws)
with imgui_utils.grayed_out(not has_noise):
imgui.same_line()
_clicked, self.noise_enable = imgui.checkbox('Noise##enable', self.noise_enable)
imgui.same_line(viz.font_size * 28.7)
with imgui_utils.grayed_out(not self.noise_enable):
with imgui_utils.item_width(-3 - viz.button_w - viz.spacing - viz.font_size * 4):
_changed, self.noise_seed = imgui.input_int('##seed', self.noise_seed)
imgui.same_line(spacing=0)
_clicked, self.noise_anim = imgui.checkbox('Anim##noise', self.noise_anim)
is_def_trunc = (self.trunc_psi == 1 and self.trunc_cutoff == num_ws)
is_def_noise = (self.noise_enable and self.noise_seed == 0 and not self.noise_anim)
with imgui_utils.grayed_out(is_def_trunc and not has_noise):
imgui.same_line(imgui.get_content_region_max()[0] - 1 - viz.button_w)
if imgui_utils.button('Reset', width=-1, enabled=(not is_def_trunc or not is_def_noise)):
self.prev_num_ws = num_ws
self.trunc_psi = 0.7
self.trunc_cutoff = 7
self.noise_enable = True
self.noise_seed = 0
self.noise_anim = False
if self.noise_anim:
self.noise_seed += 1
viz.args.update(trunc_psi=self.trunc_psi, trunc_cutoff=self.trunc_cutoff, random_seed=self.noise_seed)
viz.args.noise_mode = ('none' if not self.noise_enable else 'const' if self.noise_seed == 0 else 'random')
#----------------------------------------------------------------------------
|