(HJj ndZddlZddlmZddlmZej GddZGddZy) z A parse tree. N)Iterable)replacecdeZdZUdZeed<ejeZ eedfed<dZ e ed<y)TreeNodezj Private class that creates the tree data structure from the orthography profile for parsing. char)default_factorychildrenFsentinelN) __name__ __module__ __qualname____doc__str__annotations__ dataclassesfielddictr r boolW/Users/ahmed/devFolder/claude-voice/.venv/lib/python3.12/site-packages/segments/tree.pyrr s8 I&7k&7&7&MHd3 ?#MHdrrc deZdZdZdeefdZefdedeefdZ de dede de eee ffd Z y ) Treez The parse tree. >>> t = Tree('abcdefg') >>> t.parse('abcde') ['a', 'b', 'c', 'd', 'e'] >>> t = Tree(['ab', 'c', 'de']) >>> t.parse('abcde') ['ab', 'c', 'de'] graphemescbd}tdd|_|D]}||j|y)Ncj|D]'}|jj|t|})d|_y)NT)r setdefaultrr )nodeliners r _multigraphz"Tree.__init__.._multigraph!s-}}//htnE DMrT)r )rroot)selfrr graphemes r__init__z Tree.__init__ s/ ! R$/ !H  8 ,"rrreturnc|j|j|d\}}||d}|rX|j||d|dd}|j|j|d\}}|j|||d}|rX|S)zSegment `line` into graphemes.rN)_parser"appendextend)r#rerrorresidxremris rparsez Tree.parse+s;;tyy$2S34j JJuSV} %ab'C;;tyy#q1DAq JJqMab'C  rr"r.cJt|dk(rg|fSg}d}|}|}|t|krt|jj||}|dz }|snN|jr3|j |||d||z\}}|d|g}|j ||t|krt|r|}||fS)z :param root: Tree node. :param line: String to parse. :param idx: Global counter of characters parsed. :return: (list of parsed graphemes, incremented character count) rr(N)lenr getr r)r+) r#r"rr.r2currrcidxsubparses rr)z Tree._parse8s t9>s7NSY==$$T$Z0D AID}}!%T4;d !K$et  X&SY CczrN)r r r rrrr%rlistr2rinttupler)rrrrrsa  -(3- -&- # c 83SU49c>=Rrr) rrcollections.abcrsegments.errorsr dataclassrrrrrr?s>$# ??r