mirror of
https://repo.dec05eba.com/gpu-screen-recorder-ui
synced 2026-03-31 09:17:04 +09:00
Initial commit, showing ui above target window
This commit is contained in:
dec05eba
276
src/main.cpp
Normal file
276
src/main.cpp
Normal file
@@ -0,0 +1,276 @@
|
||||
#include "../include/gui/Button.hpp"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <errno.h>
|
||||
|
||||
#include <X11/Xlib.h>
|
||||
#include <X11/cursorfont.h>
|
||||
#include <X11/extensions/shape.h>
|
||||
|
||||
#include <mglpp/mglpp.hpp>
|
||||
#include <mglpp/graphics/Font.hpp>
|
||||
#include <mglpp/graphics/Text.hpp>
|
||||
#include <mglpp/graphics/Texture.hpp>
|
||||
#include <mglpp/graphics/Sprite.hpp>
|
||||
#include <mglpp/window/Window.hpp>
|
||||
#include <mglpp/window/Event.hpp>
|
||||
#include <mglpp/system/MemoryMappedFile.hpp>
|
||||
|
||||
#include <vector>
|
||||
|
||||
extern "C" {
|
||||
#include <mgl/mgl.h>
|
||||
}
|
||||
|
||||
static void usage() {
|
||||
fprintf(stderr, "usage: window-overlay <window>\n");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
static void startup_error(const char *msg) {
|
||||
fprintf(stderr, "Error: %s\n", msg);
|
||||
exit(1);
|
||||
}
|
||||
|
||||
static bool string_to_i64(const char *str, int64_t *result) {
|
||||
errno = 0;
|
||||
char *endptr = nullptr;
|
||||
int64_t res = strtoll(str, &endptr, 0);
|
||||
if(endptr == str || errno != 0)
|
||||
return false;
|
||||
|
||||
*result = res;
|
||||
return true;
|
||||
}
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
if(argc != 2)
|
||||
usage();
|
||||
|
||||
int64_t target_window;
|
||||
if(!string_to_i64(argv[1], &target_window)) {
|
||||
fprintf(stderr, "Error: invalid number '%s' was specific for window argument\n", argv[1]);
|
||||
return 1;
|
||||
}
|
||||
|
||||
mgl::Init init;
|
||||
Display *display = (Display*)mgl_get_context()->connection;
|
||||
|
||||
XWindowAttributes target_win_attr;
|
||||
memset(&target_win_attr, 0, sizeof(target_win_attr));
|
||||
if(!XGetWindowAttributes(display, target_window, &target_win_attr)) {
|
||||
fprintf(stderr, "Error: window argument %s is not a valid window\n", argv[1]);
|
||||
return 1;
|
||||
}
|
||||
|
||||
XSelectInput(display, target_window, StructureNotifyMask);
|
||||
|
||||
mgl::vec2i target_window_size = { target_win_attr.width, target_win_attr.height };
|
||||
|
||||
mgl::Window::CreateParams window_create_params;
|
||||
window_create_params.size = target_window_size;
|
||||
window_create_params.parent_window = target_window;
|
||||
window_create_params.hidden = true;
|
||||
|
||||
mgl::Window window;
|
||||
if(!window.create("mglpp", window_create_params))
|
||||
startup_error("failed to create window");
|
||||
|
||||
mgl::MemoryMappedFile title_font_file;
|
||||
if(!title_font_file.load("fonts/Orbitron-Bold.ttf", mgl::MemoryMappedFile::LoadOptions{true, false}))
|
||||
startup_error("failed to load file: fonts/Orbitron-Bold.ttf");
|
||||
|
||||
mgl::MemoryMappedFile font_file;
|
||||
if(!font_file.load("fonts/Orbitron-Regular.ttf", mgl::MemoryMappedFile::LoadOptions{true, false}))
|
||||
startup_error("failed to load file: fonts/Orbitron-Regular.ttf");
|
||||
|
||||
mgl::Font title_font;
|
||||
if(!title_font.load_from_file(title_font_file, 32))
|
||||
startup_error("failed to load font: fonts/Orbitron-Bold.ttf");
|
||||
|
||||
mgl::Font font;
|
||||
if(!font.load_from_file(font_file, 20))
|
||||
startup_error("failed to load font: fonts/Orbitron-Regular.ttf");
|
||||
|
||||
mgl::Texture replay_button_texture;
|
||||
if(!replay_button_texture.load_from_file("images/replay.png"))
|
||||
startup_error("failed to load texture: images/replay.png");
|
||||
|
||||
mgl::Texture record_button_texture;
|
||||
if(!record_button_texture.load_from_file("images/record.png"))
|
||||
startup_error("failed to load texture: images/record.png");
|
||||
|
||||
mgl::Texture stream_button_texture;
|
||||
if(!stream_button_texture.load_from_file("images/stream.png"))
|
||||
startup_error("failed to load texture: images/stream.png");
|
||||
|
||||
struct MainButton {
|
||||
mgl::Text title;
|
||||
mgl::Text description;
|
||||
mgl::Sprite icon;
|
||||
gsr::Button button;
|
||||
};
|
||||
|
||||
const char *titles[] = {
|
||||
"Instant Replay",
|
||||
"Record",
|
||||
"Livestream"
|
||||
};
|
||||
|
||||
const char *descriptions[] = {
|
||||
"Off",
|
||||
"Not recording",
|
||||
"Not streaming"
|
||||
};
|
||||
|
||||
mgl::Texture *textures[] = {
|
||||
&replay_button_texture,
|
||||
&record_button_texture,
|
||||
&stream_button_texture
|
||||
};
|
||||
|
||||
std::vector<MainButton> main_buttons;
|
||||
std::vector<XRectangle> shapes;
|
||||
|
||||
for(int i = 0; i < 3; ++i) {
|
||||
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);
|
||||
description.set_color(mgl::Color(255, 255, 255, 150));
|
||||
|
||||
mgl::Sprite sprite(textures[i]);
|
||||
gsr::Button button({425, 300});
|
||||
|
||||
MainButton main_button = {
|
||||
std::move(title),
|
||||
std::move(description),
|
||||
std::move(sprite),
|
||||
std::move(button)
|
||||
};
|
||||
|
||||
main_buttons.push_back(std::move(main_button));
|
||||
shapes.push_back({});
|
||||
}
|
||||
|
||||
// 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 };
|
||||
|
||||
XGCValues xgcv;
|
||||
xgcv.foreground = WhitePixel(display, DefaultScreen(display));
|
||||
xgcv.line_width = 1;
|
||||
xgcv.line_style = LineSolid;
|
||||
|
||||
Pixmap shape_pixmap = None;
|
||||
GC shape_gc = None;
|
||||
|
||||
auto update_overlay_shape = [&]() {
|
||||
const int main_button_margin = 20;
|
||||
const mgl::vec2i main_buttons_start_pos = target_window_size/2 - overlay_desired_size/2;
|
||||
mgl::vec2i main_button_pos = main_buttons_start_pos;
|
||||
|
||||
for(size_t i = 0; i < main_buttons.size(); ++i) {
|
||||
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,
|
||||
main_button_pos.y + main_button_margin).floor());
|
||||
|
||||
main_buttons[i].description.set_position(
|
||||
mgl::vec2f(
|
||||
main_button_pos.x + per_button_width * 0.5f - main_buttons[i].description.get_bounds().size.x * 0.5f,
|
||||
main_button_pos.y + overlay_desired_size.y - main_buttons[i].description.get_bounds().size.y - main_button_margin).floor());
|
||||
|
||||
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_buttons[i].button.set_position(main_button_pos.to_vec2f());
|
||||
shapes[i] = {
|
||||
(short)main_button_pos.x, (short)main_button_pos.y, (unsigned short)per_button_width, (unsigned short)overlay_desired_size.y
|
||||
};
|
||||
main_button_pos.x += per_button_width;
|
||||
}
|
||||
|
||||
const mgl::vec2i settings_button_size(128, 128);
|
||||
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
|
||||
};
|
||||
|
||||
if(shape_pixmap) {
|
||||
XFreePixmap(display, shape_pixmap);
|
||||
shape_pixmap = None;
|
||||
}
|
||||
|
||||
if(shape_gc) {
|
||||
XFreeGC(display, shape_gc);
|
||||
shape_gc = None;
|
||||
}
|
||||
|
||||
shape_pixmap = XCreatePixmap(display, window.get_system_handle(), target_window_size.x, target_window_size.y, 1);
|
||||
if(!shape_pixmap)
|
||||
fprintf(stderr, "Error: failed to create shape pixmap\n");
|
||||
|
||||
shape_gc = XCreateGC(display, shape_pixmap, 0, &xgcv);
|
||||
|
||||
XSetForeground(display, shape_gc, 0);
|
||||
XFillRectangle(display, shape_pixmap, shape_gc, 0, 0, target_window_size.x, target_window_size.y);
|
||||
|
||||
XSetForeground(display, shape_gc, 1);
|
||||
XDrawRectangles(display, shape_pixmap, shape_gc, shapes.data(), shapes.size());
|
||||
XFillRectangles(display, shape_pixmap, shape_gc, shapes.data(), shapes.size());
|
||||
|
||||
XShapeCombineMask(display, window.get_system_handle(), ShapeBounding, 0, 0, shape_pixmap, ShapeSet);
|
||||
};
|
||||
|
||||
update_overlay_shape();
|
||||
window.set_visible(true);
|
||||
|
||||
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);
|
||||
XGrabKeyboard(display, window.get_system_handle(), True, GrabModeAsync, GrabModeAsync, CurrentTime);
|
||||
|
||||
XEvent xev;
|
||||
mgl::Event event;
|
||||
|
||||
while(window.is_open()) {
|
||||
if(XCheckTypedWindowEvent(display, target_window, ConfigureNotify, &xev) && (xev.xconfigure.width != target_window_size.x || xev.xconfigure.height != target_window_size.y)) {
|
||||
while(XCheckTypedWindowEvent(display, target_window, ConfigureNotify, &xev)) {}
|
||||
target_window_size.x = xev.xconfigure.width;
|
||||
target_window_size.y = xev.xconfigure.height;
|
||||
window.set_size(target_window_size);
|
||||
update_overlay_shape();
|
||||
}
|
||||
|
||||
if(window.poll_event(event)) {
|
||||
for(auto &main_button : main_buttons) {
|
||||
main_button.button.on_event(event, window);
|
||||
}
|
||||
|
||||
if(event.type == mgl::Event::KeyPressed) {
|
||||
if(event.key.code == mgl::Keyboard::Escape) {
|
||||
window.close();
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
window.clear(mgl::Color(37, 43, 47, 255));
|
||||
for(auto &main_button : main_buttons) {
|
||||
main_button.button.draw(window);
|
||||
window.draw(main_button.icon);
|
||||
window.draw(main_button.title);
|
||||
window.draw(main_button.description);
|
||||
}
|
||||
window.display();
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
Reference in New Issue
Block a user