o i6x @s dZddlZddlZddlmZmZddlmZmZddl m Z ddl m Z ddl mZmZmZmZer:ddlmZd Zid ed ed d ddddddddddddddddddd d!d"d d#d d$d%d&did'dd(dd)dd*dd+dd,dd-dd.dd/d0d1d2d3d!d4d5d6d7d8d8d9d:d;d<d:d:Zd=Zd>Zied?eed@dedAd edBd edCd%edDdedEdedFdedGdedHdedIdedJdedKdedLd!edMd0edNd2edOd5ZdPdQeDZedAed <dRZddTedUedVefdWdXZddTedUedVefdYdZZddTedUedVefd[d\Z SddTed]edUedVefd^d_Z! dd`edaedBdVefdbdcZ"ddZ#eee deZ$dTedfeedge%e&dVefdhdiZ'e'Z(djedfeedge%e&dVefdkdlZ)djedfeedge%e&dVefdmdnZ*dTedfeedVefdodpZ+dTedfeedVefdqdrZ,ddTed`edUedVefdsdtZ-dfdeBeeBdVe.efdudvZ/d`edVefdwdxZ0 dd`edyedBdVdfdzd{Z1dfeed|e.edVe.efd}d~Z2dddee.eBdBdVdfddZ3dS)z6Handle alternate character sets for character strings.N) TYPE_CHECKINGcast)MutableSequenceSequence)config) warn_and_log)TEXT_VR_DELIMS PersonNameVRCUSTOMIZABLE_CHARSET_VR) DataElementiso8859ISO_IR 6z ISO_IR 13 shift_jisz ISO_IR 100latin_1z ISO_IR 101 iso8859_2z ISO_IR 109 iso8859_3z ISO_IR 110 iso8859_4z ISO_IR 126 iso_ir_126z ISO_IR 127 iso_ir_127z ISO_IR 138 iso_ir_138z ISO_IR 144 iso_ir_144z ISO_IR 148 iso_ir_148z ISO_IR 166 iso_ir_166z ISO 2022 IR 6zISO 2022 IR 13ISO 2022 IR 87 iso2022_jpzISO 2022 IR 100zISO 2022 IR 101zISO 2022 IR 109zISO 2022 IR 110zISO 2022 IR 126zISO 2022 IR 127zISO 2022 IR 138zISO 2022 IR 144zISO 2022 IR 148zISO 2022 IR 149euc_krISO 2022 IR 159 iso2022_jp_2zISO 2022 IR 166zISO 2022 IR 58 iso_ir_58 ISO_IR 192UTF8GB18030z ISO 2022 GBKGBKz ISO 2022 58GB2312)r!r$r#s(Bs-A)I(Js$Bs-Bs-Cs-Ds-Fs-Gs-Hs-Ls-Ms-Ts$)Cs$(Ds$)AcCsi|]\}}||qSr)).0kvr)r)E/mnt/sdb/aimis/docanh/lib/python3.10/site-packages/pydicom/charset.py ^sr.)rrr strictvalueerrorsreturnc Cshtd}|}|dks|dkr>d}|D]'}z||}Wn ty(d}Ynwt|dks5dt|kr7d}||7}q|S||d}t|dkrUtd|dt|d t|d@}t|dd dD]M\}}z||}Wnty} z || _t|| _| d } ~ wwt|dkst|d@|Adkr|dkrd nd } d | } td||t|| ||7}qd|S)aConvert a unicode string into JIS X 0201 byte string using shift_jis encodings. shift_jis is a superset of jis_x_0201. So we can regard the encoded value as jis_x_0201 if it is single byte character. Parameters ---------- value : str The unicode string as presented to the user. errors : str The behavior of a character which could not be encoded. If 'strict' is passed, raise an UnicodeEncodeError. If any other value is passed, non ISO IR 14 characters are replaced by the ASCII '?'. Returns ------- bytes The encoded string. If some characters in value could not be encoded to JIS X 0201, and `errors` is not set to 'strict', they are replaced to '?'. Raises ------ UnicodeEncodeError If errors is set to 'strict' and `value` could not be encoded with JIS X 0201. rr/r?rzillegal multibyte sequenceNz ISO IR 14z ISO IR 13Given character is out of ) codecsgetincrementalencoderencodeUnicodeEncodeErrorlenord enumeratestartend) r0r1 encoder_classencoderencodedcbmsbie character_setmsgr)r)r-_encode_to_jis_x_0201gsF         rKcCt|d|dS)z7Convert a unicode string into JIS X 0208 encoded bytes.rr1_encode_to_given_charsetr0r1r)r)r-_encode_to_jis_x_0208rQcCrL)z7Convert a unicode string into JIS X 0212 encoded bytes.rrMrNrPr)r)r-_encode_to_jis_x_0212rRrSrIc Cst|}|dkr|j||dSt|}|}||d}|t|s2t||dt|d|t|dddD];\}}z||} Wnty\} z || _ t|| _ | d} ~ ww| ddt krrt|||t|d||| 7}q;|S)aEncode a unicode string using the given character set. The escape sequence which is located at the end of the encoded value has to vary depending on the value 1 of SpecificCharacterSet. So we have to trim it and append the correct escape sequence manually. Parameters ---------- value : text type The unicode string as presented to the user. character_set: str: Character set for result. errors : str The behavior of a character which could not be encoded. This value is passed to errors argument of str.encode(). Returns ------- bytes The encoded string. If some characters in value could not be encoded to given character_set, it depends on the behavior of corresponding python encoder. Raises ------ UnicodeEncodeError If errors is set to 'strict' and `value` could not be encoded with given character_set. r/rMrr7r5N) python_encodingr:r8r9 startswithENCODINGS_TO_CODESr;r<r>r?r@ESC) r0rIr1encodingrArBrCrGrDrErHr)r)r-rOs:!   rOrXrCcCsHtd}td}|dkr|dur|S|d}d|kr|S|St|dS)aReturn an escape sequence corresponding to the given encoding. If encoding is 'shift_jis', return 'ESC)I' or 'ESC(J' depending on the first byte of encoded. Parameters ---------- encoding : str An encoding is used to specify an escape sequence. encoded : bytes The encoded value is used to choose an escape sequence if encoding is 'shift_jis'. Returns ------- bytes Escape sequence for encoded value. r(r'rNrr6r3)rWrVget)rXrC ESC_ISO_IR_14 ESC_ISO_IR_13 first_byter)r)r-!_get_escape_sequence_for_encodings r])rr)rrr encodings delimiterscst|vrNd}z||WSty.tjjtjkrtd|dt}||YSt yMtjjtjkr<td|d|j|ddYSwd}t ||}d fd d |DS) aDecode an encoded byte `value` into a unicode string using `encodings`. Parameters ---------- value : bytes The encoded byte string in the DICOM element value. encodings : list of str The encodings needed to decode the string as a list of Python encodings, converted from the encodings in (0008,0005) *Specific Character Set*. delimiters : set of int A set of characters or character codes, each of which resets the encoding in `value`. Returns ------- str The decoded unicode string. If the value could not be decoded, and :attr:`~pydicom.config.settings.reading_validation_mode` is not ``RAISE``, a warning is issued, and `value` is decoded using the first encoding with replacement characters, resulting in data loss. Raises ------ UnicodeDecodeError If :attr:`~pydicom.config.settings.reading_validation_mode` is ``RAISE`` and `value` could not be decoded with the given encodings. LookupError If :attr:`~pydicom.config.settings.reading_validation_mode` is ``RAISE`` and the given encodings are invalid. rUnknown encoding '"' - using default encoding insteadz,Failed to decode byte string with encoding 'z2' - using replacement characters in decoded stringreplacerMs(^[^]+|[][^]*)rcsg|]}t|qSr))_decode_fragment)r*fragmentr_r^r)r- usz decode_bytes..) rWdecode LookupErrorrsettingsreading_validation_modeRAISErdefault_encoding UnicodeErrorrefindalljoin)r0r^r_first_encodingregex fragmentsr)rer- decode_bytes*s6#     rtbyte_strcCstz|tr t|||WS||dWSty9tjjtjkr#t dd |d|j|dddYSw)a,Decode a byte string encoded with a single encoding. If `byte_str` starts with an escape sequence, the encoding corresponding to this sequence is used for decoding if present in `encodings`, otherwise the first value in encodings. If a delimiter occurs inside the string, it resets the encoding to the first encoding in case of single-byte encodings. Parameters ---------- byte_str : bytes The encoded string to be decoded. encodings: list of str The list of Python encodings as converted from the values in the Specific Character Set tag. delimiters: set of int A set of characters or character codes, each of which resets the encoding in `byte_str`. Returns ------- str The decoded unicode string. If the value could not be decoded, and :attr:`~pydicom.config.settings.reading_validation_mode` is not set to ``RAISE``, a warning is issued, and the value is decoded using the first encoding with replacement characters, resulting in data loss. Raises ------ UnicodeDecodeError If :attr:`~pydicom.config.settings.reading_validation_mode` is set to ``RAISE`` and `value` could not be decoded with the given encodings. References ---------- * DICOM Standard, Part 5, :dcm:`Sections 6.1.2.4` and :dcm:`6.1.2.5` * DICOM Standard, Part 3, :dcm:`Annex C.12.1.1.2` rz-Failed to decode byte string with encodings: , z1 - using replacement characters in decoded stringrbrM) rUrW_decode_escaped_fragmentrgrmrrirjrkrrp)rur^r_r)r)r-rc|s.  rccs|drdnd}t|d|d}||vs|tkrU|tvr$||S||d}tfddt|Dd}|durP|d||||d|dS||Sd }tj j tj krbt |t |d |d|j|dd d S) zDecodes a byte string starting with an escape sequence. See `_decode_fragment` for parameter description and more information. )s$(s$)Nrc3s |] \}}|vr|VqdSNr))r*idxchr_r)r- sz+_decode_escaped_fragment..rz5Found unknown escape sequence in encoded string valuez - using encoding rbrM)rUCODES_TO_ENCODINGSrYrlhandled_encodingsrgnextr>rrirjrk ValueErrorr)rur^r_ seq_lengthrXindexrJr)r}r-rws"   rwc Cst|D]5\}}z(t||}|dkr |tvr t||d}||}|tvr,|t|d7}|WSty9Yqwt|dkrPzt||WStyOYnwt j j t j kr^| |dtdd|dt||dddS) aEncode a unicode string `value` into :class:`bytes` using `encodings`. Parameters ---------- value : str The unicode string as presented to the user. encodings : list of str The encodings needed to encode the string as a list of Python encodings, converted from the encodings in (0008,0005) *Specific Character Set*. Returns ------- bytes The encoded string. If `value` could not be encoded with any of the given encodings, and :attr:`~pydicom.config.settings.reading_validation_mode` is not ``RAISE``, a warning is issued, and `value` is encoded using the first encoding with replacement characters, resulting in data loss. Raises ------ UnicodeEncodeError If :attr:`~pydicom.config.settings.writing_validation_mode` is set to ``RAISE`` and `value` could not be encoded with the supplied encodings. rrCr5z'Failed to encode value with encodings: rvz1 - using replacement characters in encoded stringrbrM)r>_encode_string_implrr]#need_tail_escape_sequence_encodingsrmr<_encode_string_partsrrriwriting_validation_moderkr:rrp)r0r^rGrXrCescape_sequencer)r)r- encode_strings4      rc Cst}|}t}|red}|D].}zt|||}t|}Wnttfy;}z|j|kr1|j}|}WYd}~q d}~ww|dkrDtdt|d||}|tvrY|t ||d7}||7}||d}|s |t vrq|t |d7}t |S)atConvert a unicode string into a byte string using the given list of encodings. This is invoked if `encode_string` failed to encode `value` with a single encoding. We try instead to use different encodings for different parts of the string, using the encoding that can encode the longest part of the rest of the string as we go along. Parameters ---------- value : str The unicode string as presented to the user. encodings : list of str The encodings needed to encode the string as a list of Python encodings, converted from the encodings in Specific Character Set. Returns ------- bytes The encoded string, including the escape sequences needed to switch between different encodings. Raises ------ ValueError If `value` could not be encoded with the given encodings. rNz:None of the given encodings can encode the first characterr) bytearrayrlrr<UnicodeDecodeErrorr;r?rrr]rbytes) r0r^rCunencoded_part best_encoding max_indexrXerr encoded_partr)r)r-r$s@   !rcCs&|tvr t|||dS|j||dS)zConvert a unicode string into a byte string. If given encoding is in `custom_encoders`, use a corresponding `custom_encoder`. If given encoding is not in `custom_encoders`, use a corresponding python handled encoder. rM)custom_encodersr:)r0rXr1r)r)r-rjsrc Cs|pdg}t|tr|g}n|dd}|dsd|d<g}|D]}z |t|Wq ty;|t|Yq wt|dkrGt||}|S)a}Convert DICOM `encodings` into corresponding Python encodings. Handles some common spelling mistakes and issues a warning in this case. Handles stand-alone encodings: if they are the first encodings, additional encodings are ignored, if they are not the first encoding, they are ignored. In both cases, a warning is issued. Invalid encodings are replaced with the default encoding with a respective warning issued, if :attr:`~pydicom.config.settings.reading_validation_mode` is ``WARN``, or an exception is raised if it is set to ``RAISE``. Parameters ---------- encodings : str or list of str The encoding or list of encodings as read from (0008,0005) *Specific Character Set*. Returns ------- list of str A :class:`list` of Python encodings corresponding to the DICOM encodings. If an encoding is already a Python encoding, it is returned unchanged. Encodings with common spelling errors are replaced by the correct encoding, and invalid encodings are replaced with the default encoding if :attr:`~pydicom.config.settings.reading_validation_mode` is not set to ``RAISE``. Raises ------ LookupError If `encodings` contains a value that could not be converted and :attr:`~pydicom.config.settings.reading_validation_mode` is ``RAISE``. rNrrr5) isinstancestrappendrTKeyError'_python_encoding_for_corrected_encodingr<$_handle_illegal_standalone_encodings)r^ py_encodingsrXr)r)r-convert_encodingss '     rcCsd}td|durd|dd}ntd|dur#d|dd}|rAz t|}t|||WSty@t|tYSwzt||WStyXt|tYSw)aTry to replace the given invalid encoding with a valid encoding by checking for common spelling errors, and return the correct Python encoding for that encoding. Otherwise check if the encoding is already a valid Python encoding, and return that. If both attempts fail, return the default encoding. Issue a warning for the invalid encoding except for the case where it is already converted. Nz ^ISO[^_]IRISO_IRz!^(?=ISO.2022.IR.)(?!ISO 2022 IR )z ISO 2022 IR ) rnmatchrT_warn_about_invalid_encodingrrlr8lookuprh)rXpatched py_encodingr)r)r-rs*     rpatched_encodingcCsV|durtjjtjkrtd|dd|d}n d|d|d}t|dddS) a?Issue a warning for the given invalid encoding. If patched_encoding is given, it is mentioned as the replacement encoding, other the default encoding. If no replacement encoding is given, and :attr:`~pydicom.config.settings.reading_validation_mode` is set to ``RAISE``, `LookupError` is raised. Nr`'raz,Incorrect value for Specific Character Set 'z' - assuming ' stacklevel)rrirjrkrhr)rXrrJr)r)r-rs rrc Cs|dtvr!td|ddd|dddd|ddSttt|ddD]\}}|tvrDtd|d dd||d=q-|S) zCheck for stand-alone encodings in multi-valued encodings. If the first encoding is a stand-alone encoding, the rest of the encodings is removed. If any other encoding is a stand-alone encoding, it is removed from the encodings. rzValue 'zG' for Specific Character Set does not allow code extensions, ignoring: rvr5Nrrz/' cannot be used as code extension, ignoring it)STAND_ALONE_ENCODINGSrrpreversedlistr>)r^rrGrXr)r)r-rs"      relemr dicom_character_setcs|jrdS|s dg}t||jtjkr2|jdkr%tt|j|_dSfdd|jD|_dS|jt vrp|jdkrNt |jt rDdSt |jt |_dSt}|jD]}t |t ra||qT|t |t qT||_dSdS)a Apply the DICOM character encoding to a data element Parameters ---------- elem : dataelem.DataElement The :class:`DataElement` instance containing an encoded byte string value to decode. dicom_character_set : str or list of str or None The value of (0008,0005) *Specific Character Set*, which may be a single value, a multiple value (code extension), or may also be ``''`` or ``None``, in which case ``'ISO_IR 6'`` will be used. Nrr5csg|] }tt|qSr))rr rg)r*vvr^r)r-rf9sz"decode_element..)is_emptyrr PNVMrr r0rgr rrrtrrr)rroutputr0r)rr-decode_elements*         r)r/rz)4__doc__r8rntypingrrcollections.abcrrpydicomr pydicom.miscrpydicom.valuereprr r r pydicom.dataelemr rlrTrrWritemsrVrrrrKrQrSrOr]rrsetintrt decode_stringrcrwrrrrrrrrrr)r)r)r-s         !"#$*      G D %"O > +?F">%