o
i7 @ sX d Z ddlmZ ddlmZmZ z ddlZddlZdZW n e y' dZY nw ddl
m Z ddl m
Z
dd lmZmZ ddlZerHdd
lmZ d Zd d
iZejjgZdefddZde defddZ dddefddZ dddefddZ d,ddde dd fd d Z d e!de"e# fd d!Z$ d,d"e!d#e#d$e#d%e#d&e#d'e de%fd(d)Z&d"e!de%fd*d+Z'dS )-ah Use the `numpy `_ package to convert RLE lossless *Pixel
Data* to a :class:`numpy.ndarray`.
**Supported transfer syntaxes**
* 1.2.840.10008.1.2.5 : RLE Lossless
**Supported data**
The RLE handler supports the conversion of data in the (7FE0,0010)
*Pixel Data* element to a 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 | N | Required |
+-------------+---------------------------+------+--------------+----------+
| (0028,0006) | PlanarConfiguration | 1C | 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, 32 | Required |
+-------------+---------------------------+------+--------------+----------+
| (0028,0103) | PixelRepresentation | 1 | 0, 1 | Required |
+-------------+---------------------------+------+--------------+----------+
)unpack)
TYPE_CHECKINGcastNTF)generate_frames) warn_and_log) pixel_dtype
get_nr_frames)Datasetz RLE Losslessnumpy)zhttps://numpy.org/NumPyreturnc C s t S )z8Return ``True`` if the handler has its dependencies met.)HAVE_RLE r r ]/mnt/sdb/aimis/docanh/lib/python3.10/site-packages/pydicom/pixel_data_handlers/rle_handler.py is_available? s 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_syntaxD 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_RGBP 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_RGBY r r >rle_segment_order
np.ndarrayc s j j}|tvr tdg d} fdd|D }|r$tdd| tt j}tt j }t
dd}tt j }tt j } t
}
t j|d D ]} t| || |||} |
| qLt|
t }
t rld
_td |
S ) av Return an :class:`numpy.ndarray` of the *Pixel Data*.
Parameters
----------
ds : dataset.Dataset
The :class:`Dataset` containing an Image Pixel module and the RLE
encoded *Pixel Data* to be converted.
rle_segment_order : str
The order of segments used by the RLE decoder when dealing with *Bits
Allocated* > 8. Each RLE segment contains 8-bits of the pixel data,
and segments are supposed to be ordered from MSB to LSB. A value of
``'>'`` means interpret the segments as being in big endian order
(default) while a value of ``'<'`` means interpret the segments as
being in little endian order which may be possible if the encoded data
is non-conformant.
Returns
-------
numpy.ndarray
The decoded contents of (7FE0,0010) *Pixel Data* as a 1D array.
Raises
------
AttributeError
If `ds` is missing a required element.
NotImplementedError
If `ds` contains pixel data in an unsupported format.
ValueError
If the actual length of the pixel data doesn't match the expected
length.
zgUnable to convert the pixel data as the transfer syntax is not supported by the RLE pixel data handler.) PixelData
BitsAllocatedRowsColumnsPixelRepresentationSamplesPerPixelc s g | ]}| vr|qS r r ).0elemr r r
s z!get_pixeldata..zbUnable to convert the pixel data as the following required elements are missing from the dataset: z, F)warn)number_of_framesRGBr ) file_metaTransferSyntaxUIDr NotImplementedErrorAttributeErrorjoinr intr r" r r r bytearrayr r _rle_decode_frameextendnp
frombufferr r PhotometricInterpretation)r r r required_elementsmissingnr_bits
nr_samples nr_framesrowscols
pixel_dataframeimarrr r r
get_pixeldatab s8
r@ headerc C sl t | dkr
tdtd| dd d }|dkr!td| d td
| d | dd|d }t|S )
a Return a list of byte offsets for the segments in RLE data.
**RLE Header Format**
The RLE Header contains the number of segments for the image and the
starting offset of each segment. Each of these numbers is represented as
an unsigned long stored in little-endian. The RLE Header is 16 long words
in length (i.e. 64 bytes) which allows it to describe a compressed image
with up to 15 segments. All unused segment offsets shall be set to zero.
As an example, the table below describes an RLE Header with 3 segments as
would typically be used with 8-bit RGB or YCbCr data (with 1 segment per
channel).
+--------------+---------------------------------+------------+
| Byte offset | Description | Value |
+==============+=================================+============+
| 0 | Number of segments | 3 |
+--------------+---------------------------------+------------+
| 4 | Offset of segment 1, N bytes | 64 |
+--------------+---------------------------------+------------+
| 8 | Offset of segment 2, M bytes | 64 + N |
+--------------+---------------------------------+------------+
| 12 | Offset of segment 3 | 64 + N + M |
+--------------+---------------------------------+------------+
| 16 | Offset of segment 4 (not used) | 0 |
+--------------+---------------------------------+------------+
| ... | ... | 0 |
+--------------+---------------------------------+------------+
| 60 | Offset of segment 15 (not used) | 0 |
+--------------+---------------------------------+------------+
Parameters
----------
header : bytes
The RLE header data (i.e. the first 64 bytes of an RLE frame).
Returns
-------
list of int
The byte offsets for each segment in the RLE data.
Raises
------
ValueError
If there are more than 15 segments or if the header is not 64 bytes
long.
References
----------
DICOM Standard, Part 5, :dcm:`Annex G`
@ z(The RLE header can only be 64 bytes longz' for big endian (default),
'<' for little endian (non-conformant).
Returns
-------
bytearray
The frame's decoded data in little endian and planar configuration 1
byte ordering (i.e. for RGB data this is all red pixels then all
green then all blue, with the bytes for each pixel ordered from
MSB to LSB when reading left to right).
zUUnable to decode RLE encoded pixel data with a (0028,0100) 'Bits Allocated' value of NrB zPThe number of RLE segments in the pixel data doesn't match the expected amount (z vs. z
segments)rF rH zJThe amount of decoded RLE segment data doesn't match the expected amount (z bytes)z:The decoded RLE segment contains non-conformant padding - z bytes expectedr )
r+ rN rI rJ appendr/ rangereversed_rle_decode_segmentr )rO r: rP r8 r7 rQ rM rL bytes_per_sampledecodedstride
sample_numberle_gen byte_offsets byte_offsetiisegment
actual_lengthstartr r r r0 sf $
r0 c C s t | } t }d}|j}z5 | | d }|d7 }|dkr/|| ||d d| |d7 }n|dk rA|| ||| ||7 }q tyK Y |S w )zReturn a single segment of decoded RLE data as bytearray.
Parameters
----------
data : bytes
The segment data to be decoded.
Returns
-------
bytearray
The decoded segment.
r TrH i )r/ r1
IndexError)rO resultpos
result_extend header_byter r r rV d s&
rV )r )(__doc__structr typingr r r
r2 r
ImportErrorpydicom.encapsr pydicom.miscr pydicom.pixels.utilsr r pydicom.uidpydicompydicom.datasetr HANDLER_NAME DEPENDENCIESuid RLELosslessr boolr strr r r r@ bytesrK r. rN r/ r0 rV r r r r sT $
NI
q