import cv2
from PIL import Image
from matplotlib import pyplot as plt


def show_image_cv2(image, maxdim=700):
    h, w, c = image.shape
    maxhw = max(h, w)
    if maxhw > maxdim:
        ratio = maxdim / maxhw
        h = int(h * ratio)
        w = int(w * ratio)
        img = cv2.resize(image, (h, w))
    cv2.imshow("", img)
    cv2.waitKey(0)
    cv2.destroyAllWindows()


def show_image_mpl(image):
    fig, ax = plt.subplots(1, 1)
    fig.set_size_inches(20, 20)
    ax.imshow(image, cmap="gray")
    plt.show()


def show_image(image, backend="mpl"):
    if backend == "mpl":
        show_image_mpl(image)
    elif backend == "cv2":
        show_image_cv2(image)
    elif backend == "pil":
        Image.fromarray(image).show()
    else:
        raise ValueError(f"Unknown backend: {backend}")


def save_image(image, path):
    cv2.imwrite(path, image)