progs/gallium/python/retrace/model.py

   1 #!/usr/bin/env python
   2 ##########################################################################
   3 #
   4 # Copyright 2008 Tungsten Graphics, Inc., Cedar Park, Texas.
   5 # All Rights Reserved.
   6 #
   7 # Permission is hereby granted, free of charge, to any person obtaining a
   8 # copy of this software and associated documentation files (the
   9 # "Software"), to deal in the Software without restriction, including
  10 # without limitation the rights to use, copy, modify, merge, publish,
  11 # distribute, sub license, and/or sell copies of the Software, and to
  12 # permit persons to whom the Software is furnished to do so, subject to
  13 # the following conditions:
  14 #
  15 # The above copyright notice and this permission notice (including the
  16 # next paragraph) shall be included in all copies or substantial portions
  17 # of the Software.
  18 #
  19 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  20 # OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  21 # MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  22 # IN NO EVENT SHALL TUNGSTEN GRAPHICS AND/OR ITS SUPPLIERS BE LIABLE FOR
  23 # ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  24 # TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  25 # SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  26 #
  27 ##########################################################################
  28
  29
  30 '''Trace data model.'''
  31
  32
  33 import sys
  34 import string
  35 import format
  36
  37 try:
  38     from cStringIO import StringIO
  39 except ImportError:
  40     from StringIO import StringIO
  41
  42
  43 class Node:
  44
  45     def visit(self, visitor):
  46         raise NotImplementedError
  47
  48     def __str__(self):
  49         stream = StringIO()
  50         formatter = format.DefaultFormatter(stream)
  51         pretty_printer = PrettyPrinter(formatter)
  52         self.visit(pretty_printer)
  53         return stream.getvalue()
  54
  55
  56 class Literal(Node):
  57
  58     def __init__(self, value):
  59         self.value = value
  60
  61     def visit(self, visitor):
  62         visitor.visit_literal(self)
  63
  64
  65 class NamedConstant(Node):
  66
  67     def __init__(self, name):
  68         self.name = name
  69
  70     def visit(self, visitor):
  71         visitor.visit_named_constant(self)
  72
  73
  74 class Array(Node):
  75
  76     def __init__(self, elements):
  77         self.elements = elements
  78
  79     def visit(self, visitor):
  80         visitor.visit_array(self)
  81
  82
  83 class Struct(Node):
  84
  85     def __init__(self, name, members):
  86         self.name = name
  87         self.members = members
  88
  89     def visit(self, visitor):
  90         visitor.visit_struct(self)
  91
  92
  93 class Pointer(Node):
  94
  95     def __init__(self, address):
  96         self.address = address
  97
  98     def visit(self, visitor):
  99         visitor.visit_pointer(self)
 100
 101
 102 class Call:
 103
 104     def __init__(self, no, klass, method, args, ret):
 105         self.no = no
 106         self.klass = klass
 107         self.method = method
 108         self.args = args
 109         self.ret = ret
 110
 111     def visit(self, visitor):
 112         visitor.visit_call(self)
 113
 114
 115 class Trace:
 116
 117     def __init__(self, calls):
 118         self.calls = calls
 119
 120     def visit(self, visitor):
 121         visitor.visit_trace(self)
 122
 123
 124 class Visitor:
 125
 126     def visit_literal(self, node):
 127         raise NotImplementedError
 128
 129     def visit_named_constant(self, node):
 130         raise NotImplementedError
 131
 132     def visit_array(self, node):
 133         raise NotImplementedError
 134
 135     def visit_struct(self, node):
 136         raise NotImplementedError
 137
 138     def visit_pointer(self, node):
 139         raise NotImplementedError
 140
 141     def visit_call(self, node):
 142         raise NotImplementedError
 143
 144     def visit_trace(self, node):
 145         raise NotImplementedError
 146
 147
 148 class PrettyPrinter:
 149
 150     def __init__(self, formatter):
 151         self.formatter = formatter
 152
 153     def visit_literal(self, node):
 154         if isinstance(node.value, basestring):
 155             if len(node.value) >= 4096 or node.value.strip(string.printable):
 156                 self.formatter.text('...')
 157                 return
 158
 159             self.formatter.literal('"' + node.value + '"')
 160             return
 161
 162         self.formatter.literal(repr(node.value))
 163
 164     def visit_named_constant(self, node):
 165         self.formatter.literal(node.name)
 166
 167     def visit_array(self, node):
 168         self.formatter.text('{')
 169         sep = ''
 170         for value in node.elements:
 171             self.formatter.text(sep)
 172             value.visit(self)
 173             sep = ', '
 174         self.formatter.text('}')
 175
 176     def visit_struct(self, node):
 177         self.formatter.text('{')
 178         sep = ''
 179         for name, value in node.members:
 180             self.formatter.text(sep)
 181             self.formatter.variable(name)
 182             self.formatter.text(' = ')
 183             value.visit(self)
 184             sep = ', '
 185         self.formatter.text('}')
 186
 187     def visit_pointer(self, node):
 188         self.formatter.address(node.address)
 189
 190     def visit_call(self, node):
 191         self.formatter.text('%s ' % node.no)
 192         if node.klass is not None:
 193             self.formatter.function(node.klass + '::' + node.method)
 194         else:
 195             self.formatter.function(node.method)
 196         self.formatter.text('(')
 197         sep = ''
 198         for name, value in node.args:
 199             self.formatter.text(sep)
 200             self.formatter.variable(name)
 201             self.formatter.text(' = ')
 202             value.visit(self)
 203             sep = ', '
 204         self.formatter.text(')')
 205         if node.ret is not None:
 206             self.formatter.text(' = ')
 207             node.ret.visit(self)
 208
 209     def visit_trace(self, node):
 210         for call in node.calls:
 211             call.visit(self)
 212             self.formatter.newline()
 213