diff options
Diffstat (limited to 'source/luametatex/source/luaoptional/lmtgraphicsmagick.c')
| -rw-r--r-- | source/luametatex/source/luaoptional/lmtgraphicsmagick.c | 199 |
1 files changed, 199 insertions, 0 deletions
diff --git a/source/luametatex/source/luaoptional/lmtgraphicsmagick.c b/source/luametatex/source/luaoptional/lmtgraphicsmagick.c new file mode 100644 index 000000000..c71c68c8c --- /dev/null +++ b/source/luametatex/source/luaoptional/lmtgraphicsmagick.c @@ -0,0 +1,199 @@ +/* + See license.txt in the root of this project. +*/ + +/* For now just a simple conversion, like in the example module. */ + +# include "luametatex.h" +# include "lmtoptional.h" + +typedef enum gmlib_NoiseType { + UniformNoise, + GaussianNoise, + MultiplicativeGaussianNoise, + ImpulseNoise, + LaplacianNoise, + PoissonNoise, + RandomNoise, + UndefinedNoise +} gmlib_NoiseType; + +typedef struct gmlib_state_info { + + int initialized; + int padding; + + void (*gm_InitializeMagick) ( + // void **argv + void *path + ); + + void (*gm_DestroyMagick) ( + void + ); + + void * (*gm_NewMagickWand) ( + void + ); + + void (*gm_DestroyMagickWand) ( + void *wand + ); + + int (*gm_MagickReadImage) ( + void *wand, + const char *name + ); + + int (*gm_MagickWriteImage) ( + void *wand, + const char *name + ); + + int (*gm_MagickBlurImage) ( + void *wand, + const double radius, + const double sigma + ); + + int (*gm_MagickAddNoiseImage) ( + void *wand, + const gmlib_NoiseType noise_type + ); + +} gmlib_state_info; + +static gmlib_state_info gmlib_state = { + + .initialized = 0, + .padding = 0, + + .gm_InitializeMagick = NULL, + .gm_DestroyMagick = NULL, + .gm_NewMagickWand = NULL, + .gm_DestroyMagickWand = NULL, + .gm_MagickReadImage = NULL, + .gm_MagickWriteImage = NULL, + + .gm_MagickBlurImage = NULL, + .gm_MagickAddNoiseImage = NULL, + +}; + +static int gmlib_initialize(lua_State * L) // todo: table +{ + if (! gmlib_state.initialized) { + const char *filename1 = lua_tostring(L,1); + const char *filename2 = lua_tostring(L,2); + if (filename1) { + + lmt_library lib = lmt_library_load(filename1); + + gmlib_state.gm_InitializeMagick = lmt_library_find(lib, "InitializeMagick"); + gmlib_state.gm_DestroyMagick = lmt_library_find(lib, "DestroyMagick"); + + gmlib_state.initialized = lmt_library_okay(lib); + } + if (gmlib_state.initialized && filename2) { + + lmt_library lib = lmt_library_load(filename2); + + gmlib_state.gm_NewMagickWand = lmt_library_find(lib, "NewMagickWand"); + gmlib_state.gm_DestroyMagickWand = lmt_library_find(lib, "DestroyMagickWand"); + gmlib_state.gm_MagickReadImage = lmt_library_find(lib, "MagickReadImage"); + gmlib_state.gm_MagickWriteImage = lmt_library_find(lib, "MagickWriteImage"); + + gmlib_state.gm_MagickBlurImage = lmt_library_find(lib, "MagickBlurImage"); + gmlib_state.gm_MagickAddNoiseImage = lmt_library_find(lib, "MagickAddNoiseImage"); + + gmlib_state.initialized = lmt_library_okay(lib); + } + } + lua_pushboolean(L, gmlib_state.initialized); + return 1; +} + +/* We could have a callback for stdout and error. */ + +/* Somehow not in gm: (void) MagickImageCommand(image_info, arg_count, args, NULL, exception); */ + +static int gmlib_execute(lua_State * L) +{ + if (gmlib_state.initialized) { + if (gmlib_state.initialized == 1) { + /* Once per run. */ + gmlib_state.gm_InitializeMagick(NULL); + gmlib_state.initialized = 2; + } + if (lua_type(L, 1) == LUA_TTABLE) { + void *wand = NULL; + const char *inpname = NULL; + const char *outname = NULL; + lua_getfield(L, -1, "inputfile" ); inpname = luaL_optstring(L, -1, NULL); lua_pop(L, 1); + lua_getfield(L, -1, "outputfile"); outname = luaL_optstring(L, -1, NULL); lua_pop(L, 1); + /* gmlib_state.gm_InitializeMagick(NULL); */ + wand = gmlib_state.gm_NewMagickWand(); + if (wand) { + int state = gmlib_state.gm_MagickReadImage(wand, inpname); /* todo: check return status */ + if (state) { + /* fun stuff */ + if (lua_getfield(L, -1, "blur" ) == LUA_TTABLE) { + lua_getfield(L, -1, "radius"); + lua_getfield(L, -2, "sigma"); + gmlib_state.gm_MagickBlurImage(wand, lua_tonumber(L, -2), lua_tonumber(L, -1)); + lua_pop(L, 3); + } else { + lua_pop(L, 1); + } + if (lua_getfield(L, -1, "noise" ) == LUA_TTABLE) { + lua_getfield(L, -1, "type"); + gmlib_state.gm_MagickAddNoiseImage(wand, lua_tointeger(L, -1)); + lua_pop(L, 2); + } else { + lua_pop(L, 1); + } + /* done */ + state = gmlib_state.gm_MagickWriteImage(wand, outname); /* todo: check return status */ + gmlib_state.gm_DestroyMagickWand(wand); + if (state) { + lua_pushboolean(L, 1); + return 1; + } else { + lua_pushboolean(L, 0); + lua_pushliteral(L, "possible write error"); + return 2; + } + } else { + gmlib_state.gm_DestroyMagickWand(wand); + lua_pushboolean(L, 0); + lua_pushliteral(L, "possible read error"); + return 2; + } + } else { + lua_pushboolean(L, 0); + lua_pushliteral(L, "possible memory issue"); + return 2; + } + /* gmlib_state.gm_DestroyMagick(); */ + } else { + lua_pushboolean(L, 0); + lua_pushliteral(L, "invalid specification"); + return 2; + } + } + lua_pushboolean(L, 0); + lua_pushliteral(L, "not initialized"); + return 2; +} + +static struct luaL_Reg gmlib_function_list[] = { + { "initialize", gmlib_initialize }, + { "execute", gmlib_execute }, + { NULL, NULL }, +}; + +int luaopen_graphicsmagick(lua_State * L) +{ + lmt_library_register(L, "graphicsmagick", gmlib_function_list); + return 0; +} |