utils.c

#include "utils.h"

THLongStorage* cutorch_checklongargs(lua_State *L, int index) {
    THLongStorage *storage;
    int i;
    int narg = lua_gettop(L)-index+1;

    if(narg == 1 && luaT_toudata(L, index, "torch.LongStorage")) {
        THLongStorage *storagesrc = luaT_toudata(L, index, "torch.LongStorage");
        storage = THLongStorage_newWithSize(storagesrc->size);
        THLongStorage_copy(storage, storagesrc);
    } else {
        storage = THLongStorage_newWithSize(narg);
        for(i = index; i < index+narg; i++) {
            if(!lua_isnumber(L, i)) {
                THLongStorage_free(storage);
                luaL_argerror(L, i, "number expected");
            }
            THLongStorage_set(storage, i-index, lua_tonumber(L, i));
        }
    }
    return storage;
}

int cutorch_islongargs(lua_State *L, int index) {
    int narg = lua_gettop(L)-index+1;

    if(narg == 1 && luaT_toudata(L, index, "torch.LongStorage")) {
        return 1;
    } else {
        int i;

        for(i = index; i < index+narg; i++) {
            if(!lua_isnumber(L, i))
                return 0;
        }
        return 1;
    }
    return 0;
}

struct THCState* cutorch_getstate(lua_State* L) {
    lua_getglobal(L, "cutorch");
    lua_getfield(L, -1, "_state");
    struct THCState *state = lua_touserdata(L, -1);
    lua_pop(L, 2);
    return state;
}