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

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

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