Upload New File

0_Python_playground/pattern.py

+import numpy as np
+import matplotlib.pyplot as plt
+
+
+class Checker:
+    res = None
+    ts = None
+    output = None
+
+    def __init__(self, res, tile_size):
+        self.res = res
+        self.ts = tile_size
+
+    def draw(self):
+        if (self.res % (2 * self.ts)) == 0:
+            base_pat = np.zeros((2 * self.ts, (2 * self.ts)))
+            base_pat[0:self.ts, self.ts:(2 * self.ts)] = 1
+            base_pat[self.ts: (2 * self.ts), 0:self.ts] = 1
+            self.output = np.tile(base_pat, ((self.res // (2 * self.ts), self.res // (2 * self.ts))))
+        return self.output.copy()
+
+    def show(self):
+        plt.imshow(self.output, cmap='gray')
+        plt.axis('off')
+        plt.show()
+        return
+
+
+class Circle:
+    res = None
+    radius = None
+    (x, y) = (None, None)
+    img = None
+
+    def __init__(self, resolution, radius, position):
+        self.res = resolution
+        self.radius = radius
+        self.x = position[0]
+        self.y = position[1]
+        self.output = np.ones((self.res, self.res))
+
+    def draw(self):
+        w = np.arange(0, self.res)
+        X, Y = np.meshgrid(w, w)
+        Z = ((X - self.x) ** 2 + (Y - self.y) ** 2) > self.radius ** 2
+        # Z = ((X - self.x) ** 2 + (Y - (self.res - 1 - self.y)) ** 2) > self.radius ** 2
+        self.output[Z] = 0
+        return np.copy(self.output)
+
+    def show(self):
+        plt.imshow(self.output, cmap='gray')
+        plt.axis('off')
+        plt.show()
+        return
+
+
+class Spectrum:
+    res = None
+    output = None
+
+    def __init__(self, resolution):
+        self.res = resolution
+
+    def draw(self):
+        gs = np.arange(0, self.res, 1) / (self.res - 1) #meshgrid
+        r = np.tile(gs, (self.res, 1))
+        b = np.tile((np.arange(self.res - 1, -1, -1) / (self.res - 1)), (self.res, 1))
+        g = np.array(r).T
+        self.output = np.dstack((r, g, b))
+        return self.output.copy()
+
+    def show(self):
+        plt.imshow(self.output)
+        plt.axis('off')
+        plt.show()