nuke-nodes-comfyui/__init__.py at main · sumitchatterjee13/nuke-nodes-comfyui · 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
"""
Nuke Nodes for ComfyUI
A collection of custom nodes that replicate Nuke compositing functionality
"""

import os

# Web directory for JavaScript extensions (preview widgets, etc.)
WEB_DIRECTORY = os.path.join(os.path.dirname(os.path.realpath(__file__)), "web")

from .blur_nodes import *
from .colorspace_nodes import *
from .grade_nodes import *
from .io_nodes import *
from .merge_nodes import *
from .multipass_nodes import *
from .transform_nodes import *
from .vectorfield_nodes import *

# Import version information
from .version import (
    __author__,
    __description__,
    __license__,
    __title__,
    __url__,
    __version__,
    get_full_version,
    get_version,
    get_version_info,
)
from .viewer_nodes import *

# Node mappings for ComfyUI
NODE_CLASS_MAPPINGS = {}
NODE_DISPLAY_NAME_MAPPINGS = {}

# Import all node classes and their mappings
from .blur_nodes import NODE_CLASS_MAPPINGS as blur_mappings
from .blur_nodes import NODE_DISPLAY_NAME_MAPPINGS as blur_display_mappings
from .colorspace_nodes import NODE_CLASS_MAPPINGS as colorspace_mappings
from .colorspace_nodes import NODE_DISPLAY_NAME_MAPPINGS as colorspace_display_mappings
from .grade_nodes import NODE_CLASS_MAPPINGS as grade_mappings
from .grade_nodes import NODE_DISPLAY_NAME_MAPPINGS as grade_display_mappings
from .io_nodes import NODE_CLASS_MAPPINGS as io_mappings
from .io_nodes import NODE_DISPLAY_NAME_MAPPINGS as io_display_mappings
from .merge_nodes import NODE_CLASS_MAPPINGS as merge_mappings
from .merge_nodes import NODE_DISPLAY_NAME_MAPPINGS as merge_display_mappings
from .multipass_nodes import NODE_CLASS_MAPPINGS as multipass_mappings
from .multipass_nodes import NODE_DISPLAY_NAME_MAPPINGS as multipass_display_mappings
from .transform_nodes import NODE_CLASS_MAPPINGS as transform_mappings
from .transform_nodes import NODE_DISPLAY_NAME_MAPPINGS as transform_display_mappings
from .vectorfield_nodes import NODE_CLASS_MAPPINGS as vectorfield_mappings
from .vectorfield_nodes import (
    NODE_DISPLAY_NAME_MAPPINGS as vectorfield_display_mappings,
)
from .viewer_nodes import NODE_CLASS_MAPPINGS as viewer_mappings
from .viewer_nodes import NODE_DISPLAY_NAME_MAPPINGS as viewer_display_mappings

# Combine all mappings
NODE_CLASS_MAPPINGS.update(blur_mappings)
NODE_CLASS_MAPPINGS.update(colorspace_mappings)
NODE_CLASS_MAPPINGS.update(grade_mappings)
NODE_CLASS_MAPPINGS.update(io_mappings)
NODE_CLASS_MAPPINGS.update(merge_mappings)
NODE_CLASS_MAPPINGS.update(multipass_mappings)
NODE_CLASS_MAPPINGS.update(transform_mappings)
NODE_CLASS_MAPPINGS.update(vectorfield_mappings)
NODE_CLASS_MAPPINGS.update(viewer_mappings)

NODE_DISPLAY_NAME_MAPPINGS.update(blur_display_mappings)
NODE_DISPLAY_NAME_MAPPINGS.update(colorspace_display_mappings)
NODE_DISPLAY_NAME_MAPPINGS.update(grade_display_mappings)
NODE_DISPLAY_NAME_MAPPINGS.update(io_display_mappings)
NODE_DISPLAY_NAME_MAPPINGS.update(merge_display_mappings)
NODE_DISPLAY_NAME_MAPPINGS.update(multipass_display_mappings)
NODE_DISPLAY_NAME_MAPPINGS.update(transform_display_mappings)
NODE_DISPLAY_NAME_MAPPINGS.update(vectorfield_display_mappings)
NODE_DISPLAY_NAME_MAPPINGS.update(viewer_display_mappings)

__all__ = [
    "NODE_CLASS_MAPPINGS",
    "NODE_DISPLAY_NAME_MAPPINGS",
    "WEB_DIRECTORY",
    "__version__",
    "__title__",
    "__description__",
    "__author__",
    "__license__",
    "__url__",
    "get_version",
    "get_version_info",
    "get_full_version",
]