Ng} dZddlmZddlmZmZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZmZmZmZmZmZmZdd lmZdd lmZdd lmZddlm Z m!Z!m"Z"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.edZ/e0e1e1e1e1e1e1fZ2e0e1e1fZ3e0e1e1e1e1e1e1e1e1fZ4ee2e3e4fZ5Gdde6Z7Gdde6Z8Gdde8Z9Gdde6Z:d6dZ;d7d!ZGd-d.eZ?Gd/d0e?Z@Gd1d2e@ZAGd3d4e?ZBd5S)9zExtract images embedded in PDF.) annotations)ABCabstractmethod)copy)Decimal)BytesIO) zip_longest)Path) copyfileobj)AnyBinaryIOCallable NamedTupleTypeVarUnioncast)Image)ImageCmsProfile)jbig2)BufferPdfPdfErrorStreamDecodeLevel)DependencyError) _transcoding)Array DictionaryNameObjectStreamStringTceZdZdZdS)UnsupportedImageTypeErrorzz0_array_str.._convert..>s#:::'HHW%%:::r+) isinstancelistrr rbytesintrstrr!NotImplementedError)itemr:r3s r,r:z_array_str.._convert<s dT5M * * ;::::T::: : dVZ< = = K dT3K ( ( t99  dV % % ;; !%(((r+)r<r=)r3resultr:s` @r, _array_strrD9sO ) ) ) ) ) )Xe__F fd # # Mr+*list[Object] | Dictionary | Array | Objectreturn list[Object]ct|tr|St|S)zEnsure value is a list of pikepdf.Object, if it was not already. To support DecodeParms which can be present as either an array of dicts or a single dict. It's easier to convert to an array of one dict. )r<r= wrap_in_arrayas_list)r3s r, _ensure_listrKMsA % ##%%--// 0 00r+objDictionary | Streamnamer@type_Callable[[Any], T]defaultT | Nonect|||} ||S#t$r|YdSYnwxYwtd|z)z@Retrieve metadata from a dictionary or stream and wrangle types.NzMetadata access for )getattr TypeErrorrA)rLrNrOrQvals r,_metadata_from_objrWXsl #tW % %CuSzz  ;44 ; 4t; < <paletteN)r&r'r(r)__annotations__r*r+r,rYrYes1NNNNNr+rYc,eZdZdZhdZehdzZddhZed/d Ze d0dZ e d0dZ e d1dZ e d2dZ e dZe dZe d3dZe dZe d4dZe d0dZe ed5dZe d1dZd Ze d1d!Ze d1d"Ze d6d$Zd%Ze d7d&Ze d'Ze d8d)Zed9d+Zd:d-Zd.S); PdfImageBasezAbstract base class for images.>/CalRGB/CalGray /DeviceRGB /DeviceGray>/CalCMYK /ICCBased /DeviceCMYK /Separation/DeviceNrNr@rOrPrQr"rFcdS)z!Get metadata for this image type.Nr*selfrNrOrQs r, _metadatazPdfImageBase._metadatayr+r?c:|dtdS)z"Width of the image data in pixels.Widthrrkr?rjs r,widthzPdfImageBase.width}s~~gsA...r+c:|dtdS)z#Height of the image data in pixels.Heightrrorps r,heightzPdfImageBase.heights~~hQ///r+boolc:|dtdS)z)Return ``True`` if this is an image mask. ImageMaskF)rkrurps r, image_maskzPdfImageBase.image_masks~~k4777r+ int | Nonec:|dtdS)z.Bits per component for this image (low-level).BitsPerComponentrrorps r,_bpczPdfImageBase._bpcs~~0#q999r+c:|dtgS)zColorspace (low-level). ColorSpacerkrDrps r, _colorspaceszPdfImageBase._colorspacess~~lJ;;;r+c:|dtgS)zBList of names of the filters that we applied to encode this image.Filterrrps r,filterszPdfImageBase.filterss~~h B777r+ DecodeArrayc|dtg}|r=t|dvr,ttt d|DS|jdvrdS|jdkrdS|jdkrd S|jd kr4|d krdS|d krdStd t|z)zExtract the /Decode array.Decode)c34K|]}t|VdSN)float)r8r3s r, z-PdfImageBase._decode_array..s(*L*LE5<<*L*L*L*L*L*Lr+rbr`)?rar_)rrrrrrre)rrrrrrrrrdLRGBz5Don't how to retrieve default /Decode array for image) rkrKlenrrtuple colorspace_approx_mode_from_iccrArepr)rjdecodes r, _decode_arrayzPdfImageBase._decode_arrays~~h bAA  Nc&kkY.. U*L*LV*L*L*L%L%LMM M ?9 9 9: ?7 7 711 ?m + +;; ?k ) )))++s22!z))++u4455! Cd4jj P   r+c:|dtgS)z/List of the /DecodeParms, arguments to filters. DecodeParms)rkrKrps r, decode_parmszPdfImageBase.decode_parmss~~m\2>>>r+ str | Nonec|jrdS|jr|jd|jvr |jdS|jddkrT|jd}t|tr ||jvr|St|t r|ddvr|dS|jddkrdSt dt|jz)z:PDF name of the colorspace that best describes this image.Nr/Indexed)rdrgr`r_rgz not sure how to get colorspace: )rxrMAIN_COLORSPACESr<r@r=rAr)rjsubspaces r,rzPdfImageBase.colorspaces ? 4   " #t'<<<(++ #z11,Q/h,,$T=R1R1R#Oh--'(1+B33 $A;& #z11!z! .d6G1H1H H   r+cJ|j |jdkr |jrdndS|jS)z!Bits per component of this image.Nrrr)r|rxrps r,bits_per_componentzPdfImageBase.bits_per_components/ 9  Q.11Q .yr+ImageCmsProfile | NonecdS)z4Return ICC profile for this image if one is defined.Nr*rps r,icczPdfImageBase.iccrlr+cd|jvS)z/Check if the image has a defined color palette.r)rrps r,indexedzPdfImageBase.indexedsT...r+c |j}|ddkr|dd|krdS|d|krdSn#tttf$rYnwxYwdS)NrrrTF)r IndexErrorAttributeErrorKeyError)rjrNcss r,_colorspace_has_namez!PdfImageBase._colorspace_has_namesx "B!u ""r!uQx4'7'7t!u}}tNH5    D us%8 8AAc,|dS)rPRINT_COLORSPACESrArrY)rj_idxbase_hivallookuprs r,r[zPdfImageBase.palette>s| 4 N)-): &D$ N N NEFFA M N 8 t' 4+= DRVAWAW tAw<ct}|5}||d|cdddS#1swxYwYdS)z!Display hook for IPython/Jupyter.PNGN)rrsavegetvalue)rjbims r, _repr_png_zPdfImageBase._repr_png_bs II     B GGAu   ::<<                  s*AA!AN)rNr@rOrPrQr"rFr"rFr?rFru)rFry)rFr)rFrrFr)rFr)rFr@)rFrrFrrFr>) r&r'r(r)SIMPLE_COLORSPACESrrrrkpropertyrqrtrxr|rrrrrrrrrrrrrrrr[rrr*r+r,r^r^rs))MMM),T,T,TT& 3000^0///X/000X0888X8:::X:<<X<88X8   X ,??X?   X 2X CCC^XC///X/   555X5888X8'''X' < < <   X D P PX P)))X)>EEE^E      r+r^c4eZdZUdZded<ded<ded<d2fd Zd2d Zd Zed Z d Z e d Z e d3dZ dZd4dZd5dZd5dZd5dZd5dZd6dZdddd7d!Zejfd8d%Zejfd9d'Zd5d(Zd:d;d,Zd-Zd> :  "&|"4"4H   Z: % %"&}"5"5H ).s):):AA!,'/t$s8}}r+c0t|j|||SrrWrLris r,rkzPdfImage._metadata!$(D%AAAr+c|jdkrI|js |jdSt|jdtsJ|jddSt d)Nrdrz+Don't know how to find ICC stream for image)rrrr<r=rArps r, _iccstreamzPdfImage._iccstreamsg ?k ) )< ,(++d/2D99 9 9 9$Q'* *!"OPPPr+rFc(|jdvrdS|jsz|j}|}t |} t ||_n:#t $r-}t|dkrtd|Yd}~nd}~wwxYw|jS)zIf an ICC profile is attached, return a Pillow object that describe it. Most of the information may be found in ``icc.profile``. )rdrNzcannot open profile from stringz#ICC profile corrupt or not readable) rrr get_stream_bufferrrOSErrorr@r$)rj iccstream iccbuffer iccbytesiors r,rz PdfImage.iccs ?"; ; ;4y I!3355I ++J +J77    q66>>>3=?>>>>  ysA B #BB chdfdt|jD}t|dkr%td|jjd|jdt|dkr&|jtjgfS|d}|dkr |j |jfSt|j}td|jd |D|_ t|j d ||_|tj|j|d fS) z4Remove simple lossless compression where it appears.> /JBIG2Decode /DCTDecode /JPXDecode/CCITTFaxDecodec"g|] \}}|v | Sr*r*)r8nfiltCOMPLEX_FILTERSs r,r;z3PdfImage._remove_simple_filters..s'WWWDt?V?V1?V?V?Vr+rzObject z has compound complex filters: z. We cannot decompress this.rc,g|]}t|Sr*)r)r8fs r,r;z3PdfImage._remove_simple_filters..s C C CQa C C Cr+N) enumeraterrrArLobjgen read_bytesr specializedread_raw_bytesrrrrr)rjindicesrobj_copyrs @r,_remove_simple_filterszPdfImage._remove_simple_filterssb    XWWWIdl$;$;WWW w<$(/>><>>>  w<<1  8&&'8'DEErI I AJ 668**,,dl: :>> C C$,rr2B C C CDD$T%6rr%:;;""#4#@AA4.normal_dct_rgbsb NB&q)!,8,Q/266%~9%>"*> >r+cd}|}jdd'jddd|}jdko||kS)Nrrr(rr))DEFAULT_CT_CMYKr+rjs r,normal_dct_cmykz1PdfImage._extract_direct..normal_dct_cmyksb O B&q)!,8,Q/266%9&@2+@ @r+rrdN)rz.tifrrz.jpgr)r$rr rwrite_generate_ccitt_headerr)rjr%r,r/rrrs` r,_extract_directzPdfImage._extract_directs ? ? ? ? ? ? A A A A A A3355 g () ) )+--o0022 LL44Ts4CC D D D LL   6 |n $ $ I   0 0 OO4E4E  LL   6tr+rcd}|jdkrdnd}|jdvr7tj||j|j|\}}nB|jdkr(t t|}ntd|jdkr/|j (|j \}}tj ||j|||}ntj ||j|}|S)zAExtract an image when there are 1/2/4/8 bits packed in byte data.rrr)rrz*BitsPerComponent must be 1, 2, 4, 8, or 16r) rrrunpack_subbyte_pixelsrrr memoryviewr r2r[image_from_buffer_and_paletteimage_from_byte_buffer)rjstridescalebuffer base_moder[rs r,_extract_transcoded_1248bitsz%PdfImage._extract_transcoded_1248bits%sY#%%1  "f , ,)?!!49d.EuNFFF $ ) )*d&<&<&>&>??FF&'STT T 9   8!% Iw;  BB4VTYOOB r+c|js|jdvrtd |}n`#tt f$rL}dt |vr5tj stddd}~wwxYwtj d|j |}|j |j\}}tj|||}|S)N)rrz$1-bit RGB and CMYK are not supportedz(read_bytes called on unfilterable streamzRjbig2dec - not installed or installed version is too old (older than version 0.15)r)rxrr$r RuntimeErrorrr@r get_decoder availablerr frombytesrr[rfix_1bit_palette_image)rjrrrr<r[s r,_extract_transcoded_1bitz!PdfImage._extract_transcoded_1bit?s T49#?#?+,RSS S ??$$DDh'   :c!ffDD)++5577E&0  _S$)T 2 2 < #!% Iw4RGLLB s6BABBc*|Sr)rDrps r,_extract_transcoded_maskz!PdfImage._extract_transcoded_maskWs,,...r+c |jr|S|jdvrt|jdkr=|jdkr2t jd|j|dddd}n|jdkr=|jdkr2t jd|j| dddd}n}|jdkr| }n]|jdvr |jdkr| }n4tt|d zt|jz|jd kr(|j!|j|jd <|S) N>rrrrrawrrr)rrz, rdr)rxrFrr0rr frombufferrrr rDr=r$rrLrrrinfo)rjrs r,_extract_transcodedzPdfImage._extract_transcodedZsy ? 30022 2 91 1 1466 6 9  $"9Q">">!ty$//"3"3UE1aBBY& T%<%A%A! 4#9#9#;#;UFAqBB $ ) )..00BB Y* $ $)@A)E)E2244BB+DJJ,=TX,NOO O ?k ) )dh.B%)X%5%5%7%7BGM " r+r@cl||}|r|Sd} |}|jdkr3||dd |r|dSdS|r2||d  |r|d Sd SnA#t $r4}d t |vrtt||d}~wwxYw |r|n#|r|wwxYwtt|) aExtract the image to a stream. If possible, the compressed data is extracted and inserted into a compressed image file format without transcoding the compressed content. If this is not possible, the data will be decompressed and extracted to an appropriate format. Args: stream: Writable stream to write data to Returns: The file format extension. r%Nrtifftiff_adobe_deflate)format compressionz.tiffpng)rPz.pngzcalled on unfilterable stream) r2rKrrcloserr@r$r)rjr%direct_extractionrrs r,_extract_to_streamzPdfImage._extract_to_streamxs!000??  %$ $  ))++Bw&  v;OPPP     u---         .#a&&88/T ;;B        (T 333s/7B%0B%$C>% C#//CC##C>>DNrr% fileprefixBinaryIO | NonerWct|t|krtd|r||St}||}|dt t t ||z}|d5}t||dddn #1swxYwYt |S)aExtract the image directly to a usable image file. If possible, the compressed data is extracted and inserted into a compressed image file format without transcoding the compressed content. If this is not possible, the data will be decompressed and extracted to an appropriate format. Because it is not known until attempted what image format will be extracted, users should not assume what format they are getting back. When saving the image to a file, use a temporary filename, and then rename the file to its final name based on the returned file extension. Images might be saved as any of .png, .jpg, or .tiff. Examples: >>> im.extract_to(stream=bytes_io) # doctest: +SKIP '.png' >>> im.extract_to(fileprefix='/tmp/image00') # doctest: +SKIP '/tmp/image00.jpg' Args: stream: Writable stream to write data to. fileprefix (str or Path): The path to write the extracted image to, without the file extension. Returns: If *fileprefix* was provided, then the fileprefix with the appropriate extension. If no *fileprefix*, then an extension indicating the file type. z%Cannot set both stream and fileprefixrMrwbN) rurrUrseekr r@openr )rjr%rWbio extensionfilepathtargets r, extract_tozPdfImage.extract_tosD <<4 ++ + +DEE E  :**&*99 9ii++3+77   D,,-- 9:: ]]4  %F V $ $ $ % % % % % % % % % % % % % % %8}}sCC#&C# decode_levelrr>c8|j|S)z7Decompress this image and return it as unencoded bytes.rb)rLrrjrbs r,rzPdfImage.read_bytessx"" "===r+rc8|j|S)z+Access this image with the buffer protocol.rd)rLr res r,r zPdfImage.get_stream_buffersx))|)DDDr+ct}||}|r)|dtj|S|}|st t||S)zuExtract the image as a Pillow Image, using decompression as necessary. Caller must close the image. rMr)rr2r[rr\rKr$r)rjr]rTrs r,rzPdfImage.as_pil_images{ ii 000<<  # HHQKKK:c?? "  % % ' ' 8+DJJ77 7 r+rr bytes | Nonec&|jstddg}|D]?\}}|jd|||krtd|d|@|jddd}d}|dkrd}n |dkrd }d }nd }|jdd d } |j} | rd nd} t | d kr | dkrd | z } t |} |d}t j|j| ||| |S)z8Construct a CCITT G3 or G4 header from the PDF metadata.z$/CCITTFaxDecode without /DecodeParms)z/EncodedByteAlignFrz&/CCITTFaxDecode with decode parameter z not equal z/KNr4rz /BlackIs1Fr)rrr+) data_length ccitt_group t4_options photometryr) rrrr$rrrgenerate_ccitt_headerr) rjrrexpected_defaultsrNrVkrmrl black_is_onerrnimg_sizes r,r1zPdfImage._generate_ccitt_headersz   ECDD D ) +  ID# #''c22c99/STSScSS:  a $ $T1 - - q55KK UUKJJK(+// UCC #'-QQA  v;;!  * 4 4ZJt99 ;C1 I #!!     r+cR|dS)z$Show the image however PIL wants to.N)rshowrps r,ruz PdfImage.shows&   """""r+rrc||_dSr)r)rjrs r,_set_pdf_sourcezPdfImage._set_pdf_source#sr+c  |j}n#t$rd}YnwxYwd|d|jd|jdt t |d S)N?zrrArqrthexidrjrs r,__repr__zPdfImage.__repr__&s 9DD"   DDD  CD C CJ C C!% C C25bhh-- C C C  )rLr rr%r rFrr)r%r rFr@)r%rXrWr@rFr@)rbrrFr>)rbrrFrr)rr>rrhrFr>)rr) r&r'r(r)r\rrr classmethodrrkrr rr$r2r=rDrFrKrUrarr rr rr1rurwr __classcell__rs@r,rrjsUKKK     ) ) ) ) ) )    %%% [4BBBQQXQX*TTT84444l40////<#4#4#4#4L,02------`1B0M>>>>>1B0MEEEEE "4 4 4 4 4 l###       r+rceZdZdZfdZdZddZdfd Zefd Z edd Z eddZ dZ xZ S)rzSupport class for JPEG 2000 images. Implements the same API as :class:`PdfImage`. If you call PdfImage(object_that_is_actually_jpeg2000_image), pikepdf will return this class instead, due to the check in PdfImage.__new__. c|t|||_dS)zInitialize a JPEG 2000 image.N)rrr_jpxpil)rjrLrs r,rzPdfJpxImage.__init__8s2 ((** r+ct|tstS|j|jko$t|to|j|jkSr)r<r^rrLrrrs r,rzPdfJpxImage.__eq__=sN%.. "! ! H ! .5+.. .  - r+r%r rFrcp|\}}|dgkrdS||dS)Nrz.jp2)r$r0)rjr%rrs r,r2zPdfJpxImage._extract_directFs@3355 g |n $ $4 Tvr+rcDtSr)rrK)rjrs r,rKzPdfJpxImage._extract_transcodedMsww**,,,r+ctj}|r|S|jjdkrdgS|jjdkrdgSt d)zReturn the effective colorspace of a JPEG 2000 image. If the ColorSpace dictionary is present, the colorspace embedded in the JPEG 2000 data will be ignored, as required by the specification. rrbrrazComplex JP2 colorspace)rrrrrA)rjsuper_colorspacesrs r,rzPdfJpxImage._colorspacesPs_"GG0  %$ $ <  # #!? " <  % % > !!":;;;r+r?cdS)z=Return 8, since bpc is not meaningful for JPEG 2000 encoding.rr*rps r,r|zPdfJpxImage._bpcbs qr+rucdS)z4Return False, since JPEG 2000 should not be indexed.Fr*rps r,rzPdfJpxImage.indexedks ur+c rd|jd|jd|jdtt |d S)Nz)zPdfInlineImage.s8881q!888r+r_dataztuple[Object, ...] _image_object image_data image_objectrc|_|_dfd|D} tjd|zdz}n8#t $r+}t d|dz|d}~wwxYw|_dS)zConstruct wrapper for inline image. Args: image_data: data stream for image, extracted from content stream image_object: the metadata for image, also from content stream  c3PK|] }|jV!dS) remap_namesN) _unparse_objABBREVS)r8rLrjs r,rz*PdfInlineImage.__init__..sI  ADD  ct|  < <      r+s<< s >>zparsing inline unicode_escapeN)rrjoinrparserrrL)rjrrreparse reparsed_objrs` r,rzPdfInlineImage.__init__s  )))    HT      X!<(86(ABBLL X X X,w~~>N/O/OOPPVW W XsA B&A==Bct|tstS|j|jkoHt|to3|j|jkSr)r<r^rrLrr_inline_image_raw_bytesrs r,rzPdfInlineImage.__eq__sk%.. "! ! H ! 5.11  2244;66889  r+ct|trnt|trC|d}t|tsJ|||S|dSt|t r|rdndSt|tttfr"t| dStt|)NT)resolvedstruesfalseascii)r<rrunparser>rrur?rrr@encoderAr)rrLrrNs r,rzPdfInlineImage._unparse_objs c6 " " .#t$$ 3{{D{11!$....."tT222;;;-- - c4  0!/77x / cC%0 1 1 ,s88??7++ +!$s)),,,r+c0t|j|||Srrris r,rkzPdfInlineImage._metadatar r+rFr>cVfdfd}d|S)zACreate the content stream bytes that reproduce this inline image.c3KjD]9}|j}t|tsJ|V:dS)Nr)rrREVERSE_ABBREVSr<r>) metadata_objunparsedrjs r,metadata_tokensz/PdfInlineImage.unparse..metadata_tokenssh $ 2   ,, d.B-"(E22222   r+c3KdVddDVdVjVdVdS)NsBI rc3K|]}|VdSrr*)r8rs r,rzFPdfInlineImage.unparse..inline_image_tokens..s"99!A999999r+s ID sEI)rrr)rrjsr,inline_image_tokensz3PdfInlineImage.unparse..inline_image_tokenssoMMM))99'8'899999 9 9 9OOO*4466 6 6 6KKKKKr+r+)r)rjrrs` @r,rzPdfInlineImage.unparsesY           xx++--...r+c td)zIRaise an exception since ICC profiles are not supported on inline images.zJInline images with ICC profiles are not supported in the PDF specification)r2rps r,rzPdfInlineImage.iccs# X   r+c  |j}n#t$rd}YnwxYwd|d|jd|jdt t |d S)Nryz#.s"#Q#Q2B#Q#Q#Q#Q#Q#Qr+)rnewadd_blank_pagerqrtpages contents_addrrexternalize_inline_imagesrr nextimagesvaluesrrw)rjtmppdfraw_imgimgs r,_convert_to_pdfimagez#PdfInlineImage._convert_to_pdfimagesT[(ABBB Q$$z 4 4 4 4 4 ; ;G D Dd %     Q$$T\\^^444  Q11333vt#Q#Qa1G1N1N1P1P#Q#Q#QQQRRw F### r+rcN|S)z&Return inline image as a Pillow Image.)rrrps r,rzPdfInlineImage.as_pil_images ((**77999r+NrrVr%rXrWr@cT|||S)zxExtract the inline image directly to a usable image file. See: :meth:`PdfImage.extract_to` rV)rra)rjr%rWs r,razPdfInlineImage.extract_tos2 ((**55j6   r+cN|S)z Return decompressed image bytes.)rrrps r,rzPdfInlineImage.read_bytess"((**55777r+cN|S)z"Return decompressed stream buffer.)rr rps r,r z PdfInlineImage.get_stream_buffers"((**<<>>>r+)rrrrr)rFrr)r%rXrWr@)r&r'r(r)ritemsrr\rrrrrkrrrrrrrarr r*r+r,rrzs.. y z %     z   ~  . { # "  # #!" #G&98 888OMMM%%%%    0     - -[ -BBB////(  X    $::::7;b      888 ?????r+rN)r3r4)r3rErFrG) rLrMrNr@rOrPrQr"rFrR)Cr) __future__rabcrrrdecimalrior itertoolsr pathlibr shutilr typingr r rrrrrPILr PIL.ImageCmsrpikepdfr pikepdf._corerrrrpikepdf._exceptionsrpikepdf.modelsrpikepdf.objectsrrrrr r!r"rrRGBDecodeArrayGrayDecodeArrayCMYKDecodeArrayr Exceptionr$r.r0r2rDrKrWrYr^rrrr*r+r,rs &%""""""########!!!!!!LLLLLLLLLLLLLLLLLL((((((BBBBBBBBBBBB//////'''''' GCLLueUE5%?@u %ueUE5%NONO_DE GGGGG GGG@@@@@)@@@UUUUU)<UUUJJJJJ9JJJ(1111 = = = =     *   u u u u u 3u u u pD D D D D |D D D NF F F F F (F F F R^?^?^?^?^?\^?^?^?^?^?r+