X-Git-Url: https://git.cworth.org/git?a=blobdiff_plain;f=wrappers%2Ftrace.py;h=79156acafbd328086f6ea021c41bf5c64a8ce2d7;hb=81301939f025407ceb284a9dcd5d5a1f05d27b8f;hp=cbdc0c07122e2a9ee4390263f83746a68decb107;hpb=2b807458f983e01e6c5c49a6c77ee28a145a1b2d;p=apitrace

diff --git a/wrappers/trace.py b/wrappers/trace.py
index cbdc0c0..79156ac 100644
--- a/wrappers/trace.py
+++ b/wrappers/trace.py
@@ -426,8 +426,9 @@ class Tracer:
         self.header(api)
 
         # Includes
-        for header in api.headers:
-            print header
+        for module in api.modules:
+            for header in module.headers:
+                print header
         print
 
         # Generate the serializer functions
@@ -442,8 +443,10 @@ class Tracer:
         # Function wrappers
         self.interface = None
         self.base = None
-        map(self.traceFunctionDecl, api.functions)
-        map(self.traceFunctionImpl, api.functions)
+        for function in api.getAllFunctions():
+            self.traceFunctionDecl(function)
+        for function in api.getAllFunctions():
+            self.traceFunctionImpl(function)
         print
 
         self.footer(api)