108 lines
3.7 KiB
JavaScript
108 lines
3.7 KiB
JavaScript
import { app } from "../../../scripts/app.js";
|
|
import {chainCallback} from "./utils.js";
|
|
import {clearInputs} from "./inputs.js";
|
|
import {addWidgets} from "./widgets.js";
|
|
import {colors, bg_colors, node_type_list} from "./constants.js";
|
|
|
|
function initialisation(node) {
|
|
node.widgets[0].callback = ( value ) => {
|
|
clearInputs(node);
|
|
node.addOutput("output", value);
|
|
node.addInput("input", value);
|
|
node.color = colors[node_type_list.indexOf(value)];
|
|
node.bgcolor = bg_colors[node_type_list.indexOf(value)];
|
|
};
|
|
node.color = colors[node_type_list.indexOf("none")];
|
|
node.bgcolor = bg_colors[node_type_list.indexOf("none")];
|
|
}
|
|
|
|
function configure(info) {
|
|
let widgetDict = info.widgets_values
|
|
if (info.widgets_values.length == undefined) {
|
|
|
|
for (let w of this.widgets) {
|
|
if (w.name in widgetDict) {
|
|
w.value = widgetDict[w.name].value;
|
|
}
|
|
}
|
|
// check if widgetDict in this.widgets
|
|
for (let [key, value] of Object.entries(widgetDict)) {
|
|
let widget = this.widgets.find(w => w.name === key);
|
|
let type = this.widgets.find(w => w.name === "type");
|
|
if(!widget){
|
|
addWidgets(this, key, value, app);
|
|
widget = this.widgets.find(w => w.name === key);
|
|
}
|
|
//this.widgets.push(value);
|
|
widget.options = info.widgets_values[key].options;
|
|
widget.value = info.widgets_values[key].value;
|
|
//if value exists in inputs
|
|
|
|
for (let input of this.inputs)
|
|
if (input.name == key){
|
|
//find if key exists in inputs array in inputs.Name
|
|
if (info.widgets_values[key].type != "converted-widget"){
|
|
this.removeInput(this.inputs.indexOf(input));
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (info.outputs_values != undefined){
|
|
// deep copy outputs
|
|
if(this.id == -1){
|
|
this.outputs[0] = {links: null, name: info.outputs_values.name, type: info.outputs_values.type};
|
|
}else{
|
|
this.outputs[0] = {...info.outputs_values};
|
|
}
|
|
}
|
|
this.setSize(info.size);
|
|
};
|
|
|
|
|
|
function serialize(info) {
|
|
info.widgets_values = {};
|
|
if (!this.widgets) {
|
|
return;
|
|
}
|
|
|
|
for (let w of this.widgets) {
|
|
info.widgets_values[w.name] = {name: w.name, options : w.options, value: w.value, type: w.type, origType: w.origType, last_y: w.last_y};
|
|
}
|
|
for (let w of this.inputs){
|
|
// if w.name exists in info.widgets_values
|
|
if (info.widgets_values[w.name]){
|
|
if(info.widgets_values[w.name].type == "converted-widget"){
|
|
if(info.widgets_values[w.name].origType == "toggle"){
|
|
w.type = "BOOLEAN";
|
|
}else if(info.widgets_values[w.name].origType == "combo"){
|
|
w.type = "COMBO";
|
|
}
|
|
}
|
|
}
|
|
}
|
|
for (let w of this.inputs){
|
|
if (w.name=="input"){
|
|
w.type = info.widgets_values.type.value;
|
|
}
|
|
}
|
|
if (this.outputs.length > 0){
|
|
if (this.outputs[0].links == null){
|
|
info.outputs_values = {links: null, name: this.outputs[0].name, type: this.outputs[0].type};
|
|
}else{
|
|
info.outputs_values = {links: [...this.outputs[0].links], name: this.outputs[0].name, slot_index: this.outputs[0].slot_index, type: this.outputs[0].type};
|
|
}
|
|
}
|
|
this.setSize(info.size);
|
|
}
|
|
|
|
|
|
export function setupContinueNode(nodeType, nodeData, app) {
|
|
nodeType.prototype.onNodeCreated = function() {
|
|
chainCallback(this, "onConfigure", configure);
|
|
chainCallback(this, "onSerialize", serialize);
|
|
initialisation(this);
|
|
}
|
|
}
|