o iC@s^dZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl Z ddl mZddlmZddlmZdd lmZdd lmZmZmZdd lmZmZd Zed ZedZde de fddZ!dede fddZ"de de fddZ#de fddZ$    d:dede de%dBde&d e dBde f d!d"Z'ddde#dfdede de%dBde&d#e e ge fd e dBde fd$d%Z(     d;dede de%dBde&d&e&d e dBde fd'd(Z) dz Produce runnable python code which can recreate DICOM objects or files. Can run as a script to produce code for an entire file, or import and use specific functions to provide code for pydicom DICOM classes N)cast)Callable)deque)dictionary_keyword) DataElement)Dataset)BaseTag)BYTES_VR AMBIGUOUS_VRVR) filespec_helpfilespec_parser z(.)([A-Z][a-z]+)z([a-z0-9])([A-Z])namereturncCstd|}td|S)z:Convert name from CamelCase to lower_case_with_underscoresz\1_\2) first_cap_resub all_cap_relower)rs1rI/mnt/sdb/aimis/docanh/lib/python3.10/site-packages/pydicom/util/codify.pycamel_to_underscore)s rtagcCsd|jdd|jddS)z'String of tag value as (0xgggg, 0xeeee)z(0x04Xz, 0x))groupelement)rrrrtag_repr0srcCs0t|}|dd}|dd}|dd}|S)zCallable to reduce some names in code to more readable short form :arg name: a sequence variable name or sequence item name :return: a shorter version of name if a known conversion, else return original name control_pointcp referencereffraction_groupfrxn_gp)rreplace)rrrrdefault_name_filter5s    r&cCsd}d}d}t|||fS)zpCode the import statements needed by other codify results :return: a string of import statement lines zimport pydicomz4from pydicom.dataset import Dataset, FileMetaDatasetz%from pydicom.sequence import Sequence) line_termjoin)line1line2line3rrr code_importsEsr,dsFdataelem dataset_name exclude_sizeinclude_private var_namesc Cs|jtjkrt|||||dSd}zt|j}Wn ty#d}Ynw|jdkr/t|j}nt|j}|rY|jt t Btj hvrYt |jt tBsYt|j|krYdt|jd}|rg|d|d|}|St|j} |j} |d | d | d |d }|S) aCode lines for a single DICOM data element Parameters ---------- dataelem : DataElement The DataElement instance to turn into code dataset_name : str The variable name of the Dataset containing `dataelem` exclude_size : int | None If specified, values longer than this (in bytes) will only have a commented string for a value, causing a syntax error when the code is run, and thus prompting the user to remove or fix that line. var_names: deque | None Used internally to ensure unique variable names in nested sequences. Returns ------- str A string containing code to recreate the data element If the data element is a sequence, calls code_sequence r2TFATz# XXX Array of z bytes excluded. = z .add_new(z, 'z', r)r SQ code_sequencerrKeyErrorrvaluereprr r US_SS isinstanceintfloatlen) r.r/r0r1r2 have_keywordkeywordvaluereplinervrrrr code_dataelemQs4       rF name_filtercsdurtdtdtffdd }g}|j}|j} | dd} zt|j} Wnty7d|jd } Ynw|d|d | || } | | } || } || d ||d | d | t |D]\}}| ddd}| ddd}t ||rtt ||}nt ||rtt ||}nt|d}|d|d | d| d|| dd|}|||}t ||||d} |}||d|| d|d||ddqk t|S)aCode lines for recreating a Sequence data element Parameters ---------- dataelem : DataElement The DataElement instance whose value is the Sequence dataset_name : str Variable name of the dataset containing the Sequence exclude_size : int, optional If not ``None``, values longer than this (in bytes) will only have a commented string for a value, causing a syntax error when the code is run, and thus prompting the user to remove or fix that line. include_private: bool If ``False`` (default) private elements are skipped, otherwise private data elements will be coded. name_filter: Callable[[str], str] A callable taking a sequence name or sequence item name, and returning a shorter name for easier code reading var_names: deque | None Used internally to ensure unique variable names in nested sequences. Returns ------- str A string containing code lines to recreate a DICOM sequence Nrrcs.tt|d}|dkr|S|d|S)Nr_)rrcount)r name_countr3rr unique_namesz"code_sequence..unique_namez SequenceTag08xz# z = Sequence()r5r6SequenceIndexNumberrHz:  _sequencer3rz.append(r)rstrr:rr%rrr9append enumeratehasattrgetattr code_datasetpop splitlinesextendr'r()r.r/r0r1rGr2rLlinesseqseq_name seq_item_name seq_keywordseq_var orig_seq_varir- index_keywordnumber_keyword index_strds_name code_item code_splitrr3rr8sV%           r8 is_file_metac Cs|durt}g}|r dnd}||||D]"}|s!|jjr!qt|||||d} || |jtjkr:|dqt|rI|ddkrI|t |S)aReturn Python code for creating `ds`. Parameters ---------- ds : pydicom.dataset.Dataset The dataset to codify. dataset_name : str, optional The Python variable name to use for the dataset, default ``'ds'``. exclude_size : int, optional If not ``None``, values longer than this (in bytes) will only have a commented string for a value, causing a syntax error when the code is run, and thus prompting the user to remove or fix that line. include_private : bool, optional If ``False`` (default) private elements are skipped, otherwise private data elements will be coded. is_file_meta : bool, optional ``True`` if `ds` contains file meta information elements. var_names: deque, optional Used internally to ensure unique variable names in nested sequences. Returns ------- str The codified dataset. Nz = FileMetaDataset()z = Dataset()r3rM) rrVr is_privaterFr r7r@r[r'r() r-r/r0r1rlr2r^ds_classr. code_linerrrrZ s$"       rZfilenamecCstj|dd}t|||S)aWrite a complete source code file to recreate a DICOM file Parameters ---------- filename : str Complete path and filename of a DICOM file to convert exclude_size : int |None If not None, values longer than this (in bytes) will only have a commented string for a value, causing a syntax error when the code is run, and thus prompting the user to remove or fix that line. include_private : bool If ``False`` (default), private elements are skipped If ``True``, private data elements will be coded. Returns ------- str A string containing code lines to recreate the entire DICOM file T)force)pydicomdcmreadcode_file_from_dataset)rqr0r1r-rrr code_fileFs rvc Csg}|d}|rd|dnd}|d|d||d|t|dt|d rL|d t|jd ||d d }|||d|d t|||d}|||dt|d rl|d|j\}} |d|d| dt|S)aWrite a complete source code file to recreate a DICOM file Parameters ---------- ds : Dataset A pydicom Dataset to convert exclude_size : int |None If not None, values longer than this (in bytes) will only have a commented string for a value, causing a syntax error when the code is run, and thus prompting the user to remove or fix that line. include_private : bool If ``False`` (default), private elements are skipped If ``True``, private data elements will be coded. Returns ------- str A string containing code lines to recreate the entire DICOM file rqz DICOM file ''znon-file datasetz# -*- coding: utf-8 -*-z# Coded version of z+# Produced by pydicom codify utility scriptrM file_metaz# File meta info data elementsT)rlz# Main data elements)r0r1zds.file_meta = file_metazds.set_original_encoding(z, r) getrVr,rXrZrxoriginal_encodingr'r() r-r0r1r^rq identifier code_metacode_ds implicit_vr little_endianrrrrubs<                ruparserdefault_exclude_sizecCsr|jdttd|jddtjddddtjd |jd d t|d |d d|jddddd|jdddddS)Nfilespec)helptypeoutfile?wzUTF-8)encodingzQFilename to write Python code to, if not specified then code is written to stdout)nargsrrdefaultz-ez--exclude-sizez4Exclude binary data larger than specified (default: z bytes))rrrz-pz--include-private store_truez:Include private data elements (default is to exclude them))actionrz-sz --save-aszjSpecify the filename for ds.save_as(save_filename); otherwise the input name + '_from_codify' will be used)r) add_argumentr r argparseFileTypesysstdoutr>)rrrrrset_parser_argumentss@    rargsc Cst|jdkr td|jd\}}|j}|r%t|ts%tdt|t|p)||j|j }|j r6|j }nt j |\}}|dd}d|d}||7}z|jjd kr_td |jjd Wn tyiYnw|j|dS) NrHz1Codify can only work on a single DICOM file inputrz&Codify can only code a Dataset, not a _from_codifyz.dcmz ds.save_as(r'z', enforce_file_format=True)zzWriting code to file 'rw)r@rNotImplementedErrorrqr=rrrur0r1save_asospathsplitextrrprintAttributeErrorwrite) rr-rrqcode_strsave_as_filenamebaserI save_linerrr do_codifys2     rcCs2tjdd|dd}t||t||dS)a7Create Python code according to user options Parameters: ----------- default_exclude_size : int Values longer than this will be coded as a commented syntax error args : List[str], optional Command-line arguments to parse. If ``None`` then :attr:`sys.argv` is used. z-Produce python/pydicom code from a DICOM filez>Binary data (e.g. pixels) larger than --exclude-size (default zy bytes) is not included. A dummy line with a syntax error is produced. Private data elements are not included by default.) descriptionepilogN)rArgumentParserrr parse_args)rrrrrrmains  r__main__d)r)r-NFN)r-NFFN)NF)N)3__doc__ros.pathrrertypingrcollections.abcr collectionsrrspydicom.datadictrpydicom.dataelemrpydicom.datasetr pydicom.tagrpydicom.valuerepr r r pydicom.cli.mainr r r'compilerrrUrrr&r,r>boolrFr8rZrvrurr Namespacerlistr__name__rrrrs          F  v >  F , $