^czddlZddlZddlZddlmZddlmZmZmZddl m Z m Z m Z ddl mZmZmZmZmZmZmZddlmZmZddlmZmZejd Zdd Z d Z!d Z"ejej#Z$ddZ%dZ&dZ'e'ej(Z)GddZ*e*Z(GddZ+dZ,GddZ-ej.e(dfdZ/ddZ0dS)N)_)getattropensetattr)binhexshort)errorlineloglockmdiffpycompatscmutilutil)hashutil stringutil)r revmapc,||SN)file)repopaths z_parents..<s3 4 4 4Aqvvxx1668833a333r_filelog) _changeidparents__dict__r_reporr%)r#followplr"s` r_parentsr,2s{ KK B 5 4 4 4 4 4 4 455QZ''!!'1668844AJ Irc}|d}|r|ds|dz }fdt|D|fS)N rcg|]}|fSrr)r!ifctxs rr$z_decorate..Ns 1 1 11dAY 1 1 1r)datacountendswithrange)r1text linecounts` r _decorater8Isl 99;;D 5!!I DMM%((Q 1 1 1 1i 0 0 1 1 14 88rcb|D]+\\}}}}}|dkr|d|||d||<,|S)N=rr)parentchildblocksa1a2b1b2ts r_pairrCTsN%//RR! 99$Qi2.E!HRUO LrFc|r(t|ts|t||}n||}|G| ||}n|||}nc||}|dkr|}n|}||kr||_|||_|S)a2(repo, str, str) -> fctx get the filectx object from repo, rev, path, in an efficient way. if resolverev is True, "rev" is a revision specified by the revset language, otherwise "rev" is a nodeid, or a revision number that can be consumed by repo.__getitem__. if adjustctx is not None, the returned fctx will point to a changeset that introduces the change (last modified the file). if adjustctx is 'linkrev', trust the linkrev and do not adjust it. this is noticeably faster for big repos but is incorrect for some cases. N)changeidlinkrev) isinstanceint _revsinglerevfilectxlinkrevintrorevr& _changectx)rrJr resolverev adjustctxctxr1rMs r resolvefctxrRbs*S#..3?s##3i 7799 t9DD<<swwyy<99DD4y  " "||~~HH}}H swwyy %DN"8nDO Krc|ddddddddS) Ns.hg/s.hg.hg/s.l/s.l.hg/s.m/s.m.hg/s.lock/s .lock.hg/)replacer s r encodedirrUsE Wj))  # #  # # L ) ) rctjtfdtjjD}t tj| ddS)Nc3<K|]}|t|fVdSr)r)r!kdiffoptss r zhashdiffopts..s2JJQ78Q''(JJJJJJr) rpprintsortedr rYdefaultsrrsha1digest)rY diffoptstrs` r hashdiffoptsrbsi"JJJJ%.2IJJJJJJ x}Z((//11 2 22A2 66rcDeZdZdZddddZdZejdZdS) annotateoptszlike mercurial.mdiff.diffopts, but is for annotate followrename: follow renames, like "hg annotate -f" followmerge: follow p2 of a merge changeset, otherwise p2 is ignored NT)sdiffoptss followrenames followmergec tj|}|jD]*\}}t |||||+dSr)r byteskwargsr^itemsrget)selfoptsrXvs r__init__zannotateopts.__init__s^#D))M'')) - -DAq D!TXXa^^ , , , , - -rcd}|js|dz }|js|dz }|jHt|jtjsJt |j}|t kr|d|zz }|pdS)z?represent opts in a short string, suitable for a directory namersr0sm0Nisdefault) followrename followmergerYrGr rb_defaultdiffopthash)riresult diffopthashs rshortstrzannotateopts.shortstrs   eOF  eOF = $dmU^<< < < <&t}55K111$,,##r) __name__ __module__ __qualname____doc__r^rlr propertycachertrrrrdrdscH ---   $ $ $ $ $rrdceZdZdZdZedZedZdZeZ dZ edZ dd Z dd Z d ZddZdZdZdZddZdZeddZddZdZejdZejdZddZd S)_annotatecontextz{do not use this class directly as it does not use lock to protect writes. use "with annotatecontext(...)" instead. c||_|j|_||_||_||_||_d|_d|_i|_dSr) ruirrj linelogpath revmappath_linelog_revmap _node2path)rirrr~rrjs rrlz_annotatecontext.__init__sH '  &$  rc|jtj|jrt |jd5} t j| |_n-#t j $rt j|_YnwxYwdddn #1swxYwYnt j|_|jS)Nsrb) rosrexistsr~r linelogmodr fromdataread LineLogErrorrir#s rr z_annotatecontext.linelogs = w~~d.// 5$*E22=a=(2(:(C(CAFFHH(M(M %2===(2(:(<(< ================ !+ 2 4 4 }s6B36A:9B3:'B$!B3#B$$B33B7:B7cZ|jtj|j|_|jSr)r revmapmodrrris rrz_annotatecontext.revmaps& < $+DO<.s^   A*4B)<)<Kdk!!"%%%"''))a P   r)2rGrHrrrrcanannotatedirectlyr} debugflagdebugrrr\r safehasattrannotatedirectlyrrrrJr Abortrappendpoprr annotateannotateresultr2 _parentfuncrhsetadd_checklastmasterheadlenr makeprogressr8 enumeratelist _diffblocksrC incrementrjrp _appendrevrcomplete_refineannotateresult)rirJrshowpath showlinesdirectlyrevfctx masterfctx initvisitvisitpcacheneededhistr#llrevrrr+r" newmainbranchprogressreadyr= ismainbranchcurrr0bs bannotateds` rrz_annotatecontext.annotate's|$ c3   5di)..s3344C!44S99'  Gw   - ")$*:7G*L*LMM(((IFF F  & &!..tt/ ,,!%J     8I  )~~'k8999::   Z ( ( (!!! 1 $ AF{{a4iiDK ++AFFHH55 %%e,,,4!16688,Q!!!$$BF1I $ $"JJq!,,q0q F??LLOOO $&  AT[((   a B qE AT[((  !  % %a ( ( ( 7    !$(Is=/A/A#BC  248I> dl1222!!! 7'' S%7%7(  . 5b ADyy EB $ $D==!ELLOOO  IIKKKF -LD====Q<>Qq 1IIINIIIIDGq  5""$$$! r77a<!$"2"23Q"@"@AAF6:6666 5,-FFHH)]. 5`     gq)   ))&'8YOOOsC22 C?>C?cxd}d}t|tsO|Mttdt |d|}|||jf|jvr|}|M|jrdnd}| ||d}||jv}|s|jr| |dd}||fS)z(str) -> bool, fctx or node. return (True, f) if we can annotate without updating the linelog, pass f to annotatedirectly. return (False, f) if we need extra calculation. f is the fctx resolved from rev. TN)(cdSrr)xs rz6_annotatecontext.canannotatedirectly..strrF)rPrO) rGrHbytesrrhrrr _perfhackr)rirJrrr#hshrPs rrz$_annotatecontext.canannotatedirectlys #s## #&&**3s88^^DDSIICC#3t{#B#B 9&*n> $I!!#t!LLA$+%F Ldn L%%cTd%KKqyrc||dd}||jvrD|g}d} |d}||}|snF|d|jvr|d}n||dft j} | |j| | j tj|jt|D]S}|} t| || } || || | }Tn|j} |j| } |} | fd| D}|r||}|r||| }||f}|S)a (rev : str) -> [(node : str, linenum : int, path : str)] the result has the same format with annotate, but include all (including deleted) lines up to rev. call this after calling annotate(rev, ...) for better performance and accuracy. TrrrrcPg|]"\}}|k ||f#Sr)r)r!rrrrs rr$z5_annotatecontext.annotatealllines...s3IIITQa5jj6>>!$$a(jjjr)rrrr2rrr copyfromrrrreversedrr _doappendrev getalllinesrr_addpathtoresult _resolvelines)rirJrrrchainar#r+r br=linesrrr linecontentsrrs @@rannotatealllinesz!_annotatecontext.annotateallliness##CDD#II $+ % %IEA $"I%%a((a5DK''1 A RU### $!(**G   T\ * * *   W^ , , ,%''F OODK ( ( (e__  FFHHd..q!4455!!'61f===  lG[F##%%llnns##IIIIIUIII  ;**66::F  ,--ffgFFLl+F rcfd|D}dgt|z}tjt}t t|D]1}|||||df|2|rcd} |D]?\\} } } | tj zr+|| dd}n#t$r|cYSwxYw|i} |D]\\} } } | | vr|| dd}|j j r-|j dt|| | fz||| }t#j|}|| | <| D]}| | | ||<t)d|DsJ|S|j dt|z|| ||| }|j}t#j|}t|t|krt/jdt3|D]$\}}||}||vr||D]}|||<||=%|c|S) ah(annotateresult) -> [line]. designed for annotatealllines. this is probably the most inefficient code in the whole fastannotate directory. but we have made a decision that the linelog does not store line contents. so getting them requires random accesses to the revlog data, since they can be many, it can be very slow. cFg|]}|dS)r)r)r!rrs rr$z2_annotatecontext._resolvelines..@s)===qt$$===rNrrs6fastannotate: reading %s line #%d to resolve lines %r c3K|]}|duV dSrr)r!rs rrZz1_annotatecontext._resolvelines..cs&99Q1D=999999rs+fastannotate: annotate %s to resolve lines sunexpected annotated lines)r collections defaultdictrr5rrgrev2flagrsidebranchflag StopIterationr}rrr rrev2pathr splitnewlinesr2allrrfaerrorCorruptedFileErrorr)rirrr revsrrkey2idxsr0rrJ_linenumidxsrevlineslinenumr1ridx annotatedlinerXs ` rrz_annotatecontext._resolvelines8sK>===n===#n---*400s6{{## @ @A d1g~a034 5 <  $00fooc6J6JKK % 3DIIKK @ @(- #==&.smG&<s =99&99999999  GMM?%**L      S ! ! !$$S&//#*>*>??D.I' 44E5zzS^^++01NOOO$U++ $ $4aL=='{++&*s   a0 $b sAC%% C43C4ct|tr|}n|}j|}|s$t jdt|zj|tj zdkr$t jdt|zj |fdj j D}||||S)alike annotate, but when we know that f is in linelog. f can be either a 20-char str (node) or a fctx. this is for perf - in the best case, the user provides a node and we don't need to read the filelog or construct any filecontext. s%s is not in revmaprs%s is not in revmap mainbranchcNg|]!\}}j||f"Sr)rr)r!rrris rr$z5_annotatecontext.annotatedirectly..s?   ,0AqT[  # #Q '   r)rGrrrrrrrrrrr rrr)rir#rrrrrrs` rrz!_annotatecontext.annotatedirectlyys  a   CC&&((C ##C(( P,-Cc#hh-NOO O K  ' ')*B Bq H H,1CHH<  e$$$    48L4O   ))&!XyIIIrc|r||}|rt|trI|j|}|||j|}n|}tj| }t|t|krtj ||f}|S)znadd the missing path or line contents, they can be expensive. f could be either node or fctx. ) rrGrrrrrr rr2rrr)rirrr#rrrr1rs rrz&_annotatecontext._refineannotateresults  3**622F  %!U##  ++A..((DK,@,@,G,GHH' 44E5zzS[[((0222e_F rcL||j|j|||dSr)rr r)rir1r=rs rrz_annotatecontext._appendrevs( $, T6:NNNNNrcBtj|||jjSr)r allblocksrjrY)rirrs rrz_annotatecontext._diffblockssq!TY%7888rc@fd}g}i|*|D]'\\}}} } } | dkr|d|| | Dz }(t|}||vr|||D]} || | <||} | |<t |D][\\}}} } } | dkr||| ||| | -fd|| | D}|| |||\dS)z'append a revision to linelog and revmapct|trtjd|}|}|*|d|}|S)z (fctx) -> intsf should not be an intNT) sidebranchr)rGrHr ProgrammingErrorrrrr)r#rrJrs rgetllrevz/_annotatecontext._doappendrev..getllrevsq!S!! H,-FGGG&&((C..%%C{mmCDqvvxxmHHJrNr:cBg|]\}}t|t|SrrGrH)r!r#rs rr$z1_annotatecontext._doappendrev..s<!!! Aq)!S11!!!!rr cXg|]&\}}t|tr|n||f'Srr)r!rr siderevmaps rr$z1_annotatecontext._doappendrev..sI1&a--@aa:a=1Er)rremoverrrr replacelinesreplacelines_vec)r rr1r=rrsiderevsr>r?r@rAopr#rblinesrs ` @rrz_annotatecontext._doappendrevs       !(.  $ RR"::!!$.r"u$5!!!H x== 8   OOD ! ! ! ( (A$HQKKJqMM diikk << 4$,V$4$4 @ @ RRbTzz!$$UBB;;;; *2b5 1((B???? @ @rcBjfdfd|DS)z6(revmap, [(node, linenum)]) -> [(node, linenum, path)]Ncj|}|2|}|j|<|Sr)rrhrr)nodeidrrris r_getpathz3_annotatecontext._addpathtoresult.._getpathsK?&&v..D|v~~f'='=>>*.'Krc2g|]\}}|||fSrr)r!nrrs rr$z5_annotatecontext._addpathtoresult..s,???1Axx{{#???r)r)rirrrs` `@rrz!_annotatecontext._addpathtoresultsL >[F      @???????rc|d}nA|j|}|stj|jj|krtjdS)z:check if fctx is the master's head last time, raise if notNr)rrrrCannotReuseErrorr r)rir1rs rrz%_annotatecontext._checklastmasterheadsi <EEK'' 44E 1.000 < % ' '*,, , ( 'rcF|jj|jjfd}|S)z-> (fctx) -> [fctx]cBt|}s |dd}|S)N)r*r)r,)r#r+rpros rr'z-_annotatecontext._parentfunc..parentss/!L111B VIr)rjrorp)rir'rpros @@rrz_annotatecontext._parentfuncs?y- i+        rc8|jddS)N fastannotatesperfhack)r} configboolrs rrz_annotatecontext._perfhacksw!!/;???rc 8t|j||p|jfi|Sr)rRrr)rirJrkwdss rrz_annotatecontext._resolvefctxs%49cD,=DIGG$GGGrT)NFF)FFr)rurvrwrxrlpropertyr rr__del__rrrrrrrrrrr staticmethodrrrrryrrrrrrr{r{s     X X !!!G::: ==X=!!!!F@P@P@P@PD.6666p???BJJJ.$OOOO999/@/@/@\/@b @ @ @ @ - - -     @@@HHHHHHrr{cZ|D]'} tj|#t$rY$wxYwdS)zsilent, best-effort unlinkN)runlinkOSError)pathsrs rrr sS  K        D s  ((cdeZdZdZefdZedZedZdZ edZ dS) pathhelperz9helper for getting paths for lockfile, linelog and revmapctjd|jt ||_||_dS)Nr&)rrjoinrtrU_vfspathr))rirrrjs rrlzpathhelper.__init__s5 T]IdOO    rctj|jj|jSr)rrdirnamer)vfsr5r6rs rr8zpathhelper.dirnames*wtz~224=AABBBrcP|jj|jdzS)Ns.lr)r9r5r6rs rr~zpathhelper.linelogpath !z~""4=5#8999rcPtj|jj|jdzS)Ns.lock)lockmodr r)r9r6rs rr zpathhelper.lock$s|DJNDMH,DEEErcP|jj|jdzS)Ns.mr;rs rrzpathhelper.revmappath'r<rN) rurvrwrx defaultoptsrlr+r8r~r rrrrr3r3sCC(3CCXC::X:FFF::X:::rr3c# Kt|||}tj|j|j}|j}d} |5t|||||}|r||Vdddn #1swxYwYnB#t$r5|||j d|zwxYw|| dSdS#|| wwxYw)a context needed to perform (fast) annotate on a file an annotatecontext of a single file consists of two structures: the linelog and the revmap. this function takes care of locking. only 1 process is allowed to write that file's linelog and revmap at a time. when something goes wrong, this function will assume the linelog and the revmap are in a bad state, and remove them from disk. use this function in the following way: with annotatecontext(...) as actx: actx. .... Ns+fastannotate: %s: cache broken and deleted ) r3rmakedirsr8rr~r r{r Exceptionr}rr)rrrjrhelperrr~actxs rannotatecontextrF,sg dD ) )FM&.!!!"J$K D  [[]]  #D$ ZNND  JJJ                    LLNNN  ELMMM    JJLLLLL  4  JJLLLL sAB.B  B BBBBC4?CC44D c|j}|j}|jdddrd}t ||}t ||||S)zdlike annotatecontext but get the context from a fctx. convenient when used in fctx.annotate r&s forcefollowT)rYro)r)_pathr}r'rdrF)r1r*rYrrraoptss rfctxannotatecontextrJQsY :D :D w/>4@@ ( @ @ @E 4ug 6 66rr*)FN)TNF)1r contextlibrmercurial.i18nrmercurial.pycompatrrrmercurial.noderrr mercurialr r rr r>r rrrmercurial.utilsrrrrr lrucachefuncrr,r8rC revsinglerIrRrUrbr@rqrdr{rr3contextmanagerrFrJrrrrUs      .999T w0 1 1 ))))Z777#l5#455$$$$$$$$Dlnn HHHHHHHHD::::::::4 %0%!!!!H 7 7 7 7 7 7r