o Si@s^ddlmZmZddlmZddlmZmZddlZ ddl m Z m Z m Z eGdddZdS))asdict dataclass)Path)OptionalTupleN)Pathlikefastcopyis_module_availablec@seZdZUdZeed<eed<eed<eed<eed<edeeeffdd Z ede fd d Z de fd d Z ede ddfddZed!ddZd"de fddZdeddfddZd!ddZddZd#dd ZdS)$Imagea6 The Image manifest describes an image that is stored somewhere: it might be in a file, in memory, in the cloud, etc. Image helps abstract away from the actual storage mechanism and location by providing a method called :meth:`.Image.load`. Image manifest can be easily created by calling :meth:`lhotse.image.io.PillowWriter.store_image`, for example:: >>> from lhotse.image.io import PillowWriter >>> with PillowWriter('images/') as writer: ... manifest = writer.store_image('image-1', 'path/to/image.jpg') storage_type storage_path storage_keywidthheightreturncCs |j|jfS)z1Return the shape of the image as (height, width).)rrselfrF/home/ubuntu/.local/lib/python3.10/site-packages/lhotse/image/image.pyshape*s z Image.shapecCsddlm}||jS)z'Check if the image is stored in memory.r) is_in_memory)lhotse.image.iorr )rrrrrr/s  zImage.is_in_memorycCst|S)z(Convert to dictionary for serialization.)rrrrrto_dict7sz Image.to_dictdatacCs.d|vrd|vrd|vrd|d<|di|S)z1Create an Image from a dictionary representation.r r r Nrr)clsrrrr from_dict;s zImage.from_dictcCsddlm}|d|S)Nr)PillowInMemoryWriter)rr store_image)rrrrrr from_pillowFs zImage.from_pillowF as_pil_imagecCs@tdsJdddlm}||j|j}|j|j|d}|S)a' Load the image from the underlying storage. :param as_numpy: If True (default), returns a numpy array with shape (height, width, channels). If False, returns a PIL Image object. :return: Loaded image as numpy array or PIL Image object. PILz8In order to load images, please run 'pip install pillow'r) get_reader)r )r rr"r r readr )rr r"storageimg_datarrrloadLs z Image.loadpathcCst|tt||jdS)zt Return a copy of the image with ``path`` added as a prefix to the ``storage_path`` member. )r )rstrrr )rr'rrrwith_path_prefix`szImage.with_path_prefixcCsPddlm}|jdvr |S|}|d}|d|}t|j|d|j|jdS)z> Return a copy of the image stored in memory. r)get_memory_writer pillow_memoryr)r r r rr) rr*r r&writer namerr)rr*imgwriterrrrrmove_to_memorygs    zImage.move_to_memoryc Cs>d|jd|jdt|jtr|jndd|jd|jd S)NzImage(storage_type='z', storage_path='z', storage_key='z z ', width=z , height=))r r isinstancer r(rrrrrr__repr__|szImage.__repr__NcKsXtdstdddlm}|}|dur|\}}|j|fi||d|S)a/ Display the image using matplotlib. :param ax: Optional matplotlib Axes to plot on. If None, a new figure and axes will be created. :param kwargs: Additional keyword arguments passed to `ax.imshow()`. :return: The matplotlib Axes object with the plotted image. matplotlibz;To use Image.plot(), please 'pip install matplotlib' first.rNoff)r ImportErrormatplotlib.pyplotpyplotr&subplotsimshowaxis)raxkwargspltr%figrrrplots   z Image.plot)rr )F)N)__name__ __module__ __qualname____doc__r(__annotations__intpropertyrrboolrdictr classmethodrrr&rr)r0r3r@rrrrr s*    r ) dataclassesrrpathlibrtypingrrnumpynp lhotse.utilsrrr r rrrrs