##// END OF EJS Templates
revlogdeltas: extra fulltext building in its own function...
Boris Feld -
r39367:fd0150a3 default
parent child Browse files
Show More
@@ -1,734 +1,739 b''
1 1 # revlogdeltas.py - Logic around delta computation for revlog
2 2 #
3 3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
4 4 # Copyright 2018 Octobus <contact@octobus.net>
5 5 #
6 6 # This software may be used and distributed according to the terms of the
7 7 # GNU General Public License version 2 or any later version.
8 8 """Helper class to compute deltas stored inside revlogs"""
9 9
10 10 from __future__ import absolute_import
11 11
12 12 import heapq
13 13 import struct
14 14
15 15 # import stuff from node for others to import from revlog
16 16 from ..node import (
17 17 nullrev,
18 18 )
19 19 from ..i18n import _
20 20
21 21 from .constants import (
22 22 REVIDX_ISCENSORED,
23 23 REVIDX_RAWTEXT_CHANGING_FLAGS,
24 24 )
25 25
26 26 from ..thirdparty import (
27 27 attr,
28 28 )
29 29
30 30 from .. import (
31 31 error,
32 32 mdiff,
33 33 )
34 34
35 35 RevlogError = error.RevlogError
36 36 CensoredNodeError = error.CensoredNodeError
37 37
38 38 # maximum <delta-chain-data>/<revision-text-length> ratio
39 39 LIMIT_DELTA2TEXT = 2
40 40
41 41 class _testrevlog(object):
42 42 """minimalist fake revlog to use in doctests"""
43 43
44 44 def __init__(self, data, density=0.5, mingap=0):
45 45 """data is an list of revision payload boundaries"""
46 46 self._data = data
47 47 self._srdensitythreshold = density
48 48 self._srmingapsize = mingap
49 49
50 50 def start(self, rev):
51 51 if rev == 0:
52 52 return 0
53 53 return self._data[rev - 1]
54 54
55 55 def end(self, rev):
56 56 return self._data[rev]
57 57
58 58 def length(self, rev):
59 59 return self.end(rev) - self.start(rev)
60 60
61 61 def __len__(self):
62 62 return len(self._data)
63 63
64 64 def slicechunk(revlog, revs, deltainfo=None, targetsize=None):
65 65 """slice revs to reduce the amount of unrelated data to be read from disk.
66 66
67 67 ``revs`` is sliced into groups that should be read in one time.
68 68 Assume that revs are sorted.
69 69
70 70 The initial chunk is sliced until the overall density (payload/chunks-span
71 71 ratio) is above `revlog._srdensitythreshold`. No gap smaller than
72 72 `revlog._srmingapsize` is skipped.
73 73
74 74 If `targetsize` is set, no chunk larger than `targetsize` will be yield.
75 75 For consistency with other slicing choice, this limit won't go lower than
76 76 `revlog._srmingapsize`.
77 77
78 78 If individual revisions chunk are larger than this limit, they will still
79 79 be raised individually.
80 80
81 81 >>> revlog = _testrevlog([
82 82 ... 5, #00 (5)
83 83 ... 10, #01 (5)
84 84 ... 12, #02 (2)
85 85 ... 12, #03 (empty)
86 86 ... 27, #04 (15)
87 87 ... 31, #05 (4)
88 88 ... 31, #06 (empty)
89 89 ... 42, #07 (11)
90 90 ... 47, #08 (5)
91 91 ... 47, #09 (empty)
92 92 ... 48, #10 (1)
93 93 ... 51, #11 (3)
94 94 ... 74, #12 (23)
95 95 ... 85, #13 (11)
96 96 ... 86, #14 (1)
97 97 ... 91, #15 (5)
98 98 ... ])
99 99
100 100 >>> list(slicechunk(revlog, list(range(16))))
101 101 [[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]]
102 102 >>> list(slicechunk(revlog, [0, 15]))
103 103 [[0], [15]]
104 104 >>> list(slicechunk(revlog, [0, 11, 15]))
105 105 [[0], [11], [15]]
106 106 >>> list(slicechunk(revlog, [0, 11, 13, 15]))
107 107 [[0], [11, 13, 15]]
108 108 >>> list(slicechunk(revlog, [1, 2, 3, 5, 8, 10, 11, 14]))
109 109 [[1, 2], [5, 8, 10, 11], [14]]
110 110
111 111 Slicing with a maximum chunk size
112 112 >>> list(slicechunk(revlog, [0, 11, 13, 15], targetsize=15))
113 113 [[0], [11], [13], [15]]
114 114 >>> list(slicechunk(revlog, [0, 11, 13, 15], targetsize=20))
115 115 [[0], [11], [13, 15]]
116 116 """
117 117 if targetsize is not None:
118 118 targetsize = max(targetsize, revlog._srmingapsize)
119 119 # targetsize should not be specified when evaluating delta candidates:
120 120 # * targetsize is used to ensure we stay within specification when reading,
121 121 # * deltainfo is used to pick are good delta chain when writing.
122 122 if not (deltainfo is None or targetsize is None):
123 123 msg = 'cannot use `targetsize` with a `deltainfo`'
124 124 raise error.ProgrammingError(msg)
125 125 for chunk in _slicechunktodensity(revlog, revs,
126 126 deltainfo,
127 127 revlog._srdensitythreshold,
128 128 revlog._srmingapsize):
129 129 for subchunk in _slicechunktosize(revlog, chunk, targetsize):
130 130 yield subchunk
131 131
132 132 def _slicechunktosize(revlog, revs, targetsize=None):
133 133 """slice revs to match the target size
134 134
135 135 This is intended to be used on chunk that density slicing selected by that
136 136 are still too large compared to the read garantee of revlog. This might
137 137 happens when "minimal gap size" interrupted the slicing or when chain are
138 138 built in a way that create large blocks next to each other.
139 139
140 140 >>> revlog = _testrevlog([
141 141 ... 3, #0 (3)
142 142 ... 5, #1 (2)
143 143 ... 6, #2 (1)
144 144 ... 8, #3 (2)
145 145 ... 8, #4 (empty)
146 146 ... 11, #5 (3)
147 147 ... 12, #6 (1)
148 148 ... 13, #7 (1)
149 149 ... 14, #8 (1)
150 150 ... ])
151 151
152 152 Cases where chunk is already small enough
153 153 >>> list(_slicechunktosize(revlog, [0], 3))
154 154 [[0]]
155 155 >>> list(_slicechunktosize(revlog, [6, 7], 3))
156 156 [[6, 7]]
157 157 >>> list(_slicechunktosize(revlog, [0], None))
158 158 [[0]]
159 159 >>> list(_slicechunktosize(revlog, [6, 7], None))
160 160 [[6, 7]]
161 161
162 162 cases where we need actual slicing
163 163 >>> list(_slicechunktosize(revlog, [0, 1], 3))
164 164 [[0], [1]]
165 165 >>> list(_slicechunktosize(revlog, [1, 3], 3))
166 166 [[1], [3]]
167 167 >>> list(_slicechunktosize(revlog, [1, 2, 3], 3))
168 168 [[1, 2], [3]]
169 169 >>> list(_slicechunktosize(revlog, [3, 5], 3))
170 170 [[3], [5]]
171 171 >>> list(_slicechunktosize(revlog, [3, 4, 5], 3))
172 172 [[3], [5]]
173 173 >>> list(_slicechunktosize(revlog, [5, 6, 7, 8], 3))
174 174 [[5], [6, 7, 8]]
175 175 >>> list(_slicechunktosize(revlog, [0, 1, 2, 3, 4, 5, 6, 7, 8], 3))
176 176 [[0], [1, 2], [3], [5], [6, 7, 8]]
177 177
178 178 Case with too large individual chunk (must return valid chunk)
179 179 >>> list(_slicechunktosize(revlog, [0, 1], 2))
180 180 [[0], [1]]
181 181 >>> list(_slicechunktosize(revlog, [1, 3], 1))
182 182 [[1], [3]]
183 183 >>> list(_slicechunktosize(revlog, [3, 4, 5], 2))
184 184 [[3], [5]]
185 185 """
186 186 assert targetsize is None or 0 <= targetsize
187 187 if targetsize is None or segmentspan(revlog, revs) <= targetsize:
188 188 yield revs
189 189 return
190 190
191 191 startrevidx = 0
192 192 startdata = revlog.start(revs[0])
193 193 endrevidx = 0
194 194 iterrevs = enumerate(revs)
195 195 next(iterrevs) # skip first rev.
196 196 for idx, r in iterrevs:
197 197 span = revlog.end(r) - startdata
198 198 if span <= targetsize:
199 199 endrevidx = idx
200 200 else:
201 201 chunk = _trimchunk(revlog, revs, startrevidx, endrevidx + 1)
202 202 if chunk:
203 203 yield chunk
204 204 startrevidx = idx
205 205 startdata = revlog.start(r)
206 206 endrevidx = idx
207 207 yield _trimchunk(revlog, revs, startrevidx)
208 208
209 209 def _slicechunktodensity(revlog, revs, deltainfo=None, targetdensity=0.5,
210 210 mingapsize=0):
211 211 """slice revs to reduce the amount of unrelated data to be read from disk.
212 212
213 213 ``revs`` is sliced into groups that should be read in one time.
214 214 Assume that revs are sorted.
215 215
216 216 ``deltainfo`` is a _deltainfo instance of a revision that we would append
217 217 to the top of the revlog.
218 218
219 219 The initial chunk is sliced until the overall density (payload/chunks-span
220 220 ratio) is above `targetdensity`. No gap smaller than `mingapsize` is
221 221 skipped.
222 222
223 223 >>> revlog = _testrevlog([
224 224 ... 5, #00 (5)
225 225 ... 10, #01 (5)
226 226 ... 12, #02 (2)
227 227 ... 12, #03 (empty)
228 228 ... 27, #04 (15)
229 229 ... 31, #05 (4)
230 230 ... 31, #06 (empty)
231 231 ... 42, #07 (11)
232 232 ... 47, #08 (5)
233 233 ... 47, #09 (empty)
234 234 ... 48, #10 (1)
235 235 ... 51, #11 (3)
236 236 ... 74, #12 (23)
237 237 ... 85, #13 (11)
238 238 ... 86, #14 (1)
239 239 ... 91, #15 (5)
240 240 ... ])
241 241
242 242 >>> list(_slicechunktodensity(revlog, list(range(16))))
243 243 [[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]]
244 244 >>> list(_slicechunktodensity(revlog, [0, 15]))
245 245 [[0], [15]]
246 246 >>> list(_slicechunktodensity(revlog, [0, 11, 15]))
247 247 [[0], [11], [15]]
248 248 >>> list(_slicechunktodensity(revlog, [0, 11, 13, 15]))
249 249 [[0], [11, 13, 15]]
250 250 >>> list(_slicechunktodensity(revlog, [1, 2, 3, 5, 8, 10, 11, 14]))
251 251 [[1, 2], [5, 8, 10, 11], [14]]
252 252 >>> list(_slicechunktodensity(revlog, [1, 2, 3, 5, 8, 10, 11, 14],
253 253 ... mingapsize=20))
254 254 [[1, 2, 3, 5, 8, 10, 11], [14]]
255 255 >>> list(_slicechunktodensity(revlog, [1, 2, 3, 5, 8, 10, 11, 14],
256 256 ... targetdensity=0.95))
257 257 [[1, 2], [5], [8, 10, 11], [14]]
258 258 >>> list(_slicechunktodensity(revlog, [1, 2, 3, 5, 8, 10, 11, 14],
259 259 ... targetdensity=0.95, mingapsize=12))
260 260 [[1, 2], [5, 8, 10, 11], [14]]
261 261 """
262 262 start = revlog.start
263 263 length = revlog.length
264 264
265 265 if len(revs) <= 1:
266 266 yield revs
267 267 return
268 268
269 269 nextrev = len(revlog)
270 270 nextoffset = revlog.end(nextrev - 1)
271 271
272 272 if deltainfo is None:
273 273 deltachainspan = segmentspan(revlog, revs)
274 274 chainpayload = sum(length(r) for r in revs)
275 275 else:
276 276 deltachainspan = deltainfo.distance
277 277 chainpayload = deltainfo.compresseddeltalen
278 278
279 279 if deltachainspan < mingapsize:
280 280 yield revs
281 281 return
282 282
283 283 readdata = deltachainspan
284 284
285 285 if deltachainspan:
286 286 density = chainpayload / float(deltachainspan)
287 287 else:
288 288 density = 1.0
289 289
290 290 if density >= targetdensity:
291 291 yield revs
292 292 return
293 293
294 294 if deltainfo is not None and deltainfo.deltalen:
295 295 revs = list(revs)
296 296 revs.append(nextrev)
297 297
298 298 # Store the gaps in a heap to have them sorted by decreasing size
299 299 gapsheap = []
300 300 heapq.heapify(gapsheap)
301 301 prevend = None
302 302 for i, rev in enumerate(revs):
303 303 if rev < nextrev:
304 304 revstart = start(rev)
305 305 revlen = length(rev)
306 306 else:
307 307 revstart = nextoffset
308 308 revlen = deltainfo.deltalen
309 309
310 310 # Skip empty revisions to form larger holes
311 311 if revlen == 0:
312 312 continue
313 313
314 314 if prevend is not None:
315 315 gapsize = revstart - prevend
316 316 # only consider holes that are large enough
317 317 if gapsize > mingapsize:
318 318 heapq.heappush(gapsheap, (-gapsize, i))
319 319
320 320 prevend = revstart + revlen
321 321
322 322 # Collect the indices of the largest holes until the density is acceptable
323 323 indicesheap = []
324 324 heapq.heapify(indicesheap)
325 325 while gapsheap and density < targetdensity:
326 326 oppgapsize, gapidx = heapq.heappop(gapsheap)
327 327
328 328 heapq.heappush(indicesheap, gapidx)
329 329
330 330 # the gap sizes are stored as negatives to be sorted decreasingly
331 331 # by the heap
332 332 readdata -= (-oppgapsize)
333 333 if readdata > 0:
334 334 density = chainpayload / float(readdata)
335 335 else:
336 336 density = 1.0
337 337
338 338 # Cut the revs at collected indices
339 339 previdx = 0
340 340 while indicesheap:
341 341 idx = heapq.heappop(indicesheap)
342 342
343 343 chunk = _trimchunk(revlog, revs, previdx, idx)
344 344 if chunk:
345 345 yield chunk
346 346
347 347 previdx = idx
348 348
349 349 chunk = _trimchunk(revlog, revs, previdx)
350 350 if chunk:
351 351 yield chunk
352 352
353 353 def _trimchunk(revlog, revs, startidx, endidx=None):
354 354 """returns revs[startidx:endidx] without empty trailing revs
355 355
356 356 Doctest Setup
357 357 >>> revlog = _testrevlog([
358 358 ... 5, #0
359 359 ... 10, #1
360 360 ... 12, #2
361 361 ... 12, #3 (empty)
362 362 ... 17, #4
363 363 ... 21, #5
364 364 ... 21, #6 (empty)
365 365 ... ])
366 366
367 367 Contiguous cases:
368 368 >>> _trimchunk(revlog, [0, 1, 2, 3, 4, 5, 6], 0)
369 369 [0, 1, 2, 3, 4, 5]
370 370 >>> _trimchunk(revlog, [0, 1, 2, 3, 4, 5, 6], 0, 5)
371 371 [0, 1, 2, 3, 4]
372 372 >>> _trimchunk(revlog, [0, 1, 2, 3, 4, 5, 6], 0, 4)
373 373 [0, 1, 2]
374 374 >>> _trimchunk(revlog, [0, 1, 2, 3, 4, 5, 6], 2, 4)
375 375 [2]
376 376 >>> _trimchunk(revlog, [0, 1, 2, 3, 4, 5, 6], 3)
377 377 [3, 4, 5]
378 378 >>> _trimchunk(revlog, [0, 1, 2, 3, 4, 5, 6], 3, 5)
379 379 [3, 4]
380 380
381 381 Discontiguous cases:
382 382 >>> _trimchunk(revlog, [1, 3, 5, 6], 0)
383 383 [1, 3, 5]
384 384 >>> _trimchunk(revlog, [1, 3, 5, 6], 0, 2)
385 385 [1]
386 386 >>> _trimchunk(revlog, [1, 3, 5, 6], 1, 3)
387 387 [3, 5]
388 388 >>> _trimchunk(revlog, [1, 3, 5, 6], 1)
389 389 [3, 5]
390 390 """
391 391 length = revlog.length
392 392
393 393 if endidx is None:
394 394 endidx = len(revs)
395 395
396 396 # If we have a non-emtpy delta candidate, there are nothing to trim
397 397 if revs[endidx - 1] < len(revlog):
398 398 # Trim empty revs at the end, except the very first revision of a chain
399 399 while (endidx > 1
400 400 and endidx > startidx
401 401 and length(revs[endidx - 1]) == 0):
402 402 endidx -= 1
403 403
404 404 return revs[startidx:endidx]
405 405
406 406 def segmentspan(revlog, revs, deltainfo=None):
407 407 """Get the byte span of a segment of revisions
408 408
409 409 revs is a sorted array of revision numbers
410 410
411 411 >>> revlog = _testrevlog([
412 412 ... 5, #0
413 413 ... 10, #1
414 414 ... 12, #2
415 415 ... 12, #3 (empty)
416 416 ... 17, #4
417 417 ... ])
418 418
419 419 >>> segmentspan(revlog, [0, 1, 2, 3, 4])
420 420 17
421 421 >>> segmentspan(revlog, [0, 4])
422 422 17
423 423 >>> segmentspan(revlog, [3, 4])
424 424 5
425 425 >>> segmentspan(revlog, [1, 2, 3,])
426 426 7
427 427 >>> segmentspan(revlog, [1, 3])
428 428 7
429 429 """
430 430 if not revs:
431 431 return 0
432 432 if deltainfo is not None and len(revlog) <= revs[-1]:
433 433 if len(revs) == 1:
434 434 return deltainfo.deltalen
435 435 offset = revlog.end(len(revlog) - 1)
436 436 end = deltainfo.deltalen + offset
437 437 else:
438 438 end = revlog.end(revs[-1])
439 439 return end - revlog.start(revs[0])
440 440
441 def _textfromdelta(fh, revlog, baserev, delta, p1, p2, flags, expectednode):
442 """build full text from a (base, delta) pair and other metadata"""
443 # special case deltas which replace entire base; no need to decode
444 # base revision. this neatly avoids censored bases, which throw when
445 # they're decoded.
446 hlen = struct.calcsize(">lll")
447 if delta[:hlen] == mdiff.replacediffheader(revlog.rawsize(baserev),
448 len(delta) - hlen):
449 fulltext = delta[hlen:]
450 else:
451 # deltabase is rawtext before changed by flag processors, which is
452 # equivalent to non-raw text
453 basetext = revlog.revision(baserev, _df=fh, raw=False)
454 fulltext = mdiff.patch(basetext, delta)
455
456 try:
457 res = revlog._processflags(fulltext, flags, 'read', raw=True)
458 fulltext, validatehash = res
459 if validatehash:
460 revlog.checkhash(fulltext, expectednode, p1=p1, p2=p2)
461 if flags & REVIDX_ISCENSORED:
462 raise RevlogError(_('node %s is not censored') % expectednode)
463 except CensoredNodeError:
464 # must pass the censored index flag to add censored revisions
465 if not flags & REVIDX_ISCENSORED:
466 raise
467 return fulltext
468
441 469 @attr.s(slots=True, frozen=True)
442 470 class _deltainfo(object):
443 471 distance = attr.ib()
444 472 deltalen = attr.ib()
445 473 data = attr.ib()
446 474 base = attr.ib()
447 475 chainbase = attr.ib()
448 476 chainlen = attr.ib()
449 477 compresseddeltalen = attr.ib()
450 478 snapshotdepth = attr.ib()
451 479
452 480 def isgooddeltainfo(revlog, deltainfo, revinfo):
453 481 """Returns True if the given delta is good. Good means that it is within
454 482 the disk span, disk size, and chain length bounds that we know to be
455 483 performant."""
456 484 if deltainfo is None:
457 485 return False
458 486
459 487 # - 'deltainfo.distance' is the distance from the base revision --
460 488 # bounding it limits the amount of I/O we need to do.
461 489 # - 'deltainfo.compresseddeltalen' is the sum of the total size of
462 490 # deltas we need to apply -- bounding it limits the amount of CPU
463 491 # we consume.
464 492
465 493 if revlog._sparserevlog:
466 494 # As sparse-read will be used, we can consider that the distance,
467 495 # instead of being the span of the whole chunk,
468 496 # is the span of the largest read chunk
469 497 base = deltainfo.base
470 498
471 499 if base != nullrev:
472 500 deltachain = revlog._deltachain(base)[0]
473 501 else:
474 502 deltachain = []
475 503
476 504 # search for the first non-snapshot revision
477 505 for idx, r in enumerate(deltachain):
478 506 if not revlog.issnapshot(r):
479 507 break
480 508 deltachain = deltachain[idx:]
481 509 chunks = slicechunk(revlog, deltachain, deltainfo)
482 510 all_span = [segmentspan(revlog, revs, deltainfo)
483 511 for revs in chunks]
484 512 distance = max(all_span)
485 513 else:
486 514 distance = deltainfo.distance
487 515
488 516 textlen = revinfo.textlen
489 517 defaultmax = textlen * 4
490 518 maxdist = revlog._maxdeltachainspan
491 519 if not maxdist:
492 520 maxdist = distance # ensure the conditional pass
493 521 maxdist = max(maxdist, defaultmax)
494 522 if revlog._sparserevlog and maxdist < revlog._srmingapsize:
495 523 # In multiple place, we are ignoring irrelevant data range below a
496 524 # certain size. Be also apply this tradeoff here and relax span
497 525 # constraint for small enought content.
498 526 maxdist = revlog._srmingapsize
499 527
500 528 # Bad delta from read span:
501 529 #
502 530 # If the span of data read is larger than the maximum allowed.
503 531 if maxdist < distance:
504 532 return False
505 533
506 534 # Bad delta from new delta size:
507 535 #
508 536 # If the delta size is larger than the target text, storing the
509 537 # delta will be inefficient.
510 538 if textlen < deltainfo.deltalen:
511 539 return False
512 540
513 541 # Bad delta from cumulated payload size:
514 542 #
515 543 # If the sum of delta get larger than K * target text length.
516 544 if textlen * LIMIT_DELTA2TEXT < deltainfo.compresseddeltalen:
517 545 return False
518 546
519 547 # Bad delta from chain length:
520 548 #
521 549 # If the number of delta in the chain gets too high.
522 550 if (revlog._maxchainlen
523 551 and revlog._maxchainlen < deltainfo.chainlen):
524 552 return False
525 553
526 554 # bad delta from intermediate snapshot size limit
527 555 #
528 556 # If an intermediate snapshot size is higher than the limit. The
529 557 # limit exist to prevent endless chain of intermediate delta to be
530 558 # created.
531 559 if (deltainfo.snapshotdepth is not None and
532 560 (textlen >> deltainfo.snapshotdepth) < deltainfo.deltalen):
533 561 return False
534 562
535 563 # bad delta if new intermediate snapshot is larger than the previous
536 564 # snapshot
537 565 if (deltainfo.snapshotdepth
538 566 and revlog.length(deltainfo.base) < deltainfo.deltalen):
539 567 return False
540 568
541 569 return True
542 570
543 571 class deltacomputer(object):
544 572 def __init__(self, revlog):
545 573 self.revlog = revlog
546 574
547 575 def _getcandidaterevs(self, p1, p2, cachedelta):
548 576 """
549 577 Provides revisions that present an interest to be diffed against,
550 578 grouped by level of easiness.
551 579 """
552 580 revlog = self.revlog
553 581 gdelta = revlog._generaldelta
554 582 curr = len(revlog)
555 583 prev = curr - 1
556 584 p1r, p2r = revlog.rev(p1), revlog.rev(p2)
557 585
558 586 # should we try to build a delta?
559 587 if prev != nullrev and revlog._storedeltachains:
560 588 tested = set()
561 589 # This condition is true most of the time when processing
562 590 # changegroup data into a generaldelta repo. The only time it
563 591 # isn't true is if this is the first revision in a delta chain
564 592 # or if ``format.generaldelta=true`` disabled ``lazydeltabase``.
565 593 if cachedelta and gdelta and revlog._lazydeltabase:
566 594 # Assume what we received from the server is a good choice
567 595 # build delta will reuse the cache
568 596 yield (cachedelta[0],)
569 597 tested.add(cachedelta[0])
570 598
571 599 if gdelta:
572 600 # exclude already lazy tested base if any
573 601 parents = [p for p in (p1r, p2r)
574 602 if p != nullrev and p not in tested]
575 603
576 604 if not revlog._deltabothparents and len(parents) == 2:
577 605 parents.sort()
578 606 # To minimize the chance of having to build a fulltext,
579 607 # pick first whichever parent is closest to us (max rev)
580 608 yield (parents[1],)
581 609 # then the other one (min rev) if the first did not fit
582 610 yield (parents[0],)
583 611 tested.update(parents)
584 612 elif len(parents) > 0:
585 613 # Test all parents (1 or 2), and keep the best candidate
586 614 yield parents
587 615 tested.update(parents)
588 616
589 617 if prev not in tested:
590 618 # other approach failed try against prev to hopefully save us a
591 619 # fulltext.
592 620 yield (prev,)
593 621 tested.add(prev)
594 622
595 623 def buildtext(self, revinfo, fh):
596 624 """Builds a fulltext version of a revision
597 625
598 626 revinfo: _revisioninfo instance that contains all needed info
599 627 fh: file handle to either the .i or the .d revlog file,
600 628 depending on whether it is inlined or not
601 629 """
602 630 btext = revinfo.btext
603 631 if btext[0] is not None:
604 632 return btext[0]
605 633
606 634 revlog = self.revlog
607 635 cachedelta = revinfo.cachedelta
608 flags = revinfo.flags
609 node = revinfo.node
610
611 636 baserev = cachedelta[0]
612 637 delta = cachedelta[1]
613 # special case deltas which replace entire base; no need to decode
614 # base revision. this neatly avoids censored bases, which throw when
615 # they're decoded.
616 hlen = struct.calcsize(">lll")
617 if delta[:hlen] == mdiff.replacediffheader(revlog.rawsize(baserev),
618 len(delta) - hlen):
619 btext[0] = delta[hlen:]
620 else:
621 # deltabase is rawtext before changed by flag processors, which is
622 # equivalent to non-raw text
623 basetext = revlog.revision(baserev, _df=fh, raw=False)
624 btext[0] = mdiff.patch(basetext, delta)
625 638
626 try:
627 res = revlog._processflags(btext[0], flags, 'read', raw=True)
628 btext[0], validatehash = res
629 if validatehash:
630 revlog.checkhash(btext[0], node, p1=revinfo.p1, p2=revinfo.p2)
631 if flags & REVIDX_ISCENSORED:
632 raise RevlogError(_('node %s is not censored') % node)
633 except CensoredNodeError:
634 # must pass the censored index flag to add censored revisions
635 if not flags & REVIDX_ISCENSORED:
636 raise
637 return btext[0]
639 fulltext = btext[0] = _textfromdelta(fh, revlog, baserev, delta,
640 revinfo.p1, revinfo.p2,
641 revinfo.flags, revinfo.node)
642 return fulltext
638 643
639 644 def _builddeltadiff(self, base, revinfo, fh):
640 645 revlog = self.revlog
641 646 t = self.buildtext(revinfo, fh)
642 647 if revlog.iscensored(base):
643 648 # deltas based on a censored revision must replace the
644 649 # full content in one patch, so delta works everywhere
645 650 header = mdiff.replacediffheader(revlog.rawsize(base), len(t))
646 651 delta = header + t
647 652 else:
648 653 ptext = revlog.revision(base, _df=fh, raw=True)
649 654 delta = mdiff.textdiff(ptext, t)
650 655
651 656 return delta
652 657
653 658 def _builddeltainfo(self, revinfo, base, fh):
654 659 # can we use the cached delta?
655 660 if revinfo.cachedelta and revinfo.cachedelta[0] == base:
656 661 delta = revinfo.cachedelta[1]
657 662 else:
658 663 delta = self._builddeltadiff(base, revinfo, fh)
659 664 revlog = self.revlog
660 665 header, data = revlog.compress(delta)
661 666 deltalen = len(header) + len(data)
662 667 chainbase = revlog.chainbase(base)
663 668 offset = revlog.end(len(revlog) - 1)
664 669 dist = deltalen + offset - revlog.start(chainbase)
665 670 if revlog._generaldelta:
666 671 deltabase = base
667 672 else:
668 673 deltabase = chainbase
669 674 chainlen, compresseddeltalen = revlog._chaininfo(base)
670 675 chainlen += 1
671 676 compresseddeltalen += deltalen
672 677
673 678 revlog = self.revlog
674 679 snapshotdepth = None
675 680 if deltabase == nullrev:
676 681 snapshotdepth = 0
677 682 elif revlog._sparserevlog and revlog.issnapshot(deltabase):
678 683 # A delta chain should always be one full snapshot,
679 684 # zero or more semi-snapshots, and zero or more deltas
680 685 p1, p2 = revlog.rev(revinfo.p1), revlog.rev(revinfo.p2)
681 686 if deltabase not in (p1, p2) and revlog.issnapshot(deltabase):
682 687 snapshotdepth = len(revlog._deltachain(deltabase)[0])
683 688
684 689 return _deltainfo(dist, deltalen, (header, data), deltabase,
685 690 chainbase, chainlen, compresseddeltalen,
686 691 snapshotdepth)
687 692
688 693 def finddeltainfo(self, revinfo, fh):
689 694 """Find an acceptable delta against a candidate revision
690 695
691 696 revinfo: information about the revision (instance of _revisioninfo)
692 697 fh: file handle to either the .i or the .d revlog file,
693 698 depending on whether it is inlined or not
694 699
695 700 Returns the first acceptable candidate revision, as ordered by
696 701 _getcandidaterevs
697 702 """
698 703 if not revinfo.textlen:
699 704 return None # empty file do not need delta
700 705
701 706 cachedelta = revinfo.cachedelta
702 707 p1 = revinfo.p1
703 708 p2 = revinfo.p2
704 709 revlog = self.revlog
705 710
706 711 deltalength = self.revlog.length
707 712 deltaparent = self.revlog.deltaparent
708 713
709 714 deltainfo = None
710 715 deltas_limit = revinfo.textlen * LIMIT_DELTA2TEXT
711 716 for candidaterevs in self._getcandidaterevs(p1, p2, cachedelta):
712 717 # filter out delta base that will never produce good delta
713 718 candidaterevs = [r for r in candidaterevs
714 719 if self.revlog.length(r) <= deltas_limit]
715 720 nominateddeltas = []
716 721 for candidaterev in candidaterevs:
717 722 # skip over empty delta (no need to include them in a chain)
718 723 while candidaterev != nullrev and not deltalength(candidaterev):
719 724 candidaterev = deltaparent(candidaterev)
720 725 # no need to try a delta against nullid, this will be handled
721 726 # by fulltext later.
722 727 if candidaterev == nullrev:
723 728 continue
724 729 # no delta for rawtext-changing revs (see "candelta" for why)
725 730 if revlog.flags(candidaterev) & REVIDX_RAWTEXT_CHANGING_FLAGS:
726 731 continue
727 732 candidatedelta = self._builddeltainfo(revinfo, candidaterev, fh)
728 733 if isgooddeltainfo(self.revlog, candidatedelta, revinfo):
729 734 nominateddeltas.append(candidatedelta)
730 735 if nominateddeltas:
731 736 deltainfo = min(nominateddeltas, key=lambda x: x.deltalen)
732 737 break
733 738
734 739 return deltainfo
General Comments 0
You need to be logged in to leave comments. Login now