git.cworth.org Git - apitrace/blob - 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 */