##// END OF EJS Templates
revlog.revision: fix cache preload for inline revlogs...
Siddharth Agarwal -
r19714:0e07c0b5 default
parent child Browse files
Show More
@@ -1,1340 +1,1342 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 = (0, 0)
203 self._basecache = (0, 0)
204 self._chunkcache = (0, '')
204 self._chunkcache = (0, '')
205 self.index = []
205 self.index = []
206 self._pcache = {}
206 self._pcache = {}
207 self._nodecache = {nullid: nullrev}
207 self._nodecache = {nullid: nullrev}
208 self._nodepos = None
208 self._nodepos = None
209
209
210 v = REVLOG_DEFAULT_VERSION
210 v = REVLOG_DEFAULT_VERSION
211 opts = getattr(opener, 'options', None)
211 opts = getattr(opener, 'options', None)
212 if opts is not None:
212 if opts is not None:
213 if 'revlogv1' in opts:
213 if 'revlogv1' in opts:
214 if 'generaldelta' in opts:
214 if 'generaldelta' in opts:
215 v |= REVLOGGENERALDELTA
215 v |= REVLOGGENERALDELTA
216 else:
216 else:
217 v = 0
217 v = 0
218
218
219 i = ''
219 i = ''
220 self._initempty = True
220 self._initempty = True
221 try:
221 try:
222 f = self.opener(self.indexfile)
222 f = self.opener(self.indexfile)
223 i = f.read()
223 i = f.read()
224 f.close()
224 f.close()
225 if len(i) > 0:
225 if len(i) > 0:
226 v = struct.unpack(versionformat, i[:4])[0]
226 v = struct.unpack(versionformat, i[:4])[0]
227 self._initempty = False
227 self._initempty = False
228 except IOError, inst:
228 except IOError, inst:
229 if inst.errno != errno.ENOENT:
229 if inst.errno != errno.ENOENT:
230 raise
230 raise
231
231
232 self.version = v
232 self.version = v
233 self._inline = v & REVLOGNGINLINEDATA
233 self._inline = v & REVLOGNGINLINEDATA
234 self._generaldelta = v & REVLOGGENERALDELTA
234 self._generaldelta = v & REVLOGGENERALDELTA
235 flags = v & ~0xFFFF
235 flags = v & ~0xFFFF
236 fmt = v & 0xFFFF
236 fmt = v & 0xFFFF
237 if fmt == REVLOGV0 and flags:
237 if fmt == REVLOGV0 and flags:
238 raise RevlogError(_("index %s unknown flags %#04x for format v0")
238 raise RevlogError(_("index %s unknown flags %#04x for format v0")
239 % (self.indexfile, flags >> 16))
239 % (self.indexfile, flags >> 16))
240 elif fmt == REVLOGNG and flags & ~REVLOGNG_FLAGS:
240 elif fmt == REVLOGNG and flags & ~REVLOGNG_FLAGS:
241 raise RevlogError(_("index %s unknown flags %#04x for revlogng")
241 raise RevlogError(_("index %s unknown flags %#04x for revlogng")
242 % (self.indexfile, flags >> 16))
242 % (self.indexfile, flags >> 16))
243 elif fmt > REVLOGNG:
243 elif fmt > REVLOGNG:
244 raise RevlogError(_("index %s unknown format %d")
244 raise RevlogError(_("index %s unknown format %d")
245 % (self.indexfile, fmt))
245 % (self.indexfile, fmt))
246
246
247 self._io = revlogio()
247 self._io = revlogio()
248 if self.version == REVLOGV0:
248 if self.version == REVLOGV0:
249 self._io = revlogoldio()
249 self._io = revlogoldio()
250 try:
250 try:
251 d = self._io.parseindex(i, self._inline)
251 d = self._io.parseindex(i, self._inline)
252 except (ValueError, IndexError):
252 except (ValueError, IndexError):
253 raise RevlogError(_("index %s is corrupted") % (self.indexfile))
253 raise RevlogError(_("index %s is corrupted") % (self.indexfile))
254 self.index, nodemap, self._chunkcache = d
254 self.index, nodemap, self._chunkcache = d
255 if nodemap is not None:
255 if nodemap is not None:
256 self.nodemap = self._nodecache = nodemap
256 self.nodemap = self._nodecache = nodemap
257 if not self._chunkcache:
257 if not self._chunkcache:
258 self._chunkclear()
258 self._chunkclear()
259
259
260 def tip(self):
260 def tip(self):
261 return self.node(len(self.index) - 2)
261 return self.node(len(self.index) - 2)
262 def __len__(self):
262 def __len__(self):
263 return len(self.index) - 1
263 return len(self.index) - 1
264 def __iter__(self):
264 def __iter__(self):
265 return iter(xrange(len(self)))
265 return iter(xrange(len(self)))
266 def revs(self, start=0, stop=None):
266 def revs(self, start=0, stop=None):
267 """iterate over all rev in this revlog (from start to stop)"""
267 """iterate over all rev in this revlog (from start to stop)"""
268 step = 1
268 step = 1
269 if stop is not None:
269 if stop is not None:
270 if start > stop:
270 if start > stop:
271 step = -1
271 step = -1
272 stop += step
272 stop += step
273 else:
273 else:
274 stop = len(self)
274 stop = len(self)
275 return xrange(start, stop, step)
275 return xrange(start, stop, step)
276
276
277 @util.propertycache
277 @util.propertycache
278 def nodemap(self):
278 def nodemap(self):
279 self.rev(self.node(0))
279 self.rev(self.node(0))
280 return self._nodecache
280 return self._nodecache
281
281
282 def hasnode(self, node):
282 def hasnode(self, node):
283 try:
283 try:
284 self.rev(node)
284 self.rev(node)
285 return True
285 return True
286 except KeyError:
286 except KeyError:
287 return False
287 return False
288
288
289 def clearcaches(self):
289 def clearcaches(self):
290 try:
290 try:
291 self._nodecache.clearcaches()
291 self._nodecache.clearcaches()
292 except AttributeError:
292 except AttributeError:
293 self._nodecache = {nullid: nullrev}
293 self._nodecache = {nullid: nullrev}
294 self._nodepos = None
294 self._nodepos = None
295
295
296 def rev(self, node):
296 def rev(self, node):
297 try:
297 try:
298 return self._nodecache[node]
298 return self._nodecache[node]
299 except RevlogError:
299 except RevlogError:
300 # parsers.c radix tree lookup failed
300 # parsers.c radix tree lookup failed
301 raise LookupError(node, self.indexfile, _('no node'))
301 raise LookupError(node, self.indexfile, _('no node'))
302 except KeyError:
302 except KeyError:
303 # pure python cache lookup failed
303 # pure python cache lookup failed
304 n = self._nodecache
304 n = self._nodecache
305 i = self.index
305 i = self.index
306 p = self._nodepos
306 p = self._nodepos
307 if p is None:
307 if p is None:
308 p = len(i) - 2
308 p = len(i) - 2
309 for r in xrange(p, -1, -1):
309 for r in xrange(p, -1, -1):
310 v = i[r][7]
310 v = i[r][7]
311 n[v] = r
311 n[v] = r
312 if v == node:
312 if v == node:
313 self._nodepos = r - 1
313 self._nodepos = r - 1
314 return r
314 return r
315 raise LookupError(node, self.indexfile, _('no node'))
315 raise LookupError(node, self.indexfile, _('no node'))
316
316
317 def node(self, rev):
317 def node(self, rev):
318 return self.index[rev][7]
318 return self.index[rev][7]
319 def linkrev(self, rev):
319 def linkrev(self, rev):
320 return self.index[rev][4]
320 return self.index[rev][4]
321 def parents(self, node):
321 def parents(self, node):
322 i = self.index
322 i = self.index
323 d = i[self.rev(node)]
323 d = i[self.rev(node)]
324 return i[d[5]][7], i[d[6]][7] # map revisions to nodes inline
324 return i[d[5]][7], i[d[6]][7] # map revisions to nodes inline
325 def parentrevs(self, rev):
325 def parentrevs(self, rev):
326 return self.index[rev][5:7]
326 return self.index[rev][5:7]
327 def start(self, rev):
327 def start(self, rev):
328 return int(self.index[rev][0] >> 16)
328 return int(self.index[rev][0] >> 16)
329 def end(self, rev):
329 def end(self, rev):
330 return self.start(rev) + self.length(rev)
330 return self.start(rev) + self.length(rev)
331 def length(self, rev):
331 def length(self, rev):
332 return self.index[rev][1]
332 return self.index[rev][1]
333 def chainbase(self, rev):
333 def chainbase(self, rev):
334 index = self.index
334 index = self.index
335 base = index[rev][3]
335 base = index[rev][3]
336 while base != rev:
336 while base != rev:
337 rev = base
337 rev = base
338 base = index[rev][3]
338 base = index[rev][3]
339 return base
339 return base
340 def flags(self, rev):
340 def flags(self, rev):
341 return self.index[rev][0] & 0xFFFF
341 return self.index[rev][0] & 0xFFFF
342 def rawsize(self, rev):
342 def rawsize(self, rev):
343 """return the length of the uncompressed text for a given revision"""
343 """return the length of the uncompressed text for a given revision"""
344 l = self.index[rev][2]
344 l = self.index[rev][2]
345 if l >= 0:
345 if l >= 0:
346 return l
346 return l
347
347
348 t = self.revision(self.node(rev))
348 t = self.revision(self.node(rev))
349 return len(t)
349 return len(t)
350 size = rawsize
350 size = rawsize
351
351
352 def ancestors(self, revs, stoprev=0, inclusive=False):
352 def ancestors(self, revs, stoprev=0, inclusive=False):
353 """Generate the ancestors of 'revs' in reverse topological order.
353 """Generate the ancestors of 'revs' in reverse topological order.
354 Does not generate revs lower than stoprev.
354 Does not generate revs lower than stoprev.
355
355
356 See the documentation for ancestor.lazyancestors for more details."""
356 See the documentation for ancestor.lazyancestors for more details."""
357
357
358 return ancestor.lazyancestors(self, revs, stoprev=stoprev,
358 return ancestor.lazyancestors(self, revs, stoprev=stoprev,
359 inclusive=inclusive)
359 inclusive=inclusive)
360
360
361 def descendants(self, revs):
361 def descendants(self, revs):
362 """Generate the descendants of 'revs' in revision order.
362 """Generate the descendants of 'revs' in revision order.
363
363
364 Yield a sequence of revision numbers starting with a child of
364 Yield a sequence of revision numbers starting with a child of
365 some rev in revs, i.e., each revision is *not* considered a
365 some rev in revs, i.e., each revision is *not* considered a
366 descendant of itself. Results are ordered by revision number (a
366 descendant of itself. Results are ordered by revision number (a
367 topological sort)."""
367 topological sort)."""
368 first = min(revs)
368 first = min(revs)
369 if first == nullrev:
369 if first == nullrev:
370 for i in self:
370 for i in self:
371 yield i
371 yield i
372 return
372 return
373
373
374 seen = set(revs)
374 seen = set(revs)
375 for i in self.revs(start=first + 1):
375 for i in self.revs(start=first + 1):
376 for x in self.parentrevs(i):
376 for x in self.parentrevs(i):
377 if x != nullrev and x in seen:
377 if x != nullrev and x in seen:
378 seen.add(i)
378 seen.add(i)
379 yield i
379 yield i
380 break
380 break
381
381
382 def findcommonmissing(self, common=None, heads=None):
382 def findcommonmissing(self, common=None, heads=None):
383 """Return a tuple of the ancestors of common and the ancestors of heads
383 """Return a tuple of the ancestors of common and the ancestors of heads
384 that are not ancestors of common. In revset terminology, we return the
384 that are not ancestors of common. In revset terminology, we return the
385 tuple:
385 tuple:
386
386
387 ::common, (::heads) - (::common)
387 ::common, (::heads) - (::common)
388
388
389 The list is sorted by revision number, meaning it is
389 The list is sorted by revision number, meaning it is
390 topologically sorted.
390 topologically sorted.
391
391
392 'heads' and 'common' are both lists of node IDs. If heads is
392 'heads' and 'common' are both lists of node IDs. If heads is
393 not supplied, uses all of the revlog's heads. If common is not
393 not supplied, uses all of the revlog's heads. If common is not
394 supplied, uses nullid."""
394 supplied, uses nullid."""
395 if common is None:
395 if common is None:
396 common = [nullid]
396 common = [nullid]
397 if heads is None:
397 if heads is None:
398 heads = self.heads()
398 heads = self.heads()
399
399
400 common = [self.rev(n) for n in common]
400 common = [self.rev(n) for n in common]
401 heads = [self.rev(n) for n in heads]
401 heads = [self.rev(n) for n in heads]
402
402
403 # we want the ancestors, but inclusive
403 # we want the ancestors, but inclusive
404 has = set(self.ancestors(common))
404 has = set(self.ancestors(common))
405 has.add(nullrev)
405 has.add(nullrev)
406 has.update(common)
406 has.update(common)
407
407
408 # take all ancestors from heads that aren't in has
408 # take all ancestors from heads that aren't in has
409 missing = set()
409 missing = set()
410 visit = util.deque(r for r in heads if r not in has)
410 visit = util.deque(r for r in heads if r not in has)
411 while visit:
411 while visit:
412 r = visit.popleft()
412 r = visit.popleft()
413 if r in missing:
413 if r in missing:
414 continue
414 continue
415 else:
415 else:
416 missing.add(r)
416 missing.add(r)
417 for p in self.parentrevs(r):
417 for p in self.parentrevs(r):
418 if p not in has:
418 if p not in has:
419 visit.append(p)
419 visit.append(p)
420 missing = list(missing)
420 missing = list(missing)
421 missing.sort()
421 missing.sort()
422 return has, [self.node(r) for r in missing]
422 return has, [self.node(r) for r in missing]
423
423
424 def findmissingrevs(self, common=None, heads=None):
424 def findmissingrevs(self, common=None, heads=None):
425 """Return the revision numbers of the ancestors of heads that
425 """Return the revision numbers of the ancestors of heads that
426 are not ancestors of common.
426 are not ancestors of common.
427
427
428 More specifically, return a list of revision numbers corresponding to
428 More specifically, return a list of revision numbers corresponding to
429 nodes N such that every N satisfies the following constraints:
429 nodes N such that every N satisfies the following constraints:
430
430
431 1. N is an ancestor of some node in 'heads'
431 1. N is an ancestor of some node in 'heads'
432 2. N is not an ancestor of any node in 'common'
432 2. N is not an ancestor of any node in 'common'
433
433
434 The list is sorted by revision number, meaning it is
434 The list is sorted by revision number, meaning it is
435 topologically sorted.
435 topologically sorted.
436
436
437 'heads' and 'common' are both lists of revision numbers. If heads is
437 'heads' and 'common' are both lists of revision numbers. If heads is
438 not supplied, uses all of the revlog's heads. If common is not
438 not supplied, uses all of the revlog's heads. If common is not
439 supplied, uses nullid."""
439 supplied, uses nullid."""
440 if common is None:
440 if common is None:
441 common = [nullrev]
441 common = [nullrev]
442 if heads is None:
442 if heads is None:
443 heads = self.headrevs()
443 heads = self.headrevs()
444
444
445 return ancestor.missingancestors(heads, common, self.parentrevs)
445 return ancestor.missingancestors(heads, common, self.parentrevs)
446
446
447 def findmissing(self, common=None, heads=None):
447 def findmissing(self, common=None, heads=None):
448 """Return the ancestors of heads that are not ancestors of common.
448 """Return the ancestors of heads that are not ancestors of common.
449
449
450 More specifically, return a list of nodes N such that every N
450 More specifically, return a list of nodes N such that every N
451 satisfies the following constraints:
451 satisfies the following constraints:
452
452
453 1. N is an ancestor of some node in 'heads'
453 1. N is an ancestor of some node in 'heads'
454 2. N is not an ancestor of any node in 'common'
454 2. N is not an ancestor of any node in 'common'
455
455
456 The list is sorted by revision number, meaning it is
456 The list is sorted by revision number, meaning it is
457 topologically sorted.
457 topologically sorted.
458
458
459 'heads' and 'common' are both lists of node IDs. If heads is
459 'heads' and 'common' are both lists of node IDs. If heads is
460 not supplied, uses all of the revlog's heads. If common is not
460 not supplied, uses all of the revlog's heads. If common is not
461 supplied, uses nullid."""
461 supplied, uses nullid."""
462 if common is None:
462 if common is None:
463 common = [nullid]
463 common = [nullid]
464 if heads is None:
464 if heads is None:
465 heads = self.heads()
465 heads = self.heads()
466
466
467 common = [self.rev(n) for n in common]
467 common = [self.rev(n) for n in common]
468 heads = [self.rev(n) for n in heads]
468 heads = [self.rev(n) for n in heads]
469
469
470 return [self.node(r) for r in
470 return [self.node(r) for r in
471 ancestor.missingancestors(heads, common, self.parentrevs)]
471 ancestor.missingancestors(heads, common, self.parentrevs)]
472
472
473 def nodesbetween(self, roots=None, heads=None):
473 def nodesbetween(self, roots=None, heads=None):
474 """Return a topological path from 'roots' to 'heads'.
474 """Return a topological path from 'roots' to 'heads'.
475
475
476 Return a tuple (nodes, outroots, outheads) where 'nodes' is a
476 Return a tuple (nodes, outroots, outheads) where 'nodes' is a
477 topologically sorted list of all nodes N that satisfy both of
477 topologically sorted list of all nodes N that satisfy both of
478 these constraints:
478 these constraints:
479
479
480 1. N is a descendant of some node in 'roots'
480 1. N is a descendant of some node in 'roots'
481 2. N is an ancestor of some node in 'heads'
481 2. N is an ancestor of some node in 'heads'
482
482
483 Every node is considered to be both a descendant and an ancestor
483 Every node is considered to be both a descendant and an ancestor
484 of itself, so every reachable node in 'roots' and 'heads' will be
484 of itself, so every reachable node in 'roots' and 'heads' will be
485 included in 'nodes'.
485 included in 'nodes'.
486
486
487 'outroots' is the list of reachable nodes in 'roots', i.e., the
487 'outroots' is the list of reachable nodes in 'roots', i.e., the
488 subset of 'roots' that is returned in 'nodes'. Likewise,
488 subset of 'roots' that is returned in 'nodes'. Likewise,
489 'outheads' is the subset of 'heads' that is also in 'nodes'.
489 'outheads' is the subset of 'heads' that is also in 'nodes'.
490
490
491 'roots' and 'heads' are both lists of node IDs. If 'roots' is
491 'roots' and 'heads' are both lists of node IDs. If 'roots' is
492 unspecified, uses nullid as the only root. If 'heads' is
492 unspecified, uses nullid as the only root. If 'heads' is
493 unspecified, uses list of all of the revlog's heads."""
493 unspecified, uses list of all of the revlog's heads."""
494 nonodes = ([], [], [])
494 nonodes = ([], [], [])
495 if roots is not None:
495 if roots is not None:
496 roots = list(roots)
496 roots = list(roots)
497 if not roots:
497 if not roots:
498 return nonodes
498 return nonodes
499 lowestrev = min([self.rev(n) for n in roots])
499 lowestrev = min([self.rev(n) for n in roots])
500 else:
500 else:
501 roots = [nullid] # Everybody's a descendant of nullid
501 roots = [nullid] # Everybody's a descendant of nullid
502 lowestrev = nullrev
502 lowestrev = nullrev
503 if (lowestrev == nullrev) and (heads is None):
503 if (lowestrev == nullrev) and (heads is None):
504 # We want _all_ the nodes!
504 # We want _all_ the nodes!
505 return ([self.node(r) for r in self], [nullid], list(self.heads()))
505 return ([self.node(r) for r in self], [nullid], list(self.heads()))
506 if heads is None:
506 if heads is None:
507 # All nodes are ancestors, so the latest ancestor is the last
507 # All nodes are ancestors, so the latest ancestor is the last
508 # node.
508 # node.
509 highestrev = len(self) - 1
509 highestrev = len(self) - 1
510 # Set ancestors to None to signal that every node is an ancestor.
510 # Set ancestors to None to signal that every node is an ancestor.
511 ancestors = None
511 ancestors = None
512 # Set heads to an empty dictionary for later discovery of heads
512 # Set heads to an empty dictionary for later discovery of heads
513 heads = {}
513 heads = {}
514 else:
514 else:
515 heads = list(heads)
515 heads = list(heads)
516 if not heads:
516 if not heads:
517 return nonodes
517 return nonodes
518 ancestors = set()
518 ancestors = set()
519 # Turn heads into a dictionary so we can remove 'fake' heads.
519 # Turn heads into a dictionary so we can remove 'fake' heads.
520 # Also, later we will be using it to filter out the heads we can't
520 # Also, later we will be using it to filter out the heads we can't
521 # find from roots.
521 # find from roots.
522 heads = dict.fromkeys(heads, False)
522 heads = dict.fromkeys(heads, False)
523 # Start at the top and keep marking parents until we're done.
523 # Start at the top and keep marking parents until we're done.
524 nodestotag = set(heads)
524 nodestotag = set(heads)
525 # Remember where the top was so we can use it as a limit later.
525 # Remember where the top was so we can use it as a limit later.
526 highestrev = max([self.rev(n) for n in nodestotag])
526 highestrev = max([self.rev(n) for n in nodestotag])
527 while nodestotag:
527 while nodestotag:
528 # grab a node to tag
528 # grab a node to tag
529 n = nodestotag.pop()
529 n = nodestotag.pop()
530 # Never tag nullid
530 # Never tag nullid
531 if n == nullid:
531 if n == nullid:
532 continue
532 continue
533 # A node's revision number represents its place in a
533 # A node's revision number represents its place in a
534 # topologically sorted list of nodes.
534 # topologically sorted list of nodes.
535 r = self.rev(n)
535 r = self.rev(n)
536 if r >= lowestrev:
536 if r >= lowestrev:
537 if n not in ancestors:
537 if n not in ancestors:
538 # If we are possibly a descendant of one of the roots
538 # If we are possibly a descendant of one of the roots
539 # and we haven't already been marked as an ancestor
539 # and we haven't already been marked as an ancestor
540 ancestors.add(n) # Mark as ancestor
540 ancestors.add(n) # Mark as ancestor
541 # Add non-nullid parents to list of nodes to tag.
541 # Add non-nullid parents to list of nodes to tag.
542 nodestotag.update([p for p in self.parents(n) if
542 nodestotag.update([p for p in self.parents(n) if
543 p != nullid])
543 p != nullid])
544 elif n in heads: # We've seen it before, is it a fake head?
544 elif n in heads: # We've seen it before, is it a fake head?
545 # So it is, real heads should not be the ancestors of
545 # So it is, real heads should not be the ancestors of
546 # any other heads.
546 # any other heads.
547 heads.pop(n)
547 heads.pop(n)
548 if not ancestors:
548 if not ancestors:
549 return nonodes
549 return nonodes
550 # Now that we have our set of ancestors, we want to remove any
550 # Now that we have our set of ancestors, we want to remove any
551 # roots that are not ancestors.
551 # roots that are not ancestors.
552
552
553 # If one of the roots was nullid, everything is included anyway.
553 # If one of the roots was nullid, everything is included anyway.
554 if lowestrev > nullrev:
554 if lowestrev > nullrev:
555 # But, since we weren't, let's recompute the lowest rev to not
555 # But, since we weren't, let's recompute the lowest rev to not
556 # include roots that aren't ancestors.
556 # include roots that aren't ancestors.
557
557
558 # Filter out roots that aren't ancestors of heads
558 # Filter out roots that aren't ancestors of heads
559 roots = [n for n in roots if n in ancestors]
559 roots = [n for n in roots if n in ancestors]
560 # Recompute the lowest revision
560 # Recompute the lowest revision
561 if roots:
561 if roots:
562 lowestrev = min([self.rev(n) for n in roots])
562 lowestrev = min([self.rev(n) for n in roots])
563 else:
563 else:
564 # No more roots? Return empty list
564 # No more roots? Return empty list
565 return nonodes
565 return nonodes
566 else:
566 else:
567 # We are descending from nullid, and don't need to care about
567 # We are descending from nullid, and don't need to care about
568 # any other roots.
568 # any other roots.
569 lowestrev = nullrev
569 lowestrev = nullrev
570 roots = [nullid]
570 roots = [nullid]
571 # Transform our roots list into a set.
571 # Transform our roots list into a set.
572 descendants = set(roots)
572 descendants = set(roots)
573 # Also, keep the original roots so we can filter out roots that aren't
573 # Also, keep the original roots so we can filter out roots that aren't
574 # 'real' roots (i.e. are descended from other roots).
574 # 'real' roots (i.e. are descended from other roots).
575 roots = descendants.copy()
575 roots = descendants.copy()
576 # Our topologically sorted list of output nodes.
576 # Our topologically sorted list of output nodes.
577 orderedout = []
577 orderedout = []
578 # Don't start at nullid since we don't want nullid in our output list,
578 # Don't start at nullid since we don't want nullid in our output list,
579 # and if nullid shows up in descendants, empty parents will look like
579 # and if nullid shows up in descendants, empty parents will look like
580 # they're descendants.
580 # they're descendants.
581 for r in self.revs(start=max(lowestrev, 0), stop=highestrev + 1):
581 for r in self.revs(start=max(lowestrev, 0), stop=highestrev + 1):
582 n = self.node(r)
582 n = self.node(r)
583 isdescendant = False
583 isdescendant = False
584 if lowestrev == nullrev: # Everybody is a descendant of nullid
584 if lowestrev == nullrev: # Everybody is a descendant of nullid
585 isdescendant = True
585 isdescendant = True
586 elif n in descendants:
586 elif n in descendants:
587 # n is already a descendant
587 # n is already a descendant
588 isdescendant = True
588 isdescendant = True
589 # This check only needs to be done here because all the roots
589 # This check only needs to be done here because all the roots
590 # will start being marked is descendants before the loop.
590 # will start being marked is descendants before the loop.
591 if n in roots:
591 if n in roots:
592 # If n was a root, check if it's a 'real' root.
592 # If n was a root, check if it's a 'real' root.
593 p = tuple(self.parents(n))
593 p = tuple(self.parents(n))
594 # If any of its parents are descendants, it's not a root.
594 # If any of its parents are descendants, it's not a root.
595 if (p[0] in descendants) or (p[1] in descendants):
595 if (p[0] in descendants) or (p[1] in descendants):
596 roots.remove(n)
596 roots.remove(n)
597 else:
597 else:
598 p = tuple(self.parents(n))
598 p = tuple(self.parents(n))
599 # A node is a descendant if either of its parents are
599 # A node is a descendant if either of its parents are
600 # descendants. (We seeded the dependents list with the roots
600 # descendants. (We seeded the dependents list with the roots
601 # up there, remember?)
601 # up there, remember?)
602 if (p[0] in descendants) or (p[1] in descendants):
602 if (p[0] in descendants) or (p[1] in descendants):
603 descendants.add(n)
603 descendants.add(n)
604 isdescendant = True
604 isdescendant = True
605 if isdescendant and ((ancestors is None) or (n in ancestors)):
605 if isdescendant and ((ancestors is None) or (n in ancestors)):
606 # Only include nodes that are both descendants and ancestors.
606 # Only include nodes that are both descendants and ancestors.
607 orderedout.append(n)
607 orderedout.append(n)
608 if (ancestors is not None) and (n in heads):
608 if (ancestors is not None) and (n in heads):
609 # We're trying to figure out which heads are reachable
609 # We're trying to figure out which heads are reachable
610 # from roots.
610 # from roots.
611 # Mark this head as having been reached
611 # Mark this head as having been reached
612 heads[n] = True
612 heads[n] = True
613 elif ancestors is None:
613 elif ancestors is None:
614 # Otherwise, we're trying to discover the heads.
614 # Otherwise, we're trying to discover the heads.
615 # Assume this is a head because if it isn't, the next step
615 # Assume this is a head because if it isn't, the next step
616 # will eventually remove it.
616 # will eventually remove it.
617 heads[n] = True
617 heads[n] = True
618 # But, obviously its parents aren't.
618 # But, obviously its parents aren't.
619 for p in self.parents(n):
619 for p in self.parents(n):
620 heads.pop(p, None)
620 heads.pop(p, None)
621 heads = [n for n, flag in heads.iteritems() if flag]
621 heads = [n for n, flag in heads.iteritems() if flag]
622 roots = list(roots)
622 roots = list(roots)
623 assert orderedout
623 assert orderedout
624 assert roots
624 assert roots
625 assert heads
625 assert heads
626 return (orderedout, roots, heads)
626 return (orderedout, roots, heads)
627
627
628 def headrevs(self):
628 def headrevs(self):
629 try:
629 try:
630 return self.index.headrevs()
630 return self.index.headrevs()
631 except AttributeError:
631 except AttributeError:
632 return self._headrevs()
632 return self._headrevs()
633
633
634 def _headrevs(self):
634 def _headrevs(self):
635 count = len(self)
635 count = len(self)
636 if not count:
636 if not count:
637 return [nullrev]
637 return [nullrev]
638 # we won't iter over filtered rev so nobody is a head at start
638 # we won't iter over filtered rev so nobody is a head at start
639 ishead = [0] * (count + 1)
639 ishead = [0] * (count + 1)
640 index = self.index
640 index = self.index
641 for r in self:
641 for r in self:
642 ishead[r] = 1 # I may be an head
642 ishead[r] = 1 # I may be an head
643 e = index[r]
643 e = index[r]
644 ishead[e[5]] = ishead[e[6]] = 0 # my parent are not
644 ishead[e[5]] = ishead[e[6]] = 0 # my parent are not
645 return [r for r, val in enumerate(ishead) if val]
645 return [r for r, val in enumerate(ishead) if val]
646
646
647 def heads(self, start=None, stop=None):
647 def heads(self, start=None, stop=None):
648 """return the list of all nodes that have no children
648 """return the list of all nodes that have no children
649
649
650 if start is specified, only heads that are descendants of
650 if start is specified, only heads that are descendants of
651 start will be returned
651 start will be returned
652 if stop is specified, it will consider all the revs from stop
652 if stop is specified, it will consider all the revs from stop
653 as if they had no children
653 as if they had no children
654 """
654 """
655 if start is None and stop is None:
655 if start is None and stop is None:
656 if not len(self):
656 if not len(self):
657 return [nullid]
657 return [nullid]
658 return [self.node(r) for r in self.headrevs()]
658 return [self.node(r) for r in self.headrevs()]
659
659
660 if start is None:
660 if start is None:
661 start = nullid
661 start = nullid
662 if stop is None:
662 if stop is None:
663 stop = []
663 stop = []
664 stoprevs = set([self.rev(n) for n in stop])
664 stoprevs = set([self.rev(n) for n in stop])
665 startrev = self.rev(start)
665 startrev = self.rev(start)
666 reachable = set((startrev,))
666 reachable = set((startrev,))
667 heads = set((startrev,))
667 heads = set((startrev,))
668
668
669 parentrevs = self.parentrevs
669 parentrevs = self.parentrevs
670 for r in self.revs(start=startrev + 1):
670 for r in self.revs(start=startrev + 1):
671 for p in parentrevs(r):
671 for p in parentrevs(r):
672 if p in reachable:
672 if p in reachable:
673 if r not in stoprevs:
673 if r not in stoprevs:
674 reachable.add(r)
674 reachable.add(r)
675 heads.add(r)
675 heads.add(r)
676 if p in heads and p not in stoprevs:
676 if p in heads and p not in stoprevs:
677 heads.remove(p)
677 heads.remove(p)
678
678
679 return [self.node(r) for r in heads]
679 return [self.node(r) for r in heads]
680
680
681 def children(self, node):
681 def children(self, node):
682 """find the children of a given node"""
682 """find the children of a given node"""
683 c = []
683 c = []
684 p = self.rev(node)
684 p = self.rev(node)
685 for r in self.revs(start=p + 1):
685 for r in self.revs(start=p + 1):
686 prevs = [pr for pr in self.parentrevs(r) if pr != nullrev]
686 prevs = [pr for pr in self.parentrevs(r) if pr != nullrev]
687 if prevs:
687 if prevs:
688 for pr in prevs:
688 for pr in prevs:
689 if pr == p:
689 if pr == p:
690 c.append(self.node(r))
690 c.append(self.node(r))
691 elif p == nullrev:
691 elif p == nullrev:
692 c.append(self.node(r))
692 c.append(self.node(r))
693 return c
693 return c
694
694
695 def descendant(self, start, end):
695 def descendant(self, start, end):
696 if start == nullrev:
696 if start == nullrev:
697 return True
697 return True
698 for i in self.descendants([start]):
698 for i in self.descendants([start]):
699 if i == end:
699 if i == end:
700 return True
700 return True
701 elif i > end:
701 elif i > end:
702 break
702 break
703 return False
703 return False
704
704
705 def ancestor(self, a, b):
705 def ancestor(self, a, b):
706 """calculate the least common ancestor of nodes a and b"""
706 """calculate the least common ancestor of nodes a and b"""
707
707
708 a, b = self.rev(a), self.rev(b)
708 a, b = self.rev(a), self.rev(b)
709 try:
709 try:
710 ancs = self.index.ancestors(a, b)
710 ancs = self.index.ancestors(a, b)
711 except (AttributeError, OverflowError):
711 except (AttributeError, OverflowError):
712 ancs = ancestor.ancestors(self.parentrevs, a, b)
712 ancs = ancestor.ancestors(self.parentrevs, a, b)
713 if ancs:
713 if ancs:
714 # choose a consistent winner when there's a tie
714 # choose a consistent winner when there's a tie
715 return min(map(self.node, ancs))
715 return min(map(self.node, ancs))
716 return nullid
716 return nullid
717
717
718 def _match(self, id):
718 def _match(self, id):
719 if isinstance(id, int):
719 if isinstance(id, int):
720 # rev
720 # rev
721 return self.node(id)
721 return self.node(id)
722 if len(id) == 20:
722 if len(id) == 20:
723 # possibly a binary node
723 # possibly a binary node
724 # odds of a binary node being all hex in ASCII are 1 in 10**25
724 # odds of a binary node being all hex in ASCII are 1 in 10**25
725 try:
725 try:
726 node = id
726 node = id
727 self.rev(node) # quick search the index
727 self.rev(node) # quick search the index
728 return node
728 return node
729 except LookupError:
729 except LookupError:
730 pass # may be partial hex id
730 pass # may be partial hex id
731 try:
731 try:
732 # str(rev)
732 # str(rev)
733 rev = int(id)
733 rev = int(id)
734 if str(rev) != id:
734 if str(rev) != id:
735 raise ValueError
735 raise ValueError
736 if rev < 0:
736 if rev < 0:
737 rev = len(self) + rev
737 rev = len(self) + rev
738 if rev < 0 or rev >= len(self):
738 if rev < 0 or rev >= len(self):
739 raise ValueError
739 raise ValueError
740 return self.node(rev)
740 return self.node(rev)
741 except (ValueError, OverflowError):
741 except (ValueError, OverflowError):
742 pass
742 pass
743 if len(id) == 40:
743 if len(id) == 40:
744 try:
744 try:
745 # a full hex nodeid?
745 # a full hex nodeid?
746 node = bin(id)
746 node = bin(id)
747 self.rev(node)
747 self.rev(node)
748 return node
748 return node
749 except (TypeError, LookupError):
749 except (TypeError, LookupError):
750 pass
750 pass
751
751
752 def _partialmatch(self, id):
752 def _partialmatch(self, id):
753 try:
753 try:
754 n = self.index.partialmatch(id)
754 n = self.index.partialmatch(id)
755 if n and self.hasnode(n):
755 if n and self.hasnode(n):
756 return n
756 return n
757 return None
757 return None
758 except RevlogError:
758 except RevlogError:
759 # parsers.c radix tree lookup gave multiple matches
759 # parsers.c radix tree lookup gave multiple matches
760 # fall through to slow path that filters hidden revisions
760 # fall through to slow path that filters hidden revisions
761 pass
761 pass
762 except (AttributeError, ValueError):
762 except (AttributeError, ValueError):
763 # we are pure python, or key was too short to search radix tree
763 # we are pure python, or key was too short to search radix tree
764 pass
764 pass
765
765
766 if id in self._pcache:
766 if id in self._pcache:
767 return self._pcache[id]
767 return self._pcache[id]
768
768
769 if len(id) < 40:
769 if len(id) < 40:
770 try:
770 try:
771 # hex(node)[:...]
771 # hex(node)[:...]
772 l = len(id) // 2 # grab an even number of digits
772 l = len(id) // 2 # grab an even number of digits
773 prefix = bin(id[:l * 2])
773 prefix = bin(id[:l * 2])
774 nl = [e[7] for e in self.index if e[7].startswith(prefix)]
774 nl = [e[7] for e in self.index if e[7].startswith(prefix)]
775 nl = [n for n in nl if hex(n).startswith(id) and
775 nl = [n for n in nl if hex(n).startswith(id) and
776 self.hasnode(n)]
776 self.hasnode(n)]
777 if len(nl) > 0:
777 if len(nl) > 0:
778 if len(nl) == 1:
778 if len(nl) == 1:
779 self._pcache[id] = nl[0]
779 self._pcache[id] = nl[0]
780 return nl[0]
780 return nl[0]
781 raise LookupError(id, self.indexfile,
781 raise LookupError(id, self.indexfile,
782 _('ambiguous identifier'))
782 _('ambiguous identifier'))
783 return None
783 return None
784 except TypeError:
784 except TypeError:
785 pass
785 pass
786
786
787 def lookup(self, id):
787 def lookup(self, id):
788 """locate a node based on:
788 """locate a node based on:
789 - revision number or str(revision number)
789 - revision number or str(revision number)
790 - nodeid or subset of hex nodeid
790 - nodeid or subset of hex nodeid
791 """
791 """
792 n = self._match(id)
792 n = self._match(id)
793 if n is not None:
793 if n is not None:
794 return n
794 return n
795 n = self._partialmatch(id)
795 n = self._partialmatch(id)
796 if n:
796 if n:
797 return n
797 return n
798
798
799 raise LookupError(id, self.indexfile, _('no match found'))
799 raise LookupError(id, self.indexfile, _('no match found'))
800
800
801 def cmp(self, node, text):
801 def cmp(self, node, text):
802 """compare text with a given file revision
802 """compare text with a given file revision
803
803
804 returns True if text is different than what is stored.
804 returns True if text is different than what is stored.
805 """
805 """
806 p1, p2 = self.parents(node)
806 p1, p2 = self.parents(node)
807 return hash(text, p1, p2) != node
807 return hash(text, p1, p2) != node
808
808
809 def _addchunk(self, offset, data):
809 def _addchunk(self, offset, data):
810 o, d = self._chunkcache
810 o, d = self._chunkcache
811 # try to add to existing cache
811 # try to add to existing cache
812 if o + len(d) == offset and len(d) + len(data) < _chunksize:
812 if o + len(d) == offset and len(d) + len(data) < _chunksize:
813 self._chunkcache = o, d + data
813 self._chunkcache = o, d + data
814 else:
814 else:
815 self._chunkcache = offset, data
815 self._chunkcache = offset, data
816
816
817 def _loadchunk(self, offset, length):
817 def _loadchunk(self, offset, length):
818 if self._inline:
818 if self._inline:
819 df = self.opener(self.indexfile)
819 df = self.opener(self.indexfile)
820 else:
820 else:
821 df = self.opener(self.datafile)
821 df = self.opener(self.datafile)
822
822
823 readahead = max(65536, length)
823 readahead = max(65536, length)
824 df.seek(offset)
824 df.seek(offset)
825 d = df.read(readahead)
825 d = df.read(readahead)
826 df.close()
826 df.close()
827 self._addchunk(offset, d)
827 self._addchunk(offset, d)
828 if readahead > length:
828 if readahead > length:
829 return util.buffer(d, 0, length)
829 return util.buffer(d, 0, length)
830 return d
830 return d
831
831
832 def _getchunk(self, offset, length):
832 def _getchunk(self, offset, length):
833 o, d = self._chunkcache
833 o, d = self._chunkcache
834 l = len(d)
834 l = len(d)
835
835
836 # is it in the cache?
836 # is it in the cache?
837 cachestart = offset - o
837 cachestart = offset - o
838 cacheend = cachestart + length
838 cacheend = cachestart + length
839 if cachestart >= 0 and cacheend <= l:
839 if cachestart >= 0 and cacheend <= l:
840 if cachestart == 0 and cacheend == l:
840 if cachestart == 0 and cacheend == l:
841 return d # avoid a copy
841 return d # avoid a copy
842 return util.buffer(d, cachestart, cacheend - cachestart)
842 return util.buffer(d, cachestart, cacheend - cachestart)
843
843
844 return self._loadchunk(offset, length)
844 return self._loadchunk(offset, length)
845
845
846 def _chunkraw(self, startrev, endrev):
846 def _chunkraw(self, startrev, endrev):
847 start = self.start(startrev)
847 start = self.start(startrev)
848 length = self.end(endrev) - start
848 end = self.end(endrev)
849 if self._inline:
849 if self._inline:
850 start += (startrev + 1) * self._io.size
850 start += (startrev + 1) * self._io.size
851 end += (endrev + 1) * self._io.size
852 length = end - start
851 return self._getchunk(start, length)
853 return self._getchunk(start, length)
852
854
853 def _chunk(self, rev):
855 def _chunk(self, rev):
854 return decompress(self._chunkraw(rev, rev))
856 return decompress(self._chunkraw(rev, rev))
855
857
856 def _chunks(self, revs):
858 def _chunks(self, revs):
857 '''faster version of [self._chunk(rev) for rev in revs]
859 '''faster version of [self._chunk(rev) for rev in revs]
858
860
859 Assumes that revs is in ascending order.'''
861 Assumes that revs is in ascending order.'''
860 start = self.start
862 start = self.start
861 length = self.length
863 length = self.length
862 inline = self._inline
864 inline = self._inline
863 iosize = self._io.size
865 iosize = self._io.size
864 getchunk = self._getchunk
866 getchunk = self._getchunk
865
867
866 l = []
868 l = []
867 ladd = l.append
869 ladd = l.append
868
870
869 for rev in revs:
871 for rev in revs:
870 chunkstart = start(rev)
872 chunkstart = start(rev)
871 if inline:
873 if inline:
872 chunkstart += (rev + 1) * iosize
874 chunkstart += (rev + 1) * iosize
873 chunklength = length(rev)
875 chunklength = length(rev)
874 ladd(decompress(getchunk(chunkstart, chunklength)))
876 ladd(decompress(getchunk(chunkstart, chunklength)))
875
877
876 return l
878 return l
877
879
878 def _chunkbase(self, rev):
880 def _chunkbase(self, rev):
879 return self._chunk(rev)
881 return self._chunk(rev)
880
882
881 def _chunkclear(self):
883 def _chunkclear(self):
882 self._chunkcache = (0, '')
884 self._chunkcache = (0, '')
883
885
884 def deltaparent(self, rev):
886 def deltaparent(self, rev):
885 """return deltaparent of the given revision"""
887 """return deltaparent of the given revision"""
886 base = self.index[rev][3]
888 base = self.index[rev][3]
887 if base == rev:
889 if base == rev:
888 return nullrev
890 return nullrev
889 elif self._generaldelta:
891 elif self._generaldelta:
890 return base
892 return base
891 else:
893 else:
892 return rev - 1
894 return rev - 1
893
895
894 def revdiff(self, rev1, rev2):
896 def revdiff(self, rev1, rev2):
895 """return or calculate a delta between two revisions"""
897 """return or calculate a delta between two revisions"""
896 if rev1 != nullrev and self.deltaparent(rev2) == rev1:
898 if rev1 != nullrev and self.deltaparent(rev2) == rev1:
897 return str(self._chunk(rev2))
899 return str(self._chunk(rev2))
898
900
899 return mdiff.textdiff(self.revision(rev1),
901 return mdiff.textdiff(self.revision(rev1),
900 self.revision(rev2))
902 self.revision(rev2))
901
903
902 def revision(self, nodeorrev):
904 def revision(self, nodeorrev):
903 """return an uncompressed revision of a given node or revision
905 """return an uncompressed revision of a given node or revision
904 number.
906 number.
905 """
907 """
906 if isinstance(nodeorrev, int):
908 if isinstance(nodeorrev, int):
907 rev = nodeorrev
909 rev = nodeorrev
908 node = self.node(rev)
910 node = self.node(rev)
909 else:
911 else:
910 node = nodeorrev
912 node = nodeorrev
911 rev = None
913 rev = None
912
914
913 cachedrev = None
915 cachedrev = None
914 if node == nullid:
916 if node == nullid:
915 return ""
917 return ""
916 if self._cache:
918 if self._cache:
917 if self._cache[0] == node:
919 if self._cache[0] == node:
918 return self._cache[2]
920 return self._cache[2]
919 cachedrev = self._cache[1]
921 cachedrev = self._cache[1]
920
922
921 # look up what we need to read
923 # look up what we need to read
922 text = None
924 text = None
923 if rev is None:
925 if rev is None:
924 rev = self.rev(node)
926 rev = self.rev(node)
925
927
926 # check rev flags
928 # check rev flags
927 if self.flags(rev) & ~REVIDX_KNOWN_FLAGS:
929 if self.flags(rev) & ~REVIDX_KNOWN_FLAGS:
928 raise RevlogError(_('incompatible revision flag %x') %
930 raise RevlogError(_('incompatible revision flag %x') %
929 (self.flags(rev) & ~REVIDX_KNOWN_FLAGS))
931 (self.flags(rev) & ~REVIDX_KNOWN_FLAGS))
930
932
931 # build delta chain
933 # build delta chain
932 chain = []
934 chain = []
933 index = self.index # for performance
935 index = self.index # for performance
934 generaldelta = self._generaldelta
936 generaldelta = self._generaldelta
935 iterrev = rev
937 iterrev = rev
936 e = index[iterrev]
938 e = index[iterrev]
937 while iterrev != e[3] and iterrev != cachedrev:
939 while iterrev != e[3] and iterrev != cachedrev:
938 chain.append(iterrev)
940 chain.append(iterrev)
939 if generaldelta:
941 if generaldelta:
940 iterrev = e[3]
942 iterrev = e[3]
941 else:
943 else:
942 iterrev -= 1
944 iterrev -= 1
943 e = index[iterrev]
945 e = index[iterrev]
944 chain.reverse()
946 chain.reverse()
945 base = iterrev
947 base = iterrev
946
948
947 if iterrev == cachedrev:
949 if iterrev == cachedrev:
948 # cache hit
950 # cache hit
949 text = self._cache[2]
951 text = self._cache[2]
950
952
951 # drop cache to save memory
953 # drop cache to save memory
952 self._cache = None
954 self._cache = None
953
955
954 self._chunkraw(base, rev)
956 self._chunkraw(base, rev)
955 if text is None:
957 if text is None:
956 text = str(self._chunkbase(base))
958 text = str(self._chunkbase(base))
957
959
958 bins = self._chunks(chain)
960 bins = self._chunks(chain)
959 text = mdiff.patches(text, bins)
961 text = mdiff.patches(text, bins)
960
962
961 text = self._checkhash(text, node, rev)
963 text = self._checkhash(text, node, rev)
962
964
963 self._cache = (node, rev, text)
965 self._cache = (node, rev, text)
964 return text
966 return text
965
967
966 def _checkhash(self, text, node, rev):
968 def _checkhash(self, text, node, rev):
967 p1, p2 = self.parents(node)
969 p1, p2 = self.parents(node)
968 self.checkhash(text, p1, p2, node, rev)
970 self.checkhash(text, p1, p2, node, rev)
969 return text
971 return text
970
972
971 def checkhash(self, text, p1, p2, node, rev=None):
973 def checkhash(self, text, p1, p2, node, rev=None):
972 if node != hash(text, p1, p2):
974 if node != hash(text, p1, p2):
973 revornode = rev
975 revornode = rev
974 if revornode is None:
976 if revornode is None:
975 revornode = templatefilters.short(hex(node))
977 revornode = templatefilters.short(hex(node))
976 raise RevlogError(_("integrity check failed on %s:%s")
978 raise RevlogError(_("integrity check failed on %s:%s")
977 % (self.indexfile, revornode))
979 % (self.indexfile, revornode))
978
980
979 def checkinlinesize(self, tr, fp=None):
981 def checkinlinesize(self, tr, fp=None):
980 if not self._inline or (self.start(-2) + self.length(-2)) < _maxinline:
982 if not self._inline or (self.start(-2) + self.length(-2)) < _maxinline:
981 return
983 return
982
984
983 trinfo = tr.find(self.indexfile)
985 trinfo = tr.find(self.indexfile)
984 if trinfo is None:
986 if trinfo is None:
985 raise RevlogError(_("%s not found in the transaction")
987 raise RevlogError(_("%s not found in the transaction")
986 % self.indexfile)
988 % self.indexfile)
987
989
988 trindex = trinfo[2]
990 trindex = trinfo[2]
989 dataoff = self.start(trindex)
991 dataoff = self.start(trindex)
990
992
991 tr.add(self.datafile, dataoff)
993 tr.add(self.datafile, dataoff)
992
994
993 if fp:
995 if fp:
994 fp.flush()
996 fp.flush()
995 fp.close()
997 fp.close()
996
998
997 df = self.opener(self.datafile, 'w')
999 df = self.opener(self.datafile, 'w')
998 try:
1000 try:
999 for r in self:
1001 for r in self:
1000 df.write(self._chunkraw(r, r))
1002 df.write(self._chunkraw(r, r))
1001 finally:
1003 finally:
1002 df.close()
1004 df.close()
1003
1005
1004 fp = self.opener(self.indexfile, 'w', atomictemp=True)
1006 fp = self.opener(self.indexfile, 'w', atomictemp=True)
1005 self.version &= ~(REVLOGNGINLINEDATA)
1007 self.version &= ~(REVLOGNGINLINEDATA)
1006 self._inline = False
1008 self._inline = False
1007 for i in self:
1009 for i in self:
1008 e = self._io.packentry(self.index[i], self.node, self.version, i)
1010 e = self._io.packentry(self.index[i], self.node, self.version, i)
1009 fp.write(e)
1011 fp.write(e)
1010
1012
1011 # if we don't call close, the temp file will never replace the
1013 # if we don't call close, the temp file will never replace the
1012 # real index
1014 # real index
1013 fp.close()
1015 fp.close()
1014
1016
1015 tr.replace(self.indexfile, trindex * self._io.size)
1017 tr.replace(self.indexfile, trindex * self._io.size)
1016 self._chunkclear()
1018 self._chunkclear()
1017
1019
1018 def addrevision(self, text, transaction, link, p1, p2, cachedelta=None,
1020 def addrevision(self, text, transaction, link, p1, p2, cachedelta=None,
1019 node=None):
1021 node=None):
1020 """add a revision to the log
1022 """add a revision to the log
1021
1023
1022 text - the revision data to add
1024 text - the revision data to add
1023 transaction - the transaction object used for rollback
1025 transaction - the transaction object used for rollback
1024 link - the linkrev data to add
1026 link - the linkrev data to add
1025 p1, p2 - the parent nodeids of the revision
1027 p1, p2 - the parent nodeids of the revision
1026 cachedelta - an optional precomputed delta
1028 cachedelta - an optional precomputed delta
1027 node - nodeid of revision; typically node is not specified, and it is
1029 node - nodeid of revision; typically node is not specified, and it is
1028 computed by default as hash(text, p1, p2), however subclasses might
1030 computed by default as hash(text, p1, p2), however subclasses might
1029 use different hashing method (and override checkhash() in such case)
1031 use different hashing method (and override checkhash() in such case)
1030 """
1032 """
1031 if link == nullrev:
1033 if link == nullrev:
1032 raise RevlogError(_("attempted to add linkrev -1 to %s")
1034 raise RevlogError(_("attempted to add linkrev -1 to %s")
1033 % self.indexfile)
1035 % self.indexfile)
1034 node = node or hash(text, p1, p2)
1036 node = node or hash(text, p1, p2)
1035 if node in self.nodemap:
1037 if node in self.nodemap:
1036 return node
1038 return node
1037
1039
1038 dfh = None
1040 dfh = None
1039 if not self._inline:
1041 if not self._inline:
1040 dfh = self.opener(self.datafile, "a")
1042 dfh = self.opener(self.datafile, "a")
1041 ifh = self.opener(self.indexfile, "a+")
1043 ifh = self.opener(self.indexfile, "a+")
1042 try:
1044 try:
1043 return self._addrevision(node, text, transaction, link, p1, p2,
1045 return self._addrevision(node, text, transaction, link, p1, p2,
1044 cachedelta, ifh, dfh)
1046 cachedelta, ifh, dfh)
1045 finally:
1047 finally:
1046 if dfh:
1048 if dfh:
1047 dfh.close()
1049 dfh.close()
1048 ifh.close()
1050 ifh.close()
1049
1051
1050 def compress(self, text):
1052 def compress(self, text):
1051 """ generate a possibly-compressed representation of text """
1053 """ generate a possibly-compressed representation of text """
1052 if not text:
1054 if not text:
1053 return ("", text)
1055 return ("", text)
1054 l = len(text)
1056 l = len(text)
1055 bin = None
1057 bin = None
1056 if l < 44:
1058 if l < 44:
1057 pass
1059 pass
1058 elif l > 1000000:
1060 elif l > 1000000:
1059 # zlib makes an internal copy, thus doubling memory usage for
1061 # zlib makes an internal copy, thus doubling memory usage for
1060 # large files, so lets do this in pieces
1062 # large files, so lets do this in pieces
1061 z = zlib.compressobj()
1063 z = zlib.compressobj()
1062 p = []
1064 p = []
1063 pos = 0
1065 pos = 0
1064 while pos < l:
1066 while pos < l:
1065 pos2 = pos + 2**20
1067 pos2 = pos + 2**20
1066 p.append(z.compress(text[pos:pos2]))
1068 p.append(z.compress(text[pos:pos2]))
1067 pos = pos2
1069 pos = pos2
1068 p.append(z.flush())
1070 p.append(z.flush())
1069 if sum(map(len, p)) < l:
1071 if sum(map(len, p)) < l:
1070 bin = "".join(p)
1072 bin = "".join(p)
1071 else:
1073 else:
1072 bin = _compress(text)
1074 bin = _compress(text)
1073 if bin is None or len(bin) > l:
1075 if bin is None or len(bin) > l:
1074 if text[0] == '\0':
1076 if text[0] == '\0':
1075 return ("", text)
1077 return ("", text)
1076 return ('u', text)
1078 return ('u', text)
1077 return ("", bin)
1079 return ("", bin)
1078
1080
1079 def _addrevision(self, node, text, transaction, link, p1, p2,
1081 def _addrevision(self, node, text, transaction, link, p1, p2,
1080 cachedelta, ifh, dfh):
1082 cachedelta, ifh, dfh):
1081 """internal function to add revisions to the log
1083 """internal function to add revisions to the log
1082
1084
1083 see addrevision for argument descriptions.
1085 see addrevision for argument descriptions.
1084 invariants:
1086 invariants:
1085 - text is optional (can be None); if not set, cachedelta must be set.
1087 - text is optional (can be None); if not set, cachedelta must be set.
1086 if both are set, they must correspond to each other.
1088 if both are set, they must correspond to each other.
1087 """
1089 """
1088 btext = [text]
1090 btext = [text]
1089 def buildtext():
1091 def buildtext():
1090 if btext[0] is not None:
1092 if btext[0] is not None:
1091 return btext[0]
1093 return btext[0]
1092 # flush any pending writes here so we can read it in revision
1094 # flush any pending writes here so we can read it in revision
1093 if dfh:
1095 if dfh:
1094 dfh.flush()
1096 dfh.flush()
1095 ifh.flush()
1097 ifh.flush()
1096 basetext = self.revision(self.node(cachedelta[0]))
1098 basetext = self.revision(self.node(cachedelta[0]))
1097 btext[0] = mdiff.patch(basetext, cachedelta[1])
1099 btext[0] = mdiff.patch(basetext, cachedelta[1])
1098 self.checkhash(btext[0], p1, p2, node)
1100 self.checkhash(btext[0], p1, p2, node)
1099 return btext[0]
1101 return btext[0]
1100
1102
1101 def builddelta(rev):
1103 def builddelta(rev):
1102 # can we use the cached delta?
1104 # can we use the cached delta?
1103 if cachedelta and cachedelta[0] == rev:
1105 if cachedelta and cachedelta[0] == rev:
1104 delta = cachedelta[1]
1106 delta = cachedelta[1]
1105 else:
1107 else:
1106 t = buildtext()
1108 t = buildtext()
1107 ptext = self.revision(self.node(rev))
1109 ptext = self.revision(self.node(rev))
1108 delta = mdiff.textdiff(ptext, t)
1110 delta = mdiff.textdiff(ptext, t)
1109 data = self.compress(delta)
1111 data = self.compress(delta)
1110 l = len(data[1]) + len(data[0])
1112 l = len(data[1]) + len(data[0])
1111 if basecache[0] == rev:
1113 if basecache[0] == rev:
1112 chainbase = basecache[1]
1114 chainbase = basecache[1]
1113 else:
1115 else:
1114 chainbase = self.chainbase(rev)
1116 chainbase = self.chainbase(rev)
1115 dist = l + offset - self.start(chainbase)
1117 dist = l + offset - self.start(chainbase)
1116 if self._generaldelta:
1118 if self._generaldelta:
1117 base = rev
1119 base = rev
1118 else:
1120 else:
1119 base = chainbase
1121 base = chainbase
1120 return dist, l, data, base, chainbase
1122 return dist, l, data, base, chainbase
1121
1123
1122 curr = len(self)
1124 curr = len(self)
1123 prev = curr - 1
1125 prev = curr - 1
1124 base = chainbase = curr
1126 base = chainbase = curr
1125 offset = self.end(prev)
1127 offset = self.end(prev)
1126 flags = 0
1128 flags = 0
1127 d = None
1129 d = None
1128 basecache = self._basecache
1130 basecache = self._basecache
1129 p1r, p2r = self.rev(p1), self.rev(p2)
1131 p1r, p2r = self.rev(p1), self.rev(p2)
1130
1132
1131 # should we try to build a delta?
1133 # should we try to build a delta?
1132 if prev != nullrev:
1134 if prev != nullrev:
1133 if self._generaldelta:
1135 if self._generaldelta:
1134 if p1r >= basecache[1]:
1136 if p1r >= basecache[1]:
1135 d = builddelta(p1r)
1137 d = builddelta(p1r)
1136 elif p2r >= basecache[1]:
1138 elif p2r >= basecache[1]:
1137 d = builddelta(p2r)
1139 d = builddelta(p2r)
1138 else:
1140 else:
1139 d = builddelta(prev)
1141 d = builddelta(prev)
1140 else:
1142 else:
1141 d = builddelta(prev)
1143 d = builddelta(prev)
1142 dist, l, data, base, chainbase = d
1144 dist, l, data, base, chainbase = d
1143
1145
1144 # full versions are inserted when the needed deltas
1146 # full versions are inserted when the needed deltas
1145 # become comparable to the uncompressed text
1147 # become comparable to the uncompressed text
1146 if text is None:
1148 if text is None:
1147 textlen = mdiff.patchedsize(self.rawsize(cachedelta[0]),
1149 textlen = mdiff.patchedsize(self.rawsize(cachedelta[0]),
1148 cachedelta[1])
1150 cachedelta[1])
1149 else:
1151 else:
1150 textlen = len(text)
1152 textlen = len(text)
1151 if d is None or dist > textlen * 2:
1153 if d is None or dist > textlen * 2:
1152 text = buildtext()
1154 text = buildtext()
1153 data = self.compress(text)
1155 data = self.compress(text)
1154 l = len(data[1]) + len(data[0])
1156 l = len(data[1]) + len(data[0])
1155 base = chainbase = curr
1157 base = chainbase = curr
1156
1158
1157 e = (offset_type(offset, flags), l, textlen,
1159 e = (offset_type(offset, flags), l, textlen,
1158 base, link, p1r, p2r, node)
1160 base, link, p1r, p2r, node)
1159 self.index.insert(-1, e)
1161 self.index.insert(-1, e)
1160 self.nodemap[node] = curr
1162 self.nodemap[node] = curr
1161
1163
1162 entry = self._io.packentry(e, self.node, self.version, curr)
1164 entry = self._io.packentry(e, self.node, self.version, curr)
1163 if not self._inline:
1165 if not self._inline:
1164 transaction.add(self.datafile, offset)
1166 transaction.add(self.datafile, offset)
1165 transaction.add(self.indexfile, curr * len(entry))
1167 transaction.add(self.indexfile, curr * len(entry))
1166 if data[0]:
1168 if data[0]:
1167 dfh.write(data[0])
1169 dfh.write(data[0])
1168 dfh.write(data[1])
1170 dfh.write(data[1])
1169 dfh.flush()
1171 dfh.flush()
1170 ifh.write(entry)
1172 ifh.write(entry)
1171 else:
1173 else:
1172 offset += curr * self._io.size
1174 offset += curr * self._io.size
1173 transaction.add(self.indexfile, offset, curr)
1175 transaction.add(self.indexfile, offset, curr)
1174 ifh.write(entry)
1176 ifh.write(entry)
1175 ifh.write(data[0])
1177 ifh.write(data[0])
1176 ifh.write(data[1])
1178 ifh.write(data[1])
1177 self.checkinlinesize(transaction, ifh)
1179 self.checkinlinesize(transaction, ifh)
1178
1180
1179 if type(text) == str: # only accept immutable objects
1181 if type(text) == str: # only accept immutable objects
1180 self._cache = (node, curr, text)
1182 self._cache = (node, curr, text)
1181 self._basecache = (curr, chainbase)
1183 self._basecache = (curr, chainbase)
1182 return node
1184 return node
1183
1185
1184 def addgroup(self, bundle, linkmapper, transaction):
1186 def addgroup(self, bundle, linkmapper, transaction):
1185 """
1187 """
1186 add a delta group
1188 add a delta group
1187
1189
1188 given a set of deltas, add them to the revision log. the
1190 given a set of deltas, add them to the revision log. the
1189 first delta is against its parent, which should be in our
1191 first delta is against its parent, which should be in our
1190 log, the rest are against the previous delta.
1192 log, the rest are against the previous delta.
1191 """
1193 """
1192
1194
1193 # track the base of the current delta log
1195 # track the base of the current delta log
1194 content = []
1196 content = []
1195 node = None
1197 node = None
1196
1198
1197 r = len(self)
1199 r = len(self)
1198 end = 0
1200 end = 0
1199 if r:
1201 if r:
1200 end = self.end(r - 1)
1202 end = self.end(r - 1)
1201 ifh = self.opener(self.indexfile, "a+")
1203 ifh = self.opener(self.indexfile, "a+")
1202 isize = r * self._io.size
1204 isize = r * self._io.size
1203 if self._inline:
1205 if self._inline:
1204 transaction.add(self.indexfile, end + isize, r)
1206 transaction.add(self.indexfile, end + isize, r)
1205 dfh = None
1207 dfh = None
1206 else:
1208 else:
1207 transaction.add(self.indexfile, isize, r)
1209 transaction.add(self.indexfile, isize, r)
1208 transaction.add(self.datafile, end)
1210 transaction.add(self.datafile, end)
1209 dfh = self.opener(self.datafile, "a")
1211 dfh = self.opener(self.datafile, "a")
1210
1212
1211 try:
1213 try:
1212 # loop through our set of deltas
1214 # loop through our set of deltas
1213 chain = None
1215 chain = None
1214 while True:
1216 while True:
1215 chunkdata = bundle.deltachunk(chain)
1217 chunkdata = bundle.deltachunk(chain)
1216 if not chunkdata:
1218 if not chunkdata:
1217 break
1219 break
1218 node = chunkdata['node']
1220 node = chunkdata['node']
1219 p1 = chunkdata['p1']
1221 p1 = chunkdata['p1']
1220 p2 = chunkdata['p2']
1222 p2 = chunkdata['p2']
1221 cs = chunkdata['cs']
1223 cs = chunkdata['cs']
1222 deltabase = chunkdata['deltabase']
1224 deltabase = chunkdata['deltabase']
1223 delta = chunkdata['delta']
1225 delta = chunkdata['delta']
1224
1226
1225 content.append(node)
1227 content.append(node)
1226
1228
1227 link = linkmapper(cs)
1229 link = linkmapper(cs)
1228 if node in self.nodemap:
1230 if node in self.nodemap:
1229 # this can happen if two branches make the same change
1231 # this can happen if two branches make the same change
1230 chain = node
1232 chain = node
1231 continue
1233 continue
1232
1234
1233 for p in (p1, p2):
1235 for p in (p1, p2):
1234 if p not in self.nodemap:
1236 if p not in self.nodemap:
1235 raise LookupError(p, self.indexfile,
1237 raise LookupError(p, self.indexfile,
1236 _('unknown parent'))
1238 _('unknown parent'))
1237
1239
1238 if deltabase not in self.nodemap:
1240 if deltabase not in self.nodemap:
1239 raise LookupError(deltabase, self.indexfile,
1241 raise LookupError(deltabase, self.indexfile,
1240 _('unknown delta base'))
1242 _('unknown delta base'))
1241
1243
1242 baserev = self.rev(deltabase)
1244 baserev = self.rev(deltabase)
1243 chain = self._addrevision(node, None, transaction, link,
1245 chain = self._addrevision(node, None, transaction, link,
1244 p1, p2, (baserev, delta), ifh, dfh)
1246 p1, p2, (baserev, delta), ifh, dfh)
1245 if not dfh and not self._inline:
1247 if not dfh and not self._inline:
1246 # addrevision switched from inline to conventional
1248 # addrevision switched from inline to conventional
1247 # reopen the index
1249 # reopen the index
1248 ifh.close()
1250 ifh.close()
1249 dfh = self.opener(self.datafile, "a")
1251 dfh = self.opener(self.datafile, "a")
1250 ifh = self.opener(self.indexfile, "a")
1252 ifh = self.opener(self.indexfile, "a")
1251 finally:
1253 finally:
1252 if dfh:
1254 if dfh:
1253 dfh.close()
1255 dfh.close()
1254 ifh.close()
1256 ifh.close()
1255
1257
1256 return content
1258 return content
1257
1259
1258 def strip(self, minlink, transaction):
1260 def strip(self, minlink, transaction):
1259 """truncate the revlog on the first revision with a linkrev >= minlink
1261 """truncate the revlog on the first revision with a linkrev >= minlink
1260
1262
1261 This function is called when we're stripping revision minlink and
1263 This function is called when we're stripping revision minlink and
1262 its descendants from the repository.
1264 its descendants from the repository.
1263
1265
1264 We have to remove all revisions with linkrev >= minlink, because
1266 We have to remove all revisions with linkrev >= minlink, because
1265 the equivalent changelog revisions will be renumbered after the
1267 the equivalent changelog revisions will be renumbered after the
1266 strip.
1268 strip.
1267
1269
1268 So we truncate the revlog on the first of these revisions, and
1270 So we truncate the revlog on the first of these revisions, and
1269 trust that the caller has saved the revisions that shouldn't be
1271 trust that the caller has saved the revisions that shouldn't be
1270 removed and that it'll re-add them after this truncation.
1272 removed and that it'll re-add them after this truncation.
1271 """
1273 """
1272 if len(self) == 0:
1274 if len(self) == 0:
1273 return
1275 return
1274
1276
1275 for rev in self:
1277 for rev in self:
1276 if self.index[rev][4] >= minlink:
1278 if self.index[rev][4] >= minlink:
1277 break
1279 break
1278 else:
1280 else:
1279 return
1281 return
1280
1282
1281 # first truncate the files on disk
1283 # first truncate the files on disk
1282 end = self.start(rev)
1284 end = self.start(rev)
1283 if not self._inline:
1285 if not self._inline:
1284 transaction.add(self.datafile, end)
1286 transaction.add(self.datafile, end)
1285 end = rev * self._io.size
1287 end = rev * self._io.size
1286 else:
1288 else:
1287 end += rev * self._io.size
1289 end += rev * self._io.size
1288
1290
1289 transaction.add(self.indexfile, end)
1291 transaction.add(self.indexfile, end)
1290
1292
1291 # then reset internal state in memory to forget those revisions
1293 # then reset internal state in memory to forget those revisions
1292 self._cache = None
1294 self._cache = None
1293 self._chunkclear()
1295 self._chunkclear()
1294 for x in xrange(rev, len(self)):
1296 for x in xrange(rev, len(self)):
1295 del self.nodemap[self.node(x)]
1297 del self.nodemap[self.node(x)]
1296
1298
1297 del self.index[rev:-1]
1299 del self.index[rev:-1]
1298
1300
1299 def checksize(self):
1301 def checksize(self):
1300 expected = 0
1302 expected = 0
1301 if len(self):
1303 if len(self):
1302 expected = max(0, self.end(len(self) - 1))
1304 expected = max(0, self.end(len(self) - 1))
1303
1305
1304 try:
1306 try:
1305 f = self.opener(self.datafile)
1307 f = self.opener(self.datafile)
1306 f.seek(0, 2)
1308 f.seek(0, 2)
1307 actual = f.tell()
1309 actual = f.tell()
1308 f.close()
1310 f.close()
1309 dd = actual - expected
1311 dd = actual - expected
1310 except IOError, inst:
1312 except IOError, inst:
1311 if inst.errno != errno.ENOENT:
1313 if inst.errno != errno.ENOENT:
1312 raise
1314 raise
1313 dd = 0
1315 dd = 0
1314
1316
1315 try:
1317 try:
1316 f = self.opener(self.indexfile)
1318 f = self.opener(self.indexfile)
1317 f.seek(0, 2)
1319 f.seek(0, 2)
1318 actual = f.tell()
1320 actual = f.tell()
1319 f.close()
1321 f.close()
1320 s = self._io.size
1322 s = self._io.size
1321 i = max(0, actual // s)
1323 i = max(0, actual // s)
1322 di = actual - (i * s)
1324 di = actual - (i * s)
1323 if self._inline:
1325 if self._inline:
1324 databytes = 0
1326 databytes = 0
1325 for r in self:
1327 for r in self:
1326 databytes += max(0, self.length(r))
1328 databytes += max(0, self.length(r))
1327 dd = 0
1329 dd = 0
1328 di = actual - len(self) * s - databytes
1330 di = actual - len(self) * s - databytes
1329 except IOError, inst:
1331 except IOError, inst:
1330 if inst.errno != errno.ENOENT:
1332 if inst.errno != errno.ENOENT:
1331 raise
1333 raise
1332 di = 0
1334 di = 0
1333
1335
1334 return (dd, di)
1336 return (dd, di)
1335
1337
1336 def files(self):
1338 def files(self):
1337 res = [self.indexfile]
1339 res = [self.indexfile]
1338 if not self._inline:
1340 if not self._inline:
1339 res.append(self.datafile)
1341 res.append(self.datafile)
1340 return res
1342 return res
General Comments 0
You need to be logged in to leave comments. Login now