set/get tweaks

This commit is contained in:
kijai 2024-04-24 21:30:07 +03:00
parent 473ea60f78
commit 6cb0e8e334
2 changed files with 9 additions and 26 deletions

3
.gitignore vendored
View File

@ -4,4 +4,5 @@ __pycache__
*.ckpt *.ckpt
*.pth *.pth
types types
models models
jsconfig.json

View File

@ -1,5 +1,5 @@
import { app } from "../../../scripts/app.js"; import { app } from "../../../scripts/app.js";
import { ComfyWidgets } from '../../../scripts/widgets.js';
//based on diffus3's SetGet: https://github.com/diffus3/ComfyUI-extensions //based on diffus3's SetGet: https://github.com/diffus3/ComfyUI-extensions
// Nodes that allow you to tunnel connections for cleaner graphs // Nodes that allow you to tunnel connections for cleaner graphs
@ -243,18 +243,10 @@ app.registerExtension({
content: `${getter.title} id: ${getter.id}`, content: `${getter.title} id: ${getter.id}`,
callback: () => { callback: () => {
if (this.canvas?.ds?.offset) { this.canvas.centerOnNode(getter);
const nodeCenterX = getter.pos[0] + (getter.size[0] / 2); this.canvas.selectNode(getter, false);
const nodeCenterY = getter.pos[1] + (getter.size[1] / 2);
this.canvas.ds.offset[0] = -nodeCenterX + this.canvas.mouse[0];
this.canvas.ds.offset[1] = -nodeCenterY + this.canvas.mouse[1];
}
if (this.canvas?.ds?.scale != null) {
this.canvas.ds.scale = Number(1);
}
this.canvas.selectNode(getter, false)
this.canvas.setDirty(true, true); this.canvas.setDirty(true, true);
}, },
})); }));
@ -437,19 +429,9 @@ app.registerExtension({
}; };
this.goToSetter = function() { this.goToSetter = function() {
const setter = this.findSetter(this.graph); const setter = this.findSetter(this.graph);
if (this.canvas?.ds?.offset) { this.canvas.centerOnNode(setter);
const nodeCenterX = setter.pos[0] + (setter.size[0] / 2); this.canvas.selectNode(setter, false);
const nodeCenterY = setter.pos[1] + (setter.size[1] / 2);
this.canvas.ds.offset[0] = -nodeCenterX + this.canvas.mouse[0];
this.canvas.ds.offset[1] = -nodeCenterY + this.canvas.mouse[1];
}
if (this.canvas?.ds?.scale != null) {
this.canvas.ds.scale = Number(1);
}
this.canvas.selectNode(setter, false)
this.canvas.setDirty(true, true);
}; };
// This node is purely frontend and does not impact the resulting prompt so should not be serialized // This node is purely frontend and does not impact the resulting prompt so should not be serialized