from .image_split import SplitImage_Tiled, CalculateTilingSpec_Tiled
from .image_stitch import StitchImage_Tiled
from .image_batch import BatchImages_Tiled
from .image_latent import SplitLatent_Tiled

NODE_CLASS_MAPPINGS = {
    "SplitImage_Tiled": SplitImage_Tiled,
    "StitchImage_Tiled": StitchImage_Tiled,
    "BatchImages_Tiled": BatchImages_Tiled,
    "CalculateTilingSpec_Tiled": CalculateTilingSpec_Tiled,
    "SplitLatent_Tiled": SplitLatent_Tiled
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "SplitImage_Tiled": "Split Image into Tiles",
    "StitchImage_Tiled": "Stitch Image Tiles",
    "BatchImages_Tiled": "Merge Image Batches",
    "CalculateTilingSpec_Tiled": "Calculate Tiling Spec",
    "SplitLatent_Tiled": "Split Latent into Tiles"
}

__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']