git.cworth.org Git - apitrace/blob - scripts/unpickle.py

   1 #!/usr/bin/env python
   2 ##########################################################################
   3 #
   4 # Copyright 2012 Jose Fonseca
   5 # All Rights Reserved.
   6 #
   7 # Permission is hereby granted, free of charge, to any person obtaining a copy
   8 # of this software and associated documentation files (the "Software"), to deal
   9 # in the Software without restriction, including without limitation the rights
  10 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  11 # copies of the Software, and to permit persons to whom the Software is
  12 # furnished to do so, subject to the following conditions:
  13 #
  14 # The above copyright notice and this permission notice shall be included in
  15 # all copies or substantial portions of the Software.
  16 #
  17 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  18 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  19 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  20 # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  21 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  22 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  23 # THE SOFTWARE.
  24 #
  25 ##########################################################################/
  26
  27 '''Sample program for apitrace pickle command.
  28
  29 Run as:
  30
  31    apitrace pickle foo.trace | python unpickle.py
  32
  33 '''
  34
  35
  36 import optparse
  37 import cPickle as pickle
  38 import sys
  39 import time
  40
  41
  42 class Call:
  43
  44     def __init__(self, callTuple):
  45         self.no, self.functionName, self.args, self.ret = callTuple
  46
  47     def __str__(self):
  48         s = self.functionName
  49         if self.no is not None:
  50             s = str(self.no) + ' ' + s
  51         s += '(' + ', '.join(map(repr, self.args)) + ')'
  52         if self.ret is not None:
  53             s += ' = '
  54             s += repr(self.ret)
  55         return s
  56
  57     def __eq__(self, other):
  58         return \
  59             self.functionName == other.functionName and \
  60             self.args == other.args and \
  61             self.ret == other.ret
  62
  63     def __hash__(self):
  64         # XXX: hack due to unhashable types
  65         #return hash(self.functionName) ^ hash(tuple(self.args)) ^ hash(self.ret)
  66         return hash(self.functionName) ^ hash(repr(self.args)) ^ hash(repr(self.ret))
  67
  68
  69
  70 class Unpickler:
  71
  72     callFactory = Call
  73
  74     def __init__(self, stream):
  75         self.stream = stream
  76
  77     def parse(self):
  78         while self.parseCall():
  79             pass
  80
  81     def parseCall(self):
  82         try:
  83             callTuple = pickle.load(self.stream)
  84         except EOFError:
  85             return False
  86         else:
  87             call = self.callFactory(callTuple)
  88             self.handleCall(call)
  89             return True
  90
  91     def handleCall(self, call):
  92         pass
  93
  94
  95 class Counter(Unpickler):
  96
  97     def __init__(self, stream, quiet):
  98         Unpickler.__init__(self, stream)
  99         self.quiet = quiet
 100         self.calls = 0
 101
 102     def handleCall(self, call):
 103         if not self.quiet:
 104             sys.stdout.write(str(call))
 105             sys.stdout.write('\n')
 106         self.calls += 1
 107
 108
 109 def main():
 110     optparser = optparse.OptionParser(
 111         usage="\n\tapitrace pickle trace. %prog [options]")
 112     optparser.add_option(
 113         '--quiet',
 114         action="store_true", dest="quiet", default=False,
 115         help="don't dump calls to stdout")
 116
 117     (options, args) = optparser.parse_args(sys.argv[1:])
 118
 119     if args:
 120         optparser.error('unexpected arguments')
 121
 122     # Change stdin to binary mode
 123     try:
 124         import msvcrt
 125     except ImportError:
 126         pass
 127     else:
 128         import os
 129         msvcrt.setmode(sys.stdin.fileno(), os.O_BINARY)
 130
 131     startTime = time.time()
 132     parser = Counter(sys.stdin, options.quiet)
 133     parser.parse()
 134     stopTime = time.time()
 135     duration = stopTime - startTime
 136     sys.stderr.write('%u calls, %.03f secs, %u calls/sec\n' % (parser.calls, duration, parser.calls/duration))
 137
 138
 139 if __name__ == '__main__':
 140     main()