37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
"""ComfyUI-CXN custom nodes entry point.
|
|
|
|
ComfyUI imports this as a package, so the relative imports below resolve. When
|
|
the directory is imported outside ComfyUI (e.g. pytest collecting the repo, where
|
|
no parent package is established) the imports are skipped and empty mappings are
|
|
exposed. Import errors raised *while* loading as a real package still propagate,
|
|
so genuine node bugs surface in ComfyUI instead of being silently swallowed.
|
|
"""
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
if __package__:
|
|
from .cxn.load_fbx import LoadFBX
|
|
from .cxn.load_glb import LoadGLB
|
|
from .cxn.render_fbx import RenderFBX
|
|
from .cxn.match_foot_pose import MatchFootPose
|
|
from .cxn.image_difference import ImageDifference
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"CXN_LoadFBX": LoadFBX,
|
|
"CXN_LoadGLB": LoadGLB,
|
|
"CXN_RenderFBX": RenderFBX,
|
|
"CXN_MatchFootPose": MatchFootPose,
|
|
"CXN_ImageDifference": ImageDifference,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"CXN_LoadFBX": "Load FBX Model",
|
|
"CXN_LoadGLB": "Load GLB Model",
|
|
"CXN_RenderFBX": "Render FBX",
|
|
"CXN_MatchFootPose": "Match Foot Pose (SAM3D Body)",
|
|
"CXN_ImageDifference": "CXN Image Difference",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|