cheetah_qrcode 1.0.1 → 1.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 105a2bf919646066ad19e7041d4b80017b23fdf389c65b02e84275f917dea5e0
-  data.tar.gz: 3854c8cdda670d87bb2c61071866cadb7cb6f3d505f7c0ce264ec718632f3a16
+  metadata.gz: 0fe4a038a4c92715c7848f221672a271bc27e9e79b2d450d9894362a8fb321b8
+  data.tar.gz: '0462619677f70ede96efd8174fa3d6b52aaada7a7b119117e12936625af33dce'
 SHA512:
-  metadata.gz: 3227f3d6391afb7910d3c1c6840408d75fcdea6127b4a99f3c701b2b896ca14673047907a145c0464f2b27eb9ce5c8821cbf1e0ab527f54a86481c6cd770e549
-  data.tar.gz: 326643675bad85067387e93ff4da7bba491e2fa5eba01c919de97c6c385bf0d3c57876af2c7f38ba3987fb373846e7e082256c40097783568b2c5954c1316272
+  metadata.gz: 772f7f33a2da1b72f887a21d895c8fec5e46a00a9fe8c164e12852d775db26f660cb4c154070d1a215f007417e6f196fe6d9195d65a8db1fbdb27fdd22030f51
+  data.tar.gz: d8fc4fced392d3ca1f0062b6db4b3b534d80d5ebd8e1d917a427f860cde33a041e95bad1cd42326ff168361938fda278f7dfa9fcb1a6da89837a8bee53c2de49

data/ext/cheetah_qrcode/cheetah_qrcode.c

@@ -24,12 +24,20 @@ static VALUE encode_text(int argc, VALUE* argv, VALUE self) {
                 rb_raise(rb_eTypeError, "Invalid text");
         }
 
-        if (rb_type(arg_ec_level) == T_SYMBOL) {
-                arg_ec_level = RB_SYM2ID(arg_ec_level);
-        } else {
+        if (rb_type(arg_ec_level) != T_SYMBOL) {
                 rb_raise(rb_eTypeError, "Invalid error correction level, use :L, :M, :Q, :H");
         }
 
+        if (rb_type(arg_border) != T_FIXNUM) {
+                rb_raise(rb_eTypeError, "Invalid border");
+        }
+
+        if (rb_type(arg_size) != T_FIXNUM) {
+                rb_raise(rb_eTypeError, "Invalid size");
+        }
+
+        arg_ec_level = RB_SYM2ID(arg_ec_level);
+
         if (arg_ec_level == rb_intern("L") || arg_ec_level == rb_intern("l")) {
                 qrcode_ec_level = qrcodegen_Ecc_LOW;
         } else if (arg_ec_level == rb_intern("M") || arg_ec_level == rb_intern("m")) {
@@ -42,16 +50,18 @@ static VALUE encode_text(int argc, VALUE* argv, VALUE self) {
                 rb_raise(rb_eTypeError, "Invalid error correction level, use :L, :M, :Q, :H");
         }
 
-        if (rb_type(arg_border) == T_FIXNUM) {
-                qrcode_border = RB_NUM2UINT(arg_border);
-        } else {
-                rb_raise(rb_eTypeError, "Invalid border");
+        qrcode_border = RB_NUM2UINT(arg_border);
+
+        // The max version qrcode is only 117 wide
+        if (qrcode_border > 128) {
+                rb_raise(rb_eTypeError, "Border too large");
         }
 
-        if (rb_type(arg_size) == T_FIXNUM) {
-                image_size = RB_NUM2UINT(arg_size);
-        } else {
-                rb_raise(rb_eTypeError, "Invalid size");
+        image_size = RB_NUM2UINT(arg_size);
+
+        // 204800x204800 will use 5000MiB memory for image buffer
+        if (image_size > 204800) {
+                rb_raise(rb_eTypeError, "Image size too large");
         }
 
         bool ok = qrcodegen_encodeText(
@@ -96,7 +106,21 @@ static VALUE encode_text(int argc, VALUE* argv, VALUE self) {
         // Create image initialized to 0 (Entirely black image)
         image = calloc(image_length, sizeof(uint8_t));
         if (!image) {
-                rb_raise(rb_eRuntimeError, "Unable to create image buffer");
+                rb_raise(rb_eRuntimeError, "Unable to create image buffer for %lu bytes", image_length);
+        }
+
+        // Lookup map for qrcode-to-image pixel conversion
+        size_t image_scale_map[qrcode_size + 1];
+
+        // First pixel always start at 0
+        image_scale_map[0] = 0;
+
+        // Last pixel always end at image_size
+        image_scale_map[qrcode_size] = image_size;
+
+        // Precompute image scale pixels
+        for (size_t i = 1; i < qrcode_size; i++) {
+                image_scale_map[i] = (size_t)(i * image_scale) + 1;
         }
 
         // Loop through qrcode to find white modules to write into image
@@ -108,33 +132,32 @@ static VALUE encode_text(int argc, VALUE* argv, VALUE self) {
                         }
 
                         // Map current qrcode module to image coordinates
-                        size_t ix_begin = (size_t)(qx * image_scale) + 1;
-                        size_t ix_end = (size_t)((qx + 1) * image_scale) + 1;
-                        size_t iy_begin = (size_t)(qy * image_scale) + 1;
-                        size_t iy_end = (size_t)((qy + 1) * image_scale) + 1;
-
-                        // Fix offset for the first pixel
-                        if (qx == 0) {
-                                ix_begin = 0;
-                        }
-                        if (qy == 0) {
-                                iy_begin = 0;
-                        }
+                        size_t ix_begin = image_scale_map[qx];
+                        size_t ix_end = image_scale_map[qx + 1];
+                        size_t iy_begin = image_scale_map[qy];
+                        size_t iy_end = image_scale_map[qy + 1];
 
-                        // For boundary safety
-                        if (ix_end > image_size) {
-                                ix_end = image_size;
-                        }
-                        if (iy_end > image_size) {
-                                iy_end = image_size;
-                        }
-
-                        // Fill white pixels into image
+                        // Fill white pixels into image at unit of bytes
                         for (size_t iy = iy_begin; iy < iy_end; iy++) {
-                                for (size_t ix = ix_begin; ix < ix_end; ix++) {
+                                size_t ix = ix_begin;
+                                size_t ix_diff = ix_end - ix_begin;
+
+                                while (ix_diff > 0) {
                                         size_t i = (iy * image_scanline_width) + (ix / 8);
 
-                                        image[i] |= 0b10000000 >> (ix % 8);
+                                        uint8_t byte_value = 0xFF;
+                                        uint8_t bits_needed = 8 - (ix % 8);
+
+                                        if (bits_needed > ix_diff) {
+                                                byte_value <<= (bits_needed - ix_diff);
+                                                bits_needed = ix_diff;
+                                        }
+
+                                        byte_value >>= (ix % 8);
+                                        image[i] |= byte_value;
+
+                                        ix += bits_needed;
+                                        ix_diff -= bits_needed;
                                 }
                         }
                 }

data/lib/cheetah_qrcode/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module CheetahQRCode
-  VERSION = '1.0.1'
+  VERSION = '1.0.3'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cheetah_qrcode
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.0.3
 platform: ruby
 authors:
 - atitan
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2025-04-27 00:00:00.000000000 Z
+date: 2025-04-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake