PozzettiAndrea
1721e2bf4e
The three.js preview's generic name made it unclear next to the VTK/Dual/ Multi/Batch previews. Renamed the node id (schema + both mappings), the widget file (mesh_preview.js -> mesh_preview_three.js, extension name geometrypack.meshpreview -> geometrypack.meshpreview.three), and the four test workflows that reference the node type. Breaking for external saved workflows that used GeomPackPreviewMesh (node type no longer exists) -- accepted; the pack's own workflows are updated.
131 lines
6.1 KiB
JavaScript
131 lines
6.1 KiB
JavaScript
/**
|
|
* ComfyUI GeomPack - 3D Mesh Preview Widget
|
|
* Minimal implementation based on ComfyUI-3D-Pack pattern
|
|
*/
|
|
|
|
import { app } from "../../../scripts/app.js";
|
|
|
|
// Auto-detect extension folder name (handles ComfyUI-GeometryPack or comfyui-geometrypack)
|
|
const EXTENSION_FOLDER = (() => {
|
|
const url = import.meta.url;
|
|
const match = url.match(/\/extensions\/([^/]+)\//);
|
|
return match ? match[1] : "ComfyUI-GeometryPack";
|
|
})();
|
|
|
|
console.log("[GeomPack] Loading mesh preview extension...");
|
|
|
|
app.registerExtension({
|
|
name: "geometrypack.meshpreview.three",
|
|
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (nodeData.name === "GeomPackPreviewMeshThree") {
|
|
console.log("[GeomPack] Registering Preview Mesh node");
|
|
|
|
const onNodeCreated = nodeType.prototype.onNodeCreated;
|
|
nodeType.prototype.onNodeCreated = function() {
|
|
const r = onNodeCreated ? onNodeCreated.apply(this, arguments) : undefined;
|
|
|
|
console.log("[GeomPack DEBUG] this.widgets before:", this.widgets);
|
|
console.log("[GeomPack DEBUG] nodeData:", nodeData);
|
|
console.log("[GeomPack] Node created, adding widget");
|
|
|
|
// Create iframe for 3D viewer
|
|
const iframe = document.createElement("iframe");
|
|
iframe.style.width = "100%";
|
|
iframe.style.height = "100%";
|
|
iframe.style.border = "none";
|
|
iframe.style.backgroundColor = "#2a2a2a";
|
|
iframe.style.aspectRatio = "1";
|
|
|
|
// Point to our HTML viewer (with cache buster)
|
|
iframe.src = `/extensions/${EXTENSION_FOLDER}/viewer.html?v=` + Date.now();
|
|
|
|
// Add widget with required options
|
|
console.log("[GeomPack DEBUG] About to call addDOMWidget");
|
|
console.log("[GeomPack DEBUG] typeof this.addDOMWidget:", typeof this.addDOMWidget);
|
|
|
|
const widget = this.addDOMWidget("preview", "MESH_PREVIEW", iframe, {
|
|
getValue() { return ""; },
|
|
setValue(v) { }
|
|
});
|
|
|
|
console.log("[GeomPack DEBUG] Widget created:", widget);
|
|
console.log("[GeomPack DEBUG] Widget properties:", Object.keys(widget || {}));
|
|
console.log("[GeomPack DEBUG] Widget.id:", widget?.id);
|
|
console.log("[GeomPack DEBUG] this.widgets after addDOMWidget:", this.widgets);
|
|
|
|
// Set widget size - computeSize returns [width, height]
|
|
widget.computeSize = function(width) {
|
|
console.log("[GeomPack DEBUG] computeSize called with width:", width);
|
|
const size = [width || 512, width || 512];
|
|
console.log("[GeomPack DEBUG] computeSize returning:", size);
|
|
return size;
|
|
};
|
|
|
|
// Also try setting element size directly
|
|
widget.element = iframe;
|
|
|
|
console.log("[GeomPack DEBUG] Widget computeSize set");
|
|
console.log("[GeomPack DEBUG] Iframe dimensions:", iframe.style.width, iframe.style.height);
|
|
|
|
// Store iframe reference
|
|
this.meshViewerIframe = iframe;
|
|
|
|
// Set initial node size to be square
|
|
this.setSize([512, 512]);
|
|
console.log("[GeomPack DEBUG] Node size set to [512, 512]");
|
|
|
|
// Handle execution
|
|
const onExecuted = this.onExecuted;
|
|
this.onExecuted = function(message) {
|
|
console.log("[GeomPack] onExecuted called with message:", message);
|
|
onExecuted?.apply(this, arguments);
|
|
|
|
// The message IS the UI data (not message.ui)
|
|
if (message?.mesh_file && message.mesh_file[0]) {
|
|
const filename = message.mesh_file[0];
|
|
console.log(`[GeomPack] Loading mesh: ${filename}`);
|
|
|
|
// ComfyUI serves files via /view API — parse type/subfolder/filename
|
|
// Handles both absolute (/home/.../output/sub/file) and relative (output/sub/file) paths
|
|
let filepath;
|
|
const normalized = filename.replace(/\\/g, '/');
|
|
const pathMatch = normalized.match(/(?:^|\/)(output|input|temp)\/(.+)$/);
|
|
if (pathMatch) {
|
|
const [, type, relPath] = pathMatch;
|
|
const parts = relPath.split('/');
|
|
const fname = parts.pop();
|
|
const subfolder = parts.join('/');
|
|
filepath = `/view?filename=${encodeURIComponent(fname)}&type=${type}&subfolder=${encodeURIComponent(subfolder)}`;
|
|
} else {
|
|
// Fallback: use just the basename
|
|
const basename = normalized.split('/').pop();
|
|
filepath = `/view?filename=${encodeURIComponent(basename)}&type=output&subfolder=`;
|
|
}
|
|
|
|
// Send message to iframe (with delay to ensure iframe is loaded)
|
|
setTimeout(() => {
|
|
if (iframe.contentWindow) {
|
|
console.log(`[GeomPack] Sending postMessage to iframe: ${filepath}`);
|
|
iframe.contentWindow.postMessage({
|
|
type: "LOAD_MESH",
|
|
filepath: filepath,
|
|
timestamp: Date.now()
|
|
}, "*");
|
|
} else {
|
|
console.error("[GeomPack] Iframe contentWindow not available");
|
|
}
|
|
}, 100);
|
|
} else {
|
|
console.log("[GeomPack] No mesh_file in message data. Keys:", Object.keys(message || {}));
|
|
}
|
|
};
|
|
|
|
return r;
|
|
};
|
|
}
|
|
}
|
|
});
|
|
|
|
console.log("[GeomPack] Mesh preview extension registered");
|