File size: 3,228 Bytes
2d368c9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 | /**
* Support for widgets in the UI
*/
const TypeSlot = {
Input: 1,
Output: 2,
};
const TypeSlotEvent = {
Connect: true,
Disconnect: false,
};
/**
* Manage the slots on a node to allow a dynamic number of inputs
*/
export function node_add_dynamic(nodeType, prefix, dynamic_type='*', index_start=0, shape=LiteGraph.GRID_SHAPE, match_output=false) {
/*
this one should just put the "prefix" as the last empty entry.
Means we have to pay attention not to collide key names in the
input list.
*/
index_start = Math.max(0, index_start);
const onNodeCreated = nodeType.prototype.onNodeCreated
nodeType.prototype.onNodeCreated = function () {
const me = onNodeCreated?.apply(this);
this.addInput(prefix, dynamic_type);
if (match_output) {
this.addOutput(prefix, dynamic_type, { shape: shape });
}
return me;
}
const onConnectionsChange = nodeType.prototype.onConnectionsChange
nodeType.prototype.onConnectionsChange = function (slotType, slot_idx, event, link_info, node_slot) {
const me = onConnectionsChange?.apply(this, arguments)
if (slotType === TypeSlot.Input) {
if (slot_idx >= index_start && link_info) {
if (event === TypeSlotEvent.Connect) {
const fromNode = this.graph._nodes.find(
(otherNode) => otherNode.id == link_info.origin_id
)
if (fromNode) {
const parent_link = fromNode.outputs[link_info.origin_slot];
node_slot.type = parent_link.type;
node_slot.name = `${slot_idx}_${parent_link.name.toLowerCase()}`;
if (match_output) {
const slot_out = this.outputs[slot_idx];
slot_out.type = parent_link.type;
slot_out.name = `${slot_idx}_${parent_link.type}`;
}
}
}
}
// check that the last slot is a dynamic entry....
let last = this.inputs[this.inputs.length-1];
if (last.type != dynamic_type || last.name != prefix) {
this.addInput(prefix, dynamic_type);
}
if (match_output) {
last = this.outputs[this.outputs.length-1];
if (last.type != dynamic_type || last.name != prefix) {
this.addOutput(prefix, dynamic_type);
}
}
}
setTimeout(() => {
// clean off missing slot connects
if (this.graph === undefined) {
return;
}
let idx = index_start;
while (idx < this.inputs.length-1) {
const slot = this.inputs[idx];
if (slot.link == null) {
if (match_output) {
this.removeOutput(slot_idx);
}
this.removeInput(slot_idx);
} else {
idx += 1;
}
}
}, 25);
return me;
}
return nodeType;
} |