bdiff: implement cffi version of bdiff
authorMaciej Fijalkowski <fijall@gmail.com>
Sat, 20 Aug 2016 23:06:01 +0200
changeset 29845 1ea77b75d266
parent 29844 a8933d992a71
child 29846 bff109e6398a
bdiff: implement cffi version of bdiff
mercurial/pure/bdiff.py
--- a/mercurial/pure/bdiff.py	Thu Jul 28 14:17:08 2016 +0200
+++ b/mercurial/pure/bdiff.py	Sat Aug 20 23:06:01 2016 +0200
@@ -113,12 +113,12 @@
             b = ffi.new("struct bdiff_line**")
             ac = ffi.new("char[]", sa)
             bc = ffi.new("char[]", sb)
+            l = ffi.new("struct bdiff_hunk*")
             try:
                 an = lib.bdiff_splitlines(ac, len(sa), a)
                 bn = lib.bdiff_splitlines(bc, len(sb), b)
                 if not a[0] or not b[0]:
                     raise MemoryError
-                l = ffi.new("struct bdiff_hunk*")
                 count = lib.bdiff_diff(a[0], an, b[0], bn, l)
                 if count < 0:
                     raise MemoryError
@@ -134,3 +134,36 @@
                 lib.free(b[0])
                 lib.bdiff_freehunks(l.next)
             return rl
+
+        def bdiff(sa, sb):
+            a = ffi.new("struct bdiff_line**")
+            b = ffi.new("struct bdiff_line**")
+            ac = ffi.new("char[]", sa)
+            bc = ffi.new("char[]", sb)
+            l = ffi.new("struct bdiff_hunk*")
+            try:
+                an = lib.bdiff_splitlines(ac, len(sa), a)
+                bn = lib.bdiff_splitlines(bc, len(sb), b)
+                if not a[0] or not b[0]:
+                    raise MemoryError
+                count = lib.bdiff_diff(a[0], an, b[0], bn, l)
+                if count < 0:
+                    raise MemoryError
+                rl = []
+                h = l.next
+                la = lb = 0
+                while h:
+                    if h.a1 != la or h.b1 != lb:
+                        lgt = (b[0] + h.b1).l - (b[0] + lb).l
+                        rl.append(struct.pack(">lll", (a[0] + la).l - a[0].l,
+                            (a[0] + h.a1).l - a[0].l, lgt))
+                        rl.append(str(ffi.buffer((b[0] + lb).l, lgt)))
+                    la = h.a2
+                    lb = h.b2
+                    h = h.next
+
+            finally:
+                lib.free(a[0])
+                lib.free(b[0])
+                lib.bdiff_freehunks(l.next)
+            return "".join(rl)