l0jedZddlmZddlmZGddZedGd d eZd ej_edGd d eZdej_edGddeZ de j_edGddeZ de j_edGddeZ de j_dede de fdZdede defdZdS)uAn improved Wadler–Lindig pretty printer. This implementation additionally: - handles new lines in the text to format. - removes some dead code from the canonical implementation. References: (1) Wadler, P., 1998. A prettier printer. Journal of Functional Programming, pp.223-244. (2) Lindig, C. 2000. Strictly Pretty. https://lindig.github.io/papers/strictly-pretty-2000.pdf Inspired by JAX's use of the same references above, but re-implemented from scratch. ) dataclass) ansi_stripc2eZdZdZd dZdeddfdZd d Zd S) AbstractDoczBase class for all document types. For more on the following shorthand methods, see [the methods example](./methods.ipynb). otherreturn ConcatDocc"t||S)zH`doc1 + doc2` offers a convenient shorthand for `ConcatDoc(doc1, doc2)`.)r selfrs _/home/wildlama/miniconda3/envs/lam/lib/python3.11/site-packages/wadler_lindig/_wadler_lindig.py__add__zAbstractDoc.__add__su%%%indentNestDocc$t||S)z]`doc.nest(indent)` offers a convenient shorthand for `NestDoc(doc, indent)`. )r)r)r rs rnestzAbstractDoc.nest"stF++++rGroupDocc t|S)z@`doc.group()` offers a convenient shorthand for `GroupDoc(doc)`.)rr s rgroupzAbstractDoc.group(s~~rN)rrr r )r r)__name__ __module__ __qualname____doc__rintrrrrrrsh &&&&,3,9,,,, rrT)frozenceZdZUdZeed<dS)TextDoczFRepresents an unbroken piece of text to display. May include newlines.textN)rrrrstr__annotations__rrrr!r!-sPP IIIIIrr!z.**Arguments:** - `text`: the string of text. c$eZdZUdZeed<dZdS)BreakDoczIf in vertical mode then this is a valid place to insert a newline. If in horizontal mode then `self.text` will be displayed instead. r"c6d|jvrtddS)N z"Cannot have newlines in BreakDocs.)r" ValueErrorrs r __post_init__zBreakDoc.__post_init__Bs& 49  ABB B  rN)rrrrr#r$r*rrrr&r&:s? IIICCCCCrr&z**Arguments:** - `text`: the string of text to display if a newline is not inserted. Common values are `" "` (for example between elements of a list) or `""` (for example between the final element of a list and a closing ']'). cJeZdZUdZeedfed<dddZdeddfd ZdS) r a\Concatenate multiple documents together, to be displayed one after another. If for example these consist only of `TextDoc`s and other `ConcatDoc`s then there is no implied breaking between them, so the formatted text may exceed the maximum width. You may wish to separate pieces with `BreakDoc`s to indicate this, for example. .childrenN)r,ct|dkr||}n$t|dks|tdt|d|dS)NrzxMust be called as either `ConcatDoc(children=(foo, bar, ...))` or as `ConcatDoc(foo, bar, ...)` or as `foo + bar + ...`.r,)lenr)object __setattr__)r r,argss r__init__zConcatDoc.__init__]si t99q==X-HH YY]]h.F  4X66666rrr c(tg|j|RS)N)r r,r s rrzConcatDoc.__add__gs/$-/////r) rrrrtuplerr$r2rrrrr r OsxK$%%%%(,777770[0[000000rr z**Arguments:** Can be called as any of: - `ConcatDoc(doc1, doc2, doc3, ...)` - `ConcatDoc(children=(doc1, doc2, doc3, ...))` - `doc1 + doc2 + doc3 + ...` c(eZdZUdZeed<eed<dS)rzkIf in vertical mode, increase the indent after each newline by `indent` whilst displaying `child`. childrN)rrrrrr$rrrrrrvs1  KKKKKrraH**Arguments:** - `child`: the child document to display. - `indent`: how much to increase the indent. Frequently `child` will be `ConcatDoc(BreakDoc(""), another_doc)`, so that the first line of `another_doc` will be indented as much as its later lines. See also the [The `(break-group).nest-break` example](./pattern.ipynb). ceZdZUdZeed<dS)raGroups the parts of a child document to be laid out all horizontally together or all vertically together. This decision will persist everywhere outside any child `GroupDoc`s, within which their own local rule is used. For example using `[...]` to denote a grouping: ``` [ foo, bar, [baz, qux] ] ``` then `foo`, `bar` and `[baz, qux]` are laid out vertically, but the sub-group `[baz, qux]` is judged to have enough space, and so is laid out horizontally. r6N)rrrrrr$rrrrrs* rrz:**Arguments:** - `child`: the child document to display. docwidthr c |g}t|dkrS|dkrL|xtdxK\}|tt tt |dz}nxtdx$\}|tt |z}nxtdx'\}| t|nfxtdx\}| |n?xtdx\}| |n}Jt|t|dkr|dkL|dkS)Nrr)default)r.popr!maxmapr splitlinesr&r extendreversedrappendrr#)r8r9todor"r,r6xs r _verticalrEs"eD d))a--EQJJhhjjSj&6&6&A&A&C&CDDaPPPP$$Z--... $$$$$$88 HX..////%"""""" E""""# %% E""""!$c!ff$$u d))a--EQJJ 19rc g}d}d}d}t|g}t|dkr |xtdx\}|}nxtdx\}|}nxt dxn\} || ddd|zz|tt| dddz }nDxtdxZ\} |r|dd|zz|}n|| |tt| z }nxtdx'\} | t| nxtdx5\} } |||| || z }njtdx[\} |r@t| ||z s-|d|| d }n|| n Jt|dk d |S) uBPretty-formats a document using a Wadler–Lindig pretty-printer. **Arguments:** - `doc`: a document to pretty-format as a string. - `width`: a best-effort maximum width to allow. May be exceeded if there are unbroken pieces of text which are wider than this. **Returns:** A string, corresponding to the pretty-formatted document. !!! info We extend the canonical Wadler–Lindig implementation with the ability to handle multiline text. We also remove what seems to be some dead code from their implementation. rTrNr( rF)rr.r<boolrr!rBreplacerrsplitr&r r@rArrEjoin) r8r9outs width_so_farverticalrrC vertical2indent2r"r,r6 extra_indents r pformat_docrTs&DLHF-5SMM?D d))a--hhjj ii$!WW  DLLtcL6H/HIIJJJJt{{4/C/CB/G$H$H I II  $$:KKsV| 3444#)LLKK%%% C 4(8(8$9$99LL  %$$$$$88 HX..////%------  F### E""",&.% 'IeU\5I$J$J 'KK%%%KK&&&$HHKK&&&&!uE d))a--F 774==rN)r dataclassesr_ansirrr!r2r&r rrrrJrEr#rTrrrrWsV""!!!!!, $k   $ C C C C C{ C C C $00000 0008  $k ${(;st&=[========r