VirtualMath/vmcore/ofunc/src/function.c

#include "__ofunc.h"

static void setFunctionData(Value *value, LinkValue *cls, Inter *inter) {
    value->data.function.function_data.pt_type = inter->data.default_pt_type;
    value->data.function.function_data.cls = cls;
    value->data.function.function_data.run = false;
    value->data.function.function_data.push = true;
}

ResultType function_new(O_FUNC){
    LinkValue *value = NULL;
    ArgumentParser ap[] = {{.type=only_value, .must=1, .long_arg=false},
                           {.must=-1}};
    int status = 1;
    setResultCore(result);
    arg = parserValueArgument(ap, arg, &status, NULL);
    if (status != 1){
        setResultError(E_ArgumentException, FEW_ARG, LINEFILE, true, CNEXT_NT);
        return R_error;
    }

    {  // 不使用make_new, 需要设定makeObject的set_out_var
        Inherit *object_father = getInheritFromValueCore(inter->data.base_obj[B_FUNCTION]);
        Value *new_object = makeObject(inter, NULL, NULL, false, object_father);  // 不加入out_var
        value = makeLinkValue(new_object, belong, auto_aut, inter);
        gc_freeTmpLink(&new_object->gc_status);
    }

    value->value->type = V_func;
    value->value->data.function.type = vm_func;
    value->value->data.function.function = NULL;
    value->value->data.function.pt = NULL;
    value->value->data.function.of = NULL;
    setFunctionData(value->value, ap->value, inter);

    run_init(value, arg, LINEFILE, CNEXT_NT);
    return result->type;
}

ResultType function_init(O_FUNC){
    ArgumentParser ap[] = {{.type=only_value, .must=1, .long_arg=false},
                           {.type=only_value, .must=0, .long_arg=false},
                           {.must=-1}};
    LinkValue *func;
    setResultCore(result);
    parserArgumentUnion(ap, arg, CNEXT_NT);
    if (!CHECK_RESULT(result))
        return result->type;
    freeResult(result);
    if ((func = ap[0].value)->value->type != V_func) {
        setResultError(E_TypeException, INSTANCE_ERROR(func), LINEFILE, true, CNEXT_NT);
        return R_error;
    }

    if (ap[1].value != NULL) {
        Statement *return_ = makeBaseLinkValueStatement(ap[1].value, LINEFILE);
        func->value->data.function.function = makeReturnStatement(return_, LINEFILE);
        func->value->data.function.function_data.pt_type = free_;
        func->value->data.function.type = vm_func;
    }

    setResult(result, inter);
    return result->type;
}

ResultType function_set(O_FUNC){  // 针对FFI设置vaargs
    ArgumentParser ap[] = {{.type=only_value, .must=1, .long_arg=false},
                           {.type=only_value, .must=0, .long_arg=true},
                           {.must=-1}};
    LinkValue *func;
    setResultCore(result);
    parserArgumentUnion(ap, arg, CNEXT_NT);
    if (!CHECK_RESULT(result))
        return result->type;
    freeResult(result);
    if ((func = ap[0].value)->value->type != V_func || (func = ap[0].value)->value->data.function.type != f_func) {
        setResultError(E_TypeException, INSTANCE_ERROR(func), LINEFILE, true, CNEXT_NT);
        return R_error;
    }

    if (ap[1].arg != NULL) {
        LinkValue *list;
        makeListValue(ap[1].arg, LINEFILE, L_tuple, CNEXT_NT);
        if (!CHECK_RESULT(result))
            return result->type;
        GET_RESULT(list, result);
        addAttributes(L"vaargs", false, list, LINEFILE, true, CFUNC_NT(var_list, result, func));
        gc_freeTmpLink(&list->gc_status);
    } else
        findFromVarList(L"vaargs", 0, del_var, CFUNC_CORE(var_list));

    if (CHECK_RESULT(result))
        setResultOperation(result, func);
    return result->type;
}

void registeredFunction(R_FUNC){
    LinkValue *object = inter->data.base_obj[B_FUNCTION];
    NameFunc tmp[] = {{L"set", function_set, object_free_, .var=nfv_notpush},
                      {inter->data.mag_func[M_INIT], function_init, object_free_, .var=nfv_notpush},
                      {NULL, NULL}};
    gc_addTmpLink(&object->gc_status);
    addBaseClassVar(L"func", object, belong, inter);
    iterBaseClassFunc(tmp, object, CFUNC_CORE(inter->var_list));
    gc_freeTmpLink(&object->gc_status);
}

void makeBaseFunction(Inter *inter){
    LinkValue *function = makeBaseChildClass(inter->data.base_obj[B_VOBJECT], inter);
    gc_addStatementLink(&function->gc_status);
    inter->data.base_obj[B_FUNCTION] = function;
}

void functionPresetting(LinkValue *func, Inter *inter) {  // 提前注册func_new
    Result result;
    VarList *object_var = func->value->object.var;
    LinkValue *func_new;
    setResultCore(&result);

    func_new = makeCFunctionFromOf(function_new, func, function_new, func, NULL, inter);  // var_list为NULL, 即声明为内联函数 (若不声明为内联函数则改为inter.var_list即可)
    func_new->value->data.function.function_data.pt_type = class_free_;
    addStrVar(inter->data.mag_func[M_NEW], false, true, func_new, LINEFILE, false, CFUNC_NT(object_var, &result, func));
    freeResult(&result);
    freeResult(&result);
}