Override MessageCallback in GPUTestingEnvironment

DiligentGraphics · Sep 30, 2023 · b99cfd0 · b99cfd0
1 parent 94786b3
commit b99cfd0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/Tests/GPUTestFramework/include/GPUTestingEnvironment.hpp b/Tests/GPUTestFramework/include/GPUTestingEnvironment.hpp
@@ -51,6 +51,7 @@ class GPUTestingEnvironment : public TestingEnvironment
 {
 public:
     static GPUTestingEnvironment* Initialize(int argc, char** argv);
+    static GPUTestingEnvironment* InitializeVk();
 
     struct CreateInfo
     {
@@ -61,6 +62,8 @@ class GPUTestingEnvironment : public TestingEnvironment
         bool               EnableDeviceSimulation = false;
 
         DeviceFeatures Features{DEVICE_FEATURE_STATE_OPTIONAL};
+
+        DebugMessageCallbackType MessageCallback  = TestingEnvironment::MessageCallback;
     };
     GPUTestingEnvironment(const CreateInfo& CI, const SwapChainDesc& SCDesc);
 

diff --git a/Tests/GPUTestFramework/src/GPUTestingEnvironment.cpp b/Tests/GPUTestFramework/src/GPUTestingEnvironment.cpp
@@ -231,7 +231,7 @@ GPUTestingEnvironment::GPUTestingEnvironment(const CreateInfo& EnvCI, const Swap
 #    endif
 
             auto* pFactoryD3D11 = GetEngineFactoryD3D11();
-            pFactoryD3D11->SetMessageCallback(MessageCallback);
+            pFactoryD3D11->SetMessageCallback(EnvCI.MessageCallback);
 
             EngineD3D11CreateInfo EngineCI;
             EngineCI.GraphicsAPIVersion = Version{11, 0};
@@ -273,7 +273,7 @@ GPUTestingEnvironment::GPUTestingEnvironment(const CreateInfo& EnvCI, const Swap
             }
 #    endif
             auto* pFactoryD3D12 = GetEngineFactoryD3D12();
-            pFactoryD3D12->SetMessageCallback(MessageCallback);
+            pFactoryD3D12->SetMessageCallback(EnvCI.MessageCallback);
 
             if (!pFactoryD3D12->LoadD3D12())
             {
@@ -338,7 +338,7 @@ GPUTestingEnvironment::GPUTestingEnvironment(const CreateInfo& EnvCI, const Swap
             }
 #    endif
             auto* pFactoryOpenGL = GetEngineFactoryOpenGL();
-            pFactoryOpenGL->SetMessageCallback(MessageCallback);
+            pFactoryOpenGL->SetMessageCallback(EnvCI.MessageCallback);
             EnumerateAdapters(pFactoryOpenGL, Version{},
                               [](const GraphicsAdapterInfo& AdapterInfo, Uint32 AdapterId) {
                                   return std::vector<DisplayModeAttribs>{};
@@ -374,7 +374,7 @@ GPUTestingEnvironment::GPUTestingEnvironment(const CreateInfo& EnvCI, const Swap
 #    endif
 
             auto* pFactoryVk = GetEngineFactoryVk();
-            pFactoryVk->SetMessageCallback(MessageCallback);
+            pFactoryVk->SetMessageCallback(EnvCI.MessageCallback);
 
             if (EnvCI.EnableDeviceSimulation)
                 pFactoryVk->EnableDeviceSimulation();
@@ -421,7 +421,7 @@ GPUTestingEnvironment::GPUTestingEnvironment(const CreateInfo& EnvCI, const Swap
         case RENDER_DEVICE_TYPE_METAL:
         {
             auto* pFactoryMtl = GetEngineFactoryMtl();
-            pFactoryMtl->SetMessageCallback(MessageCallback);
+            pFactoryMtl->SetMessageCallback(EnvCI.MessageCallback);
 
             EnumerateAdapters(pFactoryMtl, Version{},
                               [](const GraphicsAdapterInfo& AdapterInfo, Uint32 AdapterId) {
@@ -564,7 +564,7 @@ GPUTestingEnvironment::GPUTestingEnvironment(const CreateInfo& EnvCI, const Swap
 #    else
         m_ArchiverFactory = Diligent::GetArchiverFactory();
 #    endif
-        m_ArchiverFactory->SetMessageCallback(MessageCallback);
+        m_ArchiverFactory->SetMessageCallback(EnvCI.MessageCallback);
     }
 #endif
 }