66 lines
2.7 KiB
JavaScript
66 lines
2.7 KiB
JavaScript
// ComfyUI - CRZ Float to Int JS Extension
|
|
import { app } from "../../scripts/app.js";
|
|
|
|
app.registerExtension({
|
|
name: "CRZ.FloatToInt",
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (nodeData.name === "CRZFloatToInt") {
|
|
// Set node properties for transparency
|
|
nodeType.title_mode = LiteGraph.NO_TITLE;
|
|
// nodeType.collapsable = false;
|
|
|
|
const onNodeCreated = nodeType.prototype.onNodeCreated;
|
|
nodeType.prototype.onNodeCreated = function () {
|
|
const r = onNodeCreated ? onNodeCreated.apply(this, arguments) : undefined;
|
|
|
|
// Set same styling as passthrough node
|
|
this.bgcolor = "rgba(48, 48, 48, 0.829)";
|
|
this.color = "transparent";
|
|
this.boxcolor = "transparent";
|
|
|
|
// Match passthrough size
|
|
this.size = [80, 28];
|
|
this.resizable = true;
|
|
|
|
// Mark as CRZ node for connection hiding
|
|
this.isCRZNode = true;
|
|
|
|
// Hide input/output labels
|
|
if (this.inputs && this.inputs[0]) {
|
|
this.inputs[0].localized_name = " ";
|
|
this.inputs[0].color_off = "rgba(63, 126, 79, 1)";
|
|
this.inputs[0].color_on = "rgba(94, 238, 125, 1)";
|
|
}
|
|
if (this.outputs && this.outputs[0]) {
|
|
this.outputs[0].localized_name = " ";
|
|
this.outputs[0].color_off = "rgba(38, 78, 116, 1)";
|
|
this.outputs[0].color_on = "rgba(69, 147, 219, 1)";
|
|
}
|
|
|
|
return r;
|
|
};
|
|
|
|
// Override onDrawBackground
|
|
nodeType.prototype.onDrawBackground = function(ctx) {
|
|
// Keep sockets visible - don't make them transparent
|
|
// Just draw the connection line
|
|
const canvas = app.graph.list_of_graphcanvas[0];
|
|
if (this.inputs && this.outputs && this.inputs[0].pos && this.outputs[0].pos) {
|
|
let linkColor = "#999"; // Default connection color
|
|
|
|
ctx.lineWidth = canvas?.connections_width || 2;
|
|
ctx.strokeStyle = linkColor;
|
|
ctx.beginPath();
|
|
ctx.moveTo(this.inputs[0].pos[0], this.inputs[0].pos[1]);
|
|
ctx.lineTo(this.outputs[0].pos[0], this.outputs[0].pos[1]);
|
|
ctx.stroke();
|
|
}
|
|
};
|
|
|
|
// Override computeSize
|
|
nodeType.prototype.computeSize = function() {
|
|
return [80, 28];
|
|
};
|
|
}
|
|
}
|
|
});
|