2?j;dZddlmZmZmZmZmZmZmZddl Z ddl Z ddl m Z mZmZmZddlmZmZddlmZGdd eZdS) 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 edej fd Z d e e e e fdeefd Zdeej fd Z dd ededeeeffdZdd edefdZxZS) RawPyPluginzA class representing the rawpy plugin. Methods ------- .. autosummary:: :toctree: _plugins/rawpy RawPyPlugin.read requestreturnNct|d|_|jjt jkr tj| |_dS#tj tj tj f$r:|j tkrtddtd|jddwxYw|jjt jkrtdddS)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 FC:\Projects\Historical\venv\Lib\site-packages\imageio/plugins/rawpy.pyrzRawPyPlugin.__init__s  !!! < 6; . . #(<0@0@0B0B#C#C   '0# $ 11-@ .@go@@@  \ !V\ 1 1%&GHHd R2 1s +A22A Ccx|jr|j|jdSN)rclose_requestfinishr(s r*r-zRawPyPlugin.close<s<   %   " " $ $ $ r)indexr2c p |jjdi|}n#t$rYnwxYw|tur|d}|S)znRead Raw Image. Returns ------- nd_image: ndarray The image data )N.)r postprocess ExceptionEllipsis)r(r2kwargsnd_images r*rzRawPyPlugin.readBs` 3t'3==f==HH    D  H   *Hs  ""ndimagect)r)NotImplementedError)r(r:s r*r'zRawPyPlugin.writeWs!###r1c#ZK |VdS#t$rYdSwxYw)znLoad the image. Returns ------- nd_image: ndarray The image data N)rr6r0s r*iterzRawPyPlugin.iter[sG ))++         DD s  **Texclude_appliedci}|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<|rJ|dd|dd|dd|dd|dd|dd|dd|dd|d d|d d|d d|dd|dd|dd|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*+   ; 12"&!1!<#'#3#@ ,0,<,R() ,6<%?!%!1!<!%!1!<&,'. * 4!+!6 $ 0 : @'.(0#-#: "&"2">  . LL2D 9 9 9 LL94 @ @ @ LLt , , , LL . . . LL0$ 7 7 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.propertiessH$* _&"(% *0U%8888r1)rN)NTr,)__name__ __module__ __qualname____doc__rrr-intnpndarrayrr rrrbytesr'rr>boolrstrrrWrr[ __classcell__)r)s@r*rrsa  SSDSSSSSS@ $%S2:*$U9d9o#=>$8E?$$$$ hrz*    :>>>>26> c3h>>>>@99999999999r1r)r_typingrrrrrrr rnumpyra core.requestr r r rcore.v3_plugin_apirrrrr4r1r*rks EDDDDDDDDDDDDDDDDD JJJJJJJJJJJJ::::::::o9o9o9o9o9(o9o9o9o9o9r1