~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-23 15:44:24 UTC
  • mto: (1185.50.19 bzr-jam-integration)
  • mto: This revision was merged to the branch mainline in revision 1512.
  • Revision ID: john@arbash-meinel.com-20051123154424-a02f8bf990a1fed5
Renamed all of the tests from selftest/foo.py to tests/test_foo.py

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