o
iO! @ s d Z ddlmZmZ ddlmZ zddlZdZW n e y# dZY nw er,ddl
m Z dZ d d
iZ
d d d d d
dddddd
ZdefddZ d!dddeded fdd Zd!ddd ededd fd d ZdS )"a Use the `numpy `_ package to convert supported *Waveform
Data* to a :class:`numpy.ndarray`.
.. versionadded:: 2.1
**Supported data**
The numpy handler supports the conversion of data in the (5400,0100)
*Waveform Sequence* element to an :class:`~numpy.ndarray` provided the
related :dcm:`Waveform` module elements have values
given in the table below.
+-------------+------------------------------+------+-------------------------+
| Element | Supported |
+-------------+------------------------------+------+ values |
| Tag | Keyword | Type | |
+=============+==============================+======+=========================+
| (003A,0005) | NumberOfWaveformChannels | 1 | N > 0 |
+-------------+------------------------------+------+-------------------------+
| (003A,0010) | NumberOfWaveformSamples | 1 | N > 0 |
+-------------+------------------------------+------+-------------------------+
| (5400,1004) | WaveformBitsAllocated | 1 | 8, 16, 32, 64 |
+-------------+------------------------------+------+-------------------------+
| (5400,1006) | WaveformSampleInterpretation | 1 | SB, UB, MB, AB, SS, US, |
| | | | SL, UL, SV, UV |
+-------------+------------------------------+------+-------------------------+
)
TYPE_CHECKINGcast) GeneratorNTF)DatasetzNumpy Waveformnumpy)zhttps://numpy.org/NumPyint8uint8int16uint16int32uint32int64uint64)
) SB)r UB)r MB)r AB) SS)r US) SL)r UL)@ SV)r UVreturnc C s t S )zXReturn ``True`` if the handler has its dependencies met.
.. versionadded:: 2.1
)HAVE_NP r r U/mnt/sdb/aimis/docanh/lib/python3.10/site-packages/pydicom/waveforms/numpy_handler.py is_available= s r" dsr as_raw)
np.ndarrayNNc # sn d| vr t dtttd | jD ]\} g d} fdd|D }|r3t d| dd | tt jd
}tt j}tt j }tt j}tt
j } || | }
t || f } t
jtt jd |
| d } t
| ||} |s| d
} ttd j}
t|
D ]&\}}|dd}|dd}|dd}| d|f | | | | d|f< q| V qd S )a Yield an :class:`~numpy.ndarray` for each multiplex group in the
*Waveform Sequence*.
.. versionadded:: 2.1
Parameters
----------
ds : pydicom.dataset.Dataset
The :class:`Dataset` containing a :dcm:`Waveform
` module and the *Waveform Sequence* to be
converted.
as_raw : bool, optional
If ``True`` (default), then yield the raw unitless waveform data. If
``False`` then attempt to convert the raw data for each channel to the
quantity specified by the corresponding (003A,0210) *Channel
Sensitivity* unit.
Yields
------
np.ndarray
The waveform data for a multiplex group as an :class:`~numpy.ndarray`
with shape (samples, channels).
WaveformSequence=No (5400,0100) Waveform Sequence element found in the datasetr NumberOfWaveformChannelsNumberOfWaveformSamplesWaveformBitsAllocated WaveformSampleInterpretation WaveformDatac g | ]}| vr|qS r r .0elemitemr r!
l z&generate_multiplex..:Unable to convert the waveform multiplex group with index H as the following required elements are missing from the sequence item: , r NdtypefloatChannelBaseline ChannelSensitivity ?"ChannelSensitivityCorrectionFactor.)AttributeError enumerater listr& joinintr+ r* r) strr, WAVEFORM_DTYPESnp
frombufferbytesr- copyreshapeastypeChannelDefinitionSequenceget)r# r$ iirequired_elementsmissingbytes_per_sample
nr_samples nr_channelsbits_allocatedsample_interpretation expected_lenr: arrseqjjchbaseline sensitivity
correctionr r2 r! generate_multiplexE sF
"r` indexr% c sb d| vrt dttd | j| g d} fdd|D }|r.t d| dd | tt jd
}tt j}tt j}tt j}tt j
} || | }
t || f } t j
tt jd |
| d } t | ||} |s| d
} ttd j}
t|
D ]&\}}|dd}|dd}|dd}| d|f | | | | d|f< qtd| S )aW Return an :class:`~numpy.ndarray` for the multiplex group in the
*Waveform Sequence* at `index`.
.. versionadded:: 2.1
Parameters
----------
ds : pydicom.dataset.Dataset
The :class:`Dataset` containing a :dcm:`Waveform
` module and the *Waveform Sequence* to be
converted.
index : int
The index of the multiplex group to return.
as_raw : bool, optional
If ``True`` (default), then return the raw unitless waveform data. If
``False`` then attempt to convert the raw data for each channel to the
quantity specified by the corresponding (003A,0210) *Channel
Sensitivity* unit.
Returns
-------
np.ndarray
The waveform data for a multiplex group as an :class:`~numpy.ndarray`
with shape (samples, channels).
r& r' r r( c r. r r r/ r2 r r! r4 r5 z#multiplex_array..r6 r7 r8 r Nr9 r; r< r= r> r? r@ .r% )rA r rC r& rD rE r+ r* r) rF r, rG rH rI rJ r- rK rL rM rN rB rO )r# ra r$ rQ rR rS rT rU rV rW rX r: rY rZ r[ r\ r] r^ r_ r r2 r! multiplex_array sB
"
rb )T)__doc__typingr r collections.abcr r rH r ImportErrorpydicom.datasetr HANDLER_NAME DEPENDENCIESrG boolr" r` rE rb r r r r! sD
K