pikapython/package/PikaStdLib/PikaStdData_List.c

#include "BaseObj.h"
#include "PikaObj.h"
#include "dataStrs.h"

void PikaStdData_List_append(PikaObj* self, Arg* arg) {
    PikaList* list = obj_getPtr(self, "list");
    list_append(list, arg);
}

int PikaStdData_List_len(PikaObj* self) {
    PikaList* list = obj_getPtr(self, "list");
    return list_getSize(list);
}

Arg* PikaStdData_List_get(PikaObj* self, int i) {
    PikaList* list = obj_getPtr(self, "list");
    return arg_copy(list_getArg(list, i));
}

void PikaStdData_List___init__(PikaObj* self) {
    if (!obj_isArgExist(self, "list")) {
        PikaList* list = New_list();
        obj_setPtr(self, "list", list);
    }
}

void PikaStdData_List_set(PikaObj* self, Arg* arg, int i) {
    PikaList* list = obj_getPtr(self, "list");
    if (PIKA_RES_OK != list_setArg(list, i, arg)) {
        obj_setErrorCode(self, 1);
        obj_setSysOut(self, "[error]: index exceeded lengh of list.");
    }
}

Arg* PikaStdData_List___iter__(PikaObj* self) {
    obj_setInt(self, "__iter_i", 0);
    return arg_setRef(NULL, "", self);
}

Arg* PikaStdData_List___next__(PikaObj* self) {
    int __iter_i = args_getInt(self->list, "__iter_i");
    Arg* res = PikaStdData_List_get(self, __iter_i);
    if (NULL == res) {
        return arg_setNull(NULL);
    }
    args_setInt(self->list, "__iter_i", __iter_i + 1);
    return res;
}

Arg* PikaStdData_List___get__(PikaObj* self) {
    return PikaStdData_List_get(self, obj_getInt(self, "__key"));
}

void PikaStdData_List___set__(PikaObj* self) {
    PikaStdData_List_set(self, obj_getArg(self, "__val"),
                         obj_getInt(self, "__key"));
}

void PikaStdData_ByteArray_fromString(PikaObj* self, char* s) {
    for (uint32_t i = 0; i < strGetSize(s); i++) {
        obj_setInt(self, "__val", (int)s[i]);
        PIKA_PYTHON_BEGIN
        /* clang-format off */
        PIKA_PYTHON(
            append(__val)
        )
        /* clang-format on */
        const uint8_t bytes[] = {
            0x08, 0x00, /* instruct array size */
            0x10, 0x81, 0x01, 0x00, 0x00, 0x02, 0x07, 0x00, /* instruct array */
            0x0e, 0x00, /* const pool size */
            0x00, 0x5f, 0x5f, 0x76, 0x61, 0x6c, 0x00, 0x61,
            0x70, 0x70, 0x65, 0x6e, 0x64, 0x00, /* const pool */
        };
        pikaVM_runByteCode(self, (uint8_t*)bytes);
        PIKA_PYTHON_END
    }
}

void PikaStdData_List___del__(PikaObj* self) {
    Args* list = obj_getPtr(self, "list");
    args_deinit(list);
}

char* PikaStdLib_SysObj_str(PikaObj* self, Arg* arg);
char* PikaStdData_List___str__(PikaObj* self) {
    Arg* str_arg = arg_setStr(NULL, "", "[");
    PikaList* list = obj_getPtr(self, "list");

    int i = 0;
    while (PIKA_TRUE) {
        Arg* item = list_getArg(list, i);
        if (NULL == item) {
            break;
        }
        if (i != 0) {
            str_arg = arg_strAppend(str_arg, ", ");
        }
        char* item_str = PikaStdLib_SysObj_str(self, item);
        str_arg = arg_strAppend(str_arg, item_str);
        i++;
    }

    str_arg = arg_strAppend(str_arg, "]");
    obj_setStr(self, "_buf", arg_getStr(str_arg));
    arg_deinit(str_arg);
    return obj_getStr(self, "_buf");
}