~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to tools/weavebench.py

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

• reverse the comparison (3644.2.10 to 6079)
• stop comparing with revision 6079

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, seed

 

27

import tempfile

 

28

import hotshot, hotshot.stats

 

29

import sys

 

30

 

 

31

WEAVE_NAME = "bench.weave"

 

32

NUM_REVS = 2000

 

33

 

 

34

seed(0)

 

35

 

 

36

def build():

 

37

    pb = ProgressBar(show_eta=False)

 

38

 

 

39

    wf = Weave()

 

40

    lines = []

 

41

 

 

42

    parents = []

 

43

    for i in xrange(NUM_REVS):

 

44

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

 

45

 

 

46

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

 

47

            o = randint(0, len(lines))

 

48

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

 

49

 

 

50

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

 

51

            if lines:

 

52

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

 

53

 

 

54

        rev_id = wf.add("%s" % i, parents, lines)

 

55

        parents = [rev_id]

 

56

 

 

57

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

 

58

 

 

59

        

 

60

#     parents = set()

 

61

#     revno = 1

 

62

#     rev_history = b.revision_history()

 

63

#     for rev_id in rev_history:

 

64

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

 

65

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

 

66

#         weave_id = wf.add(parents, inv_xml)

 

67

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

 

68

#         revno += 1

 

69

 

 

70

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

 

71

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

 

72

 

 

73

    pb.clear()

 

74

 

 

75

 

 

76

 

 

77

def profileit(fn): 

 

78

    prof_f = tempfile.NamedTemporaryFile()

 

79

 

 

80

    prof = hotshot.Profile(prof_f.name)

 

81

 

 

82

    prof.runcall(fn) 

 

83

    prof.close()

 

84

 

 

85

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

 

86

    #stats.strip_dirs()

 

87

    stats.sort_stats('time')

 

88

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

 

89

    ## print_stats seems hardcoded to stdout

 

90

    stats.print_stats(20)

 

91

            

 

92

 

 

93

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

 

94

    opt_p = True

 

95

    sys.argv.remove('-p')

 

96

else:

 

97

    opt_p = False

 

98

 

 

99

if len(sys.argv) > 1:

 

100

    NUM_REVS = int(sys.argv[1])

 

101

 

 

102

if opt_p:

 

103

    profileit(build)

 

104

else:

 

105

    build()

 

106

    

• Older »

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