B •%Z\Å`ã@s dZddlmZmZmZddlZddlZddlZddlZddl Z ddl m Z ddl mZddl mZe e¡ZdZd Zd Zd Zd Zd ZdZdZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZeeeedœZ ddddgZ!dd „Z"e"ƒdS)!z Lytro Illum Plugin. Plugin to read Lytro Illum .lfr and .raw files as produced by the Lytro Illum light field camera. é)Úabsolute_importÚprint_functionÚdivisionNé)Úformats)ÚFormat)Úimread)iøi0)iÐ iÐ é éé-é#é éc@s.eZdZdZdZdd„ZGdd„dejƒZdS)Ú LytroFormata Base 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ú4lib/python3.7/site-packages/imageio/plugins/lytro.pyÚ _can_write>szLytroFormat._can_writec@s.eZdZd dd„Zdd„Zdd„Zdd „Zd S) zLytroFormat.WriterrcCs|j ¡|_dS)N)rÚget_fileZ_fp)rÚflagsrrrÚ_openEszLytroFormat.Writer._opencCsdS)Nr)rrrrÚ_closeHszLytroFormat.Writer._closecCs tdƒ‚dS)Nz)The lytro format cannot write image data.)Ú RuntimeError)rÚimÚmetarrrÚ _append_dataMszLytroFormat.Writer._append_datacCs tdƒ‚dS)Nz(The lytro format cannot write meta data.)r)rrrrrÚ_set_meta_dataQsz!LytroFormat.Writer._set_meta_dataN)r)Ú__name__Ú __module__Ú __qualname__rrrrrrrrÚWriterDs r"N)rr r!Ú__doc__Z_modesrrr"rrrrr3src@s6eZdZdZdd„Zedd„ƒZGdd„dejƒZdS) ÚLytroIllumRawFormatak 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 ---------------------- None cCs&|jd|jdkr"|jdkr"dSdS)Néú?)z.rawT)ÚmodeÚmodesÚ extension)rrrrrÚ _can_readds 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ér%rrr r é0éÀé)Údtypegø@) ÚnpÚ left_shiftÚ bitwise_andÚ right_shiftÚzerosÚLYTRO_ILLUM_IMAGE_SIZEÚuint16ÚreshapeÚdivideÚastypeÚfloat64)ÚarrayÚt0Út1Út2Zt3ZlsbÚimagerrrÚrearrange_bitsjs&   ((((z"LytroIllumRawFormat.rearrange_bitsc@s4eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd S) zLytroIllumRawFormat.ReadercCs|j ¡|_d|_dS)N)rrÚ_fileÚ_data)rrrrrŽs z LytroIllumRawFormat.Reader._opencCs|`dS)N)rB)rrrrr’sz!LytroIllumRawFormat.Reader._closecCsdS)Nr%r)rrrrÚ _get_length—sz&LytroIllumRawFormat.Reader._get_lengthcCsZ|dkrtdƒ‚|jdkr&|j ¡|_tj|jtjd tj¡}t   |¡}||j ddfS)N)rÚNonez$Lytro file contains only one dataset)r/r)Úindex) Ú IndexErrorrBrAÚreadr0Ú frombufferÚuint8r9r6r$r@Ú_get_meta_data)rrEÚrawÚimgrrrÚ _get_data›s   z$LytroIllumRawFormat.Reader._get_datacCsx|dkrtdƒ‚tj |j ¡¡d}d}x.dD]&}tj ||¡r0t t ||ƒ¡}q0W|dk rf|St   d¡iSdS)N)rNz.Lytro meta data file contains only one datasetr)z.txtz.TXTz.jsonz.JSONz-No metadata file found for provided raw file.) rFÚosÚpathÚsplitextrÚget_local_filenameÚisfileÚjsonÚloadÚopenÚloggerÚwarning)rrEÚ filename_baseÚ meta_dataÚextrrrrJ®s  z)LytroIllumRawFormat.Reader._get_meta_dataN)rr r!rrrCrMrJrrrrÚReaders r[N) rr r!r#r*Ú staticmethodr@rr[rrrrr$Ws  #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 ---------------------- None cCs&|jd|jdkr"|jdkr"dSdS)Nr%r&)z.lfrT)r'r(r))rrrrrr*Ós zLytroLfrFormat._can_readc@sTeZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dd„Z dS)zLytroLfrFormat.Readerc CsÚ|j ¡|_d|_i|_i|_d|_| ¡| ¡|  ¡yx|jddd}|d|jkr<|d|jkr<|d|jkr<|j|d\}}|j  |d¡|j  |¡|_ |j|d\}}|j  |d¡|j  |¡}t  | d¡¡|jd<|j|d\}}|j  |d¡|j  |¡}t  | d¡¡|_|j|jd <|jd d}|d|jkr´|j|d\}}|j  |d¡|j  |¡}t|d d }|d }|d} ||| dœ|jd<Wntk rÔtdƒ‚YnXdS)NZframesrÚframeÚ metadataRefÚimageRefÚprivateMetadataRefÚASCIIÚmetadataÚprivateMetadataZ thumbnailsZjpeg)ÚformatÚheightÚwidth)r?rfrgZ thumbnailz+The specified file is not a valid LFR file.)rrrArBÚ_chunksrcÚ_contentÚ _find_headerÚ _find_chunksÚ _find_metaÚseekrGÚraw_image_datarSÚloadsÚdecodeÚserial_numbersrÚKeyErrorr) rÚ chunk_dictÚdata_posÚsizercrqZthumbnail_dataZ thumbnail_imgZthumbnail_heightZthumbnail_widthrrrrÜsL      zLytroLfrFormat.Reader._opencCs|`dS)N)rB)rrrrrszLytroLfrFormat.Reader._closecCsdS)Nr%r)rrrrrC sz!LytroLfrFormat.Reader._get_lengthcCs0d}|j t¡}||kr tdƒ‚|j t¡dS)zH Checks if file has correct header and skip it. s ‰LFP  zThe LFR file header is invalid.N)rArGÚ HEADER_LENGTHrÚ SIZE_LENGTH)rÚ file_headerÚheaderrrrrj$s  z"LytroLfrFormat.Reader._find_headercCs:d}x0tdtƒD]"}| |¡\}}}||f|j|<qWdS)zN Gets start position and size of data chunks in file. s ‰LFC  rN)ÚrangeÚDATA_CHUNKS_ILLUMÚ _get_chunkrh)rÚ chunk_headerrrtruÚsha1rrrrk1sz"LytroLfrFormat.Reader._find_chunkscCsDd}| |¡\}}}|j |d¡|j |¡}t | d¡¡|_dS)zt Gets a data chunk that contains information over content of other data chunks. s ‰LFM  rrbN)r|rArmrGrSrorpri)rÚ meta_headerrtrur~Údatarrrrl;s  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¡}x|dkr¢|j d¡}qŒW|j  dd¡|||fS) a_ 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.Nz>irrbr%óéÿÿÿÿ) rArGrvrÚstructÚunpackrwÚstrÚ SHA1_LENGTHrpÚPADDING_LENGTHÚtellrm)rryÚ header_chunkrtr~ruÚchrrrr|Hs      z LytroLfrFormat.Reader._get_chunkcCs>|dkrtdƒ‚tj|jtjd tj¡}t |¡}||j fS)N)rNz(Lytro lfr file contains only one dataset)r/) rFr0rHrnrIr9r6r$r@rc)rrErKrrrrrMvs  zLytroLfrFormat.Reader._get_datacCs|dkrtdƒ‚|jS)N)rNz.Lytro meta data file contains only one dataset)rFrc)rrErrrrJ‚sz$LytroLfrFormat.Reader._get_meta_dataN) rr r!rrrCrjrkrlr|rMrJrrrrr[Ûs?   . r[N)rr r!r#r*rr[rrrrr]Çs r]c@s6eZdZdZdd„Zedd„ƒZGdd„dejƒZdS) ÚLytroF01RawFormata† This is the Lytro RAW format for the original F01 Lytro camera. The raw format is a 12bit image format as used by the Lytro F01 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 ---------------------- None cCs&|jd|jdkr"|jdkr"dSdS)Nr%r&)z.rawT)r'r(r))rrrrrr*™s zLytroF01RawFormat._can_readcCsÞ|ddd…}|ddd…}|ddd…}t |d¡t t |d¡d¡}t t |d¡d¡|}tjttjd }| tdtddf¡|dd…ddd…f<| tdtddf¡|dd…ddd…f<t |d ¡  tj ¡S) Nrrr%rr éðéé)r/gþ¯@) r0r1r3r2r4ÚLYTRO_F01_IMAGE_SIZEr6r7r8r9r:)r;r<r=r>Za0Za1r?rrrr@Ÿs ((z LytroF01RawFormat.rearrange_bitsc@s4eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd S) zLytroF01RawFormat.ReadercCs|j ¡|_d|_dS)N)rrrArB)rrrrr¹s zLytroF01RawFormat.Reader._opencCs|`dS)N)rB)rrrrr½szLytroF01RawFormat.Reader._closecCsdS)Nr%r)rrrrrCÂsz$LytroF01RawFormat.Reader._get_lengthcCsZ|dkrtdƒ‚|jdkr&|j ¡|_tj|jtjd tj¡}t   |¡}||j ddfS)N)rrDz$Lytro file contains only one dataset)r/r)rE) rFrBrArGr0rHrIr9r6r‹r@rJ)rrErKrLrrrrMÆs   z"LytroF01RawFormat.Reader._get_datacCsx|dkrtdƒ‚tj |j ¡¡d}d}x.dD]&}tj ||¡r0t t ||ƒ¡}q0W|dk rf|St   d¡iSdS)N)rNz.Lytro meta data file contains only one datasetr)z.txtz.TXTz.jsonz.JSONz-No metadata file found for provided raw file.) rFrNrOrPrrQrRrSrTrUrVrW)rrErXrYrZrrrrJÙs  z'LytroF01RawFormat.Reader._get_meta_dataN)rr r!rrrCrMrJrrrrr[¸s r[N) rr r!r#r*r\r@rr[rrrrr‹‹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 ---------------------- None cCs&|jd|jdkr"|jdkr"dSdS)Nr%r&)z.lfpT)r'r(r))rrrrrr*þs zLytroLfpFormat._can_readc@sTeZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dd„Z dS)zLytroLfpFormat.ReadercCsf|j ¡|_d|_i|_i|_d|_| ¡| ¡|  ¡y|jdddd}|d|jkr@|d|jkr@|d|jkr@|j|d\}}|j  |d¡|j  |¡|_ |j|d\}}|j  |d¡|j  |¡}t  | d¡¡|jd <|j|d\}}|j  |d¡|j  |¡}t  | d¡¡|_|j|jd <Wntk r`td ƒ‚YnXdS) NZpictureZ frameArrayrr^r_r`rarbrcrdz+The specified file is not a valid LFP file.)rrrArBrhrcrirjrlrkrmrGrnrSrorprqrrr)rrsrtrurcrqrrrrs6   zLytroLfpFormat.Reader._opencCs|`dS)N)rB)rrrrr2szLytroLfpFormat.Reader._closecCsdS)Nr%r)rrrrrC7sz!LytroLfpFormat.Reader._get_lengthcCs0d}|j t¡}||kr tdƒ‚|j t¡dS)zH Checks if file has correct header and skip it. s ‰LFP  zThe LFP file header is invalid.N)rArGrvrrw)rrxryrrrrj;s  z"LytroLfpFormat.Reader._find_headercCs:d}x0tdtƒD]"}| |¡\}}}||f|j|<qWdS)zN Gets start position and size of data chunks in file. s ‰LFC  rN)rzÚDATA_CHUNKS_F01r|rh)rr}rrtrur~rrrrkIsz"LytroLfpFormat.Reader._find_chunkscCsPd}| |¡\}}}|j |d¡|j |¡}t | d¡¡|_|j d¡}dS)zt Gets a data chunk that contains information over content of other data chunks. s ‰LFM  rrbr+N)r|rArmrGrSrorpri)rrrtrur~r€rrrrlSs  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¡}x|dkr¢|j d¡}qŒW|j  dd¡|||fS) a_ 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 LFP chunk header is invalid.Nz>irrbr%rr‚) rArGrvrrƒr„rwr…r†rpr‡rˆrm)rryr‰rtr~rurŠrrrr|bs      z LytroLfpFormat.Reader._get_chunkcCs>|dkrtdƒ‚tj|jtjd tj¡}t |¡}||j fS)N)rNz(Lytro lfp file contains only one dataset)r/) rFr0rHrnrIr9r6r‹r@rc)rrErKrrrrrMs  zLytroLfpFormat.Reader._get_datacCs|dkrtdƒ‚|jS)N)rNz.Lytro meta data file contains only one dataset)rFrc)rrErrrrJœsz$LytroLfpFormat.Reader._get_meta_dataN) rr r!rrrCrjrkrlr|rMrJrrrrr[s+ . r[N)rr r!r#r*rr[rrrrròs r)z lytro-lfrzlytro-illum-rawz lytro-lfpz lytro-f01-raw)z LYTRO-LFRzLytro Illum lfr image fileZlfrr)zLYTRO-ILLUM-RAWzLytro Illum raw image filerKr)z LYTRO-LFPzLytro F01 lfp image fileZlfpr)z LYTRO-F01-RAWzLytro F01 raw image filerKrcCsHxBtD]:\}}}}t | ¡t¡}|r|||||ƒ}tj|dqWdS)N)re)Ú file_formatsÚSPECIAL_CLASSESÚgetÚlowerrrZ add_format)ÚnameZdesrZrZ format_classrerrrÚ _create_predefined_lytro_formats·s r—)#r#Z __future__rrrrNrSrƒZloggingZnumpyr0ÚrZcorerrZ getLoggerrrVr5rrvrwr†r‡r{r‘rr$r]r‹rr“r’r—rrrrÚ sF     $pEg6