示例#1
0
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;
}
示例#2
0
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;
}