100 lines
4.5 KiB
JavaScript
100 lines
4.5 KiB
JavaScript
import { BaseAnyInputConnectedNode } from "./base_any_input_connected_node.js";
|
|
import { PassThroughFollowing } from "./utils.js";
|
|
import { wait } from "./shared_utils.js";
|
|
export class BaseNodeModeChanger extends BaseAnyInputConnectedNode {
|
|
constructor(title) {
|
|
super(title);
|
|
this.inputsPassThroughFollowing = PassThroughFollowing.ALL;
|
|
this.isVirtualNode = true;
|
|
this.modeOn = -1;
|
|
this.modeOff = -1;
|
|
this.stabilizedWidgetStates = [];
|
|
this.properties = this.properties || {};
|
|
this.properties['restriction'] = 'None';
|
|
wait(10).then(() => {
|
|
if (this.modeOn < 0 || this.modeOff < 0) {
|
|
throw new Error('modeOn and modeOff must be overridden.');
|
|
}
|
|
});
|
|
this.addOutput("OPT_CONNECTION", "*");
|
|
}
|
|
handleLinkedNodesStabilization(linkedNodes) {
|
|
var _a, _b, _c, _d;
|
|
let restictToOne = (_b = (_a = this.properties) === null || _a === void 0 ? void 0 : _a['restriction']) === null || _b === void 0 ? void 0 : _b.includes(' one');
|
|
let oneIsOn = false;
|
|
console.log(this.stabilizedWidgetStates.join(', '), ' | ', (_c = this.widgets) === null || _c === void 0 ? void 0 : _c.map(w => w.value).join(', '));
|
|
if (restictToOne && this.stabilizedWidgetStates.length) {
|
|
for (const [index, state] of this.stabilizedWidgetStates.entries()) {
|
|
if (linkedNodes[index] && (linkedNodes[index].mode === this.modeOn) !== state) {
|
|
this.widgets[index].doModeChange(linkedNodes[index].mode === this.modeOn);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
this.stabilizedWidgetStates = [];
|
|
for (const [index, node] of linkedNodes.entries()) {
|
|
let widget = this.widgets && this.widgets[index];
|
|
if (!widget) {
|
|
this._tempWidth = this.size[0];
|
|
widget = this.addWidget('toggle', '', false, '', { "on": 'yes', "off": 'no' });
|
|
}
|
|
node && this.setWidget(widget, node, restictToOne && oneIsOn ? false : undefined);
|
|
oneIsOn = oneIsOn || widget.value;
|
|
this.stabilizedWidgetStates.push(widget.value);
|
|
}
|
|
if (this.widgets && this.widgets.length > linkedNodes.length) {
|
|
this.widgets.length = linkedNodes.length;
|
|
}
|
|
if (((_d = this.properties) === null || _d === void 0 ? void 0 : _d['restriction']) === 'Always one' && !oneIsOn) {
|
|
this.widgets[0].doModeChange(true, true);
|
|
}
|
|
}
|
|
onConnectionsChange(type, index, connected, linkInfo, ioSlot) {
|
|
this.stabilizedWidgetStates = [];
|
|
super.onConnectionsChange && super.onConnectionsChange(type, index, connected, linkInfo, ioSlot);
|
|
}
|
|
setWidget(widget, linkedNode, forceValue) {
|
|
const value = forceValue == null ? linkedNode.mode === this.modeOn : forceValue;
|
|
widget.name = `Enable ${linkedNode.title}`;
|
|
widget.options = { 'on': 'yes', 'off': 'no' };
|
|
widget.value = value;
|
|
widget.doModeChange = (forceValue, skipOtherNodeCheck) => {
|
|
var _a, _b, _c;
|
|
let newValue = forceValue == null ? linkedNode.mode === this.modeOff : forceValue;
|
|
if (skipOtherNodeCheck !== true) {
|
|
if (newValue && ((_b = (_a = this.properties) === null || _a === void 0 ? void 0 : _a['restriction']) === null || _b === void 0 ? void 0 : _b.includes(' one'))) {
|
|
for (const widget of this.widgets) {
|
|
widget.doModeChange(false, true);
|
|
}
|
|
}
|
|
else if (!newValue && ((_c = this.properties) === null || _c === void 0 ? void 0 : _c['restriction']) === 'Always one') {
|
|
newValue = true;
|
|
}
|
|
}
|
|
linkedNode.mode = (newValue ? this.modeOn : this.modeOff);
|
|
widget.value = newValue;
|
|
this.stabilizedWidgetStates = [];
|
|
};
|
|
widget.callback = () => {
|
|
widget.doModeChange();
|
|
};
|
|
if (forceValue != null) {
|
|
linkedNode.mode = (forceValue ? this.modeOn : this.modeOff);
|
|
}
|
|
}
|
|
forceWidgetOff(widget, skipOtherNodeCheck) {
|
|
widget.doModeChange(false, skipOtherNodeCheck);
|
|
}
|
|
forceWidgetOn(widget, skipOtherNodeCheck) {
|
|
widget.doModeChange(true, skipOtherNodeCheck);
|
|
}
|
|
static setUp(clazz) {
|
|
BaseAnyInputConnectedNode.setUp(clazz);
|
|
}
|
|
}
|
|
BaseNodeModeChanger.collapsible = false;
|
|
BaseNodeModeChanger["@restriction"] = {
|
|
type: "combo",
|
|
values: ["None", "Max one", "Always one"],
|
|
};
|