neuro_llc/ComfyBox
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

Allow picking index for PickImageNode

Browse Source
This commit is contained in:
space-nuko
2023-05-17 20:59:32 -05:00
parent a9cebc23d1
commit 0cecbee21c
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/lib/nodes/ComfyPickImageNode.ts
View File

@@ -6,6 +6,7 @@ export default class ComfyPickImageNode extends ComfyGraphNode {
static slotLayout: SlotLayout = { static slotLayout: SlotLayout = {
inputs: [ inputs: [
{ name: "images", type: "COMFYBOX_IMAGES,COMFYBOX_IMAGE" }, { name: "images", type: "COMFYBOX_IMAGES,COMFYBOX_IMAGE" },
{ name: "index", type: "number" },
], ],
outputs: [ outputs: [
{ name: "image", type: "COMFYBOX_IMAGE" }, { name: "image", type: "COMFYBOX_IMAGE" },
@@ -35,13 +36,14 @@ export default class ComfyPickImageNode extends ComfyGraphNode {
_path: string | null = null; _path: string | null = null;
_index: number = 0; _index: number = 0;
private setValue(value: ComfyBoxImageMetadata[] | ComfyBoxImageMetadata | null) { private setValue(value: ComfyBoxImageMetadata[] | ComfyBoxImageMetadata | null, index: number) {
if (value != null && !Array.isArray(value)) { if (value != null && !Array.isArray(value)) {
value = [value] value = [value]
this._index = 0; index = 0;
} }
const changed = this._value != value; const changed = this._value != value || this._index != index;
this._value = value as ComfyBoxImageMetadata[]; this._value = value as ComfyBoxImageMetadata[];
this._index = index;
if (changed) { if (changed) {
if (value && value[this._index] != null) { if (value && value[this._index] != null) {
this._image = value[this._index] this._image = value[this._index]
@@ -61,7 +63,8 @@ export default class ComfyPickImageNode extends ComfyGraphNode {
override onExecute() { override onExecute() {
const data = this.getInputData(0) const data = this.getInputData(0)
this.setValue(data); const index = this.getInputData(1)
this.setValue(data, index);
if (this._image == null) { if (this._image == null) {
this.setOutputData(0, null) this.setOutputData(0, null)