← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/merge3.py

  • Committer: Martin Pool
  • Date: 2007-04-04 06:17:31 UTC
  • mto: This revision was merged to the branch mainline in revision 2397.
  • Revision ID: mbp@sourcefrog.net-20070404061731-tt2xrzllqhbodn83
Contents of TODO file moved into bug tracker

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/merge3.py
1
 
# Copyright (C) 2004, 2005 by Canonical Ltd
2
 
 
 
1
# Copyright (C) 2004, 2005 Canonical Ltd
 
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
 
 
7
#
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
12
 
 
 
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19
19
# s: "i hate that."
20
20
 
21
21
 
 
22
from bzrlib.errors import CantReprocessAndShowBase
 
23
import bzrlib.patiencediff
 
24
from bzrlib.textfile import check_text_lines
 
25
 
22
26
 
23
27
def intersect(ra, rb):
24
28
    """Given two ranges return the range where they intersect or None.
64
68
    incorporating the changes from both BASE->OTHER and BASE->THIS.
65
69
    All three will typically be sequences of lines."""
66
70
    def __init__(self, base, a, b):
 
71
        check_text_lines(base)
 
72
        check_text_lines(a)
 
73
        check_text_lines(b)
67
74
        self.base = base
68
75
        self.a = a
69
76
        self.b = b
70
 
        from difflib import SequenceMatcher
71
 
        self.a_ops = SequenceMatcher(None, base, a).get_opcodes()
72
 
        self.b_ops = SequenceMatcher(None, base, b).get_opcodes()
73
77
 
74
78
 
75
79
 
76
80
    def merge_lines(self,
77
81
                    name_a=None,
78
82
                    name_b=None,
79
 
                    start_marker='<<<<<<<<',
80
 
                    mid_marker='========',
81
 
                    end_marker='>>>>>>>>',
82
 
                    show_base=False):
 
83
                    name_base=None,
 
84
                    start_marker='<<<<<<<',
 
85
                    mid_marker='=======',
 
86
                    end_marker='>>>>>>>',
 
87
                    base_marker=None,
 
88
                    reprocess=False):
83
89
        """Return merge in cvs-like form.
84
90
        """
 
91
        newline = '\n'
 
92
        if len(self.a) > 0:
 
93
            if self.a[0].endswith('\r\n'):
 
94
                newline = '\r\n'
 
95
            elif self.a[0].endswith('\r'):
 
96
                newline = '\r'
 
97
        if base_marker and reprocess:
 
98
            raise CantReprocessAndShowBase()
85
99
        if name_a:
86
100
            start_marker = start_marker + ' ' + name_a
87
101
        if name_b:
88
102
            end_marker = end_marker + ' ' + name_b
89
 
            
90
 
        for t in self.merge_regions():
 
103
        if name_base and base_marker:
 
104
            base_marker = base_marker + ' ' + name_base
 
105
        merge_regions = self.merge_regions()
 
106
        if reprocess is True:
 
107
            merge_regions = self.reprocess_merge_regions(merge_regions)
 
108
        for t in merge_regions:
91
109
            what = t[0]
92
110
            if what == 'unchanged':
93
111
                for i in range(t[1], t[2]):
99
117
                for i in range(t[1], t[2]):
100
118
                    yield self.b[i]
101
119
            elif what == 'conflict':
102
 
                yield start_marker + '\n'
 
120
                yield start_marker + newline
103
121
                for i in range(t[3], t[4]):
104
122
                    yield self.a[i]
105
 
                yield mid_marker + '\n'
 
123
                if base_marker is not None:
 
124
                    yield base_marker + newline
 
125
                    for i in range(t[1], t[2]):
 
126
                        yield self.base[i]
 
127
                yield mid_marker + newline
106
128
                for i in range(t[5], t[6]):
107
129
                    yield self.b[i]
108
 
                yield end_marker + '\n'
 
130
                yield end_marker + newline
109
131
            else:
110
132
                raise ValueError(what)
111
133
        
263
285
                iz = zend
264
286
                ia = aend
265
287
                ib = bend
266
 
        
267
 
 
268
 
        
 
288
    
 
289
 
 
290
    def reprocess_merge_regions(self, merge_regions):
 
291
        """Where there are conflict regions, remove the agreed lines.
 
292
 
 
293
        Lines where both A and B have made the same changes are 
 
294
        eliminated.
 
295
        """
 
296
        for region in merge_regions:
 
297
            if region[0] != "conflict":
 
298
                yield region
 
299
                continue
 
300
            type, iz, zmatch, ia, amatch, ib, bmatch = region
 
301
            a_region = self.a[ia:amatch]
 
302
            b_region = self.b[ib:bmatch]
 
303
            matches = bzrlib.patiencediff.PatienceSequenceMatcher(
 
304
                    None, a_region, b_region).get_matching_blocks()
 
305
            next_a = ia
 
306
            next_b = ib
 
307
            for region_ia, region_ib, region_len in matches[:-1]:
 
308
                region_ia += ia
 
309
                region_ib += ib
 
310
                reg = self.mismatch_region(next_a, region_ia, next_b,
 
311
                                           region_ib)
 
312
                if reg is not None:
 
313
                    yield reg
 
314
                yield 'same', region_ia, region_len+region_ia
 
315
                next_a = region_ia + region_len
 
316
                next_b = region_ib + region_len
 
317
            reg = self.mismatch_region(next_a, amatch, next_b, bmatch)
 
318
            if reg is not None:
 
319
                yield reg
 
320
 
 
321
 
 
322
    @staticmethod
 
323
    def mismatch_region(next_a, region_ia,  next_b, region_ib):
 
324
        if next_a < region_ia or next_b < region_ib:
 
325
            return 'conflict', None, None, next_a, region_ia, next_b, region_ib
 
326
            
 
327
 
269
328
    def find_sync_regions(self):
270
329
        """Return a list of sync regions, where both descendents match the base.
271
330
 
272
331
        Generates a list of (base1, base2, a1, a2, b1, b2).  There is
273
332
        always a zero-length sync region at the end of all the files.
274
333
        """
275
 
        from difflib import SequenceMatcher
276
334
 
277
335
        ia = ib = 0
278
 
        amatches = SequenceMatcher(None, self.base, self.a).get_matching_blocks()
279
 
        bmatches = SequenceMatcher(None, self.base, self.b).get_matching_blocks()
 
336
        amatches = bzrlib.patiencediff.PatienceSequenceMatcher(
 
337
                None, self.base, self.a).get_matching_blocks()
 
338
        bmatches = bzrlib.patiencediff.PatienceSequenceMatcher(
 
339
                None, self.base, self.b).get_matching_blocks()
280
340
        len_a = len(amatches)
281
341
        len_b = len(bmatches)
282
342
 
332
392
 
333
393
    def find_unconflicted(self):
334
394
        """Return a list of ranges in base that are not conflicted."""
335
 
        from difflib import SequenceMatcher
336
 
 
337
 
        import re
338
 
 
339
 
        # don't sync-up on lines containing only blanks or pounds
340
 
        junk_re = re.compile(r'^[ \t#]*$')
341
 
        
342
 
        am = SequenceMatcher(junk_re.match, self.base, self.a).get_matching_blocks()
343
 
        bm = SequenceMatcher(junk_re.match, self.base, self.b).get_matching_blocks()
 
395
        am = bzrlib.patiencediff.PatienceSequenceMatcher(
 
396
                None, self.base, self.a).get_matching_blocks()
 
397
        bm = bzrlib.patiencediff.PatienceSequenceMatcher(
 
398
                None, self.base, self.b).get_matching_blocks()
344
399
 
345
400
        unc = []
346
401