o
i& @ s d Z ddlmZ ddlZddlmZ ddlmZmZ er"ddl m
Z
zddl Z dZ W n e
y5 dZ Y nw zddlZdd lmZ dZeed
ZW n
e
yV dZdZY nw dd lmZ dd lmZmZ ddlZdd
lmZmZmZ ddl m Z m Z m Z m Z dZ!dddZ"ej#j$ej#j%ej#j&ej#j'ej#j(ej#j)ej#jej#jgZ*ej#j$gZ+de,fddZ-ddde,fddZ.ddde,fddZ/d ede,fd d Z0d+d d!Z1d,d$d%Z2ddde3fd&d'Z4d-d)d*Z5dS ).zkUse the `GDCM `_ Python package to
decode pixel transfer syntaxes.
)deepcopyN)NamedTemporaryFile)
TYPE_CHECKINGcast)DatasetTF) DataElementSetByteStringValue)config)generate_framesgenerate_fragmented_frames)UIDJPEG2000JPEG2000Lossless)get_expected_length pixel_dtypeget_j2k_parameters
get_nr_framesGDCM)zhttps://numpy.org/NumPy)z&https://sourceforge.net/projects/gdcm/r )numpygdcmreturnc C s t otS )z8Return ``True`` if the handler has its dependencies met.)HAVE_NP HAVE_GDCM r r ^/mnt/sdb/aimis/docanh/lib/python3.10/site-packages/pydicom/pixel_data_handlers/gdcm_handler.py is_availableA s r dsr c C dS )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_RGBF r c C r )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_RGBO r! r" transfer_syntaxc C s | t v S )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.
)SUPPORTED_TRANSFER_SYNTAXES)r# r r r supports_transfer_syntaxX s r% r c C s | j j}ttdd}|jrCtj }t| dd}t | j
|d}|D ]}|D ]}t }| | |
| q(q$|| |S | | j
|S )a Return a ``gdcm.DataElement`` for the *Pixel Data*.
Parameters
----------
ds : dataset.Dataset
The :class:`~pydicom.dataset.Dataset` containing the *Pixel
Data*.
Returns
-------
gdcm.DataElement
The converted *Pixel Data* element.
i Fwarnnumber_of_frames) file_metaTransferSyntaxUIDr r Tag
is_compressedSequenceOfFragmentsNewr r PixelDataFragmentr AddFragmentSetValue__ref__) r tsyntax data_element fragments nr_frames fragment_genframe_fragments
fragment_datafragmentr r r create_data_elementd s$
r> r7
gdcm.Imagec C s t }t| dd}||dkrdnd || j| j|f || t j | j}|
t | | j j }t j
t|}|t
| t | j| j| j| j| j}|| d| v re|| j |S )a1 Return a ``gdcm.Image``.
Parameters
----------
ds : dataset.Dataset
The :class:`~pydicom.dataset.Dataset` containing the Image
Pixel module.
data_element : gdcm.DataElement
The ``gdcm.DataElement`` *Pixel Data* element.
Returns
-------
gdcm.Image
Fr' PlanarConfiguration)r Imager SetNumberOfDimensions
SetDimensionsColumnsRowsSetDataElementPhotometricInterpretation GetPIType SetPhotometricInterpretationr+ r, TransferSyntax GetTSTypestr__str__SetTransferSyntax PixelFormatSamplesPerPixel
BitsAllocated
BitsStoredHighBitPixelRepresentationSetPixelFormatSetPlanarConfigurationrC )r r7 imager* pi_typer6 ts_type pixel_formatr r r create_image s*
r^ c C s t }t| dd}|r't|tr'|| | s tdtt|
S | d}| d |d< | j |_ t
ddd
}|| W d n1 sLw Y ||j | s_td|
}d}t|j |S ) zReturn the pixel data from `ds` as a str.
Used for GDCM < 2.8.8.
Parameters
----------
ds : pydicom.dataset.Dataset
The dataset to create the str from.
Returns
-------
str
The UTF-8 encoded pixel data.
filenameNz GDCM could not read DICOM image( r1 wbF)delete)r ImageReadergetattr
isinstancerO SetFileNameRead TypeErrorr GetImage GetBuffer
group_datasetr+ r save_asnameosremove)r readerfnamenewt pixel_strr r r _get_pixel_str_fileio s(
ru
numpy.ndarrayc C s t stdtrt| }t| |}| }nt| }|dd}t| }| j dkr/|d d }t
||krM||d }t t
|t jrL|d| }n t| }t j
||d}t| d } |j| krmtd
|j d | | jj}
tjr|
ttfv rt| } tt| j| d } t | }
t t |
d
| j }|
dd}|s| j!dkrt t | j"| }t#| $d}d|_!t|}|%||> %||? }t&| rd| _ t d|' S )a Use the GDCM package to decode *Pixel Data*.
Returns
-------
numpy.ndarray
A correctly sized (but not shaped) array of the entire data volume
Raises
------
ImportError
If the required packages are not available.
TypeError
If the image could not be read by GDCM or if the *Pixel Data* type is
unsupported.
AttributeError
If the decoded amount of data does not match the expected amount.
z-The GDCM handler requires both gdcm and numpyzutf-8surrogateescape YBR_FULL_422rA rB N)dtypepixelszAmount of pixel data z" does not match the expected data r) precision is_signedr@ r` r RGBrv )(r ImportErrorHAVE_GDCM_IN_MEMORY_SUPPORTr> r^ rj ru encoder rJ lenr any
frombufferbyter sizeAttributeErrorr+ r, r APPLY_J2K_CORRECTIONSr
r r nextr
r1 r r int
setdefaultrU rW rT r rk astyper" copy)r gdcm_data_element
gdcm_imagert pixel_bytearrayexpected_length_bytespadding numpy_dtypearrexpected_length_pixelsr6 r9
codestreamparams
j2k_precisionj2k_signshift pixel_modulery r r r
get_pixeldata sR
r )r r r r )r r r7 r r r? )r r r rv )6__doc__r r rn tempfiler typingr r pydicom.datasetr r r r~ r r r hasattrr pydicomr pydicom.encapsr
r pydicom.uidr r
r pydicom.pixels.utilsr r r r HANDLER_NAME DEPENDENCIESuidJPEGBaseline8BitJPEGExtended12Bit JPEGLosslessJPEGLosslessSV1JPEGLSLosslessJPEGLSNearLosslessr$ $should_convert_these_syntaxes_to_RGBboolr r r" r% r> r^ rO ru r r r r r sb
#).