o 0 i@sdZddlmZmZmZmZmZmZddlm Z m Z ddl m Z ee e e fZee e e fZee e fZdZGdddZGdd d e e e e fZGd d d Zd ed edefddZGdddZde dee fddZdS)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@seZdZdZdddZdeeeefdeefddZ d e deefd d Z d eedd fd dZ d e de deefddZdS) TemplateConfz~Template Configuration Allows customization for different uses of Template parse() must return a Tree instance. NcCs ||_dSN)_parse)selfparser_/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/lark/tree_templates.py__init__s zTemplateConf.__init__varreturncCsVt|tr t|St|tr)|jdkr)t|jdkr)t|jdtr)t|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_vartemplatecCs4t|tr|js J||}t|tstd|S)Nz+template parser must return a Tree instance)rrrr TypeErrorrrrrr _get_tree3s    zTemplateConf._get_treeTemplatecCs t||dS)N)conf)r!rrrr__call__=s zTemplateConf.__call__treecCs||}|rt|tstd|||iSt|tr$||kr"iSdSt|tr.t|ts8Jd|d||j|jkrjt|jt|jkrji}t|j|jD]\}}| ||}|durbdS| |qQ|SdS)zAReturns dict of {var: match} if found a match, else None z6Template variables can only match Tree instances. Not Nz template=z tree=) rrrrrrrrzip_match_tree_templateupdate)rrr$Z template_varrest1t2matchesrrrr&@s&   (   z!TemplateConf._match_tree_templater )__name__ __module__ __qualname____doc__rrrrrr TreeOrCoder r#Branch MatchResultr&rrrrr s " r csLeZdZdedeeeefddffdd Zdeeffdd ZZ S) _ReplaceVarsr"varsrNcst||_||_dSr )superr_conf_vars)rr"r4 __class__rrr_s  z_ReplaceVars.__init__csPt|||}|j|}|r&z|j|WSty%td|dw|S)Nz"No mapping for template variable ())r5 __default__r6rr7KeyErrorr )rrrmetar$rr8rrr;ds   z_ReplaceVars.__default__) r,r-r.r rrrrr; __classcell__rrr8rr3^s&r3c@seZdZdZefdeedefddZdede e fddZ dede e eee ffd d Zd eeeefdeefd d ZdS)r!aRepresents a tree template, 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"r r$)rr$r"rrrrxszTemplate.__init__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"r r&r$)rr$rrrmatch|s zTemplate.matchccs8|j|}|D]}||}|r||fVq dS)zISearch for all occurrences of the tree template inside ``tree``. N)r"r Z iter_subtreesr?)rr$subtreer(rrrsearchs    zTemplate.searchr4cCst|j||jS)z(Apply vars to the template tree )r3r"Z transformr$)rr4rrr apply_varsszTemplate.apply_varsN)r,r-r.r/r rrrr0rr2r?rrrArrBrrrrr!ps "& r!r)r*r$cCs>|j|}||D]\}}||}||j|jq |S)z=Search tree and translate each occurrence of t1 into t2. )r"r rArBsetrr)r)r*r$r@r4r(rrr translates  rDc@s8eZdZdZdeeeffddZdeefddZ dS) TemplateTranslatorz;Utility class for translating a collection of patterns translationscCs$tdd|Ds J||_dS)Ncss(|]\}}t|tot|tVqdSr )rr!).0kvrrr s&z.TemplateTranslator.__init__..)allitemsrF)rrFrrrrs zTemplateTranslator.__init__r$cCs$|jD] \}}t|||}q|Sr )rFrLrD)rr$rHrIrrrrDszTemplateTranslator.translateN) r,r-r.r/rr!rrrrDrrrrrEsrEvaluercCs|tr |tSdSr ) startswith_TEMPLATE_MARKERlstrip)rMrrrrsrN)r/typingrrrrrrZlarkrr Zlark.exceptionsr rr1r0r2rOr r3r!rDrErrrrrs   L+