X-Git-Url: https://git.cworth.org/git?a=blobdiff_plain;f=wrappers%2Fddrawtrace.py;h=94e39e2d126e3c2d18db984af49a7f946d6c4a58;hb=48c661ea6c0f2bd9b76a3385cd946b7d07bc9b5f;hp=c95a018886f4e9186e708eb7cfd8bb951d9130e7;hpb=bcb26b220e51175b67d524d26d735865ef6cedb4;p=apitrace

diff --git a/wrappers/ddrawtrace.py b/wrappers/ddrawtrace.py
index c95a018..94e39e2 100644
--- a/wrappers/ddrawtrace.py
+++ b/wrappers/ddrawtrace.py
@@ -25,19 +25,10 @@
 
 
 from dlltrace import DllTracer
+from specs.stdapi import API
 from specs.d3d import ddraw, interfaces
 
 
-class DDrawTracer(DllTracer):
-
-    def traceFunctionImplBody(self, function):
-        if function.name in ('AcquireDDThreadLock', 'ReleaseDDThreadLock'):
-            self.invokeFunction(function)
-            return
-
-        DllTracer.traceFunctionImplBody(self, function)
-
-
 if __name__ == '__main__':
     print '#define INITGUID'
     print '#include <windows.h>'
@@ -66,5 +57,8 @@ if __name__ == '__main__':
     print '#include "trace_writer_local.hpp"'
     print '#include "os.hpp"'
     print
-    tracer = DDrawTracer('ddraw.dll')
-    tracer.trace_api(ddraw)
+
+    api = API()
+    api.addModule(ddraw)
+    tracer = DllTracer()
+    tracer.traceApi(api)