X-Git-Url: https://git.cworth.org/git?a=blobdiff_plain;f=wrappers%2Fddrawtrace.py;h=94e39e2d126e3c2d18db984af49a7f946d6c4a58;hb=HEAD;hp=ca33a3e899b44216ec3f85886d7c5ddcedf02ffb;hpb=81301939f025407ceb284a9dcd5d5a1f05d27b8f;p=apitrace

diff --git a/wrappers/ddrawtrace.py b/wrappers/ddrawtrace.py
index ca33a3e..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.traceModule(ddraw)
+
+    api = API()
+    api.addModule(ddraw)
+    tracer = DllTracer()
+    tracer.traceApi(api)