X-Git-Url: https://git.cworth.org/git?a=blobdiff_plain;f=wrappers%2Ftrace.py;h=e35108b273920561b0c87a950dbd2a41606d2086;hb=b35ae0df401ff22e3b2dc9b6bfacf1e23baebde0;hp=bb0f675a547231f3e94d7370d31f3c6fe33cf3c6;hpb=bce31f64b76d8aa9a933f0f6a1a2d6ff95b63f33;p=apitrace

diff --git a/wrappers/trace.py b/wrappers/trace.py
index bb0f675..e35108b 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)