Difforum/__init__.py at main · chillithebillis/Difforum · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
"""
Difforum - a modern, Deforum-style animation toolkit for ComfyUI.

Phase 1 ships the GPU-free schedule engine (keyframe parser + safe math
evaluator + audio-reactive hooks). Later phases add depth warping, the
feedback sampler (Classic+ mode) and the Wan 2.2 VACE video bridge
(Hybrid mode). See DESIGN.md.
"""

from .nodes.schedule_nodes import (
    NODE_CLASS_MAPPINGS as _SCHED_CLASSES,
    NODE_DISPLAY_NAME_MAPPINGS as _SCHED_NAMES,
)
from .nodes.audio_nodes import (
    NODE_CLASS_MAPPINGS as _AUDIO_CLASSES,
    NODE_DISPLAY_NAME_MAPPINGS as _AUDIO_NAMES,
)
from .nodes.hybrid_nodes import (
    NODE_CLASS_MAPPINGS as _HYBRID_CLASSES,
    NODE_DISPLAY_NAME_MAPPINGS as _HYBRID_NAMES,
)
from .nodes.catalog_nodes import (
    NODE_CLASS_MAPPINGS as _CATALOG_CLASSES,
    NODE_DISPLAY_NAME_MAPPINGS as _CATALOG_NAMES,
)
from .nodes.warp_nodes import (
    NODE_CLASS_MAPPINGS as _WARP_CLASSES,
    NODE_DISPLAY_NAME_MAPPINGS as _WARP_NAMES,
)
from .nodes.sampler_nodes import (
    NODE_CLASS_MAPPINGS as _SAMPLER_CLASSES,
    NODE_DISPLAY_NAME_MAPPINGS as _SAMPLER_NAMES,
)
from .nodes.guide_nodes import (
    NODE_CLASS_MAPPINGS as _GUIDE_CLASSES,
    NODE_DISPLAY_NAME_MAPPINGS as _GUIDE_NAMES,
)
from .nodes.prompt_nodes import (
    NODE_CLASS_MAPPINGS as _PROMPT_CLASSES,
    NODE_DISPLAY_NAME_MAPPINGS as _PROMPT_NAMES,
)
from .nodes.live_nodes import (
    NODE_CLASS_MAPPINGS as _LIVE_CLASSES,
    NODE_DISPLAY_NAME_MAPPINGS as _LIVE_NAMES,
)
from .nodes.effects_nodes import (
    NODE_CLASS_MAPPINGS as _EFFECTS_CLASSES,
    NODE_DISPLAY_NAME_MAPPINGS as _EFFECTS_NAMES,
)
from .nodes.look_nodes import (
    NODE_CLASS_MAPPINGS as _LOOK_CLASSES,
    NODE_DISPLAY_NAME_MAPPINGS as _LOOK_NAMES,
)
from .nodes.glitch_nodes import (
    NODE_CLASS_MAPPINGS as _GLITCH_CLASSES,
    NODE_DISPLAY_NAME_MAPPINGS as _GLITCH_NAMES,
)
from .nodes.video_nodes import (
    NODE_CLASS_MAPPINGS as _VIDEO_CLASSES,
    NODE_DISPLAY_NAME_MAPPINGS as _VIDEO_NAMES,
)

NODE_CLASS_MAPPINGS = {}
NODE_DISPLAY_NAME_MAPPINGS = {}

NODE_CLASS_MAPPINGS.update(_SCHED_CLASSES)
NODE_DISPLAY_NAME_MAPPINGS.update(_SCHED_NAMES)
NODE_CLASS_MAPPINGS.update(_AUDIO_CLASSES)
NODE_DISPLAY_NAME_MAPPINGS.update(_AUDIO_NAMES)
NODE_CLASS_MAPPINGS.update(_HYBRID_CLASSES)
NODE_DISPLAY_NAME_MAPPINGS.update(_HYBRID_NAMES)
NODE_CLASS_MAPPINGS.update(_CATALOG_CLASSES)
NODE_DISPLAY_NAME_MAPPINGS.update(_CATALOG_NAMES)
NODE_CLASS_MAPPINGS.update(_WARP_CLASSES)
NODE_DISPLAY_NAME_MAPPINGS.update(_WARP_NAMES)
NODE_CLASS_MAPPINGS.update(_SAMPLER_CLASSES)
NODE_DISPLAY_NAME_MAPPINGS.update(_SAMPLER_NAMES)
NODE_CLASS_MAPPINGS.update(_GUIDE_CLASSES)
NODE_DISPLAY_NAME_MAPPINGS.update(_GUIDE_NAMES)
NODE_CLASS_MAPPINGS.update(_PROMPT_CLASSES)
NODE_DISPLAY_NAME_MAPPINGS.update(_PROMPT_NAMES)
NODE_CLASS_MAPPINGS.update(_LIVE_CLASSES)
NODE_DISPLAY_NAME_MAPPINGS.update(_LIVE_NAMES)
NODE_CLASS_MAPPINGS.update(_EFFECTS_CLASSES)
NODE_DISPLAY_NAME_MAPPINGS.update(_EFFECTS_NAMES)
NODE_CLASS_MAPPINGS.update(_LOOK_CLASSES)
NODE_DISPLAY_NAME_MAPPINGS.update(_LOOK_NAMES)
NODE_CLASS_MAPPINGS.update(_VIDEO_CLASSES)
NODE_DISPLAY_NAME_MAPPINGS.update(_VIDEO_NAMES)
NODE_CLASS_MAPPINGS.update(_GLITCH_CLASSES)
NODE_DISPLAY_NAME_MAPPINGS.update(_GLITCH_NAMES)

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]

print(f"[Difforum] loaded {len(NODE_CLASS_MAPPINGS)} nodes")