mrmBot-Matrix/natives/qr.cc

51 lines
1.3 KiB
C++
Raw Normal View History

2020-07-27 23:02:59 +00:00
#include <napi.h>
#include <ZXing/ReadBarcode.h>
#include <ZXing/TextUtfEncoding.h>
#define STB_IMAGE_IMPLEMENTATION
#include <stb/stb_image.h>
using namespace std;
using namespace ZXing;
class QrReadWorker : public Napi::AsyncWorker {
public:
QrReadWorker(Napi::Function& callback, string in_path)
: Napi::AsyncWorker(callback), in_path(in_path) {}
~QrReadWorker() {}
void Execute() {
int width, height, channels;
unique_ptr<stbi_uc, void(*)(void*)> buffer(stbi_load(in_path.c_str(), &width, &height, &channels, 4), stbi_image_free);
auto result = ReadBarcode(width, height, buffer.get(), width * 4, 4, 0, 1, 2, { BarcodeFormat::QR_CODE });
if (result.isValid()) {
final = TextUtfEncoding::ToUtf8(result.text());
missing = false;
} else {
final = "";
}
}
void OnOK() {
Napi::Object object = Napi::Object::New(Env());
object.Set("qrText", final);
object.Set("missing", missing);
Callback().Call({Env().Undefined(), object});
}
private:
string in_path, final;
bool missing = true;
};
Napi::Value QrRead(const Napi::CallbackInfo &info)
{
Napi::Env env = info.Env();
string in_path = info[0].As<Napi::String>().Utf8Value();
Napi::Function cb = info[1].As<Napi::Function>();
QrReadWorker* qrReadWorker = new QrReadWorker(cb, in_path);
qrReadWorker->Queue();
return env.Undefined();
}