a &0_C@sdZddlmZddlmZddlmZddlmZm Z m Z m Z dZ dZ d Zd Ze d ed ed iZddZddZddZddZddZddZddZddZddZejGd d!d!eZd"d#Zd$d%Zd S)&a>supports walking the history as DAGs suitable for graphical output The most basic format we use is that of:: (id, type, data, [parentids]) The node and parent ids are arbitrary integers which identify a node in the context of the graph returned. Type is a constant specifying the node type. Data depends on type. )absolute_import)nullrev)attr)dagoppycompatsmartsetutilCPGM|:Nc #si}D]}||}fdd|Dfdd|D}ddtD}|D]}||}|durttjstttt||g}||<|s| t |f |qV| fdd|D |qV|t||fVqdS) acset DAG generator yielding (id, CHANGESET, ctx, [parentinfo]) tuples This generator function walks through revisions (which should be ordered from bigger to lower). It returns a tuple for each node. Each parentinfo entry is a tuple with (edgetype, parentid), where edgetype is one of PARENT, GRANDPARENT or MISSINGPARENT. The node and parent ids are arbitrary integers which identify a node in the context of the graph returned. cs h|]}|vr|qSrev.0p)revsr4/usr/lib/python3/dist-packages/mercurial/graphmod.py =zdagwalker..cs,g|]$}|tkr|vr|qSr)rrrpsetrr >szdagwalker..cSsg|] }t|fqSr)PARENTrrrrrCrNc3s|]}|vrt|fVqdSN) GRANDPARENT)rgrrr Srzdagwalker..)parentssortedget isinstancerZbasesetsetrZreachablerootsappend MISSINGPARENTaddextendupdater CHANGESET) reporZgpcacherctxZmparsr"ZmparZgpr)rrr dagwalker*s,        r/c#sJt||D]8}||}fdd|D}|t|t|fVq dS)zcset DAG generator yielding (id, CHANGESET, ctx, [parentids]) tuples This generator function walks the given nodes. It only returns parents that are in nodes, too. cs$h|]}|vrt|fqSr)noderrrZincluderrrbsznodes..N)r&r"rr,r#)r-nodesr0r.r"rr1rr2Ys r2c #sg}i}d}ijdD]f\}}d|vr|dd\}}|dkrb|rbt||i|<q|dkr|r||i|<qrtfdd} ndd} |D]b\} } } } | |vr| | ||| <|d7}| | }| | }|d d fd d | D}|||d<t |D]*\}}|s<|||<n|||<|d7}q$g}t |D]\}}|vr| |}| | |||| dd | dd fnL|| kr\| D]<\}}| |}| | ||| dd | dd fqq\| | | ||f|fV}qd S)axannotates a DAG with colored edge information For each DAG node this function emits tuples:: (id, type, data, (col, color), [(col, nextcol, color)]) with the following new elements: - Tuple (col, color) with column and color index for the current node - A list of tuples indicating the edges between the current node and its parents. rsgraph.swidthscolorcs|iSr)r$branchr)configr-rrrzcolored..cSsiSrrrrrrr6rNcsg|]\}}|vr|qSrr)rZptr)nextrrrrzcolored..r)uiZ configitemsrsplitisdigitint setdefaultisalnumr Z lrucachefuncr'indexpop enumerater$)Zdagr-seenZcolorsZnewcolorkeyvalr4ZsettingZgetconfZcurtypedatar"colZcolorZ addparentsiredgesZecolZeidZbconfptyper)r5r7r-rcoloredhsp                rKc#s|j}||vr||||g}g}|D]D\}} | |kr@q.| |vrT|| q.|| |j|d|j| <q.t|} d| d} |dd|d<fdd|D} |dd<t|dkr4| f| dfd} | d7} ||| | | | ffVd}d7| d7} g} |d=qt|dkrP| ft|dkrp| dft| } | dkr| d7} |j|d||| | | | ffVdS) z9adds edge info to changelog DAG walk suitable for ascii()rrNcsg|]}|fqSr)r?rZnextseenZnodeidxrrrrzasciiedges..\r)rBr'r?stylesr$rIlenr@)rEcharstaterr"rBZ knownparentsZ newparentsrJparentncolswidthrIZ nmorecolsrrMr asciiedgessN          rVcCs2t|D]$\}\}}||kr||df||<qdS)Nr)rA)rIrHstartendrrr_fixlongrightedgessrYc Cs|rr||krr|dkrr|dkr\t|d|}||d|dd}|ddg|||Sddg||dSn*||d} | dkr|| d dSgSdS)Nrr8rrL/ rN)maxr*) echarsidxZpidxrTcoldiffZpdiffZfix_tailrWtail remainderrrr_getnodelineedgestails rbcCs|D]\}}||dkr*d|d|d<q||dkrHd|d|d<q||krf|d||d|<qd|t|krxqd|d|<||kr||}}td|dd|D]}||dkrd||<qqdS)NrrZrLrN+-)rPrange)r]rInodelineZ interlinerWrXrHrrr _drawedges s      rgcCs|d|d}||df|vs,||f|vrL|||d|ddn|ddg||d}|dkr|||d d|S)NrLrr[r)r*)r]r^rTrIlinerarrr_getpaddingline s  ricsd|vrdS|ddd}|r8|ddur8|qt|dddd}|jsZdnd}t|||kr||ddq^|rg}g} |d} t|| dd| dD].\} | dur|dq| dqt t | | t| dd} | dd} || dD]t}dgt|| || d<t t| D]F| d}t|| | <|| krvdn || ||<qBq|jsd d d ndd it|D]4\}d|vrƐqfd d |D|dd<qdd | D}|D]}||=| |qdS)zDraw ending lines for missing parent edges None indicates an edge that ends at between this node and the next Replace with a short line ending in ~ and add / lines to any edges to the right. NrLr8rrr[rZr~)rrLcsg|]}|pdqS)r[r$rcrHmaprrr`rz$_drawendinglines..cSsg|]\}}|dur|qSrrrrrnrrrrcr) valuesr@r\count graphshortenrPr'r?rAlistreitemsremove)linesZextraedgemaprBrRZ edgecharsZ shift_sizeZminlinesZemptiesZtoshiftZ first_emptyrntargetsZ positionsrhposrwrSrror_drawendinglines5sD      *  r|c@sxeZdZdZejdeedZejdee dZ ejdddZ ejdddZ ejdee jdZejdddZdS) asciistatez State of ascii() graph renderingF)ZinitdefaultrN)__name__ __module__ __qualname____doc__rZibZFactoryrurBdictrI lastcoldiff lastindexEDGEScopyrOrtrrrrr}isr}cCs(|D]\}}|||dqdS)aXoutputs an ASCII graph of a DAG this is a helper function for 'ascii' below. takes the following arguments: - ui to write to - graph data: list of { graph nodes/edges, text } this function can be monkey-patched by extensions to alter graph display without needing to mimic all of the edge-fixup logic in ascii()  N)writerstrip)r9ZgraphZlnZlogstrrrr outputgraphus rc s|\}}}} d| kr dks&nJ|j|j} fdd| D} | dt|| t| d| dkrvt|t|dko| dkodd|D} t|dko| } | d |d}||d g|t| ||j|| |j| | d |d}t d| D]}| d q||d }| dkrNt |D]}|d d gq6nJ| dkrx|| |d d|dn t |D]}|d d gqt | ||||g}| r| t | ||||jrtdd|Dr| |n | || d || d}t|t|krFt|t|krF| |d d qt||| |t|t|krv| dqVt||| fdd|D}t|t||| |_||_d S)aprints an ASCII graph of the DAG takes the following arguments (one call per node in the graph): - ui to write to - Somewhere to keep the needed state in (init to asciistate()) - Column of the current node in the set of ongoing edges. - Type indicator of node data, usually 'C' for changesets. - Payload: (char, lines): - Character to use as node's symbol. - List of lines to display as the node's text. - Edges; a list of (col, next_col) indicating the edges between the current node and its parents. - Number of columns (ongoing edges) in the current revision. - The difference between the number of columns (ongoing edges) in the next revision and the number of columns (ongoing edges) in the current revision. That is: -1 means one column removed; 0 means no columns added or removed; 1 means one column added. rLcs&g|]}|ddfD]}|qqSrr[rlrq)ryrrrrzascii..rrr8cSs g|]\}}|d|kr|qS)rr)rxyrrrrrNr[rrZrNcss|]}|r|dvVqdS)s\/Nrrmrrrr!rzascii..rcs"g|]}ddd|fqS)s%-*s rLr)join)rrh)indentation_levelrrrs)rIrBr*r\rPrYrbrrrZxranger'rgrirtanyr|rzip)r9rRrErQtextZcoldatar^rIrTr_rBr]Zadd_padding_lineZfix_nodeline_tailrfZshift_interlinerHrsrxZextra_interliner)ryrrasciisp           r)rZ __future__rr0rZ thirdpartyrrrrr r,rrr(rr/r2rKrVrYrbrgrir|sobjectr}rrrrrr s,   /Z64