^c9dddlZddlmZddlmZmZdZedZdZGdd Z dS) N)_)errorutilisCpartial read of revlog %s; expected %d bytes from offset %d, got %dc&||dz zdko|dkS)Nr)ns H/usr/lib/python3/dist-packages/mercurial/revlogutils/randomaccessfile.py_is_power_of_twor s QK1  (!q&(ceZdZdZ d dZdZd dZejd dZ ejdZ d d Z d d Z d Z dS)randomaccessfilezDAccessing arbitrary chuncks of data within a file, with some cachingNct|sJ||_||_||_d|_d|_d|_d|_|r|\|_|_dSdSNr r)r openerfilenamedefault_cached_chunk_sizewriting_handlereading_handle _cached_chunk_cached_chunk_position)selfrrr initial_caches r __init__zrandomaccessfile.__init__sz 9:::::   )B&"" &'#  L>K ;D '););); L Lr c"d|_d|_dSr)rr)rs r clear_cachezrandomaccessfile.clear_cache1s &'###r rc:||j|S)zReturn a file object)mode)rr)rr s r _openzrandomaccessfile._open5s{{4=t{444r c#K||VdS|jr |jVdS|jr |jVdS|5}|VddddS#1swxYwYdS)z%File object suitable for reading dataN)rrr!)rexisting_file_objfps r _open_readzrandomaccessfile._open_read9s  (# # # # # #  % % % % % %   % % % % % %                   sAAAc#K|j^|jW|jP|5}||_ dVd|_n #d|_wxYw ddddS#1swxYwYdSdVdS)z4Context manager that keeps the file open for readingN)rrrr!)rr$s r readingzrandomaccessfile.readingOs   '#+ ) /&(#/EEE*.D''$D'....'  / / / / / / / / / / / / / / / / / / EEEEEs'AAA A  AA"Ac||z}|j}|t|jz}||kr9||kr3||kr ||kr|jS||z }tj|j||S||||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. )rlenrrbuffer_read_and_update_cache)roffsetlengthr#end cache_start cache_endrelative_starts r read_chunkzrandomaccessfile.read_chunk`svo1 #d&8"9"99 & SI%5%5f$$ )9)9))#k1N;t1>6JJ J**66;LMMMr c||jdz z}||z|jz|jdz z|z }||5}||||}dddn #1swxYwY|||||z }t ||z } | |kr't |j||| fz} tj | ||ks||krtj |||S|S)Nr) rr%seekread_add_cached_chunkr)PARTIAL_READ_MSGrr RevlogErrorrr*) rr,r-r# real_offset real_lengthfile_objdatarelative_offsetgotmessages r r+z'randomaccessfile._read_and_update_cacheys^!?!!CDD f_t= =.23 4  __. / / .8 MM+ & & &==--D . . . . . . . . . . . . . . . {D111 ;.$ii/) <<&$-)MMG#G,, , [ K6$9$9;t_f== = s+A77A;>A;c|jt|jz|kr?t|jt|ztkr|xj|z c_dS||_||_dS)z|Add to or replace the cached data chunk. Accepts an absolute offset and the data that is at that location. N)rr)r_MAX_CACHED_CHUNK_SIZE)rr,r<s r r6z"randomaccessfile._add_cached_chunksv  '#d.@*A*A AV K KD&''#d))36LLL   $ &    !%D *0D ' ' 'r )N)r)__name__ __module__ __qualname____doc__rrr! contextlibcontextmanagerr%r'r2r+r6r r r rrsNN LLLL((((5555* NNNN22 1 1 1 1 1r r) rFi18nrrrrAr7r rr r r rJs  !1J )))E1E1E1E1E1E1E1E1E1E1r