72 lines
2.3 KiB
C
72 lines
2.3 KiB
C
|
#include "module.h"
|
||
|
|
#include "lauxlib.h"
|
||
|
|
#include "common.h"
|
||
|
|
#include "lmem.h"
|
||
|
|
|
||
|
|
#include "qrcodegen.h"
|
||
|
|
|
||
|
|
// qrcodegen.encodeText(text, [{minver=int, maxver=int, ecl=int, mask=int, boostecl=bool}])
|
||
|
|
// all params except text are optional
|
||
|
|
static int encodeText(lua_State *L)
|
||
|
|
{
|
||
|
|
lua_settop(L, 2);
|
||
|
|
const char *text = luaL_checkstring(L, 1);
|
||
|
|
|
||
|
|
int min_ver = opt_checkint_range(L, "minver", qrcodegen_VERSION_MIN,
|
||
|
|
qrcodegen_VERSION_MIN, qrcodegen_VERSION_MAX);
|
||
|
|
|
||
|
|
int max_ver = opt_checkint_range(L, "maxver", qrcodegen_VERSION_MAX,
|
||
|
|
min_ver, qrcodegen_VERSION_MAX);
|
||
|
|
|
||
|
|
int ecl = opt_checkint_range(L, "ecl", qrcodegen_Ecc_LOW,
|
||
|
|
qrcodegen_Ecc_LOW, qrcodegen_Ecc_HIGH);
|
||
|
|
|
||
|
|
int mask = opt_checkint_range(L, "mask", qrcodegen_Mask_AUTO,
|
||
|
|
qrcodegen_Mask_AUTO, qrcodegen_Mask_7);
|
||
|
|
|
||
|
|
bool boost = opt_checkbool(L, "boostecl", false);
|
||
|
|
|
||
|
|
const size_t buf_len = qrcodegen_BUFFER_LEN_FOR_VERSION(max_ver);
|
||
|
|
uint8_t *result = (uint8_t *)luaM_malloc(L, buf_len * 2);
|
||
|
|
uint8_t *tempbuf = result + buf_len;
|
||
|
|
|
||
|
|
bool ok = qrcodegen_encodeText(text, tempbuf, result, ecl, min_ver, max_ver, mask, boost);
|
||
|
|
if (ok) {
|
||
|
|
lua_pushlstring(L, (const char *)result, buf_len);
|
||
|
|
} else {
|
||
|
|
lua_pushnil(L);
|
||
|
|
}
|
||
|
|
luaM_freemem(L, result, buf_len * 2);
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int getSize(lua_State *L)
|
||
|
|
{
|
||
|
|
const uint8_t* data = (const uint8_t *)luaL_checkstring(L, 1);
|
||
|
|
lua_pushinteger(L, qrcodegen_getSize(data));
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int getPixel(lua_State *L)
|
||
|
|
{
|
||
|
|
const uint8_t* data = (const uint8_t *)luaL_checkstring(L, 1);
|
||
|
|
int x = luaL_checkint(L, 2);
|
||
|
|
int y = luaL_checkint(L, 3);
|
||
|
|
lua_pushboolean(L, qrcodegen_getModule(data, x, y));
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static const LUA_REG_TYPE qrcodegen_map[] = {
|
||
|
|
{ LSTRKEY("encodeText"), LFUNCVAL(encodeText) },
|
||
|
|
{ LSTRKEY("getSize"), LFUNCVAL(getSize) },
|
||
|
|
{ LSTRKEY("getPixel"), LFUNCVAL(getPixel) },
|
||
|
|
{ LSTRKEY("LOW"), LNUMVAL(qrcodegen_Ecc_LOW) },
|
||
|
|
{ LSTRKEY("MEDIUM"), LNUMVAL(qrcodegen_Ecc_MEDIUM) },
|
||
|
|
{ LSTRKEY("QUARTILE"), LNUMVAL(qrcodegen_Ecc_QUARTILE) },
|
||
|
|
{ LSTRKEY("HIGH"), LNUMVAL(qrcodegen_Ecc_HIGH) },
|
||
|
|
{ LSTRKEY("AUTO"), LNUMVAL(qrcodegen_Mask_AUTO) },
|
||
|
|
{ LNILKEY, LNILVAL }
|
||
|
|
};
|
||
|
|
|
||
|
|
NODEMCU_MODULE(QRCODEGEN, "qrcodegen", qrcodegen_map, NULL);
|