from .nodes.crop_and_resize_nodes import (
    SmartImageResize,
    ResizeToMultiple,
    ExtendCanvasByPercentage,
    ImageCropByPercentage,
    RemovePadding,
)
from .nodes.mask_nodes import MaskCropByPercentage, ApplyMaskToImage
from .nodes.extractor_nodes import ImageResolutionExtractor, ImageAspectRatioExtractor

NODE_CLASS_MAPPINGS = {
    "ImageCropByPercentage": ImageCropByPercentage,
    "MaskCropByPercentage": MaskCropByPercentage,
    "ExtendCanvasByPercentage": ExtendCanvasByPercentage,
    "ResizeToMultiple": ResizeToMultiple,
    "ImageResolutionExtractor": ImageResolutionExtractor,
    "ImageAspectRatioExtractor": ImageAspectRatioExtractor,
    "SmartImageResize": SmartImageResize,
    "RemovePadding": RemovePadding,
    "ApplyMaskToImage": ApplyMaskToImage,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "ImageCropByPercentage": "Image Crop By Percentage",
    "MaskCropByPercentage": "Mask Crop By Percentage",
    "ExtendCanvasByPercentage": "Extend Canvas By Percentage",
    "ResizeToMultiple": "Resize To Multiple",
    "ImageResolutionExtractor": "Image Resolution Extractor",
    "ImageAspectRatioExtractor": "Image Aspect Ratio Extractor",
    "SmartImageResize": "Smart Image Resize",
    "RemovePadding": "Remove Padding",
    "ApplyMaskToImage": "Apply Mask To Image",
}