]> git.cworth.org Git - apitrace/blob - retrace/d3d9state_images.cpp
scripts: Don't install bogus sample.py
[apitrace] / retrace / d3d9state_images.cpp
1 /**************************************************************************
2  *
3  * Copyright 2011 Jose Fonseca
4  * All Rights Reserved.
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and associated documentation files (the "Software"), to deal
8  * in the Software without restriction, including without limitation the rights
9  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10  * copies of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in
14  * all copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
22  * THE SOFTWARE.
23  *
24  **************************************************************************/
25
26
27 #include <assert.h>
28
29 #include "image.hpp"
30 #include "json.hpp"
31 #include "d3d9imports.hpp"
32 #include "d3dstate.hpp"
33
34
35 namespace d3dstate {
36
37
38 static image::Image *
39 getRenderTargetImage(IDirect3DDevice9 *pDevice,
40                      IDirect3DSurface9 *pRenderTarget) {
41     image::Image *image = NULL;
42     D3DSURFACE_DESC Desc;
43     IDirect3DSurface9 *pStagingSurface = NULL;
44     D3DLOCKED_RECT LockedRect;
45     const unsigned char *src;
46     unsigned char *dst;
47     HRESULT hr;
48
49     if (!pRenderTarget) {
50         return NULL;
51     }
52
53     hr = pRenderTarget->GetDesc(&Desc);
54     assert(SUCCEEDED(hr));
55
56     if (Desc.Format != D3DFMT_X8R8G8B8 && Desc.Format != D3DFMT_A8R8G8B8) {
57         std::cerr << "warning: unsupported D3DFORMAT " << Desc.Format << "\n";
58         goto no_staging;
59     }
60
61     hr = pDevice->CreateOffscreenPlainSurface(Desc.Width, Desc.Height, Desc.Format, D3DPOOL_SYSTEMMEM, &pStagingSurface, NULL);
62     if (FAILED(hr)) {
63         goto no_staging;
64     }
65
66     hr = pDevice->GetRenderTargetData(pRenderTarget, pStagingSurface);
67     if (FAILED(hr)) {
68         goto no_rendertargetdata;
69     }
70
71     hr = pStagingSurface->LockRect(&LockedRect, NULL, D3DLOCK_READONLY);
72     if (FAILED(hr)) {
73         goto no_rendertargetdata;
74     }
75
76     image = new image::Image(Desc.Width, Desc.Height, 3, true);
77     if (!image) {
78         goto no_image;
79     }
80
81     dst = image->start();
82     src = (const unsigned char *)LockedRect.pBits;
83     for (unsigned y = 0; y < Desc.Height; ++y) {
84         for (unsigned x = 0; x < Desc.Width; ++x) {
85             dst[3*x + 0] = src[4*x + 2];
86             dst[3*x + 1] = src[4*x + 1];
87             dst[3*x + 2] = src[4*x + 0];
88         }
89         src += LockedRect.Pitch;
90         dst += image->stride();
91     }
92
93 no_image:
94     pStagingSurface->UnlockRect();
95 no_rendertargetdata:
96     pStagingSurface->Release();
97 no_staging:
98     return image;
99 }
100
101
102 image::Image *
103 getRenderTargetImage(IDirect3DDevice9 *pDevice) {
104     HRESULT hr;
105
106     IDirect3DSurface9 *pRenderTarget = NULL;
107     hr = pDevice->GetRenderTarget(0, &pRenderTarget);
108     if (FAILED(hr)) {
109         return NULL;
110     }
111     assert(pRenderTarget);
112
113     image::Image *image = NULL;
114     if (pRenderTarget) {
115         image = getRenderTargetImage(pDevice, pRenderTarget);
116         pRenderTarget->Release();
117     }
118
119     return image;
120 }
121
122
123 void
124 dumpFramebuffer(JSONWriter &json, IDirect3DDevice9 *pDevice)
125 {
126     HRESULT hr;
127
128     json.beginMember("framebuffer");
129     json.beginObject();
130
131     D3DCAPS9 Caps;
132     pDevice->GetDeviceCaps(&Caps);
133
134     for (UINT i = 0; i < Caps.NumSimultaneousRTs; ++i) {
135         IDirect3DSurface9 *pRenderTarget = NULL;
136         hr = pDevice->GetRenderTarget(i, &pRenderTarget);
137         if (FAILED(hr)) {
138             continue;
139         }
140
141         if (!pRenderTarget) {
142             continue;
143         }
144
145         image::Image *image;
146         image = getRenderTargetImage(pDevice, pRenderTarget);
147         if (image) {
148             char label[64];
149             _snprintf(label, sizeof label, "RENDER_TARGET_%u", i);
150             json.beginMember(label);
151             json.writeImage(image, "UNKNOWN");
152             json.endMember(); // RENDER_TARGET_*
153         }
154
155         pRenderTarget->Release();
156     }
157
158     json.endObject();
159     json.endMember(); // framebuffer
160 }
161
162
163 } /* namespace d3dstate */