Ej;dZddlmZmZmZmZmZmZmZddl Z ddl Z ddl m Z mZmZmZddlmZmZddlmZGdd eZy) zRead/Write images using rawpy. rawpy is an easy-to-use Python wrapper for the LibRaw library. It also contains some extra functionality for finding and repairing hot/dead pixels. )AnyDictIteratorListOptionalTupleUnionN) URI_BYTESInitializationErrorIOModeRequest)ImagePropertiesPluginV3) ArrayLikec eZdZdZdeddffd ZddZddd edejfd Z d e e e e fdeefd Zdeejfd Z dd ededeeeffdZdd edefdZxZS) RawPyPluginzA class representing the rawpy plugin. Methods ------- .. autosummary:: :toctree: _plugins/rawpy RawPyPlugin.read requestreturnNct||d|_|jjt j k(r* tj|j|_y|jjt j"k(r tddy#tjtjtjf$r:|jtk(r tddtd|j ddwxYw)zInstantiates a new rawpy plugin object Parameters ---------- request: Request A request object representing the resource to be operated on. Nz&RawPy can not read the provided bytes.zRawPy can not read .RawPy does not support writing.)super__init__ _image_filemodeio_moder readrawpyimreadget_fileNotSupportedErrorLibRawFileUnsupportedError LibRawIOError _uri_typer r raw_uriwrite)selfr __class__s S/root/workspace/ytshorts/venv/lib/python3.12/site-packages/imageio/plugins/rawpy.pyrzRawPyPlugin.__init__s ! <<  6;; . #(<<0@0@0B#C \\ ! !V\\ 1%&GHd R2''00## $$ 1-@ .-goo->a@  s (BA,D c|jr|jj|jjyN)rclose_requestfinishr(s r*r-zRawPyPlugin.close<s-       " " $ r)indexr2c x |jjdi|}|turd}S#t$rYwxYw)znRead Raw Image. Returns ------- nd_image: ndarray The image data )N.)r postprocess ExceptionEllipsis)r(r2kwargsnd_images r*rzRawPyPlugin.readBsO 3t''33=f=H H  *H    s - 99ndimagect)r)NotImplementedError)r(r:s r*r'zRawPyPlugin.writeWs !##r1c#PK |jy#t$rYywxYww)znLoad the image. Returns ------- nd_image: ndarray The image data N)rr6r0s r*iterzRawPyPlugin.iter[s( ))+    s&& #&#&exclude_appliedcdi}|jj}|jj|d<|jj|d<|jj|d<|jj |d<|jj |d<|jjj|d<|j|d<|jj|d<|jj|d <|j|d <|j|d <|j|d <|j|d <|jjj |d<|j"|d<|j$|d<|j&|d<|jj(|d<|r|j+dd|j+dd|j+dd|j+dd|j+dd|j+dd|j+dd|j+dd|j+d d|j+d d|j+d d|j+dd|j+dd|j+dd|j+dd|S)aRead ndimage metadata. Parameters ---------- exclude_applied : bool If True, exclude metadata fields that are applied to the image while reading. For example, if the binary data contains a rotation flag, the image is rotated by default and the rotation flag is excluded from the metadata to avoid confusion. Returns ------- metadata : dict A dictionary of format-specific metadata. black_level_per_channelcamera_white_level_per_channel color_desc color_matrixdaylight_whitebalancedtypeflip num_colors tone_curvewidthheight raw_width raw_height raw_shapeiwidthiheight pixel_aspect white_levelN)rsizesrArBrCrDrE raw_imagerFrGrHrIrJrKrLrMshaperOrPrQrRpop)r(r2r?metadata image_sizes r*rWzRawPyPlugin.metadataise(%%++ .2.>.>.V.V*+    ; ; 12"&!1!1!">  LL2D 9 LL94 @ LLt , LL . LL0$ 7 LL$ ' LL & LLt , LLt , LLd + LLt , LLd + LL4 ( LLD ) LL -r1c|jj}|j}|j}||f}|jjj }t ||S)aStandardized ndimage metadata Returns ------- properties : ImageProperties A dataclass filled with standardized image metadata. Notes ----- This does not decode pixel data and is fast for large images. )rUrF)rrSrJrKrTrFr)r(r2 ImageSizerJrKrUrFs r* propertieszRawPyPlugin.propertiessV$$** __&&"(%  **00U%88r1)rN)NTr,)__name__ __module__ __qualname____doc__rrr-intnpndarrayrr rrrbytesr'rr>boolrstrrrWrr[ __classcell__)r)s@r*rrs SSDS@ $%S2::*$U9d9o#=>$8E?$ hrzz* :>>>26> c3h>@999r1r)r_typingrrrrrrr rnumpyra core.requestr r r rcore.v3_plugin_apirrrrr4r1r*rks6 EDD JJ:o9(o9r1