junmakii_image_util_keep_aspect_ratio.py

"""A image utility function to keep image aspect ratio.
"""


def get_size(im_size, size):
    """
    :rtype: (int, int)

    >>> get_size((360, 100), (360, 180))
    (648, 180)

    >>> get_size((100, 360), (360, 180))
    (360, 1296)

    >>> get_size((512, 512), (640, 360))
    (640, 640)
    """
    if im_size[0] == im_size[1]:
        return (max(size), max(size))
    elif im_size[0] > im_size[1]:
        width = size[1] * im_size[0] / im_size[1]
        return (width, size[1])
    else:
        height = size[0] * im_size[1] / im_size[0]
        return (size[0], height)