Unified WM / DE / Resolution detection

This commit is contained in:
Linus Dierheimer
2022-01-02 16:55:26 +01:00
parent 06fa8e7478
commit bcf543e129
22 changed files with 948 additions and 937 deletions
+1 -1
View File
@@ -175,7 +175,7 @@ static bool printCursorFromEnv(FFinstance* instance)
void ffPrintCursor(FFinstance* instance)
{
const FFWMDEResult* wmde = ffDetectWMDE(instance);
const FFDisplayServerResult* wmde = ffConnectDisplayServer(instance);
if(ffStrbufIgnCaseCompS(&wmde->wmProtocolName, "TTY") == 0)
{
+2 -3
View File
@@ -3,11 +3,11 @@
#include <string.h>
#define FF_DE_MODULE_NAME "DE"
#define FF_DE_NUM_FORMAT_ARGS 4
#define FF_DE_NUM_FORMAT_ARGS 3
void ffPrintDesktopEnvironment(FFinstance* instance)
{
const FFWMDEResult* result = ffDetectWMDE(instance);
const FFDisplayServerResult* result = ffConnectDisplayServer(instance);
if(result->dePrettyName.length == 0)
{
@@ -32,7 +32,6 @@ void ffPrintDesktopEnvironment(FFinstance* instance)
else
{
ffPrintFormatString(instance, FF_DE_MODULE_NAME, 0, &instance->config.deKey, &instance->config.deFormat, NULL, FF_DE_NUM_FORMAT_ARGS, (FFformatarg[]){
{FF_FORMAT_ARG_TYPE_STRING, result->sessionDesktop},
{FF_FORMAT_ARG_TYPE_STRBUF, &result->deProcessName},
{FF_FORMAT_ARG_TYPE_STRBUF, &result->dePrettyName},
{FF_FORMAT_ARG_TYPE_STRBUF, &result->deVersion}
+1 -1
View File
@@ -5,7 +5,7 @@
void ffPrintFont(FFinstance* instance)
{
const FFWMDEResult* wmde = ffDetectWMDE(instance);
const FFDisplayServerResult* wmde = ffConnectDisplayServer(instance);
if(ffStrbufIgnCaseCompS(&wmde->wmProtocolName, "TTY") == 0)
{
+1 -1
View File
@@ -5,7 +5,7 @@
void ffPrintIcons(FFinstance* instance)
{
const FFWMDEResult* wmde = ffDetectWMDE(instance);
const FFDisplayServerResult* wmde = ffConnectDisplayServer(instance);
if(ffStrbufIgnCaseCompS(&wmde->wmProtocolName, "TTY") == 0)
{
+7 -485
View File
@@ -1,25 +1,16 @@
#include "fastfetch.h"
#include <string.h>
#include <dirent.h>
#include <pthread.h>
#define FF_RESOLUTION_MODULE_NAME "Resolution"
#define FF_RESOLUTION_NUM_FORMAT_ARGS 3
typedef struct ResolutionResult
void ffPrintResolution(FFinstance* instance)
{
uint32_t width;
uint32_t height;
uint32_t refreshRate;
} ResolutionResult;
const FFDisplayServerResult* dsResult = ffConnectDisplayServer(instance);
static bool printResolutionResultList(FFinstance* instance, FFlist* results)
{
for(uint32_t i = 0; i < results->length; i++)
for(uint32_t i = 0; i < dsResult->resolutions.length; i++)
{
ResolutionResult* result = ffListGet(results, i);
uint8_t moduleIndex = results->length == 1 ? 0 : (uint8_t) i + 1;
FFResolutionResult* result = ffListGet(&dsResult->resolutions, i);
uint8_t moduleIndex = dsResult->resolutions.length == 1 ? 0 : (uint8_t) i + 1;
if(instance->config.resolutionFormat.length == 0)
{
@@ -41,475 +32,6 @@ static bool printResolutionResultList(FFinstance* instance, FFlist* results)
}
}
bool res = results->length > 0;
ffListDestroy(results);
return res;
}
static void printResolutionDRMBackend(FFinstance* instance)
{
const char* drmDirPath = "/sys/class/drm/";
DIR* dirp = opendir(drmDirPath);
if(dirp == NULL)
{
ffPrintError(instance, FF_RESOLUTION_MODULE_NAME, 0, &instance->config.resolutionKey, &instance->config.resolutionFormat, FF_RESOLUTION_NUM_FORMAT_ARGS, "Couldn't connect to a display server or open %s", drmDirPath);
return;
}
FFstrbuf drmDir;
ffStrbufInitA(&drmDir, 64);
ffStrbufAppendS(&drmDir, drmDirPath);
uint32_t drmDirLength = drmDir.length;
FFlist modes;
ffListInitA(&modes, sizeof(ResolutionResult), 4);
struct dirent* entry;
while((entry = readdir(dirp)) != NULL)
{
ffStrbufAppendS(&drmDir, entry->d_name);
ffStrbufAppendS(&drmDir, "/modes");
FILE* modeFile = fopen(drmDir.chars, "r");
if(modeFile == NULL)
{
ffStrbufSubstrBefore(&drmDir, drmDirLength);
continue;
}
ResolutionResult* result = ffListAdd(&modes);
result->width = 0;
result->height = 0;
result->refreshRate = 0;
int scanned = fscanf(modeFile, "%ix%i", &result->width, &result->height);
if(scanned < 2 || result->width == 0 || result->height == 0)
--modes.length;
fclose(modeFile);
ffStrbufSubstrBefore(&drmDir, drmDirLength);
}
closedir(dirp);
ffStrbufDestroy(&drmDir);
if(modes.length == 0)
{
ffPrintError(instance, FF_RESOLUTION_MODULE_NAME, 0, &instance->config.resolutionKey, &instance->config.resolutionFormat, FF_RESOLUTION_NUM_FORMAT_ARGS, "Couldn't connect to a display server or find a resolution in %s", drmDirPath);
ffListDestroy(&modes);
return;
}
printResolutionResultList(instance, &modes);
}
#if defined(FF_HAVE_XRANDR) || defined(FF_HAVE_WAYLAND)
static uint32_t parseRefreshRate(int32_t refreshRate)
{
if(refreshRate <= 0)
return 0;
int remainder = refreshRate % 5;
if(remainder >= 3)
refreshRate += (5 - remainder);
else
refreshRate -= remainder;
//All other typicall refresh rates are dividable by 5
if(refreshRate == 145)
refreshRate = 144;
return (uint32_t) refreshRate;
}
#endif //FF_HAVE_XRANDR || FF_HAVE_WAYLAND
#ifdef FF_HAVE_X11
#include <X11/Xlib.h>
static void x11AddScreenAsResult(FFlist* results, Screen* screen, uint32_t refreshRate)
{
if(WidthOfScreen(screen) == 0 || HeightOfScreen(screen) == 0)
return;
ResolutionResult* result = ffListAdd(results);
result->width = (uint32_t) WidthOfScreen(screen);
result->height = (uint32_t) HeightOfScreen(screen);
result->refreshRate = refreshRate;
}
static bool printResolutionX11Backend(FFinstance* instance)
{
FF_LIBRARY_LOAD(x11, "libX11.so", instance->config.libX11, false)
FF_LIBRARY_LOAD_SYMBOL(x11, XOpenDisplay, false)
FF_LIBRARY_LOAD_SYMBOL(x11, XCloseDisplay, false)
Display* display = ffXOpenDisplay(x11);
if(display == NULL)
{
dlclose(x11);
return false;
}
FFlist results;
ffListInitA(&results, sizeof(ResolutionResult), 4);
for(int i = 0; i < ScreenCount(display); i++)
x11AddScreenAsResult(&results, ScreenOfDisplay(display, i), 0);
ffXCloseDisplay(display);
dlclose(x11);
return printResolutionResultList(instance, &results);
}
#endif //FF_HAVE_X11
#ifdef FF_HAVE_XRANDR
#include <X11/extensions/Xrandr.h>
typedef struct XrandrData
{
FF_LIBRARY_SYMBOL(XRRGetScreenInfo)
FF_LIBRARY_SYMBOL(XRRConfigCurrentConfiguration)
FF_LIBRARY_SYMBOL(XRRConfigCurrentRate)
FF_LIBRARY_SYMBOL(XRRGetMonitors)
FF_LIBRARY_SYMBOL(XRRGetScreenResources)
FF_LIBRARY_SYMBOL(XRRGetOutputInfo)
FF_LIBRARY_SYMBOL(XRRGetCrtcInfo)
FF_LIBRARY_SYMBOL(XRRFreeCrtcInfo)
FF_LIBRARY_SYMBOL(XRRFreeOutputInfo)
FF_LIBRARY_SYMBOL(XRRFreeScreenResources)
FF_LIBRARY_SYMBOL(XRRFreeMonitors)
FF_LIBRARY_SYMBOL(XRRFreeScreenConfigInfo)
//Init once
Display* display;
FFlist results;
//Init per screen
uint32_t defaultRefreshRate;
XRRScreenResources* screenResources;
} XrandrData;
static XRRModeInfo* xrandrGetModeInfo(XRRScreenResources* screenResources, RRMode mode)
{
for(int i = 0; i < screenResources->nmode; i++)
{
if(screenResources->modes[i].id == mode)
return &screenResources->modes[i];
}
return NULL;
}
static bool xrandrHandleOutputInfo(XrandrData* data, XRROutputInfo* outputInfo)
{
XRRCrtcInfo* crtcInfo = data->ffXRRGetCrtcInfo(data->display, data->screenResources, outputInfo->crtc);
if(crtcInfo == NULL)
return false;
XRRModeInfo* modeInfo = xrandrGetModeInfo(data->screenResources, crtcInfo->mode);
if(modeInfo == NULL || modeInfo->width == 0 || modeInfo->height == 0)
{
data->ffXRRFreeCrtcInfo(crtcInfo);
return false;
}
ResolutionResult* result = ffListAdd(&data->results);
result->width = modeInfo->width;
result->height = modeInfo->height;
result->refreshRate = parseRefreshRate((int32_t) (modeInfo->dotClock / (modeInfo->hTotal * modeInfo->vTotal)));
if(result->refreshRate == 0)
result->refreshRate = data->defaultRefreshRate;
data->ffXRRFreeCrtcInfo(crtcInfo);
return true;
}
static bool xrandrHandleMonitorFallback(XrandrData* data, XRRMonitorInfo* monitorInfo)
{
if(monitorInfo->width == 0 || monitorInfo->height == 0)
return false;
ResolutionResult* result = ffListAdd(&data->results);
result->width = (uint32_t) monitorInfo->width;
result->height = (uint32_t) monitorInfo->height;
result->refreshRate = (uint32_t) data->defaultRefreshRate;
return true;
}
static bool xrandrHandleMonitor(XrandrData* data, XRRMonitorInfo* monitorInfo)
{
bool foundMode = false;
for(int i = 0; i < monitorInfo->noutput; i++)
{
XRROutputInfo* outputInfo = data->ffXRRGetOutputInfo(data->display, data->screenResources, monitorInfo->outputs[i]);
if(outputInfo == NULL)
continue;
if(xrandrHandleOutputInfo(data, outputInfo))
foundMode = true;
data->ffXRRFreeOutputInfo(outputInfo);
}
if(!foundMode)
return xrandrHandleMonitorFallback(data, monitorInfo);
return true;
}
static inline bool xrandrLoopMonitors(XrandrData* data, XRRMonitorInfo* monitorInfos, int numberOfMonitors, bool(*monitorFunc)(XrandrData* data, XRRMonitorInfo* monitorInfo))
{
bool foundAMonitor = false;
for(int i = 0; i < numberOfMonitors; i++)
{
if(monitorFunc(data, &monitorInfos[i]))
foundAMonitor = true;
}
return foundAMonitor;
}
static void xrandrHandleScreen(XrandrData* data, Screen* screen)
{
Window window = RootWindowOfScreen(screen);
XRRScreenConfiguration* screenConfiguration = data->ffXRRGetScreenInfo(data->display, window);
if(screenConfiguration != NULL)
{
data->defaultRefreshRate = (uint32_t) data->ffXRRConfigCurrentRate(screenConfiguration);
data->ffXRRFreeScreenConfigInfo(screenConfiguration);
}
else
data->defaultRefreshRate = 0;
int numberOfMonitors;
XRRMonitorInfo* monitorInfos = data->ffXRRGetMonitors(data->display, window, True, &numberOfMonitors);
if(monitorInfos == NULL)
{
x11AddScreenAsResult(&data->results, screen, data->defaultRefreshRate);
return;
}
data->screenResources = data->ffXRRGetScreenResources(data->display, window);
bool foundAMonitor;
if(data->screenResources != NULL)
foundAMonitor = xrandrLoopMonitors(data, monitorInfos, numberOfMonitors, xrandrHandleMonitor);
else
foundAMonitor = xrandrLoopMonitors(data, monitorInfos, numberOfMonitors, xrandrHandleMonitorFallback);
if(!foundAMonitor)
x11AddScreenAsResult(&data->results, screen, data->defaultRefreshRate);
data->ffXRRFreeScreenResources(data->screenResources);
data->ffXRRFreeMonitors(monitorInfos);
}
static bool printResolutionXrandrBackend(FFinstance* instance)
{
FF_LIBRARY_LOAD(xrandr, "libXrandr.so", instance->config.libXrandr, false);
FF_LIBRARY_LOAD_SYMBOL(xrandr, XOpenDisplay, false)
FF_LIBRARY_LOAD_SYMBOL(xrandr, XCloseDisplay, false)
XrandrData data;
FF_LIBRARY_LOAD_SYMBOL_ADRESS(xrandr, data.ffXRRGetScreenInfo, XRRGetScreenInfo, false)
FF_LIBRARY_LOAD_SYMBOL_ADRESS(xrandr, data.ffXRRConfigCurrentRate, XRRConfigCurrentRate, false);
FF_LIBRARY_LOAD_SYMBOL_ADRESS(xrandr, data.ffXRRGetMonitors, XRRGetMonitors, false);
FF_LIBRARY_LOAD_SYMBOL_ADRESS(xrandr, data.ffXRRGetScreenResources, XRRGetScreenResources, false);
FF_LIBRARY_LOAD_SYMBOL_ADRESS(xrandr, data.ffXRRGetOutputInfo, XRRGetOutputInfo, false);
FF_LIBRARY_LOAD_SYMBOL_ADRESS(xrandr, data.ffXRRGetCrtcInfo, XRRGetCrtcInfo, false);
FF_LIBRARY_LOAD_SYMBOL_ADRESS(xrandr, data.ffXRRFreeCrtcInfo, XRRFreeCrtcInfo, false);
FF_LIBRARY_LOAD_SYMBOL_ADRESS(xrandr, data.ffXRRFreeOutputInfo, XRRFreeOutputInfo, false);
FF_LIBRARY_LOAD_SYMBOL_ADRESS(xrandr, data.ffXRRFreeScreenResources, XRRFreeScreenResources, false);
FF_LIBRARY_LOAD_SYMBOL_ADRESS(xrandr, data.ffXRRFreeMonitors, XRRFreeMonitors, false);
FF_LIBRARY_LOAD_SYMBOL_ADRESS(xrandr, data.ffXRRFreeScreenConfigInfo, XRRFreeScreenConfigInfo, false);
data.display = ffXOpenDisplay(NULL);
if(data.display == NULL)
{
dlclose(xrandr);
return false;
}
ffListInitA(&data.results, sizeof(ResolutionResult), 4);
for(int i = 0; i < ScreenCount(data.display); i++)
xrandrHandleScreen(&data, ScreenOfDisplay(data.display, i));
ffXCloseDisplay(data.display);
dlclose(xrandr);
return printResolutionResultList(instance, &data.results);
}
#endif // FF_HAVE_XRANDR
#ifdef FF_HAVE_WAYLAND
#include <wayland-client.h>
typedef struct WaylandData
{
FFinstance* instance;
FFlist results;
FF_LIBRARY_SYMBOL(wl_proxy_marshal_constructor_versioned)
FF_LIBRARY_SYMBOL(wl_proxy_add_listener)
FF_LIBRARY_SYMBOL(wl_proxy_destroy)
const struct wl_interface* ffwl_output_interface;
struct wl_output_listener output_listener;
} WaylandData;
static void waylandGlobalRemoveListener(void* data, struct wl_registry* wl_registry, uint32_t name){
FF_UNUSED(data, wl_registry, name);
return;
}
static void waylandOutputGeometryListener(void* data, struct wl_output* wl_output, int32_t x, int32_t y, int32_t physical_width, int32_t physical_height, int32_t subpixel, const char* make, const char* model, int32_t transform)
{
FF_UNUSED(data, wl_output, x, y, physical_width, physical_height, subpixel, make, model, transform);
return;
}
static void waylandOutputDoneListener(void* data, struct wl_output* wl_output)
{
FF_UNUSED(data, wl_output);
return;
}
static void waylandOutputScaleListener(void* data, struct wl_output* wl_output, int32_t factor)
{
FF_UNUSED(data, wl_output, factor);
return;
}
static void waylandOutputModeListener(void* data, struct wl_output* output, uint32_t flags, int32_t width, int32_t height, int32_t refreshRate)
{
if(!(flags & WL_OUTPUT_MODE_CURRENT))
return;
WaylandData* wldata = (WaylandData*) data;
wldata->ffwl_proxy_destroy((struct wl_proxy*) output);
if(width <= 0 || height <= 0)
return;
static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
pthread_mutex_lock(&mutex);
ResolutionResult* result = ffListAdd(&wldata->results);
pthread_mutex_unlock(&mutex);
result->width = (uint32_t) width;
result->height = (uint32_t) height;
result->refreshRate = parseRefreshRate(refreshRate / 1000);
}
static void waylandGlobalAddListener(void* data, struct wl_registry* registry, uint32_t name, const char* interface, uint32_t version)
{
if(strcmp(interface, "wl_output") == 0)
{
WaylandData* wldata = (WaylandData*) data;
struct wl_output* output = (struct wl_output*) wldata->ffwl_proxy_marshal_constructor_versioned((struct wl_proxy *) registry, WL_REGISTRY_BIND, wldata->ffwl_output_interface, version, name, wldata->ffwl_output_interface->name, version, NULL);
if(output == NULL)
return;
wldata->ffwl_proxy_add_listener((struct wl_proxy*) output, (void(**)(void)) &wldata->output_listener, data);
}
}
static bool printResolutionWaylandBackend(FFinstance* instance)
{
if(getenv("XDG_RUNTIME_DIR") == NULL)
return false;
const char* sessionType = getenv("XDG_SESSION_TYPE");
if(sessionType != NULL && strcasecmp(sessionType, "wayland") != 0)
return false;
FF_LIBRARY_LOAD(wayland, "libwayland-client.so", instance->config.libWayland, false);
FF_LIBRARY_LOAD_SYMBOL(wayland, wl_display_connect, false)
FF_LIBRARY_LOAD_SYMBOL(wayland, wl_display_dispatch, false)
FF_LIBRARY_LOAD_SYMBOL(wayland, wl_display_roundtrip, false)
FF_LIBRARY_LOAD_SYMBOL(wayland, wl_proxy_marshal_constructor, false)
FF_LIBRARY_LOAD_SYMBOL(wayland, wl_display_disconnect, false)
FF_LIBRARY_LOAD_SYMBOL(wayland, wl_registry_interface, false)
WaylandData data;
FF_LIBRARY_LOAD_SYMBOL_ADRESS(wayland, data.ffwl_proxy_marshal_constructor_versioned, wl_proxy_marshal_constructor_versioned, false)
FF_LIBRARY_LOAD_SYMBOL_ADRESS(wayland, data.ffwl_proxy_add_listener, wl_proxy_add_listener, false)
FF_LIBRARY_LOAD_SYMBOL_ADRESS(wayland, data.ffwl_output_interface, wl_output_interface, false)
FF_LIBRARY_LOAD_SYMBOL_ADRESS(wayland, data.ffwl_proxy_destroy, wl_proxy_destroy, false)
struct wl_display* display = ffwl_display_connect(NULL);
if(display == NULL)
{
dlclose(wayland);
return false;
}
struct wl_registry* registry = (struct wl_registry*) ffwl_proxy_marshal_constructor((struct wl_proxy*) display, WL_DISPLAY_GET_REGISTRY, ffwl_registry_interface, NULL);
if(registry == NULL)
{
ffwl_display_disconnect(display);
dlclose(wayland);
return false;
}
data.instance = instance;
ffListInitA(&data.results, sizeof(ResolutionResult), 4);
struct wl_registry_listener regestry_listener;
regestry_listener.global = waylandGlobalAddListener;
regestry_listener.global_remove = waylandGlobalRemoveListener;
data.output_listener.geometry = waylandOutputGeometryListener;
data.output_listener.mode = waylandOutputModeListener;
data.output_listener.done = waylandOutputDoneListener;
data.output_listener.scale = waylandOutputScaleListener;
data.ffwl_proxy_add_listener((struct wl_proxy*) registry, (void(**)(void)) &regestry_listener, &data);
ffwl_display_dispatch(display);
ffwl_display_roundtrip(display);
data.ffwl_proxy_destroy((struct wl_proxy*) registry);
ffwl_display_disconnect(display);
dlclose(wayland);
return printResolutionResultList(instance, &data.results);
}
#endif //FF_HAVE_WAYLAND
void ffPrintResolution(FFinstance* instance)
{
if(
#if FF_HAVE_WAYLAND
printResolutionWaylandBackend(instance) ||
#endif
#if FF_HAVE_XRANDR
printResolutionXrandrBackend(instance) ||
#endif
#if FF_HAVE_X11
printResolutionX11Backend(instance) ||
#endif
false
) return;
printResolutionDRMBackend(instance);
if(dsResult->resolutions.length == 0)
ffPrintError(instance, FF_RESOLUTION_MODULE_NAME, 0, &instance->config.resolutionKey, &instance->config.resolutionFormat, FF_RESOLUTION_NUM_FORMAT_ARGS, "Couldn't detect resolution");
}
+1 -1
View File
@@ -30,7 +30,7 @@ static void printTerminalFont(FFinstance* instance, const char* raw, FFfont* fon
static const char* getSystemMonospaceFont(FFinstance* instance)
{
const FFWMDEResult* wmde = ffDetectWMDE(instance);
const FFDisplayServerResult* wmde = ffConnectDisplayServer(instance);
if(ffStrbufIgnCaseCompS(&wmde->dePrettyName, "Cinnamon") == 0)
{
+1 -1
View File
@@ -5,7 +5,7 @@
void ffPrintTheme(FFinstance* instance)
{
const FFWMDEResult* wmde = ffDetectWMDE(instance);
const FFDisplayServerResult* wmde = ffConnectDisplayServer(instance);
if(ffStrbufIgnCaseCompS(&wmde->wmProtocolName, "TTY") == 0)
{
+3 -4
View File
@@ -1,11 +1,11 @@
#include "fastfetch.h"
#define FF_WM_MODULE_NAME "WM"
#define FF_WM_NUM_FORMAT_ARGS 4
#define FF_WM_NUM_FORMAT_ARGS 3
void ffPrintWM(FFinstance* instance)
{
const FFWMDEResult* result = ffDetectWMDE(instance);
const FFDisplayServerResult* result = ffConnectDisplayServer(instance);
if(result->wmPrettyName.length == 0)
{
@@ -15,7 +15,7 @@ void ffPrintWM(FFinstance* instance)
if(instance->config.wmFormat.length == 0)
{
ffPrintLogoAndKey(instance, "WM", 0, &instance->config.wmKey);
ffPrintLogoAndKey(instance, FF_WM_MODULE_NAME, 0, &instance->config.wmKey);
if(result->wmPrettyName.length == 0 && result->wmProcessName.length == 0)
{
@@ -41,7 +41,6 @@ void ffPrintWM(FFinstance* instance)
else
{
ffPrintFormatString(instance, FF_WM_MODULE_NAME, 0, &instance->config.wmKey, &instance->config.wmFormat, NULL, FF_WM_NUM_FORMAT_ARGS, (FFformatarg[]){
{FF_FORMAT_ARG_TYPE_STRING, result->sessionDesktop},
{FF_FORMAT_ARG_TYPE_STRBUF, &result->wmProcessName},
{FF_FORMAT_ARG_TYPE_STRBUF, &result->wmPrettyName},
{FF_FORMAT_ARG_TYPE_STRBUF, &result->wmProtocolName}
+1 -1
View File
@@ -223,7 +223,7 @@ static void printOpenbox(FFinstance* instance, const FFstrbuf* dePrettyName)
void ffPrintWMTheme(FFinstance* instance)
{
const FFWMDEResult* result = ffDetectWMDE(instance);
const FFDisplayServerResult* result = ffConnectDisplayServer(instance);
if(result->wmPrettyName.length == 0)
{