qrtools 1.0.0

data/ext/qrtools/qrtools_decoder.h

@@ -0,0 +1,10 @@
+#ifndef QRTOOLS_DECODER
+#define QRTOOLS_DECODER
+
+#include <qrtools.h>
+
+extern VALUE cQRToolsDecoder;
+
+void init_qrtools_decoder();
+
+#endif

data/ext/qrtools/qrtools_encoder.c

@@ -0,0 +1,63 @@
+#include <qrtools_encoder.h>
+
+static VALUE encode(int argc, VALUE *argv, VALUE klass)
+{
+  //extern QRcode *QRcode_encodeString(const char *string, int version, QRecLevel level, QRencodeMode hint, int casesensitive);
+  VALUE string, options;
+  if(rb_scan_args(argc, argv, "11", &string, &options) == 1) {
+    options = rb_hash_new();
+  }
+
+  VALUE version = rb_hash_aref(options, ID2SYM(rb_intern("version")));
+  VALUE level = rb_hash_aref(options, ID2SYM(rb_intern("error_correction")));
+  VALUE case_sensitive = rb_hash_aref(options, ID2SYM(rb_intern("case_sensitive")));
+  VALUE mode = rb_hash_aref(options, ID2SYM(rb_intern("mode")));
+
+  int cs = 0;
+  if(RTEST(case_sensitive) && case_sensitive == Qtrue) cs = 1;
+
+  QRcode * code = QRcode_encodeString(
+      StringValuePtr(string),
+      RTEST(version) ? NUM2INT(version) : 0,
+      RTEST(level) ? NUM2INT(level) : 0,
+      RTEST(mode) ? NUM2INT(mode) : 2,
+      cs
+  );
+  if(NULL == code) rb_raise(rb_eRuntimeError, "could not create code");
+  return Data_Wrap_Struct(cQRToolsQRCode, NULL, QRcode_free, code);
+}
+
+static VALUE version(VALUE self)
+{
+  QRinput * input;
+  Data_Get_Struct(self, QRinput, input);
+  return INT2NUM(QRinput_getVersion(input));
+}
+
+static VALUE set_version(VALUE self, VALUE version)
+{
+  QRinput * input;
+  Data_Get_Struct(self, QRinput, input);
+  QRinput_setVersion(input, NUM2INT(version));
+
+  return self;
+}
+
+static VALUE allocate(VALUE klass)
+{
+  return Data_Wrap_Struct(klass, NULL, QRinput_free, QRinput_new());
+}
+
+VALUE cQRToolsEncoder;
+void init_qrtools_encoder()
+{
+  VALUE qrtools = rb_define_module("QRTools");
+  VALUE klass   = rb_define_class_under(qrtools, "Encoder", rb_cObject);
+
+  cQRToolsEncoder = klass;
+
+  rb_define_alloc_func(klass, allocate);
+  rb_define_singleton_method(klass, "encode", encode, -1);
+  rb_define_method(klass, "version", version, 0);
+  rb_define_method(klass, "version=", set_version, 1);
+}

data/ext/qrtools/qrtools_encoder.h

@@ -0,0 +1,10 @@
+#ifndef QRTOOLS_ENDECODER
+#define QRTOOLS_ENDECODER
+
+#include <qrtools.h>
+
+extern VALUE cQRToolsEncoder;
+
+void init_qrtools_encoder();
+
+#endif

data/ext/qrtools/qrtools_header.c

@@ -0,0 +1,51 @@
+#include <qrtools_header.h>
+
+static VALUE model(VALUE self)
+{
+  QrCodeHeader * header;
+  Data_Get_Struct(self, QrCodeHeader, header);
+  return INT2NUM(header->model);
+}
+
+static VALUE version(VALUE self)
+{
+  QrCodeHeader * header;
+  Data_Get_Struct(self, QrCodeHeader, header);
+  return INT2NUM(header->version);
+}
+
+static VALUE level(VALUE self)
+{
+  QrCodeHeader * header;
+  Data_Get_Struct(self, QrCodeHeader, header);
+  return INT2NUM(header->level);
+}
+
+static VALUE character_size(VALUE self)
+{
+  QrCodeHeader * header;
+  Data_Get_Struct(self, QrCodeHeader, header);
+  return INT2NUM(header->charactor_size);
+}
+
+static VALUE byte_size(VALUE self)
+{
+  QrCodeHeader * header;
+  Data_Get_Struct(self, QrCodeHeader, header);
+  return INT2NUM(header->byte_size);
+}
+
+VALUE cQRToolsHeader;
+void init_qrtools_header()
+{
+  VALUE qrtools  = rb_define_module("QRTools");
+  VALUE klass    = rb_define_class_under(qrtools, "Header", rb_cObject);
+
+  cQRToolsHeader = klass;
+
+  rb_define_method(klass, "model", model, 0);
+  rb_define_method(klass, "version", version, 0);
+  rb_define_method(klass, "level", level, 0);
+  rb_define_method(klass, "character_size", character_size, 0);
+  rb_define_method(klass, "byte_size", byte_size, 0);
+}

data/ext/qrtools/qrtools_header.h

@@ -0,0 +1,10 @@
+#ifndef QRTOOLS_HEADER
+#define QRTOOLS_HEADER
+
+#include <qrtools.h>
+
+extern VALUE cQRToolsHeader;
+
+void init_qrtools_header();
+
+#endif

data/ext/qrtools/qrtools_image.c

@@ -0,0 +1,80 @@
+#include <qrtools_image.h>
+
+static void dealloc(IplImage * img)
+{
+  cvReleaseImage(&img);
+}
+
+static VALUE load(VALUE klass, VALUE path)
+{
+  IplImage *src = cvLoadImage(StringValuePtr(path), 1);
+  return Data_Wrap_Struct(klass, NULL, dealloc, src);
+}
+
+static VALUE save(VALUE self, VALUE path)
+{
+  IplImage * src;
+  Data_Get_Struct(self, IplImage, src);
+  cvSaveImage(StringValuePtr(path), src);
+  return self;
+}
+
+/*
+ * call-seq:
+ *  native_draw_line(from, to, r, g, b, thickness, type, shift)
+ *
+ * Draw line from +from+ vertex to +to+ vertex
+ */
+static VALUE native_draw_line(
+    VALUE self,
+    VALUE _from,
+    VALUE _to,
+    VALUE _r,
+    VALUE _g,
+    VALUE _b,
+    VALUE _thickness,
+    VALUE _type,
+    VALUE _shift
+) {
+  IplImage * src;
+  Data_Get_Struct(self, IplImage, src);
+  cvLine(
+      src,
+      cvPoint(
+        NUM2INT(rb_funcall(_from, rb_intern("x"), 0)),
+        NUM2INT(rb_funcall(_from, rb_intern("y"), 0))
+      ),
+      cvPoint(
+        NUM2INT(rb_funcall(_to, rb_intern("x"), 0)),
+        NUM2INT(rb_funcall(_to, rb_intern("y"), 0))
+      ),
+      CV_RGB(
+        NUM2INT(_r),
+        NUM2INT(_g),
+        NUM2INT(_b)
+      ),
+      NUM2INT(_thickness),
+      NUM2INT(_type),
+      NUM2INT(_shift)
+  );
+  return self;
+}
+
+VALUE QRTools_Wrap_Image(IplImage * image)
+{
+  return Data_Wrap_Struct(cQRToolsImage, NULL, dealloc, image);
+}
+
+VALUE cQRToolsImage;
+
+void init_qrtools_image()
+{
+  VALUE qrtools  = rb_define_module("QRTools");
+  VALUE klass    = rb_define_class_under(qrtools, "Image", rb_cObject);
+
+  cQRToolsImage = klass;
+
+  rb_define_singleton_method(klass, "load", load, 1);
+  rb_define_method(klass, "save", save, 1);
+  rb_define_method(klass, "native_draw_line", native_draw_line, 8);
+}

data/ext/qrtools/qrtools_image.h

@@ -0,0 +1,11 @@
+#ifndef QRTOOLS_IMAGE
+#define QRTOOLS_IMAGE
+
+#include <qrtools.h>
+
+extern VALUE cQRToolsImage;
+
+void init_qrtools_image();
+VALUE QRTools_Wrap_Image(IplImage * image);
+
+#endif

data/ext/qrtools/qrtools_qrcode.c

@@ -0,0 +1,36 @@
+#include <qrtools_qrcode.h>
+
+static VALUE width(VALUE self)
+{
+  QRcode * code;
+  Data_Get_Struct(self, QRcode, code);
+  return INT2NUM(code->width);
+}
+
+static VALUE version(VALUE self)
+{
+  QRcode * code;
+  Data_Get_Struct(self, QRcode, code);
+  return INT2NUM(code->version);
+}
+
+static VALUE data(VALUE self)
+{
+  QRcode * code;
+  Data_Get_Struct(self, QRcode, code);
+
+  return rb_str_new(code->data, code->width * code->width);
+}
+
+VALUE cQRToolsQRCode;
+void init_qrtools_qrcode()
+{
+  VALUE qrtools = rb_define_module("QRTools");
+  VALUE klass   = rb_define_class_under(qrtools, "QRCode", rb_cObject);
+
+  cQRToolsQRCode = klass;
+
+  rb_define_method(klass, "width", width, 0);
+  rb_define_method(klass, "version", version, 0);
+  rb_define_method(klass, "data", data, 0);
+}

data/ext/qrtools/qrtools_qrcode.h

@@ -0,0 +1,10 @@
+#ifndef QRTOOLS_QRCODE
+#define QRTOOLS_QRCODE
+
+#include <qrtools.h>
+
+extern VALUE cQRToolsQRCode;
+
+void init_qrtools_qrcode();
+
+#endif

data/ext/qrtools/qrtools_ui_camera.c

@@ -0,0 +1,58 @@
+#include <qrtools_ui_camera.h>
+
+static void deallocate(CvCapture * capture)
+{
+  if(capture) cvReleaseCapture(&capture);
+}
+
+static VALUE allocate(VALUE klass)
+{
+  return Data_Wrap_Struct(klass, NULL, deallocate, NULL);
+}
+
+static VALUE start(VALUE self)
+{
+  VALUE index = rb_iv_get(self, "@camera_number");
+  CvCapture * capture = cvCreateCameraCapture(NUM2INT(index));
+  DATA_PTR(self) = capture;
+  return self;
+}
+
+static VALUE stop(VALUE self)
+{
+  CvCapture * capture;
+  Data_Get_Struct(self, CvCapture, capture);
+  if(capture) {
+    cvReleaseCapture(&capture);
+    DATA_PTR(self) = NULL;
+  }
+  return self;
+}
+
+static VALUE capture(VALUE self)
+{
+  CvCapture * capture;
+  Data_Get_Struct(self, CvCapture, capture);
+
+  IplImage * cam = cvQueryFrame(capture);
+  if(cam)
+    return QRTools_Wrap_Image(cvCloneImage(cam));
+
+  return Qnil;
+}
+
+VALUE cQRToolsUICamera;
+
+void init_qrtools_ui_camera()
+{
+  VALUE qrtools = rb_define_module("QRTools");
+  VALUE ui      = rb_define_module_under(qrtools, "UI");
+  VALUE klass   = rb_define_class_under(ui, "Camera", rb_cObject);
+
+  cQRToolsUICamera = klass;
+  rb_define_alloc_func(klass, allocate);
+  rb_define_method(klass, "capture", capture, 0);
+
+  rb_define_private_method(klass, "native_start", start, 0);
+  rb_define_private_method(klass, "native_stop", stop, 0);
+}

data/ext/qrtools/qrtools_ui_camera.h

@@ -0,0 +1,10 @@
+#ifndef QRTOOLS_UI_CAMERA
+#define QRTOOLS_UI_CAMERA
+
+#include <qrtools.h>
+
+extern VALUE cQRToolsUICamera;
+
+void init_qrtools_ui_camera();
+
+#endif

data/ext/qrtools/qrtools_ui_window.c

@@ -0,0 +1,40 @@
+#include <qrtools_ui_window.h>
+
+static VALUE show(VALUE self)
+{
+  VALUE name = rb_iv_get(self, "@name");
+  cvNamedWindow(StringValuePtr(name), 1);
+  return self;
+}
+
+static VALUE hide(VALUE self)
+{
+  VALUE name = rb_iv_get(self, "@name");
+  cvDestroyWindow(StringValuePtr(name));
+  return self;
+}
+
+static VALUE display(VALUE self, VALUE image)
+{
+  VALUE name = rb_iv_get(self, "@name");
+
+  IplImage * src;
+  Data_Get_Struct(image, IplImage, src);
+  cvShowImage(StringValuePtr(name), src);
+  return self;
+}
+
+VALUE cQRToolsUIWindow;
+
+void init_qrtools_ui_window()
+{
+  VALUE qrtools = rb_define_module("QRTools");
+  VALUE ui      = rb_define_module_under(qrtools, "UI");
+  VALUE klass   = rb_define_class_under(ui, "Window", rb_cObject);
+
+  cQRToolsUIWindow = klass;
+
+  rb_define_method(klass, "show", show, 0);
+  rb_define_method(klass, "hide", hide, 0);
+  rb_define_method(klass, "display", display, 1);
+}

data/ext/qrtools/qrtools_ui_window.h

@@ -0,0 +1,10 @@
+#ifndef QRTOOLS_UI_WINDOW
+#define QRTOOLS_UI_WINDOW
+
+#include <qrtools.h>
+
+extern VALUE cQRToolsUIWindow;
+
+void init_qrtools_ui_window();
+
+#endif

data/ext/qrtools/qrtypes.h

@@ -0,0 +1,42 @@
+/////////////////////////////////////////////////////////////////////////
+//
+// qrtypes.h --a part of libdecodeqr
+//
+// Copyright(C) 2007 NISHI Takao <zophos@koka-in.org>
+//                   JMA  (Japan Medical Association)
+//                   NaCl (Network Applied Communication Laboratory Ltd.)
+//
+// This is free software with ABSOLUTELY NO WARRANTY.
+// You can redistribute and/or modify it under the terms of LGPL.
+//
+// $Id: qrtypes.h 36 2007-02-21 23:22:03Z zophos $
+//
+#ifndef __QR_TYPES__
+#define __QR_TYPES__
+
+#define QR_LEVEL_M 0
+#define QR_LEVEL_L 1
+#define QR_LEVEL_H 2
+#define QR_LEVEL_Q 3
+
+#define QR_MODE_NUMBER 1
+#define QR_MODE_ALPHABET 2
+#define QR_MODE_JOINT 3
+#define QR_MODE_8BIT 4
+#define QR_MODE_FNC1_1 5
+#define QR_MODE_ECI 7
+#define QR_MODE_KANJIS 8
+#define QR_MODE_FNC1_2 9
+
+typedef void * QrDecoderHandle;
+typedef struct{
+    int model;
+    int version;
+    int level;
+    //int mode;     //not supported yet
+    //int eci_mode; //not supported yet
+    int charactor_size;
+    int byte_size;
+} QrCodeHeader;
+
+#endif