##// END OF EJS Templates
pure/bdiff: fix circular import
Matt Mackall -
r7944:e9b48afd default
parent child Browse files
Show More
@@ -1,69 +1,76
1 1 # bdiff.py - Python implementation of bdiff.c
2 2 #
3 3 # Copyright 2009 Matt Mackall <mpm@selenic.com> and others
4 4 #
5 5 # This software may be used and distributed according to the terms
6 6 # of the GNU General Public License, incorporated herein by reference.
7 7
8 8 import struct, difflib
9 # mdiff import moved to bottom due to import cycle
9
10 def splitnewlines(text):
11 '''like str.splitlines, but only split on newlines.'''
12 lines = [l + '\n' for l in text.split('\n')]
13 if lines:
14 if lines[-1] == '\n':
15 lines.pop()
16 else:
17 lines[-1] = lines[-1][:-1]
18 return lines
10 19
11 20 def _normalizeblocks(a, b, blocks):
12 21 prev = None
13 22 for curr in blocks:
14 23 if prev is None:
15 24 prev = curr
16 25 continue
17 26 shift = 0
18 27
19 28 a1, b1, l1 = prev
20 29 a1end = a1 + l1
21 30 b1end = b1 + l1
22 31
23 32 a2, b2, l2 = curr
24 33 a2end = a2 + l2
25 34 b2end = b2 + l2
26 35 if a1end == a2:
27 36 while a1end+shift < a2end and a[a1end+shift] == b[b1end+shift]:
28 37 shift += 1
29 38 elif b1end == b2:
30 39 while b1end+shift < b2end and a[a1end+shift] == b[b1end+shift]:
31 40 shift += 1
32 41 yield a1, b1, l1+shift
33 42 prev = a2+shift, b2+shift, l2-shift
34 43 yield prev
35 44
36 45 def bdiff(a, b):
37 46 a = str(a).splitlines(True)
38 47 b = str(b).splitlines(True)
39 48
40 49 if not a:
41 50 s = "".join(b)
42 51 return s and (struct.pack(">lll", 0, 0, len(s)) + s)
43 52
44 53 bin = []
45 54 p = [0]
46 55 for i in a: p.append(p[-1] + len(i))
47 56
48 57 d = difflib.SequenceMatcher(None, a, b).get_matching_blocks()
49 58 d = _normalizeblocks(a, b, d)
50 59 la = 0
51 60 lb = 0
52 61 for am, bm, size in d:
53 62 s = "".join(b[lb:bm])
54 63 if am > la or s:
55 64 bin.append(struct.pack(">lll", p[la], p[am], len(s)) + s)
56 65 la = am + size
57 66 lb = bm + size
58 67
59 68 return "".join(bin)
60 69
61 70 def blocks(a, b):
62 an = mdiff.splitnewlines(a)
63 bn = mdiff.splitnewlines(b)
71 an = splitnewlines(a)
72 bn = splitnewlines(b)
64 73 d = difflib.SequenceMatcher(None, an, bn).get_matching_blocks()
65 74 d = _normalizeblocks(an, bn, d)
66 75 return [(i, i + n, j, j + n) for (i, j, n) in d]
67 76
68 # this breaks an import cycle
69 import mdiff
General Comments 0
You need to be logged in to leave comments. Login now