Integrate Vulkan Screenshot Layer with Perfetto

CMakeLists.txt

@@ -31,7 +31,7 @@ if(Qt6_FOUND)
   get_target_property(QT_TARGET_TYPE Qt6::Core TYPE)
 endif()
 
-set(CMAKE_CXX_STANDARD 17)
+set(CMAKE_CXX_STANDARD 20)
 set(CMAKE_CXX_STANDARD_REQUIRED ON)
 set(CMAKE_CXX_EXTENSIONS OFF)
 set(CMAKE_POSITION_INDEPENDENT_CODE ON)

layersvt/CMakeLists.txt

@@ -111,6 +111,8 @@ if(BUILD_SCREENSHOT)
         screenshot.cpp
         screenshot_parsing.cpp
         screenshot_parsing.h
+        perfetto/perfetto.cc
+        perfetto/screenshots_perfetto_helpers.cpp
         vk_layer_table.cpp
         vk_layer_table.h
         screenshot_layer.md
@@ -202,12 +204,17 @@ foreach(layer ${TOOL_LAYERS})
         target_link_Libraries(${layer} PRIVATE log android atomic)
     endif()
 
+    if (WIN32)
+        target_link_Libraries(${layer} PRIVATE ws2_32)
+    endif()
+
     target_include_directories(${layer} PRIVATE .)
 
     if (MSVC)
         target_link_options(${layer} PRIVATE /DEF:${CMAKE_CURRENT_SOURCE_DIR}/${layer}.def)
     elseif(MINGW)
         target_sources(${layer} PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/${layer}.def)
+        target_compile_definitions(${layer} PRIVATE uid_t=int)
     endif()
 
     if (APPLE)

layersvt/perfetto/screenshots_perfetto_helpers.cpp

@@ -0,0 +1,66 @@
+#include "screenshots_perfetto_helpers.h"
+#include <atomic>
+#include <thread>
+#include <mutex>
+#include <condition_variable>
+#include <list>
+
+#ifdef ANDROID
+#include <android/log.h>
+#endif
+
+#include "screenshot_writer.h"
+
+namespace screenshot {
+struct ScreenshotQueueData;
+extern std::atomic_bool pauseCapture;
+extern std::mutex globalLock;
+extern std::condition_variable screenshotSavedCV;
+extern std::list<std::shared_ptr<ScreenshotQueueData>> screenshotsData;
+}  // namespace screenshot
+
+PERFETTO_DEFINE_CATEGORIES(perfetto::Category("VulkanScreenshots").SetDescription("Vulkan Layer Screenshots"));
+
+PERFETTO_TRACK_EVENT_STATIC_STORAGE();
+
+void InitializeScreenshotsPerfetto() {
+    std::atomic_store(&screenshot::pauseCapture, true);
+
+    perfetto::TracingInitArgs args;
+    // The backends determine where trace events are recorded.
+    // kSystemBackend connects to the system traced service (e.g. on Android).
+    args.backends = perfetto::kSystemBackend;
+
+    perfetto::Tracing::Initialize(args);
+    perfetto::TrackEvent::Register();
+
+    perfetto::DataSourceDescriptor dsd;
+    dsd.set_name("VulkanScreenshots");
+    ScreenshotDataSource::Register(dsd);
+}
+
+void ScreenshotDataSource::OnStart(const StartArgs&) {
+#ifdef ANDROID
+    __android_log_print(ANDROID_LOG_INFO, "screenshot", "ScreenshotDataSource::OnStart called");
+#endif
+    std::atomic_store(&screenshot::pauseCapture, false);
+}
+
+void ScreenshotDataSource::OnStop(const StopArgs& args) {
+#ifdef ANDROID
+    __android_log_print(ANDROID_LOG_INFO, "screenshot", "ScreenshotDataSource::OnStop called");
+#endif
+
+    std::atomic_store(&screenshot::pauseCapture, true);
+    auto async_stop_closure = args.HandleStopAsynchronously();
+
+    std::thread([stop_closure = std::move(async_stop_closure)]() {
+        {
+            std::unique_lock<std::mutex> lock(screenshot::globalLock);
+            screenshot::screenshotSavedCV.wait(lock, [] { return screenshot::screenshotsData.empty(); });
+        }
+
+        // Explicitly notify Perfetto that this data source is now ready to be destroyed.
+        stop_closure();
+    }).detach();
+}

layersvt/perfetto/screenshots_perfetto_helpers.h

@@ -0,0 +1,14 @@
+#ifndef LAYERSVT_SCREENSHOTS_PERFETTO_HELPERS_H
+#define LAYERSVT_SCREENSHOTS_PERFETTO_HELPERS_H
+
+#include "perfetto.h"
+
+class ScreenshotDataSource : public perfetto::DataSource<ScreenshotDataSource> {
+   public:
+    void OnStart(const StartArgs&) override;
+    void OnStop(const StopArgs& args) override;
+};
+
+void InitializeScreenshotsPerfetto();
+
+#endif  // LAYERSVT_SCREENSHOTS_PERFETTO_HELPERS_H