X-Git-Url: https://git.cworth.org/git?a=blobdiff_plain;f=retrace%2Fd3dretrace.hpp;h=144a5c454b1dcdf80d24013e175e354433c17b6b;hb=eb216e6bd676535a5668bf8c389433c657f51aad;hp=c4e5bb4ae2bab0c0556edf08726e32c23f877b42;hpb=af366b25cb9db606ec37069119b428fb9be4a163;p=apitrace

diff --git a/retrace/d3dretrace.hpp b/retrace/d3dretrace.hpp
index c4e5bb4..144a5c4 100644
--- a/retrace/d3dretrace.hpp
+++ b/retrace/d3dretrace.hpp
@@ -30,24 +30,57 @@
 #include <windows.h>
 
 #include "retrace.hpp"
-
-
-struct IDirect3DDevice9;
-
-
-extern const char *g_szD3D9DllName;
+#include "d3dstate.hpp"
 
 
 namespace d3dretrace {
 
 
-extern IDirect3DDevice9 *pLastDirect3DDevice9;
-
-
 extern const retrace::Entry d3d9_callbacks[];
 extern const retrace::Entry d3d10_callbacks[];
 
 
+template< class Device >
+class D3DDumper : public retrace::Dumper {
+public:
+    Device *pLastDevice;
+
+    D3DDumper() :
+        pLastDevice(NULL)
+    {}
+
+    image::Image *
+    getSnapshot(void) {
+        if (!pLastDevice) {
+            return NULL;
+        }
+        return d3dstate::getRenderTargetImage(pLastDevice);
+    }
+
+    bool
+    dumpState(std::ostream &os) {
+        if (!pLastDevice) {
+            return false;
+        }
+        d3dstate::dumpDevice(os, pLastDevice);
+        return true;
+    }
+
+    inline void
+    bindDevice(Device *pDevice) {
+        pLastDevice = pDevice;
+        retrace::dumper = this;
+    }
+    
+    inline void
+    unbindDevice(Device *pDevice) {
+        if (pLastDevice == pDevice) {
+            pLastDevice = NULL;
+        }
+    }
+};
+
+
 HWND
 createWindow(int width, int height);