from .nodes import (
    ApplyCircularConvPaddingModel,
    ApplyCircularConvPaddingVAE,
    C2EMaskedDiffNode,
    C2ENode,
    CreatePoleMask,
    CreateSeamMask,
    Crop360To180Node,
    CropImageWithCoordsNode,
    E2CNode,
    E2ENode,
    E2Face,
    E2PNode,
    Face2E,
    MonoScopicToStereoNode,
    Pad180To360Node,
    PasteImageWithCoordsNode,
    RollImageNode,
    SplitFacesNode,
    StackFacesNode,
    StereoToMonoScopicNode,
)

NODE_CLASS_MAPPINGS = {
    "Cubemap to Equirectangular": C2ENode,
    "Equirectangular to Cubemap": E2CNode,
    "Equirectangular to Perspective": E2PNode,
    "Equirectangular Rotation": E2ENode,
    "Split Cubemap Faces": SplitFacesNode,
    "Stack Cubemap Faces": StackFacesNode,
    "Roll Image Axes": RollImageNode,
    "Masked Diff C2E": C2EMaskedDiffNode,
    "Crop Image with Coords": CropImageWithCoordsNode,
    "Paste Image with Coords": PasteImageWithCoordsNode,
    "Pad 180 to 360 Equirectangular": Pad180To360Node,
    "Crop 360 to 180 Equirectangular": Crop360To180Node,
    "Crop Stereo to Monoscopic": StereoToMonoScopicNode,
    "Merge Monoscopic into Stereo": StereoToMonoScopicNode,
    "Apply Circular Padding VAE": ApplyCircularConvPaddingVAE,
    "Apply Circular Padding Model": ApplyCircularConvPaddingModel,
    "Create Seam Mask": CreateSeamMask,
    "Create Pole Mask": CreatePoleMask,
    "Equirectangular to Face": E2Face,
    "Face to Equirectangular": Face2E,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "Cubemap to Equirectangular": "Cubemap to Equirectangular",
    "Equirectangular to Cubemap": "Equirectangular to Cubemap",
    "Equirectangular to Perspective": "Equirectangular to Perspective",
    "Equirectangular Rotation": "Equirectangular Rotation",
    "Split Cubemap Faces": "Split Cubemap Faces",
    "Stack Cubemap Faces": "Stack Cubemap Faces",
    "Roll Image Axes": "Roll Image Axes",
    "Masked Diff C2E": "Masked Diff C2E",
    "Crop Image with Coords": "Crop Image with Coords",
    "Paste Image with Coords": "Paste Image with Coords",
    "Pad 180 to 360 Equirectangular": "Pad 180 to 360 Equirectangular",
    "Crop 360 to 180 Equirectangular": "Crop 360 to 180 Equirectangular",
    "Crop Stereo to Monoscopic": "Crop Stereo to Monoscopic",
    "Merge Monoscopic into Stereo": "Merge Monoscopic into Stereo",
    "Apply Circular Padding VAE": "Apply Circular Padding VAE",
    "Apply Circular Padding Model": "Apply Circular Padding Model",
    "Create Seam Mask": "Create Seam Mask",
    "Create Pole Mask": "Create Pole Mask",
    "Equirectangular to Face": "Equirectangular to Face",
    "Face to Equirectangular": "Face to Equirectangular",
}