From 40b03ab0dca7d948051a40d68db7602a652b8351 Mon Sep 17 00:00:00 2001
From: dec05eba <dec05eba@protonmail.com>
Date: Sat, 8 Apr 2023 02:40:00 +0200
Subject: Enable use on amd/intel

---
 src/config.hpp |  4 ++--
 src/main.cpp   | 55 ++++++++++++++++++++++++++++++++++++++++++++-----------
 2 files changed, 46 insertions(+), 13 deletions(-)

(limited to 'src')

diff --git a/src/config.hpp b/src/config.hpp
index 275f9f9..7257a4a 100644
--- a/src/config.hpp
+++ b/src/config.hpp
@@ -138,8 +138,8 @@ static bool file_get_content(const char *filepath, std::string &file_content) {
 }
 
 struct StringView {
-    const char *str;
-    size_t size;
+    const char *str = nullptr;
+    size_t size = 0;
 
     bool operator == (const char *other) const {
         int len = strlen(other);
diff --git a/src/main.cpp b/src/main.cpp
index 4ce1f8f..d5b9313 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -1500,6 +1500,28 @@ static bool is_cuda_installed() {
     return lib != nullptr;
 }
 
+static bool is_program_installed(const StringView program_name) {
+    const char *path = getenv("PATH");
+    if(!path)
+        return false;
+
+    bool program_installed = false;
+    char full_program_path[PATH_MAX];
+    string_split_char(path, ':', [&](StringView line) -> bool {
+        snprintf(full_program_path, sizeof(full_program_path), "%.*s/%.*s", (int)line.size, line.str, (int)program_name.size, program_name.str);
+        if(access(full_program_path, F_OK) == 0) {
+            program_installed = true;
+            return false;
+        }
+        return true;
+    });
+    return program_installed;
+}
+
+static bool is_pkexec_installed() {
+    return is_program_installed({ "pkexec", 6 });
+}
+
 typedef gboolean (*KeyPressHandler)(GtkButton *button, gpointer userdata);
 static void keypress_toggle_recording(bool recording_state, GtkButton *record_button, KeyPressHandler keypress_handler, GtkApplication *app) {
     if(!gtk_widget_get_sensitive(GTK_WIDGET(record_button)))
@@ -2416,6 +2438,15 @@ static bool is_xwayland() {
     return xwayland_found;
 }
 
+static const char* gpu_vendor_to_name(gpu_vendor vendor) {
+    switch(vendor) {
+        case GPU_VENDOR_AMD:        return "AMD";
+        case GPU_VENDOR_INTEL:      return "Intel";
+        case GPU_VENDOR_NVIDIA:     return "NVIDIA";
+    }
+    return "";
+}
+
 static void activate(GtkApplication *app, gpointer userdata) {
     nvfbc_installed = is_nv_fbc_installed();
 
@@ -2438,16 +2469,6 @@ static void activate(GtkApplication *app, gpointer userdata) {
         return;
     }
 
-    // TODO: Remove once gpu screen recorder supports amd and intel properly
-    if(gpu_inf.vendor != GPU_VENDOR_NVIDIA) {
-        GtkWidget *dialog = gtk_message_dialog_new(NULL, GTK_DIALOG_MODAL, GTK_MESSAGE_ERROR, GTK_BUTTONS_OK,
-            "GPU Screen Recorder does currently only support NVIDIA GPUs. If you are using a laptop with a NVIDIA GPU then make sure you are using your NVIDIA GPU for all applications, everything. You can do this by switching to NVIDIA performance mode in nvidia settings and then rebooting. Make sure you have done this PROPERLY. You can verify this by using this command:\nglxinfo | grep 'client glx'\nand see if it says NVIDIA Corporation.\nNote: using prime-run won't fix this as it only runs one application on the NVIDIA GPU. All applications, everything needs to run on the NVIDIA GPU for screen capture to work.");
-        gtk_dialog_run(GTK_DIALOG(dialog));
-        gtk_widget_destroy(dialog);
-        g_application_quit(G_APPLICATION(app));
-        return;
-    }
-
     if(gpu_inf.vendor == GPU_VENDOR_NVIDIA) {
         if(!is_cuda_installed()) {
             GtkWidget *dialog = gtk_message_dialog_new(NULL, GTK_DIALOG_MODAL, GTK_MESSAGE_ERROR, GTK_BUTTONS_OK,
@@ -2466,11 +2487,23 @@ static void activate(GtkApplication *app, gpointer userdata) {
             g_application_quit(G_APPLICATION(app));
             return;
         }
+    } else {
+        if(!is_pkexec_installed()) {
+            GtkWidget *dialog = gtk_message_dialog_new(NULL, GTK_DIALOG_MODAL, GTK_MESSAGE_ERROR, GTK_BUTTONS_OK,
+                "pkexec needs to be installed to record a monitor on systems with AMD/Intel GPU. Please install polkit which provides pkexec.");
+            gtk_dialog_run(GTK_DIALOG(dialog));
+            gtk_widget_destroy(dialog);
+            g_application_quit(G_APPLICATION(app));
+            return;
+        }
     }
 
+    std::string window_title = "GPU Screen Recorder | Running on ";
+    window_title += gpu_vendor_to_name(gpu_inf.vendor);
+
     window = gtk_application_window_new(app);
     g_signal_connect(window, "destroy", G_CALLBACK(on_destroy_window), nullptr);
-    gtk_window_set_title(GTK_WINDOW(window), "GPU Screen Recorder");
+    gtk_window_set_title(GTK_WINDOW(window), window_title.c_str());
     gtk_window_set_resizable(GTK_WINDOW(window), false);
 
     select_window_userdata.app = app;
-- 
cgit v1.2.3