##// END OF EJS Templates
revlog.children: use parentrevs instead of parents
Benoit Boissinot -
r3470:c6773b7e default
parent child Browse files
Show More
@@ -1,1268 +1,1264 b''
1 """
1 """
2 revlog.py - storage back-end for mercurial
2 revlog.py - storage back-end for mercurial
3
3
4 This provides efficient delta storage with O(1) retrieve and append
4 This provides efficient delta storage with O(1) retrieve and append
5 and O(changes) merge between branches
5 and O(changes) merge between branches
6
6
7 Copyright 2005, 2006 Matt Mackall <mpm@selenic.com>
7 Copyright 2005, 2006 Matt Mackall <mpm@selenic.com>
8
8
9 This software may be used and distributed according to the terms
9 This software may be used and distributed according to the terms
10 of the GNU General Public License, incorporated herein by reference.
10 of the GNU General Public License, incorporated herein by reference.
11 """
11 """
12
12
13 from node import *
13 from node import *
14 from i18n import gettext as _
14 from i18n import gettext as _
15 from demandload import demandload
15 from demandload import demandload
16 demandload(globals(), "binascii changegroup errno ancestor mdiff os")
16 demandload(globals(), "binascii changegroup errno ancestor mdiff os")
17 demandload(globals(), "sha struct util zlib")
17 demandload(globals(), "sha struct util zlib")
18
18
19 # revlog version strings
19 # revlog version strings
20 REVLOGV0 = 0
20 REVLOGV0 = 0
21 REVLOGNG = 1
21 REVLOGNG = 1
22
22
23 # revlog flags
23 # revlog flags
24 REVLOGNGINLINEDATA = (1 << 16)
24 REVLOGNGINLINEDATA = (1 << 16)
25 REVLOG_DEFAULT_FLAGS = REVLOGNGINLINEDATA
25 REVLOG_DEFAULT_FLAGS = REVLOGNGINLINEDATA
26
26
27 REVLOG_DEFAULT_FORMAT = REVLOGNG
27 REVLOG_DEFAULT_FORMAT = REVLOGNG
28 REVLOG_DEFAULT_VERSION = REVLOG_DEFAULT_FORMAT | REVLOG_DEFAULT_FLAGS
28 REVLOG_DEFAULT_VERSION = REVLOG_DEFAULT_FORMAT | REVLOG_DEFAULT_FLAGS
29
29
30 def flagstr(flag):
30 def flagstr(flag):
31 if flag == "inline":
31 if flag == "inline":
32 return REVLOGNGINLINEDATA
32 return REVLOGNGINLINEDATA
33 raise RevlogError(_("unknown revlog flag %s" % flag))
33 raise RevlogError(_("unknown revlog flag %s" % flag))
34
34
35 def hash(text, p1, p2):
35 def hash(text, p1, p2):
36 """generate a hash from the given text and its parent hashes
36 """generate a hash from the given text and its parent hashes
37
37
38 This hash combines both the current file contents and its history
38 This hash combines both the current file contents and its history
39 in a manner that makes it easy to distinguish nodes with the same
39 in a manner that makes it easy to distinguish nodes with the same
40 content in the revision graph.
40 content in the revision graph.
41 """
41 """
42 l = [p1, p2]
42 l = [p1, p2]
43 l.sort()
43 l.sort()
44 s = sha.new(l[0])
44 s = sha.new(l[0])
45 s.update(l[1])
45 s.update(l[1])
46 s.update(text)
46 s.update(text)
47 return s.digest()
47 return s.digest()
48
48
49 def compress(text):
49 def compress(text):
50 """ generate a possibly-compressed representation of text """
50 """ generate a possibly-compressed representation of text """
51 if not text: return ("", text)
51 if not text: return ("", text)
52 if len(text) < 44:
52 if len(text) < 44:
53 if text[0] == '\0': return ("", text)
53 if text[0] == '\0': return ("", text)
54 return ('u', text)
54 return ('u', text)
55 bin = zlib.compress(text)
55 bin = zlib.compress(text)
56 if len(bin) > len(text):
56 if len(bin) > len(text):
57 if text[0] == '\0': return ("", text)
57 if text[0] == '\0': return ("", text)
58 return ('u', text)
58 return ('u', text)
59 return ("", bin)
59 return ("", bin)
60
60
61 def decompress(bin):
61 def decompress(bin):
62 """ decompress the given input """
62 """ decompress the given input """
63 if not bin: return bin
63 if not bin: return bin
64 t = bin[0]
64 t = bin[0]
65 if t == '\0': return bin
65 if t == '\0': return bin
66 if t == 'x': return zlib.decompress(bin)
66 if t == 'x': return zlib.decompress(bin)
67 if t == 'u': return bin[1:]
67 if t == 'u': return bin[1:]
68 raise RevlogError(_("unknown compression type %r") % t)
68 raise RevlogError(_("unknown compression type %r") % t)
69
69
70 indexformatv0 = ">4l20s20s20s"
70 indexformatv0 = ">4l20s20s20s"
71 v0shaoffset = 56
71 v0shaoffset = 56
72 # index ng:
72 # index ng:
73 # 6 bytes offset
73 # 6 bytes offset
74 # 2 bytes flags
74 # 2 bytes flags
75 # 4 bytes compressed length
75 # 4 bytes compressed length
76 # 4 bytes uncompressed length
76 # 4 bytes uncompressed length
77 # 4 bytes: base rev
77 # 4 bytes: base rev
78 # 4 bytes link rev
78 # 4 bytes link rev
79 # 4 bytes parent 1 rev
79 # 4 bytes parent 1 rev
80 # 4 bytes parent 2 rev
80 # 4 bytes parent 2 rev
81 # 32 bytes: nodeid
81 # 32 bytes: nodeid
82 indexformatng = ">Qiiiiii20s12x"
82 indexformatng = ">Qiiiiii20s12x"
83 ngshaoffset = 32
83 ngshaoffset = 32
84 versionformat = ">i"
84 versionformat = ">i"
85
85
86 class lazyparser(object):
86 class lazyparser(object):
87 """
87 """
88 this class avoids the need to parse the entirety of large indices
88 this class avoids the need to parse the entirety of large indices
89 """
89 """
90
90
91 # lazyparser is not safe to use on windows if win32 extensions not
91 # lazyparser is not safe to use on windows if win32 extensions not
92 # available. it keeps file handle open, which make it not possible
92 # available. it keeps file handle open, which make it not possible
93 # to break hardlinks on local cloned repos.
93 # to break hardlinks on local cloned repos.
94 safe_to_use = os.name != 'nt' or (not util.is_win_9x() and
94 safe_to_use = os.name != 'nt' or (not util.is_win_9x() and
95 hasattr(util, 'win32api'))
95 hasattr(util, 'win32api'))
96
96
97 def __init__(self, dataf, size, indexformat, shaoffset):
97 def __init__(self, dataf, size, indexformat, shaoffset):
98 self.dataf = dataf
98 self.dataf = dataf
99 self.format = indexformat
99 self.format = indexformat
100 self.s = struct.calcsize(indexformat)
100 self.s = struct.calcsize(indexformat)
101 self.indexformat = indexformat
101 self.indexformat = indexformat
102 self.datasize = size
102 self.datasize = size
103 self.l = size/self.s
103 self.l = size/self.s
104 self.index = [None] * self.l
104 self.index = [None] * self.l
105 self.map = {nullid: -1}
105 self.map = {nullid: -1}
106 self.allmap = 0
106 self.allmap = 0
107 self.all = 0
107 self.all = 0
108 self.mapfind_count = 0
108 self.mapfind_count = 0
109 self.shaoffset = shaoffset
109 self.shaoffset = shaoffset
110
110
111 def loadmap(self):
111 def loadmap(self):
112 """
112 """
113 during a commit, we need to make sure the rev being added is
113 during a commit, we need to make sure the rev being added is
114 not a duplicate. This requires loading the entire index,
114 not a duplicate. This requires loading the entire index,
115 which is fairly slow. loadmap can load up just the node map,
115 which is fairly slow. loadmap can load up just the node map,
116 which takes much less time.
116 which takes much less time.
117 """
117 """
118 if self.allmap: return
118 if self.allmap: return
119 end = self.datasize
119 end = self.datasize
120 self.allmap = 1
120 self.allmap = 1
121 cur = 0
121 cur = 0
122 count = 0
122 count = 0
123 blocksize = self.s * 256
123 blocksize = self.s * 256
124 self.dataf.seek(0)
124 self.dataf.seek(0)
125 while cur < end:
125 while cur < end:
126 data = self.dataf.read(blocksize)
126 data = self.dataf.read(blocksize)
127 off = 0
127 off = 0
128 for x in xrange(256):
128 for x in xrange(256):
129 n = data[off + self.shaoffset:off + self.shaoffset + 20]
129 n = data[off + self.shaoffset:off + self.shaoffset + 20]
130 self.map[n] = count
130 self.map[n] = count
131 count += 1
131 count += 1
132 if count >= self.l:
132 if count >= self.l:
133 break
133 break
134 off += self.s
134 off += self.s
135 cur += blocksize
135 cur += blocksize
136
136
137 def loadblock(self, blockstart, blocksize, data=None):
137 def loadblock(self, blockstart, blocksize, data=None):
138 if self.all: return
138 if self.all: return
139 if data is None:
139 if data is None:
140 self.dataf.seek(blockstart)
140 self.dataf.seek(blockstart)
141 if blockstart + blocksize > self.datasize:
141 if blockstart + blocksize > self.datasize:
142 # the revlog may have grown since we've started running,
142 # the revlog may have grown since we've started running,
143 # but we don't have space in self.index for more entries.
143 # but we don't have space in self.index for more entries.
144 # limit blocksize so that we don't get too much data.
144 # limit blocksize so that we don't get too much data.
145 blocksize = max(self.datasize - blockstart, 0)
145 blocksize = max(self.datasize - blockstart, 0)
146 data = self.dataf.read(blocksize)
146 data = self.dataf.read(blocksize)
147 lend = len(data) / self.s
147 lend = len(data) / self.s
148 i = blockstart / self.s
148 i = blockstart / self.s
149 off = 0
149 off = 0
150 for x in xrange(lend):
150 for x in xrange(lend):
151 if self.index[i + x] == None:
151 if self.index[i + x] == None:
152 b = data[off : off + self.s]
152 b = data[off : off + self.s]
153 self.index[i + x] = b
153 self.index[i + x] = b
154 n = b[self.shaoffset:self.shaoffset + 20]
154 n = b[self.shaoffset:self.shaoffset + 20]
155 self.map[n] = i + x
155 self.map[n] = i + x
156 off += self.s
156 off += self.s
157
157
158 def findnode(self, node):
158 def findnode(self, node):
159 """search backwards through the index file for a specific node"""
159 """search backwards through the index file for a specific node"""
160 if self.allmap: return None
160 if self.allmap: return None
161
161
162 # hg log will cause many many searches for the manifest
162 # hg log will cause many many searches for the manifest
163 # nodes. After we get called a few times, just load the whole
163 # nodes. After we get called a few times, just load the whole
164 # thing.
164 # thing.
165 if self.mapfind_count > 8:
165 if self.mapfind_count > 8:
166 self.loadmap()
166 self.loadmap()
167 if node in self.map:
167 if node in self.map:
168 return node
168 return node
169 return None
169 return None
170 self.mapfind_count += 1
170 self.mapfind_count += 1
171 last = self.l - 1
171 last = self.l - 1
172 while self.index[last] != None:
172 while self.index[last] != None:
173 if last == 0:
173 if last == 0:
174 self.all = 1
174 self.all = 1
175 self.allmap = 1
175 self.allmap = 1
176 return None
176 return None
177 last -= 1
177 last -= 1
178 end = (last + 1) * self.s
178 end = (last + 1) * self.s
179 blocksize = self.s * 256
179 blocksize = self.s * 256
180 while end >= 0:
180 while end >= 0:
181 start = max(end - blocksize, 0)
181 start = max(end - blocksize, 0)
182 self.dataf.seek(start)
182 self.dataf.seek(start)
183 data = self.dataf.read(end - start)
183 data = self.dataf.read(end - start)
184 findend = end - start
184 findend = end - start
185 while True:
185 while True:
186 # we're searching backwards, so weh have to make sure
186 # we're searching backwards, so weh have to make sure
187 # we don't find a changeset where this node is a parent
187 # we don't find a changeset where this node is a parent
188 off = data.rfind(node, 0, findend)
188 off = data.rfind(node, 0, findend)
189 findend = off
189 findend = off
190 if off >= 0:
190 if off >= 0:
191 i = off / self.s
191 i = off / self.s
192 off = i * self.s
192 off = i * self.s
193 n = data[off + self.shaoffset:off + self.shaoffset + 20]
193 n = data[off + self.shaoffset:off + self.shaoffset + 20]
194 if n == node:
194 if n == node:
195 self.map[n] = i + start / self.s
195 self.map[n] = i + start / self.s
196 return node
196 return node
197 else:
197 else:
198 break
198 break
199 end -= blocksize
199 end -= blocksize
200 return None
200 return None
201
201
202 def loadindex(self, i=None, end=None):
202 def loadindex(self, i=None, end=None):
203 if self.all: return
203 if self.all: return
204 all = False
204 all = False
205 if i == None:
205 if i == None:
206 blockstart = 0
206 blockstart = 0
207 blocksize = (512 / self.s) * self.s
207 blocksize = (512 / self.s) * self.s
208 end = self.datasize
208 end = self.datasize
209 all = True
209 all = True
210 else:
210 else:
211 if end:
211 if end:
212 blockstart = i * self.s
212 blockstart = i * self.s
213 end = end * self.s
213 end = end * self.s
214 blocksize = end - blockstart
214 blocksize = end - blockstart
215 else:
215 else:
216 blockstart = (i & ~(32)) * self.s
216 blockstart = (i & ~(32)) * self.s
217 blocksize = self.s * 64
217 blocksize = self.s * 64
218 end = blockstart + blocksize
218 end = blockstart + blocksize
219 while blockstart < end:
219 while blockstart < end:
220 self.loadblock(blockstart, blocksize)
220 self.loadblock(blockstart, blocksize)
221 blockstart += blocksize
221 blockstart += blocksize
222 if all: self.all = True
222 if all: self.all = True
223
223
224 class lazyindex(object):
224 class lazyindex(object):
225 """a lazy version of the index array"""
225 """a lazy version of the index array"""
226 def __init__(self, parser):
226 def __init__(self, parser):
227 self.p = parser
227 self.p = parser
228 def __len__(self):
228 def __len__(self):
229 return len(self.p.index)
229 return len(self.p.index)
230 def load(self, pos):
230 def load(self, pos):
231 if pos < 0:
231 if pos < 0:
232 pos += len(self.p.index)
232 pos += len(self.p.index)
233 self.p.loadindex(pos)
233 self.p.loadindex(pos)
234 return self.p.index[pos]
234 return self.p.index[pos]
235 def __getitem__(self, pos):
235 def __getitem__(self, pos):
236 ret = self.p.index[pos] or self.load(pos)
236 ret = self.p.index[pos] or self.load(pos)
237 if isinstance(ret, str):
237 if isinstance(ret, str):
238 ret = struct.unpack(self.p.indexformat, ret)
238 ret = struct.unpack(self.p.indexformat, ret)
239 return ret
239 return ret
240 def __setitem__(self, pos, item):
240 def __setitem__(self, pos, item):
241 self.p.index[pos] = item
241 self.p.index[pos] = item
242 def __delitem__(self, pos):
242 def __delitem__(self, pos):
243 del self.p.index[pos]
243 del self.p.index[pos]
244 def append(self, e):
244 def append(self, e):
245 self.p.index.append(e)
245 self.p.index.append(e)
246
246
247 class lazymap(object):
247 class lazymap(object):
248 """a lazy version of the node map"""
248 """a lazy version of the node map"""
249 def __init__(self, parser):
249 def __init__(self, parser):
250 self.p = parser
250 self.p = parser
251 def load(self, key):
251 def load(self, key):
252 n = self.p.findnode(key)
252 n = self.p.findnode(key)
253 if n == None:
253 if n == None:
254 raise KeyError(key)
254 raise KeyError(key)
255 def __contains__(self, key):
255 def __contains__(self, key):
256 if key in self.p.map:
256 if key in self.p.map:
257 return True
257 return True
258 self.p.loadmap()
258 self.p.loadmap()
259 return key in self.p.map
259 return key in self.p.map
260 def __iter__(self):
260 def __iter__(self):
261 yield nullid
261 yield nullid
262 for i in xrange(self.p.l):
262 for i in xrange(self.p.l):
263 ret = self.p.index[i]
263 ret = self.p.index[i]
264 if not ret:
264 if not ret:
265 self.p.loadindex(i)
265 self.p.loadindex(i)
266 ret = self.p.index[i]
266 ret = self.p.index[i]
267 if isinstance(ret, str):
267 if isinstance(ret, str):
268 ret = struct.unpack(self.p.indexformat, ret)
268 ret = struct.unpack(self.p.indexformat, ret)
269 yield ret[-1]
269 yield ret[-1]
270 def __getitem__(self, key):
270 def __getitem__(self, key):
271 try:
271 try:
272 return self.p.map[key]
272 return self.p.map[key]
273 except KeyError:
273 except KeyError:
274 try:
274 try:
275 self.load(key)
275 self.load(key)
276 return self.p.map[key]
276 return self.p.map[key]
277 except KeyError:
277 except KeyError:
278 raise KeyError("node " + hex(key))
278 raise KeyError("node " + hex(key))
279 def __setitem__(self, key, val):
279 def __setitem__(self, key, val):
280 self.p.map[key] = val
280 self.p.map[key] = val
281 def __delitem__(self, key):
281 def __delitem__(self, key):
282 del self.p.map[key]
282 del self.p.map[key]
283
283
284 class RevlogError(Exception): pass
284 class RevlogError(Exception): pass
285
285
286 class revlog(object):
286 class revlog(object):
287 """
287 """
288 the underlying revision storage object
288 the underlying revision storage object
289
289
290 A revlog consists of two parts, an index and the revision data.
290 A revlog consists of two parts, an index and the revision data.
291
291
292 The index is a file with a fixed record size containing
292 The index is a file with a fixed record size containing
293 information on each revision, includings its nodeid (hash), the
293 information on each revision, includings its nodeid (hash), the
294 nodeids of its parents, the position and offset of its data within
294 nodeids of its parents, the position and offset of its data within
295 the data file, and the revision it's based on. Finally, each entry
295 the data file, and the revision it's based on. Finally, each entry
296 contains a linkrev entry that can serve as a pointer to external
296 contains a linkrev entry that can serve as a pointer to external
297 data.
297 data.
298
298
299 The revision data itself is a linear collection of data chunks.
299 The revision data itself is a linear collection of data chunks.
300 Each chunk represents a revision and is usually represented as a
300 Each chunk represents a revision and is usually represented as a
301 delta against the previous chunk. To bound lookup time, runs of
301 delta against the previous chunk. To bound lookup time, runs of
302 deltas are limited to about 2 times the length of the original
302 deltas are limited to about 2 times the length of the original
303 version data. This makes retrieval of a version proportional to
303 version data. This makes retrieval of a version proportional to
304 its size, or O(1) relative to the number of revisions.
304 its size, or O(1) relative to the number of revisions.
305
305
306 Both pieces of the revlog are written to in an append-only
306 Both pieces of the revlog are written to in an append-only
307 fashion, which means we never need to rewrite a file to insert or
307 fashion, which means we never need to rewrite a file to insert or
308 remove data, and can use some simple techniques to avoid the need
308 remove data, and can use some simple techniques to avoid the need
309 for locking while reading.
309 for locking while reading.
310 """
310 """
311 def __init__(self, opener, indexfile, datafile,
311 def __init__(self, opener, indexfile, datafile,
312 defversion=REVLOG_DEFAULT_VERSION):
312 defversion=REVLOG_DEFAULT_VERSION):
313 """
313 """
314 create a revlog object
314 create a revlog object
315
315
316 opener is a function that abstracts the file opening operation
316 opener is a function that abstracts the file opening operation
317 and can be used to implement COW semantics or the like.
317 and can be used to implement COW semantics or the like.
318 """
318 """
319 self.indexfile = indexfile
319 self.indexfile = indexfile
320 self.datafile = datafile
320 self.datafile = datafile
321 self.opener = opener
321 self.opener = opener
322
322
323 self.indexstat = None
323 self.indexstat = None
324 self.cache = None
324 self.cache = None
325 self.chunkcache = None
325 self.chunkcache = None
326 self.defversion = defversion
326 self.defversion = defversion
327 self.load()
327 self.load()
328
328
329 def load(self):
329 def load(self):
330 v = self.defversion
330 v = self.defversion
331 try:
331 try:
332 f = self.opener(self.indexfile)
332 f = self.opener(self.indexfile)
333 i = f.read(4)
333 i = f.read(4)
334 f.seek(0)
334 f.seek(0)
335 except IOError, inst:
335 except IOError, inst:
336 if inst.errno != errno.ENOENT:
336 if inst.errno != errno.ENOENT:
337 raise
337 raise
338 i = ""
338 i = ""
339 else:
339 else:
340 try:
340 try:
341 st = util.fstat(f)
341 st = util.fstat(f)
342 except AttributeError, inst:
342 except AttributeError, inst:
343 st = None
343 st = None
344 else:
344 else:
345 oldst = self.indexstat
345 oldst = self.indexstat
346 if (oldst and st.st_dev == oldst.st_dev
346 if (oldst and st.st_dev == oldst.st_dev
347 and st.st_ino == oldst.st_ino
347 and st.st_ino == oldst.st_ino
348 and st.st_mtime == oldst.st_mtime
348 and st.st_mtime == oldst.st_mtime
349 and st.st_ctime == oldst.st_ctime):
349 and st.st_ctime == oldst.st_ctime):
350 return
350 return
351 self.indexstat = st
351 self.indexstat = st
352 if len(i) > 0:
352 if len(i) > 0:
353 v = struct.unpack(versionformat, i)[0]
353 v = struct.unpack(versionformat, i)[0]
354 flags = v & ~0xFFFF
354 flags = v & ~0xFFFF
355 fmt = v & 0xFFFF
355 fmt = v & 0xFFFF
356 if fmt == REVLOGV0:
356 if fmt == REVLOGV0:
357 if flags:
357 if flags:
358 raise RevlogError(_("index %s invalid flags %x for format v0" %
358 raise RevlogError(_("index %s invalid flags %x for format v0" %
359 (self.indexfile, flags)))
359 (self.indexfile, flags)))
360 elif fmt == REVLOGNG:
360 elif fmt == REVLOGNG:
361 if flags & ~REVLOGNGINLINEDATA:
361 if flags & ~REVLOGNGINLINEDATA:
362 raise RevlogError(_("index %s invalid flags %x for revlogng" %
362 raise RevlogError(_("index %s invalid flags %x for revlogng" %
363 (self.indexfile, flags)))
363 (self.indexfile, flags)))
364 else:
364 else:
365 raise RevlogError(_("index %s invalid format %d" %
365 raise RevlogError(_("index %s invalid format %d" %
366 (self.indexfile, fmt)))
366 (self.indexfile, fmt)))
367 self.version = v
367 self.version = v
368 if v == REVLOGV0:
368 if v == REVLOGV0:
369 self.indexformat = indexformatv0
369 self.indexformat = indexformatv0
370 shaoffset = v0shaoffset
370 shaoffset = v0shaoffset
371 else:
371 else:
372 self.indexformat = indexformatng
372 self.indexformat = indexformatng
373 shaoffset = ngshaoffset
373 shaoffset = ngshaoffset
374
374
375 if i:
375 if i:
376 if (lazyparser.safe_to_use and not self.inlinedata() and
376 if (lazyparser.safe_to_use and not self.inlinedata() and
377 st and st.st_size > 10000):
377 st and st.st_size > 10000):
378 # big index, let's parse it on demand
378 # big index, let's parse it on demand
379 parser = lazyparser(f, st.st_size, self.indexformat, shaoffset)
379 parser = lazyparser(f, st.st_size, self.indexformat, shaoffset)
380 self.index = lazyindex(parser)
380 self.index = lazyindex(parser)
381 self.nodemap = lazymap(parser)
381 self.nodemap = lazymap(parser)
382 else:
382 else:
383 self.parseindex(f, st)
383 self.parseindex(f, st)
384 if self.version != REVLOGV0:
384 if self.version != REVLOGV0:
385 e = list(self.index[0])
385 e = list(self.index[0])
386 type = self.ngtype(e[0])
386 type = self.ngtype(e[0])
387 e[0] = self.offset_type(0, type)
387 e[0] = self.offset_type(0, type)
388 self.index[0] = e
388 self.index[0] = e
389 else:
389 else:
390 self.nodemap = { nullid: -1}
390 self.nodemap = { nullid: -1}
391 self.index = []
391 self.index = []
392
392
393
393
394 def parseindex(self, fp, st):
394 def parseindex(self, fp, st):
395 s = struct.calcsize(self.indexformat)
395 s = struct.calcsize(self.indexformat)
396 self.index = []
396 self.index = []
397 self.nodemap = {nullid: -1}
397 self.nodemap = {nullid: -1}
398 inline = self.inlinedata()
398 inline = self.inlinedata()
399 n = 0
399 n = 0
400 leftover = None
400 leftover = None
401 while True:
401 while True:
402 if st:
402 if st:
403 data = fp.read(65536)
403 data = fp.read(65536)
404 else:
404 else:
405 # hack for httprangereader, it doesn't do partial reads well
405 # hack for httprangereader, it doesn't do partial reads well
406 data = fp.read()
406 data = fp.read()
407 if not data:
407 if not data:
408 break
408 break
409 if n == 0 and self.inlinedata():
409 if n == 0 and self.inlinedata():
410 # cache the first chunk
410 # cache the first chunk
411 self.chunkcache = (0, data)
411 self.chunkcache = (0, data)
412 if leftover:
412 if leftover:
413 data = leftover + data
413 data = leftover + data
414 leftover = None
414 leftover = None
415 off = 0
415 off = 0
416 l = len(data)
416 l = len(data)
417 while off < l:
417 while off < l:
418 if l - off < s:
418 if l - off < s:
419 leftover = data[off:]
419 leftover = data[off:]
420 break
420 break
421 cur = data[off:off + s]
421 cur = data[off:off + s]
422 off += s
422 off += s
423 e = struct.unpack(self.indexformat, cur)
423 e = struct.unpack(self.indexformat, cur)
424 self.index.append(e)
424 self.index.append(e)
425 self.nodemap[e[-1]] = n
425 self.nodemap[e[-1]] = n
426 n += 1
426 n += 1
427 if inline:
427 if inline:
428 off += e[1]
428 off += e[1]
429 if off > l:
429 if off > l:
430 # some things don't seek well, just read it
430 # some things don't seek well, just read it
431 fp.read(off - l)
431 fp.read(off - l)
432 if not st:
432 if not st:
433 break
433 break
434
434
435
435
436 def ngoffset(self, q):
436 def ngoffset(self, q):
437 if q & 0xFFFF:
437 if q & 0xFFFF:
438 raise RevlogError(_('%s: incompatible revision flag %x') %
438 raise RevlogError(_('%s: incompatible revision flag %x') %
439 (self.indexfile, q))
439 (self.indexfile, q))
440 return long(q >> 16)
440 return long(q >> 16)
441
441
442 def ngtype(self, q):
442 def ngtype(self, q):
443 return int(q & 0xFFFF)
443 return int(q & 0xFFFF)
444
444
445 def offset_type(self, offset, type):
445 def offset_type(self, offset, type):
446 return long(long(offset) << 16 | type)
446 return long(long(offset) << 16 | type)
447
447
448 def loadindex(self, start, end):
448 def loadindex(self, start, end):
449 """load a block of indexes all at once from the lazy parser"""
449 """load a block of indexes all at once from the lazy parser"""
450 if isinstance(self.index, lazyindex):
450 if isinstance(self.index, lazyindex):
451 self.index.p.loadindex(start, end)
451 self.index.p.loadindex(start, end)
452
452
453 def loadindexmap(self):
453 def loadindexmap(self):
454 """loads both the map and the index from the lazy parser"""
454 """loads both the map and the index from the lazy parser"""
455 if isinstance(self.index, lazyindex):
455 if isinstance(self.index, lazyindex):
456 p = self.index.p
456 p = self.index.p
457 p.loadindex()
457 p.loadindex()
458 self.nodemap = p.map
458 self.nodemap = p.map
459
459
460 def loadmap(self):
460 def loadmap(self):
461 """loads the map from the lazy parser"""
461 """loads the map from the lazy parser"""
462 if isinstance(self.nodemap, lazymap):
462 if isinstance(self.nodemap, lazymap):
463 self.nodemap.p.loadmap()
463 self.nodemap.p.loadmap()
464 self.nodemap = self.nodemap.p.map
464 self.nodemap = self.nodemap.p.map
465
465
466 def inlinedata(self): return self.version & REVLOGNGINLINEDATA
466 def inlinedata(self): return self.version & REVLOGNGINLINEDATA
467 def tip(self): return self.node(len(self.index) - 1)
467 def tip(self): return self.node(len(self.index) - 1)
468 def count(self): return len(self.index)
468 def count(self): return len(self.index)
469 def node(self, rev):
469 def node(self, rev):
470 return (rev < 0) and nullid or self.index[rev][-1]
470 return (rev < 0) and nullid or self.index[rev][-1]
471 def rev(self, node):
471 def rev(self, node):
472 try:
472 try:
473 return self.nodemap[node]
473 return self.nodemap[node]
474 except KeyError:
474 except KeyError:
475 raise RevlogError(_('%s: no node %s') % (self.indexfile, hex(node)))
475 raise RevlogError(_('%s: no node %s') % (self.indexfile, hex(node)))
476 def linkrev(self, node):
476 def linkrev(self, node):
477 return (node == nullid) and -1 or self.index[self.rev(node)][-4]
477 return (node == nullid) and -1 or self.index[self.rev(node)][-4]
478 def parents(self, node):
478 def parents(self, node):
479 if node == nullid: return (nullid, nullid)
479 if node == nullid: return (nullid, nullid)
480 r = self.rev(node)
480 r = self.rev(node)
481 d = self.index[r][-3:-1]
481 d = self.index[r][-3:-1]
482 if self.version == REVLOGV0:
482 if self.version == REVLOGV0:
483 return d
483 return d
484 return [ self.node(x) for x in d ]
484 return [ self.node(x) for x in d ]
485 def parentrevs(self, rev):
485 def parentrevs(self, rev):
486 if rev == -1:
486 if rev == -1:
487 return (-1, -1)
487 return (-1, -1)
488 d = self.index[rev][-3:-1]
488 d = self.index[rev][-3:-1]
489 if self.version == REVLOGV0:
489 if self.version == REVLOGV0:
490 return [ self.rev(x) for x in d ]
490 return [ self.rev(x) for x in d ]
491 return d
491 return d
492 def start(self, rev):
492 def start(self, rev):
493 if rev < 0:
493 if rev < 0:
494 return -1
494 return -1
495 if self.version != REVLOGV0:
495 if self.version != REVLOGV0:
496 return self.ngoffset(self.index[rev][0])
496 return self.ngoffset(self.index[rev][0])
497 return self.index[rev][0]
497 return self.index[rev][0]
498
498
499 def end(self, rev): return self.start(rev) + self.length(rev)
499 def end(self, rev): return self.start(rev) + self.length(rev)
500
500
501 def size(self, rev):
501 def size(self, rev):
502 """return the length of the uncompressed text for a given revision"""
502 """return the length of the uncompressed text for a given revision"""
503 l = -1
503 l = -1
504 if self.version != REVLOGV0:
504 if self.version != REVLOGV0:
505 l = self.index[rev][2]
505 l = self.index[rev][2]
506 if l >= 0:
506 if l >= 0:
507 return l
507 return l
508
508
509 t = self.revision(self.node(rev))
509 t = self.revision(self.node(rev))
510 return len(t)
510 return len(t)
511
511
512 # alternate implementation, The advantage to this code is it
512 # alternate implementation, The advantage to this code is it
513 # will be faster for a single revision. But, the results are not
513 # will be faster for a single revision. But, the results are not
514 # cached, so finding the size of every revision will be slower.
514 # cached, so finding the size of every revision will be slower.
515 """
515 """
516 if self.cache and self.cache[1] == rev:
516 if self.cache and self.cache[1] == rev:
517 return len(self.cache[2])
517 return len(self.cache[2])
518
518
519 base = self.base(rev)
519 base = self.base(rev)
520 if self.cache and self.cache[1] >= base and self.cache[1] < rev:
520 if self.cache and self.cache[1] >= base and self.cache[1] < rev:
521 base = self.cache[1]
521 base = self.cache[1]
522 text = self.cache[2]
522 text = self.cache[2]
523 else:
523 else:
524 text = self.revision(self.node(base))
524 text = self.revision(self.node(base))
525
525
526 l = len(text)
526 l = len(text)
527 for x in xrange(base + 1, rev + 1):
527 for x in xrange(base + 1, rev + 1):
528 l = mdiff.patchedsize(l, self.chunk(x))
528 l = mdiff.patchedsize(l, self.chunk(x))
529 return l
529 return l
530 """
530 """
531
531
532 def length(self, rev):
532 def length(self, rev):
533 if rev < 0:
533 if rev < 0:
534 return 0
534 return 0
535 else:
535 else:
536 return self.index[rev][1]
536 return self.index[rev][1]
537 def base(self, rev): return (rev < 0) and rev or self.index[rev][-5]
537 def base(self, rev): return (rev < 0) and rev or self.index[rev][-5]
538
538
539 def reachable(self, rev, stop=None):
539 def reachable(self, rev, stop=None):
540 reachable = {}
540 reachable = {}
541 visit = [rev]
541 visit = [rev]
542 reachable[rev] = 1
542 reachable[rev] = 1
543 if stop:
543 if stop:
544 stopn = self.rev(stop)
544 stopn = self.rev(stop)
545 else:
545 else:
546 stopn = 0
546 stopn = 0
547 while visit:
547 while visit:
548 n = visit.pop(0)
548 n = visit.pop(0)
549 if n == stop:
549 if n == stop:
550 continue
550 continue
551 if n == nullid:
551 if n == nullid:
552 continue
552 continue
553 for p in self.parents(n):
553 for p in self.parents(n):
554 if self.rev(p) < stopn:
554 if self.rev(p) < stopn:
555 continue
555 continue
556 if p not in reachable:
556 if p not in reachable:
557 reachable[p] = 1
557 reachable[p] = 1
558 visit.append(p)
558 visit.append(p)
559 return reachable
559 return reachable
560
560
561 def nodesbetween(self, roots=None, heads=None):
561 def nodesbetween(self, roots=None, heads=None):
562 """Return a tuple containing three elements. Elements 1 and 2 contain
562 """Return a tuple containing three elements. Elements 1 and 2 contain
563 a final list bases and heads after all the unreachable ones have been
563 a final list bases and heads after all the unreachable ones have been
564 pruned. Element 0 contains a topologically sorted list of all
564 pruned. Element 0 contains a topologically sorted list of all
565
565
566 nodes that satisfy these constraints:
566 nodes that satisfy these constraints:
567 1. All nodes must be descended from a node in roots (the nodes on
567 1. All nodes must be descended from a node in roots (the nodes on
568 roots are considered descended from themselves).
568 roots are considered descended from themselves).
569 2. All nodes must also be ancestors of a node in heads (the nodes in
569 2. All nodes must also be ancestors of a node in heads (the nodes in
570 heads are considered to be their own ancestors).
570 heads are considered to be their own ancestors).
571
571
572 If roots is unspecified, nullid is assumed as the only root.
572 If roots is unspecified, nullid is assumed as the only root.
573 If heads is unspecified, it is taken to be the output of the
573 If heads is unspecified, it is taken to be the output of the
574 heads method (i.e. a list of all nodes in the repository that
574 heads method (i.e. a list of all nodes in the repository that
575 have no children)."""
575 have no children)."""
576 nonodes = ([], [], [])
576 nonodes = ([], [], [])
577 if roots is not None:
577 if roots is not None:
578 roots = list(roots)
578 roots = list(roots)
579 if not roots:
579 if not roots:
580 return nonodes
580 return nonodes
581 lowestrev = min([self.rev(n) for n in roots])
581 lowestrev = min([self.rev(n) for n in roots])
582 else:
582 else:
583 roots = [nullid] # Everybody's a descendent of nullid
583 roots = [nullid] # Everybody's a descendent of nullid
584 lowestrev = -1
584 lowestrev = -1
585 if (lowestrev == -1) and (heads is None):
585 if (lowestrev == -1) and (heads is None):
586 # We want _all_ the nodes!
586 # We want _all_ the nodes!
587 return ([self.node(r) for r in xrange(0, self.count())],
587 return ([self.node(r) for r in xrange(0, self.count())],
588 [nullid], list(self.heads()))
588 [nullid], list(self.heads()))
589 if heads is None:
589 if heads is None:
590 # All nodes are ancestors, so the latest ancestor is the last
590 # All nodes are ancestors, so the latest ancestor is the last
591 # node.
591 # node.
592 highestrev = self.count() - 1
592 highestrev = self.count() - 1
593 # Set ancestors to None to signal that every node is an ancestor.
593 # Set ancestors to None to signal that every node is an ancestor.
594 ancestors = None
594 ancestors = None
595 # Set heads to an empty dictionary for later discovery of heads
595 # Set heads to an empty dictionary for later discovery of heads
596 heads = {}
596 heads = {}
597 else:
597 else:
598 heads = list(heads)
598 heads = list(heads)
599 if not heads:
599 if not heads:
600 return nonodes
600 return nonodes
601 ancestors = {}
601 ancestors = {}
602 # Turn heads into a dictionary so we can remove 'fake' heads.
602 # Turn heads into a dictionary so we can remove 'fake' heads.
603 # Also, later we will be using it to filter out the heads we can't
603 # Also, later we will be using it to filter out the heads we can't
604 # find from roots.
604 # find from roots.
605 heads = dict.fromkeys(heads, 0)
605 heads = dict.fromkeys(heads, 0)
606 # Start at the top and keep marking parents until we're done.
606 # Start at the top and keep marking parents until we're done.
607 nodestotag = heads.keys()
607 nodestotag = heads.keys()
608 # Remember where the top was so we can use it as a limit later.
608 # Remember where the top was so we can use it as a limit later.
609 highestrev = max([self.rev(n) for n in nodestotag])
609 highestrev = max([self.rev(n) for n in nodestotag])
610 while nodestotag:
610 while nodestotag:
611 # grab a node to tag
611 # grab a node to tag
612 n = nodestotag.pop()
612 n = nodestotag.pop()
613 # Never tag nullid
613 # Never tag nullid
614 if n == nullid:
614 if n == nullid:
615 continue
615 continue
616 # A node's revision number represents its place in a
616 # A node's revision number represents its place in a
617 # topologically sorted list of nodes.
617 # topologically sorted list of nodes.
618 r = self.rev(n)
618 r = self.rev(n)
619 if r >= lowestrev:
619 if r >= lowestrev:
620 if n not in ancestors:
620 if n not in ancestors:
621 # If we are possibly a descendent of one of the roots
621 # If we are possibly a descendent of one of the roots
622 # and we haven't already been marked as an ancestor
622 # and we haven't already been marked as an ancestor
623 ancestors[n] = 1 # Mark as ancestor
623 ancestors[n] = 1 # Mark as ancestor
624 # Add non-nullid parents to list of nodes to tag.
624 # Add non-nullid parents to list of nodes to tag.
625 nodestotag.extend([p for p in self.parents(n) if
625 nodestotag.extend([p for p in self.parents(n) if
626 p != nullid])
626 p != nullid])
627 elif n in heads: # We've seen it before, is it a fake head?
627 elif n in heads: # We've seen it before, is it a fake head?
628 # So it is, real heads should not be the ancestors of
628 # So it is, real heads should not be the ancestors of
629 # any other heads.
629 # any other heads.
630 heads.pop(n)
630 heads.pop(n)
631 if not ancestors:
631 if not ancestors:
632 return nonodes
632 return nonodes
633 # Now that we have our set of ancestors, we want to remove any
633 # Now that we have our set of ancestors, we want to remove any
634 # roots that are not ancestors.
634 # roots that are not ancestors.
635
635
636 # If one of the roots was nullid, everything is included anyway.
636 # If one of the roots was nullid, everything is included anyway.
637 if lowestrev > -1:
637 if lowestrev > -1:
638 # But, since we weren't, let's recompute the lowest rev to not
638 # But, since we weren't, let's recompute the lowest rev to not
639 # include roots that aren't ancestors.
639 # include roots that aren't ancestors.
640
640
641 # Filter out roots that aren't ancestors of heads
641 # Filter out roots that aren't ancestors of heads
642 roots = [n for n in roots if n in ancestors]
642 roots = [n for n in roots if n in ancestors]
643 # Recompute the lowest revision
643 # Recompute the lowest revision
644 if roots:
644 if roots:
645 lowestrev = min([self.rev(n) for n in roots])
645 lowestrev = min([self.rev(n) for n in roots])
646 else:
646 else:
647 # No more roots? Return empty list
647 # No more roots? Return empty list
648 return nonodes
648 return nonodes
649 else:
649 else:
650 # We are descending from nullid, and don't need to care about
650 # We are descending from nullid, and don't need to care about
651 # any other roots.
651 # any other roots.
652 lowestrev = -1
652 lowestrev = -1
653 roots = [nullid]
653 roots = [nullid]
654 # Transform our roots list into a 'set' (i.e. a dictionary where the
654 # Transform our roots list into a 'set' (i.e. a dictionary where the
655 # values don't matter.
655 # values don't matter.
656 descendents = dict.fromkeys(roots, 1)
656 descendents = dict.fromkeys(roots, 1)
657 # Also, keep the original roots so we can filter out roots that aren't
657 # Also, keep the original roots so we can filter out roots that aren't
658 # 'real' roots (i.e. are descended from other roots).
658 # 'real' roots (i.e. are descended from other roots).
659 roots = descendents.copy()
659 roots = descendents.copy()
660 # Our topologically sorted list of output nodes.
660 # Our topologically sorted list of output nodes.
661 orderedout = []
661 orderedout = []
662 # Don't start at nullid since we don't want nullid in our output list,
662 # Don't start at nullid since we don't want nullid in our output list,
663 # and if nullid shows up in descedents, empty parents will look like
663 # and if nullid shows up in descedents, empty parents will look like
664 # they're descendents.
664 # they're descendents.
665 for r in xrange(max(lowestrev, 0), highestrev + 1):
665 for r in xrange(max(lowestrev, 0), highestrev + 1):
666 n = self.node(r)
666 n = self.node(r)
667 isdescendent = False
667 isdescendent = False
668 if lowestrev == -1: # Everybody is a descendent of nullid
668 if lowestrev == -1: # Everybody is a descendent of nullid
669 isdescendent = True
669 isdescendent = True
670 elif n in descendents:
670 elif n in descendents:
671 # n is already a descendent
671 # n is already a descendent
672 isdescendent = True
672 isdescendent = True
673 # This check only needs to be done here because all the roots
673 # This check only needs to be done here because all the roots
674 # will start being marked is descendents before the loop.
674 # will start being marked is descendents before the loop.
675 if n in roots:
675 if n in roots:
676 # If n was a root, check if it's a 'real' root.
676 # If n was a root, check if it's a 'real' root.
677 p = tuple(self.parents(n))
677 p = tuple(self.parents(n))
678 # If any of its parents are descendents, it's not a root.
678 # If any of its parents are descendents, it's not a root.
679 if (p[0] in descendents) or (p[1] in descendents):
679 if (p[0] in descendents) or (p[1] in descendents):
680 roots.pop(n)
680 roots.pop(n)
681 else:
681 else:
682 p = tuple(self.parents(n))
682 p = tuple(self.parents(n))
683 # A node is a descendent if either of its parents are
683 # A node is a descendent if either of its parents are
684 # descendents. (We seeded the dependents list with the roots
684 # descendents. (We seeded the dependents list with the roots
685 # up there, remember?)
685 # up there, remember?)
686 if (p[0] in descendents) or (p[1] in descendents):
686 if (p[0] in descendents) or (p[1] in descendents):
687 descendents[n] = 1
687 descendents[n] = 1
688 isdescendent = True
688 isdescendent = True
689 if isdescendent and ((ancestors is None) or (n in ancestors)):
689 if isdescendent and ((ancestors is None) or (n in ancestors)):
690 # Only include nodes that are both descendents and ancestors.
690 # Only include nodes that are both descendents and ancestors.
691 orderedout.append(n)
691 orderedout.append(n)
692 if (ancestors is not None) and (n in heads):
692 if (ancestors is not None) and (n in heads):
693 # We're trying to figure out which heads are reachable
693 # We're trying to figure out which heads are reachable
694 # from roots.
694 # from roots.
695 # Mark this head as having been reached
695 # Mark this head as having been reached
696 heads[n] = 1
696 heads[n] = 1
697 elif ancestors is None:
697 elif ancestors is None:
698 # Otherwise, we're trying to discover the heads.
698 # Otherwise, we're trying to discover the heads.
699 # Assume this is a head because if it isn't, the next step
699 # Assume this is a head because if it isn't, the next step
700 # will eventually remove it.
700 # will eventually remove it.
701 heads[n] = 1
701 heads[n] = 1
702 # But, obviously its parents aren't.
702 # But, obviously its parents aren't.
703 for p in self.parents(n):
703 for p in self.parents(n):
704 heads.pop(p, None)
704 heads.pop(p, None)
705 heads = [n for n in heads.iterkeys() if heads[n] != 0]
705 heads = [n for n in heads.iterkeys() if heads[n] != 0]
706 roots = roots.keys()
706 roots = roots.keys()
707 assert orderedout
707 assert orderedout
708 assert roots
708 assert roots
709 assert heads
709 assert heads
710 return (orderedout, roots, heads)
710 return (orderedout, roots, heads)
711
711
712 def heads(self, start=None):
712 def heads(self, start=None):
713 """return the list of all nodes that have no children
713 """return the list of all nodes that have no children
714
714
715 if start is specified, only heads that are descendants of
715 if start is specified, only heads that are descendants of
716 start will be returned
716 start will be returned
717
717
718 """
718 """
719 if start is None:
719 if start is None:
720 start = nullid
720 start = nullid
721 startrev = self.rev(start)
721 startrev = self.rev(start)
722 reachable = {startrev: 1}
722 reachable = {startrev: 1}
723 heads = {startrev: 1}
723 heads = {startrev: 1}
724
724
725 parentrevs = self.parentrevs
725 parentrevs = self.parentrevs
726 for r in xrange(startrev + 1, self.count()):
726 for r in xrange(startrev + 1, self.count()):
727 for p in parentrevs(r):
727 for p in parentrevs(r):
728 if p in reachable:
728 if p in reachable:
729 reachable[r] = 1
729 reachable[r] = 1
730 heads[r] = 1
730 heads[r] = 1
731 if p in heads:
731 if p in heads:
732 del heads[p]
732 del heads[p]
733 return [self.node(r) for r in heads]
733 return [self.node(r) for r in heads]
734
734
735 def children(self, node):
735 def children(self, node):
736 """find the children of a given node"""
736 """find the children of a given node"""
737 c = []
737 c = []
738 p = self.rev(node)
738 p = self.rev(node)
739 for r in range(p + 1, self.count()):
739 for r in range(p + 1, self.count()):
740 n = self.node(r)
740 for pr in self.parentrevs(n):
741 for pn in self.parents(n):
741 if pr == p:
742 if pn == node:
742 c.append(self.node(r))
743 c.append(n)
744 continue
745 elif pn == nullid:
746 continue
747 return c
743 return c
748
744
749 def lookup(self, id):
745 def lookup(self, id):
750 """locate a node based on:
746 """locate a node based on:
751 - revision number or str(revision number)
747 - revision number or str(revision number)
752 - nodeid or subset of hex nodeid
748 - nodeid or subset of hex nodeid
753 """
749 """
754 if isinstance(id, (long, int)):
750 if isinstance(id, (long, int)):
755 # rev
751 # rev
756 return self.node(id)
752 return self.node(id)
757 if len(id) == 20:
753 if len(id) == 20:
758 # possibly a binary node
754 # possibly a binary node
759 # odds of a binary node being all hex in ASCII are 1 in 10**25
755 # odds of a binary node being all hex in ASCII are 1 in 10**25
760 try:
756 try:
761 node = id
757 node = id
762 r = self.rev(node) # quick search the index
758 r = self.rev(node) # quick search the index
763 return node
759 return node
764 except RevlogError:
760 except RevlogError:
765 pass # may be partial hex id
761 pass # may be partial hex id
766 try:
762 try:
767 # str(rev)
763 # str(rev)
768 rev = int(id)
764 rev = int(id)
769 if str(rev) != id: raise ValueError
765 if str(rev) != id: raise ValueError
770 if rev < 0: rev = self.count() + rev
766 if rev < 0: rev = self.count() + rev
771 if rev < 0 or rev >= self.count(): raise ValueError
767 if rev < 0 or rev >= self.count(): raise ValueError
772 return self.node(rev)
768 return self.node(rev)
773 except (ValueError, OverflowError):
769 except (ValueError, OverflowError):
774 pass
770 pass
775 try:
771 try:
776 if len(id) == 40:
772 if len(id) == 40:
777 # a full hex nodeid?
773 # a full hex nodeid?
778 node = bin(id)
774 node = bin(id)
779 r = self.rev(node)
775 r = self.rev(node)
780 return node
776 return node
781 elif len(id) < 40:
777 elif len(id) < 40:
782 # hex(node)[:...]
778 # hex(node)[:...]
783 bin_id = bin(id[:len(id) & ~1]) # grab an even number of digits
779 bin_id = bin(id[:len(id) & ~1]) # grab an even number of digits
784 node = None
780 node = None
785 for n in self.nodemap:
781 for n in self.nodemap:
786 if n.startswith(bin_id) and hex(n).startswith(id):
782 if n.startswith(bin_id) and hex(n).startswith(id):
787 if node is not None:
783 if node is not None:
788 raise RevlogError(_("Ambiguous identifier"))
784 raise RevlogError(_("Ambiguous identifier"))
789 node = n
785 node = n
790 if node is not None:
786 if node is not None:
791 return node
787 return node
792 except TypeError:
788 except TypeError:
793 pass
789 pass
794
790
795 raise RevlogError(_("No match found"))
791 raise RevlogError(_("No match found"))
796
792
797 def cmp(self, node, text):
793 def cmp(self, node, text):
798 """compare text with a given file revision"""
794 """compare text with a given file revision"""
799 p1, p2 = self.parents(node)
795 p1, p2 = self.parents(node)
800 return hash(text, p1, p2) != node
796 return hash(text, p1, p2) != node
801
797
802 def makenode(self, node, text):
798 def makenode(self, node, text):
803 """calculate a file nodeid for text, descended or possibly
799 """calculate a file nodeid for text, descended or possibly
804 unchanged from node"""
800 unchanged from node"""
805
801
806 if self.cmp(node, text):
802 if self.cmp(node, text):
807 return hash(text, node, nullid)
803 return hash(text, node, nullid)
808 return node
804 return node
809
805
810 def diff(self, a, b):
806 def diff(self, a, b):
811 """return a delta between two revisions"""
807 """return a delta between two revisions"""
812 return mdiff.textdiff(a, b)
808 return mdiff.textdiff(a, b)
813
809
814 def patches(self, t, pl):
810 def patches(self, t, pl):
815 """apply a list of patches to a string"""
811 """apply a list of patches to a string"""
816 return mdiff.patches(t, pl)
812 return mdiff.patches(t, pl)
817
813
818 def chunk(self, rev, df=None, cachelen=4096):
814 def chunk(self, rev, df=None, cachelen=4096):
819 start, length = self.start(rev), self.length(rev)
815 start, length = self.start(rev), self.length(rev)
820 inline = self.inlinedata()
816 inline = self.inlinedata()
821 if inline:
817 if inline:
822 start += (rev + 1) * struct.calcsize(self.indexformat)
818 start += (rev + 1) * struct.calcsize(self.indexformat)
823 end = start + length
819 end = start + length
824 def loadcache(df):
820 def loadcache(df):
825 cache_length = max(cachelen, length) # 4k
821 cache_length = max(cachelen, length) # 4k
826 if not df:
822 if not df:
827 if inline:
823 if inline:
828 df = self.opener(self.indexfile)
824 df = self.opener(self.indexfile)
829 else:
825 else:
830 df = self.opener(self.datafile)
826 df = self.opener(self.datafile)
831 df.seek(start)
827 df.seek(start)
832 self.chunkcache = (start, df.read(cache_length))
828 self.chunkcache = (start, df.read(cache_length))
833
829
834 if not self.chunkcache:
830 if not self.chunkcache:
835 loadcache(df)
831 loadcache(df)
836
832
837 cache_start = self.chunkcache[0]
833 cache_start = self.chunkcache[0]
838 cache_end = cache_start + len(self.chunkcache[1])
834 cache_end = cache_start + len(self.chunkcache[1])
839 if start >= cache_start and end <= cache_end:
835 if start >= cache_start and end <= cache_end:
840 # it is cached
836 # it is cached
841 offset = start - cache_start
837 offset = start - cache_start
842 else:
838 else:
843 loadcache(df)
839 loadcache(df)
844 offset = 0
840 offset = 0
845
841
846 #def checkchunk():
842 #def checkchunk():
847 # df = self.opener(self.datafile)
843 # df = self.opener(self.datafile)
848 # df.seek(start)
844 # df.seek(start)
849 # return df.read(length)
845 # return df.read(length)
850 #assert s == checkchunk()
846 #assert s == checkchunk()
851 return decompress(self.chunkcache[1][offset:offset + length])
847 return decompress(self.chunkcache[1][offset:offset + length])
852
848
853 def delta(self, node):
849 def delta(self, node):
854 """return or calculate a delta between a node and its predecessor"""
850 """return or calculate a delta between a node and its predecessor"""
855 r = self.rev(node)
851 r = self.rev(node)
856 return self.revdiff(r - 1, r)
852 return self.revdiff(r - 1, r)
857
853
858 def revdiff(self, rev1, rev2):
854 def revdiff(self, rev1, rev2):
859 """return or calculate a delta between two revisions"""
855 """return or calculate a delta between two revisions"""
860 b1 = self.base(rev1)
856 b1 = self.base(rev1)
861 b2 = self.base(rev2)
857 b2 = self.base(rev2)
862 if b1 == b2 and rev1 + 1 == rev2:
858 if b1 == b2 and rev1 + 1 == rev2:
863 return self.chunk(rev2)
859 return self.chunk(rev2)
864 else:
860 else:
865 return self.diff(self.revision(self.node(rev1)),
861 return self.diff(self.revision(self.node(rev1)),
866 self.revision(self.node(rev2)))
862 self.revision(self.node(rev2)))
867
863
868 def revision(self, node):
864 def revision(self, node):
869 """return an uncompressed revision of a given"""
865 """return an uncompressed revision of a given"""
870 if node == nullid: return ""
866 if node == nullid: return ""
871 if self.cache and self.cache[0] == node: return self.cache[2]
867 if self.cache and self.cache[0] == node: return self.cache[2]
872
868
873 # look up what we need to read
869 # look up what we need to read
874 text = None
870 text = None
875 rev = self.rev(node)
871 rev = self.rev(node)
876 base = self.base(rev)
872 base = self.base(rev)
877
873
878 if self.inlinedata():
874 if self.inlinedata():
879 # we probably have the whole chunk cached
875 # we probably have the whole chunk cached
880 df = None
876 df = None
881 else:
877 else:
882 df = self.opener(self.datafile)
878 df = self.opener(self.datafile)
883
879
884 # do we have useful data cached?
880 # do we have useful data cached?
885 if self.cache and self.cache[1] >= base and self.cache[1] < rev:
881 if self.cache and self.cache[1] >= base and self.cache[1] < rev:
886 base = self.cache[1]
882 base = self.cache[1]
887 text = self.cache[2]
883 text = self.cache[2]
888 self.loadindex(base, rev + 1)
884 self.loadindex(base, rev + 1)
889 else:
885 else:
890 self.loadindex(base, rev + 1)
886 self.loadindex(base, rev + 1)
891 text = self.chunk(base, df=df)
887 text = self.chunk(base, df=df)
892
888
893 bins = []
889 bins = []
894 for r in xrange(base + 1, rev + 1):
890 for r in xrange(base + 1, rev + 1):
895 bins.append(self.chunk(r, df=df))
891 bins.append(self.chunk(r, df=df))
896
892
897 text = self.patches(text, bins)
893 text = self.patches(text, bins)
898
894
899 p1, p2 = self.parents(node)
895 p1, p2 = self.parents(node)
900 if node != hash(text, p1, p2):
896 if node != hash(text, p1, p2):
901 raise RevlogError(_("integrity check failed on %s:%d")
897 raise RevlogError(_("integrity check failed on %s:%d")
902 % (self.datafile, rev))
898 % (self.datafile, rev))
903
899
904 self.cache = (node, rev, text)
900 self.cache = (node, rev, text)
905 return text
901 return text
906
902
907 def checkinlinesize(self, tr, fp=None):
903 def checkinlinesize(self, tr, fp=None):
908 if not self.inlinedata():
904 if not self.inlinedata():
909 return
905 return
910 if not fp:
906 if not fp:
911 fp = self.opener(self.indexfile, 'r')
907 fp = self.opener(self.indexfile, 'r')
912 fp.seek(0, 2)
908 fp.seek(0, 2)
913 size = fp.tell()
909 size = fp.tell()
914 if size < 131072:
910 if size < 131072:
915 return
911 return
916 trinfo = tr.find(self.indexfile)
912 trinfo = tr.find(self.indexfile)
917 if trinfo == None:
913 if trinfo == None:
918 raise RevlogError(_("%s not found in the transaction" %
914 raise RevlogError(_("%s not found in the transaction" %
919 self.indexfile))
915 self.indexfile))
920
916
921 trindex = trinfo[2]
917 trindex = trinfo[2]
922 dataoff = self.start(trindex)
918 dataoff = self.start(trindex)
923
919
924 tr.add(self.datafile, dataoff)
920 tr.add(self.datafile, dataoff)
925 df = self.opener(self.datafile, 'w')
921 df = self.opener(self.datafile, 'w')
926 calc = struct.calcsize(self.indexformat)
922 calc = struct.calcsize(self.indexformat)
927 for r in xrange(self.count()):
923 for r in xrange(self.count()):
928 start = self.start(r) + (r + 1) * calc
924 start = self.start(r) + (r + 1) * calc
929 length = self.length(r)
925 length = self.length(r)
930 fp.seek(start)
926 fp.seek(start)
931 d = fp.read(length)
927 d = fp.read(length)
932 df.write(d)
928 df.write(d)
933 fp.close()
929 fp.close()
934 df.close()
930 df.close()
935 fp = self.opener(self.indexfile, 'w', atomictemp=True)
931 fp = self.opener(self.indexfile, 'w', atomictemp=True)
936 self.version &= ~(REVLOGNGINLINEDATA)
932 self.version &= ~(REVLOGNGINLINEDATA)
937 if self.count():
933 if self.count():
938 x = self.index[0]
934 x = self.index[0]
939 e = struct.pack(self.indexformat, *x)[4:]
935 e = struct.pack(self.indexformat, *x)[4:]
940 l = struct.pack(versionformat, self.version)
936 l = struct.pack(versionformat, self.version)
941 fp.write(l)
937 fp.write(l)
942 fp.write(e)
938 fp.write(e)
943
939
944 for i in xrange(1, self.count()):
940 for i in xrange(1, self.count()):
945 x = self.index[i]
941 x = self.index[i]
946 e = struct.pack(self.indexformat, *x)
942 e = struct.pack(self.indexformat, *x)
947 fp.write(e)
943 fp.write(e)
948
944
949 # if we don't call rename, the temp file will never replace the
945 # if we don't call rename, the temp file will never replace the
950 # real index
946 # real index
951 fp.rename()
947 fp.rename()
952
948
953 tr.replace(self.indexfile, trindex * calc)
949 tr.replace(self.indexfile, trindex * calc)
954 self.chunkcache = None
950 self.chunkcache = None
955
951
956 def addrevision(self, text, transaction, link, p1=None, p2=None, d=None):
952 def addrevision(self, text, transaction, link, p1=None, p2=None, d=None):
957 """add a revision to the log
953 """add a revision to the log
958
954
959 text - the revision data to add
955 text - the revision data to add
960 transaction - the transaction object used for rollback
956 transaction - the transaction object used for rollback
961 link - the linkrev data to add
957 link - the linkrev data to add
962 p1, p2 - the parent nodeids of the revision
958 p1, p2 - the parent nodeids of the revision
963 d - an optional precomputed delta
959 d - an optional precomputed delta
964 """
960 """
965 if not self.inlinedata():
961 if not self.inlinedata():
966 dfh = self.opener(self.datafile, "a")
962 dfh = self.opener(self.datafile, "a")
967 else:
963 else:
968 dfh = None
964 dfh = None
969 ifh = self.opener(self.indexfile, "a+")
965 ifh = self.opener(self.indexfile, "a+")
970 return self._addrevision(text, transaction, link, p1, p2, d, ifh, dfh)
966 return self._addrevision(text, transaction, link, p1, p2, d, ifh, dfh)
971
967
972 def _addrevision(self, text, transaction, link, p1, p2, d, ifh, dfh):
968 def _addrevision(self, text, transaction, link, p1, p2, d, ifh, dfh):
973 if text is None: text = ""
969 if text is None: text = ""
974 if p1 is None: p1 = self.tip()
970 if p1 is None: p1 = self.tip()
975 if p2 is None: p2 = nullid
971 if p2 is None: p2 = nullid
976
972
977 node = hash(text, p1, p2)
973 node = hash(text, p1, p2)
978
974
979 if node in self.nodemap:
975 if node in self.nodemap:
980 return node
976 return node
981
977
982 n = self.count()
978 n = self.count()
983 t = n - 1
979 t = n - 1
984
980
985 if n:
981 if n:
986 base = self.base(t)
982 base = self.base(t)
987 start = self.start(base)
983 start = self.start(base)
988 end = self.end(t)
984 end = self.end(t)
989 if not d:
985 if not d:
990 prev = self.revision(self.tip())
986 prev = self.revision(self.tip())
991 d = self.diff(prev, text)
987 d = self.diff(prev, text)
992 data = compress(d)
988 data = compress(d)
993 l = len(data[1]) + len(data[0])
989 l = len(data[1]) + len(data[0])
994 dist = end - start + l
990 dist = end - start + l
995
991
996 # full versions are inserted when the needed deltas
992 # full versions are inserted when the needed deltas
997 # become comparable to the uncompressed text
993 # become comparable to the uncompressed text
998 if not n or dist > len(text) * 2:
994 if not n or dist > len(text) * 2:
999 data = compress(text)
995 data = compress(text)
1000 l = len(data[1]) + len(data[0])
996 l = len(data[1]) + len(data[0])
1001 base = n
997 base = n
1002 else:
998 else:
1003 base = self.base(t)
999 base = self.base(t)
1004
1000
1005 offset = 0
1001 offset = 0
1006 if t >= 0:
1002 if t >= 0:
1007 offset = self.end(t)
1003 offset = self.end(t)
1008
1004
1009 if self.version == REVLOGV0:
1005 if self.version == REVLOGV0:
1010 e = (offset, l, base, link, p1, p2, node)
1006 e = (offset, l, base, link, p1, p2, node)
1011 else:
1007 else:
1012 e = (self.offset_type(offset, 0), l, len(text),
1008 e = (self.offset_type(offset, 0), l, len(text),
1013 base, link, self.rev(p1), self.rev(p2), node)
1009 base, link, self.rev(p1), self.rev(p2), node)
1014
1010
1015 self.index.append(e)
1011 self.index.append(e)
1016 self.nodemap[node] = n
1012 self.nodemap[node] = n
1017 entry = struct.pack(self.indexformat, *e)
1013 entry = struct.pack(self.indexformat, *e)
1018
1014
1019 if not self.inlinedata():
1015 if not self.inlinedata():
1020 transaction.add(self.datafile, offset)
1016 transaction.add(self.datafile, offset)
1021 transaction.add(self.indexfile, n * len(entry))
1017 transaction.add(self.indexfile, n * len(entry))
1022 if data[0]:
1018 if data[0]:
1023 dfh.write(data[0])
1019 dfh.write(data[0])
1024 dfh.write(data[1])
1020 dfh.write(data[1])
1025 dfh.flush()
1021 dfh.flush()
1026 else:
1022 else:
1027 ifh.seek(0, 2)
1023 ifh.seek(0, 2)
1028 transaction.add(self.indexfile, ifh.tell(), self.count() - 1)
1024 transaction.add(self.indexfile, ifh.tell(), self.count() - 1)
1029
1025
1030 if len(self.index) == 1 and self.version != REVLOGV0:
1026 if len(self.index) == 1 and self.version != REVLOGV0:
1031 l = struct.pack(versionformat, self.version)
1027 l = struct.pack(versionformat, self.version)
1032 ifh.write(l)
1028 ifh.write(l)
1033 entry = entry[4:]
1029 entry = entry[4:]
1034
1030
1035 ifh.write(entry)
1031 ifh.write(entry)
1036
1032
1037 if self.inlinedata():
1033 if self.inlinedata():
1038 ifh.write(data[0])
1034 ifh.write(data[0])
1039 ifh.write(data[1])
1035 ifh.write(data[1])
1040 self.checkinlinesize(transaction, ifh)
1036 self.checkinlinesize(transaction, ifh)
1041
1037
1042 self.cache = (node, n, text)
1038 self.cache = (node, n, text)
1043 return node
1039 return node
1044
1040
1045 def ancestor(self, a, b):
1041 def ancestor(self, a, b):
1046 """calculate the least common ancestor of nodes a and b"""
1042 """calculate the least common ancestor of nodes a and b"""
1047
1043
1048 def parents(rev):
1044 def parents(rev):
1049 return [p for p in self.parentrevs(rev) if p != -1]
1045 return [p for p in self.parentrevs(rev) if p != -1]
1050
1046
1051 c = ancestor.ancestor(self.rev(a), self.rev(b), parents)
1047 c = ancestor.ancestor(self.rev(a), self.rev(b), parents)
1052 if c is None:
1048 if c is None:
1053 return nullid
1049 return nullid
1054
1050
1055 return self.node(c)
1051 return self.node(c)
1056
1052
1057 def group(self, nodelist, lookup, infocollect=None):
1053 def group(self, nodelist, lookup, infocollect=None):
1058 """calculate a delta group
1054 """calculate a delta group
1059
1055
1060 Given a list of changeset revs, return a set of deltas and
1056 Given a list of changeset revs, return a set of deltas and
1061 metadata corresponding to nodes. the first delta is
1057 metadata corresponding to nodes. the first delta is
1062 parent(nodes[0]) -> nodes[0] the receiver is guaranteed to
1058 parent(nodes[0]) -> nodes[0] the receiver is guaranteed to
1063 have this parent as it has all history before these
1059 have this parent as it has all history before these
1064 changesets. parent is parent[0]
1060 changesets. parent is parent[0]
1065 """
1061 """
1066 revs = [self.rev(n) for n in nodelist]
1062 revs = [self.rev(n) for n in nodelist]
1067
1063
1068 # if we don't have any revisions touched by these changesets, bail
1064 # if we don't have any revisions touched by these changesets, bail
1069 if not revs:
1065 if not revs:
1070 yield changegroup.closechunk()
1066 yield changegroup.closechunk()
1071 return
1067 return
1072
1068
1073 # add the parent of the first rev
1069 # add the parent of the first rev
1074 p = self.parents(self.node(revs[0]))[0]
1070 p = self.parents(self.node(revs[0]))[0]
1075 revs.insert(0, self.rev(p))
1071 revs.insert(0, self.rev(p))
1076
1072
1077 # build deltas
1073 # build deltas
1078 for d in xrange(0, len(revs) - 1):
1074 for d in xrange(0, len(revs) - 1):
1079 a, b = revs[d], revs[d + 1]
1075 a, b = revs[d], revs[d + 1]
1080 nb = self.node(b)
1076 nb = self.node(b)
1081
1077
1082 if infocollect is not None:
1078 if infocollect is not None:
1083 infocollect(nb)
1079 infocollect(nb)
1084
1080
1085 d = self.revdiff(a, b)
1081 d = self.revdiff(a, b)
1086 p = self.parents(nb)
1082 p = self.parents(nb)
1087 meta = nb + p[0] + p[1] + lookup(nb)
1083 meta = nb + p[0] + p[1] + lookup(nb)
1088 yield changegroup.genchunk("%s%s" % (meta, d))
1084 yield changegroup.genchunk("%s%s" % (meta, d))
1089
1085
1090 yield changegroup.closechunk()
1086 yield changegroup.closechunk()
1091
1087
1092 def addgroup(self, revs, linkmapper, transaction, unique=0):
1088 def addgroup(self, revs, linkmapper, transaction, unique=0):
1093 """
1089 """
1094 add a delta group
1090 add a delta group
1095
1091
1096 given a set of deltas, add them to the revision log. the
1092 given a set of deltas, add them to the revision log. the
1097 first delta is against its parent, which should be in our
1093 first delta is against its parent, which should be in our
1098 log, the rest are against the previous delta.
1094 log, the rest are against the previous delta.
1099 """
1095 """
1100
1096
1101 #track the base of the current delta log
1097 #track the base of the current delta log
1102 r = self.count()
1098 r = self.count()
1103 t = r - 1
1099 t = r - 1
1104 node = None
1100 node = None
1105
1101
1106 base = prev = -1
1102 base = prev = -1
1107 start = end = textlen = 0
1103 start = end = textlen = 0
1108 if r:
1104 if r:
1109 end = self.end(t)
1105 end = self.end(t)
1110
1106
1111 ifh = self.opener(self.indexfile, "a+")
1107 ifh = self.opener(self.indexfile, "a+")
1112 ifh.seek(0, 2)
1108 ifh.seek(0, 2)
1113 transaction.add(self.indexfile, ifh.tell(), self.count())
1109 transaction.add(self.indexfile, ifh.tell(), self.count())
1114 if self.inlinedata():
1110 if self.inlinedata():
1115 dfh = None
1111 dfh = None
1116 else:
1112 else:
1117 transaction.add(self.datafile, end)
1113 transaction.add(self.datafile, end)
1118 dfh = self.opener(self.datafile, "a")
1114 dfh = self.opener(self.datafile, "a")
1119
1115
1120 # loop through our set of deltas
1116 # loop through our set of deltas
1121 chain = None
1117 chain = None
1122 for chunk in revs:
1118 for chunk in revs:
1123 node, p1, p2, cs = struct.unpack("20s20s20s20s", chunk[:80])
1119 node, p1, p2, cs = struct.unpack("20s20s20s20s", chunk[:80])
1124 link = linkmapper(cs)
1120 link = linkmapper(cs)
1125 if node in self.nodemap:
1121 if node in self.nodemap:
1126 # this can happen if two branches make the same change
1122 # this can happen if two branches make the same change
1127 # if unique:
1123 # if unique:
1128 # raise RevlogError(_("already have %s") % hex(node[:4]))
1124 # raise RevlogError(_("already have %s") % hex(node[:4]))
1129 chain = node
1125 chain = node
1130 continue
1126 continue
1131 delta = chunk[80:]
1127 delta = chunk[80:]
1132
1128
1133 for p in (p1, p2):
1129 for p in (p1, p2):
1134 if not p in self.nodemap:
1130 if not p in self.nodemap:
1135 raise RevlogError(_("unknown parent %s") % short(p))
1131 raise RevlogError(_("unknown parent %s") % short(p))
1136
1132
1137 if not chain:
1133 if not chain:
1138 # retrieve the parent revision of the delta chain
1134 # retrieve the parent revision of the delta chain
1139 chain = p1
1135 chain = p1
1140 if not chain in self.nodemap:
1136 if not chain in self.nodemap:
1141 raise RevlogError(_("unknown base %s") % short(chain[:4]))
1137 raise RevlogError(_("unknown base %s") % short(chain[:4]))
1142
1138
1143 # full versions are inserted when the needed deltas become
1139 # full versions are inserted when the needed deltas become
1144 # comparable to the uncompressed text or when the previous
1140 # comparable to the uncompressed text or when the previous
1145 # version is not the one we have a delta against. We use
1141 # version is not the one we have a delta against. We use
1146 # the size of the previous full rev as a proxy for the
1142 # the size of the previous full rev as a proxy for the
1147 # current size.
1143 # current size.
1148
1144
1149 if chain == prev:
1145 if chain == prev:
1150 tempd = compress(delta)
1146 tempd = compress(delta)
1151 cdelta = tempd[0] + tempd[1]
1147 cdelta = tempd[0] + tempd[1]
1152 textlen = mdiff.patchedsize(textlen, delta)
1148 textlen = mdiff.patchedsize(textlen, delta)
1153
1149
1154 if chain != prev or (end - start + len(cdelta)) > textlen * 2:
1150 if chain != prev or (end - start + len(cdelta)) > textlen * 2:
1155 # flush our writes here so we can read it in revision
1151 # flush our writes here so we can read it in revision
1156 if dfh:
1152 if dfh:
1157 dfh.flush()
1153 dfh.flush()
1158 ifh.flush()
1154 ifh.flush()
1159 text = self.revision(chain)
1155 text = self.revision(chain)
1160 text = self.patches(text, [delta])
1156 text = self.patches(text, [delta])
1161 chk = self._addrevision(text, transaction, link, p1, p2, None,
1157 chk = self._addrevision(text, transaction, link, p1, p2, None,
1162 ifh, dfh)
1158 ifh, dfh)
1163 if not dfh and not self.inlinedata():
1159 if not dfh and not self.inlinedata():
1164 # addrevision switched from inline to conventional
1160 # addrevision switched from inline to conventional
1165 # reopen the index
1161 # reopen the index
1166 dfh = self.opener(self.datafile, "a")
1162 dfh = self.opener(self.datafile, "a")
1167 ifh = self.opener(self.indexfile, "a")
1163 ifh = self.opener(self.indexfile, "a")
1168 if chk != node:
1164 if chk != node:
1169 raise RevlogError(_("consistency error adding group"))
1165 raise RevlogError(_("consistency error adding group"))
1170 textlen = len(text)
1166 textlen = len(text)
1171 else:
1167 else:
1172 if self.version == REVLOGV0:
1168 if self.version == REVLOGV0:
1173 e = (end, len(cdelta), base, link, p1, p2, node)
1169 e = (end, len(cdelta), base, link, p1, p2, node)
1174 else:
1170 else:
1175 e = (self.offset_type(end, 0), len(cdelta), textlen, base,
1171 e = (self.offset_type(end, 0), len(cdelta), textlen, base,
1176 link, self.rev(p1), self.rev(p2), node)
1172 link, self.rev(p1), self.rev(p2), node)
1177 self.index.append(e)
1173 self.index.append(e)
1178 self.nodemap[node] = r
1174 self.nodemap[node] = r
1179 if self.inlinedata():
1175 if self.inlinedata():
1180 ifh.write(struct.pack(self.indexformat, *e))
1176 ifh.write(struct.pack(self.indexformat, *e))
1181 ifh.write(cdelta)
1177 ifh.write(cdelta)
1182 self.checkinlinesize(transaction, ifh)
1178 self.checkinlinesize(transaction, ifh)
1183 if not self.inlinedata():
1179 if not self.inlinedata():
1184 dfh = self.opener(self.datafile, "a")
1180 dfh = self.opener(self.datafile, "a")
1185 ifh = self.opener(self.indexfile, "a")
1181 ifh = self.opener(self.indexfile, "a")
1186 else:
1182 else:
1187 dfh.write(cdelta)
1183 dfh.write(cdelta)
1188 ifh.write(struct.pack(self.indexformat, *e))
1184 ifh.write(struct.pack(self.indexformat, *e))
1189
1185
1190 t, r, chain, prev = r, r + 1, node, node
1186 t, r, chain, prev = r, r + 1, node, node
1191 base = self.base(t)
1187 base = self.base(t)
1192 start = self.start(base)
1188 start = self.start(base)
1193 end = self.end(t)
1189 end = self.end(t)
1194
1190
1195 return node
1191 return node
1196
1192
1197 def strip(self, rev, minlink):
1193 def strip(self, rev, minlink):
1198 if self.count() == 0 or rev >= self.count():
1194 if self.count() == 0 or rev >= self.count():
1199 return
1195 return
1200
1196
1201 if isinstance(self.index, lazyindex):
1197 if isinstance(self.index, lazyindex):
1202 self.loadindexmap()
1198 self.loadindexmap()
1203
1199
1204 # When stripping away a revision, we need to make sure it
1200 # When stripping away a revision, we need to make sure it
1205 # does not actually belong to an older changeset.
1201 # does not actually belong to an older changeset.
1206 # The minlink parameter defines the oldest revision
1202 # The minlink parameter defines the oldest revision
1207 # we're allowed to strip away.
1203 # we're allowed to strip away.
1208 while minlink > self.index[rev][-4]:
1204 while minlink > self.index[rev][-4]:
1209 rev += 1
1205 rev += 1
1210 if rev >= self.count():
1206 if rev >= self.count():
1211 return
1207 return
1212
1208
1213 # first truncate the files on disk
1209 # first truncate the files on disk
1214 end = self.start(rev)
1210 end = self.start(rev)
1215 if not self.inlinedata():
1211 if not self.inlinedata():
1216 df = self.opener(self.datafile, "a")
1212 df = self.opener(self.datafile, "a")
1217 df.truncate(end)
1213 df.truncate(end)
1218 end = rev * struct.calcsize(self.indexformat)
1214 end = rev * struct.calcsize(self.indexformat)
1219 else:
1215 else:
1220 end += rev * struct.calcsize(self.indexformat)
1216 end += rev * struct.calcsize(self.indexformat)
1221
1217
1222 indexf = self.opener(self.indexfile, "a")
1218 indexf = self.opener(self.indexfile, "a")
1223 indexf.truncate(end)
1219 indexf.truncate(end)
1224
1220
1225 # then reset internal state in memory to forget those revisions
1221 # then reset internal state in memory to forget those revisions
1226 self.cache = None
1222 self.cache = None
1227 self.chunkcache = None
1223 self.chunkcache = None
1228 for x in xrange(rev, self.count()):
1224 for x in xrange(rev, self.count()):
1229 del self.nodemap[self.node(x)]
1225 del self.nodemap[self.node(x)]
1230
1226
1231 del self.index[rev:]
1227 del self.index[rev:]
1232
1228
1233 def checksize(self):
1229 def checksize(self):
1234 expected = 0
1230 expected = 0
1235 if self.count():
1231 if self.count():
1236 expected = self.end(self.count() - 1)
1232 expected = self.end(self.count() - 1)
1237
1233
1238 try:
1234 try:
1239 f = self.opener(self.datafile)
1235 f = self.opener(self.datafile)
1240 f.seek(0, 2)
1236 f.seek(0, 2)
1241 actual = f.tell()
1237 actual = f.tell()
1242 dd = actual - expected
1238 dd = actual - expected
1243 except IOError, inst:
1239 except IOError, inst:
1244 if inst.errno != errno.ENOENT:
1240 if inst.errno != errno.ENOENT:
1245 raise
1241 raise
1246 dd = 0
1242 dd = 0
1247
1243
1248 try:
1244 try:
1249 f = self.opener(self.indexfile)
1245 f = self.opener(self.indexfile)
1250 f.seek(0, 2)
1246 f.seek(0, 2)
1251 actual = f.tell()
1247 actual = f.tell()
1252 s = struct.calcsize(self.indexformat)
1248 s = struct.calcsize(self.indexformat)
1253 i = actual / s
1249 i = actual / s
1254 di = actual - (i * s)
1250 di = actual - (i * s)
1255 if self.inlinedata():
1251 if self.inlinedata():
1256 databytes = 0
1252 databytes = 0
1257 for r in xrange(self.count()):
1253 for r in xrange(self.count()):
1258 databytes += self.length(r)
1254 databytes += self.length(r)
1259 dd = 0
1255 dd = 0
1260 di = actual - self.count() * s - databytes
1256 di = actual - self.count() * s - databytes
1261 except IOError, inst:
1257 except IOError, inst:
1262 if inst.errno != errno.ENOENT:
1258 if inst.errno != errno.ENOENT:
1263 raise
1259 raise
1264 di = 0
1260 di = 0
1265
1261
1266 return (dd, di)
1262 return (dd, di)
1267
1263
1268
1264
General Comments 0
You need to be logged in to leave comments. Login now