q&fIjddlZddlmZddlmZmZdZedZdZGdd Z Gd d Z y) N)_)errorutilisCpartial read of revlog %s; expected %d bytes from offset %d, got %dc&||dz zdk(xr|dk7S)Nr)ns H/usr/lib/python3/dist-packages/mercurial/revlogutils/randomaccessfile.py_is_power_of_twor s QK1  (!q&(c`eZdZdZdZdZdZdZedZ dZ ddZ dd Z d Z d Zd Zy )appenderz`the changelog index must be updated last on disk, so we use this class to delay writes to itc||_|||}||_|j|_|j |j |_|j |_yN)datafptelloffsetfstatst_sizesize_end)selfvfsnamemodebufrs r __init__zappender.__init__sH t_ggi IIbM)) II r c|jSr)rrs r endz appender.end&s yyr c|jSr)rr!s r rz appender.tell)s {{r cyrr r!s r flushzappender.flush,s r c.|jjSr)rclosedr!s r r'zappender.closed/sww~~r c8|jjyr)rcloser!s r r)zappender.close3s  r c |dk(r||_n8|dk(r|xj|z c_n|dk(r|j|z|_|j|jkr&|jj |jyy)z,virtual file offset spans real file and datarrrN)rr"rrseek)rrwhences r r+z appender.seek6sg Q; DK q[ KK6 !K q[((*v-DK ;; " GGLL % #r c d}|j|jkrN|jj|}|}|xjt |z c_|dkDr|t |z}|dk7r|j|jz }|j j ddj|j |j dd=|j d|||z}|xjt |z c_||z }|S)z5only trick here is reads that span real file and datar rrN)rrrreadlenrinsertjoin)rcountretsdoffs r r.z appender.readAs ;; " U#AC KK3q6 !KqyQ A:;;*D II  Q 3 4 !"  ! TD5L1A KK3q6 !K 1HC r c|jjt||xjt |z c_|xj t |z c_yr)rappendbytesrr/r)rr4s r writezappender.writeSs< q" s1v  SV r c:|jj|Sr)r __enter__r!s r r;zappender.__enter__Xs  r c4|jj|Sr)r__exit__)rargss r r=zappender.__exit__\stww&&r N)r))__name__ __module__ __qualname____doc__rr"rr%propertyr'r)r+r.r9r;r=r r r rrsM  &$ 'r rceZdZdZ d dZdZedZd dZe jdZ e jdZ d Z d Zd Zy)randomaccessfilezDAccessing arbitrary chuncks of data within a file, with some cachingNct|sJ||_||_||_d|_d|_d|_d|_|r|\|_|_yyNr r)r openerfilenamedefault_cached_chunk_sizewriting_handlereading_handle _cached_chunk_cached_chunk_position)rrIrJrK initial_caches r rzrandomaccessfile.__init__csa 9:::   )B&"" &'# >K ;D '); r c d|_d|_yrH)rNrOr!s r clear_cachezrandomaccessfile.clear_cachews &'#r c>|jduxs|jduS)z[True if any file handle is being held Used for assert and debug in the python codeN)rMrLr!s r is_openzrandomaccessfile.is_open{s'   t + Nt/B/B$/N r c<|j|j|S)zReturn a file object)r)rIrJ)rrs r _openzrandomaccessfile._opens{{4==t{44r c#K|jr|jy|jr|jy|j5}|dddy#1swYyxYww)z%File object suitable for reading dataN)rLrMrVrrs r _read_handlezrandomaccessfile._read_handlesX   %% %  %% %    sAA# A A#A A#c#K|jF|j:|j.|j5}||_ dd|_ dddydy#d|_wxYw#1swYyxYww)z4Context manager that keeps the file open for readingN)rMrLrJrVrXs r readingzrandomaccessfile.readingsw    '##+ ) /&(#/*.D'  / / +/D'  / /s45A2A&AA& A2 A##A&&A/+A2c||z}|j}|t|jz}||krA||kr<||k(r||k(r |jS||z }tj|j||S|j ||S)aRead a chunk of bytes from the file. Accepts an absolute offset, length to read, and an optional existing file handle to read from. If an existing file handle is passed, it will be seeked and the original seek position will NOT be restored. Returns a str or buffer of raw byte data. Raises if the requested number of bytes could not be read. )rOr/rNrbuffer_read_and_update_cache)rrlengthr" cache_start cache_endrelative_starts r read_chunkzrandomaccessfile.read_chunksvo11 #d&8&8"99 & SI%5f$ )9)))#k1N;;t11>6J J**66::r c||jdz z}||z|jz|jdz z|z }|j5}|j||j|}ddd|j |||z }t ||z }||kr,t |j|||fz} tj| ||k7s||k7rtj|||S|S#1swYxYw)Nr) rKrYr+r._add_cached_chunkr/PARTIAL_READ_MSGrJr RevlogErrorrr]) rrr_ real_offset real_lengthfile_objrrelative_offsetgotmessages r r^z'randomaccessfile._read_and_update_caches !?!?!!CDD f_t== =..23 4      .H MM+ &==-D . {D1 ;.$i/) <&$--)MMG##G, , [ K6$9;;t_f= =  . .s #C..C7c|jt|jz|k(r>t|jt|ztkr|xj|z c_y||_||_y)z|Add to or replace the cached data chunk. Accepts an absolute offset and the data that is at that location. N)rOr/rN_MAX_CACHED_CHUNK_SIZE)rrrs r rez"randomaccessfile._add_cached_chunksb  ' '#d.@.@*A AV KD&&'#d)36LL   $ & !%D *0D 'r r)r)r@rArBrCrrRrDrTrV contextlibcontextmanagerrYr[rcr^rer r r rFrF`sqN L((  5" ;22 1r rF) rqi18nrrrrorfr rrFr r r rusF  !J )C'C'LJ1J1r