aboutsummaryrefslogtreecommitdiff
path: root/src/main.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/main.cpp')
-rw-r--r--src/main.cpp122
1 files changed, 107 insertions, 15 deletions
diff --git a/src/main.cpp b/src/main.cpp
index ad7f6d3..ac0b7dd 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -1,11 +1,14 @@
#include "../include/gui/Button.hpp"
#include "../include/window_texture.h"
+#include "../include/Process.hpp"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <libgen.h>
+#include <signal.h>
+#include <sys/wait.h>
#include <X11/Xlib.h>
#include <X11/cursorfont.h>
@@ -26,6 +29,15 @@ extern "C" {
#include <mgl/mgl.h>
}
+struct Config {
+ float scale = 1.0f;
+};
+
+static Config& get_config() {
+ static Config config;
+ return config;
+}
+
static void usage() {
fprintf(stderr, "usage: window-overlay <window>\n");
exit(1);
@@ -86,9 +98,10 @@ int main(int argc, char **argv) {
return 1;
}
- XSelectInput(display, target_window, VisibilityChangeMask | StructureNotifyMask);
+ XSelectInput(display, target_window, StructureNotifyMask | VisibilityChangeMask);
mgl::vec2i target_window_size = { target_win_attr.width, target_win_attr.height };
+ get_config().scale = std::min(1.0, (double)target_win_attr.width / 1920.0);
mgl::Window::CreateParams window_create_params;
window_create_params.size = target_window_size;
@@ -108,11 +121,11 @@ int main(int argc, char **argv) {
startup_error("failed to load file: fonts/Orbitron-Regular.ttf");
mgl::Font title_font;
- if(!title_font.load_from_file(title_font_file, 32))
+ if(!title_font.load_from_file(title_font_file, 32 * get_config().scale))
startup_error("failed to load font: fonts/Orbitron-Bold.ttf");
mgl::Font font;
- if(!font.load_from_file(font_file, 20))
+ if(!font.load_from_file(font_file, 20 * get_config().scale))
startup_error("failed to load font: fonts/Orbitron-Regular.ttf");
mgl::Texture replay_button_texture;
@@ -132,6 +145,7 @@ int main(int argc, char **argv) {
mgl::Text description;
mgl::Sprite icon;
gsr::Button button;
+ gsr::GsrMode mode;
};
const char *titles[] = {
@@ -140,12 +154,18 @@ int main(int argc, char **argv) {
"Livestream"
};
- const char *descriptions[] = {
+ const char *descriptions_off[] = {
"Off",
"Not recording",
"Not streaming"
};
+ const char *descriptions_on[] = {
+ "On",
+ "Recording",
+ "Streaming"
+ };
+
mgl::Texture *textures[] = {
&replay_button_texture,
&record_button_texture,
@@ -159,17 +179,19 @@ int main(int argc, char **argv) {
mgl::Text title(titles[i], {0.0f, 0.0f}, title_font);
title.set_color(mgl::Color(255, 255, 255));
- mgl::Text description(descriptions[i], {0.0f, 0.0f}, font);
+ mgl::Text description(descriptions_off[i], {0.0f, 0.0f}, font);
description.set_color(mgl::Color(255, 255, 255, 150));
mgl::Sprite sprite(textures[i]);
- gsr::Button button({425, 300});
+ sprite.set_scale(get_config().scale);
+ gsr::Button button(mgl::vec2f(425 * get_config().scale, 300 * get_config().scale).floor());
MainButton main_button = {
std::move(title),
std::move(description),
std::move(sprite),
- std::move(button)
+ std::move(button),
+ gsr::GsrMode::Unknown
};
main_buttons.push_back(std::move(main_button));
@@ -179,8 +201,59 @@ int main(int argc, char **argv) {
// Settings button
shapes.push_back({});
- const int per_button_width = 425;
- const mgl::vec2i overlay_desired_size = { per_button_width * (int)main_buttons.size(), 300 };
+ // Replay
+ main_buttons[0].button.on_click = [&]() {
+ /*
+ char window_to_record_str[32];
+ snprintf(window_to_record_str, sizeof(window_to_record_str), "%ld", target_window);
+
+ const char *args[] = {
+ "gpu-screen-recorder", "-w", window_to_record_str,
+ "-c", "mp4",
+ "-f", "60",
+ "-o", "/home/dec05eba/Videos/gpu-screen-recorder.mp4",
+ nullptr
+ };
+ gsr::exec_program_daemonized(args);
+ */
+ };
+ main_buttons[0].mode = gsr::GsrMode::Replay;
+
+ // TODO: Monitor /tmp/gpu-screen-recorder and update ui to match state
+
+ // Record
+ main_buttons[1].button.on_click = [&]() {
+ int gpu_screen_recorder_process = -1;
+ gsr::GsrMode gsr_mode = gsr::GsrMode::Unknown;
+ if(gsr::is_gpu_screen_recorder_running(gpu_screen_recorder_process, gsr_mode) && gpu_screen_recorder_process > 0) {
+ kill(gpu_screen_recorder_process, SIGINT);
+ int status;
+ if(waitpid(gpu_screen_recorder_process, &status, 0) == -1) {
+ perror("waitpid failed");
+ /* Ignore... */
+ }
+ exit(0);
+ }
+
+ char window_to_record_str[32];
+ snprintf(window_to_record_str, sizeof(window_to_record_str), "%ld", target_window);
+
+ const char *args[] = {
+ "gpu-screen-recorder", "-w", window_to_record_str,
+ "-c", "mp4",
+ "-f", "60",
+ "-o", "/home/dec05eba/Videos/gpu-screen-recorder.mp4",
+ nullptr
+ };
+ gsr::exec_program_daemonized(args);
+ exit(0);
+ };
+ main_buttons[1].mode = gsr::GsrMode::Record;
+
+ main_buttons[2].mode = gsr::GsrMode::Stream;
+
+ const int per_button_width = 425 * get_config().scale;
+ const mgl::vec2i overlay_desired_size(per_button_width * (int)main_buttons.size(), 300 * get_config().scale);
XGCValues xgcv;
xgcv.foreground = WhitePixel(display, DefaultScreen(display));
@@ -191,11 +264,25 @@ int main(int argc, char **argv) {
GC shape_gc = None;
auto update_overlay_shape = [&]() {
- const int main_button_margin = 20;
+ const int main_button_margin = 20 * get_config().scale;
const mgl::vec2i main_buttons_start_pos = target_window_size/2 - overlay_desired_size/2;
mgl::vec2i main_button_pos = main_buttons_start_pos;
+ int gpu_screen_recorder_process = -1;
+ gsr::GsrMode gsr_mode = gsr::GsrMode::Unknown;
+ gsr::is_gpu_screen_recorder_running(gpu_screen_recorder_process, gsr_mode);
+
for(size_t i = 0; i < main_buttons.size(); ++i) {
+ if(main_buttons[i].mode != gsr::GsrMode::Unknown && main_buttons[i].mode == gsr_mode) {
+ main_buttons[i].description.set_string(descriptions_on[i]);
+ main_buttons[i].description.set_color(mgl::Color(118, 185, 0));
+ main_buttons[i].icon.set_color(mgl::Color(118, 185, 0));
+ } else {
+ main_buttons[i].description.set_string(descriptions_off[i]);
+ main_buttons[i].description.set_color(mgl::Color(255, 255, 255));
+ main_buttons[i].icon.set_color(mgl::Color(255, 255, 255));
+ }
+
main_buttons[i].title.set_position(
mgl::vec2f(
main_button_pos.x + per_button_width * 0.5f - main_buttons[i].title.get_bounds().size.x * 0.5f,
@@ -208,8 +295,8 @@ int main(int argc, char **argv) {
main_buttons[i].icon.set_position(
mgl::vec2f(
- main_button_pos.x + per_button_width * 0.5f - main_buttons[i].icon.get_texture()->get_size().x * 0.5f,
- main_button_pos.y + overlay_desired_size.y * 0.5f - main_buttons[i].icon.get_texture()->get_size().y * 0.5f).floor());
+ main_button_pos.x + per_button_width * 0.5f - main_buttons[i].icon.get_texture()->get_size().x * main_buttons[i].icon.get_scale().x * 0.5f,
+ main_button_pos.y + overlay_desired_size.y * 0.5f - main_buttons[i].icon.get_texture()->get_size().y * main_buttons[i].icon.get_scale().y * 0.5f).floor());
main_buttons[i].button.set_position(main_button_pos.to_vec2f());
shapes[i] = {
@@ -218,7 +305,7 @@ int main(int argc, char **argv) {
main_button_pos.x += per_button_width;
}
- const mgl::vec2i settings_button_size(128, 128);
+ const mgl::vec2i settings_button_size(128 * get_config().scale, 128 * get_config().scale);
shapes[main_buttons.size()] = {
(short)(main_buttons_start_pos.x + overlay_desired_size.x), (short)(main_buttons_start_pos.y - settings_button_size.y),
(unsigned short)settings_button_size.x, (unsigned short)settings_button_size.y
@@ -256,13 +343,18 @@ int main(int argc, char **argv) {
Cursor default_cursor = XCreateFontCursor(display, XC_arrow);
// TODO: Retry if these fail
- XGrabPointer(display, window.get_system_handle(), True, ButtonPressMask|ButtonReleaseMask, GrabModeAsync, GrabModeAsync, None, default_cursor, CurrentTime);
+ XGrabPointer(display, window.get_system_handle(), True, ButtonPressMask | ButtonReleaseMask, GrabModeAsync, GrabModeAsync, None, default_cursor, CurrentTime);
XGrabKeyboard(display, window.get_system_handle(), True, GrabModeAsync, GrabModeAsync, CurrentTime);
XEvent xev;
mgl::Event event;
while(window.is_open()) {
+ if(XCheckTypedWindowEvent(display, target_window, DestroyNotify, &xev)) {
+ window.close();
+ break;
+ }
+
if(XCheckTypedWindowEvent(display, target_window, VisibilityNotify, &xev)) {
if(xev.xvisibility.state) {
@@ -290,7 +382,7 @@ int main(int argc, char **argv) {
}
}
- window.clear(mgl::Color(37, 43, 47, 255));
+ window.clear(mgl::Color(37, 43, 47));
for(auto &main_button : main_buttons) {
main_button.button.draw(window);
window.draw(main_button.icon);
generated by cgit v1.2.3 (git 2.25.1) at 2024-10-16 12:27:47 +0000