1 files changed, 25 insertions, 0 deletions
diff --git a/src/libultra_code_O2/osAiSetNextBuffer.c b/src/libultra_code_O2/osAiSetNextBuffer.c
new file mode 100644
index 000000000..0669ee506
--- /dev/null
+++ b/src/libultra_code_O2/osAiSetNextBuffer.c
@@ -0,0 +1,25 @@
+#include "global.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;
+}