| |
| from .categories import NodeCategories |
| from .shared import ALWAYS_CHANGED_FLAG, list_images_in_directory, DreamImage |
| from .dreamtypes import SharedTypes, FrameCounter |
| import os |
|
|
|
|
| class DreamImageSequenceInputWithDefaultFallback: |
| NODE_NAME = "Image Sequence Loader" |
| ICON = "💾" |
|
|
| @classmethod |
| def INPUT_TYPES(cls): |
| return { |
| "required": SharedTypes.frame_counter | { |
| "directory_path": ("STRING", {"default": '', "multiline": False}), |
| "pattern": ("STRING", {"default": '*', "multiline": False}), |
| "indexing": (["numeric", "alphabetic order"],) |
| }, |
| "optional": { |
| "default_image": ("IMAGE", {"default": None}) |
| } |
| } |
|
|
| CATEGORY = NodeCategories.IMAGE_ANIMATION |
| RETURN_TYPES = ("IMAGE","STRING") |
| RETURN_NAMES = ("image","frame_name") |
| FUNCTION = "result" |
|
|
| @classmethod |
| def IS_CHANGED(cls, *values): |
| return ALWAYS_CHANGED_FLAG |
|
|
| def result(self, frame_counter: FrameCounter, directory_path, pattern, indexing, **other): |
| default_image = other.get("default_image", None) |
| entries = list_images_in_directory(directory_path, pattern, indexing == "alphabetic order") |
| entry = entries.get(frame_counter.current_frame, None) |
| if not entry: |
| return (default_image, "") |
| else: |
| image_names = [os.path.basename(file_path) for file_path in entry] |
| images = map(lambda f: DreamImage(file_path=f), entry) |
| return (DreamImage.join_to_tensor_data(images), image_names[0]) |
|
|
