diff options
author | Vsevolod Stakhov <vsevolod@rspamd.com> | 2024-03-18 18:56:33 +0000 |
---|---|---|
committer | Vsevolod Stakhov <vsevolod@rspamd.com> | 2024-03-18 18:56:33 +0000 |
commit | 14c13854d3cae9d93c3d148be30fb72f1eaffe55 (patch) | |
tree | 7b1a3e41b75490fac4d45722c90a1847543c6796 /src/lua/lua_tensor.c | |
parent | 6b2b4167187fee09365271cca182866ecb029af3 (diff) | |
download | rspamd-14c13854d3cae9d93c3d148be30fb72f1eaffe55.tar.gz rspamd-14c13854d3cae9d93c3d148be30fb72f1eaffe55.zip |
[Rework] Further types conversion (no functional changes)
Diffstat (limited to 'src/lua/lua_tensor.c')
-rw-r--r-- | src/lua/lua_tensor.c | 90 |
1 files changed, 45 insertions, 45 deletions
diff --git a/src/lua/lua_tensor.c b/src/lua/lua_tensor.c index d4ee3dcca..fef1576e3 100644 --- a/src/lua/lua_tensor.c +++ b/src/lua/lua_tensor.c @@ -78,7 +78,7 @@ lua_newtensor(lua_State *L, int ndims, const int *dim, bool zero_fill, bool own) res->ndims = ndims; res->size = 1; - for (guint i = 0; i < ndims; i++) { + for (unsigned int i = 0; i < ndims; i++) { res->size *= dim[i]; res->dim[i] = dim[i]; } @@ -106,15 +106,15 @@ lua_newtensor(lua_State *L, int ndims, const int *dim, bool zero_fill, bool own) * Creates a new zero filled tensor with the specific number of dimensions * @return */ -static gint +static int lua_tensor_new(lua_State *L) { - gint ndims = luaL_checkinteger(L, 1); + int ndims = luaL_checkinteger(L, 1); if (ndims > 0 && ndims <= 2) { - gint *dims = g_alloca(sizeof(gint) * ndims); + int *dims = g_alloca(sizeof(int) * ndims); - for (guint i = 0; i < ndims; i++) { + for (unsigned int i = 0; i < ndims; i++) { dims[i] = lua_tointeger(L, i + 2); } @@ -132,7 +132,7 @@ lua_tensor_new(lua_State *L) * Creates a new zero filled tensor with the specific number of dimensions * @return */ -static gint +static int lua_tensor_fromtable(lua_State *L) { if (lua_istable(L, 1)) { @@ -141,14 +141,14 @@ lua_tensor_fromtable(lua_State *L) if (lua_isnumber(L, -1)) { lua_pop(L, 1); /* Input vector */ - gint dims[2]; + int dims[2]; dims[0] = 1; dims[1] = rspamd_lua_table_size(L, 1); struct rspamd_lua_tensor *res = lua_newtensor(L, 2, dims, false, true); - for (guint i = 0; i < dims[1]; i++) { + for (unsigned int i = 0; i < dims[1]; i++) { lua_rawgeti(L, 1, i + 1); res->data[i] = lua_tonumber(L, -1); lua_pop(L, 1); @@ -159,10 +159,10 @@ lua_tensor_fromtable(lua_State *L) lua_pop(L, 1); /* Calculate the overall size */ - gint nrows = rspamd_lua_table_size(L, 1), ncols = 0; - gint err; + int nrows = rspamd_lua_table_size(L, 1), ncols = 0; + int err; - for (gint i = 0; i < nrows; i++) { + for (int i = 0; i < nrows; i++) { lua_rawgeti(L, 1, i + 1); if (ncols == 0) { @@ -180,7 +180,7 @@ lua_tensor_fromtable(lua_State *L) } else { if (ncols != rspamd_lua_table_size(L, -1)) { - gint t = rspamd_lua_table_size(L, -1); + int t = rspamd_lua_table_size(L, -1); lua_pop(L, 1); err = luaL_error(L, "invalid params at pos %d: " @@ -196,17 +196,17 @@ lua_tensor_fromtable(lua_State *L) lua_pop(L, 1); } - gint dims[2]; + int dims[2]; dims[0] = nrows; dims[1] = ncols; struct rspamd_lua_tensor *res = lua_newtensor(L, 2, dims, false, true); - for (gint i = 0; i < nrows; i++) { + for (int i = 0; i < nrows; i++) { lua_rawgeti(L, 1, i + 1); - for (gint j = 0; j < ncols; j++) { + for (int j = 0; j < ncols; j++) { lua_rawgeti(L, -1, j + 1); res->data[i * ncols + j] = lua_tonumber(L, -1); @@ -235,7 +235,7 @@ lua_tensor_fromtable(lua_State *L) * Tensor destructor * @return */ -static gint +static int lua_tensor_destroy(lua_State *L) { struct rspamd_lua_tensor *t = lua_check_tensor(L, 1); @@ -254,11 +254,11 @@ lua_tensor_destroy(lua_State *L) * Tensor serialisation function * @return */ -static gint +static int lua_tensor_save(lua_State *L) { struct rspamd_lua_tensor *t = lua_check_tensor(L, 1); - gint size; + int size; if (t) { if (t->size > 0) { @@ -268,8 +268,8 @@ lua_tensor_save(lua_State *L) size = -(t->size); } - gsize sz = sizeof(gint) * 4 + size * sizeof(rspamd_tensor_num_t); - guchar *data; + gsize sz = sizeof(int) * 4 + size * sizeof(rspamd_tensor_num_t); + unsigned char *data; struct rspamd_lua_text *out = lua_new_text(L, NULL, 0, TRUE); @@ -280,7 +280,7 @@ lua_tensor_save(lua_State *L) memcpy(data + 4 * sizeof(int), t->data, size * sizeof(rspamd_tensor_num_t)); - out->start = (const gchar *) data; + out->start = (const char *) data; out->len = sz; } else { @@ -290,7 +290,7 @@ lua_tensor_save(lua_State *L) return 1; } -static gint +static int lua_tensor_tostring(lua_State *L) { struct rspamd_lua_tensor *t = lua_check_tensor(L, 1); @@ -300,15 +300,15 @@ lua_tensor_tostring(lua_State *L) if (t->ndims == 1) { /* Print as a vector */ - for (gint i = 0; i < t->dim[0]; i++) { + for (int i = 0; i < t->dim[0]; i++) { rspamd_printf_gstring(out, "%.4f ", t->data[i]); } /* Trim last space */ out->len--; } else { - for (gint i = 0; i < t->dim[0]; i++) { - for (gint j = 0; j < t->dim[1]; j++) { + for (int i = 0; i < t->dim[0]; i++) { + for (int j = 0; j < t->dim[1]; j++) { rspamd_printf_gstring(out, "%.4f ", t->data[i * t->dim[1] + j]); } @@ -331,11 +331,11 @@ lua_tensor_tostring(lua_State *L) return 1; } -static gint +static int lua_tensor_index(lua_State *L) { struct rspamd_lua_tensor *t = lua_check_tensor(L, 1); - gint idx; + int idx; if (t) { if (lua_isnumber(L, 2)) { @@ -352,7 +352,7 @@ lua_tensor_index(lua_State *L) } else { /* Push row */ - gint dim = t->dim[1]; + int dim = t->dim[1]; if (idx <= t->dim[0]) { @@ -376,11 +376,11 @@ lua_tensor_index(lua_State *L) return 1; } -static gint +static int lua_tensor_newindex(lua_State *L) { struct rspamd_lua_tensor *t = lua_check_tensor(L, 1); - gint idx; + int idx; if (t) { if (lua_isnumber(L, 2)) { @@ -451,7 +451,7 @@ lua_tensor_newindex(lua_State *L) * Multiply two tensors (optionally transposed) and return a new tensor * @return */ -static gint +static int lua_tensor_mul(lua_State *L) { struct rspamd_lua_tensor *t1 = lua_check_tensor(L, 1), @@ -467,7 +467,7 @@ lua_tensor_mul(lua_State *L) } if (t1 && t2) { - gint dims[2], shadow_dims[2]; + int dims[2], shadow_dims[2]; dims[0] = abs(transA ? t1->dim[1] : t1->dim[0]); shadow_dims[0] = abs(transB ? t2->dim[1] : t2->dim[0]); dims[1] = abs(transB ? t2->dim[0] : t2->dim[1]); @@ -517,10 +517,10 @@ lua_tensor_mul(lua_State *L) * Deserialize tensor * @return */ -static gint +static int lua_tensor_load(lua_State *L) { - const guchar *data; + const unsigned char *data; gsize sz; if (lua_type(L, 1) == LUA_TUSERDATA) { @@ -530,14 +530,14 @@ lua_tensor_load(lua_State *L) return luaL_error(L, "invalid argument"); } - data = (const guchar *) t->start; + data = (const unsigned char *) t->start; sz = t->len; } else { - data = (const guchar *) lua_tolstring(L, 1, &sz); + data = (const unsigned char *) lua_tolstring(L, 1, &sz); } - if (sz >= sizeof(gint) * 4) { + if (sz >= sizeof(int) * 4) { int ndims, nelts, dims[2]; memcpy(&ndims, data, sizeof(int)); @@ -582,11 +582,11 @@ lua_tensor_load(lua_State *L) return 1; } -static gint +static int lua_tensor_len(lua_State *L) { struct rspamd_lua_tensor *t = lua_check_tensor(L, 1); - gint nret = 1; + int nret = 1; if (t) { /* Return the main dimension first */ @@ -606,7 +606,7 @@ lua_tensor_len(lua_State *L) return nret; } -static gint +static int lua_tensor_eigen(lua_State *L) { struct rspamd_lua_tensor *t = lua_check_tensor(L, 1), *eigen; @@ -638,7 +638,7 @@ mean_vec(rspamd_tensor_num_t *x, gsize n) return sum / (rspamd_tensor_num_t) n; } -static gint +static int lua_tensor_mean(lua_State *L) { struct rspamd_lua_tensor *t = lua_check_tensor(L, 1); @@ -666,7 +666,7 @@ lua_tensor_mean(lua_State *L) return 1; } -static gint +static int lua_tensor_transpose(lua_State *L) { struct rspamd_lua_tensor *t = lua_check_tensor(L, 1), *res; @@ -707,7 +707,7 @@ lua_tensor_transpose(lua_State *L) return 1; } -static gint +static int lua_tensor_has_blas(lua_State *L) { #ifdef HAVE_CBLAS @@ -719,7 +719,7 @@ lua_tensor_has_blas(lua_State *L) return 1; } -static gint +static int lua_tensor_scatter_matrix(lua_State *L) { struct rspamd_lua_tensor *t = lua_check_tensor(L, 1), *res; @@ -789,7 +789,7 @@ lua_tensor_scatter_matrix(lua_State *L) return 1; } -static gint +static int lua_load_tensor(lua_State *L) { lua_newtable(L); |