67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
import os
|
|
from .common import COMFYUI_OUTPUT_FOLDER
|
|
|
|
|
|
class PreviewGaussianNode:
|
|
"""Preview Gaussian Splatting PLY files in an interactive gsplat.js viewer."""
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"ply_path": ("STRING", {
|
|
"forceInput": True,
|
|
"tooltip": "Path to a Gaussian Splatting PLY file",
|
|
}),
|
|
},
|
|
"optional": {
|
|
"extrinsics": ("EXTRINSICS", {
|
|
"tooltip": "4x4 camera extrinsics matrix for initial view",
|
|
}),
|
|
"intrinsics": ("INTRINSICS", {
|
|
"tooltip": "3x3 camera intrinsics matrix for FOV",
|
|
}),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ()
|
|
OUTPUT_NODE = True
|
|
FUNCTION = "preview_gaussian"
|
|
CATEGORY = "PlyPreview/visualization"
|
|
|
|
def preview_gaussian(self, ply_path: str, extrinsics=None, intrinsics=None):
|
|
if not ply_path:
|
|
print("[PreviewGaussian] No PLY path provided")
|
|
return {"ui": {"error": ["No PLY path provided"]}}
|
|
|
|
if not os.path.exists(ply_path):
|
|
print(f"[PreviewGaussian] PLY file not found: {ply_path}")
|
|
return {"ui": {"error": [f"File not found: {ply_path}"]}}
|
|
|
|
filename = os.path.basename(ply_path)
|
|
|
|
if COMFYUI_OUTPUT_FOLDER and ply_path.startswith(COMFYUI_OUTPUT_FOLDER):
|
|
relative_path = os.path.relpath(ply_path, COMFYUI_OUTPUT_FOLDER)
|
|
else:
|
|
relative_path = filename
|
|
|
|
file_size = os.path.getsize(ply_path)
|
|
file_size_mb = file_size / (1024 * 1024)
|
|
|
|
print(f"[PreviewGaussian] Loading PLY: {filename} ({file_size_mb:.2f} MB)")
|
|
|
|
ui_data: dict[str, list] = {
|
|
"ply_file": [relative_path],
|
|
"filename": [filename],
|
|
"file_size_mb": [round(file_size_mb, 2)],
|
|
}
|
|
|
|
if extrinsics is not None:
|
|
ui_data["extrinsics"] = [extrinsics]
|
|
if intrinsics is not None:
|
|
ui_data["intrinsics"] = [intrinsics]
|
|
|
|
return {"ui": ui_data}
|