o ir@sdZddlZddlZddlmZmZerddlmZzddlZdZ Wn e y-dZ Ynwzddl m Z m Z dZe dZe d ZWne ySdZdZdZYnwdd lmZdd lmZdd lmZdd lmZmZmZddlmZmZmZm Z m!Z!e"dZ#eegZ$e e!gZ%e%e$Z&dZ'dddZ(de)fddZ*dede)fddZ+ddde)fddZ,ddde)fddZ-d e.de/d!e/dd"fd#d$Z0d(d&d'Z1dS))zVUse the `pillow `_ Python package to decode *Pixel Data*. N) TYPE_CHECKINGcast)DatasetTF)Imagefeaturesjpgjpg_2000)config)generate_frames) warn_and_log) pixel_dtypeget_j2k_parameters get_nr_frames)UIDJPEG2000JPEG2000LosslessJPEGBaseline8BitJPEGExtended12BitpydicomPillow)zhttps://numpy.org/NumPy)zhttps://python-pillow.org/r)numpyPILreturncCstotS)z8Return ``True`` if the handler has its dependencies met.)HAVE_NPHAVE_PILrr`/mnt/sdb/aimis/docanh/lib/python3.10/site-packages/pydicom/pixel_data_handlers/pillow_handler.py is_availableAsrtransfer_syntaxcCs|tvS)zReturn ``True`` if the handler supports the `transfer_syntax`. Parameters ---------- transfer_syntax : uid.UID The Transfer Syntax UID of the *Pixel Data* that is to be used with the handler. )PillowSupportedTransferSyntaxes)rrrrsupports_transfer_syntaxFs r!dsrcCdS)zReturn ``True`` if the *Pixel Data* should to be converted from YCbCr to RGB. This affects JPEG transfer syntaxes. Frr"rrrneeds_to_convert_to_RGBRsr%cCr#)zReturn ``True`` if the *Photometric Interpretation* should be changed to RGB. This affects JPEG transfer syntaxes. Frr$rrr.should_change_PhotometricInterpretation_to_RGB[sr&dataphotometric_interpretationrcCs@t|}t|}|tvr|dkrd|jvr|d|j|S)aXDecompresses a single frame of an encapsulated Pixel Data element. Parameters ---------- data: bytes Compressed pixel data transfer_syntax: str Transfer Syntax UID photometric_interpretation: str Photometric Interpretation Returns ------- PIL.Image Decompressed pixel data RGBadobe_transformYCbCr)ioBytesIOropenPillowJPEGTransferSyntaxesinfodraftsize)r'rr(fioimagerrr_decompress_single_framees    r5 numpy.ndarraycCsV|jj}tstd|jdts|tvrtd|jdts-|t vr-td|jd|t krA|j dkrAtt dt jdt t |j}t t|j}t t|j}t t|j}t t|j }t|d d }t}d \} } t|j|d D]0} t| ||} d |vr| d||f|| | st| } t t| d|} | dd} qstdt|dt !|t"|}|t vr||}| r| |krt#d|d| dt$j%r| r|| }| s| |j&krt '||}n4|j&dkr|t (d|d8}|rt '||}n|j&dkr|t (d|d8}|rt '||}t)|r&d|_t d|S)aReturn a :class:`numpy.ndarray` of the *Pixel Data*. Parameters ---------- ds : Dataset The :class:`Dataset` containing an Image Pixel module and the *Pixel Data* to be decompressed and returned. Returns ------- numpy.ndarray The contents of (7FE0,0010) *Pixel Data* as a 1D array. Raises ------ ImportError If Pillow is not available. NotImplementedError If the transfer syntax is not supported zKThe pillow package is required to use pixel_array for this transfer syntax z#, and pillow could not be imported.z$The pixel data with transfer syntax z5, cannot be read because Pillow lacks the JPEG pluginz:, cannot be read because Pillow lacks the JPEG 2000 pluginz - z> is only supported by Pillow if (0028,0100) Bits Allocated = 8F)warn)NN)number_of_framesYBRr+ precision is_signedNzSuccessfully read z pixel bytesz%The (0028,0101) 'Bits Stored' value (z(-bit) doesn't match the JPEG 2000 data (z?-bit). It's recommended that you change the 'Bits Stored' valuer)r6)* file_metaTransferSyntaxUIDr ImportErrorname HAVE_JPEGr/NotImplementedError HAVE_JPEG2KPillowJPEG2000TransferSyntaxesr BitsAllocatedrstrPhotometricInterpretationintRowsColumns BitsStoredr bytearrayr PixelDatar5r1extendtobytesr setdefaultloggerdebuglenr frombufferr r r APPLY_J2K_CORRECTIONSPixelRepresentation right_shiftuint32r&)r"rr(rowscolumns bits_storedbits_allocated nr_frames pixel_bytes j2k_precisionj2k_signframeimparamsarrshiftrrr get_pixeldatas                   rh)r"rrr6)2__doc__r,loggingtypingrrpydicom.datasetrrrrArrrr check_codecrCrErr pydicom.encapsr pydicom.miscr pydicom.pixels.utilsr r r pydicom.uidrrrrr getLoggerrSrFr/r HANDLER_NAME DEPENDENCIESboolrr!r%r&bytesrHr5rhrrrrsd            &