~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to tools/weavebench.py

• browse files at revision 923
• compare with another revision
• download diff
• download tarball
• view history from revision 923

• reverse the comparison (923 to 41)
• stop comparing with revision 41

Show diffs side-by-side

added

removed

tools/weavebench.py

 

1

#! /usr/bin/python

 

2

 

 

3

# Copyright (C) 2005 Canonical Ltd

 

4

 

 

5

# This program is free software; you can redistribute it and/or modify

 

6

# it under the terms of the GNU General Public License as published by

 

7

# the Free Software Foundation; either version 2 of the License, or

 

8

# (at your option) any later version.

 

9

 

 

10

# This program is distributed in the hope that it will be useful,

 

11

# but WITHOUT ANY WARRANTY; without even the implied warranty of

 

12

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 

13

# GNU General Public License for more details.

 

14

 

 

15

# You should have received a copy of the GNU General Public License

 

16

# along with this program; if not, write to the Free Software

 

17

# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 

18

 

 

19

 

 

20

"""Weave algorithms benchmark"""

 

21

 

 

22

import bzrlib.branch

 

23

from bzrlib.weave import Weave

 

24

from bzrlib.weavefile import write_weave

 

25

from bzrlib.progress import ProgressBar

 

26

from random import randrange, randint

 

27

import tempfile

 

28

import hotshot, hotshot.stats

 

29

import sys

 

30

 

 

31

WEAVE_NAME = "bench.weave"

 

32

NUM_REVS = 10000

 

33

 

 

34

def build():

 

35

    pb = ProgressBar()

 

36

 

 

37

    wf = Weave()

 

38

    lines = []

 

39

 

 

40

    parents = []

 

41

    for i in xrange(NUM_REVS):

 

42

        pb.update('building', i, NUM_REVS)

 

43

 

 

44

        for j in range(randint(0, 4)):

 

45

            o = randint(0, len(lines))

 

46

            lines.insert(o, "new in version %i\n" % i)

 

47

 

 

48

        for j in range(randint(0, 2)):

 

49

            if lines:

 

50

                del lines[randrange(0, len(lines))]

 

51

 

 

52

        rev_id = wf.add(parents, lines)

 

53

        parents = [rev_id]

 

54

 

 

55

    write_weave(wf, file(WEAVE_NAME, 'wb'))

 

56

 

 

57

        

 

58

#     parents = set()

 

59

#     revno = 1

 

60

#     rev_history = b.revision_history()

 

61

#     for rev_id in rev_history:

 

62

#         pb.update('converting inventory', revno, len(rev_history))

 

63

#         inv_xml = b.inventory_store[rev_id].readlines()

 

64

#         weave_id = wf.add(parents, inv_xml)

 

65

#         parents = set([weave_id])       # always just one parent

 

66

#         revno += 1

 

67

 

 

68

#     pb.update('write weave', None, None)

 

69

#     write_weave(wf, file(WEAVE_NAME, 'wb'))

 

70

 

 

71

    pb.clear()

 

72

 

 

73

 

 

74

 

 

75

def profileit(fn): 

 

76

    prof_f = tempfile.NamedTemporaryFile()

 

77

 

 

78

    prof = hotshot.Profile(prof_f.name)

 

79

 

 

80

    prof.runcall(fn) 

 

81

    prof.close()

 

82

 

 

83

    stats = hotshot.stats.load(prof_f.name)

 

84

    #stats.strip_dirs()

 

85

    stats.sort_stats('time')

 

86

    ## XXX: Might like to write to stderr or the trace file instead but

 

87

    ## print_stats seems hardcoded to stdout

 

88

    stats.print_stats(20)

 

89

            

 

90

 

 

91

if '-p' in sys.argv[1:]:

 

92

    profileit(build)

 

93

else:

 

94

    build()

 

95

    

• Older »

Loggerhead is a web-based interface for Breezy
Version: 2.0.1