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
|
#include <TH.h>
#include <luaT.h>
#include <jpeglib.h>
#include <setjmp.h>
#if LUA_VERSION_NUM >= 503
#define luaL_checkint(L,n) ((int)luaL_checkinteger(L, (n)))
#endif
#define torch_(NAME) TH_CONCAT_3(torch_, Real, NAME)
#define torch_Tensor TH_CONCAT_STRING_3(torch., Real, Tensor)
#define libjpeg_(NAME) TH_CONCAT_3(libjpeg_, Real, NAME)
static void
jpeg_mem_src_dummy(j_decompress_ptr c, unsigned char *ibuf, unsigned long isiz)
{
}
static void
jpeg_mem_dest_dummy(j_compress_ptr c, unsigned char **obuf, unsigned long *osiz)
{
}
#define JPEG_MEM_SRC_NOT_DEF "`jpeg_mem_src` is not defined."
#define JPEG_MEM_DEST_NOT_DEF "`jpeg_mem_dest` is not defined."
#define JPEG_REQUIRED_VERSION " Use libjpeg v8+, libjpeg-turbo 1.3+ or build" \
" libjpeg-turbo with `--with-mem-srcdst`."
#define JPEG_MEM_SRC_ERR_MSG JPEG_MEM_SRC_NOT_DEF JPEG_REQUIRED_VERSION
#define JPEG_MEM_DEST_ERR_MSG JPEG_MEM_DEST_NOT_DEF JPEG_REQUIRED_VERSION
#if !defined(HAVE_JPEG_MEM_SRC)
#define jpeg_mem_src jpeg_mem_src_dummy
#endif
#if !defined(HAVE_JPEG_MEM_DEST)
#define jpeg_mem_dest jpeg_mem_dest_dummy
#endif
#include "generic/jpeg.c"
#include "THGenerateAllTypes.h"
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;
}
|
