o i,@sdZddlmZddlZddlmZmZddlmZm Z ddl m Z ddl m ZddlmZmZd d eDZd e ed e ed eefddZeeeeeeffZeeefZeeeeeffZGdddZGdddZddeDZ e !ddeDee Z"dS)z"Access code dictionary information)chainN)castAny)KeysViewIterable)Code)concepts) name_for_cid cid_conceptscCsi|]\}}||qSr ).0kvr r I/mnt/sdb/aimis/docanh/lib/python3.10/site-packages/pydicom/sr/codedict.py srsourcefiltersreturncs8stt|SddDttfdd|DS)aReturn a sorted list of filtered str. Parameters ---------- source : Iterable[str] The iterable of str to be filtered. filters : Iterable[str] An iterable containing patterns for which values are to be included in the results. Returns ------- List[str] A sorted list of unique values from `source`, filtered by including case-insensitive partial or full matches against the values in `filters`. cSsg|]}|qSr lowerr fr r r 'z_filtered..c3s*|]tfddDrVqdS)c3s|] }|vVqdSNrrvalr r *z&_filtered...N)any)r rrrr*s(z_filtered..)sortedset)rrr r r _filtereds  r#c@seZdZdZdZdeddfddZedeee ffdd Z d ee Bde fd d Z de efd dZdede efddZdede fddZede fddZedefddZdefddZedefddZdefddZde efddZdS) CollectionzjInterface for a collection of concepts, such as SNOMED-CT, or a DICOM CID. .. versionadded:: 3.0 z{} = {}namerNcCsT|ds||_t||_nd|dd|_tt|dd|_i|_dS)a6Create a new collection. Parameters ---------- name : str The name of the collection, should either be a key in the ``sr._concepts_dict.concepts`` :class:`dict` or a CID name for a CID in ``sr._cid_dict.cid_concepts`` such as ``"CID1234"``. CIDN) upper startswith_nameCONCEPTS _scheme_data CID_CONCEPTSint _cid_data _conceptsselfr%r r r__init__;s    zCollection.__init__cs$jsfddD_jS)z1Return a :class:`dict` of {SR identifiers: codes}csi|]}|t|qSr )getattr)r r%r2r rrVsz'Collection.concepts..)r0dirr5r r5rrRszCollection.conceptsitemcCs@t|trzt||}Wn tyYdSw|}||jvS)aZChecks whether a given code is a member of the collection. Parameters ---------- item : pydicom.sr.coding.Code | str The code to check for as either the code or the corresponding keyword. Returns ------- bool Whether the collection contains the `code` F) isinstancestrr4AttributeErrorrvalues)r2r7coder r r __contains__Zs  zCollection.__contains__cCsTddtt|tjD}ddtt|tjD}t|}t||B|BS)zReturn a list of available concept keywords. List of attributes is used, for example, in auto-completion in editors or command-line environments. cSh|]}|dqSrr r rr r r xrz%Collection.__dir__..cSr>r?r r@r r rrAyr)inspect getmemberstype isroutineisdatadescriptorr"r6r!)r2methspropssr_namesr r r__dir__rs zCollection.__dir__rcGs(|jrtt|j|St|j|S)aReturn an sorted list of concept keywords based on a partial match. Parameters ---------- filters : str Zero or more string arguments to the function. Used for case-insensitive match to any part of the SR keyword. Returns ------- list of str The matching keywords. If no `filters` are used then all keywords are returned. )is_cidr#r from_iterabler/r;r,)r2rr r rr6~s zCollection.dirc s|jdrfdd|jD}|stdd|jt|dkr6tdd|jdd ||d }tt t |}t|dkrTt |d \}}n8t |jd d fd d|D}t|dkrd dd|D}tdd|jd||d \}}t ||d |dSz tt |j} Wntytdd|jdwt| dkrd | } tdd|jd| t | d }| |\} } t || |jdS)a~Return the :class:`~pydicom.sr.Code` corresponding to `name`. Parameters ---------- name : str A camel case version of the concept's code meaning, such as ``"FontanelOfSkull" in the SCT coding scheme. Returns ------- pydicom.sr.Code The :class:`~pydicom.sr.Code` corresponding to `name`. r&csg|] \}}|vr|qSr r )r schemekeywordsr%r rrs z*Collection.__getattr__..zNo matching code for keyword 'z' in z/Multiple schemes found to contain the keyword 'z: z, rr'Ncs$g|]\}}|dvr||fqS)rPr )r r<r)cidr rrscss|]}|dVqdSrNr r@r r rrz)Collection.__getattr__..z"Multiple codes found for keyword ')valuemeaningscheme_designatorz ' in scheme ''z': )r%r)r/itemsr:len RuntimeErrorjoinr CIDValueTyper+listr.rr,KeyErrorkeys) r2r%matchesrM identifiersr<r_matchescodesentries code_valuesrUcidsr )rQr%r __getattr__sd          zCollection.__getattr__cCs |jdS)z:Return ``True`` if the collection is one of the DICOM CIDsr&)r%r)r5r r rrKs zCollection.is_cidcC|jS)z"Return the name of the collection.)r*r5r r rr%zCollection.namecs.fddjD}jdd|S)z*Return a representation of the collection.csg|] \}}j||qSr ) repr_formatformatr r%conceptr5r rrs z'Collection.__repr__.. )rrXr%r[)r2rr r5r__repr__s zCollection.__repr__cCrh)z0Return the scheme designator for the collection.rOr5r r rrVrizCollection.scheme_designatorcs`tdd|jDd}tdd|jDd}tdd|jDd}t|d}t|d}t|d}|jrud |d |d |d |jg}|d d dd|dddd|dfdd|j Dn6d |d |d d|jg}|d d d|ddd|dfdd|j Dd|S)z1Return a string representation of the collection.css|]}t|VqdSrrY)r nr r rrrSz%Collection.__str__..cs|] }t|dVqdSrRrpr cr r rrrcsrs)rPNrprtr r rrr z{:z} {:z} {} AttributerSchemeMeaningz ---------z----z------z-------rnc3s&|]\}}j|g|RVqdSrrkrlfmtr rr s  zScheme: c3s*|]\}}||d|dVqdS)rrrNr|rlr}r rrs  ) maxrr_r;rKr%appendrkr[rX)r2 len_names len_codes len_schemessr r}r__str__s4    zCollection.__str__cCst|S)zReturn a list of valid names for auto-completion code. Used in IPython, so that data element names can be found and offered for autocompletion on the IPython command line. )r6r5r r r trait_namesszCollection.trait_names)__name__ __module__ __qualname____doc__rjr9r3propertydictrrboolr=r]rJr6rgrKr%rorVrrr r r rr$3s& L 'r$c@sreZdZdZdeeddfddZedee fddZ d e de fd d Z dee fd d Z dee fddZdS)ConceptszWManagement class for the available concept collections. .. versionadded:: 3.0 collectionsrNcCsdd|D|_dS)zCreate a new concepts management class instance. Parameters ---------- collections : list[Collection] A list of the available concept collections. cSsi|]}|j|qSr rOrtr r rr5rz%Concepts.__init__..N) _collections)r2rr r rr3-szConcepts.__init__cCs |jS)z6Return the names of the available concept collections.rr_r5r r rr7s zConcepts.collectionsr%cCsP|drd|dd}||jvr|j|Stdt|jd|d)zReturn the concept collection corresponding to `name`. Parameters ---------- name : str The scheme designator or CID name for the collection to be returned. r&r'NrWz' object has no attribute ')r(r)rr:rDrr1r r rrg<s  zConcepts.__getattr__cCdd|jDS)z/Return a list of available scheme designations.cSsg|] }|ds|qSr&r)rtr r rrPz$Concepts.schemes..rr5r r rschemesNzConcepts.schemescCr)z%Return a list of available CID names.cSsg|] }|dr|qSrrrtr r rrTrz!Concepts.CIDs..rr5r r rCIDsRrz Concepts.CIDs)rrrrr]r$r3rrr9rrrgrrr r r rr's rcCsg|]}t|qSr r$)r designatorr r rrXrrccs|] }td|VqdS)r&Nr)r rQr r rrZsr)#r itertoolsrrBtypingrrcollections.abcrrpydicom.sr.codingrpydicom.sr._concepts_dictrr+pydicom.sr._cid_dictr r r-rX cid_for_namer9r]r#rtupler.r\ ConceptsTypeSnomedMappingTyper$rrextendrcr r r rs&   " u1