|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/hgext/fastannotate/__pycache__/ |
Upload File : |
o
�]Lb�" � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl
m Z m
Z
ddlm Z dZdZd ZG d
d � d e�Zd d
� ZdS )� )�absolute_importN)�hex)�open)�error�pycompat� )r � � c @ s� e Zd ZdZdZd!dd�Zdd� Zedd � �Zd"d d �Z d
d� Z
dd� Z dd� Z dd� Z
d#dd�Zdd� Zdd� Zdd � Zed d � �Zd d � ZdS )$�revmapz~trivial hg bin hash - linelog rev bidirectional map
also stores a flag (uint8) for each revision, and track renames.
s REVMAP1 Nc C s^ || _ dg| _dg| _i | _dg| _dg| _d| _|r-tj �|�r'| � � dS | �
� dS dS )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� _lastmaxrev�os�exists�_load�flush)�selfr
� r �;/usr/lib/python3/dist-packages/hgext/fastannotate/revmap.py�__init__< s �zrevmap.__init__c C sV |j dd� | _ |jdd� | _|j�� | _|jdd� | _|jdd� | _d| _dS )z>copy the map data from another revmap. do not affect self.pathNr )r r r �copyr r r )r �rhsr r r �copyfromS s
zrevmap.copyfromc C s t | j�d S )z"return max linelog revision numberr )�lenr )r r r r �maxrev\ s z
revmap.maxrevFc C s� || j v rt�dt|� ��t|�tkrt�dt ��t| j�}d}|r(|t O }|durC|| j
d krC|t O }| j �
|� | j
�
|� | j�
|� | j�
|� || j |< |rZ| �� |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 alreadys hsh must be %d-char longr Nr )r r �CorruptedFileErrorr r �_hshlen�hgerror�ProgrammingErrorr �sidebranchflagr �
renameflagr �appendr r )r �hsh�
sidebranchr
r �idx�flagr r r r' a s,
� �
z
revmap.appendc C � || j ks |dk r dS | j| S )z>convert linelog revision to hg hash. return None if not found.r N)r r �r �revr r r �rev2hsh| s
zrevmap.rev2hshc C r, )zkget the flag (uint8) for a given linelog revision.
return None if revision does not exist.
r N)r r r- r r r �rev2flag� s
zrevmap.rev2flagc C s2 || j ks |dk r dS t�| j|�d }| j| S )zcget the path for a given linelog revision.
return None if revision does not exist.
r Nr )r �bisect� bisect_rightr r )r r. r* r r r �rev2path� s
zrevmap.rev2pathc C s | j �|�S )z>convert hg hash to linelog revision. return None if not found.)r �get)r r( r r r �hsh2rev� s zrevmap.hsh2revc C s8 dg| _ dg| _i | _dg| _d| _|r| �� dS dS )z3make the map empty. if flush is True, write to diskNr r )r r r � _rev2pathr r )r r r r r �clear� s �z revmap.clearc C s� | j sdS | jdkr9t| j d�� }|�| j� t�dt| j��D ]}| � ||� q W d � n1 s3w Y n+t| j d�� }t�| jd t| j��D ]}| � ||� qLW d � n1 s_w Y | j
| _dS )z write the state down to the fileNr s wbr s ab) r
r r �write�HEADERr �xranger r � _writerevr )r �f�ir r r r � s$
� ���� � z revmap.flushc C s | j sdS d}t}t| j d��e}|�t| j��| jkr t�� �| jdd� |�|�}|s.n?t |�}t| j
�}|t @ rL| � |�}| j
�|� | j�|� |�|�}t|�|kr[t�� �|| j|< | j�|� | j
�|� q&W d � n1 sww Y | j| _dS )zload state from fileNr � rbF)r )r
r" r �readr r9 r r! r7 �ordr r& � _readcstrr r' r r r r r ) r �flaglen�hshlenr< �bufr+ r. r
r( r r r r � s8
� � z revmap._loadc C sj | j | }| j| }|�t�d|�� |t@ r.| �|�}|du r't�d| ��|�|d � |�|� dS )z append a revision data to file� BNs cannot find path for %s� ) r r r8 �struct�packr& r3 r r! )r r. r< r+ r( r
r r r r; � s
zrevmap._writerevc C s4 d} | � d�}|st�� �|dkr |S ||7 }q)z,read a C-language-like ' '-terminated stringr Tr rF )r? r r! )r< rD �chr r r rA � s
��zrevmap._readcstrc C sh t |t�r
|\}}n |�� |�� }}| �|�}|du r dS |dur+|| �|�kr+dS | �|�t@ dkS )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 )�
isinstance�tuple�noder
r5 r3 r0 r% )r r< r( r
r. r r r � __contains__� s
zrevmap.__contains__)N)FNF)F)�__name__�
__module__� __qualname__�__doc__r9 r r �propertyr r' r/ r0 r3 r5 r7 r r r; � staticmethodrA rM r r r r r
4 s&
r
c C s� d}z9t | d��)}|�t tj� |�� ttj�kr(|� t�}W d � W |S W d � W |S 1 s4w Y W |S t
yE Y |S w )z�return the last hash in a revmap, without loading its full content.
this is equivalent to `m = revmap(path); m.rev2hsh(m.maxrev)`, but faster.
Nr> ) r �seekr" �io�SEEK_END�tellr r
r9 r? �IOError)r
r( r< r r r � getlastnode� s" �� �� ��rY )�
__future__r r1 rU r rG �mercurial.noder �mercurial.pycompatr � mercurialr r# r � r% r& r" �objectr
rY r r r r �<module> s H