Schedule texture copy

Author: nums11

Samples/WinMLSamplesGallery/WinMLSamplesGalleryNative/D3D12Quad.cpp

@@ -43,6 +43,7 @@ void D3D12Quad::OnInit()
     LoadPipeline();
     LoadAssets();
     is_initialized = true;
+    copy_texture = false;
 }
 
 // Load the rendering pipeline dependencies.
@@ -532,6 +533,12 @@ void D3D12Quad::PopulateCommandList()
     auto render_to_present = CD3DX12_RESOURCE_BARRIER::Transition(m_renderTargets[m_frameIndex].Get(), D3D12_RESOURCE_STATE_RENDER_TARGET, D3D12_RESOURCE_STATE_PRESENT);
     m_commandList->ResourceBarrier(1, &render_to_present);
 
+    if (copy_texture)
+    {
+        CopyTextureIntoCurrentBuffer();
+        copy_texture = false;
+    }
+
     ThrowIfFailed(m_commandList->Close());
 }
 
@@ -850,3 +857,8 @@ ComPtr<ID3D12Resource> D3D12Quad::GetCurrentBuffer()
     return currentBuffer;
 }
 
+void D3D12Quad::ScheduleTextureCopy()
+{
+    copy_texture = true;
+}
+

Samples/WinMLSamplesGallery/WinMLSamplesGalleryNative/D3D12Quad.h

@@ -22,6 +22,7 @@ class D3D12Quad
     const WCHAR* GetTitle() const { return m_title.c_str(); }
 
     ComPtr<ID3D12Resource> GetCurrentBuffer();
+    void ScheduleTextureCopy();
 
     bool is_initialized = false;
 
@@ -102,5 +103,6 @@ class D3D12Quad
     // a resource dimension buffer that will be used by ORT for inference
     ComPtr<ID3D12Resource> currentBuffer;
     int imageBytesPerRow;
+    bool copy_texture;
 };
 

Samples/WinMLSamplesGallery/WinMLSamplesGalleryNative/DXResourceBinding.cpp

@@ -222,6 +222,8 @@ namespace winrt::WinMLSamplesGalleryNative::implementation
         // Classify the image using EfficientNet and return the results
         winrt::com_array<float> results = Eval(*inferenceSession, preprocessedInput);
 
+        sample.ScheduleTextureCopy();
+
         return results;
     }