44 lines
1.7 KiB
Python
44 lines
1.7 KiB
Python
from nodes import PreviewImage
|
|
|
|
|
|
class CompareFrames(PreviewImage):
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"compare_frame": ("INT", {"default": 0, "min": 0, "max": 99999, "step": 1}),
|
|
"skip_a": ("INT", {"default": 0, "min": -99999, "max": 99999, "step": 1}),
|
|
"skip_b": ("INT", {"default": 0, "min": -99999, "max": 99999, "step": 1}),
|
|
"fps": ("FLOAT", {"default": 12.0, "min": 0.1, "max": 60.0, "step": 0.1}),
|
|
},
|
|
"optional": {
|
|
"images_a": ("IMAGE",),
|
|
"images_b": ("IMAGE",),
|
|
},
|
|
"hidden": {
|
|
"prompt": "PROMPT",
|
|
"extra_pnginfo": "EXTRA_PNGINFO"
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ()
|
|
OUTPUT_NODE = True
|
|
FUNCTION = "compare_frames"
|
|
CATEGORY = "preview"
|
|
|
|
def compare_frames(self, compare_frame, skip_a, skip_b, fps=12.0, images_a=None, images_b=None,
|
|
prompt=None, extra_pnginfo=None, filename_prefix="temp.compare_frames."):
|
|
result = {"ui": {"a_images": [], "b_images": [], "fps": [fps]}}
|
|
|
|
print(f"CompareFrames: A={len(images_a) if images_a is not None else 0} frames, "
|
|
f"B={len(images_b) if images_b is not None else 0} frames, fps={fps}")
|
|
|
|
if images_a is not None:
|
|
res_a = self.save_images(images_a, filename_prefix, prompt, extra_pnginfo)
|
|
result["ui"]["a_images"] = res_a["ui"]["images"]
|
|
|
|
if images_b is not None:
|
|
res_b = self.save_images(images_b, filename_prefix, prompt, extra_pnginfo)
|
|
result["ui"]["b_images"] = res_b["ui"]["images"]
|
|
|
|
return result |