You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
551 lines
15 KiB
C++
551 lines
15 KiB
C++
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <errno.h>
|
|
#include <sys/sysinfo.h>
|
|
#include <wordexp.h>
|
|
#include "imgui.h"
|
|
#include <iostream>
|
|
#include <string>
|
|
#include <sstream>
|
|
#include <algorithm>
|
|
#include <cctype>
|
|
|
|
#include "overlay_params.h"
|
|
#include "overlay.h"
|
|
#include "config.h"
|
|
#include "string_utils.h"
|
|
|
|
#include "mesa/util/os_socket.h"
|
|
|
|
#ifdef HAVE_X11
|
|
#include <X11/keysym.h>
|
|
#include "loaders/loader_x11.h"
|
|
#endif
|
|
|
|
#ifdef HAVE_DBUS
|
|
#include "dbus_info.h"
|
|
#endif
|
|
|
|
static enum overlay_param_position
|
|
parse_position(const char *str)
|
|
{
|
|
if (!str || !strcmp(str, "top-left"))
|
|
return LAYER_POSITION_TOP_LEFT;
|
|
if (!strcmp(str, "top-right"))
|
|
return LAYER_POSITION_TOP_RIGHT;
|
|
if (!strcmp(str, "bottom-left"))
|
|
return LAYER_POSITION_BOTTOM_LEFT;
|
|
if (!strcmp(str, "bottom-right"))
|
|
return LAYER_POSITION_BOTTOM_RIGHT;
|
|
if (!strcmp(str, "top-center"))
|
|
return LAYER_POSITION_TOP_CENTER;
|
|
return LAYER_POSITION_TOP_LEFT;
|
|
}
|
|
|
|
static int
|
|
parse_control(const char *str)
|
|
{
|
|
int ret = os_socket_listen_abstract(str, 1);
|
|
if (ret < 0) {
|
|
fprintf(stderr, "ERROR: Couldn't create socket pipe at '%s'\n", str);
|
|
fprintf(stderr, "ERROR: '%s'\n", strerror(errno));
|
|
return ret;
|
|
}
|
|
|
|
os_socket_block(ret, false);
|
|
|
|
return ret;
|
|
}
|
|
|
|
static float
|
|
parse_float(const char *str)
|
|
{
|
|
float val = 0;
|
|
std::stringstream ss(str);
|
|
ss.imbue(std::locale::classic());
|
|
ss >> val;
|
|
return val;
|
|
}
|
|
|
|
#ifdef HAVE_X11
|
|
static std::vector<KeySym>
|
|
parse_string_to_keysym_vec(const char *str)
|
|
{
|
|
std::vector<KeySym> keys;
|
|
if(g_x11->IsLoaded())
|
|
{
|
|
std::stringstream keyStrings(str);
|
|
std::string ks;
|
|
while (std::getline(keyStrings, ks, '+')) {
|
|
trim(ks);
|
|
KeySym xk = g_x11->XStringToKeysym(ks.c_str());
|
|
if (xk)
|
|
keys.push_back(xk);
|
|
else
|
|
std::cerr << "MANGOHUD: Unrecognized key: '" << ks << "'\n";
|
|
}
|
|
}
|
|
return keys;
|
|
}
|
|
|
|
static std::vector<KeySym>
|
|
parse_toggle_hud(const char *str)
|
|
{
|
|
return parse_string_to_keysym_vec(str);
|
|
}
|
|
|
|
static std::vector<KeySym>
|
|
parse_toggle_logging(const char *str)
|
|
{
|
|
return parse_string_to_keysym_vec(str);
|
|
}
|
|
|
|
static std::vector<KeySym>
|
|
parse_reload_cfg(const char *str)
|
|
{
|
|
return parse_string_to_keysym_vec(str);
|
|
}
|
|
|
|
static std::vector<KeySym>
|
|
parse_upload_log(const char *str)
|
|
{
|
|
return parse_string_to_keysym_vec(str);
|
|
}
|
|
|
|
#else
|
|
#define parse_toggle_hud(x) {}
|
|
#define parse_toggle_logging(x) {}
|
|
#define parse_reload_cfg(x) {}
|
|
#define parse_upload_log(x) {}
|
|
#endif
|
|
|
|
static uint32_t
|
|
parse_fps_sampling_period(const char *str)
|
|
{
|
|
return strtol(str, NULL, 0) * 1000;
|
|
}
|
|
|
|
static uint32_t
|
|
parse_fps_limit(const char *str)
|
|
{
|
|
return strtol(str, NULL, 0);
|
|
}
|
|
|
|
static bool
|
|
parse_no_display(const char *str)
|
|
{
|
|
return strtol(str, NULL, 0) != 0;
|
|
}
|
|
|
|
static unsigned
|
|
parse_color(const char *str)
|
|
{
|
|
return strtol(str, NULL, 16);
|
|
}
|
|
|
|
static unsigned
|
|
parse_unsigned(const char *str)
|
|
{
|
|
return strtol(str, NULL, 0);
|
|
}
|
|
|
|
static signed
|
|
parse_signed(const char *str)
|
|
{
|
|
return strtol(str, NULL, 0);
|
|
}
|
|
|
|
static std::string
|
|
parse_str(const char *str)
|
|
{
|
|
return str;
|
|
}
|
|
|
|
static std::string
|
|
parse_path(const char *str)
|
|
{
|
|
#ifdef _XOPEN_SOURCE
|
|
// Expand ~/ to home dir
|
|
if (str[0] == '~') {
|
|
std::string s;
|
|
wordexp_t e;
|
|
int ret;
|
|
|
|
if (!(ret = wordexp(str, &e, 0)))
|
|
s = e.we_wordv[0];
|
|
wordfree(&e);
|
|
|
|
if (!ret)
|
|
return s;
|
|
}
|
|
#endif
|
|
return str;
|
|
}
|
|
|
|
static std::vector<media_player_order>
|
|
parse_media_player_order(const char *str)
|
|
{
|
|
std::vector<media_player_order> order;
|
|
std::stringstream ss(str);
|
|
std::string token;
|
|
while (std::getline(ss, token, ',')) {
|
|
trim(token);
|
|
std::transform(token.begin(), token.end(), token.begin(), ::tolower);
|
|
if (token == "title")
|
|
order.push_back(MP_ORDER_TITLE);
|
|
else if (token == "artist")
|
|
order.push_back(MP_ORDER_ARTIST);
|
|
else if (token == "album")
|
|
order.push_back(MP_ORDER_ALBUM);
|
|
}
|
|
return order;
|
|
}
|
|
|
|
static uint32_t
|
|
parse_font_glyph_ranges(const char *str)
|
|
{
|
|
uint32_t fg = 0;
|
|
std::stringstream ss(str);
|
|
std::string token;
|
|
while (std::getline(ss, token, ',')) {
|
|
trim(token);
|
|
std::transform(token.begin(), token.end(), token.begin(), ::tolower);
|
|
|
|
if (token == "korean")
|
|
fg |= FG_KOREAN;
|
|
else if (token == "chinese")
|
|
fg |= FG_CHINESE_FULL;
|
|
else if (token == "chinese_simplified")
|
|
fg |= FG_CHINESE_SIMPLIFIED;
|
|
else if (token == "japanese")
|
|
fg |= FG_JAPANESE;
|
|
else if (token == "cyrillic")
|
|
fg |= FG_CYRILLIC;
|
|
else if (token == "thai")
|
|
fg |= FG_THAI;
|
|
else if (token == "vietnamese")
|
|
fg |= FG_VIETNAMESE;
|
|
else if (token == "latin_ext_a")
|
|
fg |= FG_LATIN_EXT_A;
|
|
else if (token == "latin_ext_b")
|
|
fg |= FG_LATIN_EXT_B;
|
|
}
|
|
return fg;
|
|
}
|
|
|
|
#define parse_width(s) parse_unsigned(s)
|
|
#define parse_height(s) parse_unsigned(s)
|
|
#define parse_vsync(s) parse_unsigned(s)
|
|
#define parse_gl_vsync(s) parse_signed(s)
|
|
#define parse_offset_x(s) parse_unsigned(s)
|
|
#define parse_offset_y(s) parse_unsigned(s)
|
|
#define parse_log_duration(s) parse_unsigned(s)
|
|
#define parse_time_format(s) parse_str(s)
|
|
#define parse_output_file(s) parse_path(s)
|
|
#define parse_font_file(s) parse_path(s)
|
|
#define parse_font_file_text(s) parse_path(s)
|
|
#define parse_io_read(s) parse_unsigned(s)
|
|
#define parse_io_write(s) parse_unsigned(s)
|
|
#define parse_pci_dev(s) parse_str(s)
|
|
#define parse_media_player_name(s) parse_str(s)
|
|
#define parse_font_scale_media_player(s) parse_float(s)
|
|
#define parse_cpu_text(s) parse_str(s)
|
|
#define parse_gpu_text(s) parse_str(s)
|
|
#define parse_log_interval(s) parse_unsigned(s)
|
|
#define parse_font_size(s) parse_float(s)
|
|
#define parse_font_size_text(s) parse_float(s)
|
|
#define parse_font_scale(s) parse_float(s)
|
|
#define parse_background_alpha(s) parse_float(s)
|
|
#define parse_alpha(s) parse_float(s)
|
|
#define parse_permit_upload(s) parse_unsigned(s)
|
|
|
|
#define parse_cpu_color(s) parse_color(s)
|
|
#define parse_gpu_color(s) parse_color(s)
|
|
#define parse_vram_color(s) parse_color(s)
|
|
#define parse_ram_color(s) parse_color(s)
|
|
#define parse_engine_color(s) parse_color(s)
|
|
#define parse_io_color(s) parse_color(s)
|
|
#define parse_frametime_color(s) parse_color(s)
|
|
#define parse_background_color(s) parse_color(s)
|
|
#define parse_text_color(s) parse_color(s)
|
|
#define parse_media_player_color(s) parse_color(s)
|
|
|
|
static bool
|
|
parse_help(const char *str)
|
|
{
|
|
fprintf(stderr, "Layer params using VK_LAYER_MESA_OVERLAY_CONFIG=\n");
|
|
#define OVERLAY_PARAM_BOOL(name) \
|
|
fprintf(stderr, "\t%s=0|1\n", #name);
|
|
#define OVERLAY_PARAM_CUSTOM(name)
|
|
OVERLAY_PARAMS
|
|
#undef OVERLAY_PARAM_BOOL
|
|
#undef OVERLAY_PARAM_CUSTOM
|
|
fprintf(stderr, "\tposition=top-left|top-right|bottom-left|bottom-right\n");
|
|
fprintf(stderr, "\tfps_sampling_period=number-of-milliseconds\n");
|
|
fprintf(stderr, "\tno_display=0|1\n");
|
|
fprintf(stderr, "\toutput_file=/path/to/output.txt\n");
|
|
fprintf(stderr, "\twidth=width-in-pixels\n");
|
|
fprintf(stderr, "\theight=height-in-pixels\n");
|
|
|
|
return true;
|
|
}
|
|
|
|
static bool is_delimiter(char c)
|
|
{
|
|
return c == 0 || c == ',' || c == ':' || c == ';' || c == '=';
|
|
}
|
|
|
|
static int
|
|
parse_string(const char *s, char *out_param, char *out_value)
|
|
{
|
|
int i = 0;
|
|
|
|
for (; !is_delimiter(*s); s++, out_param++, i++)
|
|
*out_param = *s;
|
|
|
|
*out_param = 0;
|
|
|
|
if (*s == '=') {
|
|
s++;
|
|
i++;
|
|
for (; !is_delimiter(*s); s++, out_value++, i++) {
|
|
*out_value = *s;
|
|
// Consume escaped delimiter, but don't escape null. Might be end of string.
|
|
if (*s == '\\' && *(s + 1) != 0 && is_delimiter(*(s + 1))) {
|
|
s++;
|
|
i++;
|
|
*out_value = *s;
|
|
}
|
|
}
|
|
} else
|
|
*(out_value++) = '1';
|
|
*out_value = 0;
|
|
|
|
if (*s && is_delimiter(*s)) {
|
|
s++;
|
|
i++;
|
|
}
|
|
|
|
if (*s && !i) {
|
|
fprintf(stderr, "mesa-overlay: syntax error: unexpected '%c' (%i) while "
|
|
"parsing a string\n", *s, *s);
|
|
fflush(stderr);
|
|
}
|
|
|
|
return i;
|
|
}
|
|
|
|
const char *overlay_param_names[] = {
|
|
#define OVERLAY_PARAM_BOOL(name) #name,
|
|
#define OVERLAY_PARAM_CUSTOM(name)
|
|
OVERLAY_PARAMS
|
|
#undef OVERLAY_PARAM_BOOL
|
|
#undef OVERLAY_PARAM_CUSTOM
|
|
};
|
|
|
|
void
|
|
parse_overlay_env(struct overlay_params *params,
|
|
const char *env)
|
|
{
|
|
uint32_t num;
|
|
char key[256], value[256];
|
|
while ((num = parse_string(env, key, value)) != 0) {
|
|
env += num;
|
|
if (!strcmp("full", key)) {
|
|
bool read_cfg = params->enabled[OVERLAY_PARAM_ENABLED_read_cfg];
|
|
#define OVERLAY_PARAM_BOOL(name) \
|
|
params->enabled[OVERLAY_PARAM_ENABLED_##name] = 1;
|
|
#define OVERLAY_PARAM_CUSTOM(name)
|
|
OVERLAY_PARAMS
|
|
#undef OVERLAY_PARAM_BOOL
|
|
#undef OVERLAY_PARAM_CUSTOM
|
|
params->enabled[OVERLAY_PARAM_ENABLED_histogram] = 0;
|
|
params->enabled[OVERLAY_PARAM_ENABLED_read_cfg] = read_cfg;
|
|
}
|
|
#define OVERLAY_PARAM_BOOL(name) \
|
|
if (!strcmp(#name, key)) { \
|
|
params->enabled[OVERLAY_PARAM_ENABLED_##name] = \
|
|
strtol(value, NULL, 0); \
|
|
continue; \
|
|
}
|
|
#define OVERLAY_PARAM_CUSTOM(name) \
|
|
if (!strcmp(#name, key)) { \
|
|
params->name = parse_##name(value); \
|
|
continue; \
|
|
}
|
|
OVERLAY_PARAMS
|
|
#undef OVERLAY_PARAM_BOOL
|
|
#undef OVERLAY_PARAM_CUSTOM
|
|
fprintf(stderr, "Unknown option '%s'\n", key);
|
|
}
|
|
}
|
|
|
|
void
|
|
parse_overlay_config(struct overlay_params *params,
|
|
const char *env)
|
|
{
|
|
|
|
*params = {};
|
|
|
|
/* Visible by default */
|
|
params->enabled[OVERLAY_PARAM_ENABLED_fps] = true;
|
|
params->enabled[OVERLAY_PARAM_ENABLED_frame_timing] = true;
|
|
params->enabled[OVERLAY_PARAM_ENABLED_core_load] = false;
|
|
params->enabled[OVERLAY_PARAM_ENABLED_cpu_temp] = false;
|
|
params->enabled[OVERLAY_PARAM_ENABLED_gpu_temp] = false;
|
|
params->enabled[OVERLAY_PARAM_ENABLED_cpu_stats] = true;
|
|
params->enabled[OVERLAY_PARAM_ENABLED_gpu_stats] = true;
|
|
params->enabled[OVERLAY_PARAM_ENABLED_ram] = false;
|
|
params->enabled[OVERLAY_PARAM_ENABLED_vram] = false;
|
|
params->enabled[OVERLAY_PARAM_ENABLED_read_cfg] = false;
|
|
params->enabled[OVERLAY_PARAM_ENABLED_io_read] = false;
|
|
params->enabled[OVERLAY_PARAM_ENABLED_io_write] = false;
|
|
params->fps_sampling_period = 500000; /* 500ms */
|
|
params->width = 0;
|
|
params->height = 140;
|
|
params->control = -1;
|
|
params->fps_limit = 0;
|
|
params->vsync = -1;
|
|
params->gl_vsync = -2;
|
|
params->offset_x = 0;
|
|
params->offset_y = 0;
|
|
params->background_alpha = 0.5;
|
|
params->alpha = 1.0;
|
|
params->time_format = "%T";
|
|
params->gpu_color = 0x2e9762;
|
|
params->cpu_color = 0x2e97cb;
|
|
params->vram_color = 0xad64c1;
|
|
params->ram_color = 0xc26693;
|
|
params->engine_color = 0xeb5b5b;
|
|
params->io_color = 0xa491d3;
|
|
params->frametime_color = 0x00ff00;
|
|
params->background_color = 0x020202;
|
|
params->text_color = 0xffffff;
|
|
params->media_player_color = 0xffffff;
|
|
params->media_player_name = "spotify";
|
|
params->font_scale = 1.0f;
|
|
params->font_scale_media_player = 0.55f;
|
|
params->log_interval = 100;
|
|
params->media_player_order = { MP_ORDER_TITLE, MP_ORDER_ARTIST, MP_ORDER_ALBUM };
|
|
params->permit_upload = 0;
|
|
|
|
#ifdef HAVE_X11
|
|
params->toggle_hud = { XK_Shift_R, XK_F12 };
|
|
params->toggle_logging = { XK_Shift_L, XK_F2 };
|
|
params->reload_cfg = { XK_Shift_L, XK_F4 };
|
|
params->upload_log = { XK_Shift_L, XK_F3 };
|
|
#endif
|
|
|
|
// first pass with env var
|
|
if (env)
|
|
parse_overlay_env(params, env);
|
|
|
|
bool read_cfg = params->enabled[OVERLAY_PARAM_ENABLED_read_cfg];
|
|
if (!env || read_cfg) {
|
|
|
|
// Get config options
|
|
parseConfigFile(*params);
|
|
if (params->options.find("full") != params->options.end() && params->options.find("full")->second != "0") {
|
|
#define OVERLAY_PARAM_BOOL(name) \
|
|
params->enabled[OVERLAY_PARAM_ENABLED_##name] = 1;
|
|
#define OVERLAY_PARAM_CUSTOM(name)
|
|
OVERLAY_PARAMS
|
|
#undef OVERLAY_PARAM_BOOL
|
|
#undef OVERLAY_PARAM_CUSTOM
|
|
params->enabled[OVERLAY_PARAM_ENABLED_histogram] = 0;
|
|
params->options.erase("full");
|
|
}
|
|
for (auto& it : params->options) {
|
|
#define OVERLAY_PARAM_BOOL(name) \
|
|
if (it.first == #name) { \
|
|
params->enabled[OVERLAY_PARAM_ENABLED_##name] = \
|
|
strtol(it.second.c_str(), NULL, 0); \
|
|
continue; \
|
|
}
|
|
#define OVERLAY_PARAM_CUSTOM(name) \
|
|
if (it.first == #name) { \
|
|
params->name = parse_##name(it.second.c_str()); \
|
|
continue; \
|
|
}
|
|
OVERLAY_PARAMS
|
|
#undef OVERLAY_PARAM_BOOL
|
|
#undef OVERLAY_PARAM_CUSTOM
|
|
fprintf(stderr, "Unknown option '%s'\n", it.first.c_str());
|
|
}
|
|
|
|
}
|
|
|
|
// second pass, override config file settings with MANGOHUD_CONFIG
|
|
if (env && read_cfg)
|
|
parse_overlay_env(params, env);
|
|
|
|
if (params->font_scale_media_player <= 0.f)
|
|
params->font_scale_media_player = 0.55f;
|
|
|
|
// Convert from 0xRRGGBB to ImGui's format
|
|
std::array<unsigned *, 10> colors = {
|
|
¶ms->cpu_color,
|
|
¶ms->gpu_color,
|
|
¶ms->vram_color,
|
|
¶ms->ram_color,
|
|
¶ms->engine_color,
|
|
¶ms->io_color,
|
|
¶ms->background_color,
|
|
¶ms->frametime_color,
|
|
¶ms->text_color,
|
|
¶ms->media_player_color,
|
|
};
|
|
|
|
for (auto color : colors){
|
|
*color =
|
|
IM_COL32(RGBGetRValue(*color),
|
|
RGBGetGValue(*color),
|
|
RGBGetBValue(*color),
|
|
255);
|
|
}
|
|
|
|
params->tableCols = 3;
|
|
|
|
if (!params->font_size) {
|
|
params->font_size = 24;
|
|
}
|
|
|
|
//increase hud width if io read and write
|
|
if (!params->width) {
|
|
if ((params->enabled[OVERLAY_PARAM_ENABLED_io_read] || params->enabled[OVERLAY_PARAM_ENABLED_io_write])) {
|
|
params->width = 13 * params->font_size * params->font_scale;
|
|
} else {
|
|
params->width = params->font_size * params->font_scale * 11.7;
|
|
}
|
|
}
|
|
|
|
// set frametime limit
|
|
using namespace std::chrono;
|
|
if (params->fps_limit >= 0)
|
|
fps_limit_stats.targetFrameTime = duration_cast<Clock::duration>(duration<double>(1) / params->fps_limit);
|
|
|
|
#ifdef HAVE_DBUS
|
|
if (params->enabled[OVERLAY_PARAM_ENABLED_media_player]) {
|
|
// lock mutexes for config file change notifier thread
|
|
{
|
|
std::lock_guard<std::mutex> lk(main_metadata.mutex);
|
|
main_metadata.clear();
|
|
}
|
|
{
|
|
std::lock_guard<std::mutex> lk(generic_mpris.mutex);
|
|
generic_mpris.clear();
|
|
}
|
|
if (dbusmgr::dbus_mgr.init(params->media_player_name)) {
|
|
if (!get_media_player_metadata(dbusmgr::dbus_mgr, params->media_player_name, main_metadata))
|
|
std::cerr << "MANGOHUD: Failed to get initial media player metadata." << std::endl;
|
|
}
|
|
} else {
|
|
dbusmgr::dbus_mgr.deinit();
|
|
main_metadata.valid = false;
|
|
generic_mpris.valid = false;
|
|
}
|
|
#endif
|
|
|
|
}
|