use the standard nerv error reporting in io lib

1 files changed, 108 insertions, 0 deletions

diff --git a/nerv/lib/common.c b/nerv/lib/common.c
new file mode 100644
index 0000000..db667b2
--- /dev/null
+++ b/nerv/lib/common.c
@@ -0,0 +1,108 @@
+#include "common.h"
+#include <stdarg.h>
+int nerv_error(lua_State *L, const char *err_mesg_fmt, ...) {
+    va_list ap;
+    va_start(ap, err_mesg_fmt);
+    lua_pushstring(L, "[nerv] internal error: ");
+    lua_pushvfstring(L, err_mesg_fmt, ap); 
+    lua_concat(L, 2);
+    lua_error(L);
+    va_end(ap);
+    return 0;
+}
+
+int nerv_error_status(lua_State *L, Status *status) {
+    const char *mmesg = NULL;
+    switch (status->err_code)
+    {
+        case MAT_GENERAL_ERR: mmesg = "general error"; break;
+        case MAT_INSUF_MEM: mmesg = "insufficient memory"; break;
+        case MAT_INVALID_FORMAT: mmesg = "invalid matrix format"; break;
+        case MAT_WRITE_ERROR: mmesg = "error while writing matrix"; break;
+        case MAT_INVALID_COPY_INTERVAL: mmesg = "invalid copy interval"; break;
+        case MAT_MISMATCH_DIM: mmesg = "mismatching matrix dimension"; break;
+        case MAT_WRONG_MULT_DIM: mmesg = "wrong multipier dimension"; break;
+        case MAT_ROW_VECTOR_EXP: mmesg = "row vector expected"; break;
+        case MAT_COL_VECTOR_EXP: mmesg = "column vector expected"; break;
+        case MAT_IDX_VECTOR_EXP: mmesg = "index vector expected"; break;
+        case MAT_INVALID_IDX: mmesg = "invalid index"; break;
+        case MAT_CUDA_ERR: mmesg = "cuda error"; break;
+        case MAT_CUBLAS_ERR: mmesg = "cublas error"; break;
+        case CF_INVALID_FORMAT: mmesg = "invalid format"; break;
+        case CF_END_OF_FILE: mmesg = "unexpected end of file"; break;
+        case CF_SECTION_OVERFLOW: mmesg = "section overflow"; break;
+        case CF_WRITE_ERROR: mmesg = "error while writing"; break;
+        case CF_ERR_OPEN_FILE: mmesg = "error while opening file"; break;
+        case CF_INVALID_OP: mmesg = "invalid operation"; break;
+        default: mmesg = "unknown"; break;
+    }
+    if (status->msg)
+        nerv_error(L, "%s: %s @%s:%d", mmesg, status->msg,
+                                        status->file, status->lineno);
+    else
+        nerv_error(L, "%s @%s:%d", mmesg, status->file, status->lineno);
+}
+
+int nerv_error_method_not_implemented(lua_State *L) {
+    return nerv_error(L, "method not implemented"); 
+}
+
+void luaN_append_methods(lua_State *L, const luaL_Reg *mlist) {
+    for (; mlist->func; mlist++)
+    {
+        lua_pushcfunction(L, mlist->func);
+        lua_setfield(L, -2, mlist->name);
+    }
+}
+
+HashMap *hashmap_create(size_t size, HashKey_t hfunc, HashMapCmp_t cmp) {
+    HashMap *res = (HashMap *)malloc(sizeof(HashMap));
+    res->bucket = calloc(size, sizeof(HashNode));
+    res->cmp = cmp;
+    res->hfunc = hfunc;
+    res->size = size;
+    return res;
+}
+
+void *hashmap_getval(HashMap *h, const char *key) {
+    size_t idx = h->hfunc(key) % h->size;
+    HashNode *ptr;
+    for (ptr = h->bucket[idx]; ptr; ptr = ptr->next)
+    {
+        if (!h->cmp(ptr->key, key))
+            return ptr->val;
+    }
+    return NULL;
+}
+
+void hashmap_setval(HashMap *h, const char *key, void *val) {
+    size_t idx = h->hfunc(key) % h->size;
+    HashNode *ptr = malloc(sizeof(HashNode));
+    ptr->next = h->bucket[idx];
+    h->bucket[idx] = ptr;
+    ptr->key = key;
+    ptr->val = val;
+}
+
+void hashmap_clear(HashMap *h) {
+    size_t i;
+    for (i = 0; i < h->size; i++)
+    {
+        HashNode *ptr, *nptr;
+        for (ptr = h->bucket[i]; ptr; ptr = nptr)
+        {
+            nptr = ptr->next;
+            free(ptr->val);
+            free(ptr);
+        }
+        h->bucket[i] = NULL;
+    }
+}
+
+size_t bkdr_hash(const char *key) {
+    unsigned int seed = 131;
+    unsigned int res = 0;
+    while (*key)
+        res = res * seed + *key++;
+    return res;
+}