← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to ancestry_test.py

  • Committer: John Arbash Meinel
  • Date: 2009-08-13 20:15:12 UTC
  • mto: This revision was merged to the branch mainline in revision 4613.
  • Revision ID: john@arbash-meinel.com-20090813201512-yewwimsurqgc0ym0
Remove NOTES and ancestry_test as they were just steps along the way.

Show diffs side-by-side

added added

removed removed

Lines of Context:
ancestry_test.py
1
 
 
2
 
from bzrlib import branch
3
 
 
4
 
def get_gcindex(path):
5
 
    b = branch.Branch.open(path)
6
 
    b.lock_read()
7
 
    r = b.repository
8
 
    rev_id = b.last_revision()
9
 
    rev_key = (rev_id,)
10
 
    gcindex = r.revisions._index
11
 
    return b, rev_key, gcindex
12
 
 
13
 
 
14
 
def get_bindex(path):
15
 
    b = branch.Branch.open(path)
16
 
    b.lock_read()
17
 
    r = b.repository
18
 
    rev_id = b.last_revision()
19
 
    rev_key = (rev_id,)
20
 
    bindex = r.revisions._index._graph_index._indices[0]
21
 
    return b, rev_key, bindex
22
 
 
23
 
def get_cindex(path):
24
 
    b = branch.Branch.open(path)
25
 
    b.lock_read()
26
 
    r = b.repository
27
 
    rev_id = b.last_revision()
28
 
    rev_key = (rev_id,)
29
 
    cindex = r.revisions._index._graph_index
30
 
    return b, rev_key, cindex
31
 
 
32
 
 
33
 
 
34
 
def ancestry_from_get_ancestry(path):
35
 
    b, rev_key, cindex = get_cindex(path)
36
 
    parent_map, missing_keys = cindex.get_ancestry([rev_key])
37
 
    b.unlock()
38
 
 
39
 
 
40
 
def ancestry_from_get_parent_map(path):
41
 
    b, rev_key, gcindex = get_gcindex(path)
42
 
    search_keys = set([rev_key])
43
 
    parent_map = {}
44
 
    generation = 0
45
 
    while search_keys:
46
 
        next_parent_map = gcindex.get_parent_map(search_keys)
47
 
        next_parent_keys = set()
48
 
        map(next_parent_keys.update, next_parent_map.itervalues())
49
 
        parent_map.update(next_parent_map)
50
 
        next_parent_keys = next_parent_keys.difference(parent_map)
51
 
        generation += 1
52
 
        # print '%4d\t%5d\t%5d' % (generation, len(search_keys),
53
 
        #                          len(parent_map))
54
 
        search_keys = next_parent_keys
55
 
    b.unlock()