b9bc530ad2
This patch adds a very basic pretty-printing of the test status (passed or failed) to highlight failing tests even more: green for passed, and red for failed. The printing only uses ANSI it the target output is a tty and supports ANSI colours. Hence, any regression scripts that are outputting to files or sending e-mails etc should still be fine.
123 lines
4.1 KiB
Python
123 lines
4.1 KiB
Python
# Copyright (c) 2011 Advanced Micro Devices, Inc.
|
|
# All rights reserved.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without
|
|
# modification, are permitted provided that the following conditions are
|
|
# met: redistributions of source code must retain the above copyright
|
|
# notice, this list of conditions and the following disclaimer;
|
|
# redistributions in binary form must reproduce the above copyright
|
|
# notice, this list of conditions and the following disclaimer in the
|
|
# documentation and/or other materials provided with the distribution;
|
|
# neither the name of the copyright holders nor the names of its
|
|
# contributors may be used to endorse or promote products derived from
|
|
# this software without specific prior written permission.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
#
|
|
# Author: Steve Reinhardt
|
|
|
|
import sys
|
|
|
|
# Intended usage example:
|
|
#
|
|
# if force_colors:
|
|
# from m5.util.terminal import termcap
|
|
# elif no_colors:
|
|
# from m5.util.terminal import no_termcap as termcap
|
|
# else:
|
|
# from m5.util.terminal import tty_termcap as termcap
|
|
# print termcap.Blue + "This could be blue!" + termcap.Normal
|
|
|
|
# ANSI color names in index order
|
|
color_names = "Black Red Green Yellow Blue Magenta Cyan".split()
|
|
|
|
# Character attribute capabilities. Note that not all terminals
|
|
# support all of these capabilities, or support them
|
|
# differently/meaningfully. For example:
|
|
#
|
|
# - In PuTTY (with the default settings), Dim has no effect, Standout
|
|
# is the same as Reverse, and Blink does not blink but switches to a
|
|
# gray background.
|
|
#
|
|
# Please feel free to add information about other terminals here.
|
|
#
|
|
capability_map = {
|
|
'Bold': 'bold',
|
|
'Dim': 'dim',
|
|
'Blink': 'blink',
|
|
'Underline': 'smul',
|
|
'Reverse': 'rev',
|
|
'Standout': 'smso',
|
|
'Normal': 'sgr0'
|
|
}
|
|
|
|
capability_names = capability_map.keys()
|
|
|
|
def null_cap_string(s, *args):
|
|
return ''
|
|
|
|
try:
|
|
import curses
|
|
curses.setupterm()
|
|
def cap_string(s, *args):
|
|
cap = curses.tigetstr(s)
|
|
if cap:
|
|
return curses.tparm(cap, *args)
|
|
else:
|
|
return ''
|
|
except:
|
|
cap_string = null_cap_string
|
|
|
|
class ColorStrings(object):
|
|
def __init__(self, cap_string):
|
|
for i, c in enumerate(color_names):
|
|
setattr(self, c, cap_string('setaf', i))
|
|
for name, cap in capability_map.iteritems():
|
|
setattr(self, name, cap_string(cap))
|
|
|
|
termcap = ColorStrings(cap_string)
|
|
no_termcap = ColorStrings(null_cap_string)
|
|
|
|
if sys.stdout.isatty():
|
|
tty_termcap = termcap
|
|
else:
|
|
tty_termcap = no_termcap
|
|
|
|
def get_termcap(use_colors = None):
|
|
if use_colors:
|
|
return termcap
|
|
elif use_colors is None:
|
|
# option unspecified; default behavior is to use colors iff isatty
|
|
return tty_termcap
|
|
else:
|
|
return no_termcap
|
|
|
|
def test_termcap(obj):
|
|
for c_name in color_names:
|
|
c_str = getattr(obj, c_name)
|
|
print c_str + c_name + obj.Normal
|
|
for attr_name in capability_names:
|
|
if attr_name == 'Normal':
|
|
continue
|
|
attr_str = getattr(obj, attr_name)
|
|
print attr_str + c_str + attr_name + " " + c_name + obj.Normal
|
|
print obj.Bold + obj.Underline + \
|
|
c_name + "Bold Underline " + c + obj.Normal
|
|
|
|
if __name__ == '__main__':
|
|
print "=== termcap enabled ==="
|
|
test_termcap(termcap)
|
|
print termcap.Normal
|
|
print "=== termcap disabled ==="
|
|
test_termcap(no_termcap)
|