gdb/testsuite/gdb.python/py-pp-registration.py

   1 # Copyright (C) 2010-2017 Free Software Foundation, Inc.
   2
   3 # This program is free software; you can redistribute it and/or modify
   4 # it under the terms of the GNU General Public License as published by
   5 # the Free Software Foundation; either version 3 of the License, or
   6 # (at your option) any later version.
   7 #
   8 # This program is distributed in the hope that it will be useful,
   9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 # GNU General Public License for more details.
  12 #
  13 # You should have received a copy of the GNU General Public License
  14 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  15
  16 # This file is part of the GDB testsuite.  It tests python pretty
  17 # printer registration.
  18
  19 import re
  20 import gdb.types
  21 import gdb.printing
  22
  23
  24 def lookup_function_lookup_test(val):
  25     class PrintFunctionLookup(object):
  26         def __init__(self, val):
  27             self.val = val
  28
  29         def to_string(self):
  30             return ("x=<" + str(self.val["x"]) +
  31                     "> y=<" + str(self.val["y"]) + ">")
  32
  33     typename = gdb.types.get_basic_type(val.type).tag
  34     # Note: typename could be None.
  35     if typename == "function_lookup_test":
  36         return PrintFunctionLookup(val)
  37     return None
  38
  39
  40 class pp_s1 (object):
  41     def __init__(self, val):
  42         self.val = val
  43
  44     def to_string(self):
  45         a = self.val["a"]
  46         b = self.val["b"]
  47         return "s1 a=<" + str(self.val["a"]) + "> b=<" + str(self.val["b"]) + ">"
  48
  49
  50 class pp_s2 (object):
  51     def __init__(self, val):
  52         self.val = val
  53
  54     def to_string(self):
  55         a = self.val["a"]
  56         b = self.val["b"]
  57         return "s2 a=<" + str(self.val["a"]) + "> b=<" + str(self.val["b"]) + ">"
  58
  59
  60 def build_pretty_printer1():
  61     pp = gdb.printing.RegexpCollectionPrettyPrinter("pp-test")
  62
  63     pp.add_printer('struct s', '^struct s$', pp_s1)
  64     pp.add_printer('s', '^s$', pp_s1)
  65
  66     return pp
  67
  68
  69 def build_pretty_printer2():
  70     # This intentionally has the same name as build_pretty_printer1.
  71     # It is used to test the "replace" functionality of
  72     # register_pretty_printer.
  73     pp = gdb.printing.RegexpCollectionPrettyPrinter("pp-test")
  74
  75     pp.add_printer('struct s', '^struct s$', pp_s2)
  76     pp.add_printer('s', '^s$', pp_s2)
  77
  78     return pp
  79
  80 # Note: Registering the printers is done in the .exp file.