a d@sdZddlmZmZmZmZmZmZddlm Z m Z ddl m Z ee e e fZdZGdddZGdd d e e e e fZGd d d Zeeed d dZGdddZe ee dddZdS)zThis module defines utilities for matching and translation tree templates. A tree templates is a tree that contains nodes that are template variables. )UnionOptionalMappingDictTupleIterator)Tree Transformer)MissingVariableError$c@seZdZdZdddZeeeefeedddZ e eedd d Z eed dd d Z e e ee ee fdddZdS) TemplateConfzTTemplate Configuration Allows customization for different uses of Template NcCs ||_dSN)_parse)selfparser\/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/lark/tree_templates.py__init__szTemplateConf.__init__)varreturncCsVt|trt|St|trR|jdkrRt|jdkrRt|jdtrRt|jdSdS)zGiven a tree node, if it is a template variable return its name. Otherwise, return None. This method may be overridden for customization Parameters: var: Tree | str - The tree node to test rrN) isinstancestr_get_template_namerdatalenchildren)rrrrrtest_vars  zTemplateConf.test_var)templatercCs0t|tr|jsJ||}t|ts,J|Sr )rrrrrrrrr _get_tree/s    zTemplateConf._get_treeTemplatecCs t||dS)N)conf)r rrrr__call__7szTemplateConf.__call__)rtreercCs||}|r||iSt|tr0||kr,iSdSt|trDt|tsXJd|d||j|jkrt|jt|jkri}t|j|jD],\}}|||}|durdS| |q|SdS)Nz template=z tree=) rrrrrrrzip_match_tree_templateupdate)rrr#Z template_varrest1t2matchesrrrr%:s"  (   z!TemplateConf._match_tree_template)N)__name__ __module__ __qualname____doc__rrrrrr TreeOrCoderr"rr%rrrrr s   r csHeZdZeeeeefddfdd Zeedfdd ZZ S) _ReplaceVarsN)r!varsrcst||_||_dSr )superr_conf_vars)rr!r1 __class__rrrUs z_ReplaceVars.__init__)rcsTt|||}|j|}|rPz |j|WStyNtd|dYn0|S)Nz"No mapping for template variable ())r2 __default__r3rr4KeyErrorr )rrrmetar#rr5rrr8Zs   z_ReplaceVars.__default__) r+r,r-r rrrrr8 __classcell__rrr5rr0Ts"r0c@seZdZdZefeeedddZee e eefdddZ ee e eee eeffddd Zeeeefeed d d Zd S)r aRepresents a tree templates, tied to a specific configuration A tree template is a tree that contains nodes that are template variables. Those variables will match any tree. (future versions may support annotations on the variables, to allow more complex templates) )r#r!cCs||_|||_dSr )r!rr#)rr#r!rrrrnszTemplate.__init__)r#rcCs|j|}|j|j|S)aMatch a tree template to a tree. A tree template without variables will only match ``tree`` if it is equal to the template. Parameters: tree (Tree): The tree to match to the template Returns: Optional[Dict[str, Tree]]: If match is found, returns a dictionary mapping template variable names to their matching tree nodes. If no match was found, returns None. )r!rr%r#)rr#rrrmatchrs zTemplate.matchccs6|j|}|D]}||}|r||fVqdS)zHSearch for all occurances of the tree template inside ``tree``. N)r!rZ iter_subtreesr<)rr#subtreer'rrrsearchs    zTemplate.search)r1rcCst|j||jS)z(Apply vars to the template tree )r0r!Z transformr#)rr1rrr apply_varsszTemplate.apply_varsN)r+r,r-r.r rrrr/rrr<rrr>rr?rrrrr fs ( r )r(r)r#cCs>|j|}||D]"\}}||}||j|jq|S)z=Search tree and translate each occurrance of t1 into t2. )r!rr>r?setrr)r(r)r#r=r1r'rrr translates   rAc@s8eZdZdZeeefdddZeedddZ dS) TemplateTranslatorz;Utility class for translating a collection of patterns ) translationscCs$tdd|DsJ||_dS)Ncss&|]\}}t|tot|tVqdSr )rr ).0kvrrr z.TemplateTranslator.__init__..)allitemsrC)rrCrrrrszTemplateTranslator.__init__)r#cCs$|jD]\}}t|||}q |Sr )rCrJrA)rr#rErFrrrrAszTemplateTranslator.translateN) r+r,r-r.rr rrrrArrrrrBsrB)valuercCs|tr|tSdSr ) startswith_TEMPLATE_MARKERlstrip)rKrrrrsrN)r.typingrrrrrrZlarkrr Zlark.exceptionsr rr/rMr r0r rArBrrrrrs  D+