liba 0.1.15

An algorithm library based on C/C++
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160

#include "a.h"
#include "a/tf.h"

static JSClassID liba_tf_class_id;

static void liba_tf_finalizer(JSRuntime *rt, JSValue val)
{
    a_tf *const self = (a_tf *)JS_GetOpaque(val, liba_tf_class_id);
    js_free_rt(rt, self->output);
    js_free_rt(rt, self->input);
    js_free_rt(rt, self);
}

static JSClassDef liba_tf_class = {"tf", .finalizer = liba_tf_finalizer};

static int liba_tf_set_num_(JSContext *ctx, a_tf *self, JSValueConst num)
{
    a_u32 num_n = 0;
    a_float *num_p = self->input;
    int ret = js_array_length(ctx, num, &num_n);
    if (ret) { return ret; }
    if (num_n > self->num_n)
    {
        num_p = (a_float *)js_realloc(ctx, num_p, sizeof(a_float) * num_n * 2);
        if (!num_p) { return ~0; }
        self->input = num_p;
        num_p += num_n;
        self->num_p = num_p;
    }
    else { num_p += self->num_n; }
    self->num_n = (unsigned int)num_n;
    a_zero(self->input, sizeof(a_float) * num_n);
    return js_array_num_get(ctx, num, num_p, num_n);
}

static int liba_tf_set_den_(JSContext *ctx, a_tf *self, JSValueConst den)
{
    a_u32 den_n = 0;
    a_float *den_p = self->output;
    int ret = js_array_length(ctx, den, &den_n);
    if (ret) { return ret; }
    if (den_n > self->den_n)
    {
        den_p = (a_float *)js_realloc(ctx, den_p, sizeof(a_float) * den_n * 2);
        if (!den_p) { return ~0; }
        self->output = den_p;
        den_p += den_n;
        self->den_p = den_p;
    }
    else { den_p += self->den_n; }
    self->den_n = (unsigned int)den_n;
    a_zero(self->output, sizeof(a_float) * den_n);
    return js_array_num_get(ctx, den, den_p, den_n);
}

static JSValue liba_tf_ctor(JSContext *ctx, JSValueConst new_target, int argc, JSValueConst *argv)
{
    (void)argc;
    JSValue proto, clazz = JS_UNDEFINED;
    a_tf *const self = (a_tf *)js_mallocz(ctx, sizeof(a_tf));
    if (!self) { return JS_EXCEPTION; }
    if (JS_IsObject(argv[0]) && liba_tf_set_num_(ctx, self, argv[0])) { goto fail; }
    if (JS_IsObject(argv[1]) && liba_tf_set_den_(ctx, self, argv[1])) { goto fail; }
    proto = JS_GetPropertyStr(ctx, new_target, "prototype");
    if (JS_IsException(proto)) { goto fail; }
    clazz = JS_NewObjectProtoClass(ctx, proto, liba_tf_class_id);
    JS_FreeValue(ctx, proto);
    if (JS_IsException(clazz)) { goto fail; }
    JS_SetOpaque(clazz, self);
    return clazz;
fail:
    js_free(ctx, self);
    JS_FreeValue(ctx, clazz);
    return JS_EXCEPTION;
}

static JSValue liba_tf_iter(JSContext *ctx, JSValueConst this_val, int argc, JSValueConst *argv)
{
    (void)argc;
    a_tf *const self = (a_tf *)JS_GetOpaque2(ctx, this_val, liba_tf_class_id);
    if (!self) { return JS_EXCEPTION; }
    double x;
    if (JS_ToFloat64(ctx, &x, argv[0])) { return JS_EXCEPTION; }
    return JS_NewFloat64(ctx, (double)a_tf_iter(self, (a_float)x));
}

static JSValue liba_tf_zero(JSContext *ctx, JSValueConst this_val, int argc, JSValueConst *argv)
{
    (void)argc;
    (void)argv;
    a_tf *const self = (a_tf *)JS_GetOpaque2(ctx, this_val, liba_tf_class_id);
    if (!self) { return JS_EXCEPTION; }
    a_tf_zero(self);
    return JS_UNDEFINED;
}

enum
{
    self_coef,
    self_bias,
    self_input,
    self_output,
};

static JSValue liba_tf_get(JSContext *ctx, JSValueConst this_val, int magic)
{
    a_tf *const self = (a_tf *)JS_GetOpaque2(ctx, this_val, liba_tf_class_id);
    if (!self) { return JS_EXCEPTION; }
    switch (magic)
    {
    case self_coef: return js_array_num_new(ctx, self->num_p, self->num_n);
    case self_bias: return js_array_num_new(ctx, self->den_p, self->den_n);
    case self_input: return js_array_num_new(ctx, self->input, self->num_n);
    case self_output: return js_array_num_new(ctx, self->output, self->den_n);
    default: return JS_UNDEFINED;
    }
}

static JSValue liba_tf_set(JSContext *ctx, JSValueConst this_val, JSValueConst val, int magic)
{
    a_tf *const self = (a_tf *)JS_GetOpaque2(ctx, this_val, liba_tf_class_id);
    if (!self) { return JS_EXCEPTION; }
    switch (magic)
    {
    case self_coef:
        if (JS_IsObject(val)) { liba_tf_set_num_(ctx, self, val); }
        break;
    case self_bias:
        if (JS_IsObject(val)) { liba_tf_set_den_(ctx, self, val); }
        break;
    default:
        break;
    }
    return JS_UNDEFINED;
}

static JSCFunctionListEntry const liba_tf_proto[] = {
    JS_PROP_STRING_DEF("[Symbol.toStringTag]", "a.tf", 0),
    JS_CGETSET_MAGIC_DEF("num", liba_tf_get, liba_tf_set, self_coef),
    JS_CGETSET_MAGIC_DEF("den", liba_tf_get, liba_tf_set, self_bias),
    JS_CGETSET_MAGIC_DEF("input", liba_tf_get, liba_tf_set, self_input),
    JS_CGETSET_MAGIC_DEF("output", liba_tf_get, liba_tf_set, self_output),
    JS_CFUNC_DEF("iter", 1, liba_tf_iter),
    JS_CFUNC_DEF("zero", 0, liba_tf_zero),
};

int js_liba_tf_init(JSContext *ctx, JSModuleDef *m)
{
    JS_NewClassID(&liba_tf_class_id);
    JS_NewClass(JS_GetRuntime(ctx), liba_tf_class_id, &liba_tf_class);

    JSValue const proto = JS_NewObject(ctx);
    JS_SetPropertyFunctionList(ctx, proto, liba_tf_proto, A_LEN(liba_tf_proto));

    JSValue const clazz = JS_NewCFunction2(ctx, liba_tf_ctor, "tf", 2, JS_CFUNC_constructor, 0);
    JS_SetConstructor(ctx, clazz, proto);
    JS_SetClassProto(ctx, liba_tf_class_id, proto);

    return JS_SetModuleExport(ctx, m, "tf", clazz);
}