Merge branch 'develop' of github.com:OUDON/rmqr-generator into develop

OUDON · OUDON · commit e69278528ed1 · 2022-06-10T11:28:33.000+09:00
diff --git a/README.md b/README.md
@@ -1,7 +1,7 @@
 # An rMQR Code Generator
 ![Lorem_ipsum](https://user-images.githubusercontent.com/14174940/171996095-4707be09-506e-4ef2-ab90-9942d6efc2ed.png)
 
-This is an rMQR Code image generator implemented in Python.
+This is an rMQR Code image generator implemented in Python. This is implemented based on [ISO/IEC 23941:2022](https://www.iso.org/standard/77404.html).
 
 ## Important Notice
 Please verify an image generated by this software whether it can decode correctly before use.
diff --git a/example.py b/example.py
@@ -22,7 +22,7 @@ def main():
     # print(qr)
 
     # Save as png
-    image = QRImage(qr)
+    image = QRImage(qr, module_size=8)
     image.show()
     image.save("my_qr.png")
 
diff --git a/src/rmqrcode/qr_image.py b/src/rmqrcode/qr_image.py
@@ -1,11 +1,17 @@
 from .enums.color import Color
 
 from PIL import Image
+from PIL import ImageDraw
 
 
 class QRImage:
-    def __init__(self, qr):
-        self._img = Image.new('RGB', (qr.width() + 4, qr.height() + 4), (255, 255, 255))
+    def __init__(self, qr, module_size=10):
+        self._module_size = module_size
+        self._img = Image.new(
+            'RGB',
+            ((qr.width() + 4) * module_size, (qr.height() + 4) * module_size),
+            (255, 255, 255)
+        )
         self._make_image(qr)
 
 
@@ -19,11 +25,15 @@ def save(self, name):
 
 
     def _make_image(self, qr):
+        draw = ImageDraw.Draw(self._img)
         for y in range(qr.height()):
             for x in range(qr.width()):
                 r, g, b = 125, 125, 125
                 if qr.value_at(x, y) == Color.BLACK:
                     r, g, b = 0, 0, 0
                 elif qr.value_at(x, y) == Color.WHITE:
                     r, g, b, = 255, 255, 255
-                self._img.putpixel((x+2, y+2), (r, g, b))
+                draw.rectangle(
+                    xy=((x + 2) * self._module_size, (y + 2) * self._module_size, (x + 1 + 2) * self._module_size, (y + 1 + 2) * self._module_size),
+                    fill=(r, g, b)
+                )
