13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
17
"""Python implementations of Dirstate Helper functions."""
19
from __future__ import absolute_import
25
21
# We cannot import the dirstate module, because it loads this module
26
22
# All we really need is the IN_MEMORY_MODIFIED constant
28
24
from bzrlib.dirstate import DirState
31
def pack_stat(st, _b64=binascii.b2a_base64, _pack=struct.Struct('>6L').pack):
32
"""Convert stat values into a packed representation
34
Not all of the fields from the stat included are strictly needed, and by
35
just encoding the mtime and mode a slight speed increase could be gained.
36
However, using the pyrex version instead is a bigger win.
38
# base64 encoding always adds a final newline, so strip it off
39
return _b64(_pack(st.st_size & 0xFFFFFFFF, int(st.st_mtime) & 0xFFFFFFFF,
40
int(st.st_ctime) & 0xFFFFFFFF, st.st_dev & 0xFFFFFFFF,
41
st.st_ino & 0xFFFFFFFF, st.st_mode))[:-1]
44
def _unpack_stat(packed_stat):
45
"""Turn a packed_stat back into the stat fields.
47
This is meant as a debugging tool, should not be used in real code.
49
(st_size, st_mtime, st_ctime, st_dev, st_ino,
50
st_mode) = struct.unpack('>6L', binascii.a2b_base64(packed_stat))
51
return dict(st_size=st_size, st_mtime=st_mtime, st_ctime=st_ctime,
52
st_dev=st_dev, st_ino=st_ino, st_mode=st_mode)
55
def _bisect_path_left(paths, path):
27
def _bisect_path_left_py(paths, path):
56
28
"""Return the index where to insert path into paths.
58
30
This uses the dirblock sorting. So all children in a directory come before
91
63
mid = (lo + hi) // 2
92
64
# Grab the dirname for the current dirblock
94
if _cmp_path_by_dirblock(cur, path) < 0:
66
if _cmp_path_by_dirblock_py(cur, path) < 0:
101
def _bisect_path_right(paths, path):
73
def _bisect_path_right_py(paths, path):
102
74
"""Return the index where to insert path into paths.
104
76
This uses a path-wise comparison so we get::
123
95
# Grab the dirname for the current dirblock
125
if _cmp_path_by_dirblock(path, cur) < 0:
97
if _cmp_path_by_dirblock_py(path, cur) < 0:
132
def bisect_dirblock(dirblocks, dirname, lo=0, hi=None, cache={}):
104
def bisect_dirblock_py(dirblocks, dirname, lo=0, hi=None, cache={}):
133
105
"""Return the index where to insert dirname into the dirblocks.
135
107
The return value idx is such that all directories blocks in dirblock[:idx]
163
def cmp_by_dirs(path1, path2):
135
def cmp_by_dirs_py(path1, path2):
164
136
"""Compare two paths directory by directory.
166
138
This is equivalent to doing::
186
158
return cmp(path1.split('/'), path2.split('/'))
189
def _cmp_path_by_dirblock(path1, path2):
161
def _cmp_path_by_dirblock_py(path1, path2):
190
162
"""Compare two paths based on what directory they are in.
192
164
This generates a sort order, such that all children of a directory are