B Afxal6ã@sªdZddlZddlmZddlmZddlmZmZddl m Z m Z ddl m Z mZmZmZmZGd d „d e ƒZGd d „d e ƒZGd d„dƒZeZeZdd„Zdd„ZdS)zã pint.systems ~~~~~~~~~~~~ Functions and classes related to system definitions and conversions. :copyright: 2016 by Pint Authors, see AUTHORS for more details. :license: BSD, see LICENSE for more details. éNé)Ú_babel_systems)Ú babel_parse)Ú DefinitionÚUnitDefinition)ÚDefinitionSyntaxErrorÚRedefinitionError)ÚSharedRegistryObjectÚSourceIteratorÚgetattr_maybe_raiseÚloggerÚto_units_containerc@sŠeZdZdZe d¡Zdd„Zedd„ƒZ dd„Z d d „Z d d „Z d d„Z edd„ƒZdd„Zdd„Zdd„Zeefdd„ƒZdd„ZdS)ÚGroupaÂA group is a set of units. Units can be added directly or by including other groups. Members are computed dynamically, that is if a unit is added to a group X all groups that include X are affected. The group belongs to one Registry. It can be specified in the definition file as:: @group [using , ..., ] ... @end z6@group\s+(?P\w+)\s*(using\s(?P.*))*cCsP||_tƒ|_tƒ|_tƒ|_||jj|j<|dkrF|jjd |¡d|_dS)a, :param name: Name of the group. If not given, a root Group will be created. :type name: str :param groups: dictionary like object groups and system. The newly created group will be added after creation. :type groups: dict[str | Group] ÚrootN) ÚnameÚsetÚ _unit_namesÚ _used_groupsÚ_used_byÚ _REGISTRYÚ_groupsÚ add_groupsÚ_computed_members)Úselfr©rú+lib/python3.7/site-packages/pint/systems.pyÚ__init__0s zGroup.__init__cCsN|jdkrHt|jƒ|_x$| ¡D]\}}|j|jO_q Wt|jƒ|_|jS)z…Names of the units that are members of the group. Calculated to include to all units in all included _used_groups. N)rrrÚiter_used_groupsÚmembersÚ frozenset)rÚ_ÚgrouprrrrUs    z Group.memberscCs.d|_|jj}x|jD]}|| ¡qWdS)z?Invalidate computed members in this Group and all parent nodes.N)rrrrÚinvalidate_members)rÚdrrrrr"fs zGroup.invalidate_membersccsHt|jƒ}|jj}x0|rB| ¡}||}||jO}|||fVqWdS)N)rrrrÚpop)rÚpendingr#rr!rrrrms  zGroup.iter_used_groupscCs&x | ¡D]\}}||kr dSq WdS)NTF)r)rÚ group_namerr rrrÚ is_used_groupvszGroup.is_used_groupcGs&x|D]}|j |¡qW| ¡dS)zAdd units to group.N)rÚaddr")rÚ unit_namesÚ unit_namerrrÚ add_units|s zGroup.add_unitscCs t|jƒS)N)rr)rrrrÚnon_inherited_unit_namesƒszGroup.non_inherited_unit_namescGs&x|D]}|j |¡qW| ¡dS)zRemove units from group.N)rÚremover")rr)r*rrrÚ remove_units‡s zGroup.remove_unitscGsb|jj}xL|D]D}||}| |j¡r8td|j|fƒ‚|j |¡|j |j¡qW| ¡dS)zAdd groups to group.z+Cyclic relationship found between %s and %sN) rrr'rÚ ValueErrorrr(rr")rÚ group_namesr#r&ÚgrprrrrŽs   zGroup.add_groupscGsD|jj}x.|D]&}||}|j |¡|j |j¡qW| ¡dS)zRemove groups from group.N)rrrr-rrr")rr0r#r&r1rrrÚ remove_groups s   zGroup.remove_groupsc CsJt|ƒ}t|ƒ\}}|j |¡}|dkr4td|ƒ‚| ¡d ¡}| ¡d}|rntdd„| d¡Dƒƒ} nd} g} x¬|D]¤\}} d | krt j | |d } t | t ƒsºt d t| ƒ|d ‚y || ƒWn<tt fk r} z| jdkrî|| _| ‚Wdd} ~ XYnX|  | j¡q||  |  ¡¡q|W||ƒ}|j| Ž| rF|j| Ž|S) aEReturn a Group object parsing an iterable of lines. Parameters ---------- lines : list[str] iterable define_func : callable Function to define a unit in the registry; it must accept a single string as a parameter. Returns ------- Nz!Invalid Group header syntax: '%s'rÚ used_groupscss|]}| ¡VqdS)N)Ústrip)Ú.0Úarrrú Æsz#Group.from_lines..ú,rú=)Ú non_int_typez7Only UnitDefinition are valid inside _used_groups, not )Úlineno)r ÚnextÚ _header_reÚsearchr/Ú groupdictr4ÚtupleÚsplitrZ from_stringÚ isinstancerrÚstrrr;Úappendrr+r)ÚclsÚlinesZ define_funcr:r;ÚheaderÚrrÚgroupsr0r)ÚlineZ definitionÚexr1rrrÚ from_lines«s@          zGroup.from_linescCst||ƒ|jS)N)r r)rÚitemrrrÚ __getattr__ês zGroup.__getattr__N)Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚreÚcompiler=rÚpropertyrr"rr'r+r,r.rr2Ú classmethodÚfloatrLrNrrrrrs %    >rc@sneZdZdZe d¡Zdd„Zdd„Zdd„Z e d d „ƒZ d d „Z d d„Z dd„Zdd„Zeefdd„ƒZdS)ÚSystemaâA system is a Group plus a set of base units. Members are computed dynamically, that is if a unit is added to a group X all groups that include X are affected. The System belongs to one Registry. It can be specified in the definition file as:: @system [using , ..., ] ... @end The syntax for the rule is: new_unit_name : old_unit_name where: - old_unit_name: a root unit part which is going to be removed from the system. - new_unit_name: a non root unit which is going to replace the old_unit. If the new_unit_name and the old_unit_name, the later and the colon can be omitted. z7@system\s+(?P\w+)\s*(using\s(?P.*))*cCs4||_i|_tƒ|_tƒ|_d|_||jj|j<dS)zH :param name: Name of the group :type name: str N)rÚ base_unitsrÚ derived_unitsrrrZ_systems)rrrrrr s zSystem.__init__cCs t|jƒS)N)Úlistr)rrrrÚ__dir__)szSystem.__dir__cCs:t||ƒt|j|jd|dƒ}|dk r.|St|j|ƒS)Nr )r Úgetattrrr)rrMÚurrrrN,s  zSystem.__getattr__c Cs~|jj}|jdkrxtƒ|_xP|jD]F}y|j||jO_Wq"tk rft d  ||j ¡¡Yq"Xq"Wt |jƒ|_|jS)Nz!Could not resolve {} in System {}) rrrrrrÚKeyErrorr ZwarningÚformatrr)rr#r&rrrr3s   zSystem.memberscCs d|_dS)z?Invalidate computed members in this Group and all parent nodes.N)r)rrrrr"GszSystem.invalidate_memberscGs|jt|ƒO_| ¡dS)zAdd groups to group.N)rrr")rr0rrrrKszSystem.add_groupscGs|jt|ƒ8_| ¡dS)zRemove groups from group.N)rrr")rr0rrrr2QszSystem.remove_groupscCs0|r*|jtkr*t|j}t|ƒ}|j|S|jS)z!translate the name of the system.)rrrZmeasurement_systems)rÚlocalerrrrÚ format_babelWs   zSystem.format_babelcs¶t|ƒ}t|ƒ\}}|j |¡}|dkr4td|ƒ‚| ¡d ¡}| ¡d}|rntdd„| d¡Dƒƒ} nd} i} g} x|D]ú\}} |  ¡} d| kr2|  d¡\} ‰|  ¡ˆ ¡} ‰ˆt |ˆƒd ƒkrÜtd | ƒ‚t || ƒd |j ƒ}ˆ|krtd ƒ‚‡fd d „|  ¡Dƒ}d |ˆ|| <|| ˆ<q‚| } t || ƒd ƒ}t |ƒd kr\tdƒ‚t|ƒ ¡\‰}| d |i| ˆ<q‚W||ƒ}|j| Ž|jjf| Ž|jt| ƒO_|S)Nz!Invalid System header syntax '%s'rr3css|]}| ¡VqdS)N)r4)r5r6rrrr7osz$System.from_lines..r8)rú:rz>In `%s`, the unit at the right of the `:` must be a root unit.z(Old unit must be a component of new unitcs"i|]\}}|ˆkrd||“qS)éÿÿÿÿr)r5Únew_unitÚvalue)Úold_unitrrú ”sz%System.from_lines..zIThe new base must be a root dimension if not discarded unit is specified.)r r<r=r>r/r?r4r@rArCr rÚitemsÚlenÚdictÚpopitemrrYÚupdaterZr)rErFZ get_root_funcr:r;rGrHrrIr0Zbase_unit_namesZderived_unit_namesrJreZnew_unit_expandedZ new_unit_dictZ old_unit_dictrfÚsystemr)rgrrL_sR          zSystem.from_linesN)rOrPrQrRrSrTr=rr\rNrUrr"rr2rbrVrWrLrrrrrXïs  rXc@s$eZdZdd„Zdd„Zdd„ZdS)ÚListercCs ||_dS)N)r#)rr#rrrr¸szLister.__init__cCst|j ¡ƒS)N)r[r#Úkeys)rrrrr\»szLister.__dir__cCst||ƒ|j|S)N)r r#)rrMrrrrN¾s zLister.__getattr__N)rOrPrQrr\rNrrrrro·srocsG‡fdd„dtƒ}|S)NcseZdZ”ZdS)z build_group_class..GroupN)rOrPrQrr)ÚregistryrrrÈsr)Ú_Group)rqrr)rqrÚbuild_group_classÇsrscsG‡fdd„dtƒ}|S)NcseZdZ”ZdS)z"build_system_class..SystemN)rOrPrQrr)rqrrrXÏsrX)Ú_System)rqrXr)rqrÚbuild_system_classÎsru)rRrSZ babel_namesrÚcompatrZ definitionsrrÚerrorsrrÚutilr r r r r rrXrorrrtrsrurrrrÚ s   VI