210 lines
10 KiB
JavaScript
210 lines
10 KiB
JavaScript
import { app } from "/scripts/app.js";
|
|
import { ComfyWidgets } from "/scripts/widgets.js";
|
|
import {CUSTOM_INT, recursiveLinkUpstream, transformFunc, swapInputs,swapOutputs, renameNodeInputs,renameNodeOutputs, removeNodeInputs,removeNodeOutputs, getDrawColor, computeCanvasSize} from "./utils.js"
|
|
|
|
// Displays input text on a node
|
|
|
|
app.registerExtension({
|
|
name: "Comfy.lam.MultiTextConcatenate",
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
var showNames=["IfInnerExecute",'MultiIntFormula','MultiParamFormula']
|
|
if (showNames.indexOf(nodeData.name)>=0) {
|
|
const onDrawForeground = nodeType.prototype.onDrawForeground;
|
|
nodeType.prototype.onDrawForeground = function (ctx) {
|
|
const r = onDrawForeground?.apply?.(this, arguments);
|
|
const v = app.nodeOutputs?.[this.id + ""];
|
|
if (!this.flags.collapsed && v) {
|
|
const text = v.value[0] + "";
|
|
ctx.save();
|
|
ctx.font = "bold 12px sans-serif";
|
|
ctx.fillStyle = "dodgerblue";
|
|
const sz = ctx.measureText(text);
|
|
ctx.fillText(text, this.size[0] - sz.width - 5, LiteGraph.NODE_SLOT_HEIGHT * 3);
|
|
ctx.restore();
|
|
}
|
|
return r;
|
|
};
|
|
}
|
|
var names=["MultiTextConcatenate",'MultiTextSelelct',"MultiIntFormula","MultiParamFormula"]
|
|
if (names.indexOf(nodeData.name)>=0) {
|
|
const onNodeCreated = nodeType.prototype.onNodeCreated;
|
|
nodeType.prototype.onNodeCreated = function () {
|
|
const r = onNodeCreated ? onNodeCreated.apply(this, arguments) : undefined;
|
|
this.originalsize=0
|
|
this.inputType="STRING"
|
|
this.inputPrefix="text"
|
|
this.outputPrefix=""
|
|
if('MultiTextSelelct'==nodeData.name){
|
|
this.originalsize=1
|
|
}
|
|
if('MultiIntFormula'==nodeData.name){
|
|
this.inputType="INT,FLOAT"
|
|
this.inputPrefix="n"
|
|
}
|
|
if('MultiParamFormula'==nodeData.name){
|
|
this.inputType="*"
|
|
this.inputPrefix="p"
|
|
this.outputPrefix="r"
|
|
}
|
|
function changCustomtext(){
|
|
let cuTexts=[]
|
|
const pos = this.widgets.findIndex((w) => w.type === "customtext");
|
|
if (pos !== -1) {
|
|
for (let i = pos; i < this.widgets.length; i++) {
|
|
cuTexts.push({
|
|
name: this.widgets[i].name,
|
|
value: this.widgets[i].value,
|
|
})
|
|
this.widgets[i].onRemove?.();
|
|
}
|
|
this.widgets.length = pos;
|
|
}
|
|
for(let i = 0; i < cuTexts.length; i++){
|
|
let val=cuTexts[i].value
|
|
let name=cuTexts[i].name
|
|
const w =ComfyWidgets["STRING"](this, name, ["STRING", { multiline: true }, app]).widget;
|
|
w.value = val;
|
|
}
|
|
}
|
|
this.getExtraMenuOptions = function(_, options) {
|
|
if(this.outputPrefix){
|
|
options.unshift(
|
|
{
|
|
content: `最前插入参 /\\`,
|
|
callback: () => {
|
|
this.addInput(this.inputPrefix, this.inputType)
|
|
const inputLenth = this.inputs.length-1
|
|
const index = 0+this.originalsize
|
|
|
|
for (let i = inputLenth; i > index; i--) {
|
|
swapInputs(this, i, i-1)
|
|
}
|
|
renameNodeInputs(this, this.inputPrefix,this.originalsize)
|
|
this.setDirtyCanvas(true);
|
|
changCustomtext.call(this)
|
|
},
|
|
},
|
|
{
|
|
content: `最后插入参 \\/`,
|
|
callback: () => {
|
|
this.addInput(this.inputPrefix, this.inputType)
|
|
renameNodeInputs(this, this.inputPrefix,this.originalsize)
|
|
this.setDirtyCanvas(true);
|
|
changCustomtext.call(this)
|
|
},
|
|
},
|
|
{
|
|
content: "删除全部未连接的入参框",
|
|
callback: () => {
|
|
let indexesToRemove = []
|
|
|
|
for (let i = 0; i < this.inputs.length; i++) {
|
|
if (!this.inputs[i].link&&i>=this.originalsize) {
|
|
indexesToRemove.push(i)
|
|
}
|
|
}
|
|
|
|
removeNodeInputs(this, indexesToRemove,this.originalsize)
|
|
renameNodeInputs(this, this.inputPrefix,this.originalsize)
|
|
changCustomtext.call(this)
|
|
},
|
|
},
|
|
{
|
|
content: `最前插出参 /\\`,
|
|
callback: () => {
|
|
this.addOutput(this.outputPrefix, this.inputType)
|
|
const inputLenth = this.outputs.length-1
|
|
const index = 0+this.originalsize
|
|
|
|
for (let i = inputLenth; i > index; i--) {
|
|
swapOutputs(this, i, i-1)
|
|
}
|
|
renameNodeOutputs(this, this.outputPrefix,this.originalsize)
|
|
this.setDirtyCanvas(true);
|
|
changCustomtext.call(this)
|
|
},
|
|
},
|
|
{
|
|
content: `最后插出参 \\/`,
|
|
callback: () => {
|
|
this.addOutput(this.outputPrefix, this.inputType)
|
|
renameNodeOutputs(this, this.outputPrefix,this.originalsize)
|
|
this.setDirtyCanvas(true);
|
|
changCustomtext.call(this)
|
|
},
|
|
},
|
|
{
|
|
content: "删除全部未连接的出参框",
|
|
callback: () => {
|
|
let indexesToRemove = []
|
|
for (let i = 0; i < this.outputs.length; i++) {
|
|
if ((!this.outputs[i].links||this.outputs[i].links.length==0)&&i>=this.originalsize) {
|
|
indexesToRemove.push(i)
|
|
}
|
|
}
|
|
removeNodeOutputs(this, indexesToRemove,this.originalsize)
|
|
renameNodeOutputs(this, this.outputPrefix,this.originalsize)
|
|
changCustomtext.call(this)
|
|
},
|
|
}
|
|
);
|
|
}else{
|
|
options.unshift(
|
|
{
|
|
content: `最前插入参 /\\`,
|
|
callback: () => {
|
|
this.addInput(this.inputPrefix, this.inputType)
|
|
const inputLenth = this.inputs.length-1
|
|
const index = 0+this.originalsize
|
|
|
|
for (let i = inputLenth; i > index; i--) {
|
|
swapInputs(this, i, i-1)
|
|
}
|
|
renameNodeInputs(this, this.inputPrefix,this.originalsize)
|
|
this.setDirtyCanvas(true);
|
|
changCustomtext.call(this)
|
|
},
|
|
},
|
|
{
|
|
content: `最后插入参 \\/`,
|
|
callback: () => {
|
|
this.addInput(this.inputPrefix, this.inputType)
|
|
renameNodeInputs(this, this.inputPrefix,this.originalsize)
|
|
this.setDirtyCanvas(true);
|
|
changCustomtext.call(this)
|
|
},
|
|
},
|
|
{
|
|
content: "删除全部未连接的入参框",
|
|
callback: () => {
|
|
let indexesToRemove = []
|
|
|
|
for (let i = 0; i < this.inputs.length; i++) {
|
|
if (!this.inputs[i].link&&i>=this.originalsize) {
|
|
indexesToRemove.push(i)
|
|
}
|
|
}
|
|
|
|
removeNodeInputs(this, indexesToRemove,this.originalsize)
|
|
renameNodeInputs(this, this.inputPrefix,this.originalsize)
|
|
changCustomtext.call(this)
|
|
},
|
|
}
|
|
);
|
|
}
|
|
}
|
|
|
|
|
|
this.onSelected = function () {
|
|
this.selected = true
|
|
}
|
|
this.onDeselected = function () {
|
|
this.selected = false
|
|
}
|
|
|
|
return r;
|
|
}
|
|
|
|
}
|
|
},
|
|
}); |