mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-11 18:44:21 +08:00
6a7c8820b8
* Copy in `viser` communication infrastructure * Update background image * Add build directory to fix viser HTTP server May want to remove these later * Copy in `viser` communication infrastructure * Update background image * Pass dataset vis using viser * initial impl of ViewerParameter * idk what i did but brent and matt know * Suppress eslint typescript warnings * implement ViewerElement base class, then ViewerButton, ViewerSlider, etc * client -> server message setup * multi folder stuff * Stream controls over viser * import * add types to viewer_param.py * Move another message to viser * Update folder color * Use non throttled messages * multiple folders works * Cleanup, more porting * starting impl of control panel in viser * More porting * scenebox to viser * Cleanup * Fix render updates * More cleanup, docs, typing * start control_panel.py * set_hidden logic but not merged in * control_panel replicates behavior of existing control panel (visually), except for setting the list of output options. not yet integrated with viewer_utils.py for logic flow. also implemented set_disabled and set_hidden in javscript/ViewerElement side * Remove some original leva components * add some train util logic, attempt set_value, doesn't work on javascript side fully yet (value updates but not slider visually * bye zmq * missing call to decoder * 'gui_set' works properly, control panel now replicates behavior of original * remove print * Fix status viewer message * More cleanup * support set_options in dropdown * hiding buttons works correctly * fix set_options, switch from cur_value to get_value() * Fix slowdown * output options * colormap logic converted * add handle to cb_hook * fix some lambda bugs * cleanup * Save checkpoint logic readded * Cleanup and minor fixes * snappier ui * Refactoring * Fix state update bugs * Bug fixes * Write less to redux * added comments to reducer code * Render state machine (#1727) * state machine works, but train step throttling not implemented * more step logic, cleanup * remove random try catcH * mutex logic for render * Update lock logic * types and docs * added some state machine logic to help rerenders * merge --------- Co-authored-by: Justin <justin.g.kerr@gmail.com> * Refactor and cleanup * More cleanup and docs * docstring to gui_utils * docstrings, rename viewer_param.py to viewer_elements.py * Fix viewer, remove bridge stuff * Render on connect * Fix lint errors * More eslint fixes * render_freq logic * better rerender logic for crop * More cleanup * Purge * Cleanup * Lint fixes * Fix cache * better parsing logic * Fix infinite test * More cleanup * typo * Update webdocs * Update docs * rerender on custom hooks * add more docs * Use shared utilities in `viser.infra` * rework callback to use elements instead of handles * Fix ESlint errors * change naming of handle to element, add setter for value * cleanup --------- Co-authored-by: Matt Tancik <mattt4001@gmail.com> Co-authored-by: Justin <justin.g.kerr@gmail.com> Co-authored-by: Ethan Weber <eweb0124@gmail.com> Co-authored-by: Matthew Tancik <tancik@berkeley.edu>
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
"""Generate viser message definitions for TypeScript, by parsing Python dataclasses."""
|
|
|
|
import pathlib
|
|
import subprocess
|
|
|
|
from viser.infra import generate_typescript_interfaces
|
|
|
|
from nerfstudio.viewer.viser import NerfstudioMessage
|
|
|
|
|
|
def entrypoint() -> None:
|
|
"""Generate viser message definitions for TypeScript, by parsing Python dataclasses."""
|
|
|
|
# Generate typescript source.
|
|
defs = generate_typescript_interfaces(NerfstudioMessage)
|
|
|
|
# Write to file.
|
|
# Three parents from nerfstudio/scripts/viewer/sync_viser_message_defs.py:
|
|
# - nerfstudio/scripts/viewer/
|
|
# - nerfstudio/scripts/
|
|
# - nerfstudio/
|
|
target_path = pathlib.Path(__file__).absolute().parent.parent.parent / pathlib.Path(
|
|
"nerfstudio/viewer/app/src/modules/WebSocket/ViserMessages.tsx"
|
|
)
|
|
assert target_path.exists()
|
|
target_path.write_text(defs, encoding="utf_8")
|
|
print(f"Wrote to {target_path}")
|
|
|
|
# Run prettier.
|
|
# TODO: if this is not installed maybe we should print some error with installation
|
|
# instructions?
|
|
subprocess.run(args=["prettier", "-w", str(target_path)], check=False)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
entrypoint()
|