##// END OF EJS Templates
merge with stable
Matt Mackall -
r20965:774ff56c merge default
parent child Browse files
Show More
@@ -1,1432 +1,1437 b''
1 # revlog.py - storage back-end for mercurial
1 # revlog.py - storage back-end for mercurial
2 #
2 #
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2 or any later version.
6 # GNU General Public License version 2 or any later version.
7
7
8 """Storage back-end for Mercurial.
8 """Storage back-end for Mercurial.
9
9
10 This provides efficient delta storage with O(1) retrieve and append
10 This provides efficient delta storage with O(1) retrieve and append
11 and O(changes) merge between branches.
11 and O(changes) merge between branches.
12 """
12 """
13
13
14 # import stuff from node for others to import from revlog
14 # import stuff from node for others to import from revlog
15 from node import bin, hex, nullid, nullrev
15 from node import bin, hex, nullid, nullrev
16 from i18n import _
16 from i18n import _
17 import ancestor, mdiff, parsers, error, util, templatefilters
17 import ancestor, mdiff, parsers, error, util, templatefilters
18 import struct, zlib, errno
18 import struct, zlib, errno
19
19
20 _pack = struct.pack
20 _pack = struct.pack
21 _unpack = struct.unpack
21 _unpack = struct.unpack
22 _compress = zlib.compress
22 _compress = zlib.compress
23 _decompress = zlib.decompress
23 _decompress = zlib.decompress
24 _sha = util.sha1
24 _sha = util.sha1
25
25
26 # revlog header flags
26 # revlog header flags
27 REVLOGV0 = 0
27 REVLOGV0 = 0
28 REVLOGNG = 1
28 REVLOGNG = 1
29 REVLOGNGINLINEDATA = (1 << 16)
29 REVLOGNGINLINEDATA = (1 << 16)
30 REVLOGGENERALDELTA = (1 << 17)
30 REVLOGGENERALDELTA = (1 << 17)
31 REVLOG_DEFAULT_FLAGS = REVLOGNGINLINEDATA
31 REVLOG_DEFAULT_FLAGS = REVLOGNGINLINEDATA
32 REVLOG_DEFAULT_FORMAT = REVLOGNG
32 REVLOG_DEFAULT_FORMAT = REVLOGNG
33 REVLOG_DEFAULT_VERSION = REVLOG_DEFAULT_FORMAT | REVLOG_DEFAULT_FLAGS
33 REVLOG_DEFAULT_VERSION = REVLOG_DEFAULT_FORMAT | REVLOG_DEFAULT_FLAGS
34 REVLOGNG_FLAGS = REVLOGNGINLINEDATA | REVLOGGENERALDELTA
34 REVLOGNG_FLAGS = REVLOGNGINLINEDATA | REVLOGGENERALDELTA
35
35
36 # revlog index flags
36 # revlog index flags
37 REVIDX_KNOWN_FLAGS = 0
37 REVIDX_KNOWN_FLAGS = 0
38
38
39 # max size of revlog with inline data
39 # max size of revlog with inline data
40 _maxinline = 131072
40 _maxinline = 131072
41 _chunksize = 1048576
41 _chunksize = 1048576
42
42
43 RevlogError = error.RevlogError
43 RevlogError = error.RevlogError
44 LookupError = error.LookupError
44 LookupError = error.LookupError
45
45
46 def getoffset(q):
46 def getoffset(q):
47 return int(q >> 16)
47 return int(q >> 16)
48
48
49 def gettype(q):
49 def gettype(q):
50 return int(q & 0xFFFF)
50 return int(q & 0xFFFF)
51
51
52 def offset_type(offset, type):
52 def offset_type(offset, type):
53 return long(long(offset) << 16 | type)
53 return long(long(offset) << 16 | type)
54
54
55 nullhash = _sha(nullid)
55 nullhash = _sha(nullid)
56
56
57 def hash(text, p1, p2):
57 def hash(text, p1, p2):
58 """generate a hash from the given text and its parent hashes
58 """generate a hash from the given text and its parent hashes
59
59
60 This hash combines both the current file contents and its history
60 This hash combines both the current file contents and its history
61 in a manner that makes it easy to distinguish nodes with the same
61 in a manner that makes it easy to distinguish nodes with the same
62 content in the revision graph.
62 content in the revision graph.
63 """
63 """
64 # As of now, if one of the parent node is null, p2 is null
64 # As of now, if one of the parent node is null, p2 is null
65 if p2 == nullid:
65 if p2 == nullid:
66 # deep copy of a hash is faster than creating one
66 # deep copy of a hash is faster than creating one
67 s = nullhash.copy()
67 s = nullhash.copy()
68 s.update(p1)
68 s.update(p1)
69 else:
69 else:
70 # none of the parent nodes are nullid
70 # none of the parent nodes are nullid
71 l = [p1, p2]
71 l = [p1, p2]
72 l.sort()
72 l.sort()
73 s = _sha(l[0])
73 s = _sha(l[0])
74 s.update(l[1])
74 s.update(l[1])
75 s.update(text)
75 s.update(text)
76 return s.digest()
76 return s.digest()
77
77
78 def decompress(bin):
78 def decompress(bin):
79 """ decompress the given input """
79 """ decompress the given input """
80 if not bin:
80 if not bin:
81 return bin
81 return bin
82 t = bin[0]
82 t = bin[0]
83 if t == '\0':
83 if t == '\0':
84 return bin
84 return bin
85 if t == 'x':
85 if t == 'x':
86 try:
86 try:
87 return _decompress(bin)
87 return _decompress(bin)
88 except zlib.error, e:
88 except zlib.error, e:
89 raise RevlogError(_("revlog decompress error: %s") % str(e))
89 raise RevlogError(_("revlog decompress error: %s") % str(e))
90 if t == 'u':
90 if t == 'u':
91 return bin[1:]
91 return bin[1:]
92 raise RevlogError(_("unknown compression type %r") % t)
92 raise RevlogError(_("unknown compression type %r") % t)
93
93
94 # index v0:
94 # index v0:
95 # 4 bytes: offset
95 # 4 bytes: offset
96 # 4 bytes: compressed length
96 # 4 bytes: compressed length
97 # 4 bytes: base rev
97 # 4 bytes: base rev
98 # 4 bytes: link rev
98 # 4 bytes: link rev
99 # 32 bytes: parent 1 nodeid
99 # 32 bytes: parent 1 nodeid
100 # 32 bytes: parent 2 nodeid
100 # 32 bytes: parent 2 nodeid
101 # 32 bytes: nodeid
101 # 32 bytes: nodeid
102 indexformatv0 = ">4l20s20s20s"
102 indexformatv0 = ">4l20s20s20s"
103 v0shaoffset = 56
103 v0shaoffset = 56
104
104
105 class revlogoldio(object):
105 class revlogoldio(object):
106 def __init__(self):
106 def __init__(self):
107 self.size = struct.calcsize(indexformatv0)
107 self.size = struct.calcsize(indexformatv0)
108
108
109 def parseindex(self, data, inline):
109 def parseindex(self, data, inline):
110 s = self.size
110 s = self.size
111 index = []
111 index = []
112 nodemap = {nullid: nullrev}
112 nodemap = {nullid: nullrev}
113 n = off = 0
113 n = off = 0
114 l = len(data)
114 l = len(data)
115 while off + s <= l:
115 while off + s <= l:
116 cur = data[off:off + s]
116 cur = data[off:off + s]
117 off += s
117 off += s
118 e = _unpack(indexformatv0, cur)
118 e = _unpack(indexformatv0, cur)
119 # transform to revlogv1 format
119 # transform to revlogv1 format
120 e2 = (offset_type(e[0], 0), e[1], -1, e[2], e[3],
120 e2 = (offset_type(e[0], 0), e[1], -1, e[2], e[3],
121 nodemap.get(e[4], nullrev), nodemap.get(e[5], nullrev), e[6])
121 nodemap.get(e[4], nullrev), nodemap.get(e[5], nullrev), e[6])
122 index.append(e2)
122 index.append(e2)
123 nodemap[e[6]] = n
123 nodemap[e[6]] = n
124 n += 1
124 n += 1
125
125
126 # add the magic null revision at -1
126 # add the magic null revision at -1
127 index.append((0, 0, 0, -1, -1, -1, -1, nullid))
127 index.append((0, 0, 0, -1, -1, -1, -1, nullid))
128
128
129 return index, nodemap, None
129 return index, nodemap, None
130
130
131 def packentry(self, entry, node, version, rev):
131 def packentry(self, entry, node, version, rev):
132 if gettype(entry[0]):
132 if gettype(entry[0]):
133 raise RevlogError(_("index entry flags need RevlogNG"))
133 raise RevlogError(_("index entry flags need RevlogNG"))
134 e2 = (getoffset(entry[0]), entry[1], entry[3], entry[4],
134 e2 = (getoffset(entry[0]), entry[1], entry[3], entry[4],
135 node(entry[5]), node(entry[6]), entry[7])
135 node(entry[5]), node(entry[6]), entry[7])
136 return _pack(indexformatv0, *e2)
136 return _pack(indexformatv0, *e2)
137
137
138 # index ng:
138 # index ng:
139 # 6 bytes: offset
139 # 6 bytes: offset
140 # 2 bytes: flags
140 # 2 bytes: flags
141 # 4 bytes: compressed length
141 # 4 bytes: compressed length
142 # 4 bytes: uncompressed length
142 # 4 bytes: uncompressed length
143 # 4 bytes: base rev
143 # 4 bytes: base rev
144 # 4 bytes: link rev
144 # 4 bytes: link rev
145 # 4 bytes: parent 1 rev
145 # 4 bytes: parent 1 rev
146 # 4 bytes: parent 2 rev
146 # 4 bytes: parent 2 rev
147 # 32 bytes: nodeid
147 # 32 bytes: nodeid
148 indexformatng = ">Qiiiiii20s12x"
148 indexformatng = ">Qiiiiii20s12x"
149 ngshaoffset = 32
149 ngshaoffset = 32
150 versionformat = ">I"
150 versionformat = ">I"
151
151
152 class revlogio(object):
152 class revlogio(object):
153 def __init__(self):
153 def __init__(self):
154 self.size = struct.calcsize(indexformatng)
154 self.size = struct.calcsize(indexformatng)
155
155
156 def parseindex(self, data, inline):
156 def parseindex(self, data, inline):
157 # call the C implementation to parse the index data
157 # call the C implementation to parse the index data
158 index, cache = parsers.parse_index2(data, inline)
158 index, cache = parsers.parse_index2(data, inline)
159 return index, getattr(index, 'nodemap', None), cache
159 return index, getattr(index, 'nodemap', None), cache
160
160
161 def packentry(self, entry, node, version, rev):
161 def packentry(self, entry, node, version, rev):
162 p = _pack(indexformatng, *entry)
162 p = _pack(indexformatng, *entry)
163 if rev == 0:
163 if rev == 0:
164 p = _pack(versionformat, version) + p[4:]
164 p = _pack(versionformat, version) + p[4:]
165 return p
165 return p
166
166
167 class revlog(object):
167 class revlog(object):
168 """
168 """
169 the underlying revision storage object
169 the underlying revision storage object
170
170
171 A revlog consists of two parts, an index and the revision data.
171 A revlog consists of two parts, an index and the revision data.
172
172
173 The index is a file with a fixed record size containing
173 The index is a file with a fixed record size containing
174 information on each revision, including its nodeid (hash), the
174 information on each revision, including its nodeid (hash), the
175 nodeids of its parents, the position and offset of its data within
175 nodeids of its parents, the position and offset of its data within
176 the data file, and the revision it's based on. Finally, each entry
176 the data file, and the revision it's based on. Finally, each entry
177 contains a linkrev entry that can serve as a pointer to external
177 contains a linkrev entry that can serve as a pointer to external
178 data.
178 data.
179
179
180 The revision data itself is a linear collection of data chunks.
180 The revision data itself is a linear collection of data chunks.
181 Each chunk represents a revision and is usually represented as a
181 Each chunk represents a revision and is usually represented as a
182 delta against the previous chunk. To bound lookup time, runs of
182 delta against the previous chunk. To bound lookup time, runs of
183 deltas are limited to about 2 times the length of the original
183 deltas are limited to about 2 times the length of the original
184 version data. This makes retrieval of a version proportional to
184 version data. This makes retrieval of a version proportional to
185 its size, or O(1) relative to the number of revisions.
185 its size, or O(1) relative to the number of revisions.
186
186
187 Both pieces of the revlog are written to in an append-only
187 Both pieces of the revlog are written to in an append-only
188 fashion, which means we never need to rewrite a file to insert or
188 fashion, which means we never need to rewrite a file to insert or
189 remove data, and can use some simple techniques to avoid the need
189 remove data, and can use some simple techniques to avoid the need
190 for locking while reading.
190 for locking while reading.
191 """
191 """
192 def __init__(self, opener, indexfile):
192 def __init__(self, opener, indexfile):
193 """
193 """
194 create a revlog object
194 create a revlog object
195
195
196 opener is a function that abstracts the file opening operation
196 opener is a function that abstracts the file opening operation
197 and can be used to implement COW semantics or the like.
197 and can be used to implement COW semantics or the like.
198 """
198 """
199 self.indexfile = indexfile
199 self.indexfile = indexfile
200 self.datafile = indexfile[:-2] + ".d"
200 self.datafile = indexfile[:-2] + ".d"
201 self.opener = opener
201 self.opener = opener
202 self._cache = None
202 self._cache = None
203 self._basecache = None
203 self._basecache = None
204 self._chunkcache = (0, '')
204 self._chunkcache = (0, '')
205 self._chunkcachesize = 65536
205 self._chunkcachesize = 65536
206 self.index = []
206 self.index = []
207 self._pcache = {}
207 self._pcache = {}
208 self._nodecache = {nullid: nullrev}
208 self._nodecache = {nullid: nullrev}
209 self._nodepos = None
209 self._nodepos = None
210
210
211 v = REVLOG_DEFAULT_VERSION
211 v = REVLOG_DEFAULT_VERSION
212 opts = getattr(opener, 'options', None)
212 opts = getattr(opener, 'options', None)
213 if opts is not None:
213 if opts is not None:
214 if 'revlogv1' in opts:
214 if 'revlogv1' in opts:
215 if 'generaldelta' in opts:
215 if 'generaldelta' in opts:
216 v |= REVLOGGENERALDELTA
216 v |= REVLOGGENERALDELTA
217 else:
217 else:
218 v = 0
218 v = 0
219 if 'chunkcachesize' in opts:
219 if 'chunkcachesize' in opts:
220 self._chunkcachesize = opts['chunkcachesize']
220 self._chunkcachesize = opts['chunkcachesize']
221
221
222 if self._chunkcachesize <= 0:
222 if self._chunkcachesize <= 0:
223 raise RevlogError(_('revlog chunk cache size %r is not greater '
223 raise RevlogError(_('revlog chunk cache size %r is not greater '
224 'than 0') % self._chunkcachesize)
224 'than 0') % self._chunkcachesize)
225 elif self._chunkcachesize & (self._chunkcachesize - 1):
225 elif self._chunkcachesize & (self._chunkcachesize - 1):
226 raise RevlogError(_('revlog chunk cache size %r is not a power '
226 raise RevlogError(_('revlog chunk cache size %r is not a power '
227 'of 2') % self._chunkcachesize)
227 'of 2') % self._chunkcachesize)
228
228
229 i = ''
229 i = ''
230 self._initempty = True
230 self._initempty = True
231 try:
231 try:
232 f = self.opener(self.indexfile)
232 f = self.opener(self.indexfile)
233 i = f.read()
233 i = f.read()
234 f.close()
234 f.close()
235 if len(i) > 0:
235 if len(i) > 0:
236 v = struct.unpack(versionformat, i[:4])[0]
236 v = struct.unpack(versionformat, i[:4])[0]
237 self._initempty = False
237 self._initempty = False
238 except IOError, inst:
238 except IOError, inst:
239 if inst.errno != errno.ENOENT:
239 if inst.errno != errno.ENOENT:
240 raise
240 raise
241
241
242 self.version = v
242 self.version = v
243 self._inline = v & REVLOGNGINLINEDATA
243 self._inline = v & REVLOGNGINLINEDATA
244 self._generaldelta = v & REVLOGGENERALDELTA
244 self._generaldelta = v & REVLOGGENERALDELTA
245 flags = v & ~0xFFFF
245 flags = v & ~0xFFFF
246 fmt = v & 0xFFFF
246 fmt = v & 0xFFFF
247 if fmt == REVLOGV0 and flags:
247 if fmt == REVLOGV0 and flags:
248 raise RevlogError(_("index %s unknown flags %#04x for format v0")
248 raise RevlogError(_("index %s unknown flags %#04x for format v0")
249 % (self.indexfile, flags >> 16))
249 % (self.indexfile, flags >> 16))
250 elif fmt == REVLOGNG and flags & ~REVLOGNG_FLAGS:
250 elif fmt == REVLOGNG and flags & ~REVLOGNG_FLAGS:
251 raise RevlogError(_("index %s unknown flags %#04x for revlogng")
251 raise RevlogError(_("index %s unknown flags %#04x for revlogng")
252 % (self.indexfile, flags >> 16))
252 % (self.indexfile, flags >> 16))
253 elif fmt > REVLOGNG:
253 elif fmt > REVLOGNG:
254 raise RevlogError(_("index %s unknown format %d")
254 raise RevlogError(_("index %s unknown format %d")
255 % (self.indexfile, fmt))
255 % (self.indexfile, fmt))
256
256
257 self._io = revlogio()
257 self._io = revlogio()
258 if self.version == REVLOGV0:
258 if self.version == REVLOGV0:
259 self._io = revlogoldio()
259 self._io = revlogoldio()
260 try:
260 try:
261 d = self._io.parseindex(i, self._inline)
261 d = self._io.parseindex(i, self._inline)
262 except (ValueError, IndexError):
262 except (ValueError, IndexError):
263 raise RevlogError(_("index %s is corrupted") % (self.indexfile))
263 raise RevlogError(_("index %s is corrupted") % (self.indexfile))
264 self.index, nodemap, self._chunkcache = d
264 self.index, nodemap, self._chunkcache = d
265 if nodemap is not None:
265 if nodemap is not None:
266 self.nodemap = self._nodecache = nodemap
266 self.nodemap = self._nodecache = nodemap
267 if not self._chunkcache:
267 if not self._chunkcache:
268 self._chunkclear()
268 self._chunkclear()
269
269
270 def tip(self):
270 def tip(self):
271 return self.node(len(self.index) - 2)
271 return self.node(len(self.index) - 2)
272 def __len__(self):
272 def __len__(self):
273 return len(self.index) - 1
273 return len(self.index) - 1
274 def __iter__(self):
274 def __iter__(self):
275 return iter(xrange(len(self)))
275 return iter(xrange(len(self)))
276 def revs(self, start=0, stop=None):
276 def revs(self, start=0, stop=None):
277 """iterate over all rev in this revlog (from start to stop)"""
277 """iterate over all rev in this revlog (from start to stop)"""
278 step = 1
278 step = 1
279 if stop is not None:
279 if stop is not None:
280 if start > stop:
280 if start > stop:
281 step = -1
281 step = -1
282 stop += step
282 stop += step
283 else:
283 else:
284 stop = len(self)
284 stop = len(self)
285 return xrange(start, stop, step)
285 return xrange(start, stop, step)
286
286
287 @util.propertycache
287 @util.propertycache
288 def nodemap(self):
288 def nodemap(self):
289 self.rev(self.node(0))
289 self.rev(self.node(0))
290 return self._nodecache
290 return self._nodecache
291
291
292 def hasnode(self, node):
292 def hasnode(self, node):
293 try:
293 try:
294 self.rev(node)
294 self.rev(node)
295 return True
295 return True
296 except KeyError:
296 except KeyError:
297 return False
297 return False
298
298
299 def clearcaches(self):
299 def clearcaches(self):
300 try:
300 try:
301 self._nodecache.clearcaches()
301 self._nodecache.clearcaches()
302 except AttributeError:
302 except AttributeError:
303 self._nodecache = {nullid: nullrev}
303 self._nodecache = {nullid: nullrev}
304 self._nodepos = None
304 self._nodepos = None
305
305
306 def rev(self, node):
306 def rev(self, node):
307 try:
307 try:
308 return self._nodecache[node]
308 return self._nodecache[node]
309 except RevlogError:
309 except RevlogError:
310 # parsers.c radix tree lookup failed
310 # parsers.c radix tree lookup failed
311 raise LookupError(node, self.indexfile, _('no node'))
311 raise LookupError(node, self.indexfile, _('no node'))
312 except KeyError:
312 except KeyError:
313 # pure python cache lookup failed
313 # pure python cache lookup failed
314 n = self._nodecache
314 n = self._nodecache
315 i = self.index
315 i = self.index
316 p = self._nodepos
316 p = self._nodepos
317 if p is None:
317 if p is None:
318 p = len(i) - 2
318 p = len(i) - 2
319 for r in xrange(p, -1, -1):
319 for r in xrange(p, -1, -1):
320 v = i[r][7]
320 v = i[r][7]
321 n[v] = r
321 n[v] = r
322 if v == node:
322 if v == node:
323 self._nodepos = r - 1
323 self._nodepos = r - 1
324 return r
324 return r
325 raise LookupError(node, self.indexfile, _('no node'))
325 raise LookupError(node, self.indexfile, _('no node'))
326
326
327 def node(self, rev):
327 def node(self, rev):
328 return self.index[rev][7]
328 return self.index[rev][7]
329 def linkrev(self, rev):
329 def linkrev(self, rev):
330 return self.index[rev][4]
330 return self.index[rev][4]
331 def parents(self, node):
331 def parents(self, node):
332 i = self.index
332 i = self.index
333 d = i[self.rev(node)]
333 d = i[self.rev(node)]
334 return i[d[5]][7], i[d[6]][7] # map revisions to nodes inline
334 return i[d[5]][7], i[d[6]][7] # map revisions to nodes inline
335 def parentrevs(self, rev):
335 def parentrevs(self, rev):
336 return self.index[rev][5:7]
336 return self.index[rev][5:7]
337 def start(self, rev):
337 def start(self, rev):
338 return int(self.index[rev][0] >> 16)
338 return int(self.index[rev][0] >> 16)
339 def end(self, rev):
339 def end(self, rev):
340 return self.start(rev) + self.length(rev)
340 return self.start(rev) + self.length(rev)
341 def length(self, rev):
341 def length(self, rev):
342 return self.index[rev][1]
342 return self.index[rev][1]
343 def chainbase(self, rev):
343 def chainbase(self, rev):
344 index = self.index
344 index = self.index
345 base = index[rev][3]
345 base = index[rev][3]
346 while base != rev:
346 while base != rev:
347 rev = base
347 rev = base
348 base = index[rev][3]
348 base = index[rev][3]
349 return base
349 return base
350 def flags(self, rev):
350 def flags(self, rev):
351 return self.index[rev][0] & 0xFFFF
351 return self.index[rev][0] & 0xFFFF
352 def rawsize(self, rev):
352 def rawsize(self, rev):
353 """return the length of the uncompressed text for a given revision"""
353 """return the length of the uncompressed text for a given revision"""
354 l = self.index[rev][2]
354 l = self.index[rev][2]
355 if l >= 0:
355 if l >= 0:
356 return l
356 return l
357
357
358 t = self.revision(self.node(rev))
358 t = self.revision(self.node(rev))
359 return len(t)
359 return len(t)
360 size = rawsize
360 size = rawsize
361
361
362 def ancestors(self, revs, stoprev=0, inclusive=False):
362 def ancestors(self, revs, stoprev=0, inclusive=False):
363 """Generate the ancestors of 'revs' in reverse topological order.
363 """Generate the ancestors of 'revs' in reverse topological order.
364 Does not generate revs lower than stoprev.
364 Does not generate revs lower than stoprev.
365
365
366 See the documentation for ancestor.lazyancestors for more details."""
366 See the documentation for ancestor.lazyancestors for more details."""
367
367
368 return ancestor.lazyancestors(self, revs, stoprev=stoprev,
368 return ancestor.lazyancestors(self, revs, stoprev=stoprev,
369 inclusive=inclusive)
369 inclusive=inclusive)
370
370
371 def descendants(self, revs):
371 def descendants(self, revs):
372 """Generate the descendants of 'revs' in revision order.
372 """Generate the descendants of 'revs' in revision order.
373
373
374 Yield a sequence of revision numbers starting with a child of
374 Yield a sequence of revision numbers starting with a child of
375 some rev in revs, i.e., each revision is *not* considered a
375 some rev in revs, i.e., each revision is *not* considered a
376 descendant of itself. Results are ordered by revision number (a
376 descendant of itself. Results are ordered by revision number (a
377 topological sort)."""
377 topological sort)."""
378 first = min(revs)
378 first = min(revs)
379 if first == nullrev:
379 if first == nullrev:
380 for i in self:
380 for i in self:
381 yield i
381 yield i
382 return
382 return
383
383
384 seen = set(revs)
384 seen = set(revs)
385 for i in self.revs(start=first + 1):
385 for i in self.revs(start=first + 1):
386 for x in self.parentrevs(i):
386 for x in self.parentrevs(i):
387 if x != nullrev and x in seen:
387 if x != nullrev and x in seen:
388 seen.add(i)
388 seen.add(i)
389 yield i
389 yield i
390 break
390 break
391
391
392 def findcommonmissing(self, common=None, heads=None):
392 def findcommonmissing(self, common=None, heads=None):
393 """Return a tuple of the ancestors of common and the ancestors of heads
393 """Return a tuple of the ancestors of common and the ancestors of heads
394 that are not ancestors of common. In revset terminology, we return the
394 that are not ancestors of common. In revset terminology, we return the
395 tuple:
395 tuple:
396
396
397 ::common, (::heads) - (::common)
397 ::common, (::heads) - (::common)
398
398
399 The list is sorted by revision number, meaning it is
399 The list is sorted by revision number, meaning it is
400 topologically sorted.
400 topologically sorted.
401
401
402 'heads' and 'common' are both lists of node IDs. If heads is
402 'heads' and 'common' are both lists of node IDs. If heads is
403 not supplied, uses all of the revlog's heads. If common is not
403 not supplied, uses all of the revlog's heads. If common is not
404 supplied, uses nullid."""
404 supplied, uses nullid."""
405 if common is None:
405 if common is None:
406 common = [nullid]
406 common = [nullid]
407 if heads is None:
407 if heads is None:
408 heads = self.heads()
408 heads = self.heads()
409
409
410 common = [self.rev(n) for n in common]
410 common = [self.rev(n) for n in common]
411 heads = [self.rev(n) for n in heads]
411 heads = [self.rev(n) for n in heads]
412
412
413 # we want the ancestors, but inclusive
413 # we want the ancestors, but inclusive
414 class lazyset(object):
414 class lazyset(object):
415 def __init__(self, lazyvalues):
415 def __init__(self, lazyvalues):
416 self.addedvalues = set()
416 self.addedvalues = set()
417 self.lazyvalues = lazyvalues
417 self.lazyvalues = lazyvalues
418
418
419 def __contains__(self, value):
419 def __contains__(self, value):
420 return value in self.addedvalues or value in self.lazyvalues
420 return value in self.addedvalues or value in self.lazyvalues
421
421
422 def __iter__(self):
422 def __iter__(self):
423 added = self.addedvalues
423 added = self.addedvalues
424 for r in added:
424 for r in added:
425 yield r
425 yield r
426 for r in self.lazyvalues:
426 for r in self.lazyvalues:
427 if not r in added:
427 if not r in added:
428 yield r
428 yield r
429
429
430 def add(self, value):
430 def add(self, value):
431 self.addedvalues.add(value)
431 self.addedvalues.add(value)
432
432
433 def update(self, values):
433 def update(self, values):
434 self.addedvalues.update(values)
434 self.addedvalues.update(values)
435
435
436 has = lazyset(self.ancestors(common))
436 has = lazyset(self.ancestors(common))
437 has.add(nullrev)
437 has.add(nullrev)
438 has.update(common)
438 has.update(common)
439
439
440 # take all ancestors from heads that aren't in has
440 # take all ancestors from heads that aren't in has
441 missing = set()
441 missing = set()
442 visit = util.deque(r for r in heads if r not in has)
442 visit = util.deque(r for r in heads if r not in has)
443 while visit:
443 while visit:
444 r = visit.popleft()
444 r = visit.popleft()
445 if r in missing:
445 if r in missing:
446 continue
446 continue
447 else:
447 else:
448 missing.add(r)
448 missing.add(r)
449 for p in self.parentrevs(r):
449 for p in self.parentrevs(r):
450 if p not in has:
450 if p not in has:
451 visit.append(p)
451 visit.append(p)
452 missing = list(missing)
452 missing = list(missing)
453 missing.sort()
453 missing.sort()
454 return has, [self.node(r) for r in missing]
454 return has, [self.node(r) for r in missing]
455
455
456 def findmissingrevs(self, common=None, heads=None):
456 def findmissingrevs(self, common=None, heads=None):
457 """Return the revision numbers of the ancestors of heads that
457 """Return the revision numbers of the ancestors of heads that
458 are not ancestors of common.
458 are not ancestors of common.
459
459
460 More specifically, return a list of revision numbers corresponding to
460 More specifically, return a list of revision numbers corresponding to
461 nodes N such that every N satisfies the following constraints:
461 nodes N such that every N satisfies the following constraints:
462
462
463 1. N is an ancestor of some node in 'heads'
463 1. N is an ancestor of some node in 'heads'
464 2. N is not an ancestor of any node in 'common'
464 2. N is not an ancestor of any node in 'common'
465
465
466 The list is sorted by revision number, meaning it is
466 The list is sorted by revision number, meaning it is
467 topologically sorted.
467 topologically sorted.
468
468
469 'heads' and 'common' are both lists of revision numbers. If heads is
469 'heads' and 'common' are both lists of revision numbers. If heads is
470 not supplied, uses all of the revlog's heads. If common is not
470 not supplied, uses all of the revlog's heads. If common is not
471 supplied, uses nullid."""
471 supplied, uses nullid."""
472 if common is None:
472 if common is None:
473 common = [nullrev]
473 common = [nullrev]
474 if heads is None:
474 if heads is None:
475 heads = self.headrevs()
475 heads = self.headrevs()
476
476
477 return ancestor.missingancestors(heads, common, self.parentrevs)
477 return ancestor.missingancestors(heads, common, self.parentrevs)
478
478
479 def findmissing(self, common=None, heads=None):
479 def findmissing(self, common=None, heads=None):
480 """Return the ancestors of heads that are not ancestors of common.
480 """Return the ancestors of heads that are not ancestors of common.
481
481
482 More specifically, return a list of nodes N such that every N
482 More specifically, return a list of nodes N such that every N
483 satisfies the following constraints:
483 satisfies the following constraints:
484
484
485 1. N is an ancestor of some node in 'heads'
485 1. N is an ancestor of some node in 'heads'
486 2. N is not an ancestor of any node in 'common'
486 2. N is not an ancestor of any node in 'common'
487
487
488 The list is sorted by revision number, meaning it is
488 The list is sorted by revision number, meaning it is
489 topologically sorted.
489 topologically sorted.
490
490
491 'heads' and 'common' are both lists of node IDs. If heads is
491 'heads' and 'common' are both lists of node IDs. If heads is
492 not supplied, uses all of the revlog's heads. If common is not
492 not supplied, uses all of the revlog's heads. If common is not
493 supplied, uses nullid."""
493 supplied, uses nullid."""
494 if common is None:
494 if common is None:
495 common = [nullid]
495 common = [nullid]
496 if heads is None:
496 if heads is None:
497 heads = self.heads()
497 heads = self.heads()
498
498
499 common = [self.rev(n) for n in common]
499 common = [self.rev(n) for n in common]
500 heads = [self.rev(n) for n in heads]
500 heads = [self.rev(n) for n in heads]
501
501
502 return [self.node(r) for r in
502 return [self.node(r) for r in
503 ancestor.missingancestors(heads, common, self.parentrevs)]
503 ancestor.missingancestors(heads, common, self.parentrevs)]
504
504
505 def nodesbetween(self, roots=None, heads=None):
505 def nodesbetween(self, roots=None, heads=None):
506 """Return a topological path from 'roots' to 'heads'.
506 """Return a topological path from 'roots' to 'heads'.
507
507
508 Return a tuple (nodes, outroots, outheads) where 'nodes' is a
508 Return a tuple (nodes, outroots, outheads) where 'nodes' is a
509 topologically sorted list of all nodes N that satisfy both of
509 topologically sorted list of all nodes N that satisfy both of
510 these constraints:
510 these constraints:
511
511
512 1. N is a descendant of some node in 'roots'
512 1. N is a descendant of some node in 'roots'
513 2. N is an ancestor of some node in 'heads'
513 2. N is an ancestor of some node in 'heads'
514
514
515 Every node is considered to be both a descendant and an ancestor
515 Every node is considered to be both a descendant and an ancestor
516 of itself, so every reachable node in 'roots' and 'heads' will be
516 of itself, so every reachable node in 'roots' and 'heads' will be
517 included in 'nodes'.
517 included in 'nodes'.
518
518
519 'outroots' is the list of reachable nodes in 'roots', i.e., the
519 'outroots' is the list of reachable nodes in 'roots', i.e., the
520 subset of 'roots' that is returned in 'nodes'. Likewise,
520 subset of 'roots' that is returned in 'nodes'. Likewise,
521 'outheads' is the subset of 'heads' that is also in 'nodes'.
521 'outheads' is the subset of 'heads' that is also in 'nodes'.
522
522
523 'roots' and 'heads' are both lists of node IDs. If 'roots' is
523 'roots' and 'heads' are both lists of node IDs. If 'roots' is
524 unspecified, uses nullid as the only root. If 'heads' is
524 unspecified, uses nullid as the only root. If 'heads' is
525 unspecified, uses list of all of the revlog's heads."""
525 unspecified, uses list of all of the revlog's heads."""
526 nonodes = ([], [], [])
526 nonodes = ([], [], [])
527 if roots is not None:
527 if roots is not None:
528 roots = list(roots)
528 roots = list(roots)
529 if not roots:
529 if not roots:
530 return nonodes
530 return nonodes
531 lowestrev = min([self.rev(n) for n in roots])
531 lowestrev = min([self.rev(n) for n in roots])
532 else:
532 else:
533 roots = [nullid] # Everybody's a descendant of nullid
533 roots = [nullid] # Everybody's a descendant of nullid
534 lowestrev = nullrev
534 lowestrev = nullrev
535 if (lowestrev == nullrev) and (heads is None):
535 if (lowestrev == nullrev) and (heads is None):
536 # We want _all_ the nodes!
536 # We want _all_ the nodes!
537 return ([self.node(r) for r in self], [nullid], list(self.heads()))
537 return ([self.node(r) for r in self], [nullid], list(self.heads()))
538 if heads is None:
538 if heads is None:
539 # All nodes are ancestors, so the latest ancestor is the last
539 # All nodes are ancestors, so the latest ancestor is the last
540 # node.
540 # node.
541 highestrev = len(self) - 1
541 highestrev = len(self) - 1
542 # Set ancestors to None to signal that every node is an ancestor.
542 # Set ancestors to None to signal that every node is an ancestor.
543 ancestors = None
543 ancestors = None
544 # Set heads to an empty dictionary for later discovery of heads
544 # Set heads to an empty dictionary for later discovery of heads
545 heads = {}
545 heads = {}
546 else:
546 else:
547 heads = list(heads)
547 heads = list(heads)
548 if not heads:
548 if not heads:
549 return nonodes
549 return nonodes
550 ancestors = set()
550 ancestors = set()
551 # Turn heads into a dictionary so we can remove 'fake' heads.
551 # Turn heads into a dictionary so we can remove 'fake' heads.
552 # Also, later we will be using it to filter out the heads we can't
552 # Also, later we will be using it to filter out the heads we can't
553 # find from roots.
553 # find from roots.
554 heads = dict.fromkeys(heads, False)
554 heads = dict.fromkeys(heads, False)
555 # Start at the top and keep marking parents until we're done.
555 # Start at the top and keep marking parents until we're done.
556 nodestotag = set(heads)
556 nodestotag = set(heads)
557 # Remember where the top was so we can use it as a limit later.
557 # Remember where the top was so we can use it as a limit later.
558 highestrev = max([self.rev(n) for n in nodestotag])
558 highestrev = max([self.rev(n) for n in nodestotag])
559 while nodestotag:
559 while nodestotag:
560 # grab a node to tag
560 # grab a node to tag
561 n = nodestotag.pop()
561 n = nodestotag.pop()
562 # Never tag nullid
562 # Never tag nullid
563 if n == nullid:
563 if n == nullid:
564 continue
564 continue
565 # A node's revision number represents its place in a
565 # A node's revision number represents its place in a
566 # topologically sorted list of nodes.
566 # topologically sorted list of nodes.
567 r = self.rev(n)
567 r = self.rev(n)
568 if r >= lowestrev:
568 if r >= lowestrev:
569 if n not in ancestors:
569 if n not in ancestors:
570 # If we are possibly a descendant of one of the roots
570 # If we are possibly a descendant of one of the roots
571 # and we haven't already been marked as an ancestor
571 # and we haven't already been marked as an ancestor
572 ancestors.add(n) # Mark as ancestor
572 ancestors.add(n) # Mark as ancestor
573 # Add non-nullid parents to list of nodes to tag.
573 # Add non-nullid parents to list of nodes to tag.
574 nodestotag.update([p for p in self.parents(n) if
574 nodestotag.update([p for p in self.parents(n) if
575 p != nullid])
575 p != nullid])
576 elif n in heads: # We've seen it before, is it a fake head?
576 elif n in heads: # We've seen it before, is it a fake head?
577 # So it is, real heads should not be the ancestors of
577 # So it is, real heads should not be the ancestors of
578 # any other heads.
578 # any other heads.
579 heads.pop(n)
579 heads.pop(n)
580 if not ancestors:
580 if not ancestors:
581 return nonodes
581 return nonodes
582 # Now that we have our set of ancestors, we want to remove any
582 # Now that we have our set of ancestors, we want to remove any
583 # roots that are not ancestors.
583 # roots that are not ancestors.
584
584
585 # If one of the roots was nullid, everything is included anyway.
585 # If one of the roots was nullid, everything is included anyway.
586 if lowestrev > nullrev:
586 if lowestrev > nullrev:
587 # But, since we weren't, let's recompute the lowest rev to not
587 # But, since we weren't, let's recompute the lowest rev to not
588 # include roots that aren't ancestors.
588 # include roots that aren't ancestors.
589
589
590 # Filter out roots that aren't ancestors of heads
590 # Filter out roots that aren't ancestors of heads
591 roots = [n for n in roots if n in ancestors]
591 roots = [n for n in roots if n in ancestors]
592 # Recompute the lowest revision
592 # Recompute the lowest revision
593 if roots:
593 if roots:
594 lowestrev = min([self.rev(n) for n in roots])
594 lowestrev = min([self.rev(n) for n in roots])
595 else:
595 else:
596 # No more roots? Return empty list
596 # No more roots? Return empty list
597 return nonodes
597 return nonodes
598 else:
598 else:
599 # We are descending from nullid, and don't need to care about
599 # We are descending from nullid, and don't need to care about
600 # any other roots.
600 # any other roots.
601 lowestrev = nullrev
601 lowestrev = nullrev
602 roots = [nullid]
602 roots = [nullid]
603 # Transform our roots list into a set.
603 # Transform our roots list into a set.
604 descendants = set(roots)
604 descendants = set(roots)
605 # Also, keep the original roots so we can filter out roots that aren't
605 # Also, keep the original roots so we can filter out roots that aren't
606 # 'real' roots (i.e. are descended from other roots).
606 # 'real' roots (i.e. are descended from other roots).
607 roots = descendants.copy()
607 roots = descendants.copy()
608 # Our topologically sorted list of output nodes.
608 # Our topologically sorted list of output nodes.
609 orderedout = []
609 orderedout = []
610 # Don't start at nullid since we don't want nullid in our output list,
610 # Don't start at nullid since we don't want nullid in our output list,
611 # and if nullid shows up in descendants, empty parents will look like
611 # and if nullid shows up in descendants, empty parents will look like
612 # they're descendants.
612 # they're descendants.
613 for r in self.revs(start=max(lowestrev, 0), stop=highestrev + 1):
613 for r in self.revs(start=max(lowestrev, 0), stop=highestrev + 1):
614 n = self.node(r)
614 n = self.node(r)
615 isdescendant = False
615 isdescendant = False
616 if lowestrev == nullrev: # Everybody is a descendant of nullid
616 if lowestrev == nullrev: # Everybody is a descendant of nullid
617 isdescendant = True
617 isdescendant = True
618 elif n in descendants:
618 elif n in descendants:
619 # n is already a descendant
619 # n is already a descendant
620 isdescendant = True
620 isdescendant = True
621 # This check only needs to be done here because all the roots
621 # This check only needs to be done here because all the roots
622 # will start being marked is descendants before the loop.
622 # will start being marked is descendants before the loop.
623 if n in roots:
623 if n in roots:
624 # If n was a root, check if it's a 'real' root.
624 # If n was a root, check if it's a 'real' root.
625 p = tuple(self.parents(n))
625 p = tuple(self.parents(n))
626 # If any of its parents are descendants, it's not a root.
626 # If any of its parents are descendants, it's not a root.
627 if (p[0] in descendants) or (p[1] in descendants):
627 if (p[0] in descendants) or (p[1] in descendants):
628 roots.remove(n)
628 roots.remove(n)
629 else:
629 else:
630 p = tuple(self.parents(n))
630 p = tuple(self.parents(n))
631 # A node is a descendant if either of its parents are
631 # A node is a descendant if either of its parents are
632 # descendants. (We seeded the dependents list with the roots
632 # descendants. (We seeded the dependents list with the roots
633 # up there, remember?)
633 # up there, remember?)
634 if (p[0] in descendants) or (p[1] in descendants):
634 if (p[0] in descendants) or (p[1] in descendants):
635 descendants.add(n)
635 descendants.add(n)
636 isdescendant = True
636 isdescendant = True
637 if isdescendant and ((ancestors is None) or (n in ancestors)):
637 if isdescendant and ((ancestors is None) or (n in ancestors)):
638 # Only include nodes that are both descendants and ancestors.
638 # Only include nodes that are both descendants and ancestors.
639 orderedout.append(n)
639 orderedout.append(n)
640 if (ancestors is not None) and (n in heads):
640 if (ancestors is not None) and (n in heads):
641 # We're trying to figure out which heads are reachable
641 # We're trying to figure out which heads are reachable
642 # from roots.
642 # from roots.
643 # Mark this head as having been reached
643 # Mark this head as having been reached
644 heads[n] = True
644 heads[n] = True
645 elif ancestors is None:
645 elif ancestors is None:
646 # Otherwise, we're trying to discover the heads.
646 # Otherwise, we're trying to discover the heads.
647 # Assume this is a head because if it isn't, the next step
647 # Assume this is a head because if it isn't, the next step
648 # will eventually remove it.
648 # will eventually remove it.
649 heads[n] = True
649 heads[n] = True
650 # But, obviously its parents aren't.
650 # But, obviously its parents aren't.
651 for p in self.parents(n):
651 for p in self.parents(n):
652 heads.pop(p, None)
652 heads.pop(p, None)
653 heads = [n for n, flag in heads.iteritems() if flag]
653 heads = [n for n, flag in heads.iteritems() if flag]
654 roots = list(roots)
654 roots = list(roots)
655 assert orderedout
655 assert orderedout
656 assert roots
656 assert roots
657 assert heads
657 assert heads
658 return (orderedout, roots, heads)
658 return (orderedout, roots, heads)
659
659
660 def headrevs(self):
660 def headrevs(self):
661 try:
661 try:
662 return self.index.headrevs()
662 return self.index.headrevs()
663 except AttributeError:
663 except AttributeError:
664 return self._headrevs()
664 return self._headrevs()
665
665
666 def _headrevs(self):
666 def _headrevs(self):
667 count = len(self)
667 count = len(self)
668 if not count:
668 if not count:
669 return [nullrev]
669 return [nullrev]
670 # we won't iter over filtered rev so nobody is a head at start
670 # we won't iter over filtered rev so nobody is a head at start
671 ishead = [0] * (count + 1)
671 ishead = [0] * (count + 1)
672 index = self.index
672 index = self.index
673 for r in self:
673 for r in self:
674 ishead[r] = 1 # I may be an head
674 ishead[r] = 1 # I may be an head
675 e = index[r]
675 e = index[r]
676 ishead[e[5]] = ishead[e[6]] = 0 # my parent are not
676 ishead[e[5]] = ishead[e[6]] = 0 # my parent are not
677 return [r for r, val in enumerate(ishead) if val]
677 return [r for r, val in enumerate(ishead) if val]
678
678
679 def heads(self, start=None, stop=None):
679 def heads(self, start=None, stop=None):
680 """return the list of all nodes that have no children
680 """return the list of all nodes that have no children
681
681
682 if start is specified, only heads that are descendants of
682 if start is specified, only heads that are descendants of
683 start will be returned
683 start will be returned
684 if stop is specified, it will consider all the revs from stop
684 if stop is specified, it will consider all the revs from stop
685 as if they had no children
685 as if they had no children
686 """
686 """
687 if start is None and stop is None:
687 if start is None and stop is None:
688 if not len(self):
688 if not len(self):
689 return [nullid]
689 return [nullid]
690 return [self.node(r) for r in self.headrevs()]
690 return [self.node(r) for r in self.headrevs()]
691
691
692 if start is None:
692 if start is None:
693 start = nullid
693 start = nullid
694 if stop is None:
694 if stop is None:
695 stop = []
695 stop = []
696 stoprevs = set([self.rev(n) for n in stop])
696 stoprevs = set([self.rev(n) for n in stop])
697 startrev = self.rev(start)
697 startrev = self.rev(start)
698 reachable = set((startrev,))
698 reachable = set((startrev,))
699 heads = set((startrev,))
699 heads = set((startrev,))
700
700
701 parentrevs = self.parentrevs
701 parentrevs = self.parentrevs
702 for r in self.revs(start=startrev + 1):
702 for r in self.revs(start=startrev + 1):
703 for p in parentrevs(r):
703 for p in parentrevs(r):
704 if p in reachable:
704 if p in reachable:
705 if r not in stoprevs:
705 if r not in stoprevs:
706 reachable.add(r)
706 reachable.add(r)
707 heads.add(r)
707 heads.add(r)
708 if p in heads and p not in stoprevs:
708 if p in heads and p not in stoprevs:
709 heads.remove(p)
709 heads.remove(p)
710
710
711 return [self.node(r) for r in heads]
711 return [self.node(r) for r in heads]
712
712
713 def children(self, node):
713 def children(self, node):
714 """find the children of a given node"""
714 """find the children of a given node"""
715 c = []
715 c = []
716 p = self.rev(node)
716 p = self.rev(node)
717 for r in self.revs(start=p + 1):
717 for r in self.revs(start=p + 1):
718 prevs = [pr for pr in self.parentrevs(r) if pr != nullrev]
718 prevs = [pr for pr in self.parentrevs(r) if pr != nullrev]
719 if prevs:
719 if prevs:
720 for pr in prevs:
720 for pr in prevs:
721 if pr == p:
721 if pr == p:
722 c.append(self.node(r))
722 c.append(self.node(r))
723 elif p == nullrev:
723 elif p == nullrev:
724 c.append(self.node(r))
724 c.append(self.node(r))
725 return c
725 return c
726
726
727 def descendant(self, start, end):
727 def descendant(self, start, end):
728 if start == nullrev:
728 if start == nullrev:
729 return True
729 return True
730 for i in self.descendants([start]):
730 for i in self.descendants([start]):
731 if i == end:
731 if i == end:
732 return True
732 return True
733 elif i > end:
733 elif i > end:
734 break
734 break
735 return False
735 return False
736
736
737 def commonancestors(self, a, b):
737 def commonancestors(self, a, b):
738 """calculate the least common ancestors of nodes a and b"""
738 """calculate the least common ancestors of nodes a and b"""
739 a, b = self.rev(a), self.rev(b)
739 a, b = self.rev(a), self.rev(b)
740 try:
740 try:
741 ancs = self.index.ancestors(a, b)
741 ancs = self.index.ancestors(a, b)
742 except (AttributeError, OverflowError): # C implementation failed
742 except (AttributeError, OverflowError): # C implementation failed
743 ancs = ancestor.ancestors(self.parentrevs, a, b)
743 ancs = ancestor.ancestors(self.parentrevs, a, b)
744 return map(self.node, ancs)
744 return map(self.node, ancs)
745
745
746 def ancestor(self, a, b):
746 def ancestor(self, a, b):
747 """calculate a least common ancestor of nodes a and b"""
747 """calculate a least common ancestor of nodes a and b"""
748 ancs = self.commonancestors(a, b)
748 ancs = self.commonancestors(a, b)
749 if ancs:
749 if ancs:
750 # choose a consistent winner when there's a tie
750 # choose a consistent winner when there's a tie
751 return min(ancs)
751 return min(ancs)
752 return nullid
752 return nullid
753
753
754 def _match(self, id):
754 def _match(self, id):
755 if isinstance(id, int):
755 if isinstance(id, int):
756 # rev
756 # rev
757 return self.node(id)
757 return self.node(id)
758 if len(id) == 20:
758 if len(id) == 20:
759 # possibly a binary node
759 # possibly a binary node
760 # odds of a binary node being all hex in ASCII are 1 in 10**25
760 # odds of a binary node being all hex in ASCII are 1 in 10**25
761 try:
761 try:
762 node = id
762 node = id
763 self.rev(node) # quick search the index
763 self.rev(node) # quick search the index
764 return node
764 return node
765 except LookupError:
765 except LookupError:
766 pass # may be partial hex id
766 pass # may be partial hex id
767 try:
767 try:
768 # str(rev)
768 # str(rev)
769 rev = int(id)
769 rev = int(id)
770 if str(rev) != id:
770 if str(rev) != id:
771 raise ValueError
771 raise ValueError
772 if rev < 0:
772 if rev < 0:
773 rev = len(self) + rev
773 rev = len(self) + rev
774 if rev < 0 or rev >= len(self):
774 if rev < 0 or rev >= len(self):
775 raise ValueError
775 raise ValueError
776 return self.node(rev)
776 return self.node(rev)
777 except (ValueError, OverflowError):
777 except (ValueError, OverflowError):
778 pass
778 pass
779 if len(id) == 40:
779 if len(id) == 40:
780 try:
780 try:
781 # a full hex nodeid?
781 # a full hex nodeid?
782 node = bin(id)
782 node = bin(id)
783 self.rev(node)
783 self.rev(node)
784 return node
784 return node
785 except (TypeError, LookupError):
785 except (TypeError, LookupError):
786 pass
786 pass
787
787
788 def _partialmatch(self, id):
788 def _partialmatch(self, id):
789 try:
789 try:
790 n = self.index.partialmatch(id)
790 n = self.index.partialmatch(id)
791 if n and self.hasnode(n):
791 if n and self.hasnode(n):
792 return n
792 return n
793 return None
793 return None
794 except RevlogError:
794 except RevlogError:
795 # parsers.c radix tree lookup gave multiple matches
795 # parsers.c radix tree lookup gave multiple matches
796 # fall through to slow path that filters hidden revisions
796 # fall through to slow path that filters hidden revisions
797 pass
797 pass
798 except (AttributeError, ValueError):
798 except (AttributeError, ValueError):
799 # we are pure python, or key was too short to search radix tree
799 # we are pure python, or key was too short to search radix tree
800 pass
800 pass
801
801
802 if id in self._pcache:
802 if id in self._pcache:
803 return self._pcache[id]
803 return self._pcache[id]
804
804
805 if len(id) < 40:
805 if len(id) < 40:
806 try:
806 try:
807 # hex(node)[:...]
807 # hex(node)[:...]
808 l = len(id) // 2 # grab an even number of digits
808 l = len(id) // 2 # grab an even number of digits
809 prefix = bin(id[:l * 2])
809 prefix = bin(id[:l * 2])
810 nl = [e[7] for e in self.index if e[7].startswith(prefix)]
810 nl = [e[7] for e in self.index if e[7].startswith(prefix)]
811 nl = [n for n in nl if hex(n).startswith(id) and
811 nl = [n for n in nl if hex(n).startswith(id) and
812 self.hasnode(n)]
812 self.hasnode(n)]
813 if len(nl) > 0:
813 if len(nl) > 0:
814 if len(nl) == 1:
814 if len(nl) == 1:
815 self._pcache[id] = nl[0]
815 self._pcache[id] = nl[0]
816 return nl[0]
816 return nl[0]
817 raise LookupError(id, self.indexfile,
817 raise LookupError(id, self.indexfile,
818 _('ambiguous identifier'))
818 _('ambiguous identifier'))
819 return None
819 return None
820 except TypeError:
820 except TypeError:
821 pass
821 pass
822
822
823 def lookup(self, id):
823 def lookup(self, id):
824 """locate a node based on:
824 """locate a node based on:
825 - revision number or str(revision number)
825 - revision number or str(revision number)
826 - nodeid or subset of hex nodeid
826 - nodeid or subset of hex nodeid
827 """
827 """
828 n = self._match(id)
828 n = self._match(id)
829 if n is not None:
829 if n is not None:
830 return n
830 return n
831 n = self._partialmatch(id)
831 n = self._partialmatch(id)
832 if n:
832 if n:
833 return n
833 return n
834
834
835 raise LookupError(id, self.indexfile, _('no match found'))
835 raise LookupError(id, self.indexfile, _('no match found'))
836
836
837 def cmp(self, node, text):
837 def cmp(self, node, text):
838 """compare text with a given file revision
838 """compare text with a given file revision
839
839
840 returns True if text is different than what is stored.
840 returns True if text is different than what is stored.
841 """
841 """
842 p1, p2 = self.parents(node)
842 p1, p2 = self.parents(node)
843 return hash(text, p1, p2) != node
843 return hash(text, p1, p2) != node
844
844
845 def _addchunk(self, offset, data):
845 def _addchunk(self, offset, data):
846 o, d = self._chunkcache
846 o, d = self._chunkcache
847 # try to add to existing cache
847 # try to add to existing cache
848 if o + len(d) == offset and len(d) + len(data) < _chunksize:
848 if o + len(d) == offset and len(d) + len(data) < _chunksize:
849 self._chunkcache = o, d + data
849 self._chunkcache = o, d + data
850 else:
850 else:
851 self._chunkcache = offset, data
851 self._chunkcache = offset, data
852
852
853 def _loadchunk(self, offset, length):
853 def _loadchunk(self, offset, length):
854 if self._inline:
854 if self._inline:
855 df = self.opener(self.indexfile)
855 df = self.opener(self.indexfile)
856 else:
856 else:
857 df = self.opener(self.datafile)
857 df = self.opener(self.datafile)
858
858
859 # Cache data both forward and backward around the requested
859 # Cache data both forward and backward around the requested
860 # data, in a fixed size window. This helps speed up operations
860 # data, in a fixed size window. This helps speed up operations
861 # involving reading the revlog backwards.
861 # involving reading the revlog backwards.
862 cachesize = self._chunkcachesize
862 cachesize = self._chunkcachesize
863 realoffset = offset & ~(cachesize - 1)
863 realoffset = offset & ~(cachesize - 1)
864 reallength = (((offset + length + cachesize) & ~(cachesize - 1))
864 reallength = (((offset + length + cachesize) & ~(cachesize - 1))
865 - realoffset)
865 - realoffset)
866 df.seek(realoffset)
866 df.seek(realoffset)
867 d = df.read(reallength)
867 d = df.read(reallength)
868 df.close()
868 df.close()
869 self._addchunk(realoffset, d)
869 self._addchunk(realoffset, d)
870 if offset != realoffset or reallength != length:
870 if offset != realoffset or reallength != length:
871 return util.buffer(d, offset - realoffset, length)
871 return util.buffer(d, offset - realoffset, length)
872 return d
872 return d
873
873
874 def _getchunk(self, offset, length):
874 def _getchunk(self, offset, length):
875 o, d = self._chunkcache
875 o, d = self._chunkcache
876 l = len(d)
876 l = len(d)
877
877
878 # is it in the cache?
878 # is it in the cache?
879 cachestart = offset - o
879 cachestart = offset - o
880 cacheend = cachestart + length
880 cacheend = cachestart + length
881 if cachestart >= 0 and cacheend <= l:
881 if cachestart >= 0 and cacheend <= l:
882 if cachestart == 0 and cacheend == l:
882 if cachestart == 0 and cacheend == l:
883 return d # avoid a copy
883 return d # avoid a copy
884 return util.buffer(d, cachestart, cacheend - cachestart)
884 return util.buffer(d, cachestart, cacheend - cachestart)
885
885
886 return self._loadchunk(offset, length)
886 return self._loadchunk(offset, length)
887
887
888 def _chunkraw(self, startrev, endrev):
888 def _chunkraw(self, startrev, endrev):
889 start = self.start(startrev)
889 start = self.start(startrev)
890 end = self.end(endrev)
890 end = self.end(endrev)
891 if self._inline:
891 if self._inline:
892 start += (startrev + 1) * self._io.size
892 start += (startrev + 1) * self._io.size
893 end += (endrev + 1) * self._io.size
893 end += (endrev + 1) * self._io.size
894 length = end - start
894 length = end - start
895 return self._getchunk(start, length)
895 return self._getchunk(start, length)
896
896
897 def _chunk(self, rev):
897 def _chunk(self, rev):
898 return decompress(self._chunkraw(rev, rev))
898 return decompress(self._chunkraw(rev, rev))
899
899
900 def _chunks(self, revs):
900 def _chunks(self, revs):
901 '''faster version of [self._chunk(rev) for rev in revs]
901 '''faster version of [self._chunk(rev) for rev in revs]
902
902
903 Assumes that revs is in ascending order.'''
903 Assumes that revs is in ascending order.'''
904 if not revs:
904 if not revs:
905 return []
905 return []
906 start = self.start
906 start = self.start
907 length = self.length
907 length = self.length
908 inline = self._inline
908 inline = self._inline
909 iosize = self._io.size
909 iosize = self._io.size
910 buffer = util.buffer
910 buffer = util.buffer
911
911
912 l = []
912 l = []
913 ladd = l.append
913 ladd = l.append
914
914
915 # preload the cache
915 # preload the cache
916 self._chunkraw(revs[0], revs[-1])
916 try:
917 offset, data = self._chunkcache
917 self._chunkraw(revs[0], revs[-1])
918 offset, data = self._chunkcache
919 except OverflowError:
920 # issue4215 - we can't cache a run of chunks greater than
921 # 2G on Windows
922 return [self._chunk(rev) for rev in revs]
918
923
919 for rev in revs:
924 for rev in revs:
920 chunkstart = start(rev)
925 chunkstart = start(rev)
921 if inline:
926 if inline:
922 chunkstart += (rev + 1) * iosize
927 chunkstart += (rev + 1) * iosize
923 chunklength = length(rev)
928 chunklength = length(rev)
924 ladd(decompress(buffer(data, chunkstart - offset, chunklength)))
929 ladd(decompress(buffer(data, chunkstart - offset, chunklength)))
925
930
926 return l
931 return l
927
932
928 def _chunkclear(self):
933 def _chunkclear(self):
929 self._chunkcache = (0, '')
934 self._chunkcache = (0, '')
930
935
931 def deltaparent(self, rev):
936 def deltaparent(self, rev):
932 """return deltaparent of the given revision"""
937 """return deltaparent of the given revision"""
933 base = self.index[rev][3]
938 base = self.index[rev][3]
934 if base == rev:
939 if base == rev:
935 return nullrev
940 return nullrev
936 elif self._generaldelta:
941 elif self._generaldelta:
937 return base
942 return base
938 else:
943 else:
939 return rev - 1
944 return rev - 1
940
945
941 def revdiff(self, rev1, rev2):
946 def revdiff(self, rev1, rev2):
942 """return or calculate a delta between two revisions"""
947 """return or calculate a delta between two revisions"""
943 if rev1 != nullrev and self.deltaparent(rev2) == rev1:
948 if rev1 != nullrev and self.deltaparent(rev2) == rev1:
944 return str(self._chunk(rev2))
949 return str(self._chunk(rev2))
945
950
946 return mdiff.textdiff(self.revision(rev1),
951 return mdiff.textdiff(self.revision(rev1),
947 self.revision(rev2))
952 self.revision(rev2))
948
953
949 def revision(self, nodeorrev):
954 def revision(self, nodeorrev):
950 """return an uncompressed revision of a given node or revision
955 """return an uncompressed revision of a given node or revision
951 number.
956 number.
952 """
957 """
953 if isinstance(nodeorrev, int):
958 if isinstance(nodeorrev, int):
954 rev = nodeorrev
959 rev = nodeorrev
955 node = self.node(rev)
960 node = self.node(rev)
956 else:
961 else:
957 node = nodeorrev
962 node = nodeorrev
958 rev = None
963 rev = None
959
964
960 cachedrev = None
965 cachedrev = None
961 if node == nullid:
966 if node == nullid:
962 return ""
967 return ""
963 if self._cache:
968 if self._cache:
964 if self._cache[0] == node:
969 if self._cache[0] == node:
965 return self._cache[2]
970 return self._cache[2]
966 cachedrev = self._cache[1]
971 cachedrev = self._cache[1]
967
972
968 # look up what we need to read
973 # look up what we need to read
969 text = None
974 text = None
970 if rev is None:
975 if rev is None:
971 rev = self.rev(node)
976 rev = self.rev(node)
972
977
973 # check rev flags
978 # check rev flags
974 if self.flags(rev) & ~REVIDX_KNOWN_FLAGS:
979 if self.flags(rev) & ~REVIDX_KNOWN_FLAGS:
975 raise RevlogError(_('incompatible revision flag %x') %
980 raise RevlogError(_('incompatible revision flag %x') %
976 (self.flags(rev) & ~REVIDX_KNOWN_FLAGS))
981 (self.flags(rev) & ~REVIDX_KNOWN_FLAGS))
977
982
978 # build delta chain
983 # build delta chain
979 chain = []
984 chain = []
980 index = self.index # for performance
985 index = self.index # for performance
981 generaldelta = self._generaldelta
986 generaldelta = self._generaldelta
982 iterrev = rev
987 iterrev = rev
983 e = index[iterrev]
988 e = index[iterrev]
984 while iterrev != e[3] and iterrev != cachedrev:
989 while iterrev != e[3] and iterrev != cachedrev:
985 chain.append(iterrev)
990 chain.append(iterrev)
986 if generaldelta:
991 if generaldelta:
987 iterrev = e[3]
992 iterrev = e[3]
988 else:
993 else:
989 iterrev -= 1
994 iterrev -= 1
990 e = index[iterrev]
995 e = index[iterrev]
991
996
992 if iterrev == cachedrev:
997 if iterrev == cachedrev:
993 # cache hit
998 # cache hit
994 text = self._cache[2]
999 text = self._cache[2]
995 else:
1000 else:
996 chain.append(iterrev)
1001 chain.append(iterrev)
997 chain.reverse()
1002 chain.reverse()
998
1003
999 # drop cache to save memory
1004 # drop cache to save memory
1000 self._cache = None
1005 self._cache = None
1001
1006
1002 bins = self._chunks(chain)
1007 bins = self._chunks(chain)
1003 if text is None:
1008 if text is None:
1004 text = str(bins[0])
1009 text = str(bins[0])
1005 bins = bins[1:]
1010 bins = bins[1:]
1006
1011
1007 text = mdiff.patches(text, bins)
1012 text = mdiff.patches(text, bins)
1008
1013
1009 text = self._checkhash(text, node, rev)
1014 text = self._checkhash(text, node, rev)
1010
1015
1011 self._cache = (node, rev, text)
1016 self._cache = (node, rev, text)
1012 return text
1017 return text
1013
1018
1014 def _checkhash(self, text, node, rev):
1019 def _checkhash(self, text, node, rev):
1015 p1, p2 = self.parents(node)
1020 p1, p2 = self.parents(node)
1016 self.checkhash(text, p1, p2, node, rev)
1021 self.checkhash(text, p1, p2, node, rev)
1017 return text
1022 return text
1018
1023
1019 def checkhash(self, text, p1, p2, node, rev=None):
1024 def checkhash(self, text, p1, p2, node, rev=None):
1020 if node != hash(text, p1, p2):
1025 if node != hash(text, p1, p2):
1021 revornode = rev
1026 revornode = rev
1022 if revornode is None:
1027 if revornode is None:
1023 revornode = templatefilters.short(hex(node))
1028 revornode = templatefilters.short(hex(node))
1024 raise RevlogError(_("integrity check failed on %s:%s")
1029 raise RevlogError(_("integrity check failed on %s:%s")
1025 % (self.indexfile, revornode))
1030 % (self.indexfile, revornode))
1026
1031
1027 def checkinlinesize(self, tr, fp=None):
1032 def checkinlinesize(self, tr, fp=None):
1028 if not self._inline or (self.start(-2) + self.length(-2)) < _maxinline:
1033 if not self._inline or (self.start(-2) + self.length(-2)) < _maxinline:
1029 return
1034 return
1030
1035
1031 trinfo = tr.find(self.indexfile)
1036 trinfo = tr.find(self.indexfile)
1032 if trinfo is None:
1037 if trinfo is None:
1033 raise RevlogError(_("%s not found in the transaction")
1038 raise RevlogError(_("%s not found in the transaction")
1034 % self.indexfile)
1039 % self.indexfile)
1035
1040
1036 trindex = trinfo[2]
1041 trindex = trinfo[2]
1037 dataoff = self.start(trindex)
1042 dataoff = self.start(trindex)
1038
1043
1039 tr.add(self.datafile, dataoff)
1044 tr.add(self.datafile, dataoff)
1040
1045
1041 if fp:
1046 if fp:
1042 fp.flush()
1047 fp.flush()
1043 fp.close()
1048 fp.close()
1044
1049
1045 df = self.opener(self.datafile, 'w')
1050 df = self.opener(self.datafile, 'w')
1046 try:
1051 try:
1047 for r in self:
1052 for r in self:
1048 df.write(self._chunkraw(r, r))
1053 df.write(self._chunkraw(r, r))
1049 finally:
1054 finally:
1050 df.close()
1055 df.close()
1051
1056
1052 fp = self.opener(self.indexfile, 'w', atomictemp=True)
1057 fp = self.opener(self.indexfile, 'w', atomictemp=True)
1053 self.version &= ~(REVLOGNGINLINEDATA)
1058 self.version &= ~(REVLOGNGINLINEDATA)
1054 self._inline = False
1059 self._inline = False
1055 for i in self:
1060 for i in self:
1056 e = self._io.packentry(self.index[i], self.node, self.version, i)
1061 e = self._io.packentry(self.index[i], self.node, self.version, i)
1057 fp.write(e)
1062 fp.write(e)
1058
1063
1059 # if we don't call close, the temp file will never replace the
1064 # if we don't call close, the temp file will never replace the
1060 # real index
1065 # real index
1061 fp.close()
1066 fp.close()
1062
1067
1063 tr.replace(self.indexfile, trindex * self._io.size)
1068 tr.replace(self.indexfile, trindex * self._io.size)
1064 self._chunkclear()
1069 self._chunkclear()
1065
1070
1066 def addrevision(self, text, transaction, link, p1, p2, cachedelta=None,
1071 def addrevision(self, text, transaction, link, p1, p2, cachedelta=None,
1067 node=None):
1072 node=None):
1068 """add a revision to the log
1073 """add a revision to the log
1069
1074
1070 text - the revision data to add
1075 text - the revision data to add
1071 transaction - the transaction object used for rollback
1076 transaction - the transaction object used for rollback
1072 link - the linkrev data to add
1077 link - the linkrev data to add
1073 p1, p2 - the parent nodeids of the revision
1078 p1, p2 - the parent nodeids of the revision
1074 cachedelta - an optional precomputed delta
1079 cachedelta - an optional precomputed delta
1075 node - nodeid of revision; typically node is not specified, and it is
1080 node - nodeid of revision; typically node is not specified, and it is
1076 computed by default as hash(text, p1, p2), however subclasses might
1081 computed by default as hash(text, p1, p2), however subclasses might
1077 use different hashing method (and override checkhash() in such case)
1082 use different hashing method (and override checkhash() in such case)
1078 """
1083 """
1079 if link == nullrev:
1084 if link == nullrev:
1080 raise RevlogError(_("attempted to add linkrev -1 to %s")
1085 raise RevlogError(_("attempted to add linkrev -1 to %s")
1081 % self.indexfile)
1086 % self.indexfile)
1082 node = node or hash(text, p1, p2)
1087 node = node or hash(text, p1, p2)
1083 if node in self.nodemap:
1088 if node in self.nodemap:
1084 return node
1089 return node
1085
1090
1086 dfh = None
1091 dfh = None
1087 if not self._inline:
1092 if not self._inline:
1088 dfh = self.opener(self.datafile, "a")
1093 dfh = self.opener(self.datafile, "a")
1089 ifh = self.opener(self.indexfile, "a+")
1094 ifh = self.opener(self.indexfile, "a+")
1090 try:
1095 try:
1091 return self._addrevision(node, text, transaction, link, p1, p2,
1096 return self._addrevision(node, text, transaction, link, p1, p2,
1092 cachedelta, ifh, dfh)
1097 cachedelta, ifh, dfh)
1093 finally:
1098 finally:
1094 if dfh:
1099 if dfh:
1095 dfh.close()
1100 dfh.close()
1096 ifh.close()
1101 ifh.close()
1097
1102
1098 def compress(self, text):
1103 def compress(self, text):
1099 """ generate a possibly-compressed representation of text """
1104 """ generate a possibly-compressed representation of text """
1100 if not text:
1105 if not text:
1101 return ("", text)
1106 return ("", text)
1102 l = len(text)
1107 l = len(text)
1103 bin = None
1108 bin = None
1104 if l < 44:
1109 if l < 44:
1105 pass
1110 pass
1106 elif l > 1000000:
1111 elif l > 1000000:
1107 # zlib makes an internal copy, thus doubling memory usage for
1112 # zlib makes an internal copy, thus doubling memory usage for
1108 # large files, so lets do this in pieces
1113 # large files, so lets do this in pieces
1109 z = zlib.compressobj()
1114 z = zlib.compressobj()
1110 p = []
1115 p = []
1111 pos = 0
1116 pos = 0
1112 while pos < l:
1117 while pos < l:
1113 pos2 = pos + 2**20
1118 pos2 = pos + 2**20
1114 p.append(z.compress(text[pos:pos2]))
1119 p.append(z.compress(text[pos:pos2]))
1115 pos = pos2
1120 pos = pos2
1116 p.append(z.flush())
1121 p.append(z.flush())
1117 if sum(map(len, p)) < l:
1122 if sum(map(len, p)) < l:
1118 bin = "".join(p)
1123 bin = "".join(p)
1119 else:
1124 else:
1120 bin = _compress(text)
1125 bin = _compress(text)
1121 if bin is None or len(bin) > l:
1126 if bin is None or len(bin) > l:
1122 if text[0] == '\0':
1127 if text[0] == '\0':
1123 return ("", text)
1128 return ("", text)
1124 return ('u', text)
1129 return ('u', text)
1125 return ("", bin)
1130 return ("", bin)
1126
1131
1127 def _addrevision(self, node, text, transaction, link, p1, p2,
1132 def _addrevision(self, node, text, transaction, link, p1, p2,
1128 cachedelta, ifh, dfh):
1133 cachedelta, ifh, dfh):
1129 """internal function to add revisions to the log
1134 """internal function to add revisions to the log
1130
1135
1131 see addrevision for argument descriptions.
1136 see addrevision for argument descriptions.
1132 invariants:
1137 invariants:
1133 - text is optional (can be None); if not set, cachedelta must be set.
1138 - text is optional (can be None); if not set, cachedelta must be set.
1134 if both are set, they must correspond to each other.
1139 if both are set, they must correspond to each other.
1135 """
1140 """
1136 btext = [text]
1141 btext = [text]
1137 def buildtext():
1142 def buildtext():
1138 if btext[0] is not None:
1143 if btext[0] is not None:
1139 return btext[0]
1144 return btext[0]
1140 # flush any pending writes here so we can read it in revision
1145 # flush any pending writes here so we can read it in revision
1141 if dfh:
1146 if dfh:
1142 dfh.flush()
1147 dfh.flush()
1143 ifh.flush()
1148 ifh.flush()
1144 basetext = self.revision(self.node(cachedelta[0]))
1149 basetext = self.revision(self.node(cachedelta[0]))
1145 btext[0] = mdiff.patch(basetext, cachedelta[1])
1150 btext[0] = mdiff.patch(basetext, cachedelta[1])
1146 self.checkhash(btext[0], p1, p2, node)
1151 self.checkhash(btext[0], p1, p2, node)
1147 return btext[0]
1152 return btext[0]
1148
1153
1149 def builddelta(rev):
1154 def builddelta(rev):
1150 # can we use the cached delta?
1155 # can we use the cached delta?
1151 if cachedelta and cachedelta[0] == rev:
1156 if cachedelta and cachedelta[0] == rev:
1152 delta = cachedelta[1]
1157 delta = cachedelta[1]
1153 else:
1158 else:
1154 t = buildtext()
1159 t = buildtext()
1155 ptext = self.revision(self.node(rev))
1160 ptext = self.revision(self.node(rev))
1156 delta = mdiff.textdiff(ptext, t)
1161 delta = mdiff.textdiff(ptext, t)
1157 data = self.compress(delta)
1162 data = self.compress(delta)
1158 l = len(data[1]) + len(data[0])
1163 l = len(data[1]) + len(data[0])
1159 if basecache[0] == rev:
1164 if basecache[0] == rev:
1160 chainbase = basecache[1]
1165 chainbase = basecache[1]
1161 else:
1166 else:
1162 chainbase = self.chainbase(rev)
1167 chainbase = self.chainbase(rev)
1163 dist = l + offset - self.start(chainbase)
1168 dist = l + offset - self.start(chainbase)
1164 if self._generaldelta:
1169 if self._generaldelta:
1165 base = rev
1170 base = rev
1166 else:
1171 else:
1167 base = chainbase
1172 base = chainbase
1168 return dist, l, data, base, chainbase
1173 return dist, l, data, base, chainbase
1169
1174
1170 curr = len(self)
1175 curr = len(self)
1171 prev = curr - 1
1176 prev = curr - 1
1172 base = chainbase = curr
1177 base = chainbase = curr
1173 offset = self.end(prev)
1178 offset = self.end(prev)
1174 flags = 0
1179 flags = 0
1175 d = None
1180 d = None
1176 if self._basecache is None:
1181 if self._basecache is None:
1177 self._basecache = (prev, self.chainbase(prev))
1182 self._basecache = (prev, self.chainbase(prev))
1178 basecache = self._basecache
1183 basecache = self._basecache
1179 p1r, p2r = self.rev(p1), self.rev(p2)
1184 p1r, p2r = self.rev(p1), self.rev(p2)
1180
1185
1181 # should we try to build a delta?
1186 # should we try to build a delta?
1182 if prev != nullrev:
1187 if prev != nullrev:
1183 if self._generaldelta:
1188 if self._generaldelta:
1184 if p1r >= basecache[1]:
1189 if p1r >= basecache[1]:
1185 d = builddelta(p1r)
1190 d = builddelta(p1r)
1186 elif p2r >= basecache[1]:
1191 elif p2r >= basecache[1]:
1187 d = builddelta(p2r)
1192 d = builddelta(p2r)
1188 else:
1193 else:
1189 d = builddelta(prev)
1194 d = builddelta(prev)
1190 else:
1195 else:
1191 d = builddelta(prev)
1196 d = builddelta(prev)
1192 dist, l, data, base, chainbase = d
1197 dist, l, data, base, chainbase = d
1193
1198
1194 # full versions are inserted when the needed deltas
1199 # full versions are inserted when the needed deltas
1195 # become comparable to the uncompressed text
1200 # become comparable to the uncompressed text
1196 if text is None:
1201 if text is None:
1197 textlen = mdiff.patchedsize(self.rawsize(cachedelta[0]),
1202 textlen = mdiff.patchedsize(self.rawsize(cachedelta[0]),
1198 cachedelta[1])
1203 cachedelta[1])
1199 else:
1204 else:
1200 textlen = len(text)
1205 textlen = len(text)
1201 if d is None or dist > textlen * 2:
1206 if d is None or dist > textlen * 2:
1202 text = buildtext()
1207 text = buildtext()
1203 data = self.compress(text)
1208 data = self.compress(text)
1204 l = len(data[1]) + len(data[0])
1209 l = len(data[1]) + len(data[0])
1205 base = chainbase = curr
1210 base = chainbase = curr
1206
1211
1207 e = (offset_type(offset, flags), l, textlen,
1212 e = (offset_type(offset, flags), l, textlen,
1208 base, link, p1r, p2r, node)
1213 base, link, p1r, p2r, node)
1209 self.index.insert(-1, e)
1214 self.index.insert(-1, e)
1210 self.nodemap[node] = curr
1215 self.nodemap[node] = curr
1211
1216
1212 entry = self._io.packentry(e, self.node, self.version, curr)
1217 entry = self._io.packentry(e, self.node, self.version, curr)
1213 self._writeentry(transaction, ifh, dfh, entry, data, link, offset)
1218 self._writeentry(transaction, ifh, dfh, entry, data, link, offset)
1214
1219
1215 if type(text) == str: # only accept immutable objects
1220 if type(text) == str: # only accept immutable objects
1216 self._cache = (node, curr, text)
1221 self._cache = (node, curr, text)
1217 self._basecache = (curr, chainbase)
1222 self._basecache = (curr, chainbase)
1218 return node
1223 return node
1219
1224
1220 def _writeentry(self, transaction, ifh, dfh, entry, data, link, offset):
1225 def _writeentry(self, transaction, ifh, dfh, entry, data, link, offset):
1221 curr = len(self) - 1
1226 curr = len(self) - 1
1222 if not self._inline:
1227 if not self._inline:
1223 transaction.add(self.datafile, offset)
1228 transaction.add(self.datafile, offset)
1224 transaction.add(self.indexfile, curr * len(entry))
1229 transaction.add(self.indexfile, curr * len(entry))
1225 if data[0]:
1230 if data[0]:
1226 dfh.write(data[0])
1231 dfh.write(data[0])
1227 dfh.write(data[1])
1232 dfh.write(data[1])
1228 dfh.flush()
1233 dfh.flush()
1229 ifh.write(entry)
1234 ifh.write(entry)
1230 else:
1235 else:
1231 offset += curr * self._io.size
1236 offset += curr * self._io.size
1232 transaction.add(self.indexfile, offset, curr)
1237 transaction.add(self.indexfile, offset, curr)
1233 ifh.write(entry)
1238 ifh.write(entry)
1234 ifh.write(data[0])
1239 ifh.write(data[0])
1235 ifh.write(data[1])
1240 ifh.write(data[1])
1236 self.checkinlinesize(transaction, ifh)
1241 self.checkinlinesize(transaction, ifh)
1237
1242
1238 def addgroup(self, bundle, linkmapper, transaction):
1243 def addgroup(self, bundle, linkmapper, transaction):
1239 """
1244 """
1240 add a delta group
1245 add a delta group
1241
1246
1242 given a set of deltas, add them to the revision log. the
1247 given a set of deltas, add them to the revision log. the
1243 first delta is against its parent, which should be in our
1248 first delta is against its parent, which should be in our
1244 log, the rest are against the previous delta.
1249 log, the rest are against the previous delta.
1245 """
1250 """
1246
1251
1247 # track the base of the current delta log
1252 # track the base of the current delta log
1248 content = []
1253 content = []
1249 node = None
1254 node = None
1250
1255
1251 r = len(self)
1256 r = len(self)
1252 end = 0
1257 end = 0
1253 if r:
1258 if r:
1254 end = self.end(r - 1)
1259 end = self.end(r - 1)
1255 ifh = self.opener(self.indexfile, "a+")
1260 ifh = self.opener(self.indexfile, "a+")
1256 isize = r * self._io.size
1261 isize = r * self._io.size
1257 if self._inline:
1262 if self._inline:
1258 transaction.add(self.indexfile, end + isize, r)
1263 transaction.add(self.indexfile, end + isize, r)
1259 dfh = None
1264 dfh = None
1260 else:
1265 else:
1261 transaction.add(self.indexfile, isize, r)
1266 transaction.add(self.indexfile, isize, r)
1262 transaction.add(self.datafile, end)
1267 transaction.add(self.datafile, end)
1263 dfh = self.opener(self.datafile, "a")
1268 dfh = self.opener(self.datafile, "a")
1264
1269
1265 try:
1270 try:
1266 # loop through our set of deltas
1271 # loop through our set of deltas
1267 chain = None
1272 chain = None
1268 while True:
1273 while True:
1269 chunkdata = bundle.deltachunk(chain)
1274 chunkdata = bundle.deltachunk(chain)
1270 if not chunkdata:
1275 if not chunkdata:
1271 break
1276 break
1272 node = chunkdata['node']
1277 node = chunkdata['node']
1273 p1 = chunkdata['p1']
1278 p1 = chunkdata['p1']
1274 p2 = chunkdata['p2']
1279 p2 = chunkdata['p2']
1275 cs = chunkdata['cs']
1280 cs = chunkdata['cs']
1276 deltabase = chunkdata['deltabase']
1281 deltabase = chunkdata['deltabase']
1277 delta = chunkdata['delta']
1282 delta = chunkdata['delta']
1278
1283
1279 content.append(node)
1284 content.append(node)
1280
1285
1281 link = linkmapper(cs)
1286 link = linkmapper(cs)
1282 if node in self.nodemap:
1287 if node in self.nodemap:
1283 # this can happen if two branches make the same change
1288 # this can happen if two branches make the same change
1284 chain = node
1289 chain = node
1285 continue
1290 continue
1286
1291
1287 for p in (p1, p2):
1292 for p in (p1, p2):
1288 if p not in self.nodemap:
1293 if p not in self.nodemap:
1289 raise LookupError(p, self.indexfile,
1294 raise LookupError(p, self.indexfile,
1290 _('unknown parent'))
1295 _('unknown parent'))
1291
1296
1292 if deltabase not in self.nodemap:
1297 if deltabase not in self.nodemap:
1293 raise LookupError(deltabase, self.indexfile,
1298 raise LookupError(deltabase, self.indexfile,
1294 _('unknown delta base'))
1299 _('unknown delta base'))
1295
1300
1296 baserev = self.rev(deltabase)
1301 baserev = self.rev(deltabase)
1297 chain = self._addrevision(node, None, transaction, link,
1302 chain = self._addrevision(node, None, transaction, link,
1298 p1, p2, (baserev, delta), ifh, dfh)
1303 p1, p2, (baserev, delta), ifh, dfh)
1299 if not dfh and not self._inline:
1304 if not dfh and not self._inline:
1300 # addrevision switched from inline to conventional
1305 # addrevision switched from inline to conventional
1301 # reopen the index
1306 # reopen the index
1302 ifh.close()
1307 ifh.close()
1303 dfh = self.opener(self.datafile, "a")
1308 dfh = self.opener(self.datafile, "a")
1304 ifh = self.opener(self.indexfile, "a")
1309 ifh = self.opener(self.indexfile, "a")
1305 finally:
1310 finally:
1306 if dfh:
1311 if dfh:
1307 dfh.close()
1312 dfh.close()
1308 ifh.close()
1313 ifh.close()
1309
1314
1310 return content
1315 return content
1311
1316
1312 def getstrippoint(self, minlink):
1317 def getstrippoint(self, minlink):
1313 """find the minimum rev that must be stripped to strip the linkrev
1318 """find the minimum rev that must be stripped to strip the linkrev
1314
1319
1315 Returns a tuple containing the minimum rev and a set of all revs that
1320 Returns a tuple containing the minimum rev and a set of all revs that
1316 have linkrevs that will be broken by this strip.
1321 have linkrevs that will be broken by this strip.
1317 """
1322 """
1318 brokenrevs = set()
1323 brokenrevs = set()
1319 strippoint = len(self)
1324 strippoint = len(self)
1320
1325
1321 heads = {}
1326 heads = {}
1322 futurelargelinkrevs = set()
1327 futurelargelinkrevs = set()
1323 for head in self.headrevs():
1328 for head in self.headrevs():
1324 headlinkrev = self.linkrev(head)
1329 headlinkrev = self.linkrev(head)
1325 heads[head] = headlinkrev
1330 heads[head] = headlinkrev
1326 if headlinkrev >= minlink:
1331 if headlinkrev >= minlink:
1327 futurelargelinkrevs.add(headlinkrev)
1332 futurelargelinkrevs.add(headlinkrev)
1328
1333
1329 # This algorithm involves walking down the rev graph, starting at the
1334 # This algorithm involves walking down the rev graph, starting at the
1330 # heads. Since the revs are topologically sorted according to linkrev,
1335 # heads. Since the revs are topologically sorted according to linkrev,
1331 # once all head linkrevs are below the minlink, we know there are
1336 # once all head linkrevs are below the minlink, we know there are
1332 # no more revs that could have a linkrev greater than minlink.
1337 # no more revs that could have a linkrev greater than minlink.
1333 # So we can stop walking.
1338 # So we can stop walking.
1334 while futurelargelinkrevs:
1339 while futurelargelinkrevs:
1335 strippoint -= 1
1340 strippoint -= 1
1336 linkrev = heads.pop(strippoint)
1341 linkrev = heads.pop(strippoint)
1337
1342
1338 if linkrev < minlink:
1343 if linkrev < minlink:
1339 brokenrevs.add(strippoint)
1344 brokenrevs.add(strippoint)
1340 else:
1345 else:
1341 futurelargelinkrevs.remove(linkrev)
1346 futurelargelinkrevs.remove(linkrev)
1342
1347
1343 for p in self.parentrevs(strippoint):
1348 for p in self.parentrevs(strippoint):
1344 if p != nullrev:
1349 if p != nullrev:
1345 plinkrev = self.linkrev(p)
1350 plinkrev = self.linkrev(p)
1346 heads[p] = plinkrev
1351 heads[p] = plinkrev
1347 if plinkrev >= minlink:
1352 if plinkrev >= minlink:
1348 futurelargelinkrevs.add(plinkrev)
1353 futurelargelinkrevs.add(plinkrev)
1349
1354
1350 return strippoint, brokenrevs
1355 return strippoint, brokenrevs
1351
1356
1352 def strip(self, minlink, transaction):
1357 def strip(self, minlink, transaction):
1353 """truncate the revlog on the first revision with a linkrev >= minlink
1358 """truncate the revlog on the first revision with a linkrev >= minlink
1354
1359
1355 This function is called when we're stripping revision minlink and
1360 This function is called when we're stripping revision minlink and
1356 its descendants from the repository.
1361 its descendants from the repository.
1357
1362
1358 We have to remove all revisions with linkrev >= minlink, because
1363 We have to remove all revisions with linkrev >= minlink, because
1359 the equivalent changelog revisions will be renumbered after the
1364 the equivalent changelog revisions will be renumbered after the
1360 strip.
1365 strip.
1361
1366
1362 So we truncate the revlog on the first of these revisions, and
1367 So we truncate the revlog on the first of these revisions, and
1363 trust that the caller has saved the revisions that shouldn't be
1368 trust that the caller has saved the revisions that shouldn't be
1364 removed and that it'll re-add them after this truncation.
1369 removed and that it'll re-add them after this truncation.
1365 """
1370 """
1366 if len(self) == 0:
1371 if len(self) == 0:
1367 return
1372 return
1368
1373
1369 rev, _ = self.getstrippoint(minlink)
1374 rev, _ = self.getstrippoint(minlink)
1370 if rev == len(self):
1375 if rev == len(self):
1371 return
1376 return
1372
1377
1373 # first truncate the files on disk
1378 # first truncate the files on disk
1374 end = self.start(rev)
1379 end = self.start(rev)
1375 if not self._inline:
1380 if not self._inline:
1376 transaction.add(self.datafile, end)
1381 transaction.add(self.datafile, end)
1377 end = rev * self._io.size
1382 end = rev * self._io.size
1378 else:
1383 else:
1379 end += rev * self._io.size
1384 end += rev * self._io.size
1380
1385
1381 transaction.add(self.indexfile, end)
1386 transaction.add(self.indexfile, end)
1382
1387
1383 # then reset internal state in memory to forget those revisions
1388 # then reset internal state in memory to forget those revisions
1384 self._cache = None
1389 self._cache = None
1385 self._chunkclear()
1390 self._chunkclear()
1386 for x in xrange(rev, len(self)):
1391 for x in xrange(rev, len(self)):
1387 del self.nodemap[self.node(x)]
1392 del self.nodemap[self.node(x)]
1388
1393
1389 del self.index[rev:-1]
1394 del self.index[rev:-1]
1390
1395
1391 def checksize(self):
1396 def checksize(self):
1392 expected = 0
1397 expected = 0
1393 if len(self):
1398 if len(self):
1394 expected = max(0, self.end(len(self) - 1))
1399 expected = max(0, self.end(len(self) - 1))
1395
1400
1396 try:
1401 try:
1397 f = self.opener(self.datafile)
1402 f = self.opener(self.datafile)
1398 f.seek(0, 2)
1403 f.seek(0, 2)
1399 actual = f.tell()
1404 actual = f.tell()
1400 f.close()
1405 f.close()
1401 dd = actual - expected
1406 dd = actual - expected
1402 except IOError, inst:
1407 except IOError, inst:
1403 if inst.errno != errno.ENOENT:
1408 if inst.errno != errno.ENOENT:
1404 raise
1409 raise
1405 dd = 0
1410 dd = 0
1406
1411
1407 try:
1412 try:
1408 f = self.opener(self.indexfile)
1413 f = self.opener(self.indexfile)
1409 f.seek(0, 2)
1414 f.seek(0, 2)
1410 actual = f.tell()
1415 actual = f.tell()
1411 f.close()
1416 f.close()
1412 s = self._io.size
1417 s = self._io.size
1413 i = max(0, actual // s)
1418 i = max(0, actual // s)
1414 di = actual - (i * s)
1419 di = actual - (i * s)
1415 if self._inline:
1420 if self._inline:
1416 databytes = 0
1421 databytes = 0
1417 for r in self:
1422 for r in self:
1418 databytes += max(0, self.length(r))
1423 databytes += max(0, self.length(r))
1419 dd = 0
1424 dd = 0
1420 di = actual - len(self) * s - databytes
1425 di = actual - len(self) * s - databytes
1421 except IOError, inst:
1426 except IOError, inst:
1422 if inst.errno != errno.ENOENT:
1427 if inst.errno != errno.ENOENT:
1423 raise
1428 raise
1424 di = 0
1429 di = 0
1425
1430
1426 return (dd, di)
1431 return (dd, di)
1427
1432
1428 def files(self):
1433 def files(self):
1429 res = [self.indexfile]
1434 res = [self.indexfile]
1430 if not self._inline:
1435 if not self._inline:
1431 res.append(self.datafile)
1436 res.append(self.datafile)
1432 return res
1437 return res
General Comments 0
You need to be logged in to leave comments. Login now