| 613 | |
| 614 | |
| 615 | class IpaConfigurationNode: |
| 616 | WEIGHT_TYPES = ["linear", "ease in", "ease out", 'ease in-out', 'reverse in-out', 'weak input', 'weak output', 'weak middle', 'strong middle'] |
| 617 | IPA_EMBEDS_SCALING_OPTIONS = ["V only", "K+V", "K+V w/ C penalty", "K+mean(V) w/ C penalty"] |
| 618 | |
| 619 | @classmethod |
| 620 | def INPUT_TYPES(cls): |
| 621 | return { |
| 622 | "required": { |
| 623 | "ipa_starts_at": ("FLOAT", {"default": 0.0, "min": 0.0, "max": 1.0, "step": 0.01}), |
| 624 | "ipa_ends_at": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01}), |
| 625 | "ipa_weight_type": (cls.WEIGHT_TYPES,), |
| 626 | "ipa_weight": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 2.0, "step": 0.01}), |
| 627 | "ipa_embeds_scaling": (cls.IPA_EMBEDS_SCALING_OPTIONS,), |
| 628 | "ipa_noise_strength": ("FLOAT", {"default": 0.3, "min": 0.0, "max": 1.0, "step": 0.01}), |
| 629 | "use_image_for_noise": ("BOOLEAN", {"default": False}), |
| 630 | "type_of_noise": (["fade", "dissolve", "gaussian", "shuffle"], ), |
| 631 | "noise_blur": ("INT", { "default": 0, "min": 0, "max": 32, "step": 1 }), |
| 632 | }, |
| 633 | "optional": {} |
| 634 | } |
| 635 | |
| 636 | FUNCTION = "process_inputs" |
| 637 | RETURN_TYPES = ("ADVANCED_IPA_SETTINGS",) |
| 638 | RETURN_NAMES = ("configuration",) |
| 639 | CATEGORY = "Steerable-Motion" |
| 640 | |
| 641 | @classmethod |
| 642 | def process_inputs(cls, ipa_starts_at, ipa_ends_at, ipa_weight_type, ipa_weight, ipa_embeds_scaling, ipa_noise_strength, use_image_for_noise, type_of_noise, noise_blur): |
| 643 | return { |
| 644 | "ipa_starts_at": ipa_starts_at, |
| 645 | "ipa_ends_at": ipa_ends_at, |
| 646 | "ipa_weight_type": ipa_weight_type, |
| 647 | "ipa_weight": ipa_weight, |
| 648 | "ipa_embeds_scaling": ipa_embeds_scaling, |
| 649 | "ipa_noise_strength": ipa_noise_strength, |
| 650 | "use_image_for_noise": use_image_for_noise, |
| 651 | "type_of_noise": type_of_noise, |
| 652 | "noise_blur": noise_blur, |
| 653 | }, |
| 654 | |
| 655 | class VideoFrameExtractorAndMaskGenerator: |
| 656 | @classmethod |
nothing calls this directly
no outgoing calls
no test coverage detected