#include <global.h>
#include <ultra64/hardware.h>

s32 osAiSetNextBuffer(void* buf, u32 size) {
    static u8 D_80130500 = false;
    u32 bufAdjusted = (u32)buf;
    s32 status;

    if (D_80130500) {
        bufAdjusted = (u32)buf - 0x2000;
    }
    if ((((u32)buf + size) & 0x1FFF) == 0) {
        D_80130500 = true;
    } else {
        D_80130500 = false;
    }

    status = HW_REG(AI_STATUS_REG, s32);
    if (status & AI_STATUS_AI_FULL) {
        return -1;
    }

    HW_REG(AI_DRAM_ADDR_REG, u32) = PHYSICAL_TO_VIRTUAL(bufAdjusted);
    HW_REG(AI_LEN_REG, u32) = size;
    return 0;
}