diff --git a/project_loader.py b/project_loader.py index b840499..63817c0 100644 --- a/project_loader.py +++ b/project_loader.py @@ -393,11 +393,11 @@ class BinaryIndexDecoder: OUTPUT_NODE = False def decode(self, index: int): - return ( - bool((index >> 0) & 1), - bool((index >> 1) & 1), - bool((index >> 2) & 1), - ) + f0 = bool((index >> 0) & 1) + f1 = bool((index >> 1) & 1) + f2 = bool((index >> 2) & 1) + return {"ui": {"values": [str(f0).lower(), str(f1).lower(), str(f2).lower()]}, + "result": (f0, f1, f2)} # --- Mappings --- diff --git a/web/binary_index_decoder.js b/web/binary_index_decoder.js new file mode 100644 index 0000000..9d74a4b --- /dev/null +++ b/web/binary_index_decoder.js @@ -0,0 +1,17 @@ +import { app } from "../../scripts/app.js"; + +app.registerExtension({ + name: "json.manager.binary_index_decoder", + + async beforeRegisterNodeDef(nodeType, nodeData, app) { + if (nodeData.name !== "BinaryIndexDecoder") return; + + nodeType.prototype.onExecuted = function (output) { + if (!output?.values) return; + for (let i = 0; i < Math.min(output.values.length, this.outputs.length); i++) { + this.outputs[i].label = `${this.outputs[i].name} ${output.values[i]}`; + } + app.graph?.setDirtyCanvas(true, true); + }; + }, +});