##// END OF EJS Templates
revlog._chunks: inline getchunk...
Siddharth Agarwal -
r19715:1aab406b default
parent child Browse files
Show More
@@ -1,1342 +1,1345 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 end = self.end(endrev)
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
851 end += (endrev + 1) * self._io.size
852 length = end - start
852 length = end - start
853 return self._getchunk(start, length)
853 return self._getchunk(start, length)
854
854
855 def _chunk(self, rev):
855 def _chunk(self, rev):
856 return decompress(self._chunkraw(rev, rev))
856 return decompress(self._chunkraw(rev, rev))
857
857
858 def _chunks(self, revs):
858 def _chunks(self, revs):
859 '''faster version of [self._chunk(rev) for rev in revs]
859 '''faster version of [self._chunk(rev) for rev in revs]
860
860
861 Assumes that revs is in ascending order.'''
861 Assumes that revs is in ascending order.'''
862 start = self.start
862 start = self.start
863 length = self.length
863 length = self.length
864 inline = self._inline
864 inline = self._inline
865 iosize = self._io.size
865 iosize = self._io.size
866 getchunk = self._getchunk
866 buffer = util.buffer
867
867
868 l = []
868 l = []
869 ladd = l.append
869 ladd = l.append
870
870
871 # XXX assume for now that chunkcache is preloaded
872 offset, data = self._chunkcache
873
871 for rev in revs:
874 for rev in revs:
872 chunkstart = start(rev)
875 chunkstart = start(rev)
873 if inline:
876 if inline:
874 chunkstart += (rev + 1) * iosize
877 chunkstart += (rev + 1) * iosize
875 chunklength = length(rev)
878 chunklength = length(rev)
876 ladd(decompress(getchunk(chunkstart, chunklength)))
879 ladd(decompress(buffer(data, chunkstart - offset, chunklength)))
877
880
878 return l
881 return l
879
882
880 def _chunkbase(self, rev):
883 def _chunkbase(self, rev):
881 return self._chunk(rev)
884 return self._chunk(rev)
882
885
883 def _chunkclear(self):
886 def _chunkclear(self):
884 self._chunkcache = (0, '')
887 self._chunkcache = (0, '')
885
888
886 def deltaparent(self, rev):
889 def deltaparent(self, rev):
887 """return deltaparent of the given revision"""
890 """return deltaparent of the given revision"""
888 base = self.index[rev][3]
891 base = self.index[rev][3]
889 if base == rev:
892 if base == rev:
890 return nullrev
893 return nullrev
891 elif self._generaldelta:
894 elif self._generaldelta:
892 return base
895 return base
893 else:
896 else:
894 return rev - 1
897 return rev - 1
895
898
896 def revdiff(self, rev1, rev2):
899 def revdiff(self, rev1, rev2):
897 """return or calculate a delta between two revisions"""
900 """return or calculate a delta between two revisions"""
898 if rev1 != nullrev and self.deltaparent(rev2) == rev1:
901 if rev1 != nullrev and self.deltaparent(rev2) == rev1:
899 return str(self._chunk(rev2))
902 return str(self._chunk(rev2))
900
903
901 return mdiff.textdiff(self.revision(rev1),
904 return mdiff.textdiff(self.revision(rev1),
902 self.revision(rev2))
905 self.revision(rev2))
903
906
904 def revision(self, nodeorrev):
907 def revision(self, nodeorrev):
905 """return an uncompressed revision of a given node or revision
908 """return an uncompressed revision of a given node or revision
906 number.
909 number.
907 """
910 """
908 if isinstance(nodeorrev, int):
911 if isinstance(nodeorrev, int):
909 rev = nodeorrev
912 rev = nodeorrev
910 node = self.node(rev)
913 node = self.node(rev)
911 else:
914 else:
912 node = nodeorrev
915 node = nodeorrev
913 rev = None
916 rev = None
914
917
915 cachedrev = None
918 cachedrev = None
916 if node == nullid:
919 if node == nullid:
917 return ""
920 return ""
918 if self._cache:
921 if self._cache:
919 if self._cache[0] == node:
922 if self._cache[0] == node:
920 return self._cache[2]
923 return self._cache[2]
921 cachedrev = self._cache[1]
924 cachedrev = self._cache[1]
922
925
923 # look up what we need to read
926 # look up what we need to read
924 text = None
927 text = None
925 if rev is None:
928 if rev is None:
926 rev = self.rev(node)
929 rev = self.rev(node)
927
930
928 # check rev flags
931 # check rev flags
929 if self.flags(rev) & ~REVIDX_KNOWN_FLAGS:
932 if self.flags(rev) & ~REVIDX_KNOWN_FLAGS:
930 raise RevlogError(_('incompatible revision flag %x') %
933 raise RevlogError(_('incompatible revision flag %x') %
931 (self.flags(rev) & ~REVIDX_KNOWN_FLAGS))
934 (self.flags(rev) & ~REVIDX_KNOWN_FLAGS))
932
935
933 # build delta chain
936 # build delta chain
934 chain = []
937 chain = []
935 index = self.index # for performance
938 index = self.index # for performance
936 generaldelta = self._generaldelta
939 generaldelta = self._generaldelta
937 iterrev = rev
940 iterrev = rev
938 e = index[iterrev]
941 e = index[iterrev]
939 while iterrev != e[3] and iterrev != cachedrev:
942 while iterrev != e[3] and iterrev != cachedrev:
940 chain.append(iterrev)
943 chain.append(iterrev)
941 if generaldelta:
944 if generaldelta:
942 iterrev = e[3]
945 iterrev = e[3]
943 else:
946 else:
944 iterrev -= 1
947 iterrev -= 1
945 e = index[iterrev]
948 e = index[iterrev]
946 chain.reverse()
949 chain.reverse()
947 base = iterrev
950 base = iterrev
948
951
949 if iterrev == cachedrev:
952 if iterrev == cachedrev:
950 # cache hit
953 # cache hit
951 text = self._cache[2]
954 text = self._cache[2]
952
955
953 # drop cache to save memory
956 # drop cache to save memory
954 self._cache = None
957 self._cache = None
955
958
956 self._chunkraw(base, rev)
959 self._chunkraw(base, rev)
957 if text is None:
960 if text is None:
958 text = str(self._chunkbase(base))
961 text = str(self._chunkbase(base))
959
962
960 bins = self._chunks(chain)
963 bins = self._chunks(chain)
961 text = mdiff.patches(text, bins)
964 text = mdiff.patches(text, bins)
962
965
963 text = self._checkhash(text, node, rev)
966 text = self._checkhash(text, node, rev)
964
967
965 self._cache = (node, rev, text)
968 self._cache = (node, rev, text)
966 return text
969 return text
967
970
968 def _checkhash(self, text, node, rev):
971 def _checkhash(self, text, node, rev):
969 p1, p2 = self.parents(node)
972 p1, p2 = self.parents(node)
970 self.checkhash(text, p1, p2, node, rev)
973 self.checkhash(text, p1, p2, node, rev)
971 return text
974 return text
972
975
973 def checkhash(self, text, p1, p2, node, rev=None):
976 def checkhash(self, text, p1, p2, node, rev=None):
974 if node != hash(text, p1, p2):
977 if node != hash(text, p1, p2):
975 revornode = rev
978 revornode = rev
976 if revornode is None:
979 if revornode is None:
977 revornode = templatefilters.short(hex(node))
980 revornode = templatefilters.short(hex(node))
978 raise RevlogError(_("integrity check failed on %s:%s")
981 raise RevlogError(_("integrity check failed on %s:%s")
979 % (self.indexfile, revornode))
982 % (self.indexfile, revornode))
980
983
981 def checkinlinesize(self, tr, fp=None):
984 def checkinlinesize(self, tr, fp=None):
982 if not self._inline or (self.start(-2) + self.length(-2)) < _maxinline:
985 if not self._inline or (self.start(-2) + self.length(-2)) < _maxinline:
983 return
986 return
984
987
985 trinfo = tr.find(self.indexfile)
988 trinfo = tr.find(self.indexfile)
986 if trinfo is None:
989 if trinfo is None:
987 raise RevlogError(_("%s not found in the transaction")
990 raise RevlogError(_("%s not found in the transaction")
988 % self.indexfile)
991 % self.indexfile)
989
992
990 trindex = trinfo[2]
993 trindex = trinfo[2]
991 dataoff = self.start(trindex)
994 dataoff = self.start(trindex)
992
995
993 tr.add(self.datafile, dataoff)
996 tr.add(self.datafile, dataoff)
994
997
995 if fp:
998 if fp:
996 fp.flush()
999 fp.flush()
997 fp.close()
1000 fp.close()
998
1001
999 df = self.opener(self.datafile, 'w')
1002 df = self.opener(self.datafile, 'w')
1000 try:
1003 try:
1001 for r in self:
1004 for r in self:
1002 df.write(self._chunkraw(r, r))
1005 df.write(self._chunkraw(r, r))
1003 finally:
1006 finally:
1004 df.close()
1007 df.close()
1005
1008
1006 fp = self.opener(self.indexfile, 'w', atomictemp=True)
1009 fp = self.opener(self.indexfile, 'w', atomictemp=True)
1007 self.version &= ~(REVLOGNGINLINEDATA)
1010 self.version &= ~(REVLOGNGINLINEDATA)
1008 self._inline = False
1011 self._inline = False
1009 for i in self:
1012 for i in self:
1010 e = self._io.packentry(self.index[i], self.node, self.version, i)
1013 e = self._io.packentry(self.index[i], self.node, self.version, i)
1011 fp.write(e)
1014 fp.write(e)
1012
1015
1013 # if we don't call close, the temp file will never replace the
1016 # if we don't call close, the temp file will never replace the
1014 # real index
1017 # real index
1015 fp.close()
1018 fp.close()
1016
1019
1017 tr.replace(self.indexfile, trindex * self._io.size)
1020 tr.replace(self.indexfile, trindex * self._io.size)
1018 self._chunkclear()
1021 self._chunkclear()
1019
1022
1020 def addrevision(self, text, transaction, link, p1, p2, cachedelta=None,
1023 def addrevision(self, text, transaction, link, p1, p2, cachedelta=None,
1021 node=None):
1024 node=None):
1022 """add a revision to the log
1025 """add a revision to the log
1023
1026
1024 text - the revision data to add
1027 text - the revision data to add
1025 transaction - the transaction object used for rollback
1028 transaction - the transaction object used for rollback
1026 link - the linkrev data to add
1029 link - the linkrev data to add
1027 p1, p2 - the parent nodeids of the revision
1030 p1, p2 - the parent nodeids of the revision
1028 cachedelta - an optional precomputed delta
1031 cachedelta - an optional precomputed delta
1029 node - nodeid of revision; typically node is not specified, and it is
1032 node - nodeid of revision; typically node is not specified, and it is
1030 computed by default as hash(text, p1, p2), however subclasses might
1033 computed by default as hash(text, p1, p2), however subclasses might
1031 use different hashing method (and override checkhash() in such case)
1034 use different hashing method (and override checkhash() in such case)
1032 """
1035 """
1033 if link == nullrev:
1036 if link == nullrev:
1034 raise RevlogError(_("attempted to add linkrev -1 to %s")
1037 raise RevlogError(_("attempted to add linkrev -1 to %s")
1035 % self.indexfile)
1038 % self.indexfile)
1036 node = node or hash(text, p1, p2)
1039 node = node or hash(text, p1, p2)
1037 if node in self.nodemap:
1040 if node in self.nodemap:
1038 return node
1041 return node
1039
1042
1040 dfh = None
1043 dfh = None
1041 if not self._inline:
1044 if not self._inline:
1042 dfh = self.opener(self.datafile, "a")
1045 dfh = self.opener(self.datafile, "a")
1043 ifh = self.opener(self.indexfile, "a+")
1046 ifh = self.opener(self.indexfile, "a+")
1044 try:
1047 try:
1045 return self._addrevision(node, text, transaction, link, p1, p2,
1048 return self._addrevision(node, text, transaction, link, p1, p2,
1046 cachedelta, ifh, dfh)
1049 cachedelta, ifh, dfh)
1047 finally:
1050 finally:
1048 if dfh:
1051 if dfh:
1049 dfh.close()
1052 dfh.close()
1050 ifh.close()
1053 ifh.close()
1051
1054
1052 def compress(self, text):
1055 def compress(self, text):
1053 """ generate a possibly-compressed representation of text """
1056 """ generate a possibly-compressed representation of text """
1054 if not text:
1057 if not text:
1055 return ("", text)
1058 return ("", text)
1056 l = len(text)
1059 l = len(text)
1057 bin = None
1060 bin = None
1058 if l < 44:
1061 if l < 44:
1059 pass
1062 pass
1060 elif l > 1000000:
1063 elif l > 1000000:
1061 # zlib makes an internal copy, thus doubling memory usage for
1064 # zlib makes an internal copy, thus doubling memory usage for
1062 # large files, so lets do this in pieces
1065 # large files, so lets do this in pieces
1063 z = zlib.compressobj()
1066 z = zlib.compressobj()
1064 p = []
1067 p = []
1065 pos = 0
1068 pos = 0
1066 while pos < l:
1069 while pos < l:
1067 pos2 = pos + 2**20
1070 pos2 = pos + 2**20
1068 p.append(z.compress(text[pos:pos2]))
1071 p.append(z.compress(text[pos:pos2]))
1069 pos = pos2
1072 pos = pos2
1070 p.append(z.flush())
1073 p.append(z.flush())
1071 if sum(map(len, p)) < l:
1074 if sum(map(len, p)) < l:
1072 bin = "".join(p)
1075 bin = "".join(p)
1073 else:
1076 else:
1074 bin = _compress(text)
1077 bin = _compress(text)
1075 if bin is None or len(bin) > l:
1078 if bin is None or len(bin) > l:
1076 if text[0] == '\0':
1079 if text[0] == '\0':
1077 return ("", text)
1080 return ("", text)
1078 return ('u', text)
1081 return ('u', text)
1079 return ("", bin)
1082 return ("", bin)
1080
1083
1081 def _addrevision(self, node, text, transaction, link, p1, p2,
1084 def _addrevision(self, node, text, transaction, link, p1, p2,
1082 cachedelta, ifh, dfh):
1085 cachedelta, ifh, dfh):
1083 """internal function to add revisions to the log
1086 """internal function to add revisions to the log
1084
1087
1085 see addrevision for argument descriptions.
1088 see addrevision for argument descriptions.
1086 invariants:
1089 invariants:
1087 - text is optional (can be None); if not set, cachedelta must be set.
1090 - text is optional (can be None); if not set, cachedelta must be set.
1088 if both are set, they must correspond to each other.
1091 if both are set, they must correspond to each other.
1089 """
1092 """
1090 btext = [text]
1093 btext = [text]
1091 def buildtext():
1094 def buildtext():
1092 if btext[0] is not None:
1095 if btext[0] is not None:
1093 return btext[0]
1096 return btext[0]
1094 # flush any pending writes here so we can read it in revision
1097 # flush any pending writes here so we can read it in revision
1095 if dfh:
1098 if dfh:
1096 dfh.flush()
1099 dfh.flush()
1097 ifh.flush()
1100 ifh.flush()
1098 basetext = self.revision(self.node(cachedelta[0]))
1101 basetext = self.revision(self.node(cachedelta[0]))
1099 btext[0] = mdiff.patch(basetext, cachedelta[1])
1102 btext[0] = mdiff.patch(basetext, cachedelta[1])
1100 self.checkhash(btext[0], p1, p2, node)
1103 self.checkhash(btext[0], p1, p2, node)
1101 return btext[0]
1104 return btext[0]
1102
1105
1103 def builddelta(rev):
1106 def builddelta(rev):
1104 # can we use the cached delta?
1107 # can we use the cached delta?
1105 if cachedelta and cachedelta[0] == rev:
1108 if cachedelta and cachedelta[0] == rev:
1106 delta = cachedelta[1]
1109 delta = cachedelta[1]
1107 else:
1110 else:
1108 t = buildtext()
1111 t = buildtext()
1109 ptext = self.revision(self.node(rev))
1112 ptext = self.revision(self.node(rev))
1110 delta = mdiff.textdiff(ptext, t)
1113 delta = mdiff.textdiff(ptext, t)
1111 data = self.compress(delta)
1114 data = self.compress(delta)
1112 l = len(data[1]) + len(data[0])
1115 l = len(data[1]) + len(data[0])
1113 if basecache[0] == rev:
1116 if basecache[0] == rev:
1114 chainbase = basecache[1]
1117 chainbase = basecache[1]
1115 else:
1118 else:
1116 chainbase = self.chainbase(rev)
1119 chainbase = self.chainbase(rev)
1117 dist = l + offset - self.start(chainbase)
1120 dist = l + offset - self.start(chainbase)
1118 if self._generaldelta:
1121 if self._generaldelta:
1119 base = rev
1122 base = rev
1120 else:
1123 else:
1121 base = chainbase
1124 base = chainbase
1122 return dist, l, data, base, chainbase
1125 return dist, l, data, base, chainbase
1123
1126
1124 curr = len(self)
1127 curr = len(self)
1125 prev = curr - 1
1128 prev = curr - 1
1126 base = chainbase = curr
1129 base = chainbase = curr
1127 offset = self.end(prev)
1130 offset = self.end(prev)
1128 flags = 0
1131 flags = 0
1129 d = None
1132 d = None
1130 basecache = self._basecache
1133 basecache = self._basecache
1131 p1r, p2r = self.rev(p1), self.rev(p2)
1134 p1r, p2r = self.rev(p1), self.rev(p2)
1132
1135
1133 # should we try to build a delta?
1136 # should we try to build a delta?
1134 if prev != nullrev:
1137 if prev != nullrev:
1135 if self._generaldelta:
1138 if self._generaldelta:
1136 if p1r >= basecache[1]:
1139 if p1r >= basecache[1]:
1137 d = builddelta(p1r)
1140 d = builddelta(p1r)
1138 elif p2r >= basecache[1]:
1141 elif p2r >= basecache[1]:
1139 d = builddelta(p2r)
1142 d = builddelta(p2r)
1140 else:
1143 else:
1141 d = builddelta(prev)
1144 d = builddelta(prev)
1142 else:
1145 else:
1143 d = builddelta(prev)
1146 d = builddelta(prev)
1144 dist, l, data, base, chainbase = d
1147 dist, l, data, base, chainbase = d
1145
1148
1146 # full versions are inserted when the needed deltas
1149 # full versions are inserted when the needed deltas
1147 # become comparable to the uncompressed text
1150 # become comparable to the uncompressed text
1148 if text is None:
1151 if text is None:
1149 textlen = mdiff.patchedsize(self.rawsize(cachedelta[0]),
1152 textlen = mdiff.patchedsize(self.rawsize(cachedelta[0]),
1150 cachedelta[1])
1153 cachedelta[1])
1151 else:
1154 else:
1152 textlen = len(text)
1155 textlen = len(text)
1153 if d is None or dist > textlen * 2:
1156 if d is None or dist > textlen * 2:
1154 text = buildtext()
1157 text = buildtext()
1155 data = self.compress(text)
1158 data = self.compress(text)
1156 l = len(data[1]) + len(data[0])
1159 l = len(data[1]) + len(data[0])
1157 base = chainbase = curr
1160 base = chainbase = curr
1158
1161
1159 e = (offset_type(offset, flags), l, textlen,
1162 e = (offset_type(offset, flags), l, textlen,
1160 base, link, p1r, p2r, node)
1163 base, link, p1r, p2r, node)
1161 self.index.insert(-1, e)
1164 self.index.insert(-1, e)
1162 self.nodemap[node] = curr
1165 self.nodemap[node] = curr
1163
1166
1164 entry = self._io.packentry(e, self.node, self.version, curr)
1167 entry = self._io.packentry(e, self.node, self.version, curr)
1165 if not self._inline:
1168 if not self._inline:
1166 transaction.add(self.datafile, offset)
1169 transaction.add(self.datafile, offset)
1167 transaction.add(self.indexfile, curr * len(entry))
1170 transaction.add(self.indexfile, curr * len(entry))
1168 if data[0]:
1171 if data[0]:
1169 dfh.write(data[0])
1172 dfh.write(data[0])
1170 dfh.write(data[1])
1173 dfh.write(data[1])
1171 dfh.flush()
1174 dfh.flush()
1172 ifh.write(entry)
1175 ifh.write(entry)
1173 else:
1176 else:
1174 offset += curr * self._io.size
1177 offset += curr * self._io.size
1175 transaction.add(self.indexfile, offset, curr)
1178 transaction.add(self.indexfile, offset, curr)
1176 ifh.write(entry)
1179 ifh.write(entry)
1177 ifh.write(data[0])
1180 ifh.write(data[0])
1178 ifh.write(data[1])
1181 ifh.write(data[1])
1179 self.checkinlinesize(transaction, ifh)
1182 self.checkinlinesize(transaction, ifh)
1180
1183
1181 if type(text) == str: # only accept immutable objects
1184 if type(text) == str: # only accept immutable objects
1182 self._cache = (node, curr, text)
1185 self._cache = (node, curr, text)
1183 self._basecache = (curr, chainbase)
1186 self._basecache = (curr, chainbase)
1184 return node
1187 return node
1185
1188
1186 def addgroup(self, bundle, linkmapper, transaction):
1189 def addgroup(self, bundle, linkmapper, transaction):
1187 """
1190 """
1188 add a delta group
1191 add a delta group
1189
1192
1190 given a set of deltas, add them to the revision log. the
1193 given a set of deltas, add them to the revision log. the
1191 first delta is against its parent, which should be in our
1194 first delta is against its parent, which should be in our
1192 log, the rest are against the previous delta.
1195 log, the rest are against the previous delta.
1193 """
1196 """
1194
1197
1195 # track the base of the current delta log
1198 # track the base of the current delta log
1196 content = []
1199 content = []
1197 node = None
1200 node = None
1198
1201
1199 r = len(self)
1202 r = len(self)
1200 end = 0
1203 end = 0
1201 if r:
1204 if r:
1202 end = self.end(r - 1)
1205 end = self.end(r - 1)
1203 ifh = self.opener(self.indexfile, "a+")
1206 ifh = self.opener(self.indexfile, "a+")
1204 isize = r * self._io.size
1207 isize = r * self._io.size
1205 if self._inline:
1208 if self._inline:
1206 transaction.add(self.indexfile, end + isize, r)
1209 transaction.add(self.indexfile, end + isize, r)
1207 dfh = None
1210 dfh = None
1208 else:
1211 else:
1209 transaction.add(self.indexfile, isize, r)
1212 transaction.add(self.indexfile, isize, r)
1210 transaction.add(self.datafile, end)
1213 transaction.add(self.datafile, end)
1211 dfh = self.opener(self.datafile, "a")
1214 dfh = self.opener(self.datafile, "a")
1212
1215
1213 try:
1216 try:
1214 # loop through our set of deltas
1217 # loop through our set of deltas
1215 chain = None
1218 chain = None
1216 while True:
1219 while True:
1217 chunkdata = bundle.deltachunk(chain)
1220 chunkdata = bundle.deltachunk(chain)
1218 if not chunkdata:
1221 if not chunkdata:
1219 break
1222 break
1220 node = chunkdata['node']
1223 node = chunkdata['node']
1221 p1 = chunkdata['p1']
1224 p1 = chunkdata['p1']
1222 p2 = chunkdata['p2']
1225 p2 = chunkdata['p2']
1223 cs = chunkdata['cs']
1226 cs = chunkdata['cs']
1224 deltabase = chunkdata['deltabase']
1227 deltabase = chunkdata['deltabase']
1225 delta = chunkdata['delta']
1228 delta = chunkdata['delta']
1226
1229
1227 content.append(node)
1230 content.append(node)
1228
1231
1229 link = linkmapper(cs)
1232 link = linkmapper(cs)
1230 if node in self.nodemap:
1233 if node in self.nodemap:
1231 # this can happen if two branches make the same change
1234 # this can happen if two branches make the same change
1232 chain = node
1235 chain = node
1233 continue
1236 continue
1234
1237
1235 for p in (p1, p2):
1238 for p in (p1, p2):
1236 if p not in self.nodemap:
1239 if p not in self.nodemap:
1237 raise LookupError(p, self.indexfile,
1240 raise LookupError(p, self.indexfile,
1238 _('unknown parent'))
1241 _('unknown parent'))
1239
1242
1240 if deltabase not in self.nodemap:
1243 if deltabase not in self.nodemap:
1241 raise LookupError(deltabase, self.indexfile,
1244 raise LookupError(deltabase, self.indexfile,
1242 _('unknown delta base'))
1245 _('unknown delta base'))
1243
1246
1244 baserev = self.rev(deltabase)
1247 baserev = self.rev(deltabase)
1245 chain = self._addrevision(node, None, transaction, link,
1248 chain = self._addrevision(node, None, transaction, link,
1246 p1, p2, (baserev, delta), ifh, dfh)
1249 p1, p2, (baserev, delta), ifh, dfh)
1247 if not dfh and not self._inline:
1250 if not dfh and not self._inline:
1248 # addrevision switched from inline to conventional
1251 # addrevision switched from inline to conventional
1249 # reopen the index
1252 # reopen the index
1250 ifh.close()
1253 ifh.close()
1251 dfh = self.opener(self.datafile, "a")
1254 dfh = self.opener(self.datafile, "a")
1252 ifh = self.opener(self.indexfile, "a")
1255 ifh = self.opener(self.indexfile, "a")
1253 finally:
1256 finally:
1254 if dfh:
1257 if dfh:
1255 dfh.close()
1258 dfh.close()
1256 ifh.close()
1259 ifh.close()
1257
1260
1258 return content
1261 return content
1259
1262
1260 def strip(self, minlink, transaction):
1263 def strip(self, minlink, transaction):
1261 """truncate the revlog on the first revision with a linkrev >= minlink
1264 """truncate the revlog on the first revision with a linkrev >= minlink
1262
1265
1263 This function is called when we're stripping revision minlink and
1266 This function is called when we're stripping revision minlink and
1264 its descendants from the repository.
1267 its descendants from the repository.
1265
1268
1266 We have to remove all revisions with linkrev >= minlink, because
1269 We have to remove all revisions with linkrev >= minlink, because
1267 the equivalent changelog revisions will be renumbered after the
1270 the equivalent changelog revisions will be renumbered after the
1268 strip.
1271 strip.
1269
1272
1270 So we truncate the revlog on the first of these revisions, and
1273 So we truncate the revlog on the first of these revisions, and
1271 trust that the caller has saved the revisions that shouldn't be
1274 trust that the caller has saved the revisions that shouldn't be
1272 removed and that it'll re-add them after this truncation.
1275 removed and that it'll re-add them after this truncation.
1273 """
1276 """
1274 if len(self) == 0:
1277 if len(self) == 0:
1275 return
1278 return
1276
1279
1277 for rev in self:
1280 for rev in self:
1278 if self.index[rev][4] >= minlink:
1281 if self.index[rev][4] >= minlink:
1279 break
1282 break
1280 else:
1283 else:
1281 return
1284 return
1282
1285
1283 # first truncate the files on disk
1286 # first truncate the files on disk
1284 end = self.start(rev)
1287 end = self.start(rev)
1285 if not self._inline:
1288 if not self._inline:
1286 transaction.add(self.datafile, end)
1289 transaction.add(self.datafile, end)
1287 end = rev * self._io.size
1290 end = rev * self._io.size
1288 else:
1291 else:
1289 end += rev * self._io.size
1292 end += rev * self._io.size
1290
1293
1291 transaction.add(self.indexfile, end)
1294 transaction.add(self.indexfile, end)
1292
1295
1293 # then reset internal state in memory to forget those revisions
1296 # then reset internal state in memory to forget those revisions
1294 self._cache = None
1297 self._cache = None
1295 self._chunkclear()
1298 self._chunkclear()
1296 for x in xrange(rev, len(self)):
1299 for x in xrange(rev, len(self)):
1297 del self.nodemap[self.node(x)]
1300 del self.nodemap[self.node(x)]
1298
1301
1299 del self.index[rev:-1]
1302 del self.index[rev:-1]
1300
1303
1301 def checksize(self):
1304 def checksize(self):
1302 expected = 0
1305 expected = 0
1303 if len(self):
1306 if len(self):
1304 expected = max(0, self.end(len(self) - 1))
1307 expected = max(0, self.end(len(self) - 1))
1305
1308
1306 try:
1309 try:
1307 f = self.opener(self.datafile)
1310 f = self.opener(self.datafile)
1308 f.seek(0, 2)
1311 f.seek(0, 2)
1309 actual = f.tell()
1312 actual = f.tell()
1310 f.close()
1313 f.close()
1311 dd = actual - expected
1314 dd = actual - expected
1312 except IOError, inst:
1315 except IOError, inst:
1313 if inst.errno != errno.ENOENT:
1316 if inst.errno != errno.ENOENT:
1314 raise
1317 raise
1315 dd = 0
1318 dd = 0
1316
1319
1317 try:
1320 try:
1318 f = self.opener(self.indexfile)
1321 f = self.opener(self.indexfile)
1319 f.seek(0, 2)
1322 f.seek(0, 2)
1320 actual = f.tell()
1323 actual = f.tell()
1321 f.close()
1324 f.close()
1322 s = self._io.size
1325 s = self._io.size
1323 i = max(0, actual // s)
1326 i = max(0, actual // s)
1324 di = actual - (i * s)
1327 di = actual - (i * s)
1325 if self._inline:
1328 if self._inline:
1326 databytes = 0
1329 databytes = 0
1327 for r in self:
1330 for r in self:
1328 databytes += max(0, self.length(r))
1331 databytes += max(0, self.length(r))
1329 dd = 0
1332 dd = 0
1330 di = actual - len(self) * s - databytes
1333 di = actual - len(self) * s - databytes
1331 except IOError, inst:
1334 except IOError, inst:
1332 if inst.errno != errno.ENOENT:
1335 if inst.errno != errno.ENOENT:
1333 raise
1336 raise
1334 di = 0
1337 di = 0
1335
1338
1336 return (dd, di)
1339 return (dd, di)
1337
1340
1338 def files(self):
1341 def files(self):
1339 res = [self.indexfile]
1342 res = [self.indexfile]
1340 if not self._inline:
1343 if not self._inline:
1341 res.append(self.datafile)
1344 res.append(self.datafile)
1342 return res
1345 return res
General Comments 0
You need to be logged in to leave comments. Login now