#include "PikaStdTask_Task.h"
#include "BaseObj.h"
#include "PikaVM.h"

extern PikaObj* __pikaMain;
void PikaStdTask_Task___init__(PikaObj* self) {
    pikaVM_runAsm(self,
                  "B0\n"
                  "0 RUN calls_always.__init__\n"
                  "0 RUN calls_when.__init__\n"
                  "0 RUN assert_when.__init__\n");
    obj_setPtr(__pikaMain, "__calls_always", obj_getPtr(self, "calls_always"));
    obj_setPtr(__pikaMain, "__calls_when", obj_getPtr(self, "calls_when"));
    obj_setPtr(__pikaMain, "__assert_when", obj_getPtr(self, "assert_when"));
}

void PikaStdTask_Task_call_always(PikaObj* self, Arg* fun_todo) {
    obj_setArg(self, "fun_todo", fun_todo);
    pikaVM_runAsm(self,
                  "B0\n"
                  "1 REF fun_todo\n"
                  "0 RUN calls_always.append\n");
}

void PikaStdTask_Task_call_when(PikaObj* self, Arg* fun_todo, Arg* fun_when) {
    obj_setArg(self, "fun_todo", fun_todo);
    obj_setArg(self, "fun_when", fun_when);
    pikaVM_runAsm(self,
                  "B0\n"
                  "1 REF fun_todo\n"
                  "0 RUN calls_when.append\n"
                  "B0\n"
                  "1 REF fun_when\n"
                  "0 RUN assert_when.append\n");
}

void PikaStdTask_Task_run_once(PikaObj* self) {
    /* reference the calls_always in __pikaMain */
    pikaVM_runAsm(__pikaMain,
                  "B0\n"
                  "1 REF __calls_always\n"
                  "0 RUN iter\n"
                  "0 OUT _l0\n"
                  "B0\n"
                  "0 RUN _l0.__next__\n"
                  "0 OUT fun_todo\n"
                  "0 EST fun_todo\n"
                  "0 JEZ 2\n"
                  "B1\n"
                  "0 RUN fun_todo\n"
                  "B0\n"
                  "0 JMP -1\n"
                  "B0\n"
                  "0 DEL _l0\n"
                  "B0\n");
    obj_run(__pikaMain,
            "for i in range(0, __calls_when.len()):\n"
            "    if __assert_when[i]():\n"
            "        __calls_when()\n");
}

void PikaStdTask_Task_run_always(PikaObj* self) {
    while (1) {
        PikaStdTask_Task_run_once(self);
    }
}