src/gallium/state_trackers/python/retrace/format.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 class Formatter:
  31     '''Plain formatter'''
  32
  33     def __init__(self, stream):
  34         self.stream = stream
  35
  36     def text(self, text):
  37         self.stream.write(text)
  38
  39     def newline(self):
  40         self.text('\n')
  41
  42     def function(self, name):
  43         self.text(name)
  44
  45     def variable(self, name):
  46         self.text(name)
  47
  48     def literal(self, value):
  49         self.text(str(value))
  50
  51     def address(self, addr):
  52         self.text(str(addr))
  53
  54
  55 class AnsiFormatter(Formatter):
  56     '''Formatter for plain-text files which outputs ANSI escape codes. See
  57     http://en.wikipedia.org/wiki/ANSI_escape_code for more information
  58     concerning ANSI escape codes.
  59     '''
  60
  61     _csi = '\33['
  62
  63     _normal = '0m'
  64     _bold = '1m'
  65     _italic = '3m'
  66     _red = '31m'
  67     _green = '32m'
  68     _blue = '34m'
  69
  70     def _escape(self, code):
  71         self.text(self._csi + code)
  72
  73     def function(self, name):
  74         self._escape(self._bold)
  75         Formatter.function(self, name)
  76         self._escape(self._normal)
  77
  78     def variable(self, name):
  79         self._escape(self._italic)
  80         Formatter.variable(self, name)
  81         self._escape(self._normal)
  82
  83     def literal(self, value):
  84         self._escape(self._blue)
  85         Formatter.literal(self, value)
  86         self._escape(self._normal)
  87
  88     def address(self, value):
  89         self._escape(self._green)
  90         Formatter.address(self, value)
  91         self._escape(self._normal)
  92
  93
  94
  95 def DefaultFormatter(stream):
  96     if stream.isatty():
  97         return AnsiFormatter(stream)
  98     else:
  99         return Formatter(stream)
 100