Add get attribute TYPE_METHODS to record types

Author: philberty

docs/tree.rst

@@ -628,6 +628,10 @@ Additional attributes for various :py:class:`gcc.Type` subclasses:
 
       The fields of this type, as a list of :py:class:`gcc.FieldDecl` instances
 
+   .. py:attribute:: methods
+
+      The methods of this type, as a list of :py:class:`gcc.MethodType` instances
+
    You can look up C structures by looking within the top-level
    :py:class:`gcc.Block` within the current translation unit.  For example,
    given this sample C code:

generate-tree-c.py

@@ -474,6 +474,9 @@ def add_complex_getter(name, doc):
             add_simple_getter('fields',
                               'PyGcc_TreeListFromChain(TYPE_FIELDS(self->t.inner))',
                               "The fields of this type")
+            add_simple_getter('methods',
+                              'PyGcc_TreeListFromChain(TYPE_METHODS(self->t.inner))',
+                              "The methods of this type")
 
         if tree_type.SYM == 'IDENTIFIER_NODE':
             add_simple_getter('name',

tests/examples/cplusplus/methods/input.cc

@@ -0,0 +1,45 @@
+/*
+   Copyright 2014 Philip Herron <redbrain@gcc.gnu.org>
+   Copyright 2011 Red Hat, Inc.
+
+   This is free software: you can redistribute it and/or modify it
+   under the terms of the GNU General Public License as published by
+   the Free Software Foundation, either version 3 of the License, or
+   (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful, but
+   WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+   General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program.  If not, see
+   <http://www.gnu.org/licenses/>.
+*/
+
+class test {
+    int a, b;
+public:
+    test (int, int);
+    int somefunc (void);
+};
+
+test::test (int x, int y)
+{
+    a = x;
+    b = y;
+}
+
+int test::somefunc (void)
+{
+    return a * b;
+}
+
+
+/*
+  PEP-7
+Local variables:
+c-basic-offset: 4
+indent-tabs-mode: nil
+End:
+*/

tests/examples/cplusplus/methods/script.py

@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*-
+#   Copyright 2014 Philip Herron <redbrain@gcc.gnu.org>
+#   Copyright 2011 Red Hat, Inc.
+#
+#   This is free software: you can redistribute it and/or modify it
+#   under the terms of the GNU General Public License as published by
+#   the Free Software Foundation, either version 3 of the License, or
+#   (at your option) any later version.
+#
+#   This program is distributed in the hope that it will be useful, but
+#   WITHOUT ANY WARRANTY; without even the implied warranty of
+#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+#   General Public License for more details.
+#
+#   You should have received a copy of the GNU General Public License
+#   along with this program.  If not, see
+#   <http://www.gnu.org/licenses/>.
+
+import gcc
+from gccutils import pprint
+
+def gccPassHook(passname, _):
+    if passname.name == '*free_lang_data':
+        for i in gcc.get_translation_units ():
+            gns = gcc.get_global_namespace ()
+            for decl in gns.declarations:
+                if decl.is_builtin is False:
+                    pprint (decl.type)
+
+gcc.register_callback (gcc.PLUGIN_PASS_EXECUTION, gccPassHook)

tests/examples/cplusplus/methods/stdout.txt

@@ -0,0 +1,33 @@
+<RecordType
+  repr() = <gcc.RecordType object at 0x2be4738>
+  str() = 'struct test'
+  superclasses = (<type 'gcc.Type'>, <type 'gcc.Tree'>)
+  .attributes = {}
+  .fields = [gcc.FieldDecl('a'), gcc.FieldDecl('b'), gcc.TypeDecl('test')]
+  .methods = [gcc.FunctionDecl('test'), gcc.FunctionDecl('__base_ctor '), gcc.FunctionDecl('__comp_ctor '), gcc.FunctionDecl('test'), gcc.FunctionDecl('__base_ctor '), gcc.FunctionDecl('__comp_ctor '), gcc.FunctionDecl('somefunc')]
+  .name = <TypeDecl
+            repr() = gcc.TypeDecl('test')
+            str() = 'test'
+            superclasses = (<type 'gcc.Declaration'>, <type 'gcc.Tree'>)
+            .is_artificial = True
+            .is_builtin = False
+            .location = tests/examples/cplusplus/methods/input.cc:20
+            .name = 'test'
+            .pointer = <PointerType
+                         repr() = gcc.PointerType(dereference=<gcc.RecordType object at 0x2be4738>)
+                         str() = 'struct test *'
+                         superclasses = (<type 'gcc.Type'>, <type 'gcc.Tree'>)
+                         .attributes = {}
+                         .dereference = ... ('<gcc.RecordType object at 0x2be4738>')
+                         .name = None
+                         .sizeof = 8
+                         .str_no_uid = 'struct test *'
+                         .type = ... ('<gcc.RecordType object at 0x2be4738>')
+                       >
+            .str_no_uid = 'test'
+            .type = ... ('<gcc.RecordType object at 0x2be4738>')
+          >
+  .sizeof = 8
+  .str_no_uid = 'struct test'
+  .type = None
+>