File size: 3,409 Bytes
82ea528 |
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 92 93 94 95 96 97 98 99 |
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)
|