| import { app } from '../../../scripts/app.js' | |
| app.ui.settings.addSetting({ | |
| id: "ADE.ShowExperimental", | |
| name: "🎭🅐🅓 Show experimental nodes", | |
| type: "boolean", | |
| defaultValue: false, | |
| }); | |
| app.ui.settings.addSetting({ | |
| id: "ADE.ShowDeprecated", | |
| name: "🎭🅐🅓 Show deprecated nodes", | |
| type: "boolean", | |
| defaultValue: false, | |
| }); | |
| const deprecate_nodes = { | |
| name: 'AnimateDiff.deprecate_nodes', | |
| async beforeRegisterNodeDef(nodeType, nodeData, app) { | |
| let showDeprecated = app.ui.settings.getSettingValue("ADE.ShowDeprecated", false) | |
| let showExperimental = app.ui.settings.getSettingValue("ADE.ShowExperimental", false) | |
| let warnWidget = false | |
| if (nodeData?.input?.optional) { | |
| for (let w in nodeData.input.optional) { | |
| if (nodeData.input.optional[w][0] == "ADEWARN") { | |
| warnWidget = nodeData.input.optional[w] | |
| break | |
| } | |
| } | |
| } | |
| if (warnWidget) { | |
| if (!((warnWidget[1].warn_type || "deprecated") == "deprecated" && showDeprecated) && | |
| !(warnWidget[1].warn_type == "experimental" && showExperimental)) { | |
| nodeType.filter = "hidden" | |
| } | |
| } | |
| if (!nodeType.filter) { | |
| nodeType.filter = app.graph.filter | |
| } | |
| }, | |
| async init() { | |
| app.graph.filter = app.graph.filter || "shown" | |
| }, | |
| async setup() { | |
| for (let k in LiteGraph.registered_node_types) { | |
| let nodeType = LiteGraph.registered_node_types[k] | |
| if (!nodeType.filter) { | |
| nodeType.filter = app.graph.filter | |
| } | |
| } | |
| }, | |
| async getCustomWidgets() { | |
| return { | |
| ADEWARN(node, inputName, inputData) { | |
| let w = { | |
| name : inputName, | |
| type : "ADE.WARN", | |
| value : "", | |
| draw : function(ctx, node, widget_width, y, H) { | |
| var show_text = app.canvas.ds.scale > 0.5; | |
| var margin = 15; | |
| var text_color = inputData[1]['color'] || "#FCC" | |
| ctx.textAlign = "center"; | |
| if (show_text) { | |
| if(!this.disabled) | |
| ctx.stroke(); | |
| ctx.save(); | |
| ctx.beginPath(); | |
| ctx.rect(margin, y, widget_width - margin * 2, H); | |
| ctx.clip(); | |
| ctx.fillStyle = text_color; | |
| let disp_text = inputData[1]['text'] | |
| ctx.fillText(disp_text, widget_width/2, y + H * 0.7); | |
| ctx.restore(); | |
| } | |
| }, | |
| options : {"serialize": false}, | |
| computeSize : function(width) { | |
| if (inputData[1]['text']) { | |
| return [width, 20] | |
| } | |
| return [0, -4] | |
| } | |
| } | |
| if (!node.widgets) { | |
| node.widgets = [] | |
| } | |
| node.widgets.push(w) | |
| return w | |
| } | |
| } | |
| } | |
| } | |
| app.registerExtension(deprecate_nodes) | |