Function registerVisualizer

comfyui/visualization.js:108–126

(nodeType, nodeData)

function registerVisualizer(nodeType, nodeData) {
    if (nodeData.name !== "StableFast3DSave" && nodeData.name !== "StableFast3DPreview")
        return;

    const originalOnNodeCreated = nodeType.prototype.onNodeCreated;

    nodeType.prototype.onNodeCreated = async function () {
        const result = originalOnNodeCreated?.apply(this, arguments);
        await createWidget.apply(this, [this, app]);
        this.setSize([512, 512]);
        return result;
    };

    nodeType.prototype.onExecuted = function (message) {
        if (message?.glbs?.[0]) {
            this.updateParameters(message.glbs[0]);
        }
    };
}

app.registerExtension({
    name: "StableFast3D.Visualizer",