调整capture

Author: wenye01

Lens/Lens.vcxproj

@@ -185,6 +185,7 @@ xcopy /Y "$(SolutionDir)\3rd\imgui\backends\imgui_impl_win32.cpp" "$(ProjectDir)
     <ClInclude Include="include\gui\UIManager.h" />
     <ClInclude Include="include\gui\DemoPanel.h" />
     <ClInclude Include="include\gui\DebugPanel.h" />
+    <ClInclude Include="include\gui\CapturePanel.h" />
   </ItemGroup>
   <ItemGroup>
     <ClCompile Include="src\graphics\Buffer.cpp" />
@@ -196,16 +197,16 @@ xcopy /Y "$(SolutionDir)\3rd\imgui\backends\imgui_impl_win32.cpp" "$(ProjectDir)
     <ClCompile Include="src\gui\UIManager.cpp" />
     <ClCompile Include="src\gui\DemoPanel.cpp" />
     <ClCompile Include="src\gui\DebugPanel.cpp" />
+    <ClCompile Include="src\gui\CapturePanel.cpp" />
     <ClCompile Include="src\LensPch.cpp">
       <PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">Create</PrecompiledHeader>
       <PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">Create</PrecompiledHeader>
       <PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">Create</PrecompiledHeader>
       <PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|x64'">Create</PrecompiledHeader>
     </ClCompile>
-    <ClCompile Include="src\main_wgc_test.cpp" />
+    <ClCompile Include="src\main.cpp" />
     <!-- <ClCompile Include="src\main.cpp" /> -->
     <ClCompile Include="src\Application.cpp" />
-    <ClCompile Include="src\WGCTestApp.cpp" />
     <!-- 以下为imgui引入需要编译的文件 -->
     <ClCompile Include="src\imgui\imgui.cpp">
       <PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">NotUsing</PrecompiledHeader>

Lens/imgui.ini

@@ -3,15 +3,23 @@ Pos=60,60
 Size=400,400
 
 [Window][Dear ImGui Demo]
-Pos=-1,0
+Pos=52,23
 Size=550,658
 Collapsed=1
 
 [Window][Demo Panel]
-Pos=60,60
+Pos=231,171
 Size=219,170
 
 [Window][Debug Info]
-Pos=60,60
-Size=226,219
+Pos=174,106
+Size=226,228
+
+[Window][Window Capture]
+Pos=618,239
+Size=1544,911
+
+[Window][Capture]
+Pos=436,239
+Size=1608,871
 

Lens/include/Application.h

@@ -1,7 +1,12 @@
 #pragma once
 #include <Windows.h>
 #include "graphics/GraphicsDevice.h"
+#include "graphics/Shader.h"
 #include "ImguiManager.h"
+#include "capturer/WGCCapturer.h"
+#include <memory>
+#include <d3dcompiler.h>
+#include <wrl/client.h>
 
 namespace lens
 {
@@ -19,8 +24,14 @@ namespace lens
 
         static LRESULT CALLBACK WindowProcProxy(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
 
+        // Public getter for capturer access from UI panels
+        capturer::WGCCapturer* GetCapturer() const { return m_capturer.get(); }
+
     private:
         bool CreateLenWindow(int width = 800, int height = 600);
+        bool CreateCaptureShaders();
+        bool CreateCaptureSampler();
+        bool CompileShader(const std::string& code, const char* entryPoint, const char* target, Microsoft::WRL::ComPtr<ID3DBlob>& blob);
 
         LRESULT CALLBACK WndProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
 
@@ -31,6 +42,13 @@ namespace lens
         graphics::GraphicsDevice* m_graphicsDevice;
         ImguiManager* m_imgui;
 
+        // WGC Capture related
+        std::unique_ptr<capturer::WGCCapturer> m_capturer;
+        graphics::Shader m_captureShader;
+        Microsoft::WRL::ComPtr<ID3D11SamplerState> m_captureSampler;
+        Microsoft::WRL::ComPtr<ID3D11Buffer> m_captureConstantBuffer;
+        std::shared_ptr<graphics::Texture> m_lastCapturedFrame;
+
         int width;
         int height;
         std::wstring m_className;

Lens/include/gui/CapturePanel.h

@@ -0,0 +1,29 @@
+#pragma once
+
+#include "UIPanel.h"
+#include "capturer/WGCCapturer.h"
+
+namespace lens
+{
+    class CapturePanel : public UIPanel
+    {
+    private:
+        bool m_visible = true;
+        capturer::WGCCapturer* m_capturer;
+        std::shared_ptr<lens::graphics::Texture> m_lastFrame; // Cache the latest frame
+
+    public:
+        CapturePanel();
+        virtual ~CapturePanel() = default;
+
+        void SetCapturer(capturer::WGCCapturer* capturer) { m_capturer = capturer; }
+
+        const char* GetName() const override { return "Capture"; }
+        bool IsVisible() const override { return m_visible; }
+        void SetVisible(bool visible) override { m_visible = visible; }
+        void Render() override;
+
+        void Initialize() override;
+        void Shutdown() override;
+    };
+}

Lens/include/gui/UIManager.h

@@ -8,17 +8,32 @@
 
 namespace lens
 {
+    // 前向声明
+    namespace capturer
+    {
+        class WGCCapturer;
+    }
+
     class UIManager
     {
     private:
         std::vector<std::unique_ptr<UIPanel>> m_panels;
         std::map<std::string, UIPanel*> m_panelMap;
         bool m_showMenu = true;
 
+        // 注册所有默认面板
+        void RegisterDefaultPanels();
+        void RegisterCorePanels();
+        void RegisterCapturePanels(capturer::WGCCapturer* capturer);
+
     public:
         UIManager() = default;
         ~UIManager() = default;
 
+        // 统一初始化所有面板（推荐使用）
+        void InitializeAllPanels(capturer::WGCCapturer* capturer = nullptr);
+
+        // 手动添加面板的方法
         template<typename T, typename... Args>
         T* AddPanel(Args&&... args)
         {