hyprland-plugins/csgo-vulkan-fix/main.cpp

92 lines
3.6 KiB
C++
Raw Normal View History

2023-02-27 15:02:21 +01:00
#define WLR_USE_UNSTABLE
#include <unistd.h>
#include <hyprland/src/Compositor.hpp>
#include <hyprland/src/Window.hpp>
#include <hyprland/src/config/ConfigManager.hpp>
2023-02-27 15:02:21 +01:00
#include "globals.hpp"
// Methods
2023-09-03 16:40:39 +02:00
inline CFunctionHook* g_pSurfaceSizeHook = nullptr;
2023-02-27 15:02:21 +01:00
inline CFunctionHook* g_pSurfaceDamageHook = nullptr;
typedef void (*origSurfaceSize)(wlr_xwayland_surface*, int16_t, int16_t, uint16_t, uint16_t);
typedef void (*origSurfaceDamage)(wlr_surface*, pixman_region32_t*);
// store the surface for csgo. May be invalid, only compare against
2023-09-03 16:40:39 +02:00
inline wlr_surface* pCSGOSurface = nullptr;
2023-02-27 15:02:21 +01:00
inline wlr_xwayland_surface* pCSGOXWSurface = nullptr;
2023-09-03 16:40:39 +02:00
inline int csgoMonitor = 0;
2023-02-27 15:02:21 +01:00
// Do NOT change this function.
2023-09-03 16:40:39 +02:00
APICALL EXPORT std::string PLUGIN_API_VERSION() {
return HYPRLAND_API_VERSION;
}
2023-02-27 15:02:21 +01:00
void onNewWindow(void* self, std::any data) {
// data is guaranteed
auto* const PWINDOW = std::any_cast<CWindow*>(data);
2023-09-28 00:06:14 +02:00
if (g_pXWaylandManager->getAppIDClass(PWINDOW) == "cs2") {
2023-09-03 16:40:39 +02:00
pCSGOSurface = g_pXWaylandManager->getWindowSurface(PWINDOW);
2023-02-27 15:02:21 +01:00
pCSGOXWSurface = PWINDOW->m_uSurface.xwayland;
2023-09-03 16:40:39 +02:00
csgoMonitor = PWINDOW->m_iMonitorID;
2023-02-27 15:02:21 +01:00
}
}
void hkSetWindowSize(wlr_xwayland_surface* surface, int16_t x, int16_t y, uint16_t width, uint16_t height) {
static auto* const RESX = &HyprlandAPI::getConfigValue(PHANDLE, "plugin:csgo-vulkan-fix:res_w")->intValue;
static auto* const RESY = &HyprlandAPI::getConfigValue(PHANDLE, "plugin:csgo-vulkan-fix:res_h")->intValue;
2023-09-03 16:40:39 +02:00
if (!surface) {
(*(origSurfaceSize)g_pSurfaceSizeHook->m_pOriginal)(surface, x, y, width, height);
return;
}
const auto SURF = surface->surface;
const auto PWINDOW = g_pCompositor->getWindowFromSurface(SURF);
2023-09-28 00:06:14 +02:00
if (PWINDOW && PWINDOW->m_szInitialClass == "cs2") {
2023-09-03 16:40:39 +02:00
width = *RESX;
2023-02-27 15:02:21 +01:00
height = *RESY;
}
(*(origSurfaceSize)g_pSurfaceSizeHook->m_pOriginal)(surface, x, y, width, height);
}
void hkSurfaceDamage(wlr_surface* surface, pixman_region32_t* damage) {
(*(origSurfaceDamage)g_pSurfaceDamageHook->m_pOriginal)(surface, damage);
if (surface == pCSGOSurface) {
const auto PMONITOR = g_pCompositor->getMonitorFromID(csgoMonitor);
2023-09-03 16:40:39 +02:00
if (PMONITOR)
pixman_region32_union_rect(damage, damage, 0, 0, PMONITOR->vecSize.x, PMONITOR->vecSize.y);
2023-02-27 15:02:21 +01:00
}
}
APICALL EXPORT PLUGIN_DESCRIPTION_INFO PLUGIN_INIT(HANDLE handle) {
PHANDLE = handle;
HyprlandAPI::addConfigValue(PHANDLE, "plugin:csgo-vulkan-fix:res_w", SConfigValue{.intValue = 1680});
HyprlandAPI::addConfigValue(PHANDLE, "plugin:csgo-vulkan-fix:res_h", SConfigValue{.intValue = 1050});
2023-09-03 16:40:39 +02:00
g_pSurfaceSizeHook = HyprlandAPI::createFunctionHook(PHANDLE, (void*)&wlr_xwayland_surface_configure, (void*)&hkSetWindowSize);
2023-02-27 15:02:21 +01:00
g_pSurfaceDamageHook = HyprlandAPI::createFunctionHook(PHANDLE, (void*)&wlr_surface_get_effective_damage, (void*)&hkSurfaceDamage);
2023-09-28 00:06:14 +02:00
bool hkResult = hkResult && g_pSurfaceSizeHook->hook();
2023-09-03 16:40:39 +02:00
hkResult = hkResult && g_pSurfaceDamageHook->hook();
2023-02-27 15:02:21 +01:00
HyprlandAPI::registerCallbackDynamic(PHANDLE, "openWindow", [&](void* self, std::any data) { onNewWindow(self, data); });
if (hkResult)
2023-09-28 00:06:14 +02:00
HyprlandAPI::addNotification(PHANDLE, "[csgo-vulkan-fix] Initialized successfully! (CS2 version)", CColor{0.2, 1.0, 0.2, 1.0}, 5000);
2023-02-27 15:02:21 +01:00
else
HyprlandAPI::addNotification(PHANDLE, "[csgo-vulkan-fix] Failure in initialization (hook failed)!", CColor{1.0, 0.2, 0.2, 1.0}, 5000);
2023-09-28 00:06:14 +02:00
return {"csgo-vulkan-fix", "A plugin to fix incorrect mouse offsets on csgo in Vulkan", "Vaxry", "1.1"};
2023-02-27 15:02:21 +01:00
}
2023-09-03 16:40:39 +02:00
APICALL EXPORT void PLUGIN_EXIT() {
;
}