from comfy_api.latest import ComfyExtension, io

from .nodes_ad_settings import (
    AnimateDiffSettingsNode,
    FullStretchPENode,
    ManualAdjustPENode,
    SweetspotStretchPENode,
    WeightAdjustAllAddNode,
    WeightAdjustAllMultNode,
    WeightAdjustIndivAddNode,
    WeightAdjustIndivAttnAddNode,
    WeightAdjustIndivAttnMultNode,
    WeightAdjustIndivMultNode,
)
from .nodes_animatelcmi2v import (
    ApplyAnimateLCMI2VModel,
    LoadAnimateDiffAndInjectI2VNode,
    LoadAnimateLCMI2VModelNode,
    UpscaleAndVaeEncode,
)
from .nodes_cameractrl import (
    ApplyAnimateDiffWithCameraCtrl,
    CameraCtrlADKeyframeNode,
    CameraCtrlManualAppendPose,
    CameraCtrlPoseAdvanced,
    CameraCtrlPoseBasic,
    CameraCtrlPoseCombo,
    CameraCtrlReplaceCameraParameters,
    CameraCtrlSetOriginalAspectRatio,
    LoadAnimateDiffModelWithCameraCtrl,
    LoadCameraPosesFromFile,
    LoadCameraPosesFromPath,
)
from .nodes_conditioning import (
    CombineLoraHookEightOptionalDEPR,
    CombineLoraHookFourOptionalDEPR,
    CombineLoraHooksDEPR,
    ConditioningCombineDEPR,
    ConditioningSetMaskAndCombineHookedDEPR,
    ConditioningSetMaskHookedDEPR,
    ConditioningSetUnmaskedAndCombineHookedDEPR,
    ConditioningTimestepsNodeDEPR,
    CreateLoraHookKeyframeDEPR,
    CreateLoraHookKeyframeFromStrengthListDEPR,
    CreateLoraHookKeyframeInterpolationDEPR,
    MaskableLoraLoaderDEPR,
    MaskableLoraLoaderModelOnlyDEPR,
    MaskableSDModelLoaderDEPR,
    MaskableSDModelLoaderModelOnlyDEPR,
    PairedConditioningCombineDEPR,
    PairedConditioningSetMaskAndCombineHookedDEPR,
    PairedConditioningSetMaskHookedDEPR,
    PairedConditioningSetUnmaskedAndCombineHookedDEPR,
    SetClipLoraHookDEPR,
    SetLoraHookKeyframesDEPR,
    SetModelLoraHookDEPR,
)
from .nodes_context import (
    BatchedContextOptionsNode,
    LegacyLoopedUniformContextOptionsNode,
    LoopedUniformContextOptionsNode,
    LoopedUniformViewOptionsNode,
    StandardStaticContextOptionsNode,
    StandardStaticViewOptionsNode,
    StandardUniformContextOptionsNode,
    StandardUniformViewOptionsNode,
    ViewAsContextOptionsNode,
    VisualizeContextOptionsK,
    VisualizeContextOptionsKAdv,
    VisualizeContextOptionsSCustom,
)
from .nodes_context_extras import (
    ContextExtras_ContextRef,
    ContextExtras_NaiveReuse,
    ContextRef_KeyframeFromListNode,
    ContextRef_KeyframeInterpolationNode,
    ContextRef_KeyframeMultivalNode,
    ContextRef_ModeFirst,
    ContextRef_ModeIndexes,
    ContextRef_ModeSliding,
    ContextRef_TuneAttn,
    ContextRef_TuneAttnAdain,
    NaiveReuse_KeyframeFromListNode,
    NaiveReuse_KeyframeInterpolationNode,
    NaiveReuse_KeyframeMultivalNode,
    SetContextExtrasOnContextOptions,
)
from .nodes_deprecated import (
    AnimateDiffCombineDEPR,
    AnimateDiffLoaderAdvancedDEPR,
    AnimateDiffLoaderDEPR,
    AnimateDiffModelSettingsAdvancedAttnStrengthsDEPR,
    AnimateDiffModelSettingsAdvancedDEPR,
    AnimateDiffModelSettingsDEPR,
    AnimateDiffModelSettingsSimpleDEPR,
    LegacyAnimateDiffLoaderWithContextDEPR,
)
from .nodes_extras import (
    AnimateDiffUnload,
    CheckpointLoaderSimpleWithNoiseSelect,
    EmptyLatentImageLarge,
    PerturbedAttentionGuidanceMultival,
    RescaleCFGMultival,
)
from .nodes_gen1 import AnimateDiffLoaderGen1
from .nodes_gen2 import (
    ADKeyframeNode,
    ApplyAnimateDiffModelBasicNode,
    ApplyAnimateDiffModelNode,
    LoadAnimateDiffModelNode,
    UseEvolvedSamplingNode,
)
from .nodes_lora import AnimateDiffLoraLoader
from .nodes_multival import (
    MultivalConvertToMaskNode,
    MultivalDynamicFloatInputNode,
    MultivalDynamicFloatsNode,
    MultivalDynamicNode,
    MultivalScaledMaskNode,
)
from .nodes_per_block import (
    ADBlockComboNode,
    ADBlockIndivNode,
    PerBlockHighLevelNode,
    PerBlock_SD15_FromFloatsNode,
    PerBlock_SD15_LowLevelNode,
    PerBlock_SD15_MidLevelNode,
    PerBlock_SDXL_FromFloatsNode,
    PerBlock_SDXL_LowLevelNode,
    PerBlock_SDXL_MidLevelNode,
)
from .nodes_pia import (
    ApplyAnimateDiffPIAModel,
    InputPIA_MultivalNode,
    InputPIA_PaperPresetsNode,
    LoadAnimateDiffAndInjectPIANode,
    PIA_ADKeyframeNode,
)
from .nodes_sample import (
    AncestralOptionsNode,
    CFGExtrasPAGNode,
    CFGExtrasPAGSimpleNode,
    CFGExtrasRescaleCFGNode,
    CFGExtrasRescaleCFGSimpleNode,
    CustomCFGKeyframeFromListNode,
    CustomCFGKeyframeInterpolationNode,
    CustomCFGKeyframeNode,
    CustomCFGKeyframeSimpleNode,
    CustomCFGNode,
    CustomCFGSimpleNode,
    FreeInitOptionsNode,
    IterationOptionsNode,
    NoisedImageInjectionNode,
    NoisedImageInjectOptionsNode,
    NoiseLayerAddNode,
    NoiseLayerAddWeightedNode,
    NoiseLayerNormalizedSumNode,
    NoiseLayerReplaceNode,
    SampleSettingsNode,
)
from .nodes_scheduling import (
    AddValuesReplaceNode,
    ConditionExtractionNode,
    FloatToFloatsNode,
    PromptSchedulingLatentsNode,
    PromptSchedulingNode,
    ValueSchedulingLatentsNode,
    ValueSchedulingNode,
)
from .nodes_sigma_schedule import (
    InterpolatedWeightedAverageSigmaScheduleNode,
    RawSigmaScheduleNode,
    SigmaScheduleNode,
    SigmaScheduleToSigmasNode,
    SplitAndCombineSigmaScheduleNode,
    WeightedAverageSigmaScheduleNode,
)


class AnimateDiffExtension(ComfyExtension):
    async def get_node_list(self) -> list[type[io.ComfyNode]]:
        return [
            AnimateDiffLoraLoader,
            SampleSettingsNode,
            ADKeyframeNode,
            MultivalDynamicNode,
            MultivalDynamicFloatInputNode,
            MultivalDynamicFloatsNode,
            MultivalScaledMaskNode,
            MultivalConvertToMaskNode,
            StandardStaticContextOptionsNode,
            StandardUniformContextOptionsNode,
            LoopedUniformContextOptionsNode,
            ViewAsContextOptionsNode,
            BatchedContextOptionsNode,
            LegacyLoopedUniformContextOptionsNode,
            VisualizeContextOptionsK,
            VisualizeContextOptionsKAdv,
            VisualizeContextOptionsSCustom,
            StandardStaticViewOptionsNode,
            StandardUniformViewOptionsNode,
            LoopedUniformViewOptionsNode,
            SetContextExtrasOnContextOptions,
            ContextExtras_ContextRef,
            ContextRef_ModeFirst,
            ContextRef_ModeSliding,
            ContextRef_ModeIndexes,
            ContextRef_TuneAttn,
            ContextRef_TuneAttnAdain,
            ContextRef_KeyframeMultivalNode,
            ContextRef_KeyframeInterpolationNode,
            ContextRef_KeyframeFromListNode,
            ContextExtras_NaiveReuse,
            NaiveReuse_KeyframeMultivalNode,
            NaiveReuse_KeyframeInterpolationNode,
            NaiveReuse_KeyframeFromListNode,
            IterationOptionsNode,
            FreeInitOptionsNode,
            MaskableLoraLoaderDEPR,
            MaskableLoraLoaderModelOnlyDEPR,
            MaskableSDModelLoaderDEPR,
            MaskableSDModelLoaderModelOnlyDEPR,
            CombineLoraHooksDEPR,
            CombineLoraHookFourOptionalDEPR,
            CombineLoraHookEightOptionalDEPR,
            SetLoraHookKeyframesDEPR,
            SetClipLoraHookDEPR,
            CreateLoraHookKeyframeDEPR,
            CreateLoraHookKeyframeInterpolationDEPR,
            CreateLoraHookKeyframeFromStrengthListDEPR,
            SetModelLoraHookDEPR,
            PairedConditioningSetMaskHookedDEPR,
            ConditioningSetMaskHookedDEPR,
            PairedConditioningSetMaskAndCombineHookedDEPR,
            ConditioningSetMaskAndCombineHookedDEPR,
            PairedConditioningSetUnmaskedAndCombineHookedDEPR,
            ConditioningSetUnmaskedAndCombineHookedDEPR,
            PairedConditioningCombineDEPR,
            ConditioningCombineDEPR,
            ConditioningTimestepsNodeDEPR,
            NoiseLayerAddNode,
            NoiseLayerAddWeightedNode,
            NoiseLayerNormalizedSumNode,
            NoiseLayerReplaceNode,
            AnimateDiffSettingsNode,
            SweetspotStretchPENode,
            FullStretchPENode,
            ManualAdjustPENode,
            WeightAdjustAllAddNode,
            WeightAdjustAllMultNode,
            WeightAdjustIndivAddNode,
            WeightAdjustIndivMultNode,
            WeightAdjustIndivAttnAddNode,
            WeightAdjustIndivAttnMultNode,
            CustomCFGSimpleNode,
            CustomCFGNode,
            CustomCFGKeyframeSimpleNode,
            CustomCFGKeyframeNode,
            CustomCFGKeyframeInterpolationNode,
            CustomCFGKeyframeFromListNode,
            CFGExtrasPAGSimpleNode,
            CFGExtrasPAGNode,
            CFGExtrasRescaleCFGSimpleNode,
            CFGExtrasRescaleCFGNode,
            SigmaScheduleNode,
            RawSigmaScheduleNode,
            WeightedAverageSigmaScheduleNode,
            InterpolatedWeightedAverageSigmaScheduleNode,
            SplitAndCombineSigmaScheduleNode,
            SigmaScheduleToSigmasNode,
            NoisedImageInjectionNode,
            NoisedImageInjectOptionsNode,
            AncestralOptionsNode,
            PromptSchedulingNode,
            PromptSchedulingLatentsNode,
            ValueSchedulingNode,
            ValueSchedulingLatentsNode,
            ConditionExtractionNode,
            AddValuesReplaceNode,
            FloatToFloatsNode,
            ADBlockComboNode,
            ADBlockIndivNode,
            PerBlockHighLevelNode,
            PerBlock_SD15_MidLevelNode,
            PerBlock_SD15_LowLevelNode,
            PerBlock_SD15_FromFloatsNode,
            PerBlock_SDXL_MidLevelNode,
            PerBlock_SDXL_LowLevelNode,
            PerBlock_SDXL_FromFloatsNode,
            AnimateDiffUnload,
            EmptyLatentImageLarge,
            CheckpointLoaderSimpleWithNoiseSelect,
            PerturbedAttentionGuidanceMultival,
            RescaleCFGMultival,
            AnimateDiffLoaderGen1,
            UseEvolvedSamplingNode,
            ApplyAnimateDiffModelBasicNode,
            ApplyAnimateDiffModelNode,
            LoadAnimateDiffModelNode,
            ApplyAnimateLCMI2VModel,
            LoadAnimateLCMI2VModelNode,
            UpscaleAndVaeEncode,
            LoadAnimateDiffAndInjectI2VNode,
            ApplyAnimateDiffWithCameraCtrl,
            LoadAnimateDiffModelWithCameraCtrl,
            CameraCtrlADKeyframeNode,
            LoadCameraPosesFromFile,
            LoadCameraPosesFromPath,
            CameraCtrlPoseBasic,
            CameraCtrlPoseCombo,
            CameraCtrlPoseAdvanced,
            CameraCtrlManualAppendPose,
            CameraCtrlReplaceCameraParameters,
            CameraCtrlSetOriginalAspectRatio,
            ApplyAnimateDiffPIAModel,
            InputPIA_MultivalNode,
            InputPIA_PaperPresetsNode,
            PIA_ADKeyframeNode,
            LoadAnimateDiffAndInjectPIANode,
            LegacyAnimateDiffLoaderWithContextDEPR,
            AnimateDiffLoaderDEPR,
            AnimateDiffLoaderAdvancedDEPR,
            AnimateDiffCombineDEPR,
            AnimateDiffModelSettingsDEPR,
            AnimateDiffModelSettingsSimpleDEPR,
            AnimateDiffModelSettingsAdvancedDEPR,
            AnimateDiffModelSettingsAdvancedAttnStrengthsDEPR,
        ]