init commit

imggen.py

@@ -0,0 +1,70 @@
+from PIL import Image
+import numpy
+import numba
+
+@numba.jit(nopython=True)
+def _hilbertMap(n, size):
+    # https://en.wikipedia.org/wiki/Hilbert_curve
+    def rot(n, x, y, rx, ry):
+        if ry == 0:
+            if rx == 1:
+                x = n-1 - x
+                y = n-1 - y
+            
+            x, y = y, x
+        
+        return (x, y)
+
+    x, y = (0, 0)
+    t = n
+    s = 1
+    while (s < size):
+        rx = 1 & int(t / 2)
+        ry = 1 & (t ^ rx)
+        x, y = rot(s, x, y, rx, ry)
+        x += s * rx
+        y += s * ry
+        t = int(t / 4)
+        s *= 2
+    
+    return x, y
+
+class imggen:
+    x, y = (0,0)
+    imgraw = None
+
+    def __init__(self, size):
+        if size % 2 != 0:
+            raise ValueError("Size needs to be divisible by 2")
+        
+        self.x = int(pow(2, size/2))
+        self.y = self.x
+
+        self.imgraw = numpy.zeros((self.x, self.y, 3), dtype=numpy.uint8)
+
+
+    def colorRange(self, start, end, color):
+        for p in range(start, end+1):
+            x, y = _hilbertMap(p, self.x)
+            self.imgraw[x, y] = color
+
+    def saveImage(self, path, scale=1):
+        img = Image.fromarray(self.imgraw, mode="RGB")
+        img = img.resize([self.x * scale, self.y * scale], Image.NEAREST)
+        img.save(path)
+
+    def showImage(self, scale=1):
+        img = Image.fromarray(self.imgraw, mode="RGB")
+        img = img.resize([self.x * scale, self.y * scale], Image.NEAREST)
+        img.show()
+
+    def clear(self, color=(0,0,0)):
+        self.imgraw = numpy.full((self.x, self.y, 3), color, dtype=numpy.uint8)
+
+    
+
+if __name__ == "__main__":
+    gen = imggen(24)
+    gen.clear((255, 255, 255))
+    gen.colorRange(0, pow(2, 22), (73, 255, 51))
+    gen.showImage()