U Z$dÞbã@sÂdZddlZddlZddlZddlZddlZddlmZddl m Z e  e ¡Z dZdZdZd Zd Zd Zd Zd ZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZdS)a½ Read LFR files (Lytro Illum). Backend: internal Plugin to read Lytro Illum .lfr and .raw files as produced by the Lytro Illum light field camera. It is actually a collection of plugins, each supporting slightly different keyword arguments Parameters ---------- meta_only : bool Whether to only read the metadata. include_thumbnail : bool (only for lytro-lfr and lytro-lfp) Whether to include an image thumbnail in the metadata. éNé)ÚFormat)Úimread)iøi0)éÐ ré éé-é#é éc@s.eZdZdZdZdd„ZGdd„dejƒZdS)Ú LytroFormataBase class for Lytro format. The subclasses LytroLfrFormat, LytroLfpFormat, LytroIllumRawFormat and LytroF01RawFormat implement the Lytro-LFR, Lytro-LFP and Lytro-RAW format for the Illum and original F01 camera respectively. Writing is not supported. ÚicCsdS)NF©©ÚselfÚrequestrrú9/tmp/pip-unpacked-wheel-zwww1ayp/imageio/plugins/lytro.pyÚ _can_writeIszLytroFormat._can_writec@s.eZdZd dd„Zdd„Zdd„Zdd „Zd S) zLytroFormat.WriterrcCs|j ¡|_dS©N)rÚget_fileÚ_fp)rÚflagsrrrÚ_openPszLytroFormat.Writer._opencCsdSrr©rrrrÚ_closeSszLytroFormat.Writer._closecCs tdƒ‚dS)Nz)The lytro format cannot write image data.©Ú RuntimeError)rÚimÚmetarrrÚ _append_dataXszLytroFormat.Writer._append_datacCs tdƒ‚dS)Nz(The lytro format cannot write meta data.r)rrrrrÚ_set_meta_data\sz!LytroFormat.Writer._set_meta_dataN)r)Ú__name__Ú __module__Ú __qualname__rrrr rrrrÚWriterOs r$N)r!r"r#Ú__doc__Z_modesrrr$rrrrr >sr c@s6eZdZdZdd„Zedd„ƒZGdd„dejƒZdS) ÚLytroIllumRawFormata¡This is the Lytro Illum RAW format. The raw format is a 10bit image format as used by the Lytro Illum light field camera. The format will read the specified raw file and will try to load a .txt or .json file with the associated meta data. This format does not support writing. Parameters for reading ---------------------- meta_only : bool Whether to only read the metadata. cCs|jdkrdSdS©N)z.rawT©Ú extensionrrrrÚ _can_readps zLytroIllumRawFormat._can_readcCs’|ddd…}|ddd…}|ddd…}|ddd…}|ddd…}t |d¡t |d¡}t |d¡t t |d¡d¡}t |d¡t t |d¡d¡}t |d¡t t |d ¡d ¡}tjttjd }| tdtddf¡|dd…ddd…f<| tdtddf¡|dd…ddd…f<| tdtddf¡|dd…ddd…f<| tdtddf¡|dd…ddd…f<t |d ¡  tj ¡S) Nréérr rré0éÀé©Zdtypegø@) ÚnpÚ left_shiftÚ bitwise_andÚ right_shiftÚzerosÚLYTRO_ILLUM_IMAGE_SIZEÚuint16ÚreshapeÚdivideÚastypeÚfloat64)ÚarrayÚt0Út1Út2Út3ZlsbÚimagerrrÚrearrange_bitsus.   ÿÿÿÿz"LytroIllumRawFormat.rearrange_bitsc@s6eZdZd dd„Zdd„Zdd„Zdd „Zd d „Zd S)zLytroIllumRawFormat.ReaderFcCs|j ¡|_d|_||_dSr©rrÚ_fileÚ_dataÚ _meta_only©rÚ meta_onlyrrrr™s z LytroIllumRawFormat.Reader._opencCs|`dSr©rErrrrržsz!LytroIllumRawFormat.Reader._closecCsdS©Nr,rrrrrÚ _get_length£sz&LytroIllumRawFormat.Reader._get_lengthcCsl|dkrtdƒ‚|jsR|jdkr,|j ¡|_tj|jtjd tj ¡}t   |¡}n t  g¡}||j ddfS©N)rÚNonez$Lytro file contains only one datasetr0r)Úindex)Ú IndexErrorrFrErDÚreadr1Ú frombufferÚuint8r:r7r&rBr<Ú_get_meta_data©rrNÚrawÚimgrrrÚ _get_data§s    z$LytroIllumRawFormat.Reader._get_datacCst|dkrtdƒ‚tj |j ¡¡d}d}dD]&}tj ||¡r.t t ||ƒ¡}q.|dk rb|St   d¡iSdS©N©rNú.Lytro meta data file contains only one datasetr)z.txtz.TXTz.jsonz.JSONz-No metadata file found for provided raw file.© rOÚosÚpathÚsplitextrZget_local_filenameÚisfileÚjsonÚloadÚopenÚloggerÚwarning©rrNÚ filename_baseZ meta_dataÚextrrrrS¿s z)LytroIllumRawFormat.Reader._get_meta_dataN)F©r!r"r#rrrKrWrSrrrrÚReader˜s  riN© r!r"r#r%r*Ú staticmethodrBrrirrrrr&bs   "r&c@s*eZdZdZdd„ZGdd„dejƒZdS)ÚLytroLfrFormata¿This is the Lytro Illum LFR format. The lfr is a image and meta data container format as used by the Lytro Illum light field camera. The format will read the specified lfr file. This format does not support writing. Parameters for reading ---------------------- meta_only : bool Whether to only read the metadata. include_thumbnail : bool Whether to include an image thumbnail in the metadata. cCs|jdkrdSdS)N)z.lfrTr(rrrrr*çs zLytroLfrFormat._can_readc@sVeZdZddd„Zdd„Zdd„Zd d „Zd d „Zd d„Zdd„Z dd„Z dd„Z dS)zLytroLfrFormat.ReaderFTc Csô|j ¡|_d|_i|_i|_d|_||_||_|  ¡|  ¡|  ¡z†|jddd}|d|jkrN|d|jkrN|d|jkrN|jsÂ|j|d\}}|j  |d¡|j  |¡|_|j|d\}}|j  |d¡|j  |¡}t | d¡¡|jd<|j|d\}}|j  |d¡|j  |¡}t | d¡¡|_|j|jd <|jrÎ|jd d}|d|jkrÎ|j|d\}}|j  |d¡|j  |¡}t|d d } |d } |d} | | | dœ|jd<Wntk rîtdƒ‚YnXdS)NÚframesrÚframeÚ metadataRefÚimageRefÚprivateMetadataRefÚASCIIÚmetadataÚprivateMetadataZ thumbnailsZjpeg)ÚformatÚheightÚwidth)rArvrwZ thumbnailz+The specified file is not a valid LFR file.)rrrDrEÚ_chunksrsÚ_contentrFZ_include_thumbnailÚ _find_headerÚ _find_chunksÚ _find_metaÚseekrPÚraw_image_datar`ÚloadsÚdecodeÚserial_numbersrÚKeyErrorr) rrHZinclude_thumbnailÚ chunk_dictÚdata_posÚsizersrZthumbnail_dataZ thumbnail_imgZthumbnail_heightZthumbnail_widthrrrrïs\  ÿ þ ý     ýzLytroLfrFormat.Reader._opencCs|`dSrrIrrrrr1szLytroLfrFormat.Reader._closecCsdSrJrrrrrrK6sz!LytroLfrFormat.Reader._get_lengthcCs0d}|j t¡}||kr tdƒ‚|j t¡dS)úH Checks if file has correct header and skip it. ó ‰LFP  zThe LFR file header is invalid.N©rDrPÚ HEADER_LENGTHrÚ SIZE_LENGTH©rZ file_headerÚheaderrrrrz:s  z"LytroLfrFormat.Reader._find_headercCs6d}tdtƒD]"}| |¡\}}}||f|j|<qdS©zN Gets start position and size of data chunks in file. s ‰LFC  rN)ÚrangeÚDATA_CHUNKS_ILLUMÚ _get_chunkrx©rZ chunk_headerr r„r…Úsha1rrrr{Gsz"LytroLfrFormat.Reader._find_chunkscCsDd}| |¡\}}}|j |d¡|j |¡}t | d¡¡|_dS)út Gets a data chunk that contains information over content of other data chunks. ó ‰LFM  rrrN©rrDr}rPr`rr€ry©rZ meta_headerr„r…r’Údatarrrr|Qs  z LytroLfrFormat.Reader._find_metacCs¸|j t¡}||krtdƒ‚d}d}t d|j t¡¡d}|dkr®t|j t¡  d¡ƒ}|j t ¡|j  ¡}|j  |d¡|j d¡}|dkr |j d¡}qŠ|j  dd¡|||fS) á_ Checks if chunk has correct header and skips it. Finds start position and length of next chunk and reads sha1-string that identifies the following data chunk. Parameters ---------- header : bytes Byte string that identifies start of chunk. Returns ------- data_pos : int Start position of data chunk in file. size : int Size of data chunk. sha1 : str Sha1 value of chunk. z The LFR chunk header is invalid.Nú>irrrr,óéÿÿÿÿ© rDrPr‰rÚstructÚunpackrŠÚstrÚ SHA1_LENGTHr€ÚPADDING_LENGTHÚtellr}©rrŒZ header_chunkr„r’r…Úchrrrr^s     z LytroLfrFormat.Reader._get_chunkcCsP|dkrtdƒ‚|jsr?Za0Za1rArrrrBºs ÿÿz LytroF01RawFormat.rearrange_bitsc@s6eZdZd dd„Zdd„Zdd„Zdd „Zd d „Zd S)zLytroF01RawFormat.ReaderFcCs|j ¡|_d|_||_dSrrCrGrrrrÔs zLytroF01RawFormat.Reader._opencCs|`dSrrIrrrrrÙszLytroF01RawFormat.Reader._closecCsdSrJrrrrrrKÞsz$LytroF01RawFormat.Reader._get_lengthcCsl|dkrtdƒ‚|jsR|jdkr,|j ¡|_tj|jtjd tj ¡}t   |¡}n t  g¡}||j ddfSrL)rOrFrErDrPr1rQrRr:r7r«rBr<rSrTrrrrWâs    z"LytroF01RawFormat.Reader._get_datacCst|dkrtdƒ‚tj |j ¡¡d}d}dD]&}tj ||¡r.t t ||ƒ¡}q.|dk rb|St   d¡iSdSrXr[rerrrrSùs z'LytroF01RawFormat.Reader._get_meta_dataN)FrhrrrrriÓs  riNrjrrrrr«¦s  r«c@s*eZdZdZdd„ZGdd„dejƒZdS)ÚLytroLfpFormata½This is the Lytro Illum LFP format. The lfp is a image and meta data container format as used by the Lytro F01 light field camera. The format will read the specified lfp file. This format does not support writing. Parameters for reading ---------------------- meta_only : bool Whether to only read the metadata. include_thumbnail : bool Whether to include an image thumbnail in the metadata. cCs|jdkrdSdS)N)z.lfpTr(rrrrr*!s zLytroLfpFormat._can_readc@sVeZdZddd„Zdd„Zdd„Zdd „Zd d „Zd d „Zdd„Z dd„Z dd„Z dS)zLytroLfpFormat.ReaderFcCsr|j ¡|_d|_i|_i|_d|_||_| ¡|  ¡|  ¡z |jdddd}|d|jkrL|d|jkrL|d|jkrL|jsÀ|j|d\}}|j  |d¡|j  |¡|_ |j|d\}}|j  |d¡|j  |¡}t | d¡¡|jd <|j|d\}}|j  |d¡|j  |¡}t | d¡¡|_|j|jd <Wntk rltd ƒ‚YnXdS) NZpictureZ frameArrayrrnrorprqrrrsrtz+The specified file is not a valid LFP file.)rrrDrErxrsryrFrzr|r{r}rPr~r`rr€rr‚r)rrHrƒr„r…rsrrrrr)s@  ÿ þ ý  zLytroLfpFormat.Reader._opencCs|`dSrrIrrrrrUszLytroLfpFormat.Reader._closecCsdSrJrrrrrrKZsz!LytroLfpFormat.Reader._get_lengthcCs0d}|j t¡}||kr tdƒ‚|j t¡dS)r†r‡zThe LFP file header is invalid.Nrˆr‹rrrrz^s  z"LytroLfpFormat.Reader._find_headercCs6d}tdtƒD]"}| |¡\}}}||f|j|<qdSr)rŽÚDATA_CHUNKS_F01rrxr‘rrrr{lsz"LytroLfpFormat.Reader._find_chunkscCsPd}| |¡\}}}|j |d¡|j |¡}t | d¡¡|_|j d¡}dS)r“r”rrrr+Nr•r–rrrr|vs  z LytroLfpFormat.Reader._find_metacCs¸|j t¡}||krtdƒ‚d}d}t d|j t¡¡d}|dkr®t|j t¡  d¡ƒ}|j t ¡|j  ¡}|j  |d¡|j d¡}|dkr |j d¡}qŠ|j  dd¡|||fS) r˜z The LFP chunk header is invalid.Nr™rrrr,ršr›rœr£rrrr…s     z LytroLfpFormat.Reader._get_chunkcCsP|dkrtdƒ‚|jss,   $vOl