`iL8dZddlmZddlmZmZddlmZddlm Z ddl m Z ddl m Z m Z dd lmZmZdd lmZGd d ZGd dZdZGddZGddeZGddeZdZdeefdZGddZdZGddZdZdZ Gd d!Z!d"S)#zLProvides functions for the automatic building and shaping of the parse-tree.)List) GrammarErrorConfigurationError)Token)Tree)Transformer_InPlace) _vargs_meta_vargs_meta_inline)partialwraps)productceZdZdZdZdS)ExpandSingleChildc||_dSN) node_builder)selfrs k/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/lark/parse_tree_builder.py__init__zExpandSingleChild.__init__s(cbt|dkr|dS||S)Nrr)lenr)rchildrens r__call__zExpandSingleChild.__call__s1 x==A  A; $$X.. .rN__name__ __module__ __qualname__rrrrrrs2)))/////rrc"eZdZddZdZdZdS)PropagatePositionsNc"||_||_dSr)r node_filter)rrr$s rrzPropagatePositions.__init__s(&rc||}t|tr|j}||}|t |dsXt |d|j|_t |d|j|_t |d|j |_ d|_ t |d|j|_ t |d|j|_ t |d|j |_ |t|}|t |dsXt |d|j|_t |d|j|_t |d |j|_d|_ t |d|j|_t |d|j|_t |d |j|_|S) Nlinecontainer_linecontainer_columncontainer_start_posFend_linecontainer_end_linecontainer_end_columncontainer_end_pos)r isinstancermeta _pp_get_metahasattrgetattrr&column start_posemptyr'r(r)reversedr* end_columnend_posr+r,r-)rrresres_meta first_meta last_metas rrzPropagatePositions.__call__!s)) c4  h xH**844J%x00+$+J8H*/$Z$ZHM&-j:LjN_&`&`HO)0=RT^Th)i)iH&%*HN*1*>NPZP_*`*`',3J@RT^Te,f,f)/6zCXZdZn/o/o,))(8*<*<==I$x44+(/ ;OQZQc(d(dH%*1)=SU^Ui*j*jH''.y:MyO`'a'aH$%*HN.5iAUW`Wi.j.j+07 CY[d[o0p0p--4Y@SU^Uf-g-g* rc|D]}|j||st|tr|jjs |jcSJt|t r|cSt |dr|cSdS)N __lark_meta__)r$r.rr/r5rr1r>)rrcs rr0zPropagatePositions._pp_get_metaGs ) )A+D4D4DQ4G4G+!T"" )v|"6MMM"Au%% )O,, )((((( ) ) )rr)rrrrrr0r rrr"r"sG''''$$$L ) ) ) ) )rr"ct|rtt|S|durtS|durdStd|z)N)r$TFz*Invalid option for propagate_positions: %r)callabler r"r)options rmake_propagate_positionsrCSsW)v>>>> 4!! 5t IFR S SSrceZdZdZdZdS) ChildFilterc0||_||_||_dSr)r to_include append_none)rrGrHrs rrzChildFilter.__init___s($&rcg}|jD]?\}}}|r |dg|zz }|r|||jz }$|||@|jr|dg|jzz }||SrrGrappendrHrrrfilteredi to_expandadd_nones rrzChildFilter.__call__ds&*o - - "Ay( .TFX-- -HQK00 ,,,,   2 !11 1H  ***rNrr rrrErE^s2''' +++++rrEceZdZdZdZdS)ChildFilterLALR`Optimized childfilter for LALR (assumes no duplication in parse tree, so it's safe to change it)cg}|jD]O\}}}|r |dg|zz }|r!|r|||jz }&||j}4|||P|jr|dg|jzz }||SrrJrLs rrzChildFilterLALR.__call__xs&*o - - "Ay( .TFX-- -4 44HH'{3HH ,,,,   2 !11 1H  ***rN)rrr__doc__rr rrrRrRus)jj+++++rrRceZdZdZdZdZdS)ChildFilterLALR_NoPlaceholdersrSc"||_||_dSr)rrG)rrGrs rrz'ChildFilterLALR_NoPlaceholders.__init__($rcg}|jD]C\}}|r!|r|||jz }||j}(|||D||Sr)rGrrKr)rrrMrNrOs rrz'ChildFilterLALR_NoPlaceholders.__call__sz O - -LAy -4 44HH'{3HH ,,,,  ***rNrrrrUrrr rrrWrWs8ff%%% + + + + +rrWcF|j o|jdS)N_)is_termname startswith)syms r_should_expandrbs!{? 7sx223777r_empty_indicesc\|r|dt|ksJdd|D}d|dD}t|t|dzksJ|t|fndgt|dzz}g}d}t |D]G\}} |||z }|s| jr| js'||t| |fd}H||t|z }|s9t|t|kstd|DrB|s|rt|rtnt||Sttd |DSdS) NFc3NK|] }tt|V!dSr)strint).0bs r z,maybe_create_child_filter..s.88ACAKK888888rc,g|]}t|Sr )r)rioness r z-maybe_create_child_filter..s<<.s*@l@l~qR[\]@l@l@l@l@l@lrcg|] \}}}||f Sr r )rirNxr]s rrnz-maybe_create_child_filter..s";[;[;[uq1QF;[;[;[r)countrjoinsplit enumerater^ filter_outrKrbanyr rErRrW) expansionkeep_all_tokens ambiguousrcs empty_indicesrG nones_to_addrNras rmaybe_create_child_filterrs1##E**c)nn<<<< GG88888 8 8<    q."5"5|D E E ELM#i..11L]Z3y>>99S@l@lak@l@l@l=l=l9  ]Y ])H;;*Vbcc c9;[;[PZ;[;[;[\\ \ :9rceZdZdZdZdZdS)AmbiguousExpanderaDeal with the case where we're expanding children ('_rule') into a parent but the children are ambiguous. i.e. (parent->_ambig->_expand_this_rule). In this case, make the parent itself ambiguous with as many copies as there are ambiguous children, and then copy the ambiguous children into the right parents in the right places, essentially shifting the ambiguity up the tree.c0||_||_||_dSr)r tree_classrO)rrOrrs rrzAmbiguousExpander.__init__s($"rcld}gt|D]C\}}||r3|jvr||dDs|Sfdt|D}dfdt |DS)Nc8t|do |jdkS)Ndata_ambigr1r)ts r_is_ambig_treez2AmbiguousExpander.__call__.._is_ambig_trees1f%%.s-eeeXQANN%..eeercTg|]$}t|%Sr )rlist)rifrs rrnz.AmbiguousExpander.__call__..s-)_)_)_$*;*;DGG*D*D)_)_)_r)rvrOrKexpand_kids_by_datarrr)rrrrNrexpandr{s` @rrzAmbiguousExpander.__call__s = = = !(++ 4 4HAu~e$$ 4&&$$Q'''))(333 /$$X.. .eeeeQZ[cQdQdeeex)_)_)_)_gW]N^)_)_)_```rNr[r rrrrsCff### aaaaarrcnfdt|D}|rtt||SdS)NcXg|]&\}}s|jr|jt|$|'Sr )r^rwrb)rirNrarzs rrnz3maybe_create_ambiguous_expander..s`iiivq##i.1ki>AniSabeSfSfiiiir)rvr r)rryrzrOs ` rmaybe_create_ambiguous_expanderrsaiiii9!5!5iiiIA()Z@@@AArceZdZdZdZdZdS)AmbiguousIntermediateExpanderaq Propagate ambiguous intermediate nodes and their derivations up to the current rule. In general, converts rule _iambig _inter someChildren1 ... _inter someChildren2 ... someChildren3 ... to _ambig rule someChildren1 ... someChildren3 ... rule someChildren2 ... someChildren3 ... rule childrenFromNestedIambigs ... someChildren3 ... ... propagating up any nested '_iambig' nodes along the way. c"||_||_dSr)rr)rrrs rrz&AmbiguousIntermediateExpander.__init__rYrcdfd|}|r$fd|D}d|S|S)Nc8t|do |jdkS)Nr_iambigr)rs r_is_iambig_treez?AmbiguousIntermediateExpander.__call__.._is_iambig_trees5&))EejI.E Erc>|r|dr|d}g}|jD]r}|j}|r#|D]}|xj|ddz c_||z }7 d|j|ddz}||s|SdSdS)a Recursively flatten the derivations of the parent of an '_iambig' node. Returns a list of '_inter' nodes guaranteed not to contain any nested '_iambig' nodes, or None if children does not contain an '_iambig' node. rrN_inter)rrrK) r iambig_noderesult grandchild collapsedrnew_tree_collapse_iambigrrs rrz@AmbiguousIntermediateExpander.__call__.._collapse_iambigs OOHQK88 &qk "-"600J 0 01D E EI 0%.;;E!NNhqrrl:NNN)+#'??8Z=PS[\]\^\^S_=_#`#` h////     rcDg|]}|jSr )rr)rir?rs rrnz:AmbiguousIntermediateExpander.__call__..1s)PPPt00<<PPPrr)rr)rrrprocessed_nodesrrs` @@rrz&AmbiguousIntermediateExpander.__call__s F F F       0%$X..  >PPPPiPPPO??8_== =  ***rNr[r rrrrs=&&P%%%!+!+!+!+!+rrc<tfd}|S)NcDtj|}|Sr)rr)rtreefuncs rrzinplace_transformer..f9s$DM8,,tDzzr)r )rrs` rinplace_transformerr8s3 4[[[ Hrctus turtdtfd}|S)Nz0Meta args not supported for internal transformerc |dSrr )rrr_wrappers rrzapply_visit_wrapper..fEswtT8T222r)r r NotImplementedErrorr )rr_rrs``` rapply_visit_wrapperrAsb+,>!>!>!"TUUU 4[[333333[3 Hrc$eZdZddZdZddZdS)ParseTreeBuilderFc||_||_||_||_t |||_dSr)rpropagate_positionsr{maybe_placeholdersr_init_builders rule_builders)rrulesrrr{rs rrzParseTreeBuilder.__init__LsD$#6 ""4!$"5"5e"<"<==rc#Kt|j}|D]}|j}|j}|j}t t d|o|j ott|j ||j |j r|j nd||j ot|j|j ||j ot!t"|jg}||fVdSr)rCroptionsrzexpand1rfilteraliasrrryr{rr}rrr r)rrrrulerrzexpand_single_child wrapper_chains rrzParseTreeBuilder._init_buildersTs6t7OPP & &DlG%5O")/  $7TZN=N)$./4>txuL\V[b[p[pRVWW#t#B4?TXTbds#t#tZ7+H$/#Z#Z /""M % % % % % & &rNc i}t|dd r fd}n|j}|jD]\}}|jp|jjp |jj} t||}t|dd}|t|||}n$t|trt|}n #t$rt||}YnwxYw|D] } | |}||vrtd|d|||<|S)N __default__c||dSrr )rrdefault_handlers rdefault_callbackz:ParseTreeBuilder.create_callback..default_callbackks&tXt<<>>>&&&$rrN)"rUtypingr exceptionsrrlexerrrrvisitorsr r r functoolsr r itertoolsrrr"rCrErRrWrbboolrrrrrrrr rrrsRR88888888))))))55555555%$$$$$$$////////5)5)5)5)5)5)5)5)nTTT++++++++.+++++k+++,+++++[+++&888]UYZ^U_]]]]8aaaaaaaaBAAAN+N+N+N+N+N+N+N+d      ::::::::::r