DLL_EXPORT int luaopen_libjpeg(lua_State *L) { torch_FloatTensor_id = luaT_checktypename2id(L, "torch.FloatTensor"); torch_DoubleTensor_id = luaT_checktypename2id(L, "torch.DoubleTensor"); libjpeg_FloatMain_init(L); libjpeg_DoubleMain_init(L); luaL_register(L, "libjpeg.double", libjpeg_DoubleMain__); luaL_register(L, "libjpeg.float", libjpeg_FloatMain__); return 1; }
DLL_EXPORT int luaopen_libjpeg(lua_State *L) { libjpeg_FloatMain_init(L); libjpeg_DoubleMain_init(L); libjpeg_ByteMain_init(L); lua_newtable(L); lua_pushvalue(L, -1); lua_setglobal(L, "libjpeg"); lua_newtable(L); luaT_setfuncs(L, libjpeg_DoubleMain__, 0); lua_setfield(L, -2, "double"); lua_newtable(L); luaT_setfuncs(L, libjpeg_FloatMain__, 0); lua_setfield(L, -2, "float"); lua_newtable(L); luaT_setfuncs(L, libjpeg_ByteMain__, 0); lua_setfield(L, -2, "byte"); return 1; }