~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/merge.py

  • Committer: Martin Pool
  • Date: 2009-08-20 04:53:23 UTC
  • mto: This revision was merged to the branch mainline in revision 4632.
  • Revision ID: mbp@sourcefrog.net-20090820045323-4hsicfa87pdq3l29
Correction to xdg_cache_dir and add a simple test

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2008 Canonical Ltd
2
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
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
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
 
 
17
 
 
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
 
17
 
 
18
import errno
 
19
from itertools import chain
18
20
import os
19
 
import errno
20
 
from tempfile import mkdtemp
21
21
import warnings
22
22
 
 
23
from bzrlib import (
 
24
    debug,
 
25
    errors,
 
26
    graph as _mod_graph,
 
27
    osutils,
 
28
    patiencediff,
 
29
    registry,
 
30
    revision as _mod_revision,
 
31
    tree as _mod_tree,
 
32
    tsort,
 
33
    )
23
34
from bzrlib.branch import Branch
24
35
from bzrlib.conflicts import ConflictList, Conflict
25
36
from bzrlib.errors import (BzrCommandError,
35
46
                           WorkingTreeNotRevision,
36
47
                           BinaryFile,
37
48
                           )
 
49
from bzrlib.graph import Graph
38
50
from bzrlib.merge3 import Merge3
39
 
import bzrlib.osutils
40
 
from bzrlib.osutils import rename, pathjoin, rmtree
 
51
from bzrlib.osutils import rename, pathjoin
41
52
from progress import DummyProgress, ProgressPhase
42
 
from bzrlib.revision import common_ancestor, is_ancestor, NULL_REVISION
 
53
from bzrlib.revision import (NULL_REVISION, ensure_null)
43
54
from bzrlib.textfile import check_text_lines
44
 
from bzrlib.trace import mutter, warning, note
45
 
from bzrlib.transform import (TreeTransform, resolve_conflicts, cook_conflicts,
46
 
                              FinalPaths, create_by_entry, unique_add)
47
 
from bzrlib.versionedfile import WeaveMerge
 
55
from bzrlib.trace import mutter, warning, note, is_quiet
 
56
from bzrlib.transform import (TransformPreview, TreeTransform,
 
57
                              resolve_conflicts, cook_conflicts,
 
58
                              conflict_pass, FinalPaths, create_from_tree,
 
59
                              unique_add, ROOT_PARENT)
 
60
from bzrlib.versionedfile import PlanWeaveMerge
48
61
from bzrlib import ui
49
62
 
50
63
# TODO: Report back as changes are merged in
51
64
 
52
 
def _get_tree(treespec, local_branch=None):
53
 
    from bzrlib import workingtree
54
 
    location, revno = treespec
55
 
    if revno is None:
56
 
        tree = workingtree.WorkingTree.open_containing(location)[0]
57
 
        return tree.branch, tree
58
 
    branch = Branch.open_containing(location)[0]
59
 
    if revno == -1:
60
 
        revision = branch.last_revision()
61
 
    else:
62
 
        revision = branch.get_rev_id(revno)
63
 
        if revision is None:
64
 
            revision = NULL_REVISION
65
 
    return branch, _get_revid_tree(branch, revision, local_branch)
66
 
 
67
 
 
68
 
def _get_revid_tree(branch, revision, local_branch):
69
 
    if revision is None:
70
 
        base_tree = branch.bzrdir.open_workingtree()
71
 
    else:
72
 
        if local_branch is not None:
73
 
            if local_branch.base != branch.base:
74
 
                local_branch.fetch(branch, revision)
75
 
            base_tree = local_branch.repository.revision_tree(revision)
76
 
        else:
77
 
            base_tree = branch.repository.revision_tree(revision)
78
 
    return base_tree
79
 
 
80
65
 
81
66
def transform_tree(from_tree, to_tree, interesting_ids=None):
82
67
    merge_inner(from_tree.branch, to_tree, from_tree, ignore_zero=True,
84
69
 
85
70
 
86
71
class Merger(object):
87
 
    def __init__(self, this_branch, other_tree=None, base_tree=None, 
88
 
                 this_tree=None, pb=DummyProgress()):
 
72
    def __init__(self, this_branch, other_tree=None, base_tree=None,
 
73
                 this_tree=None, pb=None, change_reporter=None,
 
74
                 recurse='down', revision_graph=None):
89
75
        object.__init__(self)
90
 
        assert this_tree is not None, "this_tree is required"
91
76
        self.this_branch = this_branch
92
 
        self.this_basis = this_branch.last_revision()
 
77
        self.this_basis = _mod_revision.ensure_null(
 
78
            this_branch.last_revision())
93
79
        self.this_rev_id = None
94
80
        self.this_tree = this_tree
95
81
        self.this_revision_tree = None
96
82
        self.this_basis_tree = None
97
83
        self.other_tree = other_tree
 
84
        self.other_branch = None
98
85
        self.base_tree = base_tree
99
86
        self.ignore_zero = False
100
87
        self.backup_files = False
101
88
        self.interesting_ids = None
 
89
        self.interesting_files = None
102
90
        self.show_base = False
103
91
        self.reprocess = False
104
 
        self._pb = pb 
 
92
        if pb is None:
 
93
            pb = DummyProgress()
 
94
        self._pb = pb
105
95
        self.pp = None
106
 
 
107
 
 
108
 
    def revision_tree(self, revision_id):
109
 
        return self.this_branch.repository.revision_tree(revision_id)
 
96
        self.recurse = recurse
 
97
        self.change_reporter = change_reporter
 
98
        self._cached_trees = {}
 
99
        self._revision_graph = revision_graph
 
100
        self._base_is_ancestor = None
 
101
        self._base_is_other_ancestor = None
 
102
        self._is_criss_cross = None
 
103
        self._lca_trees = None
 
104
 
 
105
    @property
 
106
    def revision_graph(self):
 
107
        if self._revision_graph is None:
 
108
            self._revision_graph = self.this_branch.repository.get_graph()
 
109
        return self._revision_graph
 
110
 
 
111
    def _set_base_is_ancestor(self, value):
 
112
        self._base_is_ancestor = value
 
113
 
 
114
    def _get_base_is_ancestor(self):
 
115
        if self._base_is_ancestor is None:
 
116
            self._base_is_ancestor = self.revision_graph.is_ancestor(
 
117
                self.base_rev_id, self.this_basis)
 
118
        return self._base_is_ancestor
 
119
 
 
120
    base_is_ancestor = property(_get_base_is_ancestor, _set_base_is_ancestor)
 
121
 
 
122
    def _set_base_is_other_ancestor(self, value):
 
123
        self._base_is_other_ancestor = value
 
124
 
 
125
    def _get_base_is_other_ancestor(self):
 
126
        if self._base_is_other_ancestor is None:
 
127
            if self.other_basis is None:
 
128
                return True
 
129
            self._base_is_other_ancestor = self.revision_graph.is_ancestor(
 
130
                self.base_rev_id, self.other_basis)
 
131
        return self._base_is_other_ancestor
 
132
 
 
133
    base_is_other_ancestor = property(_get_base_is_other_ancestor,
 
134
                                      _set_base_is_other_ancestor)
 
135
 
 
136
    @staticmethod
 
137
    def from_uncommitted(tree, other_tree, pb=None, base_tree=None):
 
138
        """Return a Merger for uncommitted changes in other_tree.
 
139
 
 
140
        :param tree: The tree to merge into
 
141
        :param other_tree: The tree to get uncommitted changes from
 
142
        :param pb: A progress indicator
 
143
        :param base_tree: The basis to use for the merge.  If unspecified,
 
144
            other_tree.basis_tree() will be used.
 
145
        """
 
146
        if base_tree is None:
 
147
            base_tree = other_tree.basis_tree()
 
148
        merger = Merger(tree.branch, other_tree, base_tree, tree, pb)
 
149
        merger.base_rev_id = merger.base_tree.get_revision_id()
 
150
        merger.other_rev_id = None
 
151
        merger.other_basis = merger.base_rev_id
 
152
        return merger
 
153
 
 
154
    @classmethod
 
155
    def from_mergeable(klass, tree, mergeable, pb):
 
156
        """Return a Merger for a bundle or merge directive.
 
157
 
 
158
        :param tree: The tree to merge changes into
 
159
        :param mergeable: A merge directive or bundle
 
160
        :param pb: A progress indicator
 
161
        """
 
162
        mergeable.install_revisions(tree.branch.repository)
 
163
        base_revision_id, other_revision_id, verified =\
 
164
            mergeable.get_merge_request(tree.branch.repository)
 
165
        revision_graph = tree.branch.repository.get_graph()
 
166
        if base_revision_id is not None:
 
167
            if (base_revision_id != _mod_revision.NULL_REVISION and
 
168
                revision_graph.is_ancestor(
 
169
                base_revision_id, tree.branch.last_revision())):
 
170
                base_revision_id = None
 
171
            else:
 
172
                warning('Performing cherrypick')
 
173
        merger = klass.from_revision_ids(pb, tree, other_revision_id,
 
174
                                         base_revision_id, revision_graph=
 
175
                                         revision_graph)
 
176
        return merger, verified
 
177
 
 
178
    @staticmethod
 
179
    def from_revision_ids(pb, tree, other, base=None, other_branch=None,
 
180
                          base_branch=None, revision_graph=None,
 
181
                          tree_branch=None):
 
182
        """Return a Merger for revision-ids.
 
183
 
 
184
        :param pb: A progress indicator
 
185
        :param tree: The tree to merge changes into
 
186
        :param other: The revision-id to use as OTHER
 
187
        :param base: The revision-id to use as BASE.  If not specified, will
 
188
            be auto-selected.
 
189
        :param other_branch: A branch containing the other revision-id.  If
 
190
            not supplied, tree.branch is used.
 
191
        :param base_branch: A branch containing the base revision-id.  If
 
192
            not supplied, other_branch or tree.branch will be used.
 
193
        :param revision_graph: If you have a revision_graph precomputed, pass
 
194
            it in, otherwise it will be created for you.
 
195
        :param tree_branch: The branch associated with tree.  If not supplied,
 
196
            tree.branch will be used.
 
197
        """
 
198
        if tree_branch is None:
 
199
            tree_branch = tree.branch
 
200
        merger = Merger(tree_branch, this_tree=tree, pb=pb,
 
201
                        revision_graph=revision_graph)
 
202
        if other_branch is None:
 
203
            other_branch = tree.branch
 
204
        merger.set_other_revision(other, other_branch)
 
205
        if base is None:
 
206
            merger.find_base()
 
207
        else:
 
208
            if base_branch is None:
 
209
                base_branch = other_branch
 
210
            merger.set_base_revision(base, base_branch)
 
211
        return merger
 
212
 
 
213
    def revision_tree(self, revision_id, branch=None):
 
214
        if revision_id not in self._cached_trees:
 
215
            if branch is None:
 
216
                branch = self.this_branch
 
217
            try:
 
218
                tree = self.this_tree.revision_tree(revision_id)
 
219
            except errors.NoSuchRevisionInTree:
 
220
                tree = branch.repository.revision_tree(revision_id)
 
221
            self._cached_trees[revision_id] = tree
 
222
        return self._cached_trees[revision_id]
 
223
 
 
224
    def _get_tree(self, treespec, possible_transports=None):
 
225
        from bzrlib import workingtree
 
226
        location, revno = treespec
 
227
        if revno is None:
 
228
            tree = workingtree.WorkingTree.open_containing(location)[0]
 
229
            return tree.branch, tree
 
230
        branch = Branch.open_containing(location, possible_transports)[0]
 
231
        if revno == -1:
 
232
            revision_id = branch.last_revision()
 
233
        else:
 
234
            revision_id = branch.get_rev_id(revno)
 
235
        revision_id = ensure_null(revision_id)
 
236
        return branch, self.revision_tree(revision_id, branch)
110
237
 
111
238
    def ensure_revision_trees(self):
112
239
        if self.this_revision_tree is None:
113
 
            self.this_basis_tree = self.this_branch.repository.revision_tree(
114
 
                self.this_basis)
 
240
            self.this_basis_tree = self.revision_tree(self.this_basis)
115
241
            if self.this_basis == self.this_rev_id:
116
242
                self.this_revision_tree = self.this_basis_tree
117
243
 
118
244
        if self.other_rev_id is None:
119
245
            other_basis_tree = self.revision_tree(self.other_basis)
120
 
            changes = other_basis_tree.changes_from(self.other_tree)
121
 
            if changes.has_changed():
 
246
            if other_basis_tree.has_changes(self.other_tree):
122
247
                raise WorkingTreeNotRevision(self.this_tree)
123
248
            other_rev_id = self.other_basis
124
249
            self.other_tree = other_basis_tree
127
252
        self.ensure_revision_trees()
128
253
        def get_id(tree, file_id):
129
254
            revision_id = tree.inventory[file_id].revision
130
 
            assert revision_id is not None
131
255
            return revision_id
132
256
        if self.this_rev_id is None:
133
257
            if self.this_basis_tree.get_file_sha1(file_id) != \
139
263
 
140
264
    def check_basis(self, check_clean, require_commits=True):
141
265
        if self.this_basis is None and require_commits is True:
142
 
            raise BzrCommandError("This branch has no commits")
 
266
            raise BzrCommandError("This branch has no commits."
 
267
                                  " (perhaps you would prefer 'bzr pull')")
143
268
        if check_clean:
144
269
            self.compare_basis()
145
270
            if self.this_basis != self.this_rev_id:
146
 
                raise BzrCommandError("Working tree has uncommitted changes.")
 
271
                raise errors.UncommittedChanges(self.this_tree)
147
272
 
148
273
    def compare_basis(self):
149
 
        changes = self.this_tree.changes_from(self.this_tree.basis_tree())
150
 
        if not changes.has_changed():
 
274
        try:
 
275
            basis_tree = self.revision_tree(self.this_tree.last_revision())
 
276
        except errors.NoSuchRevision:
 
277
            basis_tree = self.this_tree.basis_tree()
 
278
        if not self.this_tree.has_changes(basis_tree):
151
279
            self.this_rev_id = self.this_basis
152
280
 
153
281
    def set_interesting_files(self, file_list):
154
 
        try:
155
 
            self._set_interesting_files(file_list)
156
 
        except NotVersionedError, e:
157
 
            raise BzrCommandError("%s is not a source file in any"
158
 
                                      " tree." % e.path)
159
 
 
160
 
    def _set_interesting_files(self, file_list):
161
 
        """Set the list of interesting ids from a list of files."""
162
 
        if file_list is None:
163
 
            self.interesting_ids = None
164
 
            return
165
 
 
166
 
        interesting_ids = set()
167
 
        for path in file_list:
168
 
            found_id = False
169
 
            for tree in (self.this_tree, self.base_tree, self.other_tree):
170
 
                file_id = tree.inventory.path2id(path)
171
 
                if file_id is not None:
172
 
                    interesting_ids.add(file_id)
173
 
                    found_id = True
174
 
            if not found_id:
175
 
                raise NotVersionedError(path=path)
176
 
        self.interesting_ids = interesting_ids
 
282
        self.interesting_files = file_list
177
283
 
178
284
    def set_pending(self):
179
 
        if not self.base_is_ancestor:
180
 
            return
181
 
        if self.other_rev_id is None:
182
 
            return
183
 
        ancestry = self.this_branch.repository.get_ancestry(self.this_basis)
184
 
        if self.other_rev_id in ancestry:
185
 
            return
186
 
        self.this_tree.add_pending_merge(self.other_rev_id)
187
 
 
188
 
    def set_other(self, other_revision):
189
 
        other_branch, self.other_tree = _get_tree(other_revision, 
190
 
                                                  self.this_branch)
 
285
        if not self.base_is_ancestor or not self.base_is_other_ancestor or self.other_rev_id is None:
 
286
            return
 
287
        self._add_parent()
 
288
 
 
289
    def _add_parent(self):
 
290
        new_parents = self.this_tree.get_parent_ids() + [self.other_rev_id]
 
291
        new_parent_trees = []
 
292
        for revision_id in new_parents:
 
293
            try:
 
294
                tree = self.revision_tree(revision_id)
 
295
            except errors.NoSuchRevision:
 
296
                tree = None
 
297
            else:
 
298
                tree.lock_read()
 
299
            new_parent_trees.append((revision_id, tree))
 
300
        try:
 
301
            self.this_tree.set_parent_trees(new_parent_trees,
 
302
                                            allow_leftmost_as_ghost=True)
 
303
        finally:
 
304
            for _revision_id, tree in new_parent_trees:
 
305
                if tree is not None:
 
306
                    tree.unlock()
 
307
 
 
308
    def set_other(self, other_revision, possible_transports=None):
 
309
        """Set the revision and tree to merge from.
 
310
 
 
311
        This sets the other_tree, other_rev_id, other_basis attributes.
 
312
 
 
313
        :param other_revision: The [path, revision] list to merge from.
 
314
        """
 
315
        self.other_branch, self.other_tree = self._get_tree(other_revision,
 
316
                                                            possible_transports)
191
317
        if other_revision[1] == -1:
192
 
            self.other_rev_id = other_branch.last_revision()
193
 
            if self.other_rev_id is None:
194
 
                raise NoCommits(other_branch)
 
318
            self.other_rev_id = _mod_revision.ensure_null(
 
319
                self.other_branch.last_revision())
 
320
            if _mod_revision.is_null(self.other_rev_id):
 
321
                raise NoCommits(self.other_branch)
195
322
            self.other_basis = self.other_rev_id
196
323
        elif other_revision[1] is not None:
197
 
            self.other_rev_id = other_branch.get_rev_id(other_revision[1])
 
324
            self.other_rev_id = self.other_branch.get_rev_id(other_revision[1])
198
325
            self.other_basis = self.other_rev_id
199
326
        else:
200
327
            self.other_rev_id = None
201
 
            self.other_basis = other_branch.last_revision()
 
328
            self.other_basis = self.other_branch.last_revision()
202
329
            if self.other_basis is None:
203
 
                raise NoCommits(other_branch)
204
 
        if other_branch.base != self.this_branch.base:
205
 
            self.this_branch.fetch(other_branch, last_revision=self.other_basis)
 
330
                raise NoCommits(self.other_branch)
 
331
        if self.other_rev_id is not None:
 
332
            self._cached_trees[self.other_rev_id] = self.other_tree
 
333
        self._maybe_fetch(self.other_branch,self.this_branch, self.other_basis)
 
334
 
 
335
    def set_other_revision(self, revision_id, other_branch):
 
336
        """Set 'other' based on a branch and revision id
 
337
 
 
338
        :param revision_id: The revision to use for a tree
 
339
        :param other_branch: The branch containing this tree
 
340
        """
 
341
        self.other_rev_id = revision_id
 
342
        self.other_branch = other_branch
 
343
        self._maybe_fetch(other_branch, self.this_branch, self.other_rev_id)
 
344
        self.other_tree = self.revision_tree(revision_id)
 
345
        self.other_basis = revision_id
 
346
 
 
347
    def set_base_revision(self, revision_id, branch):
 
348
        """Set 'base' based on a branch and revision id
 
349
 
 
350
        :param revision_id: The revision to use for a tree
 
351
        :param branch: The branch containing this tree
 
352
        """
 
353
        self.base_rev_id = revision_id
 
354
        self.base_branch = branch
 
355
        self._maybe_fetch(branch, self.this_branch, revision_id)
 
356
        self.base_tree = self.revision_tree(revision_id)
 
357
 
 
358
    def _maybe_fetch(self, source, target, revision_id):
 
359
        if not source.repository.has_same_location(target.repository):
 
360
            target.fetch(source, revision_id)
206
361
 
207
362
    def find_base(self):
208
 
        self.set_base([None, None])
 
363
        revisions = [ensure_null(self.this_basis),
 
364
                     ensure_null(self.other_basis)]
 
365
        if NULL_REVISION in revisions:
 
366
            self.base_rev_id = NULL_REVISION
 
367
            self.base_tree = self.revision_tree(self.base_rev_id)
 
368
            self._is_criss_cross = False
 
369
        else:
 
370
            lcas = self.revision_graph.find_lca(revisions[0], revisions[1])
 
371
            self._is_criss_cross = False
 
372
            if len(lcas) == 0:
 
373
                self.base_rev_id = NULL_REVISION
 
374
            elif len(lcas) == 1:
 
375
                self.base_rev_id = list(lcas)[0]
 
376
            else: # len(lcas) > 1
 
377
                if len(lcas) > 2:
 
378
                    # find_unique_lca can only handle 2 nodes, so we have to
 
379
                    # start back at the beginning. It is a shame to traverse
 
380
                    # the graph again, but better than re-implementing
 
381
                    # find_unique_lca.
 
382
                    self.base_rev_id = self.revision_graph.find_unique_lca(
 
383
                                            revisions[0], revisions[1])
 
384
                else:
 
385
                    self.base_rev_id = self.revision_graph.find_unique_lca(
 
386
                                            *lcas)
 
387
                self._is_criss_cross = True
 
388
            if self.base_rev_id == NULL_REVISION:
 
389
                raise UnrelatedBranches()
 
390
            if self._is_criss_cross:
 
391
                warning('Warning: criss-cross merge encountered.  See bzr'
 
392
                        ' help criss-cross.')
 
393
                mutter('Criss-cross lcas: %r' % lcas)
 
394
                interesting_revision_ids = [self.base_rev_id]
 
395
                interesting_revision_ids.extend(lcas)
 
396
                interesting_trees = dict((t.get_revision_id(), t)
 
397
                    for t in self.this_branch.repository.revision_trees(
 
398
                        interesting_revision_ids))
 
399
                self._cached_trees.update(interesting_trees)
 
400
                self.base_tree = interesting_trees.pop(self.base_rev_id)
 
401
                sorted_lca_keys = self.revision_graph.find_merge_order(
 
402
                    revisions[0], lcas)
 
403
                self._lca_trees = [interesting_trees[key]
 
404
                                   for key in sorted_lca_keys]
 
405
            else:
 
406
                self.base_tree = self.revision_tree(self.base_rev_id)
 
407
        self.base_is_ancestor = True
 
408
        self.base_is_other_ancestor = True
 
409
        mutter('Base revid: %r' % self.base_rev_id)
209
410
 
210
411
    def set_base(self, base_revision):
 
412
        """Set the base revision to use for the merge.
 
413
 
 
414
        :param base_revision: A 2-list containing a path and revision number.
 
415
        """
211
416
        mutter("doing merge() with no base_revision specified")
212
417
        if base_revision == [None, None]:
213
 
            try:
214
 
                pb = bzrlib.ui.ui_factory.nested_progress_bar()
215
 
                try:
216
 
                    this_repo = self.this_branch.repository
217
 
                    self.base_rev_id = common_ancestor(self.this_basis, 
218
 
                                                       self.other_basis, 
219
 
                                                       this_repo, pb)
220
 
                finally:
221
 
                    pb.finished()
222
 
            except NoCommonAncestor:
223
 
                raise UnrelatedBranches()
224
 
            self.base_tree = _get_revid_tree(self.this_branch, self.base_rev_id,
225
 
                                            None)
226
 
            self.base_is_ancestor = True
 
418
            self.find_base()
227
419
        else:
228
 
            base_branch, self.base_tree = _get_tree(base_revision)
 
420
            base_branch, self.base_tree = self._get_tree(base_revision)
229
421
            if base_revision[1] == -1:
230
422
                self.base_rev_id = base_branch.last_revision()
231
423
            elif base_revision[1] is None:
232
 
                self.base_rev_id = None
 
424
                self.base_rev_id = _mod_revision.NULL_REVISION
233
425
            else:
234
 
                self.base_rev_id = base_branch.get_rev_id(base_revision[1])
235
 
            if self.this_branch.base != base_branch.base:
236
 
                self.this_branch.fetch(base_branch)
237
 
            self.base_is_ancestor = is_ancestor(self.this_basis, 
238
 
                                                self.base_rev_id,
239
 
                                                self.this_branch)
 
426
                self.base_rev_id = _mod_revision.ensure_null(
 
427
                    base_branch.get_rev_id(base_revision[1]))
 
428
            self._maybe_fetch(base_branch, self.this_branch, self.base_rev_id)
240
429
 
241
 
    def do_merge(self):
242
 
        kwargs = {'working_tree':self.this_tree, 'this_tree': self.this_tree, 
243
 
                  'other_tree': self.other_tree, 
 
430
    def make_merger(self):
 
431
        kwargs = {'working_tree':self.this_tree, 'this_tree': self.this_tree,
 
432
                  'other_tree': self.other_tree,
244
433
                  'interesting_ids': self.interesting_ids,
245
 
                  'pp': self.pp}
 
434
                  'interesting_files': self.interesting_files,
 
435
                  'pp': self.pp,
 
436
                  'do_merge': False}
246
437
        if self.merge_type.requires_base:
247
438
            kwargs['base_tree'] = self.base_tree
248
439
        if self.merge_type.supports_reprocess:
254
445
            kwargs['show_base'] = self.show_base
255
446
        elif self.show_base:
256
447
            raise BzrError("Showing base is not supported for this"
257
 
                                  " merge type. %s" % self.merge_type)
258
 
        merge = self.merge_type(pb=self._pb, **kwargs)
 
448
                           " merge type. %s" % self.merge_type)
 
449
        if (not getattr(self.merge_type, 'supports_reverse_cherrypick', True)
 
450
            and not self.base_is_other_ancestor):
 
451
            raise errors.CannotReverseCherrypick()
 
452
        if self.merge_type.supports_cherrypick:
 
453
            kwargs['cherrypick'] = (not self.base_is_ancestor or
 
454
                                    not self.base_is_other_ancestor)
 
455
        if self._is_criss_cross and getattr(self.merge_type,
 
456
                                            'supports_lca_trees', False):
 
457
            kwargs['lca_trees'] = self._lca_trees
 
458
        return self.merge_type(pb=self._pb,
 
459
                               change_reporter=self.change_reporter,
 
460
                               **kwargs)
 
461
 
 
462
    def _do_merge_to(self, merge):
 
463
        if self.other_branch is not None:
 
464
            self.other_branch.update_references(self.this_branch)
 
465
        merge.do_merge()
 
466
        if self.recurse == 'down':
 
467
            for relpath, file_id in self.this_tree.iter_references():
 
468
                sub_tree = self.this_tree.get_nested_tree(file_id, relpath)
 
469
                other_revision = self.other_tree.get_reference_revision(
 
470
                    file_id, relpath)
 
471
                if  other_revision == sub_tree.last_revision():
 
472
                    continue
 
473
                sub_merge = Merger(sub_tree.branch, this_tree=sub_tree)
 
474
                sub_merge.merge_type = self.merge_type
 
475
                other_branch = self.other_branch.reference_parent(file_id, relpath)
 
476
                sub_merge.set_other_revision(other_revision, other_branch)
 
477
                base_revision = self.base_tree.get_reference_revision(file_id)
 
478
                sub_merge.base_tree = \
 
479
                    sub_tree.branch.repository.revision_tree(base_revision)
 
480
                sub_merge.base_rev_id = base_revision
 
481
                sub_merge.do_merge()
 
482
 
 
483
    def do_merge(self):
 
484
        self.this_tree.lock_tree_write()
 
485
        try:
 
486
            if self.base_tree is not None:
 
487
                self.base_tree.lock_read()
 
488
            try:
 
489
                if self.other_tree is not None:
 
490
                    self.other_tree.lock_read()
 
491
                try:
 
492
                    merge = self.make_merger()
 
493
                    self._do_merge_to(merge)
 
494
                finally:
 
495
                    if self.other_tree is not None:
 
496
                        self.other_tree.unlock()
 
497
            finally:
 
498
                if self.base_tree is not None:
 
499
                    self.base_tree.unlock()
 
500
        finally:
 
501
            self.this_tree.unlock()
259
502
        if len(merge.cooked_conflicts) == 0:
260
 
            if not self.ignore_zero:
 
503
            if not self.ignore_zero and not is_quiet():
261
504
                note("All changes applied successfully.")
262
505
        else:
263
506
            note("%d conflicts encountered." % len(merge.cooked_conflicts))
264
507
 
265
508
        return len(merge.cooked_conflicts)
266
509
 
267
 
    def regen_inventory(self, new_entries):
268
 
        old_entries = self.this_tree.read_working_inventory()
269
 
        new_inventory = {}
270
 
        by_path = {}
271
 
        new_entries_map = {} 
272
 
        for path, file_id in new_entries:
273
 
            if path is None:
274
 
                continue
275
 
            new_entries_map[file_id] = path
276
 
 
277
 
        def id2path(file_id):
278
 
            path = new_entries_map.get(file_id)
279
 
            if path is not None:
280
 
                return path
281
 
            entry = old_entries[file_id]
282
 
            if entry.parent_id is None:
283
 
                return entry.name
284
 
            return pathjoin(id2path(entry.parent_id), entry.name)
285
 
            
286
 
        for file_id in old_entries:
287
 
            entry = old_entries[file_id]
288
 
            path = id2path(file_id)
289
 
            if file_id in self.base_tree.inventory:
290
 
                executable = getattr(self.base_tree.inventory[file_id], 'executable', False)
291
 
            else:
292
 
                executable = getattr(entry, 'executable', False)
293
 
            new_inventory[file_id] = (path, file_id, entry.parent_id, 
294
 
                                      entry.kind, executable)
295
 
                                      
296
 
            by_path[path] = file_id
297
 
        
298
 
        deletions = 0
299
 
        insertions = 0
300
 
        new_path_list = []
301
 
        for path, file_id in new_entries:
302
 
            if path is None:
303
 
                del new_inventory[file_id]
304
 
                deletions += 1
305
 
            else:
306
 
                new_path_list.append((path, file_id))
307
 
                if file_id not in old_entries:
308
 
                    insertions += 1
309
 
        # Ensure no file is added before its parent
310
 
        new_path_list.sort()
311
 
        for path, file_id in new_path_list:
312
 
            if path == '':
313
 
                parent = None
314
 
            else:
315
 
                parent = by_path[os.path.dirname(path)]
316
 
            abspath = pathjoin(self.this_tree.basedir, path)
317
 
            kind = bzrlib.osutils.file_kind(abspath)
318
 
            if file_id in self.base_tree.inventory:
319
 
                executable = getattr(self.base_tree.inventory[file_id], 'executable', False)
320
 
            else:
321
 
                executable = False
322
 
            new_inventory[file_id] = (path, file_id, parent, kind, executable)
323
 
            by_path[path] = file_id 
324
 
 
325
 
        # Get a list in insertion order
326
 
        new_inventory_list = new_inventory.values()
327
 
        mutter ("""Inventory regeneration:
328
 
    old length: %i insertions: %i deletions: %i new_length: %i"""\
329
 
            % (len(old_entries), insertions, deletions, 
330
 
               len(new_inventory_list)))
331
 
        assert len(new_inventory_list) == len(old_entries) + insertions\
332
 
            - deletions
333
 
        new_inventory_list.sort()
334
 
        return new_inventory_list
 
510
 
 
511
class _InventoryNoneEntry(object):
 
512
    """This represents an inventory entry which *isn't there*.
 
513
 
 
514
    It simplifies the merging logic if we always have an InventoryEntry, even
 
515
    if it isn't actually present
 
516
    """
 
517
    executable = None
 
518
    kind = None
 
519
    name = None
 
520
    parent_id = None
 
521
    revision = None
 
522
    symlink_target = None
 
523
    text_sha1 = None
 
524
 
 
525
_none_entry = _InventoryNoneEntry()
335
526
 
336
527
 
337
528
class Merge3Merger(object):
340
531
    supports_reprocess = True
341
532
    supports_show_base = True
342
533
    history_based = False
 
534
    supports_cherrypick = True
 
535
    supports_reverse_cherrypick = True
 
536
    winner_idx = {"this": 2, "other": 1, "conflict": 1}
 
537
    supports_lca_trees = True
343
538
 
344
 
    def __init__(self, working_tree, this_tree, base_tree, other_tree, 
 
539
    def __init__(self, working_tree, this_tree, base_tree, other_tree,
345
540
                 interesting_ids=None, reprocess=False, show_base=False,
346
 
                 pb=DummyProgress(), pp=None):
347
 
        """Initialize the merger object and perform the merge."""
 
541
                 pb=DummyProgress(), pp=None, change_reporter=None,
 
542
                 interesting_files=None, do_merge=True,
 
543
                 cherrypick=False, lca_trees=None):
 
544
        """Initialize the merger object and perform the merge.
 
545
 
 
546
        :param working_tree: The working tree to apply the merge to
 
547
        :param this_tree: The local tree in the merge operation
 
548
        :param base_tree: The common tree in the merge operation
 
549
        :param other_tree: The other tree to merge changes from
 
550
        :param interesting_ids: The file_ids of files that should be
 
551
            participate in the merge.  May not be combined with
 
552
            interesting_files.
 
553
        :param: reprocess If True, perform conflict-reduction processing.
 
554
        :param show_base: If True, show the base revision in text conflicts.
 
555
            (incompatible with reprocess)
 
556
        :param pb: A Progress bar
 
557
        :param pp: A ProgressPhase object
 
558
        :param change_reporter: An object that should report changes made
 
559
        :param interesting_files: The tree-relative paths of files that should
 
560
            participate in the merge.  If these paths refer to directories,
 
561
            the contents of those directories will also be included.  May not
 
562
            be combined with interesting_ids.  If neither interesting_files nor
 
563
            interesting_ids is specified, all files may participate in the
 
564
            merge.
 
565
        :param lca_trees: Can be set to a dictionary of {revision_id:rev_tree}
 
566
            if the ancestry was found to include a criss-cross merge.
 
567
            Otherwise should be None.
 
568
        """
348
569
        object.__init__(self)
 
570
        if interesting_files is not None and interesting_ids is not None:
 
571
            raise ValueError(
 
572
                'specify either interesting_ids or interesting_files')
 
573
        self.interesting_ids = interesting_ids
 
574
        self.interesting_files = interesting_files
349
575
        self.this_tree = working_tree
350
576
        self.base_tree = base_tree
351
577
        self.other_tree = other_tree
353
579
        self.cooked_conflicts = []
354
580
        self.reprocess = reprocess
355
581
        self.show_base = show_base
 
582
        self._lca_trees = lca_trees
 
583
        # Uncommenting this will change the default algorithm to always use
 
584
        # _entries_lca. This can be useful for running the test suite and
 
585
        # making sure we haven't missed any corner cases.
 
586
        # if lca_trees is None:
 
587
        #     self._lca_trees = [self.base_tree]
356
588
        self.pb = pb
357
589
        self.pp = pp
 
590
        self.change_reporter = change_reporter
 
591
        self.cherrypick = cherrypick
358
592
        if self.pp is None:
359
593
            self.pp = ProgressPhase("Merge phase", 3, self.pb)
 
594
        if do_merge:
 
595
            self.do_merge()
360
596
 
361
 
        if interesting_ids is not None:
362
 
            all_ids = interesting_ids
363
 
        else:
364
 
            all_ids = set(base_tree)
365
 
            all_ids.update(other_tree)
366
 
        working_tree.lock_write()
367
 
        self.tt = TreeTransform(working_tree, self.pb)
 
597
    def do_merge(self):
 
598
        self.this_tree.lock_tree_write()
 
599
        self.base_tree.lock_read()
 
600
        self.other_tree.lock_read()
 
601
        self.tt = TreeTransform(self.this_tree, self.pb)
368
602
        try:
369
603
            self.pp.next_phase()
370
 
            child_pb = ui.ui_factory.nested_progress_bar()
371
 
            try:
372
 
                for num, file_id in enumerate(all_ids):
373
 
                    child_pb.update('Preparing file merge', num, len(all_ids))
374
 
                    self.merge_names(file_id)
375
 
                    file_status = self.merge_contents(file_id)
376
 
                    self.merge_executable(file_id, file_status)
377
 
            finally:
378
 
                child_pb.finished()
379
 
                
380
 
            self.pp.next_phase()
381
 
            child_pb = ui.ui_factory.nested_progress_bar()
382
 
            try:
383
 
                fs_conflicts = resolve_conflicts(self.tt, child_pb)
384
 
            finally:
385
 
                child_pb.finished()
386
 
            self.cook_conflicts(fs_conflicts)
387
 
            for conflict in self.cooked_conflicts:
388
 
                warning(conflict)
389
 
            self.pp.next_phase()
390
 
            results = self.tt.apply()
 
604
            self._compute_transform()
 
605
            self.pp.next_phase()
 
606
            results = self.tt.apply(no_conflicts=True)
391
607
            self.write_modified(results)
392
608
            try:
393
 
                working_tree.add_conflicts(self.cooked_conflicts)
 
609
                self.this_tree.add_conflicts(self.cooked_conflicts)
394
610
            except UnsupportedOperation:
395
611
                pass
396
612
        finally:
397
613
            self.tt.finalize()
398
 
            working_tree.unlock()
399
 
            self.pb.clear()
 
614
            self.other_tree.unlock()
 
615
            self.base_tree.unlock()
 
616
            self.this_tree.unlock()
 
617
            self.pb.clear()
 
618
 
 
619
    def make_preview_transform(self):
 
620
        self.base_tree.lock_read()
 
621
        self.other_tree.lock_read()
 
622
        self.tt = TransformPreview(self.this_tree)
 
623
        try:
 
624
            self.pp.next_phase()
 
625
            self._compute_transform()
 
626
            self.pp.next_phase()
 
627
        finally:
 
628
            self.other_tree.unlock()
 
629
            self.base_tree.unlock()
 
630
            self.pb.clear()
 
631
        return self.tt
 
632
 
 
633
    def _compute_transform(self):
 
634
        if self._lca_trees is None:
 
635
            entries = self._entries3()
 
636
            resolver = self._three_way
 
637
        else:
 
638
            entries = self._entries_lca()
 
639
            resolver = self._lca_multi_way
 
640
        child_pb = ui.ui_factory.nested_progress_bar()
 
641
        try:
 
642
            for num, (file_id, changed, parents3, names3,
 
643
                      executable3) in enumerate(entries):
 
644
                child_pb.update('Preparing file merge', num, len(entries))
 
645
                self._merge_names(file_id, parents3, names3, resolver=resolver)
 
646
                if changed:
 
647
                    file_status = self.merge_contents(file_id)
 
648
                else:
 
649
                    file_status = 'unmodified'
 
650
                self._merge_executable(file_id,
 
651
                    executable3, file_status, resolver=resolver)
 
652
        finally:
 
653
            child_pb.finished()
 
654
        self.fix_root()
 
655
        self.pp.next_phase()
 
656
        child_pb = ui.ui_factory.nested_progress_bar()
 
657
        try:
 
658
            fs_conflicts = resolve_conflicts(self.tt, child_pb,
 
659
                lambda t, c: conflict_pass(t, c, self.other_tree))
 
660
        finally:
 
661
            child_pb.finished()
 
662
        if self.change_reporter is not None:
 
663
            from bzrlib import delta
 
664
            delta.report_changes(
 
665
                self.tt.iter_changes(), self.change_reporter)
 
666
        self.cook_conflicts(fs_conflicts)
 
667
        for conflict in self.cooked_conflicts:
 
668
            warning(conflict)
 
669
 
 
670
    def _entries3(self):
 
671
        """Gather data about files modified between three trees.
 
672
 
 
673
        Return a list of tuples of file_id, changed, parents3, names3,
 
674
        executable3.  changed is a boolean indicating whether the file contents
 
675
        or kind were changed.  parents3 is a tuple of parent ids for base,
 
676
        other and this.  names3 is a tuple of names for base, other and this.
 
677
        executable3 is a tuple of execute-bit values for base, other and this.
 
678
        """
 
679
        result = []
 
680
        iterator = self.other_tree.iter_changes(self.base_tree,
 
681
                include_unchanged=True, specific_files=self.interesting_files,
 
682
                extra_trees=[self.this_tree])
 
683
        this_entries = dict((e.file_id, e) for p, e in
 
684
                            self.this_tree.iter_entries_by_dir(
 
685
                            self.interesting_ids))
 
686
        for (file_id, paths, changed, versioned, parents, names, kind,
 
687
             executable) in iterator:
 
688
            if (self.interesting_ids is not None and
 
689
                file_id not in self.interesting_ids):
 
690
                continue
 
691
            entry = this_entries.get(file_id)
 
692
            if entry is not None:
 
693
                this_name = entry.name
 
694
                this_parent = entry.parent_id
 
695
                this_executable = entry.executable
 
696
            else:
 
697
                this_name = None
 
698
                this_parent = None
 
699
                this_executable = None
 
700
            parents3 = parents + (this_parent,)
 
701
            names3 = names + (this_name,)
 
702
            executable3 = executable + (this_executable,)
 
703
            result.append((file_id, changed, parents3, names3, executable3))
 
704
        return result
 
705
 
 
706
    def _entries_lca(self):
 
707
        """Gather data about files modified between multiple trees.
 
708
 
 
709
        This compares OTHER versus all LCA trees, and for interesting entries,
 
710
        it then compares with THIS and BASE.
 
711
 
 
712
        For the multi-valued entries, the format will be (BASE, [lca1, lca2])
 
713
        :return: [(file_id, changed, parents, names, executable)]
 
714
            file_id     Simple file_id of the entry
 
715
            changed     Boolean, True if the kind or contents changed
 
716
                        else False
 
717
            parents     ((base, [parent_id, in, lcas]), parent_id_other,
 
718
                         parent_id_this)
 
719
            names       ((base, [name, in, lcas]), name_in_other, name_in_this)
 
720
            executable  ((base, [exec, in, lcas]), exec_in_other, exec_in_this)
 
721
        """
 
722
        if self.interesting_files is not None:
 
723
            lookup_trees = [self.this_tree, self.base_tree]
 
724
            lookup_trees.extend(self._lca_trees)
 
725
            # I think we should include the lca trees as well
 
726
            interesting_ids = self.other_tree.paths2ids(self.interesting_files,
 
727
                                                        lookup_trees)
 
728
        else:
 
729
            interesting_ids = self.interesting_ids
 
730
        result = []
 
731
        walker = _mod_tree.MultiWalker(self.other_tree, self._lca_trees)
 
732
 
 
733
        base_inventory = self.base_tree.inventory
 
734
        this_inventory = self.this_tree.inventory
 
735
        for path, file_id, other_ie, lca_values in walker.iter_all():
 
736
            # Is this modified at all from any of the other trees?
 
737
            if other_ie is None:
 
738
                other_ie = _none_entry
 
739
            if interesting_ids is not None and file_id not in interesting_ids:
 
740
                continue
 
741
 
 
742
            # If other_revision is found in any of the lcas, that means this
 
743
            # node is uninteresting. This is because when merging, if there are
 
744
            # multiple heads(), we have to create a new node. So if we didn't,
 
745
            # we know that the ancestry is linear, and that OTHER did not
 
746
            # modify anything
 
747
            # See doc/developers/lca_merge_resolution.txt for details
 
748
            other_revision = other_ie.revision
 
749
            if other_revision is not None:
 
750
                # We can't use this shortcut when other_revision is None,
 
751
                # because it may be None because things are WorkingTrees, and
 
752
                # not because it is *actually* None.
 
753
                is_unmodified = False
 
754
                for lca_path, ie in lca_values:
 
755
                    if ie is not None and ie.revision == other_revision:
 
756
                        is_unmodified = True
 
757
                        break
 
758
                if is_unmodified:
 
759
                    continue
 
760
 
 
761
            lca_entries = []
 
762
            for lca_path, lca_ie in lca_values:
 
763
                if lca_ie is None:
 
764
                    lca_entries.append(_none_entry)
 
765
                else:
 
766
                    lca_entries.append(lca_ie)
 
767
 
 
768
            if file_id in base_inventory:
 
769
                base_ie = base_inventory[file_id]
 
770
            else:
 
771
                base_ie = _none_entry
 
772
 
 
773
            if file_id in this_inventory:
 
774
                this_ie = this_inventory[file_id]
 
775
            else:
 
776
                this_ie = _none_entry
 
777
 
 
778
            lca_kinds = []
 
779
            lca_parent_ids = []
 
780
            lca_names = []
 
781
            lca_executable = []
 
782
            for lca_ie in lca_entries:
 
783
                lca_kinds.append(lca_ie.kind)
 
784
                lca_parent_ids.append(lca_ie.parent_id)
 
785
                lca_names.append(lca_ie.name)
 
786
                lca_executable.append(lca_ie.executable)
 
787
 
 
788
            kind_winner = self._lca_multi_way(
 
789
                (base_ie.kind, lca_kinds),
 
790
                other_ie.kind, this_ie.kind)
 
791
            parent_id_winner = self._lca_multi_way(
 
792
                (base_ie.parent_id, lca_parent_ids),
 
793
                other_ie.parent_id, this_ie.parent_id)
 
794
            name_winner = self._lca_multi_way(
 
795
                (base_ie.name, lca_names),
 
796
                other_ie.name, this_ie.name)
 
797
 
 
798
            content_changed = True
 
799
            if kind_winner == 'this':
 
800
                # No kind change in OTHER, see if there are *any* changes
 
801
                if other_ie.kind == 'directory':
 
802
                    if parent_id_winner == 'this' and name_winner == 'this':
 
803
                        # No change for this directory in OTHER, skip
 
804
                        continue
 
805
                    content_changed = False
 
806
                elif other_ie.kind is None or other_ie.kind == 'file':
 
807
                    def get_sha1(ie, tree):
 
808
                        if ie.kind != 'file':
 
809
                            return None
 
810
                        return tree.get_file_sha1(file_id)
 
811
                    base_sha1 = get_sha1(base_ie, self.base_tree)
 
812
                    lca_sha1s = [get_sha1(ie, tree) for ie, tree
 
813
                                 in zip(lca_entries, self._lca_trees)]
 
814
                    this_sha1 = get_sha1(this_ie, self.this_tree)
 
815
                    other_sha1 = get_sha1(other_ie, self.other_tree)
 
816
                    sha1_winner = self._lca_multi_way(
 
817
                        (base_sha1, lca_sha1s), other_sha1, this_sha1,
 
818
                        allow_overriding_lca=False)
 
819
                    exec_winner = self._lca_multi_way(
 
820
                        (base_ie.executable, lca_executable),
 
821
                        other_ie.executable, this_ie.executable)
 
822
                    if (parent_id_winner == 'this' and name_winner == 'this'
 
823
                        and sha1_winner == 'this' and exec_winner == 'this'):
 
824
                        # No kind, parent, name, exec, or content change for
 
825
                        # OTHER, so this node is not considered interesting
 
826
                        continue
 
827
                    if sha1_winner == 'this':
 
828
                        content_changed = False
 
829
                elif other_ie.kind == 'symlink':
 
830
                    def get_target(ie, tree):
 
831
                        if ie.kind != 'symlink':
 
832
                            return None
 
833
                        return tree.get_symlink_target(file_id)
 
834
                    base_target = get_target(base_ie, self.base_tree)
 
835
                    lca_targets = [get_target(ie, tree) for ie, tree
 
836
                                   in zip(lca_entries, self._lca_trees)]
 
837
                    this_target = get_target(this_ie, self.this_tree)
 
838
                    other_target = get_target(other_ie, self.other_tree)
 
839
                    target_winner = self._lca_multi_way(
 
840
                        (base_target, lca_targets),
 
841
                        other_target, this_target)
 
842
                    if (parent_id_winner == 'this' and name_winner == 'this'
 
843
                        and target_winner == 'this'):
 
844
                        # No kind, parent, name, or symlink target change
 
845
                        # not interesting
 
846
                        continue
 
847
                    if target_winner == 'this':
 
848
                        content_changed = False
 
849
                elif other_ie.kind == 'tree-reference':
 
850
                    # The 'changed' information seems to be handled at a higher
 
851
                    # level. At least, _entries3 returns False for content
 
852
                    # changed, even when at a new revision_id.
 
853
                    content_changed = False
 
854
                    if (parent_id_winner == 'this' and name_winner == 'this'):
 
855
                        # Nothing interesting
 
856
                        continue
 
857
                else:
 
858
                    raise AssertionError('unhandled kind: %s' % other_ie.kind)
 
859
                # XXX: We need to handle kind == 'symlink'
 
860
 
 
861
            # If we have gotten this far, that means something has changed
 
862
            result.append((file_id, content_changed,
 
863
                           ((base_ie.parent_id, lca_parent_ids),
 
864
                            other_ie.parent_id, this_ie.parent_id),
 
865
                           ((base_ie.name, lca_names),
 
866
                            other_ie.name, this_ie.name),
 
867
                           ((base_ie.executable, lca_executable),
 
868
                            other_ie.executable, this_ie.executable)
 
869
                          ))
 
870
        return result
 
871
 
 
872
 
 
873
    def fix_root(self):
 
874
        try:
 
875
            self.tt.final_kind(self.tt.root)
 
876
        except NoSuchFile:
 
877
            self.tt.cancel_deletion(self.tt.root)
 
878
        if self.tt.final_file_id(self.tt.root) is None:
 
879
            self.tt.version_file(self.tt.tree_file_id(self.tt.root),
 
880
                                 self.tt.root)
 
881
        other_root_file_id = self.other_tree.get_root_id()
 
882
        if other_root_file_id is None:
 
883
            return
 
884
        other_root = self.tt.trans_id_file_id(other_root_file_id)
 
885
        if other_root == self.tt.root:
 
886
            return
 
887
        try:
 
888
            self.tt.final_kind(other_root)
 
889
        except NoSuchFile:
 
890
            return
 
891
        if self.other_tree.inventory.root.file_id in self.this_tree.inventory:
 
892
            # the other tree's root is a non-root in the current tree
 
893
            return
 
894
        self.reparent_children(self.other_tree.inventory.root, self.tt.root)
 
895
        self.tt.cancel_creation(other_root)
 
896
        self.tt.cancel_versioning(other_root)
 
897
 
 
898
    def reparent_children(self, ie, target):
 
899
        for thing, child in ie.children.iteritems():
 
900
            trans_id = self.tt.trans_id_file_id(child.file_id)
 
901
            self.tt.adjust_path(self.tt.final_name(trans_id), target, trans_id)
400
902
 
401
903
    def write_modified(self, results):
402
904
        modified_hashes = {}
423
925
        if entry is None:
424
926
            return None
425
927
        return entry.name
426
 
    
 
928
 
427
929
    @staticmethod
428
930
    def contents_sha1(tree, file_id):
429
931
        """Determine the sha1 of the file contents (used as a key method)."""
448
950
        return tree.kind(file_id)
449
951
 
450
952
    @staticmethod
 
953
    def _three_way(base, other, this):
 
954
        #if base == other, either they all agree, or only THIS has changed.
 
955
        if base == other:
 
956
            return 'this'
 
957
        elif this not in (base, other):
 
958
            return 'conflict'
 
959
        # "Ambiguous clean merge" -- both sides have made the same change.
 
960
        elif this == other:
 
961
            return "this"
 
962
        # this == base: only other has changed.
 
963
        else:
 
964
            return "other"
 
965
 
 
966
    @staticmethod
 
967
    def _lca_multi_way(bases, other, this, allow_overriding_lca=True):
 
968
        """Consider LCAs when determining whether a change has occurred.
 
969
 
 
970
        If LCAS are all identical, this is the same as a _three_way comparison.
 
971
 
 
972
        :param bases: value in (BASE, [LCAS])
 
973
        :param other: value in OTHER
 
974
        :param this: value in THIS
 
975
        :param allow_overriding_lca: If there is more than one unique lca
 
976
            value, allow OTHER to override THIS if it has a new value, and
 
977
            THIS only has an lca value, or vice versa. This is appropriate for
 
978
            truly scalar values, not as much for non-scalars.
 
979
        :return: 'this', 'other', or 'conflict' depending on whether an entry
 
980
            changed or not.
 
981
        """
 
982
        # See doc/developers/lca_tree_merging.txt for details about this
 
983
        # algorithm.
 
984
        if other == this:
 
985
            # Either Ambiguously clean, or nothing was actually changed. We
 
986
            # don't really care
 
987
            return 'this'
 
988
        base_val, lca_vals = bases
 
989
        # Remove 'base_val' from the lca_vals, because it is not interesting
 
990
        filtered_lca_vals = [lca_val for lca_val in lca_vals
 
991
                                      if lca_val != base_val]
 
992
        if len(filtered_lca_vals) == 0:
 
993
            return Merge3Merger._three_way(base_val, other, this)
 
994
 
 
995
        unique_lca_vals = set(filtered_lca_vals)
 
996
        if len(unique_lca_vals) == 1:
 
997
            return Merge3Merger._three_way(unique_lca_vals.pop(), other, this)
 
998
 
 
999
        if allow_overriding_lca:
 
1000
            if other in unique_lca_vals:
 
1001
                if this in unique_lca_vals:
 
1002
                    # Each side picked a different lca, conflict
 
1003
                    return 'conflict'
 
1004
                else:
 
1005
                    # This has a value which supersedes both lca values, and
 
1006
                    # other only has an lca value
 
1007
                    return 'this'
 
1008
            elif this in unique_lca_vals:
 
1009
                # OTHER has a value which supersedes both lca values, and this
 
1010
                # only has an lca value
 
1011
                return 'other'
 
1012
 
 
1013
        # At this point, the lcas disagree, and the tips disagree
 
1014
        return 'conflict'
 
1015
 
 
1016
    @staticmethod
451
1017
    def scalar_three_way(this_tree, base_tree, other_tree, file_id, key):
452
1018
        """Do a three-way test on a scalar.
453
1019
        Return "this", "other" or "conflict", depending whether a value wins.
458
1024
        if key_base == key_other:
459
1025
            return "this"
460
1026
        key_this = key(this_tree, file_id)
461
 
        if key_this not in (key_base, key_other):
 
1027
        # "Ambiguous clean merge"
 
1028
        if key_this == key_other:
 
1029
            return "this"
 
1030
        elif key_this == key_base:
 
1031
            return "other"
 
1032
        else:
462
1033
            return "conflict"
463
 
        # "Ambiguous clean merge"
464
 
        elif key_this == key_other:
465
 
            return "this"
466
 
        else:
467
 
            assert key_this == key_base
468
 
            return "other"
469
1034
 
470
1035
    def merge_names(self, file_id):
471
 
        """Perform a merge on file_id names and parents"""
472
1036
        def get_entry(tree):
473
1037
            if file_id in tree.inventory:
474
1038
                return tree.inventory[file_id]
477
1041
        this_entry = get_entry(self.this_tree)
478
1042
        other_entry = get_entry(self.other_tree)
479
1043
        base_entry = get_entry(self.base_tree)
480
 
        name_winner = self.scalar_three_way(this_entry, base_entry, 
481
 
                                            other_entry, file_id, self.name)
482
 
        parent_id_winner = self.scalar_three_way(this_entry, base_entry, 
483
 
                                                 other_entry, file_id, 
484
 
                                                 self.parent)
485
 
        if this_entry is None:
 
1044
        entries = (base_entry, other_entry, this_entry)
 
1045
        names = []
 
1046
        parents = []
 
1047
        for entry in entries:
 
1048
            if entry is None:
 
1049
                names.append(None)
 
1050
                parents.append(None)
 
1051
            else:
 
1052
                names.append(entry.name)
 
1053
                parents.append(entry.parent_id)
 
1054
        return self._merge_names(file_id, parents, names,
 
1055
                                 resolver=self._three_way)
 
1056
 
 
1057
    def _merge_names(self, file_id, parents, names, resolver):
 
1058
        """Perform a merge on file_id names and parents"""
 
1059
        base_name, other_name, this_name = names
 
1060
        base_parent, other_parent, this_parent = parents
 
1061
 
 
1062
        name_winner = resolver(*names)
 
1063
 
 
1064
        parent_id_winner = resolver(*parents)
 
1065
        if this_name is None:
486
1066
            if name_winner == "this":
487
1067
                name_winner = "other"
488
1068
            if parent_id_winner == "this":
491
1071
            return
492
1072
        if name_winner == "conflict":
493
1073
            trans_id = self.tt.trans_id_file_id(file_id)
494
 
            self._raw_conflicts.append(('name conflict', trans_id, 
495
 
                                        self.name(this_entry, file_id), 
496
 
                                        self.name(other_entry, file_id)))
 
1074
            self._raw_conflicts.append(('name conflict', trans_id,
 
1075
                                        this_name, other_name))
497
1076
        if parent_id_winner == "conflict":
498
1077
            trans_id = self.tt.trans_id_file_id(file_id)
499
 
            self._raw_conflicts.append(('parent conflict', trans_id, 
500
 
                                        self.parent(this_entry, file_id), 
501
 
                                        self.parent(other_entry, file_id)))
502
 
        if other_entry is None:
503
 
            # it doesn't matter whether the result was 'other' or 
 
1078
            self._raw_conflicts.append(('parent conflict', trans_id,
 
1079
                                        this_parent, other_parent))
 
1080
        if other_name is None:
 
1081
            # it doesn't matter whether the result was 'other' or
504
1082
            # 'conflict'-- if there's no 'other', we leave it alone.
505
1083
            return
506
1084
        # if we get here, name_winner and parent_winner are set to safe values.
507
 
        winner_entry = {"this": this_entry, "other": other_entry, 
508
 
                        "conflict": other_entry}
509
1085
        trans_id = self.tt.trans_id_file_id(file_id)
510
 
        parent_id = winner_entry[parent_id_winner].parent_id
511
 
        parent_trans_id = self.tt.trans_id_file_id(parent_id)
512
 
        self.tt.adjust_path(winner_entry[name_winner].name, parent_trans_id,
513
 
                            trans_id)
 
1086
        parent_id = parents[self.winner_idx[parent_id_winner]]
 
1087
        if parent_id is not None:
 
1088
            parent_trans_id = self.tt.trans_id_file_id(parent_id)
 
1089
            self.tt.adjust_path(names[self.winner_idx[name_winner]],
 
1090
                                parent_trans_id, trans_id)
514
1091
 
515
1092
    def merge_contents(self, file_id):
516
 
        """Performa a merge on file_id contents."""
 
1093
        """Performs a merge on file_id contents."""
517
1094
        def contents_pair(tree):
518
1095
            if file_id not in tree:
519
1096
                return (None, None)
532
1109
            parent_id = self.tt.final_parent(trans_id)
533
1110
            if file_id in self.this_tree.inventory:
534
1111
                self.tt.unversion_file(trans_id)
535
 
                self.tt.delete_contents(trans_id)
536
 
            file_group = self._dump_conflicts(name, parent_id, file_id, 
 
1112
                if file_id in self.this_tree:
 
1113
                    self.tt.delete_contents(trans_id)
 
1114
            file_group = self._dump_conflicts(name, parent_id, file_id,
537
1115
                                              set_version=True)
538
1116
            self._raw_conflicts.append(('contents conflict', file_group))
539
1117
 
542
1120
        # file kind...
543
1121
        base_pair = contents_pair(self.base_tree)
544
1122
        other_pair = contents_pair(self.other_tree)
545
 
        if base_pair == other_pair:
546
 
            # OTHER introduced no changes
547
 
            return "unmodified"
548
 
        this_pair = contents_pair(self.this_tree)
549
 
        if this_pair == other_pair:
550
 
            # THIS and OTHER introduced the same changes
551
 
            return "unmodified"
552
 
        else:
553
 
            trans_id = self.tt.trans_id_file_id(file_id)
554
 
            if this_pair == base_pair:
555
 
                # only OTHER introduced changes
556
 
                if file_id in self.this_tree:
557
 
                    # Remove any existing contents
558
 
                    self.tt.delete_contents(trans_id)
559
 
                if file_id in self.other_tree:
560
 
                    # OTHER changed the file
561
 
                    create_by_entry(self.tt, 
562
 
                                    self.other_tree.inventory[file_id], 
563
 
                                    self.other_tree, trans_id)
564
 
                    if file_id not in self.this_tree.inventory:
565
 
                        self.tt.version_file(file_id, trans_id)
566
 
                    return "modified"
567
 
                elif file_id in self.this_tree.inventory:
568
 
                    # OTHER deleted the file
569
 
                    self.tt.unversion_file(trans_id)
570
 
                    return "deleted"
571
 
            #BOTH THIS and OTHER introduced changes; scalar conflict
572
 
            elif this_pair[0] == "file" and other_pair[0] == "file":
 
1123
        if self._lca_trees:
 
1124
            this_pair = contents_pair(self.this_tree)
 
1125
            lca_pairs = [contents_pair(tree) for tree in self._lca_trees]
 
1126
            winner = self._lca_multi_way((base_pair, lca_pairs), other_pair,
 
1127
                                         this_pair, allow_overriding_lca=False)
 
1128
        else:
 
1129
            if base_pair == other_pair:
 
1130
                winner = 'this'
 
1131
            else:
 
1132
                # We delayed evaluating this_pair as long as we can to avoid
 
1133
                # unnecessary sha1 calculation
 
1134
                this_pair = contents_pair(self.this_tree)
 
1135
                winner = self._three_way(base_pair, other_pair, this_pair)
 
1136
        if winner == 'this':
 
1137
            # No interesting changes introduced by OTHER
 
1138
            return "unmodified"
 
1139
        trans_id = self.tt.trans_id_file_id(file_id)
 
1140
        if winner == 'other':
 
1141
            # OTHER is a straight winner, so replace this contents with other
 
1142
            file_in_this = file_id in self.this_tree
 
1143
            if file_in_this:
 
1144
                # Remove any existing contents
 
1145
                self.tt.delete_contents(trans_id)
 
1146
            if file_id in self.other_tree:
 
1147
                # OTHER changed the file
 
1148
                create_from_tree(self.tt, trans_id,
 
1149
                                 self.other_tree, file_id)
 
1150
                if not file_in_this:
 
1151
                    self.tt.version_file(file_id, trans_id)
 
1152
                return "modified"
 
1153
            elif file_in_this:
 
1154
                # OTHER deleted the file
 
1155
                self.tt.unversion_file(trans_id)
 
1156
                return "deleted"
 
1157
        else:
 
1158
            # We have a hypothetical conflict, but if we have files, then we
 
1159
            # can try to merge the content
 
1160
            if this_pair[0] == 'file' and other_pair[0] == 'file':
573
1161
                # THIS and OTHER are both files, so text merge.  Either
574
1162
                # BASE is a file, or both converted to files, so at least we
575
1163
                # have agreement that output should be a file.
577
1165
                    self.text_merge(file_id, trans_id)
578
1166
                except BinaryFile:
579
1167
                    return contents_conflict()
580
 
                if file_id not in self.this_tree.inventory:
 
1168
                if file_id not in self.this_tree:
581
1169
                    self.tt.version_file(file_id, trans_id)
582
1170
                try:
583
1171
                    self.tt.tree_kind(trans_id)
586
1174
                    pass
587
1175
                return "modified"
588
1176
            else:
589
 
                # Scalar conflict, can't text merge.  Dump conflicts
590
1177
                return contents_conflict()
591
1178
 
592
1179
    def get_lines(self, tree, file_id):
607
1194
            base_lines = []
608
1195
        other_lines = self.get_lines(self.other_tree, file_id)
609
1196
        this_lines = self.get_lines(self.this_tree, file_id)
610
 
        m3 = Merge3(base_lines, this_lines, other_lines)
 
1197
        m3 = Merge3(base_lines, this_lines, other_lines,
 
1198
                    is_cherrypick=self.cherrypick)
611
1199
        start_marker = "!START OF MERGE CONFLICT!" + "I HOPE THIS IS UNIQUE"
612
1200
        if self.show_base is True:
613
1201
            base_marker = '|' * 7
616
1204
 
617
1205
        def iter_merge3(retval):
618
1206
            retval["text_conflicts"] = False
619
 
            for line in m3.merge_lines(name_a = "TREE", 
620
 
                                       name_b = "MERGE-SOURCE", 
 
1207
            for line in m3.merge_lines(name_a = "TREE",
 
1208
                                       name_b = "MERGE-SOURCE",
621
1209
                                       name_base = "BASE-REVISION",
622
 
                                       start_marker=start_marker, 
 
1210
                                       start_marker=start_marker,
623
1211
                                       base_marker=base_marker,
624
1212
                                       reprocess=self.reprocess):
625
1213
                if line.startswith(start_marker):
634
1222
            self._raw_conflicts.append(('text conflict', trans_id))
635
1223
            name = self.tt.final_name(trans_id)
636
1224
            parent_id = self.tt.final_parent(trans_id)
637
 
            file_group = self._dump_conflicts(name, parent_id, file_id, 
 
1225
            file_group = self._dump_conflicts(name, parent_id, file_id,
638
1226
                                              this_lines, base_lines,
639
1227
                                              other_lines)
640
1228
            file_group.append(trans_id)
641
1229
 
642
 
    def _dump_conflicts(self, name, parent_id, file_id, this_lines=None, 
 
1230
    def _dump_conflicts(self, name, parent_id, file_id, this_lines=None,
643
1231
                        base_lines=None, other_lines=None, set_version=False,
644
1232
                        no_base=False):
645
1233
        """Emit conflict files.
647
1235
        determined automatically.  If set_version is true, the .OTHER, .THIS
648
1236
        or .BASE (in that order) will be created as versioned files.
649
1237
        """
650
 
        data = [('OTHER', self.other_tree, other_lines), 
 
1238
        data = [('OTHER', self.other_tree, other_lines),
651
1239
                ('THIS', self.this_tree, this_lines)]
652
1240
        if not no_base:
653
1241
            data.append(('BASE', self.base_tree, base_lines))
662
1250
                    self.tt.version_file(file_id, trans_id)
663
1251
                    versioned = True
664
1252
        return file_group
665
 
           
666
 
    def _conflict_file(self, name, parent_id, tree, file_id, suffix, 
 
1253
 
 
1254
    def _conflict_file(self, name, parent_id, tree, file_id, suffix,
667
1255
                       lines=None):
668
1256
        """Emit a single conflict file."""
669
1257
        name = name + '.' + suffix
670
1258
        trans_id = self.tt.create_path(name, parent_id)
671
 
        entry = tree.inventory[file_id]
672
 
        create_by_entry(self.tt, entry, tree, trans_id, lines)
 
1259
        create_from_tree(self.tt, trans_id, tree, file_id, lines)
673
1260
        return trans_id
674
1261
 
675
1262
    def merge_executable(self, file_id, file_status):
676
1263
        """Perform a merge on the execute bit."""
 
1264
        executable = [self.executable(t, file_id) for t in (self.base_tree,
 
1265
                      self.other_tree, self.this_tree)]
 
1266
        self._merge_executable(file_id, executable, file_status,
 
1267
                               resolver=self._three_way)
 
1268
 
 
1269
    def _merge_executable(self, file_id, executable, file_status,
 
1270
                          resolver):
 
1271
        """Perform a merge on the execute bit."""
 
1272
        base_executable, other_executable, this_executable = executable
677
1273
        if file_status == "deleted":
678
1274
            return
679
 
        trans_id = self.tt.trans_id_file_id(file_id)
680
 
        try:
681
 
            if self.tt.final_kind(trans_id) != "file":
682
 
                return
683
 
        except NoSuchFile:
684
 
            return
685
 
        winner = self.scalar_three_way(self.this_tree, self.base_tree, 
686
 
                                       self.other_tree, file_id, 
687
 
                                       self.executable)
 
1275
        winner = resolver(*executable)
688
1276
        if winner == "conflict":
689
1277
        # There must be a None in here, if we have a conflict, but we
690
1278
        # need executability since file status was not deleted.
692
1280
                winner = "this"
693
1281
            else:
694
1282
                winner = "other"
 
1283
        if winner == 'this' and file_status != "modified":
 
1284
            return
 
1285
        trans_id = self.tt.trans_id_file_id(file_id)
 
1286
        try:
 
1287
            if self.tt.final_kind(trans_id) != "file":
 
1288
                return
 
1289
        except NoSuchFile:
 
1290
            return
695
1291
        if winner == "this":
696
 
            if file_status == "modified":
697
 
                executability = self.this_tree.is_executable(file_id)
698
 
                if executability is not None:
699
 
                    trans_id = self.tt.trans_id_file_id(file_id)
700
 
                    self.tt.set_executability(executability, trans_id)
 
1292
            executability = this_executable
701
1293
        else:
702
 
            assert winner == "other"
703
1294
            if file_id in self.other_tree:
704
 
                executability = self.other_tree.is_executable(file_id)
 
1295
                executability = other_executable
705
1296
            elif file_id in self.this_tree:
706
 
                executability = self.this_tree.is_executable(file_id)
 
1297
                executability = this_executable
707
1298
            elif file_id in self.base_tree:
708
 
                executability = self.base_tree.is_executable(file_id)
709
 
            if executability is not None:
710
 
                trans_id = self.tt.trans_id_file_id(file_id)
711
 
                self.tt.set_executability(executability, trans_id)
 
1299
                executability = base_executable
 
1300
        if executability is not None:
 
1301
            trans_id = self.tt.trans_id_file_id(file_id)
 
1302
            self.tt.set_executability(executability, trans_id)
712
1303
 
713
1304
    def cook_conflicts(self, fs_conflicts):
714
1305
        """Convert all conflicts into a form that doesn't depend on trans_id"""
723
1314
                conflict_args = conflict[2:]
724
1315
                if trans_id not in name_conflicts:
725
1316
                    name_conflicts[trans_id] = {}
726
 
                unique_add(name_conflicts[trans_id], conflict_type, 
 
1317
                unique_add(name_conflicts[trans_id], conflict_type,
727
1318
                           conflict_args)
728
1319
            if conflict_type == 'contents conflict':
729
1320
                for trans_id in conflict[1]:
747
1338
        for trans_id, conflicts in name_conflicts.iteritems():
748
1339
            try:
749
1340
                this_parent, other_parent = conflicts['parent conflict']
750
 
                assert this_parent != other_parent
 
1341
                if this_parent == other_parent:
 
1342
                    raise AssertionError()
751
1343
            except KeyError:
752
1344
                this_parent = other_parent = \
753
1345
                    self.tt.final_file_id(self.tt.final_parent(trans_id))
754
1346
            try:
755
1347
                this_name, other_name = conflicts['name conflict']
756
 
                assert this_name != other_name
 
1348
                if this_name == other_name:
 
1349
                    raise AssertionError()
757
1350
            except KeyError:
758
1351
                this_name = other_name = self.tt.final_name(trans_id)
759
1352
            other_path = fp.get_path(trans_id)
760
 
            if this_parent is not None:
 
1353
            if this_parent is not None and this_name is not None:
761
1354
                this_parent_path = \
762
1355
                    fp.get_path(self.tt.trans_id_file_id(this_parent))
763
1356
                this_path = pathjoin(this_parent_path, this_name)
774
1367
    """Three-way tree merger, text weave merger."""
775
1368
    supports_reprocess = True
776
1369
    supports_show_base = False
777
 
 
778
 
    def __init__(self, working_tree, this_tree, base_tree, other_tree, 
779
 
                 interesting_ids=None, pb=DummyProgress(), pp=None,
780
 
                 reprocess=False):
781
 
        self.this_revision_tree = self._get_revision_tree(this_tree)
782
 
        self.other_revision_tree = self._get_revision_tree(other_tree)
783
 
        super(WeaveMerger, self).__init__(working_tree, this_tree, 
784
 
                                          base_tree, other_tree, 
785
 
                                          interesting_ids=interesting_ids, 
786
 
                                          pb=pb, pp=pp, reprocess=reprocess)
787
 
 
788
 
    def _get_revision_tree(self, tree):
789
 
        """Return a revision tree related to this tree.
790
 
        If the tree is a WorkingTree, the basis will be returned.
791
 
        """
792
 
        if getattr(tree, 'get_weave', False) is False:
793
 
            # If we have a WorkingTree, try using the basis
794
 
            return tree.branch.basis_tree()
795
 
        else:
796
 
            return tree
797
 
 
798
 
    def _check_file(self, file_id):
799
 
        """Check that the revision tree's version of the file matches."""
800
 
        for tree, rt in ((self.this_tree, self.this_revision_tree), 
801
 
                         (self.other_tree, self.other_revision_tree)):
802
 
            if rt is tree:
803
 
                continue
804
 
            if tree.get_file_sha1(file_id) != rt.get_file_sha1(file_id):
805
 
                raise WorkingTreeNotRevision(self.this_tree)
 
1370
    supports_reverse_cherrypick = False
 
1371
    history_based = True
806
1372
 
807
1373
    def _merged_lines(self, file_id):
808
1374
        """Generate the merged lines.
809
1375
        There is no distinction between lines that are meant to contain <<<<<<<
810
1376
        and conflicts.
811
1377
        """
812
 
        weave = self.this_revision_tree.get_weave(file_id)
813
 
        this_revision_id = self.this_revision_tree.inventory[file_id].revision
814
 
        other_revision_id = \
815
 
            self.other_revision_tree.inventory[file_id].revision
816
 
        wm = WeaveMerge(weave, this_revision_id, other_revision_id, 
817
 
                        '<<<<<<< TREE\n', '>>>>>>> MERGE-SOURCE\n')
818
 
        return wm.merge_lines(self.reprocess)
 
1378
        if self.cherrypick:
 
1379
            base = self.base_tree
 
1380
        else:
 
1381
            base = None
 
1382
        plan = self.this_tree.plan_file_merge(file_id, self.other_tree,
 
1383
                                              base=base)
 
1384
        if 'merge' in debug.debug_flags:
 
1385
            plan = list(plan)
 
1386
            trans_id = self.tt.trans_id_file_id(file_id)
 
1387
            name = self.tt.final_name(trans_id) + '.plan'
 
1388
            contents = ('%10s|%s' % l for l in plan)
 
1389
            self.tt.new_file(name, self.tt.final_parent(trans_id), contents)
 
1390
        textmerge = PlanWeaveMerge(plan, '<<<<<<< TREE\n',
 
1391
            '>>>>>>> MERGE-SOURCE\n')
 
1392
        return textmerge.merge_lines(self.reprocess)
819
1393
 
820
1394
    def text_merge(self, file_id, trans_id):
821
1395
        """Perform a (weave) text merge for a given file and file-id.
822
1396
        If conflicts are encountered, .THIS and .OTHER files will be emitted,
823
1397
        and a conflict will be noted.
824
1398
        """
825
 
        self._check_file(file_id)
826
1399
        lines, conflicts = self._merged_lines(file_id)
827
1400
        lines = list(lines)
828
 
        # Note we're checking whether the OUTPUT is binary in this case, 
 
1401
        # Note we're checking whether the OUTPUT is binary in this case,
829
1402
        # because we don't want to get into weave merge guts.
830
1403
        check_text_lines(lines)
831
1404
        self.tt.create_file(lines, trans_id)
833
1406
            self._raw_conflicts.append(('text conflict', trans_id))
834
1407
            name = self.tt.final_name(trans_id)
835
1408
            parent_id = self.tt.final_parent(trans_id)
836
 
            file_group = self._dump_conflicts(name, parent_id, file_id, 
 
1409
            file_group = self._dump_conflicts(name, parent_id, file_id,
837
1410
                                              no_base=True)
838
1411
            file_group.append(trans_id)
839
1412
 
840
1413
 
 
1414
class LCAMerger(WeaveMerger):
 
1415
 
 
1416
    def _merged_lines(self, file_id):
 
1417
        """Generate the merged lines.
 
1418
        There is no distinction between lines that are meant to contain <<<<<<<
 
1419
        and conflicts.
 
1420
        """
 
1421
        if self.cherrypick:
 
1422
            base = self.base_tree
 
1423
        else:
 
1424
            base = None
 
1425
        plan = self.this_tree.plan_file_lca_merge(file_id, self.other_tree,
 
1426
                                                  base=base)
 
1427
        if 'merge' in debug.debug_flags:
 
1428
            plan = list(plan)
 
1429
            trans_id = self.tt.trans_id_file_id(file_id)
 
1430
            name = self.tt.final_name(trans_id) + '.plan'
 
1431
            contents = ('%10s|%s' % l for l in plan)
 
1432
            self.tt.new_file(name, self.tt.final_parent(trans_id), contents)
 
1433
        textmerge = PlanWeaveMerge(plan, '<<<<<<< TREE\n',
 
1434
            '>>>>>>> MERGE-SOURCE\n')
 
1435
        return textmerge.merge_lines(self.reprocess)
 
1436
 
 
1437
 
841
1438
class Diff3Merger(Merge3Merger):
842
1439
    """Three-way merger using external diff3 for text merging"""
843
1440
 
858
1455
        will be dumped, and a will be conflict noted.
859
1456
        """
860
1457
        import bzrlib.patch
861
 
        temp_dir = mkdtemp(prefix="bzr-")
 
1458
        temp_dir = osutils.mkdtemp(prefix="bzr-")
862
1459
        try:
863
1460
            new_file = pathjoin(temp_dir, "new")
864
1461
            this = self.dump_file(temp_dir, "this", self.this_tree, file_id)
876
1473
                name = self.tt.final_name(trans_id)
877
1474
                parent_id = self.tt.final_parent(trans_id)
878
1475
                self._dump_conflicts(name, parent_id, file_id)
879
 
            self._raw_conflicts.append(('text conflict', trans_id))
 
1476
                self._raw_conflicts.append(('text conflict', trans_id))
880
1477
        finally:
881
 
            rmtree(temp_dir)
 
1478
            osutils.rmtree(temp_dir)
882
1479
 
883
1480
 
884
1481
def merge_inner(this_branch, other_tree, base_tree, ignore_zero=False,
885
 
                backup_files=False, 
886
 
                merge_type=Merge3Merger, 
887
 
                interesting_ids=None, 
888
 
                show_base=False, 
889
 
                reprocess=False, 
 
1482
                backup_files=False,
 
1483
                merge_type=Merge3Merger,
 
1484
                interesting_ids=None,
 
1485
                show_base=False,
 
1486
                reprocess=False,
890
1487
                other_rev_id=None,
891
1488
                interesting_files=None,
892
1489
                this_tree=None,
893
 
                pb=DummyProgress()):
894
 
    """Primary interface for merging. 
 
1490
                pb=DummyProgress(),
 
1491
                change_reporter=None):
 
1492
    """Primary interface for merging.
895
1493
 
896
 
        typical use is probably 
 
1494
        typical use is probably
897
1495
        'merge_inner(branch, branch.get_revision_tree(other_revision),
898
1496
                     branch.get_revision_tree(base_revision))'
899
1497
        """
900
1498
    if this_tree is None:
901
 
        warnings.warn("bzrlib.merge.merge_inner requires a this_tree parameter as of "
902
 
             "bzrlib version 0.8.",
903
 
             DeprecationWarning,
904
 
             stacklevel=2)
905
 
        this_tree = this_branch.bzrdir.open_workingtree()
906
 
    merger = Merger(this_branch, other_tree, base_tree, this_tree=this_tree, 
907
 
                    pb=pb)
 
1499
        raise BzrError("bzrlib.merge.merge_inner requires a this_tree "
 
1500
            "parameter as of bzrlib version 0.8.")
 
1501
    merger = Merger(this_branch, other_tree, base_tree, this_tree=this_tree,
 
1502
                    pb=pb, change_reporter=change_reporter)
908
1503
    merger.backup_files = backup_files
909
1504
    merger.merge_type = merge_type
910
1505
    merger.interesting_ids = interesting_ids
911
1506
    merger.ignore_zero = ignore_zero
912
1507
    if interesting_files:
913
 
        assert not interesting_ids, ('Only supply interesting_ids'
914
 
                                     ' or interesting_files')
915
 
        merger._set_interesting_files(interesting_files)
916
 
    merger.show_base = show_base 
 
1508
        if interesting_ids:
 
1509
            raise ValueError('Only supply interesting_ids'
 
1510
                             ' or interesting_files')
 
1511
        merger.interesting_files = interesting_files
 
1512
    merger.show_base = show_base
917
1513
    merger.reprocess = reprocess
918
1514
    merger.other_rev_id = other_rev_id
919
1515
    merger.other_basis = other_rev_id
 
1516
    get_revision_id = getattr(base_tree, 'get_revision_id', None)
 
1517
    if get_revision_id is None:
 
1518
        get_revision_id = base_tree.last_revision
 
1519
    merger.set_base_revision(get_revision_id(), this_branch)
920
1520
    return merger.do_merge()
921
1521
 
922
 
 
923
 
merge_types = {     "merge3": (Merge3Merger, "Native diff3-style merge"), 
924
 
                     "diff3": (Diff3Merger,  "Merge using external diff3"),
925
 
                     'weave': (WeaveMerger, "Weave-based merge")
926
 
              }
927
 
 
928
 
 
929
 
def merge_type_help():
930
 
    templ = '%s%%7s: %%s' % (' '*12)
931
 
    lines = [templ % (f[0], f[1][1]) for f in merge_types.iteritems()]
932
 
    return '\n'.join(lines)
 
1522
def get_merge_type_registry():
 
1523
    """Merge type registry is in bzrlib.option to avoid circular imports.
 
1524
 
 
1525
    This method provides a sanctioned way to retrieve it.
 
1526
    """
 
1527
    from bzrlib import option
 
1528
    return option._merge_type_registry
 
1529
 
 
1530
 
 
1531
def _plan_annotate_merge(annotated_a, annotated_b, ancestors_a, ancestors_b):
 
1532
    def status_a(revision, text):
 
1533
        if revision in ancestors_b:
 
1534
            return 'killed-b', text
 
1535
        else:
 
1536
            return 'new-a', text
 
1537
 
 
1538
    def status_b(revision, text):
 
1539
        if revision in ancestors_a:
 
1540
            return 'killed-a', text
 
1541
        else:
 
1542
            return 'new-b', text
 
1543
 
 
1544
    plain_a = [t for (a, t) in annotated_a]
 
1545
    plain_b = [t for (a, t) in annotated_b]
 
1546
    matcher = patiencediff.PatienceSequenceMatcher(None, plain_a, plain_b)
 
1547
    blocks = matcher.get_matching_blocks()
 
1548
    a_cur = 0
 
1549
    b_cur = 0
 
1550
    for ai, bi, l in blocks:
 
1551
        # process all mismatched sections
 
1552
        # (last mismatched section is handled because blocks always
 
1553
        # includes a 0-length last block)
 
1554
        for revision, text in annotated_a[a_cur:ai]:
 
1555
            yield status_a(revision, text)
 
1556
        for revision, text in annotated_b[b_cur:bi]:
 
1557
            yield status_b(revision, text)
 
1558
        # and now the matched section
 
1559
        a_cur = ai + l
 
1560
        b_cur = bi + l
 
1561
        for text_a in plain_a[ai:a_cur]:
 
1562
            yield "unchanged", text_a
 
1563
 
 
1564
 
 
1565
class _PlanMergeBase(object):
 
1566
 
 
1567
    def __init__(self, a_rev, b_rev, vf, key_prefix):
 
1568
        """Contructor.
 
1569
 
 
1570
        :param a_rev: Revision-id of one revision to merge
 
1571
        :param b_rev: Revision-id of the other revision to merge
 
1572
        :param vf: A VersionedFiles containing both revisions
 
1573
        :param key_prefix: A prefix for accessing keys in vf, typically
 
1574
            (file_id,).
 
1575
        """
 
1576
        self.a_rev = a_rev
 
1577
        self.b_rev = b_rev
 
1578
        self.vf = vf
 
1579
        self._last_lines = None
 
1580
        self._last_lines_revision_id = None
 
1581
        self._cached_matching_blocks = {}
 
1582
        self._key_prefix = key_prefix
 
1583
        self._precache_tip_lines()
 
1584
 
 
1585
    def _precache_tip_lines(self):
 
1586
        lines = self.get_lines([self.a_rev, self.b_rev])
 
1587
        self.lines_a = lines[self.a_rev]
 
1588
        self.lines_b = lines[self.b_rev]
 
1589
 
 
1590
    def get_lines(self, revisions):
 
1591
        """Get lines for revisions from the backing VersionedFiles.
 
1592
 
 
1593
        :raises RevisionNotPresent: on absent texts.
 
1594
        """
 
1595
        keys = [(self._key_prefix + (rev,)) for rev in revisions]
 
1596
        result = {}
 
1597
        for record in self.vf.get_record_stream(keys, 'unordered', True):
 
1598
            if record.storage_kind == 'absent':
 
1599
                raise errors.RevisionNotPresent(record.key, self.vf)
 
1600
            result[record.key[-1]] = osutils.chunks_to_lines(
 
1601
                record.get_bytes_as('chunked'))
 
1602
        return result
 
1603
 
 
1604
    def plan_merge(self):
 
1605
        """Generate a 'plan' for merging the two revisions.
 
1606
 
 
1607
        This involves comparing their texts and determining the cause of
 
1608
        differences.  If text A has a line and text B does not, then either the
 
1609
        line was added to text A, or it was deleted from B.  Once the causes
 
1610
        are combined, they are written out in the format described in
 
1611
        VersionedFile.plan_merge
 
1612
        """
 
1613
        blocks = self._get_matching_blocks(self.a_rev, self.b_rev)
 
1614
        unique_a, unique_b = self._unique_lines(blocks)
 
1615
        new_a, killed_b = self._determine_status(self.a_rev, unique_a)
 
1616
        new_b, killed_a = self._determine_status(self.b_rev, unique_b)
 
1617
        return self._iter_plan(blocks, new_a, killed_b, new_b, killed_a)
 
1618
 
 
1619
    def _iter_plan(self, blocks, new_a, killed_b, new_b, killed_a):
 
1620
        last_i = 0
 
1621
        last_j = 0
 
1622
        for i, j, n in blocks:
 
1623
            for a_index in range(last_i, i):
 
1624
                if a_index in new_a:
 
1625
                    if a_index in killed_b:
 
1626
                        yield 'conflicted-a', self.lines_a[a_index]
 
1627
                    else:
 
1628
                        yield 'new-a', self.lines_a[a_index]
 
1629
                else:
 
1630
                    yield 'killed-b', self.lines_a[a_index]
 
1631
            for b_index in range(last_j, j):
 
1632
                if b_index in new_b:
 
1633
                    if b_index in killed_a:
 
1634
                        yield 'conflicted-b', self.lines_b[b_index]
 
1635
                    else:
 
1636
                        yield 'new-b', self.lines_b[b_index]
 
1637
                else:
 
1638
                    yield 'killed-a', self.lines_b[b_index]
 
1639
            # handle common lines
 
1640
            for a_index in range(i, i+n):
 
1641
                yield 'unchanged', self.lines_a[a_index]
 
1642
            last_i = i+n
 
1643
            last_j = j+n
 
1644
 
 
1645
    def _get_matching_blocks(self, left_revision, right_revision):
 
1646
        """Return a description of which sections of two revisions match.
 
1647
 
 
1648
        See SequenceMatcher.get_matching_blocks
 
1649
        """
 
1650
        cached = self._cached_matching_blocks.get((left_revision,
 
1651
                                                   right_revision))
 
1652
        if cached is not None:
 
1653
            return cached
 
1654
        if self._last_lines_revision_id == left_revision:
 
1655
            left_lines = self._last_lines
 
1656
            right_lines = self.get_lines([right_revision])[right_revision]
 
1657
        else:
 
1658
            lines = self.get_lines([left_revision, right_revision])
 
1659
            left_lines = lines[left_revision]
 
1660
            right_lines = lines[right_revision]
 
1661
        self._last_lines = right_lines
 
1662
        self._last_lines_revision_id = right_revision
 
1663
        matcher = patiencediff.PatienceSequenceMatcher(None, left_lines,
 
1664
                                                       right_lines)
 
1665
        return matcher.get_matching_blocks()
 
1666
 
 
1667
    def _unique_lines(self, matching_blocks):
 
1668
        """Analyse matching_blocks to determine which lines are unique
 
1669
 
 
1670
        :return: a tuple of (unique_left, unique_right), where the values are
 
1671
            sets of line numbers of unique lines.
 
1672
        """
 
1673
        last_i = 0
 
1674
        last_j = 0
 
1675
        unique_left = []
 
1676
        unique_right = []
 
1677
        for i, j, n in matching_blocks:
 
1678
            unique_left.extend(range(last_i, i))
 
1679
            unique_right.extend(range(last_j, j))
 
1680
            last_i = i + n
 
1681
            last_j = j + n
 
1682
        return unique_left, unique_right
 
1683
 
 
1684
    @staticmethod
 
1685
    def _subtract_plans(old_plan, new_plan):
 
1686
        """Remove changes from new_plan that came from old_plan.
 
1687
 
 
1688
        It is assumed that the difference between the old_plan and new_plan
 
1689
        is their choice of 'b' text.
 
1690
 
 
1691
        All lines from new_plan that differ from old_plan are emitted
 
1692
        verbatim.  All lines from new_plan that match old_plan but are
 
1693
        not about the 'b' revision are emitted verbatim.
 
1694
 
 
1695
        Lines that match and are about the 'b' revision are the lines we
 
1696
        don't want, so we convert 'killed-b' -> 'unchanged', and 'new-b'
 
1697
        is skipped entirely.
 
1698
        """
 
1699
        matcher = patiencediff.PatienceSequenceMatcher(None, old_plan,
 
1700
                                                       new_plan)
 
1701
        last_j = 0
 
1702
        for i, j, n in matcher.get_matching_blocks():
 
1703
            for jj in range(last_j, j):
 
1704
                yield new_plan[jj]
 
1705
            for jj in range(j, j+n):
 
1706
                plan_line = new_plan[jj]
 
1707
                if plan_line[0] == 'new-b':
 
1708
                    pass
 
1709
                elif plan_line[0] == 'killed-b':
 
1710
                    yield 'unchanged', plan_line[1]
 
1711
                else:
 
1712
                    yield plan_line
 
1713
            last_j = j + n
 
1714
 
 
1715
 
 
1716
class _PlanMerge(_PlanMergeBase):
 
1717
    """Plan an annotate merge using on-the-fly annotation"""
 
1718
 
 
1719
    def __init__(self, a_rev, b_rev, vf, key_prefix):
 
1720
        super(_PlanMerge, self).__init__(a_rev, b_rev, vf, key_prefix)
 
1721
        self.a_key = self._key_prefix + (self.a_rev,)
 
1722
        self.b_key = self._key_prefix + (self.b_rev,)
 
1723
        self.graph = Graph(self.vf)
 
1724
        heads = self.graph.heads((self.a_key, self.b_key))
 
1725
        if len(heads) == 1:
 
1726
            # one side dominates, so we can just return its values, yay for
 
1727
            # per-file graphs
 
1728
            # Ideally we would know that before we get this far
 
1729
            self._head_key = heads.pop()
 
1730
            if self._head_key == self.a_key:
 
1731
                other = b_rev
 
1732
            else:
 
1733
                other = a_rev
 
1734
            mutter('found dominating revision for %s\n%s > %s', self.vf,
 
1735
                   self._head_key[-1], other)
 
1736
            self._weave = None
 
1737
        else:
 
1738
            self._head_key = None
 
1739
            self._build_weave()
 
1740
 
 
1741
    def _precache_tip_lines(self):
 
1742
        # Turn this into a no-op, because we will do this later
 
1743
        pass
 
1744
 
 
1745
    def _find_recursive_lcas(self):
 
1746
        """Find all the ancestors back to a unique lca"""
 
1747
        cur_ancestors = (self.a_key, self.b_key)
 
1748
        # graph.find_lca(uncommon, keys) now returns plain NULL_REVISION,
 
1749
        # rather than a key tuple. We will just map that directly to no common
 
1750
        # ancestors.
 
1751
        parent_map = {}
 
1752
        while True:
 
1753
            next_lcas = self.graph.find_lca(*cur_ancestors)
 
1754
            # Map a plain NULL_REVISION to a simple no-ancestors
 
1755
            if next_lcas == set([NULL_REVISION]):
 
1756
                next_lcas = ()
 
1757
            # Order the lca's based on when they were merged into the tip
 
1758
            # While the actual merge portion of weave merge uses a set() of
 
1759
            # active revisions, the order of insertion *does* effect the
 
1760
            # implicit ordering of the texts.
 
1761
            for rev_key in cur_ancestors:
 
1762
                ordered_parents = tuple(self.graph.find_merge_order(rev_key,
 
1763
                                                                    next_lcas))
 
1764
                parent_map[rev_key] = ordered_parents
 
1765
            if len(next_lcas) == 0:
 
1766
                break
 
1767
            elif len(next_lcas) == 1:
 
1768
                parent_map[list(next_lcas)[0]] = ()
 
1769
                break
 
1770
            elif len(next_lcas) > 2:
 
1771
                # More than 2 lca's, fall back to grabbing all nodes between
 
1772
                # this and the unique lca.
 
1773
                mutter('More than 2 LCAs, falling back to all nodes for:'
 
1774
                       ' %s, %s\n=> %s', self.a_key, self.b_key, cur_ancestors)
 
1775
                cur_lcas = next_lcas
 
1776
                while len(cur_lcas) > 1:
 
1777
                    cur_lcas = self.graph.find_lca(*cur_lcas)
 
1778
                if len(cur_lcas) == 0:
 
1779
                    # No common base to find, use the full ancestry
 
1780
                    unique_lca = None
 
1781
                else:
 
1782
                    unique_lca = list(cur_lcas)[0]
 
1783
                    if unique_lca == NULL_REVISION:
 
1784
                        # find_lca will return a plain 'NULL_REVISION' rather
 
1785
                        # than a key tuple when there is no common ancestor, we
 
1786
                        # prefer to just use None, because it doesn't confuse
 
1787
                        # _get_interesting_texts()
 
1788
                        unique_lca = None
 
1789
                parent_map.update(self._find_unique_parents(next_lcas,
 
1790
                                                            unique_lca))
 
1791
                break
 
1792
            cur_ancestors = next_lcas
 
1793
        return parent_map
 
1794
 
 
1795
    def _find_unique_parents(self, tip_keys, base_key):
 
1796
        """Find ancestors of tip that aren't ancestors of base.
 
1797
 
 
1798
        :param tip_keys: Nodes that are interesting
 
1799
        :param base_key: Cull all ancestors of this node
 
1800
        :return: The parent map for all revisions between tip_keys and
 
1801
            base_key. base_key will be included. References to nodes outside of
 
1802
            the ancestor set will also be removed.
 
1803
        """
 
1804
        # TODO: this would be simpler if find_unique_ancestors took a list
 
1805
        #       instead of a single tip, internally it supports it, but it
 
1806
        #       isn't a "backwards compatible" api change.
 
1807
        if base_key is None:
 
1808
            parent_map = dict(self.graph.iter_ancestry(tip_keys))
 
1809
            # We remove NULL_REVISION because it isn't a proper tuple key, and
 
1810
            # thus confuses things like _get_interesting_texts, and our logic
 
1811
            # to add the texts into the memory weave.
 
1812
            if NULL_REVISION in parent_map:
 
1813
                parent_map.pop(NULL_REVISION)
 
1814
        else:
 
1815
            interesting = set()
 
1816
            for tip in tip_keys:
 
1817
                interesting.update(
 
1818
                    self.graph.find_unique_ancestors(tip, [base_key]))
 
1819
            parent_map = self.graph.get_parent_map(interesting)
 
1820
            parent_map[base_key] = ()
 
1821
        culled_parent_map, child_map, tails = self._remove_external_references(
 
1822
            parent_map)
 
1823
        # Remove all the tails but base_key
 
1824
        if base_key is not None:
 
1825
            tails.remove(base_key)
 
1826
            self._prune_tails(culled_parent_map, child_map, tails)
 
1827
        # Now remove all the uninteresting 'linear' regions
 
1828
        simple_map = _mod_graph.collapse_linear_regions(culled_parent_map)
 
1829
        return simple_map
 
1830
 
 
1831
    @staticmethod
 
1832
    def _remove_external_references(parent_map):
 
1833
        """Remove references that go outside of the parent map.
 
1834
 
 
1835
        :param parent_map: Something returned from Graph.get_parent_map(keys)
 
1836
        :return: (filtered_parent_map, child_map, tails)
 
1837
            filtered_parent_map is parent_map without external references
 
1838
            child_map is the {parent_key: [child_keys]} mapping
 
1839
            tails is a list of nodes that do not have any parents in the map
 
1840
        """
 
1841
        # TODO: The basic effect of this function seems more generic than
 
1842
        #       _PlanMerge. But the specific details of building a child_map,
 
1843
        #       and computing tails seems very specific to _PlanMerge.
 
1844
        #       Still, should this be in Graph land?
 
1845
        filtered_parent_map = {}
 
1846
        child_map = {}
 
1847
        tails = []
 
1848
        for key, parent_keys in parent_map.iteritems():
 
1849
            culled_parent_keys = [p for p in parent_keys if p in parent_map]
 
1850
            if not culled_parent_keys:
 
1851
                tails.append(key)
 
1852
            for parent_key in culled_parent_keys:
 
1853
                child_map.setdefault(parent_key, []).append(key)
 
1854
            # TODO: Do we want to do this, it adds overhead for every node,
 
1855
            #       just to say that the node has no children
 
1856
            child_map.setdefault(key, [])
 
1857
            filtered_parent_map[key] = culled_parent_keys
 
1858
        return filtered_parent_map, child_map, tails
 
1859
 
 
1860
    @staticmethod
 
1861
    def _prune_tails(parent_map, child_map, tails_to_remove):
 
1862
        """Remove tails from the parent map.
 
1863
 
 
1864
        This will remove the supplied revisions until no more children have 0
 
1865
        parents.
 
1866
 
 
1867
        :param parent_map: A dict of {child: [parents]}, this dictionary will
 
1868
            be modified in place.
 
1869
        :param tails_to_remove: A list of tips that should be removed,
 
1870
            this list will be consumed
 
1871
        :param child_map: The reverse dict of parent_map ({parent: [children]})
 
1872
            this dict will be modified
 
1873
        :return: None, parent_map will be modified in place.
 
1874
        """
 
1875
        while tails_to_remove:
 
1876
            next = tails_to_remove.pop()
 
1877
            parent_map.pop(next)
 
1878
            children = child_map.pop(next)
 
1879
            for child in children:
 
1880
                child_parents = parent_map[child]
 
1881
                child_parents.remove(next)
 
1882
                if len(child_parents) == 0:
 
1883
                    tails_to_remove.append(child)
 
1884
 
 
1885
    def _get_interesting_texts(self, parent_map):
 
1886
        """Return a dict of texts we are interested in.
 
1887
 
 
1888
        Note that the input is in key tuples, but the output is in plain
 
1889
        revision ids.
 
1890
 
 
1891
        :param parent_map: The output from _find_recursive_lcas
 
1892
        :return: A dict of {'revision_id':lines} as returned by
 
1893
            _PlanMergeBase.get_lines()
 
1894
        """
 
1895
        all_revision_keys = set(parent_map)
 
1896
        all_revision_keys.add(self.a_key)
 
1897
        all_revision_keys.add(self.b_key)
 
1898
 
 
1899
        # Everything else is in 'keys' but get_lines is in 'revision_ids'
 
1900
        all_texts = self.get_lines([k[-1] for k in all_revision_keys])
 
1901
        return all_texts
 
1902
 
 
1903
    def _build_weave(self):
 
1904
        from bzrlib import weave
 
1905
        self._weave = weave.Weave(weave_name='in_memory_weave',
 
1906
                                  allow_reserved=True)
 
1907
        parent_map = self._find_recursive_lcas()
 
1908
 
 
1909
        all_texts = self._get_interesting_texts(parent_map)
 
1910
 
 
1911
        # Note: Unfortunately, the order given by topo_sort will effect the
 
1912
        # ordering resolution in the output. Specifically, if you add A then B,
 
1913
        # then in the output text A lines will show up before B lines. And, of
 
1914
        # course, topo_sort doesn't guarantee any real ordering.
 
1915
        # So we use merge_sort, and add a fake node on the tip.
 
1916
        # This ensures that left-hand parents will always be inserted into the
 
1917
        # weave before right-hand parents.
 
1918
        tip_key = self._key_prefix + (_mod_revision.CURRENT_REVISION,)
 
1919
        parent_map[tip_key] = (self.a_key, self.b_key)
 
1920
 
 
1921
        for seq_num, key, depth, eom in reversed(tsort.merge_sort(parent_map,
 
1922
                                                                  tip_key)):
 
1923
            if key == tip_key:
 
1924
                continue
 
1925
        # for key in tsort.topo_sort(parent_map):
 
1926
            parent_keys = parent_map[key]
 
1927
            revision_id = key[-1]
 
1928
            parent_ids = [k[-1] for k in parent_keys]
 
1929
            self._weave.add_lines(revision_id, parent_ids,
 
1930
                                  all_texts[revision_id])
 
1931
 
 
1932
    def plan_merge(self):
 
1933
        """Generate a 'plan' for merging the two revisions.
 
1934
 
 
1935
        This involves comparing their texts and determining the cause of
 
1936
        differences.  If text A has a line and text B does not, then either the
 
1937
        line was added to text A, or it was deleted from B.  Once the causes
 
1938
        are combined, they are written out in the format described in
 
1939
        VersionedFile.plan_merge
 
1940
        """
 
1941
        if self._head_key is not None: # There was a single head
 
1942
            if self._head_key == self.a_key:
 
1943
                plan = 'new-a'
 
1944
            else:
 
1945
                if self._head_key != self.b_key:
 
1946
                    raise AssertionError('There was an invalid head: %s != %s'
 
1947
                                         % (self.b_key, self._head_key))
 
1948
                plan = 'new-b'
 
1949
            head_rev = self._head_key[-1]
 
1950
            lines = self.get_lines([head_rev])[head_rev]
 
1951
            return ((plan, line) for line in lines)
 
1952
        return self._weave.plan_merge(self.a_rev, self.b_rev)
 
1953
 
 
1954
 
 
1955
class _PlanLCAMerge(_PlanMergeBase):
 
1956
    """
 
1957
    This merge algorithm differs from _PlanMerge in that:
 
1958
    1. comparisons are done against LCAs only
 
1959
    2. cases where a contested line is new versus one LCA but old versus
 
1960
       another are marked as conflicts, by emitting the line as conflicted-a
 
1961
       or conflicted-b.
 
1962
 
 
1963
    This is faster, and hopefully produces more useful output.
 
1964
    """
 
1965
 
 
1966
    def __init__(self, a_rev, b_rev, vf, key_prefix, graph):
 
1967
        _PlanMergeBase.__init__(self, a_rev, b_rev, vf, key_prefix)
 
1968
        lcas = graph.find_lca(key_prefix + (a_rev,), key_prefix + (b_rev,))
 
1969
        self.lcas = set()
 
1970
        for lca in lcas:
 
1971
            if lca == NULL_REVISION:
 
1972
                self.lcas.add(lca)
 
1973
            else:
 
1974
                self.lcas.add(lca[-1])
 
1975
        for lca in self.lcas:
 
1976
            if _mod_revision.is_null(lca):
 
1977
                lca_lines = []
 
1978
            else:
 
1979
                lca_lines = self.get_lines([lca])[lca]
 
1980
            matcher = patiencediff.PatienceSequenceMatcher(None, self.lines_a,
 
1981
                                                           lca_lines)
 
1982
            blocks = list(matcher.get_matching_blocks())
 
1983
            self._cached_matching_blocks[(a_rev, lca)] = blocks
 
1984
            matcher = patiencediff.PatienceSequenceMatcher(None, self.lines_b,
 
1985
                                                           lca_lines)
 
1986
            blocks = list(matcher.get_matching_blocks())
 
1987
            self._cached_matching_blocks[(b_rev, lca)] = blocks
 
1988
 
 
1989
    def _determine_status(self, revision_id, unique_line_numbers):
 
1990
        """Determines the status unique lines versus all lcas.
 
1991
 
 
1992
        Basically, determines why the line is unique to this revision.
 
1993
 
 
1994
        A line may be determined new, killed, or both.
 
1995
 
 
1996
        If a line is determined new, that means it was not present in at least
 
1997
        one LCA, and is not present in the other merge revision.
 
1998
 
 
1999
        If a line is determined killed, that means the line was present in
 
2000
        at least one LCA.
 
2001
 
 
2002
        If a line is killed and new, this indicates that the two merge
 
2003
        revisions contain differing conflict resolutions.
 
2004
        :param revision_id: The id of the revision in which the lines are
 
2005
            unique
 
2006
        :param unique_line_numbers: The line numbers of unique lines.
 
2007
        :return a tuple of (new_this, killed_other):
 
2008
        """
 
2009
        new = set()
 
2010
        killed = set()
 
2011
        unique_line_numbers = set(unique_line_numbers)
 
2012
        for lca in self.lcas:
 
2013
            blocks = self._get_matching_blocks(revision_id, lca)
 
2014
            unique_vs_lca, _ignored = self._unique_lines(blocks)
 
2015
            new.update(unique_line_numbers.intersection(unique_vs_lca))
 
2016
            killed.update(unique_line_numbers.difference(unique_vs_lca))
 
2017
        return new, killed