79 lines
3.8 KiB
Python
79 lines
3.8 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 numpy as np
|
|
import imgui
|
|
import dnnlib
|
|
from gui_utils import imgui_utils
|
|
|
|
#----------------------------------------------------------------------------
|
|
|
|
class LatentWidget:
|
|
def __init__(self, viz):
|
|
self.viz = viz
|
|
self.latent = dnnlib.EasyDict(x=0, y=0, anim=False, speed=0.25)
|
|
self.latent_def = dnnlib.EasyDict(self.latent)
|
|
self.step_y = 100
|
|
|
|
def drag(self, dx, dy):
|
|
viz = self.viz
|
|
self.latent.x += dx / viz.font_size * 4e-2
|
|
self.latent.y += dy / viz.font_size * 4e-2
|
|
|
|
@imgui_utils.scoped_by_object_id
|
|
def __call__(self, show=True):
|
|
viz = self.viz
|
|
if show:
|
|
imgui.text('Latent')
|
|
imgui.same_line(viz.label_w)
|
|
seed = round(self.latent.x) + round(self.latent.y) * self.step_y
|
|
with imgui_utils.item_width(viz.font_size * 8):
|
|
changed, seed = imgui.input_int('##seed', seed)
|
|
if changed:
|
|
self.latent.x = seed
|
|
self.latent.y = 0
|
|
imgui.same_line(viz.label_w + viz.font_size * 8 + viz.spacing)
|
|
frac_x = self.latent.x - round(self.latent.x)
|
|
frac_y = self.latent.y - round(self.latent.y)
|
|
with imgui_utils.item_width(viz.font_size * 5):
|
|
changed, (new_frac_x, new_frac_y) = imgui.input_float2('##frac', frac_x, frac_y, format='%+.2f', flags=imgui.INPUT_TEXT_ENTER_RETURNS_TRUE)
|
|
if changed:
|
|
self.latent.x += new_frac_x - frac_x
|
|
self.latent.y += new_frac_y - frac_y
|
|
imgui.same_line(viz.label_w + viz.font_size * 13 + viz.spacing * 2)
|
|
_clicked, dragging, dx, dy = imgui_utils.drag_button('Drag', width=viz.button_w)
|
|
if dragging:
|
|
self.drag(dx, dy)
|
|
imgui.same_line(viz.label_w + viz.font_size * 13 + viz.button_w + viz.spacing * 3)
|
|
_clicked, self.latent.anim = imgui.checkbox('Anim', self.latent.anim)
|
|
imgui.same_line(round(viz.font_size * 27.7))
|
|
with imgui_utils.item_width(-1 - viz.button_w * 2 - viz.spacing * 2), imgui_utils.grayed_out(not self.latent.anim):
|
|
changed, speed = imgui.slider_float('##speed', self.latent.speed, -5, 5, format='Speed %.3f', power=3)
|
|
if changed:
|
|
self.latent.speed = speed
|
|
imgui.same_line()
|
|
snapped = dnnlib.EasyDict(self.latent, x=round(self.latent.x), y=round(self.latent.y))
|
|
if imgui_utils.button('Snap', width=viz.button_w, enabled=(self.latent != snapped)):
|
|
self.latent = snapped
|
|
imgui.same_line()
|
|
if imgui_utils.button('Reset', width=-1, enabled=(self.latent != self.latent_def)):
|
|
self.latent = dnnlib.EasyDict(self.latent_def)
|
|
|
|
if self.latent.anim:
|
|
self.latent.x += viz.frame_delta * self.latent.speed
|
|
viz.args.w0_seeds = [] # [[seed, weight], ...]
|
|
for ofs_x, ofs_y in [[0, 0], [1, 0], [0, 1], [1, 1]]:
|
|
seed_x = np.floor(self.latent.x) + ofs_x
|
|
seed_y = np.floor(self.latent.y) + ofs_y
|
|
seed = (int(seed_x) + int(seed_y) * self.step_y) & ((1 << 32) - 1)
|
|
weight = (1 - abs(self.latent.x - seed_x)) * (1 - abs(self.latent.y - seed_y))
|
|
if weight > 0:
|
|
viz.args.w0_seeds.append([seed, weight])
|
|
|
|
#----------------------------------------------------------------------------
|