main.c

#include <vitasdk.h>
#include <taihen.h>

#include "osd.h"
#include "main.h"

#define VGi_VERSION  "v0.7"

#define HOOKS_NUM 35
SceUID g_hooks[HOOKS_NUM] = {0};
tai_hook_ref_t g_hookrefs[HOOKS_NUM] = {0};

#define BUTTONS_FAST_MOVE_DELAY 500000
uint8_t g_menuScroll = 0;
uint8_t g_menuVisible = 0;
uint8_t g_vsyncKillswitch = 0;
static VGi_MenuSection g_menuSection = MENU_APP_INFO;
static unsigned int g_buttonsPressed = 0;
static SceUInt32 g_buttonsLastChange = 0;

static void checkButtons() {
    SceCtrlData ctrl;
    sceCtrlPeekBufferPositive(0, &ctrl, 1);
    unsigned int buttons = ctrl.buttons & ~g_buttonsPressed;
    SceUInt32 time_now = sceKernelGetProcessTimeLow();

    if (ctrl.buttons & SCE_CTRL_SELECT) {
        // Toggle menu
        if (buttons & SCE_CTRL_LTRIGGER) {
            g_menuVisible = !g_menuVisible;
        }
    } else if (g_menuVisible) {
        // Move between sections
        if (buttons & SCE_CTRL_LTRIGGER) {
            if (g_menuSection > MENU_APP_INFO) {
                g_menuSection--;
                g_menuScroll = 0;
            }
        } else if (buttons & SCE_CTRL_RTRIGGER) {
            if (g_menuSection < MENU_MAX - 1) {
                g_menuSection++;
                g_menuScroll = 0;
            }
        } else if (buttons & SCE_CTRL_UP) {
            if (g_menuScroll > 0)
                g_menuScroll--;
            g_buttonsLastChange = time_now;
        } else if (buttons & SCE_CTRL_DOWN) {
            if (g_menuScroll < 255)
                g_menuScroll++;
            g_buttonsLastChange = time_now;
        }
        // Fast scroll UP
        else if (ctrl.buttons & SCE_CTRL_UP &&
                time_now - g_buttonsLastChange > BUTTONS_FAST_MOVE_DELAY) {
            if (g_menuScroll > 0)
                g_menuScroll--;
        }
        // Fast scroll DOWN
        else if (ctrl.buttons & SCE_CTRL_DOWN &&
                time_now - g_buttonsLastChange > BUTTONS_FAST_MOVE_DELAY) {
            if (g_menuScroll < 255)
                g_menuScroll++;
        }
    }

    // Vsync Killswitch
    if (g_menuVisible && g_menuSection == MENU_GRAPHICS) {
        if (buttons & SCE_CTRL_CIRCLE) {
            if (g_vsyncKillswitch < 2)
                g_vsyncKillswitch++;
            else
                g_vsyncKillswitch = 0;
        }
    }

    g_buttonsPressed = ctrl.buttons;
}

static void drawNextSectionIndicator(const SceDisplayFrameBuf *pParam, const char *L, const char *R) {
    char buf[64];

    snprintf(buf, 64, "%s%s", strlen(L) > 0 ? "< " : "", L);
    osdDrawStringF(5, pParam->height - 22 - 5, "%s", buf);

    snprintf(buf, 64, "%s%s", R, strlen(R) > 0 ? " >" : "");
    osdDrawStringF(pParam->width - osdGetTextWidth(buf) - 5, pParam->height - 22 - 5, "%s", buf);
}

static int sceDisplaySetFrameBuf_patched(const SceDisplayFrameBuf *pParam, int sync) {

    // Check for pressed buttons
    checkButtons();

    if (!g_menuVisible)
        goto CONT;

    osdUpdateFrameBuf(pParam);

    // BG
    osdSetBgColor(0, 0, 0, 225);
    osdFastDrawRectangle(0, 0, pParam->width, pParam->height);

    // Watermark
    osdSetBgColor(0, 0, 0, 0);
    osdSetTextColor(10, 20, 50, 255);
    osdSetTextScale(pParam->width == 640 ? 3 : 5);
    osdDrawStringF(0, pParam->height - 115, "VGi %s", VGi_VERSION);
    osdSetTextScale(2);
    osdDrawStringF(pParam->width - osdGetTextWidth("by Electry"), pParam->height - 71, "by Electry");

    // Section content
    osdSetTextColor(255, 255, 255, 255);

    switch (g_menuSection) {
        case MENU_APP_INFO:
            drawAppInfoMenu(pParam);
            drawNextSectionIndicator(pParam, "", MENU_TITLE_GRAPHICS);
            break;
        case MENU_GRAPHICS:
            drawGraphicsMenu(pParam);
            drawNextSectionIndicator(pParam, MENU_TITLE_APP_INFO, MENU_TITLE_GRAPHICS_2);
            break;
        case MENU_GRAPHICS_2:
            drawGraphics2Menu(pParam);
            drawNextSectionIndicator(pParam, MENU_TITLE_GRAPHICS, MENU_TITLE_GRAPHICS_3);
            break;
        case MENU_GRAPHICS_3:
            drawGraphics3Menu(pParam);
            drawNextSectionIndicator(pParam, MENU_TITLE_GRAPHICS_2, MENU_TITLE_GRAPHICS_4);
            break;
        case MENU_GRAPHICS_4:
            drawGraphics4Menu(pParam);
            drawNextSectionIndicator(pParam, MENU_TITLE_GRAPHICS_3, MENU_TITLE_GRAPHICS_5);
            break;
        case MENU_GRAPHICS_5:
            drawGraphics5Menu(pParam);
            drawNextSectionIndicator(pParam, MENU_TITLE_GRAPHICS_4, MENU_TITLE_MEMORY);
            break;
        case MENU_MEMORY:
            drawMemoryMenu(pParam);
            drawNextSectionIndicator(pParam, MENU_TITLE_GRAPHICS_5, MENU_TITLE_THREADS);
            break;
        case MENU_THREADS:
            drawThreadsMenu(pParam);
            drawNextSectionIndicator(pParam, MENU_TITLE_MEMORY, MENU_TITLE_MISC);
            break;
        case MENU_MISC:
            drawMiscMenu(pParam);
            drawNextSectionIndicator(pParam, MENU_TITLE_THREADS, MENU_TITLE_DEVICE);
            break;
        case MENU_DEVICE:
            drawDeviceMenu(pParam);
            drawNextSectionIndicator(pParam, MENU_TITLE_MISC, "");
            break;
        case MENU_MAX:
            break;
    }

    sceDisplayWaitSetFrameBufMulti(4);
CONT:
    return TAI_CONTINUE(int, g_hookrefs[0], pParam, sync);
}

void _start() __attribute__ ((weak, alias ("module_start")));
int module_start(SceSize argc, const void *args) {

    g_hooks[0] = taiHookFunctionImport(
                    &g_hookrefs[0],
                    TAI_MAIN_MODULE,
                    TAI_ANY_LIBRARY,
                    0x7A410B64,
                    sceDisplaySetFrameBuf_patched);

    // Setup hooks, etc...
    setupAppInfoMenu();
    setupGraphicsMenu();
    setupMemoryMenu();
    setupThreadsMenu();
    setupMiscMenu();
    setupDeviceMenu();

    return SCE_KERNEL_START_SUCCESS;
}

int module_stop(SceSize argc, const void *args) {

    // Release all hooks
    for (int i = 0; i < HOOKS_NUM; i++) {
        if (g_hooks[i] >= 0)
            taiHookRelease(g_hooks[i], g_hookrefs[i]);
    }

    return SCE_KERNEL_STOP_SUCCESS;
}