o
    -i                  	   @   sv   d dl mZ 	ddejdededejfddZ		ddejd
eeeef ee B dejfddZdejdefddZ	dS )    )Imageimagesize_factor	transposereturnc                 C   sF   t | j| }t | j| }| ||f} |dkr!| t|} | S )z8Rescale the dimensions of an image by a constant factor.r   )intwidthheightresizer   r   	Transpose)r   r   r   	new_width
new_height r   R/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/vllm/multimodal/image.pyrescale_image_size   s   r      r   r   background_colorc                 C   s8   | j dksJ td| j|}|j| |  d d |S )z:Convert an RGBA image to RGB with filled background color.RGBARGB   )mask)moder   newsizepastesplit)r   r   	convertedr   r   r   rgba_to_rgb   s   r   to_modec                 C   s2   | j |kr| S | j dkr|dkrt| S | |S )Nr   r   )r   r   convert)r   r    r   r   r   convert_image_mode   s
   

r"   N)r   )r   )
PILr   floatr   r   tuplelistr   strr"   r   r   r   r   <module>   s(   

