WebRtcXR/Plugin~/WebRTCPlugin/GraphicsDevice/D3D12/D3D12Texture2D.cpp

#include "pch.h"

#include "D3D12Constants.h"
#include "D3D12Texture2D.h"

// nonstandard extension used : class rvalue used as lvalue
#pragma clang diagnostic ignored "-Wlanguage-extension-token"

namespace unity
{
namespace webrtc
{

    //---------------------------------------------------------------------------------------------------------------------

    D3D12Texture2D::D3D12Texture2D(
        uint32_t w, uint32_t h, ID3D12Resource* nativeTex, HANDLE handle, ID3D11Texture2D* sharedTex)
        : ITexture2D(w, h)
        , m_nativeTexture(nativeTex)
        , m_sharedHandle(handle)
        , m_sharedTexture(sharedTex)
        , m_readbackResource(nullptr)
    {
    }

    //----------------------------------------------------------------------------------------------------------------------

    HRESULT D3D12Texture2D::CreateReadbackResource(ID3D12Device* device)
    {
        SAFE_RELEASE(m_readbackResource)

        D3D12_RESOURCE_DESC origDesc = m_nativeTexture->GetDesc();
        device->GetCopyableFootprints(
            &origDesc,
            0,
            1,
            0,
            &m_nativeTextureFootprint.Footprint,
            &m_nativeTextureFootprint.NumRows,
            &m_nativeTextureFootprint.RowSize,
            &m_nativeTextureFootprint.ResourceSize);

        // Create the readback buffer for the texture.
        D3D12_RESOURCE_DESC desc {};
        desc.Dimension = D3D12_RESOURCE_DIMENSION_BUFFER;
        desc.Alignment = 0;
        desc.Width = m_nativeTextureFootprint.ResourceSize;
        desc.Height = 1;
        desc.DepthOrArraySize = 1;
        desc.MipLevels = 1;
        desc.Format = DXGI_FORMAT_UNKNOWN;
        desc.SampleDesc.Count = 1;
        desc.SampleDesc.Quality = 0;
        desc.Layout = D3D12_TEXTURE_LAYOUT_ROW_MAJOR;
        desc.Flags = D3D12_RESOURCE_FLAG_NONE;

        const HRESULT hr = device->CreateCommittedResource(
            &D3D12_READBACK_HEAP_PROPS,
            D3D12_HEAP_FLAG_NONE,
            &desc,
            D3D12_RESOURCE_STATE_COPY_DEST,
            nullptr,
            IID_PPV_ARGS(&m_readbackResource));
        return hr;
    }
} // end namespace webrtc
} // end namespace unity