add double precision matrix implementation

1 files changed, 41 insertions, 0 deletions

diff --git a/matrix/generic/mmatrix.c b/matrix/generic/mmatrix.c
new file mode 100644
index 0000000..ac71c3d
--- /dev/null
+++ b/matrix/generic/mmatrix.c
@@ -0,0 +1,41 @@
+#ifdef NERV_GENERIC_MMATRIX
+#include "matrix.h"
+#include "elem_type.h"
+#define MATRIX_DATA_FREE(ptr) free(ptr)
+#define MATRIX_DATA_ALLOC(dptr, stride, width, height) \
+                            host_matrix_(alloc)(dptr, stride, width, height)
+#define MATRIX_DATA_STRIDE(ncol) (sizeof(float) * (ncol))
+#define MATRIX_DATA_WRITE(data, idx, val) (data[idx] = val)
+#define MATRIX_DATA_READ(data, idx) (data[idx])
+#define NERV_GENERIC_MATRIX
+#include "../../common.h"
+
+const char *nerv_matrix_(tname) = "nerv.FloatMMatrix";
+
+static void host_matrix_(alloc)(float **dptr, size_t *stride,
+                                    long width, long height) {
+    *dptr = (float *)malloc(width * height);
+    *stride = width;
+}
+
+int nerv_matrix_(get_elem)(lua_State *L) {
+    Matrix *self = luaT_checkudata(L, 1, nerv_matrix_(tname));
+    int idx = luaL_checkinteger(L, 2);
+    if (idx < 0 || idx >= self->nmax)
+        nerv_error(L, "index must be within range [0, %d)", self->nmax);
+    lua_pushnumber(L, self->data.f[idx]);
+    return 1;
+}
+
+int nerv_matrix_(set_elem)(lua_State *L) {
+    Matrix *self = luaT_checkudata(L, 1, nerv_matrix_(tname));
+    int idx = luaL_checkinteger(L, 2);
+    float v = luaL_checknumber(L, 3);
+    if (idx < 0 || idx >= self->nmax)
+        nerv_error(L, "index must be within range [0, %d)", self->nmax);
+    self->data.f[idx] = v;
+    return 0;
+}
+
+#include "matrix.c"
+#endif