o i=+@sFdZddlmZddlZddlmZmZddlmZer"ddl m Z zddl Z dZ Wn ey5dZ YnwzddlZdZWn eyIdZYnwerTdd lmZmZzddlZdZWn eygdZYnwzddlZdZWn ey{dZYnwzddlZdZWn eydZYnwdd lmZdd lmZdd lm Z m!Z!m"Z"m#Z#m$Z$dd l%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/e0dZ1dZ2ereZ3e&e'e(e)e*e+gZ4e,e-gZ5e.gZ6e4e5e6Z7ddiZ8de9fddZ:de/de9fddZ;ddde9fddZd'ddd!e9dedfd"d#Zd&d$d%Z?dS)(a Use the :gh:`pylibjpeg ` package to convert supported pixel data to a :class:`numpy.ndarray`. .. versionadded:: 2.1 **Supported data** The pylibjpeg handler supports the conversion of data in the (7FE0,0010) *Pixel Data* elements to a :class:`~numpy.ndarray` provided the related :dcm:`Image Pixel` module elements have values given in the table below. +------------------------------------------------+---------------+----------+ | Element | Supported | | +-------------+---------------------------+------+ values | | | Tag | Keyword | Type | | | +=============+===========================+======+===============+==========+ | (0028,0002) | SamplesPerPixel | 1 | 1, 3 | Required | +-------------+---------------------------+------+---------------+----------+ | (0028,0004) | PhotometricInterpretation | 1 | MONOCHROME1, | Required | | | | | MONOCHROME2, | | | | | | RGB, | | | | | | YBR_FULL, | | | | | | YBR_FULL_422, | | | | | | YBR_ICT, | | | | | | YBR_RCT | | +-------------+---------------------------+------+---------------+----------+ | (0028,0006) | PlanarConfiguration | 1C | 0, 1 | Optional | +-------------+---------------------------+------+---------------+----------+ | (0028,0008) | NumberOfFrames | 1C | N | Optional | +-------------+---------------------------+------+---------------+----------+ | (0028,0010) | Rows | 1 | N | Required | +-------------+---------------------------+------+---------------+----------+ | (0028,0011) | Columns | 1 | N | Required | +-------------+---------------------------+------+---------------+----------+ | (0028,0100) | BitsAllocated | 1 | 8, 16 | Required | +-------------+---------------------------+------+---------------+----------+ | (0028,0101) | BitsStored | 1 | Up to 16 | Required | +-------------+---------------------------+------+---------------+----------+ | (0028,0103) | PixelRepresentation | 1 | 0, 1 | Required | +-------------+---------------------------+------+---------------+----------+ .. versionchanged:: 2.2 Added support for *RLE Lossless* via the `pylibjpeg-rle` plugin. )deepcopyN) TYPE_CHECKINGcast)Iterable)DatasetTF)get_pixel_data_decodersDecoder)config)generate_frames) pixel_dtypeget_expected_lengthreshape_pixel_arrayget_j2k_parameters get_nr_frames) JPEGBaseline8BitJPEGExtended12Bit JPEGLosslessJPEGLosslessSV1JPEGLSLosslessJPEGLSNearLosslessJPEG2000LosslessJPEG2000 RLELosslessUIDpydicom pylibjpegnumpy)zhttps://numpy.org/NumPyreturncCstotS)z8Return ``True`` if the handler has its dependencies met.)HAVE_NPHAVE_PYLIBJPEGr!r!c/mnt/sdb/aimis/docanh/lib/python3.10/site-packages/pydicom/pixel_data_handlers/pylibjpeg_handler.py is_availablesr#tsyntaxcCs|tvS)zReturn ``True`` if the handler supports the `tsyntax`. Parameters ---------- tsyntax : pydicom.uid.UID The *Transfer Syntax UID* of the *Pixel Data* that is to be used with the handler. )SUPPORTED_TRANSFER_SYNTAXES)r$r!r!r"supports_transfer_syntaxs r&dsrcCdS)zReturn ``True`` if the *Pixel Data* should to be converted from YCbCr to RGB. This affects JPEG transfer syntaxes. Fr!r'r!r!r"needs_to_convert_to_RGBr*cCr()zReturn ``True`` if the *Photometric Interpretation* should be changed to RGB. This affects JPEG transfer syntaxes. Fr!r)r!r!r".should_change_PhotometricInterpretation_to_RGBr+r, np.ndarraycCst|t|S)aXReturn the entire *Pixel Data* as an :class:`~numpy.ndarray`. .. versionadded:: 2.1 Parameters ---------- ds : pydicom.dataset.Dataset The :class:`Dataset` containing an :dcm:`Image Pixel ` module and the *Pixel Data* to be converted. Returns ------- numpy.ndarray The contents of (7FE0,0010) *Pixel Data* as an :class:`~numpy.ndarray` with shape (rows, columns), (rows, columns, components), (frames, rows, columns), or (frames, rows, columns, components) depending on the dataset. )r get_pixeldatar)r!r!r"as_arraysr/reshapec#sjj}|tvr!|tvrd}n |tvrd}nd}td|dgd}fdd|D}|r:td d |tt t|}t d |j d |t d d}td}t} ttj} ttj} tj|dD]} || |d} |ttfvrtjrt| }|dd}tt|d| }| |}|r|s|jkrd|_| t|} t| |} | | } t!| |} | j"| kr| | } |s| Vqqj#dkr| $j%j&Vqq|t'kr| $j#j%j&} | (dddVqq| $j%j&j#VqqdS)alYield a *Pixel Data* frame from `ds` as an :class:`~numpy.ndarray`. .. versionadded:: 2.1 Parameters ---------- ds : pydicom.dataset.Dataset The :class:`Dataset` containing an :dcm:`Image Pixel ` module and the *Pixel Data* to be converted. reshape : bool, optional If ``True`` (default), then the returned :class:`~numpy.ndarray` will be reshaped to the correct dimensions. If ``False`` then no reshaping will be performed. Yields ------- numpy.ndarray A single frame of (7FE0,0010) *Pixel Data* as an :class:`~numpy.ndarray` with an appropriate dtype for the data. Raises ------ AttributeError If `ds` is missing a required element. RuntimeError If the plugin required to decode the pixel data is not installed. zpylibjpeg-openjpegzpylibjpeg-libjpegz pylibjpeg-rlez)Unable to convert the Pixel Data as the 'z' plugin is not installed) BitsAllocatedRowsColumnsPixelRepresentationSamplesPerPixelPhotometricInterpretation PixelDatacsg|]}|vr|qSr!r!).0elemr)r!r" sz#generate_frames..zbUnable to convert the pixel data as the following required elements are missing from the dataset: z, z Decoding z encoded Pixel Data using F)warn()number_of_framesr) is_signedT precisionrN)) file_metaTransferSyntaxUID _DECODERS_OPENJPEG_SYNTAXES_LIBJPEG_SYNTAXES RuntimeErrorAttributeErrorjoinrrLOGGERdebugnamerr group_datasetr int BitsStoredr1frame_generatorr7rrr APPLY_J2K_CORRECTIONSr setdefaultr4viewnp left_shiftastype right_shiftdtyper5r0r2r3r transpose)r'r0r$pluginrequired_elementsmissingdecoder nr_frames pixel_modulerX bits_storedbits_allocatedframearrparamj2k_sign j2k_precisionshiftr!r)r"r sf             r cCs`t|d}|t|}t|t|}td||}tt|d|D] \}}|||||<q!|S)aReturn a :class:`numpy.ndarray` of the pixel data. .. versionadded:: 2.1 Parameters ---------- ds : pydicom.dataset.Dataset The :class:`Dataset` containing an :dcm:`Image Pixel ` module and the *Pixel Data* to be converted. Returns ------- numpy.ndarray The contents of (7FE0,0010) *Pixel Data* as a 1D array. pixelsrF)r rrTemptyr rangezipr )r' expected_len frame_lenrcgenerate_offsetsrboffsetr!r!r"r.4s   r.)r'rrr-)T)@__doc__copyrloggingtypingrrcollections.abcrpydicom.datasetrrrTr ImportErrorrr pylibjpeg.utilsrropenjpeg HAVE_OPENJPEGlibjpeg HAVE_LIBJPEGrleHAVE_RLErr pydicom.encapsr rPpydicom.pixels.utilsr r r rr pydicom.uidrrrrrrrrrr getLoggerrJ HANDLER_NAMErDrFrE _RLE_SYNTAXESr% DEPENDENCIESboolr#r&r*r,r/r.r!r!r!r"s 0         0    m