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
|
#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);
return 0;
}
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 *nerv_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 *nerv_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 nerv_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 nerv_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;
}
}
void nerv_hashmap_destroy(HashMap *h) {
nerv_hashmap_clear(h);
free(h);
}
size_t bkdr_hash(const char *key) {
unsigned int seed = 131;
unsigned int res = 0;
while (*key)
res = res * seed + *key++;
return res;
}
|