o i3@sUddlmZmZmZddlmZmZddlmZddl m Z m Z ddl m Z ddlmZddlmZmZddlmZerQdd lmZdd lmZGd d d eZGd ddZdddedefddZddddddeeefdeeeBdBdddeddf ddZddddddeeefdeeeBdBdddeddf ddZdd dd!dddeeefdeeeBdBddd"ee Bd#e!ed$fdBdeddfd%d&Z"ddd'dddeeefdeeeBdBddd#eee!ed$ffdBdeddfd(d)Z#eZ$ee%d*< ee$_ee$_dS)+)Any TYPE_CHECKINGProtocol)MutableSequenceCallable)config) dictionary_VRprivate_dictionary_VR)BytesLengthException) warn_and_log)BaseTag_LUT_DESCRIPTOR_TAGS)VR)Dataset)RawDataElementc @sNeZdZddddedeeefdeeeBdBdedBded df d d Z dS) RawDataHook.encodingdsrawdatarNrkwargsreturncKsdS)N)selfrrrrrrrC/mnt/sdb/aimis/docanh/lib/python3.10/site-packages/pydicom/hooks.py__call__szRawDataHook.__call__) __name__ __module__ __qualname__rdictstrrrrrrrrrrs  rc@sNeZdZdZd ddZdededdfdd Zded eee fddfd d Z dS)HooksaManagement class for callback functions. .. versionadded:: 3.0 .. warning:: New instances of this class should not be created, instead use the instance imported with ``from pydicom.hooks import hooks``. **Available Hooks** For conversion of raw element data to :class:`~pydicom.dataelem.DataElement` using :func:`~pydicom.dataelem.convert_raw_data_element`: * ``"raw_element_vr"``: function to perform VR lookups for raw elements, default :func:`raw_element_vr`. * ``"raw_element_value"``: function to convert raw element values to types appropriate for the VR, default :func:`raw_element_value`. * ``"raw_element_kwargs"``: `kwargs` :class:`dict` passed to the callback functions. rNcCs||i|_dS)z$Initialize a new ``Hooks`` instance.N)raw_element_kwargs)rrrr__init__5s zHooks.__init__hookfunccCsDt|std|dkr||_dS|dkr||_dStd|d)aRegister the callback function `func` to a hook. Example ------- .. code-block:: python from pydicom import dcmread from pydicom.hooks import hooks, raw_element_value_fix_separator hooks.register_callback( "raw_element_value", raw_element_value_fix_separator ) kwargs = {"target_VRs": ("DS", "IS")} hooks.register_kwargs("raw_element_kwargs", kwargs) ds = dcmread("path/to/dataset.dcm") Parameters ---------- hook : str The name of the hook to register the function to, allowed values ``"raw_element_vr"`` and ``"raw_element_value"``. func : Callable The callback function to use with the hook. Only one callback function can be used per hook. For details on the required function signatures please see the documentation for the corresponding calling function. z"'func' must be a callable functionraw_element_valueraw_element_vrUnknown hook ''N)callable TypeErrorr'r( ValueError)rr%r&rrrregister_callback;s  zHooks.register_callbackrcCsBt|tstdt|jd|dkr||_dStd|d)aRegister a `kwargs` :class:`dict` to be passed to the corresponding callback function(s). Parameters ---------- hook : str The name of the hook to register `kwargs` to, allowed value ``"raw_element_kwargs"``. kwargs : dict[str, Any] A :class:`dict` containing keyword arguments to be passed to the hook's corresponding callback function(s). z'kwargs' must be a dict, not 'r*r#r)N) isinstancer r,typerr#r-)rr%rrrrregister_kwargsbs  zHooks.register_kwargs)rN) rrr__doc__r$r!rr.r rr1rrrrr"s  "'r"rzDataset | NonetagrcCsn|jrtjS|dur4|jd@r4|jd>|jd?B}||d}|r4zt||jWSty3Ytj Swtj S)aReturn the VR for a known private tag, otherwise "UN". Parameters ---------- ds : Dataset, optional The dataset needed for the private creator lookup. If not given, "UN" is returned. tag : BaseTag The private tag to lookup. The caller has to ensure that the tag is private. Returns ------- str "LO" if the tag is a private creator, the VR of the private tag if found in the private dictionary, or "UN". Ni) is_private_creatorrLOelementgroupgetr valueKeyErrorUN)rr3private_creator_tagprivate_creatorrrr_private_vr_for_tagxs  rANrrrrrrcKs|j}|durHzt|j}WnityG|jjr t||j}n%|jjdkr*tj}nd|j}tj j tj kr;t|tj }t |dYn1w|tj krxtjrx|jjr[t||j}n|jdusgt|jdkrxzt|j}Wn tywYnw||d<dS)a Determine the VR to use for `raw`. .. versionadded:: 3.0 Default callback function for the ``"raw_element_vr"`` hook. Parameters ---------- raw : RawDataElement The raw data element to determine the VR for. data : dict[str, Any] A dict to store the results of the VR lookup, which should be added as ``{"VR": str}``. ds : pydicom.dataset.Dataset | None The dataset the element belongs to. **kwargs: dict[str, Any] Additional keyword arguments. Nrz.VR lookup failed for the raw element with tag z - setting VR to 'UN'ir)rrr3r= is_privaterAr9ULrsettingsreading_validation_modeRAISEr>r replace_un_with_known_vrr<len)rrrrrvrmsgrrrr(s4     r(c Ksddlm}|d}z||||}WnMty(}z t|d|jd}~wty_}z,|d|jd|jd} tjsFt| d t| d tj |d<|j }WYd}~nd}~ww|jt vr|rt |t rz|ddkr{|dd 7<Wn tyYnw||d <dS) aConvert the encoded value for `raw` to an appropriate type. .. versionadded:: 3.0 Will set the VR to **UN** if unable to convert the value. Default callback function for the ``"raw_element_value"`` hook. Parameters ---------- raw : RawDataElement The raw data element to determine the value for. data : dict[str, Any] A dict to store the results of the value conversion, which should be added as ``{"value": Any}``. encoding : str | MutableSequence[str] | None The character set encoding to use for text VRs. **kwargs: dict[str, Any] Additional keyword arguments. r convert_valuerz in tag Nz% This occurred while trying to parse z according to VR 'z'.z[ To replace this error with a warning set pydicom.config.convert_wrong_length_to_UN = True.z Setting VR to 'UN'.ir<)pydicom.valuesrLNotImplementedErrorr3r rrconvert_wrong_length_to_UNr r>r<r r/list Exception) rrrrrrLrIr<excrJrrrr's>      r',)r separator target_VRsrTrU.cKsf|d}|r%||vr%t|jtr%t|tr|d}|j|j|dd}t||f||d|dS)a>Convenience function to fix values with an invalid multivalue separator. .. versionadded:: 3.0 Alternative callback function for the ``"raw_element_value"`` hook. Example ------- Fix **DS** and **IS** elements that use an invalid ":" character as the multivalue separator:: from pydicom import dcmread from pydicom.hooks import hooks, raw_element_value_fix_separator hooks.register_callback( "raw_element_value", raw_element_value_fix_separator, ) hooks.register_kwargs( "raw_element", {"target_VRs": ("DS", "IS"), "separator": b":"}, ) ds = dcmread("path/to/dataset.dcm") Parameters ---------- raw : RawDataElement The raw data element to determine the value for. data : dict[str, Any] A dict to store the results of the value conversion, which should be added as ``{"value": Any}``. encoding : str | MutableSequence[str] | None The character set encoding to use for text VRs. separator : str | bytes, optional The invalid separator character to be replaced by an ASCII backslash (0x5C). target_VRs : tuple[str, ...], optional The VRs the fix should apply. **kwargs: dict[str, Any] Additional keyword arguments. rascii\)r<rN)r/r<bytesr!encode_replacereplacer')rrrrrTrUrrIrrrraw_element_value_fix_separators 5  r\)rrUc Ksddlm}zt||f||d|WdStyQ}z0|d}|rK||vrK||D]} z|| ||d<| |d<WWYd}~dStyJYq+w|d}~ww)aConvenience function to retry value conversion using a different VR. .. versionadded:: 3.0 Alternative callback function for the ``"raw_element_value"`` hook. Example ------- Retry the value conversion for **DS** elements using **US** or **SS**:: from pydicom import dcmread from pydicom.hooks import hooks, raw_element_value_retry hooks.register_callback( "raw_element_value", raw_element_value_retry, ) hooks.register_kwargs( "raw_element", {"target_VRs": {"DS": ("US", "SS")}}, ) ds = dcmread("path/to/dataset.dcm") Parameters ---------- raw : RawDataElement The raw data element to determine the value for. data : dict[str, Any] A dict to store the results of the value conversion, which should be added as ``{"value": Any}``. encoding : str | MutableSequence[str] | None The character set encoding to use for text VRs. target_VRs : dict[str, tuple[str, ...]], optional The ``{VRs the fix should apply: tuple of alternative VRs to try}``. **kwargs: dict[str, Any] Additional keyword arguments. rrKrrr<N)rMrLr'rQ) rrrrrUrrLrRrI candidaterrrraw_element_value_retryTs" 0   r^hooks)&typingrrrcollections.abcrrpydicomrpydicom.datadictrr pydicom.errorsr pydicom.miscr pydicom.tagr r pydicom.valuereprpydicom.datasetrpydicom.dataelemrrr"r!rAr r(r'rXtupler\r^r___annotations__rrrrs       Z&  @  C  C  B