^c"ddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z dZ dZ dZGdd Zd ZdS) N)hex)open)errorceZdZdZdZddZdZedZddZ d Z d Z d Z d Z dd ZdZdZdZedZdZdS)revmapz~trivial hg bin hash - linelog rev bidirectional map also stores a flag (uint8) for each revision, and track renames. sREVMAP1Nc ||_dg|_dg|_i|_dg|_dg|_d|_|rKtj|r| dS| dSdS)a=create or load the revmap, optionally associate to a file if path is None, the revmap is entirely in-memory. the caller is responsible for locking. concurrent writes to a same file is unsafe. the caller needs to make sure one file is associated to at most one revmap object at a time.Nr) path_rev2hsh _rev2flag_hsh2rev _renamerevs _renamepaths _lastmaxrevosexists_loadflush)selfrs ;/usr/lib/python3/dist-packages/hgext/fastannotate/revmap.py__init__zrevmap.__init__:s   3 E  w~~d##     r c|jdd|_|jdd|_|j|_|jdd|_|jdd|_d|_dS)z>copy the map data from another revmap. do not affect self.pathNr )rrrcopyrrr)rrhss rcopyfromzrevmap.copyfromQso QQQ qqq) ))++ ?111-,QQQ/r c0t|jdz S)z"return max linelog revision numberr)lenr)rs rmaxrevz revmap.maxrevZs4=!!A%%r FcV||jvr$tjdt|zt |t krt jdt zt |j}d}|r |tz}|O||j dkr>|tz}|j ||j ||j ||j |||j|<|r||S)z~add a binary hg hash and return the mapped linelog revision. if flush is True, incrementally update the file. s%r is in revmap alreadyshsh must be %d-char longrNr )rrCorruptedFileErrorrr!_hshlenhgerrorProgrammingErrorrsidebranchflagr renameflagrappendrr)rhsh sidebranchrridxflags rr*z revmap.append_s. $-  **SXX5  s88w  *+g5 $-    # N "D  (9"(= = = J D   # #C ( ( (   $ $T * * * S!!! d###  c   JJLLL r cB||jks|dkrdS|j|S)z>convert linelog revision to hg hash. return None if not found.rN)r"rrrevs rrev2hshzrevmap.rev2hshzs)   a4}S!!r cB||jks|dkrdS|j|S)zkget the flag (uint8) for a given linelog revision. return None if revision does not exist. rN)r"rr0s rrev2flagzrevmap.rev2flags+   a4~c""r c|||jks|dkrdStj|j|dz }|j|S)zcget the path for a given linelog revision. return None if revision does not exist. rNr)r"bisect bisect_rightrr)rr1r-s rrev2pathzrevmap.rev2pathsF   a4!$"2C881< %%r c6|j|S)z>convert hg hash to linelog revision. return None if not found.)rget)rr+s rhsh2revzrevmap.hsh2revs}  %%%r cdg|_dg|_i|_dg|_d|_|r|dSdS)z3make the map empty. if flush is True, write to diskNr r )rrr _rev2pathrr)rrs rclearz revmap.clearsQ     JJLLLLL  r c0|jsdS|jdkrt|jd5}||jt dt |jD]}||| dddn #1swxYwYnqt|jd5}t |jdzt |jD]}||| dddn #1swxYwY|j |_dS)z write the state down to the fileNr swbrsab) rrrwriteHEADERranger!r _writerevr")rfis rrz revmap.flushsy  F  r ! !di'' )1 $$$q#dm"4"455))ANN1a(((() ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) di'' )1t/!3S5G5GHH))ANN1a(((() ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ;s%AB  BB.AC??DDcd|jsdSd}t}t|jd5}|t |j|jkrt j|d ||}|snt|}t |j }|tzrI| |}|j ||j|||}t ||krt j||j|<|j||j | dddn #1swxYwY|j|_dS)zload state from fileNrrbF)r)rr%rreadr!rArr$r>ordrr) _readcstrrr*rrrr"r) rflaglenhshlenrDbufr.r1rr+s rrz revmap._loadsy  F $)U # # *qvvc$+&&''4;66.000 JJUJ # # # *ffWoo3xx$-((*$3>>!,,D$++C000%,,T222ffVnns88v%%2444%( c"%%d+++ $$S))) * * * * * * * * * * * * * * * *( ;sE$FF FcT|j|}|j|}|tjd||t zrF||}|tjd|z||dz||dS)zappend a revision data to fileBNscannot find path for %s) rrr@structpackr)r8rr$)rr1rDr.r+rs rrCzrevmap._writerevs~c"mC   D$''((( *  "==%%D|./IC/OPPP GGD5L ! ! !  r czd} |d}|stj|dkrn||z }7|S)z,read a C-language-like ''-terminated stringr TrrP)rHrr$)rDrMchs rrJzrevmap._readcstrsU B 1.000U{{ 2IC   r c6t|tr|\}}n(||}}||}|dS||||krdS||tzdkS)z(fctx or (node, path)) -> bool. test if (node, path) is in the map, and is not in a side branch. f can be either a tuple of (node, path), or a fctx. NFr) isinstancetuplenoderr;r8r4r()rrDr+rr1s r __contains__zrevmap.__contains__s a   +IC!&&((Cll3 ;5   c(:(: : :5 c""^399r )N)FNF)F)__name__ __module__ __qualname____doc__rArrpropertyr"r*r2r4r8r;r>rrrC staticmethodrJrYr rr r 2s F.&&X&6""" ###&&&&&&     ' ' ''''<     \ :::::r r c\d} t|d5}|t tj|t tjkr| t}dddn #1swxYwYn#t$rYnwxYw|S)zreturn the last hash in a revmap, without loading its full content. this is equivalent to `m = revmap(path); m.rev2hsh(m.maxrev)`, but faster. NrG) rseekr%ioSEEK_ENDtellr!r rArHIOError)rr+rDs r getlastnodergs C $   &! FFG8R[ ) ) )vvxx#fm,,,,ffWoo & & & & & & & & & & & & & & &      Js5BA0B BBBBB B)(B))r6rcrrQmercurial.nodermercurial.pycompatr mercurialrr&r(r)r%r rgr`r rrls ######,  B:B:B:B:B:B:B:B:J     r