~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/weave.py

  • Committer: John Arbash Meinel
  • Date: 2005-11-30 15:43:57 UTC
  • mto: (1185.50.1 jam-integration)
  • mto: This revision was merged to the branch mainline in revision 1518.
  • Revision ID: john@arbash-meinel.com-20051130154357-614206b3a7b83cd0
Refactored bzrlib/ui.py into a module with the possibility for multiple ui forms.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
1
#! /usr/bin/python
2
2
 
3
3
# Copyright (C) 2005 Canonical Ltd
4
 
#
 
4
 
5
5
# This program is free software; you can redistribute it and/or modify
6
6
# it under the terms of the GNU General Public License as published by
7
7
# the Free Software Foundation; either version 2 of the License, or
8
8
# (at your option) any later version.
9
 
#
 
9
 
10
10
# This program is distributed in the hope that it will be useful,
11
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
13
# GNU General Public License for more details.
14
 
#
 
14
 
15
15
# You should have received a copy of the GNU General Public License
16
16
# along with this program; if not, write to the Free Software
17
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
27
27
# property.
28
28
 
29
29
# TODO: Nothing here so far assumes the lines are really \n newlines,
30
 
# rather than being split up in some other way.  We could accommodate
 
30
# rather than being split up in some other way.  We could accomodate
31
31
# binaries, perhaps by naively splitting on \n or perhaps using
32
32
# something like a rolling checksum.
33
33
 
66
66
# be done fairly efficiently because the sequence numbers constrain
67
67
# the possible relationships.
68
68
 
69
 
# FIXME: the conflict markers should be *7* characters
70
 
 
71
 
from copy import copy
72
 
from cStringIO import StringIO
73
 
import os
74
 
import time
75
 
import warnings
76
 
 
77
 
from bzrlib.lazy_import import lazy_import
78
 
lazy_import(globals(), """
79
 
from bzrlib import tsort
80
 
""")
81
 
from bzrlib import (
82
 
    errors,
83
 
    osutils,
84
 
    progress,
85
 
    )
86
 
from bzrlib.errors import (WeaveError, WeaveFormatError, WeaveParentMismatch,
87
 
        RevisionAlreadyPresent,
88
 
        RevisionNotPresent,
89
 
        UnavailableRepresentation,
90
 
        WeaveRevisionAlreadyPresent,
91
 
        WeaveRevisionNotPresent,
92
 
        )
93
 
from bzrlib.osutils import dirname, sha, sha_strings, split_lines
94
 
import bzrlib.patiencediff
95
 
from bzrlib.revision import NULL_REVISION
96
 
from bzrlib.symbol_versioning import *
 
69
 
 
70
import sha
 
71
from difflib import SequenceMatcher
 
72
 
97
73
from bzrlib.trace import mutter
98
 
from bzrlib.versionedfile import (
99
 
    AbsentContentFactory,
100
 
    adapter_registry,
101
 
    ContentFactory,
102
 
    VersionedFile,
103
 
    )
104
 
from bzrlib.weavefile import _read_weave_v5, write_weave_v5
105
 
 
106
 
 
107
 
class WeaveContentFactory(ContentFactory):
108
 
    """Content factory for streaming from weaves.
109
 
 
110
 
    :seealso ContentFactory:
111
 
    """
112
 
 
113
 
    def __init__(self, version, weave):
114
 
        """Create a WeaveContentFactory for version from weave."""
115
 
        ContentFactory.__init__(self)
116
 
        self.sha1 = weave.get_sha1s([version])[version]
117
 
        self.key = (version,)
118
 
        parents = weave.get_parent_map([version])[version]
119
 
        self.parents = tuple((parent,) for parent in parents)
120
 
        self.storage_kind = 'fulltext'
121
 
        self._weave = weave
122
 
 
123
 
    def get_bytes_as(self, storage_kind):
124
 
        if storage_kind == 'fulltext':
125
 
            return self._weave.get_text(self.key[-1])
126
 
        elif storage_kind == 'chunked':
127
 
            return self._weave.get_lines(self.key[-1])
128
 
        else:
129
 
            raise UnavailableRepresentation(self.key, storage_kind, 'fulltext')
130
 
 
131
 
 
132
 
class Weave(VersionedFile):
 
74
from bzrlib.errors import WeaveError, WeaveFormatError, WeaveParentMismatch, \
 
75
        WeaveRevisionNotPresent, WeaveRevisionAlreadyPresent
 
76
from bzrlib.tsort import topo_sort
 
77
 
 
78
 
 
79
class Weave(object):
133
80
    """weave - versioned text file storage.
134
81
    
135
82
    A Weave manages versions of line-based text files, keeping track
219
166
    """
220
167
 
221
168
    __slots__ = ['_weave', '_parents', '_sha1s', '_names', '_name_map',
222
 
                 '_weave_name', '_matcher', '_allow_reserved']
223
 
 
224
 
    def __init__(self, weave_name=None, access_mode='w', matcher=None,
225
 
                 get_scope=None, allow_reserved=False):
226
 
        """Create a weave.
227
 
 
228
 
        :param get_scope: A callable that returns an opaque object to be used
229
 
            for detecting when this weave goes out of scope (should stop
230
 
            answering requests or allowing mutation).
231
 
        """
232
 
        super(Weave, self).__init__()
 
169
                 '_weave_name']
 
170
    
 
171
    def __init__(self, weave_name=None):
233
172
        self._weave = []
234
173
        self._parents = []
235
174
        self._sha1s = []
236
175
        self._names = []
237
176
        self._name_map = {}
238
177
        self._weave_name = weave_name
239
 
        if matcher is None:
240
 
            self._matcher = bzrlib.patiencediff.PatienceSequenceMatcher
241
 
        else:
242
 
            self._matcher = matcher
243
 
        if get_scope is None:
244
 
            get_scope = lambda:None
245
 
        self._get_scope = get_scope
246
 
        self._scope = get_scope()
247
 
        self._access_mode = access_mode
248
 
        self._allow_reserved = allow_reserved
249
178
 
250
179
    def __repr__(self):
251
180
        return "Weave(%r)" % self._weave_name
252
181
 
253
 
    def _check_write_ok(self):
254
 
        """Is the versioned file marked as 'finished' ? Raise if it is."""
255
 
        if self._get_scope() != self._scope:
256
 
            raise errors.OutSideTransaction()
257
 
        if self._access_mode != 'w':
258
 
            raise errors.ReadOnlyObjectDirtiedError(self)
259
182
 
260
183
    def copy(self):
261
184
        """Return a deep copy of self.
276
199
        return self._parents == other._parents \
277
200
               and self._weave == other._weave \
278
201
               and self._sha1s == other._sha1s 
 
202
 
279
203
    
280
204
    def __ne__(self, other):
281
205
        return not self.__eq__(other)
282
206
 
283
 
    def _idx_to_name(self, version):
284
 
        return self._names[version]
285
 
 
286
 
    def _lookup(self, name):
 
207
    def __contains__(self, name):
 
208
        return self._name_map.has_key(name)
 
209
 
 
210
    def maybe_lookup(self, name_or_index):
 
211
        """Convert possible symbolic name to index, or pass through indexes."""
 
212
        if isinstance(name_or_index, (int, long)):
 
213
            return name_or_index
 
214
        else:
 
215
            return self.lookup(name_or_index)
 
216
 
 
217
        
 
218
    def lookup(self, name):
287
219
        """Convert symbolic version name to index."""
288
 
        if not self._allow_reserved:
289
 
            self.check_not_reserved_id(name)
290
220
        try:
291
221
            return self._name_map[name]
292
222
        except KeyError:
293
 
            raise RevisionNotPresent(name, self._weave_name)
 
223
            raise WeaveRevisionNotPresent(name, self)
294
224
 
295
 
    def versions(self):
296
 
        """See VersionedFile.versions."""
 
225
    def names(self):
297
226
        return self._names[:]
298
227
 
299
 
    def has_version(self, version_id):
300
 
        """See VersionedFile.has_version."""
301
 
        return (version_id in self._name_map)
302
 
 
303
 
    __contains__ = has_version
304
 
 
305
 
    def get_record_stream(self, versions, ordering, include_delta_closure):
306
 
        """Get a stream of records for versions.
307
 
 
308
 
        :param versions: The versions to include. Each version is a tuple
309
 
            (version,).
310
 
        :param ordering: Either 'unordered' or 'topological'. A topologically
311
 
            sorted stream has compression parents strictly before their
312
 
            children.
313
 
        :param include_delta_closure: If True then the closure across any
314
 
            compression parents will be included (in the opaque data).
315
 
        :return: An iterator of ContentFactory objects, each of which is only
316
 
            valid until the iterator is advanced.
317
 
        """
318
 
        versions = [version[-1] for version in versions]
319
 
        if ordering == 'topological':
320
 
            parents = self.get_parent_map(versions)
321
 
            new_versions = tsort.topo_sort(parents)
322
 
            new_versions.extend(set(versions).difference(set(parents)))
323
 
            versions = new_versions
324
 
        for version in versions:
325
 
            if version in self:
326
 
                yield WeaveContentFactory(version, self)
327
 
            else:
328
 
                yield AbsentContentFactory((version,))
329
 
 
330
 
    def get_parent_map(self, version_ids):
331
 
        """See VersionedFile.get_parent_map."""
332
 
        result = {}
333
 
        for version_id in version_ids:
334
 
            if version_id == NULL_REVISION:
335
 
                parents = ()
336
 
            else:
337
 
                try:
338
 
                    parents = tuple(
339
 
                        map(self._idx_to_name,
340
 
                            self._parents[self._lookup(version_id)]))
341
 
                except RevisionNotPresent:
342
 
                    continue
343
 
            result[version_id] = parents
344
 
        return result
345
 
 
346
 
    def get_parents_with_ghosts(self, version_id):
347
 
        raise NotImplementedError(self.get_parents_with_ghosts)
348
 
 
349
 
    def insert_record_stream(self, stream):
350
 
        """Insert a record stream into this versioned file.
351
 
 
352
 
        :param stream: A stream of records to insert. 
353
 
        :return: None
354
 
        :seealso VersionedFile.get_record_stream:
355
 
        """
356
 
        adapters = {}
357
 
        for record in stream:
358
 
            # Raise an error when a record is missing.
359
 
            if record.storage_kind == 'absent':
360
 
                raise RevisionNotPresent([record.key[0]], self)
361
 
            # adapt to non-tuple interface
362
 
            parents = [parent[0] for parent in record.parents]
363
 
            if (record.storage_kind == 'fulltext'
364
 
                or record.storage_kind == 'chunked'):
365
 
                self.add_lines(record.key[0], parents,
366
 
                    osutils.chunks_to_lines(record.get_bytes_as('chunked')))
367
 
            else:
368
 
                adapter_key = record.storage_kind, 'fulltext'
369
 
                try:
370
 
                    adapter = adapters[adapter_key]
371
 
                except KeyError:
372
 
                    adapter_factory = adapter_registry.get(adapter_key)
373
 
                    adapter = adapter_factory(self)
374
 
                    adapters[adapter_key] = adapter
375
 
                lines = split_lines(adapter.get_bytes(
376
 
                    record, record.get_bytes_as(record.storage_kind)))
377
 
                try:
378
 
                    self.add_lines(record.key[0], parents, lines)
379
 
                except RevisionAlreadyPresent:
380
 
                    pass
 
228
    def iter_names(self):
 
229
        """Yield a list of all names in this weave."""
 
230
        return iter(self._names)
 
231
 
 
232
    def idx_to_name(self, version):
 
233
        return self._names[version]
381
234
 
382
235
    def _check_repeated_add(self, name, parents, text, sha1):
383
236
        """Check that a duplicated add is OK.
384
237
 
385
238
        If it is, return the (old) index; otherwise raise an exception.
386
239
        """
387
 
        idx = self._lookup(name)
 
240
        idx = self.lookup(name)
388
241
        if sorted(self._parents[idx]) != sorted(parents) \
389
242
            or sha1 != self._sha1s[idx]:
390
 
            raise RevisionAlreadyPresent(name, self._weave_name)
 
243
            raise WeaveRevisionAlreadyPresent(name, self)
391
244
        return idx
392
 
 
393
 
    def _add_lines(self, version_id, parents, lines, parent_texts,
394
 
       left_matching_blocks, nostore_sha, random_id, check_content):
395
 
        """See VersionedFile.add_lines."""
396
 
        idx = self._add(version_id, lines, map(self._lookup, parents),
397
 
            nostore_sha=nostore_sha)
398
 
        return sha_strings(lines), sum(map(len, lines)), idx
399
 
 
400
 
    def _add(self, version_id, lines, parents, sha1=None, nostore_sha=None):
 
245
        
 
246
    def add(self, name, parents, text, sha1=None):
401
247
        """Add a single text on top of the weave.
402
248
  
403
249
        Returns the index number of the newly added version.
404
250
 
405
 
        version_id
 
251
        name
406
252
            Symbolic name for this version.
407
253
            (Typically the revision-id of the revision that added it.)
408
254
 
409
255
        parents
410
256
            List or set of direct parent version numbers.
411
257
            
412
 
        lines
 
258
        text
413
259
            Sequence of lines to be added in the new version.
414
260
 
415
 
        :param nostore_sha: See VersionedFile.add_lines.
 
261
        sha -- SHA-1 of the file, if known.  This is trusted to be
 
262
            correct if supplied.
416
263
        """
417
 
        self._check_lines_not_unicode(lines)
418
 
        self._check_lines_are_lines(lines)
419
 
        if not sha1:
420
 
            sha1 = sha_strings(lines)
421
 
        if sha1 == nostore_sha:
422
 
            raise errors.ExistingContent
423
 
        if version_id in self._name_map:
424
 
            return self._check_repeated_add(version_id, parents, lines, sha1)
425
 
 
 
264
        from bzrlib.osutils import sha_strings
 
265
 
 
266
        assert isinstance(name, basestring)
 
267
        if sha1 is None:
 
268
            sha1 = sha_strings(text)
 
269
        if name in self._name_map:
 
270
            return self._check_repeated_add(name, parents, text, sha1)
 
271
 
 
272
        parents = map(self.maybe_lookup, parents)
426
273
        self._check_versions(parents)
427
 
        ## self._check_lines(lines)
 
274
        ## self._check_lines(text)
428
275
        new_version = len(self._parents)
429
276
 
 
277
 
430
278
        # if we abort after here the (in-memory) weave will be corrupt because only
431
279
        # some fields are updated
432
 
        # XXX: FIXME implement a succeed-or-fail of the rest of this routine.
433
 
        #      - Robert Collins 20060226
434
280
        self._parents.append(parents[:])
435
281
        self._sha1s.append(sha1)
436
 
        self._names.append(version_id)
437
 
        self._name_map[version_id] = new_version
 
282
        self._names.append(name)
 
283
        self._name_map[name] = new_version
438
284
 
439
285
            
440
286
        if not parents:
442
288
            # this more quickly by just appending unconditionally.
443
289
            # even more specially, if we're adding an empty text we
444
290
            # need do nothing at all.
445
 
            if lines:
 
291
            if text:
446
292
                self._weave.append(('{', new_version))
447
 
                self._weave.extend(lines)
 
293
                self._weave.extend(text)
448
294
                self._weave.append(('}', None))
 
295
        
449
296
            return new_version
450
297
 
451
298
        if len(parents) == 1:
455
302
                return new_version
456
303
            
457
304
 
458
 
        ancestors = self._inclusions(parents)
 
305
        ancestors = self.inclusions(parents)
459
306
 
460
307
        l = self._weave
461
308
 
468
315
 
469
316
        # another small special case: a merge, producing the same text
470
317
        # as auto-merge
471
 
        if lines == basis_lines:
472
 
            return new_version
 
318
        if text == basis_lines:
 
319
            return new_version            
473
320
 
474
 
        # add a sentinel, because we can also match against the final line
 
321
        # add a sentinal, because we can also match against the final line
475
322
        basis_lineno.append(len(self._weave))
476
323
 
477
324
        # XXX: which line of the weave should we really consider
481
328
        #print 'basis_lines:', basis_lines
482
329
        #print 'new_lines:  ', lines
483
330
 
484
 
        s = self._matcher(None, basis_lines, lines)
 
331
        s = SequenceMatcher(None, basis_lines, text)
485
332
 
486
333
        # offset gives the number of lines that have been inserted
487
334
        # into the weave up to the current point; if the original edit instruction
494
341
            #print 'raw match', tag, i1, i2, j1, j2
495
342
            if tag == 'equal':
496
343
                continue
 
344
 
497
345
            i1 = basis_lineno[i1]
498
346
            i2 = basis_lineno[i2]
 
347
 
 
348
            assert 0 <= j1 <= j2 <= len(text)
 
349
 
 
350
            #print tag, i1, i2, j1, j2
 
351
 
499
352
            # the deletion and insertion are handled separately.
500
353
            # first delete the region.
501
354
            if i1 != i2:
509
362
                # we don't destroy ourselves
510
363
                i = i2 + offset
511
364
                self._weave[i:i] = ([('{', new_version)] 
512
 
                                    + lines[j1:j2] 
 
365
                                    + text[j1:j2] 
513
366
                                    + [('}', None)])
514
367
                offset += 2 + (j2 - j1)
 
368
 
515
369
        return new_version
516
370
 
517
 
    def _inclusions(self, versions):
 
371
    def add_identical(self, old_rev_id, new_rev_id, parents):
 
372
        """Add an identical text to old_rev_id as new_rev_id."""
 
373
        old_lines = self.get(self.lookup(old_rev_id))
 
374
        self.add(new_rev_id, parents, old_lines)
 
375
 
 
376
    def inclusions(self, versions):
518
377
        """Return set of all ancestors of given version(s)."""
519
 
        if not len(versions):
520
 
            return []
521
378
        i = set(versions)
522
379
        for v in xrange(max(versions), 0, -1):
523
380
            if v in i:
527
384
        ## except IndexError:
528
385
        ##     raise ValueError("version %d not present in weave" % v)
529
386
 
530
 
    def get_ancestry(self, version_ids, topo_sorted=True):
531
 
        """See VersionedFile.get_ancestry."""
532
 
        if isinstance(version_ids, basestring):
533
 
            version_ids = [version_ids]
534
 
        i = self._inclusions([self._lookup(v) for v in version_ids])
535
 
        return [self._idx_to_name(v) for v in i]
 
387
 
 
388
    def parents(self, version):
 
389
        return self._parents[version]
 
390
 
 
391
 
 
392
    def parent_names(self, version):
 
393
        """Return version names for parents of a version."""
 
394
        return map(self.idx_to_name, self._parents[self.lookup(version)])
 
395
 
 
396
 
 
397
    def minimal_parents(self, version):
 
398
        """Find the minimal set of parents for the version."""
 
399
        included = self._parents[version]
 
400
        if not included:
 
401
            return []
 
402
        
 
403
        li = list(included)
 
404
        li.sort(reverse=True)
 
405
 
 
406
        mininc = []
 
407
        gotit = set()
 
408
 
 
409
        for pv in li:
 
410
            if pv not in gotit:
 
411
                mininc.append(pv)
 
412
                gotit.update(self.inclusions(pv))
 
413
 
 
414
        assert mininc[0] >= 0
 
415
        assert mininc[-1] < version
 
416
        return mininc
 
417
 
 
418
 
536
419
 
537
420
    def _check_lines(self, text):
538
421
        if not isinstance(text, list):
553
436
            except IndexError:
554
437
                raise IndexError("invalid version number %r" % i)
555
438
 
556
 
    def _compatible_parents(self, my_parents, other_parents):
557
 
        """During join check that other_parents are joinable with my_parents.
558
 
 
559
 
        Joinable is defined as 'is a subset of' - supersets may require 
560
 
        regeneration of diffs, but subsets do not.
561
 
        """
562
 
        return len(other_parents.difference(my_parents)) == 0
563
 
 
564
 
    def annotate(self, version_id):
565
 
        """Return a list of (version-id, line) tuples for version_id.
 
439
    
 
440
    def annotate(self, name_or_index):
 
441
        return list(self.annotate_iter(name_or_index))
 
442
 
 
443
 
 
444
    def annotate_iter(self, name_or_index):
 
445
        """Yield list of (index-id, line) pairs for the specified version.
566
446
 
567
447
        The index indicates when the line originated in the weave."""
568
 
        incls = [self._lookup(version_id)]
569
 
        return [(self._idx_to_name(origin), text) for origin, lineno, text in
570
 
            self._extract(incls)]
571
 
 
572
 
    def iter_lines_added_or_present_in_versions(self, version_ids=None,
573
 
                                                pb=None):
574
 
        """See VersionedFile.iter_lines_added_or_present_in_versions()."""
575
 
        if version_ids is None:
576
 
            version_ids = self.versions()
577
 
        version_ids = set(version_ids)
578
 
        for lineno, inserted, deletes, line in self._walk_internal(version_ids):
579
 
            # if inserted not in version_ids then it was inserted before the
580
 
            # versions we care about, but because weaves cannot represent ghosts
581
 
            # properly, we do not filter down to that
582
 
            # if inserted not in version_ids: continue
583
 
            if line[-1] != '\n':
584
 
                yield line + '\n', inserted
585
 
            else:
586
 
                yield line, inserted
587
 
 
588
 
    def _walk_internal(self, version_ids=None):
589
 
        """Helper method for weave actions."""
 
448
        incls = [self.maybe_lookup(name_or_index)]
 
449
        for origin, lineno, text in self._extract(incls):
 
450
            yield origin, text
 
451
 
 
452
    def _walk(self):
 
453
        """Walk the weave.
 
454
 
 
455
        Yields sequence of
 
456
        (lineno, insert, deletes, text)
 
457
        for each literal line.
 
458
        """
590
459
        
591
460
        istack = []
592
461
        dset = set()
594
463
        lineno = 0         # line of weave, 0-based
595
464
 
596
465
        for l in self._weave:
597
 
            if l.__class__ == tuple:
 
466
            if isinstance(l, tuple):
598
467
                c, v = l
599
468
                isactive = None
600
469
                if c == '{':
601
 
                    istack.append(self._names[v])
 
470
                    istack.append(v)
602
471
                elif c == '}':
603
472
                    istack.pop()
604
473
                elif c == '[':
605
 
                    dset.add(self._names[v])
 
474
                    assert v not in dset
 
475
                    dset.add(v)
606
476
                elif c == ']':
607
 
                    dset.remove(self._names[v])
 
477
                    dset.remove(v)
608
478
                else:
609
479
                    raise WeaveFormatError('unexpected instruction %r' % v)
610
480
            else:
611
 
                yield lineno, istack[-1], frozenset(dset), l
612
 
            lineno += 1
613
 
 
 
481
                assert isinstance(l, basestring)
 
482
                assert istack
 
483
                yield lineno, istack[-1], dset, l
 
484
            lineno += 1
 
485
 
 
486
 
 
487
 
 
488
    def _extract(self, versions):
 
489
        """Yield annotation of lines in included set.
 
490
 
 
491
        Yields a sequence of tuples (origin, lineno, text), where
 
492
        origin is the origin version, lineno the index in the weave,
 
493
        and text the text of the line.
 
494
 
 
495
        The set typically but not necessarily corresponds to a version.
 
496
        """
 
497
        for i in versions:
 
498
            if not isinstance(i, int):
 
499
                raise ValueError(i)
 
500
            
 
501
        included = self.inclusions(versions)
 
502
 
 
503
        istack = []
 
504
        dset = set()
 
505
 
 
506
        lineno = 0         # line of weave, 0-based
 
507
 
 
508
        isactive = None
 
509
 
 
510
        result = []
 
511
 
 
512
        WFE = WeaveFormatError
 
513
 
 
514
        for l in self._weave:
 
515
            if isinstance(l, tuple):
 
516
                c, v = l
 
517
                isactive = None
 
518
                if c == '{':
 
519
                    assert v not in istack
 
520
                    istack.append(v)
 
521
                elif c == '}':
 
522
                    istack.pop()
 
523
                elif c == '[':
 
524
                    if v in included:
 
525
                        assert v not in dset
 
526
                        dset.add(v)
 
527
                else:
 
528
                    assert c == ']'
 
529
                    if v in included:
 
530
                        assert v in dset
 
531
                        dset.remove(v)
 
532
            else:
 
533
                assert isinstance(l, basestring)
 
534
                if isactive is None:
 
535
                    isactive = (not dset) and istack and (istack[-1] in included)
 
536
                if isactive:
 
537
                    result.append((istack[-1], lineno, l))
 
538
            lineno += 1
614
539
        if istack:
615
540
            raise WeaveFormatError("unclosed insertion blocks "
616
541
                    "at end of weave: %s" % istack)
617
542
        if dset:
618
543
            raise WeaveFormatError("unclosed deletion blocks at end of weave: %s"
619
544
                                   % dset)
620
 
 
 
545
        return result
 
546
 
 
547
 
 
548
    def get_iter(self, name_or_index):
 
549
        """Yield lines for the specified version."""
 
550
        incls = [self.maybe_lookup(name_or_index)]
 
551
        for origin, lineno, line in self._extract(incls):
 
552
            yield line
 
553
 
 
554
 
 
555
    def get_text(self, name_or_index):
 
556
        return ''.join(self.get_iter(name_or_index))
 
557
        assert isinstance(version, int)
 
558
 
 
559
 
 
560
    def get_lines(self, name_or_index):
 
561
        return list(self.get_iter(name_or_index))
 
562
 
 
563
 
 
564
    get = get_lines
 
565
 
 
566
 
 
567
    def mash_iter(self, included):
 
568
        """Return composed version of multiple included versions."""
 
569
        included = map(self.maybe_lookup, included)
 
570
        for origin, lineno, text in self._extract(included):
 
571
            yield text
 
572
 
 
573
 
 
574
    def dump(self, to_file):
 
575
        from pprint import pprint
 
576
        print >>to_file, "Weave._weave = ",
 
577
        pprint(self._weave, to_file)
 
578
        print >>to_file, "Weave._parents = ",
 
579
        pprint(self._parents, to_file)
 
580
 
 
581
 
 
582
 
 
583
    def numversions(self):
 
584
        l = len(self._parents)
 
585
        assert l == len(self._sha1s)
 
586
        return l
 
587
 
 
588
 
 
589
    def __len__(self):
 
590
        return self.numversions()
 
591
 
 
592
 
 
593
    def check(self, progress_bar=None):
 
594
        # check no circular inclusions
 
595
        for version in range(self.numversions()):
 
596
            inclusions = list(self._parents[version])
 
597
            if inclusions:
 
598
                inclusions.sort()
 
599
                if inclusions[-1] >= version:
 
600
                    raise WeaveFormatError("invalid included version %d for index %d"
 
601
                                           % (inclusions[-1], version))
 
602
 
 
603
        # try extracting all versions; this is a bit slow and parallel
 
604
        # extraction could be used
 
605
        nv = self.numversions()
 
606
        for version in range(nv):
 
607
            if progress_bar:
 
608
                progress_bar.update('checking text', version, nv)
 
609
            s = sha.new()
 
610
            for l in self.get_iter(version):
 
611
                s.update(l)
 
612
            hd = s.hexdigest()
 
613
            expected = self._sha1s[version]
 
614
            if hd != expected:
 
615
                raise WeaveError("mismatched sha1 for version %d; "
 
616
                                 "got %s, expected %s"
 
617
                                 % (version, hd, expected))
 
618
 
 
619
        # TODO: check insertions are properly nested, that there are
 
620
        # no lines outside of insertion blocks, that deletions are
 
621
        # properly paired, etc.
 
622
 
 
623
 
 
624
    def _delta(self, included, lines):
 
625
        """Return changes from basis to new revision.
 
626
 
 
627
        The old text for comparison is the union of included revisions.
 
628
 
 
629
        This is used in inserting a new text.
 
630
 
 
631
        Delta is returned as a sequence of
 
632
        (weave1, weave2, newlines).
 
633
 
 
634
        This indicates that weave1:weave2 of the old weave should be
 
635
        replaced by the sequence of lines in newlines.  Note that
 
636
        these line numbers are positions in the total weave and don't
 
637
        correspond to the lines in any extracted version, or even the
 
638
        extracted union of included versions.
 
639
 
 
640
        If line1=line2, this is a pure insert; if newlines=[] this is a
 
641
        pure delete.  (Similar to difflib.)
 
642
        """
 
643
        raise NotImplementedError()
 
644
 
 
645
            
621
646
    def plan_merge(self, ver_a, ver_b):
622
647
        """Return pseudo-annotation indicating how the two versions merge.
623
648
 
626
651
 
627
652
        Weave lines present in none of them are skipped entirely.
628
653
        """
629
 
        inc_a = set(self.get_ancestry([ver_a]))
630
 
        inc_b = set(self.get_ancestry([ver_b]))
 
654
        inc_a = self.inclusions([ver_a])
 
655
        inc_b = self.inclusions([ver_b])
631
656
        inc_c = inc_a & inc_b
632
657
 
633
 
        for lineno, insert, deleteset, line in self._walk_internal([ver_a, ver_b]):
 
658
        for lineno, insert, deleteset, line in self._walk():
634
659
            if deleteset & inc_c:
635
660
                # killed in parent; can't be in either a or b
636
661
                # not relevant to our work
662
687
                # not in either revision
663
688
                yield 'irrelevant', line
664
689
 
665
 
    def _extract(self, versions):
666
 
        """Yield annotation of lines in included set.
667
 
 
668
 
        Yields a sequence of tuples (origin, lineno, text), where
669
 
        origin is the origin version, lineno the index in the weave,
670
 
        and text the text of the line.
671
 
 
672
 
        The set typically but not necessarily corresponds to a version.
673
 
        """
674
 
        for i in versions:
675
 
            if not isinstance(i, int):
676
 
                raise ValueError(i)
677
 
            
678
 
        included = self._inclusions(versions)
679
 
 
680
 
        istack = []
681
 
        iset = set()
682
 
        dset = set()
683
 
 
684
 
        lineno = 0         # line of weave, 0-based
685
 
 
686
 
        isactive = None
687
 
 
688
 
        result = []
689
 
 
690
 
        WFE = WeaveFormatError
691
 
 
692
 
        # wow. 
693
 
        #  449       0   4474.6820   2356.5590   bzrlib.weave:556(_extract)
694
 
        #  +285282   0   1676.8040   1676.8040   +<isinstance>
695
 
        # 1.6 seconds in 'isinstance'.
696
 
        # changing the first isinstance:
697
 
        #  449       0   2814.2660   1577.1760   bzrlib.weave:556(_extract)
698
 
        #  +140414   0    762.8050    762.8050   +<isinstance>
699
 
        # note that the inline time actually dropped (less function calls)
700
 
        # and total processing time was halved.
701
 
        # we're still spending ~1/4 of the method in isinstance though.
702
 
        # so lets hard code the acceptable string classes we expect:
703
 
        #  449       0   1202.9420    786.2930   bzrlib.weave:556(_extract)
704
 
        # +71352     0    377.5560    377.5560   +<method 'append' of 'list' 
705
 
        #                                          objects>
706
 
        # yay, down to ~1/4 the initial extract time, and our inline time
707
 
        # has shrunk again, with isinstance no longer dominating.
708
 
        # tweaking the stack inclusion test to use a set gives:
709
 
        #  449       0   1122.8030    713.0080   bzrlib.weave:556(_extract)
710
 
        # +71352     0    354.9980    354.9980   +<method 'append' of 'list' 
711
 
        #                                          objects>
712
 
        # - a 5% win, or possibly just noise. However with large istacks that
713
 
        # 'in' test could dominate, so I'm leaving this change in place -
714
 
        # when its fast enough to consider profiling big datasets we can review.
715
 
 
716
 
              
717
 
             
718
 
 
719
 
        for l in self._weave:
720
 
            if l.__class__ == tuple:
721
 
                c, v = l
722
 
                isactive = None
723
 
                if c == '{':
724
 
                    istack.append(v)
725
 
                    iset.add(v)
726
 
                elif c == '}':
727
 
                    iset.remove(istack.pop())
728
 
                elif c == '[':
729
 
                    if v in included:
730
 
                        dset.add(v)
731
 
                elif c == ']':
732
 
                    if v in included:
733
 
                        dset.remove(v)
 
690
        yield 'unchanged', ''           # terminator
 
691
 
 
692
 
 
693
 
 
694
    def weave_merge(self, plan):
 
695
        lines_a = []
 
696
        lines_b = []
 
697
        ch_a = ch_b = False
 
698
 
 
699
        for state, line in plan:
 
700
            if state == 'unchanged' or state == 'killed-both':
 
701
                # resync and flush queued conflicts changes if any
 
702
                if not lines_a and not lines_b:
 
703
                    pass
 
704
                elif ch_a and not ch_b:
 
705
                    # one-sided change:                    
 
706
                    for l in lines_a: yield l
 
707
                elif ch_b and not ch_a:
 
708
                    for l in lines_b: yield l
 
709
                elif lines_a == lines_b:
 
710
                    for l in lines_a: yield l
734
711
                else:
735
 
                    raise AssertionError()
 
712
                    yield '<<<<<<<\n'
 
713
                    for l in lines_a: yield l
 
714
                    yield '=======\n'
 
715
                    for l in lines_b: yield l
 
716
                    yield '>>>>>>>\n'
 
717
 
 
718
                del lines_a[:]
 
719
                del lines_b[:]
 
720
                ch_a = ch_b = False
 
721
                
 
722
            if state == 'unchanged':
 
723
                if line:
 
724
                    yield line
 
725
            elif state == 'killed-a':
 
726
                ch_a = True
 
727
                lines_b.append(line)
 
728
            elif state == 'killed-b':
 
729
                ch_b = True
 
730
                lines_a.append(line)
 
731
            elif state == 'new-a':
 
732
                ch_a = True
 
733
                lines_a.append(line)
 
734
            elif state == 'new-b':
 
735
                ch_b = True
 
736
                lines_b.append(line)
736
737
            else:
737
 
                if isactive is None:
738
 
                    isactive = (not dset) and istack and (istack[-1] in included)
739
 
                if isactive:
740
 
                    result.append((istack[-1], lineno, l))
741
 
            lineno += 1
742
 
        if istack:
743
 
            raise WeaveFormatError("unclosed insertion blocks "
744
 
                    "at end of weave: %s" % istack)
745
 
        if dset:
746
 
            raise WeaveFormatError("unclosed deletion blocks at end of weave: %s"
747
 
                                   % dset)
748
 
        return result
749
 
 
750
 
    def _maybe_lookup(self, name_or_index):
751
 
        """Convert possible symbolic name to index, or pass through indexes.
752
 
        
753
 
        NOT FOR PUBLIC USE.
 
738
                assert state in ('irrelevant', 'ghost-a', 'ghost-b', 'killed-base',
 
739
                                 'killed-both'), \
 
740
                       state
 
741
 
 
742
                
 
743
    def join(self, other):
 
744
        """Integrate versions from other into this weave.
 
745
 
 
746
        The resulting weave contains all the history of both weaves; 
 
747
        any version you could retrieve from either self or other can be 
 
748
        retrieved from self after this call.
 
749
 
 
750
        It is illegal for the two weaves to contain different values 
 
751
        or different parents for any version.  See also reweave().
754
752
        """
755
 
        if isinstance(name_or_index, (int, long)):
756
 
            return name_or_index
757
 
        else:
758
 
            return self._lookup(name_or_index)
759
 
 
760
 
    def get_lines(self, version_id):
761
 
        """See VersionedFile.get_lines()."""
762
 
        int_index = self._maybe_lookup(version_id)
763
 
        result = [line for (origin, lineno, line) in self._extract([int_index])]
764
 
        expected_sha1 = self._sha1s[int_index]
765
 
        measured_sha1 = sha_strings(result)
766
 
        if measured_sha1 != expected_sha1:
767
 
            raise errors.WeaveInvalidChecksum(
768
 
                    'file %s, revision %s, expected: %s, measured %s' 
769
 
                    % (self._weave_name, version_id,
770
 
                       expected_sha1, measured_sha1))
771
 
        return result
772
 
 
773
 
    def get_sha1s(self, version_ids):
774
 
        """See VersionedFile.get_sha1s()."""
775
 
        result = {}
776
 
        for v in version_ids:
777
 
            result[v] = self._sha1s[self._lookup(v)]
778
 
        return result
779
 
 
780
 
    def num_versions(self):
781
 
        """How many versions are in this weave?"""
782
 
        l = len(self._parents)
783
 
        return l
784
 
 
785
 
    __len__ = num_versions
786
 
 
787
 
    def check(self, progress_bar=None):
788
 
        # TODO evaluate performance hit of using string sets in this routine.
789
 
        # TODO: check no circular inclusions
790
 
        # TODO: create a nested progress bar
791
 
        for version in range(self.num_versions()):
792
 
            inclusions = list(self._parents[version])
793
 
            if inclusions:
794
 
                inclusions.sort()
795
 
                if inclusions[-1] >= version:
796
 
                    raise WeaveFormatError("invalid included version %d for index %d"
797
 
                                           % (inclusions[-1], version))
798
 
 
799
 
        # try extracting all versions; parallel extraction is used
800
 
        nv = self.num_versions()
801
 
        sha1s = {}
802
 
        texts = {}
803
 
        inclusions = {}
804
 
        for i in range(nv):
805
 
            # For creating the ancestry, IntSet is much faster (3.7s vs 0.17s)
806
 
            # The problem is that set membership is much more expensive
807
 
            name = self._idx_to_name(i)
808
 
            sha1s[name] = sha()
809
 
            texts[name] = []
810
 
            new_inc = set([name])
811
 
            for p in self._parents[i]:
812
 
                new_inc.update(inclusions[self._idx_to_name(p)])
813
 
 
814
 
            if set(new_inc) != set(self.get_ancestry(name)):
815
 
                raise AssertionError(
816
 
                    'failed %s != %s' 
817
 
                    % (set(new_inc), set(self.get_ancestry(name))))
818
 
            inclusions[name] = new_inc
819
 
 
820
 
        nlines = len(self._weave)
821
 
 
822
 
        update_text = 'checking weave'
823
 
        if self._weave_name:
824
 
            short_name = os.path.basename(self._weave_name)
825
 
            update_text = 'checking %s' % (short_name,)
826
 
            update_text = update_text[:25]
827
 
 
828
 
        for lineno, insert, deleteset, line in self._walk_internal():
829
 
            if progress_bar:
830
 
                progress_bar.update(update_text, lineno, nlines)
831
 
 
832
 
            for name, name_inclusions in inclusions.items():
833
 
                # The active inclusion must be an ancestor,
834
 
                # and no ancestors must have deleted this line,
835
 
                # because we don't support resurrection.
836
 
                if (insert in name_inclusions) and not (deleteset & name_inclusions):
837
 
                    sha1s[name].update(line)
838
 
 
839
 
        for i in range(nv):
840
 
            version = self._idx_to_name(i)
841
 
            hd = sha1s[version].hexdigest()
842
 
            expected = self._sha1s[i]
843
 
            if hd != expected:
844
 
                raise errors.WeaveInvalidChecksum(
845
 
                        "mismatched sha1 for version %s: "
846
 
                        "got %s, expected %s"
847
 
                        % (version, hd, expected))
848
 
 
849
 
        # TODO: check insertions are properly nested, that there are
850
 
        # no lines outside of insertion blocks, that deletions are
851
 
        # properly paired, etc.
 
753
        if other.numversions() == 0:
 
754
            return          # nothing to update, easy
 
755
        # two loops so that we do not change ourselves before verifying it
 
756
        # will be ok
 
757
        # work through in index order to make sure we get all dependencies
 
758
        for other_idx, name in enumerate(other._names):
 
759
            if self._check_version_consistent(other, other_idx, name):
 
760
                continue
 
761
        for other_idx, name in enumerate(other._names):
 
762
            # TODO: If all the parents of the other version are already 
 
763
            # present then we can avoid some work by just taking the delta
 
764
            # and adjusting the offsets.
 
765
            new_parents = self._imported_parents(other, other_idx)
 
766
            lines = other.get_lines(other_idx)
 
767
            sha1 = other._sha1s[other_idx]
 
768
            self.add(name, new_parents, lines, sha1)
 
769
 
852
770
 
853
771
    def _imported_parents(self, other, other_idx):
854
772
        """Return list of parents in self corresponding to indexes in other."""
855
773
        new_parents = []
856
774
        for parent_idx in other._parents[other_idx]:
857
775
            parent_name = other._names[parent_idx]
858
 
            if parent_name not in self._name_map:
 
776
            if parent_name not in self._names:
859
777
                # should not be possible
860
778
                raise WeaveError("missing parent {%s} of {%s} in %r" 
861
779
                                 % (parent_name, other._name_map[other_idx], self))
877
795
        this_idx = self._name_map.get(name, -1)
878
796
        if this_idx != -1:
879
797
            if self._sha1s[this_idx] != other._sha1s[other_idx]:
880
 
                raise errors.WeaveTextDiffers(name, self, other)
 
798
                raise WeaveError("inconsistent texts for version {%s} "
 
799
                                 "when joining weaves"
 
800
                                 % (name))
881
801
            self_parents = self._parents[this_idx]
882
802
            other_parents = other._parents[other_idx]
883
 
            n1 = set([self._names[i] for i in self_parents])
884
 
            n2 = set([other._names[i] for i in other_parents])
885
 
            if not self._compatible_parents(n1, n2):
 
803
            n1 = [self._names[i] for i in self_parents]
 
804
            n2 = [other._names[i] for i in other_parents]
 
805
            n1.sort()
 
806
            n2.sort()
 
807
            if n1 != n2:
886
808
                raise WeaveParentMismatch("inconsistent parents "
887
809
                    "for version {%s}: %s vs %s" % (name, n1, n2))
888
810
            else:
890
812
        else:
891
813
            return False
892
814
 
893
 
    def _reweave(self, other, pb, msg):
894
 
        """Reweave self with other - internal helper for join().
895
 
 
896
 
        :param other: The other weave to merge
897
 
        :param pb: An optional progress bar, indicating how far done we are
898
 
        :param msg: An optional message for the progress
899
 
        """
900
 
        new_weave = _reweave(self, other, pb=pb, msg=msg)
901
 
        self._copy_weave_content(new_weave)
902
 
 
903
 
    def _copy_weave_content(self, otherweave):
904
 
        """adsorb the content from otherweave."""
 
815
    def reweave(self, other):
 
816
        """Reweave self with other."""
 
817
        new_weave = reweave(self, other)
905
818
        for attr in self.__slots__:
906
 
            if attr != '_weave_name':
907
 
                setattr(self, attr, copy(getattr(otherweave, attr)))
908
 
 
909
 
 
910
 
class WeaveFile(Weave):
911
 
    """A WeaveFile represents a Weave on disk and writes on change."""
912
 
 
913
 
    WEAVE_SUFFIX = '.weave'
914
 
    
915
 
    def __init__(self, name, transport, filemode=None, create=False, access_mode='w', get_scope=None):
916
 
        """Create a WeaveFile.
917
 
        
918
 
        :param create: If not True, only open an existing knit.
919
 
        """
920
 
        super(WeaveFile, self).__init__(name, access_mode, get_scope=get_scope,
921
 
            allow_reserved=False)
922
 
        self._transport = transport
923
 
        self._filemode = filemode
924
 
        try:
925
 
            _read_weave_v5(self._transport.get(name + WeaveFile.WEAVE_SUFFIX), self)
926
 
        except errors.NoSuchFile:
927
 
            if not create:
928
 
                raise
929
 
            # new file, save it
930
 
            self._save()
931
 
 
932
 
    def _add_lines(self, version_id, parents, lines, parent_texts,
933
 
        left_matching_blocks, nostore_sha, random_id, check_content):
934
 
        """Add a version and save the weave."""
935
 
        self.check_not_reserved_id(version_id)
936
 
        result = super(WeaveFile, self)._add_lines(version_id, parents, lines,
937
 
            parent_texts, left_matching_blocks, nostore_sha, random_id,
938
 
            check_content)
939
 
        self._save()
940
 
        return result
941
 
 
942
 
    def copy_to(self, name, transport):
943
 
        """See VersionedFile.copy_to()."""
944
 
        # as we are all in memory always, just serialise to the new place.
945
 
        sio = StringIO()
946
 
        write_weave_v5(self, sio)
947
 
        sio.seek(0)
948
 
        transport.put_file(name + WeaveFile.WEAVE_SUFFIX, sio, self._filemode)
949
 
 
950
 
    def _save(self):
951
 
        """Save the weave."""
952
 
        self._check_write_ok()
953
 
        sio = StringIO()
954
 
        write_weave_v5(self, sio)
955
 
        sio.seek(0)
956
 
        bytes = sio.getvalue()
957
 
        path = self._weave_name + WeaveFile.WEAVE_SUFFIX
958
 
        try:
959
 
            self._transport.put_bytes(path, bytes, self._filemode)
960
 
        except errors.NoSuchFile:
961
 
            self._transport.mkdir(dirname(path))
962
 
            self._transport.put_bytes(path, bytes, self._filemode)
963
 
 
964
 
    @staticmethod
965
 
    def get_suffixes():
966
 
        """See VersionedFile.get_suffixes()."""
967
 
        return [WeaveFile.WEAVE_SUFFIX]
968
 
 
969
 
    def insert_record_stream(self, stream):
970
 
        super(WeaveFile, self).insert_record_stream(stream)
971
 
        self._save()
972
 
 
973
 
    @deprecated_method(one_five)
974
 
    def join(self, other, pb=None, msg=None, version_ids=None,
975
 
             ignore_missing=False):
976
 
        """Join other into self and save."""
977
 
        super(WeaveFile, self).join(other, pb, msg, version_ids, ignore_missing)
978
 
        self._save()
979
 
 
980
 
 
981
 
def _reweave(wa, wb, pb=None, msg=None):
 
819
            setattr(self, attr, getattr(new_weave, attr))
 
820
 
 
821
 
 
822
def reweave(wa, wb):
982
823
    """Combine two weaves and return the result.
983
824
 
984
825
    This works even if a revision R has different parents in 
989
830
    might be possible but it should only be necessary to do 
990
831
    this operation rarely, when a new previously ghost version is 
991
832
    inserted.
992
 
 
993
 
    :param pb: An optional progress bar, indicating how far done we are
994
 
    :param msg: An optional message for the progress
995
833
    """
996
834
    wr = Weave()
997
835
    ia = ib = 0
998
 
    queue_a = range(wa.num_versions())
999
 
    queue_b = range(wb.num_versions())
 
836
    queue_a = range(wa.numversions())
 
837
    queue_b = range(wb.numversions())
1000
838
    # first determine combined parents of all versions
1001
839
    # map from version name -> all parent names
1002
840
    combined_parents = _reweave_parent_graphs(wa, wb)
1003
841
    mutter("combined parents: %r", combined_parents)
1004
 
    order = tsort.topo_sort(combined_parents.iteritems())
 
842
    order = topo_sort(combined_parents.iteritems())
1005
843
    mutter("order to reweave: %r", order)
1006
 
 
1007
 
    if pb and not msg:
1008
 
        msg = 'reweave'
1009
 
 
1010
 
    for idx, name in enumerate(order):
1011
 
        if pb:
1012
 
            pb.update(msg, idx, len(order))
 
844
    for name in order:
1013
845
        if name in wa._name_map:
1014
846
            lines = wa.get_lines(name)
1015
847
            if name in wb._name_map:
1016
 
                lines_b = wb.get_lines(name)
1017
 
                if lines != lines_b:
1018
 
                    mutter('Weaves differ on content. rev_id {%s}', name)
1019
 
                    mutter('weaves: %s, %s', wa._weave_name, wb._weave_name)
1020
 
                    import difflib
1021
 
                    lines = list(difflib.unified_diff(lines, lines_b,
1022
 
                            wa._weave_name, wb._weave_name))
1023
 
                    mutter('lines:\n%s', ''.join(lines))
1024
 
                    raise errors.WeaveTextDiffers(name, wa, wb)
 
848
                assert lines == wb.get_lines(name)
1025
849
        else:
1026
850
            lines = wb.get_lines(name)
1027
 
        wr._add(name, lines, [wr._lookup(i) for i in combined_parents[name]])
 
851
        wr.add(name, combined_parents[name], lines)
1028
852
    return wr
1029
853
 
 
854
 
1030
855
def _reweave_parent_graphs(wa, wb):
1031
856
    """Return combined parent ancestry for two weaves.
1032
857
    
1035
860
    for weave in [wa, wb]:
1036
861
        for idx, name in enumerate(weave._names):
1037
862
            p = combined.setdefault(name, set())
1038
 
            p.update(map(weave._idx_to_name, weave._parents[idx]))
 
863
            p.update(map(weave.idx_to_name, weave._parents[idx]))
1039
864
    return combined
1040
865
 
1041
866
 
1045
870
    for i in (6, 50, 10, 10):
1046
871
        print '-' * i,
1047
872
    print
1048
 
    for i in range(w.num_versions()):
 
873
    for i in range(w.numversions()):
1049
874
        sha1 = w._sha1s[i]
1050
875
        name = w._names[i]
1051
876
        parent_str = ' '.join(map(str, w._parents[i]))
1098
923
        Add NEWTEXT, with specified parent versions.
1099
924
    weave annotate WEAVEFILE VERSION
1100
925
        Display origin of each line.
 
926
    weave mash WEAVEFILE VERSION...
 
927
        Display composite of all selected versions.
1101
928
    weave merge WEAVEFILE VERSION1 VERSION2 > OUT
1102
929
        Auto-merge two versions and display conflicts.
1103
930
    weave diff WEAVEFILE VERSION1 VERSION2 
1177
1004
        w = readit()
1178
1005
        sys.stdout.writelines(w.get_iter(int(argv[3])))
1179
1006
        
 
1007
    elif cmd == 'mash': # get composite
 
1008
        w = readit()
 
1009
        sys.stdout.writelines(w.mash_iter(map(int, argv[3:])))
 
1010
 
1180
1011
    elif cmd == 'diff':
 
1012
        from difflib import unified_diff
1181
1013
        w = readit()
1182
1014
        fn = argv[2]
1183
1015
        v1, v2 = map(int, argv[3:5])
1184
1016
        lines1 = w.get(v1)
1185
1017
        lines2 = w.get(v2)
1186
 
        diff_gen = bzrlib.patiencediff.unified_diff(lines1, lines2,
 
1018
        diff_gen = unified_diff(lines1, lines2,
1187
1019
                                '%s version %d' % (fn, v1),
1188
1020
                                '%s version %d' % (fn, v2))
1189
1021
        sys.stdout.writelines(diff_gen)
1212
1044
        pb = ProgressBar()
1213
1045
        w.check(pb)
1214
1046
        pb.clear()
1215
 
        print '%d versions ok' % w.num_versions()
 
1047
        print '%d versions ok' % w.numversions()
1216
1048
 
1217
1049
    elif cmd == 'inclusions':
1218
1050
        w = readit()
1223
1055
        print ' '.join(map(str, w._parents[int(argv[3])]))
1224
1056
 
1225
1057
    elif cmd == 'plan-merge':
1226
 
        # replaced by 'bzr weave-plan-merge'
1227
1058
        w = readit()
1228
1059
        for state, line in w.plan_merge(int(argv[3]), int(argv[4])):
1229
1060
            if line:
1230
1061
                print '%14s | %s' % (state, line),
 
1062
 
1231
1063
    elif cmd == 'merge':
1232
 
        # replaced by 'bzr weave-merge-text'
1233
1064
        w = readit()
1234
1065
        p = w.plan_merge(int(argv[3]), int(argv[4]))
1235
1066
        sys.stdout.writelines(w.weave_merge(p))
 
1067
            
 
1068
    elif cmd == 'mash-merge':
 
1069
        if len(argv) != 5:
 
1070
            usage()
 
1071
            return 1
 
1072
 
 
1073
        w = readit()
 
1074
        v1, v2 = map(int, argv[3:5])
 
1075
 
 
1076
        basis = w.inclusions([v1]).intersection(w.inclusions([v2]))
 
1077
 
 
1078
        base_lines = list(w.mash_iter(basis))
 
1079
        a_lines = list(w.get(v1))
 
1080
        b_lines = list(w.get(v2))
 
1081
 
 
1082
        from bzrlib.merge3 import Merge3
 
1083
        m3 = Merge3(base_lines, a_lines, b_lines)
 
1084
 
 
1085
        name_a = 'version %d' % v1
 
1086
        name_b = 'version %d' % v2
 
1087
        sys.stdout.writelines(m3.merge_lines(name_a=name_a, name_b=name_b))
1236
1088
    else:
1237
1089
        raise ValueError('unknown command %r' % cmd)
1238
1090
    
1239
1091
 
 
1092
 
 
1093
def profile_main(argv): 
 
1094
    import tempfile, hotshot, hotshot.stats
 
1095
 
 
1096
    prof_f = tempfile.NamedTemporaryFile()
 
1097
 
 
1098
    prof = hotshot.Profile(prof_f.name)
 
1099
 
 
1100
    ret = prof.runcall(main, argv)
 
1101
    prof.close()
 
1102
 
 
1103
    stats = hotshot.stats.load(prof_f.name)
 
1104
    #stats.strip_dirs()
 
1105
    stats.sort_stats('cumulative')
 
1106
    ## XXX: Might like to write to stderr or the trace file instead but
 
1107
    ## print_stats seems hardcoded to stdout
 
1108
    stats.print_stats(20)
 
1109
            
 
1110
    return ret
 
1111
 
 
1112
 
1240
1113
if __name__ == '__main__':
1241
1114
    import sys
1242
 
    sys.exit(main(sys.argv))
 
1115
    if '--profile' in sys.argv:
 
1116
        args = sys.argv[:]
 
1117
        args.remove('--profile')
 
1118
        sys.exit(profile_main(args))
 
1119
    else:
 
1120
        sys.exit(main(sys.argv))
 
1121