X-Git-Url: https://git.cworth.org/git?a=blobdiff_plain;f=retrace%2Fd3d11state_images.cpp;h=64169d41917c23aaa2c196c49124ad4a39b26008;hb=48a92b99752143a34a6b5f6b94bad183564cecab;hp=9c6ff87d1835b45640b7b6d2adbc96f10b927484;hpb=77373c35010d89e5ab3f030e4c60f8f6e3fdbe82;p=apitrace diff --git a/retrace/d3d11state_images.cpp b/retrace/d3d11state_images.cpp index 9c6ff87..64169d4 100644 --- a/retrace/d3d11state_images.cpp +++ b/retrace/d3d11state_images.cpp @@ -29,8 +29,11 @@ #include #include -#include "image.hpp" +#include "os.hpp" +#include "json.hpp" #include "d3d11imports.hpp" +#include "d3d10state.hpp" +#include "dxgistate.hpp" namespace d3dstate { @@ -139,41 +142,126 @@ stageResource(ID3D11DeviceContext *pDeviceContext, return hr; } -image::Image * -getRenderTargetImage(ID3D11DeviceContext *pDevice) { +static image::Image * +getSubResourceImage(ID3D11DeviceContext *pDevice, + ID3D11Resource *pResource, + DXGI_FORMAT Format, + UINT MipSlice) +{ image::Image *image = NULL; - ID3D11RenderTargetView *pRenderTargetView = NULL; - D3D11_RENDER_TARGET_VIEW_DESC Desc; - ID3D11Resource *pResource = NULL; ID3D11Resource *pStagingResource = NULL; UINT Width, Height, Depth; - UINT MipSlice; - UINT Subresource; - D3D11_MAPPED_SUBRESOURCE MappedSubresource; + UINT SubResource = MipSlice; + D3D11_MAPPED_SUBRESOURCE MappedSubResource; HRESULT hr; - const unsigned char *src; - unsigned char *dst; - pDevice->OMGetRenderTargets(1, &pRenderTargetView, NULL); - if (!pRenderTargetView) { - goto no_rendertarget; + if (!pResource) { + return NULL; } - pRenderTargetView->GetResource(&pResource); - assert(pResource); - - pRenderTargetView->GetDesc(&Desc); - if (Desc.Format != DXGI_FORMAT_R8G8B8A8_UNORM && - Desc.Format != DXGI_FORMAT_R32G32B32A32_FLOAT) { - std::cerr << "warning: unsupported DXGI format " << Desc.Format << "\n"; + hr = stageResource(pDevice, pResource, &pStagingResource, &Width, &Height, &Depth); + if (FAILED(hr)) { goto no_staging; } - hr = stageResource(pDevice, pResource, &pStagingResource, &Width, &Height, &Depth); + Width = std::max(Width >> MipSlice, 1U); + Height = std::max(Height >> MipSlice, 1U); + Depth = std::max(Depth >> MipSlice, 1U); + + hr = pDevice->Map(pStagingResource, SubResource, D3D11_MAP_READ, 0, &MappedSubResource); if (FAILED(hr)) { - goto no_staging; + goto no_map; } + image = ConvertImage(Format, + MappedSubResource.pData, + MappedSubResource.RowPitch, + Width, Height); + + pDevice->Unmap(pStagingResource, SubResource); +no_map: + if (pStagingResource) { + pStagingResource->Release(); + } +no_staging: + if (pResource) { + pResource->Release(); + } + return image; +} + + +static image::Image * +getShaderResourceViewImage(ID3D11DeviceContext *pDevice, + ID3D11ShaderResourceView *pShaderResourceView) { + D3D11_SHADER_RESOURCE_VIEW_DESC Desc; + ID3D11Resource *pResource = NULL; + UINT MipSlice; + + if (!pShaderResourceView) { + return NULL; + } + + pShaderResourceView->GetResource(&pResource); + assert(pResource); + + pShaderResourceView->GetDesc(&Desc); + + // TODO: Take the slice in consideration + switch (Desc.ViewDimension) { + case D3D11_SRV_DIMENSION_BUFFER: + MipSlice = 0; + break; + case D3D11_SRV_DIMENSION_TEXTURE1D: + MipSlice = Desc.Texture1D.MostDetailedMip; + break; + case D3D11_SRV_DIMENSION_TEXTURE1DARRAY: + MipSlice = Desc.Texture1DArray.MostDetailedMip; + break; + case D3D11_SRV_DIMENSION_TEXTURE2D: + MipSlice = Desc.Texture2D.MostDetailedMip; + break; + case D3D11_SRV_DIMENSION_TEXTURE2DARRAY: + MipSlice = Desc.Texture2DArray.MostDetailedMip; + break; + case D3D11_SRV_DIMENSION_TEXTURE2DMS: + MipSlice = 0; + break; + case D3D11_SRV_DIMENSION_TEXTURE2DMSARRAY: + MipSlice = 0; + break; + case D3D11_SRV_DIMENSION_TEXTURE3D: + MipSlice = Desc.Texture3D.MostDetailedMip; + break; + case D3D11_SRV_DIMENSION_TEXTURECUBE: + MipSlice = Desc.TextureCube.MostDetailedMip; + break; + case D3D11_SRV_DIMENSION_UNKNOWN: + default: + assert(0); + return NULL; + } + + return getSubResourceImage(pDevice, pResource, Desc.Format, MipSlice); +} + + +static image::Image * +getRenderTargetViewImage(ID3D11DeviceContext *pDevice, + ID3D11RenderTargetView *pRenderTargetView) { + D3D11_RENDER_TARGET_VIEW_DESC Desc; + ID3D11Resource *pResource = NULL; + UINT MipSlice; + + if (!pRenderTargetView) { + return NULL; + } + + pRenderTargetView->GetResource(&pResource); + assert(pResource); + + pRenderTargetView->GetDesc(&Desc); + // TODO: Take the slice in consideration switch (Desc.ViewDimension) { case D3D11_RTV_DIMENSION_BUFFER: @@ -201,63 +289,154 @@ getRenderTargetImage(ID3D11DeviceContext *pDevice) { case D3D11_RTV_DIMENSION_TEXTURE3D: MipSlice = Desc.Texture3D.MipSlice; break; - case D3D11_SRV_DIMENSION_UNKNOWN: + case D3D11_RTV_DIMENSION_UNKNOWN: default: assert(0); - goto no_map; + return NULL; } - Subresource = MipSlice; - Width = std::max(Width >> MipSlice, 1U); - Height = std::max(Height >> MipSlice, 1U); - Depth = std::max(Depth >> MipSlice, 1U); + return getSubResourceImage(pDevice, pResource, Desc.Format, MipSlice); +} - hr = pDevice->Map(pStagingResource, Subresource, D3D11_MAP_READ, 0, &MappedSubresource); - if (FAILED(hr)) { - goto no_map; + +static image::Image * +getDepthStencilViewImage(ID3D11DeviceContext *pDevice, + ID3D11DepthStencilView *pDepthStencilView) { + D3D11_DEPTH_STENCIL_VIEW_DESC Desc; + ID3D11Resource *pResource = NULL; + UINT MipSlice; + + if (!pDepthStencilView) { + return NULL; } - image = new image::Image(Width, Height, 4, true); - if (!image) { - goto no_image; + pDepthStencilView->GetResource(&pResource); + assert(pResource); + + pDepthStencilView->GetDesc(&Desc); + + // TODO: Take the slice in consideration + switch (Desc.ViewDimension) { + case D3D11_DSV_DIMENSION_TEXTURE1D: + MipSlice = Desc.Texture1D.MipSlice; + break; + case D3D11_DSV_DIMENSION_TEXTURE1DARRAY: + MipSlice = Desc.Texture1DArray.MipSlice; + break; + case D3D11_DSV_DIMENSION_TEXTURE2D: + MipSlice = Desc.Texture2D.MipSlice; + break; + case D3D11_DSV_DIMENSION_TEXTURE2DARRAY: + MipSlice = Desc.Texture2DArray.MipSlice; + break; + case D3D11_DSV_DIMENSION_TEXTURE2DMS: + MipSlice = 0; + break; + case D3D11_DSV_DIMENSION_TEXTURE2DMSARRAY: + MipSlice = 0; + break; + case D3D11_DSV_DIMENSION_UNKNOWN: + default: + assert(0); + return NULL; } - dst = image->start(); - src = (const unsigned char *)MappedSubresource.pData; - for (unsigned y = 0; y < Height; ++y) { - if (Desc.Format == DXGI_FORMAT_R8G8B8A8_UNORM) { - memcpy(dst, src, Width * 4); - } else if (Desc.Format == DXGI_FORMAT_R32G32B32A32_FLOAT) { - float scale = 1.0f/255.0f; - for (unsigned x = 0; x < Width; ++x) { - dst[4*x + 0] = ((float *)src)[4*x + 0] * scale; - dst[4*x + 1] = ((float *)src)[4*x + 1] * scale; - dst[4*x + 2] = ((float *)src)[4*x + 2] * scale; - dst[4*x + 3] = ((float *)src)[4*x + 3] * scale; - } - } else { - assert(0); + return getSubResourceImage(pDevice, pResource, Desc.Format, MipSlice); +} + + +void +dumpTextures(JSONWriter &json, ID3D11DeviceContext *pDevice) +{ + json.beginMember("textures"); + json.beginObject(); + + ID3D11ShaderResourceView *pShaderResourceViews[D3D11_COMMONSHADER_SAMPLER_SLOT_COUNT]; + pDevice->PSGetShaderResources(0, ARRAYSIZE(pShaderResourceViews), pShaderResourceViews); + + for (UINT i = 0; i < ARRAYSIZE(pShaderResourceViews); ++i) { + if (!pShaderResourceViews[i]) { + continue; } - src += MappedSubresource.RowPitch; - dst += image->stride(); - } -no_image: - pDevice->Unmap(pStagingResource, Subresource); -no_map: - if (pStagingResource) { - pStagingResource->Release(); - } -no_staging: - if (pResource) { - pResource->Release(); + image::Image *image; + image = getShaderResourceViewImage(pDevice, pShaderResourceViews[i]); + if (image) { + char label[64]; + _snprintf(label, sizeof label, "PS_RESOURCE_%u", i); + json.beginMember(label); + json.writeImage(image, "UNKNOWN"); + json.endMember(); // PS_RESOURCE_* + } + + pShaderResourceViews[i]->Release(); } + + json.endObject(); + json.endMember(); // textures +} + + +image::Image * +getRenderTargetImage(ID3D11DeviceContext *pDevice) { + ID3D11RenderTargetView *pRenderTargetView = NULL; + pDevice->OMGetRenderTargets(1, &pRenderTargetView, NULL); + + image::Image *image = NULL; if (pRenderTargetView) { + image = getRenderTargetViewImage(pDevice, pRenderTargetView); pRenderTargetView->Release(); } -no_rendertarget: + return image; } +void +dumpFramebuffer(JSONWriter &json, ID3D11DeviceContext *pDevice) +{ + json.beginMember("framebuffer"); + json.beginObject(); + + ID3D11RenderTargetView *pRenderTargetViews[D3D11_SIMULTANEOUS_RENDER_TARGET_COUNT]; + ID3D11DepthStencilView *pDepthStencilView; + pDevice->OMGetRenderTargets(ARRAYSIZE(pRenderTargetViews), pRenderTargetViews, + &pDepthStencilView); + + for (UINT i = 0; i < ARRAYSIZE(pRenderTargetViews); ++i) { + if (!pRenderTargetViews[i]) { + continue; + } + + image::Image *image; + image = getRenderTargetViewImage(pDevice, pRenderTargetViews[i]); + if (image) { + char label[64]; + _snprintf(label, sizeof label, "RENDER_TARGET_%u", i); + json.beginMember(label); + json.writeImage(image, "UNKNOWN"); + json.endMember(); // RENDER_TARGET_* + } + + pRenderTargetViews[i]->Release(); + } + + if (pDepthStencilView) { + image::Image *image; + image = getDepthStencilViewImage(pDevice, pDepthStencilView); + if (image) { + json.beginMember("DEPTH_STENCIL"); + json.writeImage(image, "UNKNOWN"); + json.endMember(); + } + + pDepthStencilView->Release(); + + } + + json.endObject(); + json.endMember(); // framebuffer +} + + } /* namespace d3dstate */