Add example files

Author: Gavin Lüdemann

README.md

@@ -3,3 +3,5 @@
 A Python Lighthouse Adapter
 
 (README WIP; please consult pyghthouse/ph.py for usage information.)
+
+Example scripts can be found [here](examples)

examples/README.md

@@ -0,0 +1,8 @@
+# Pyghthouse Examples
+
+This is a loose collection of example scripts for pyghthouse.
+
+Run any script by entering `python <filename>`.
+
+If you set up config.py with your username and API token, you won't have to enter them every time you run a script.
+Be aware that API tokens are only valid for a few days.

examples/config.py

@@ -0,0 +1,11 @@
+##### enter your username and optionally your valid API token #####
+
+UNAME = ""
+TOKEN = ""
+
+##### leave everything below unchanged #####
+
+if UNAME == "":
+    UNAME = input("Enter your username: ")
+if TOKEN == "":
+    TOKEN = input("Enter your valid API token: ")

examples/huecircle.py

@@ -0,0 +1,34 @@
+from pyghthouse import Pyghthouse
+from pyghthouse.utils._color import from_hsv
+from config import UNAME, TOKEN
+import numpy as np
+
+
+def cart2pol(x, y):
+    rho = np.sqrt(x**2 + y**2)
+    phi = (np.arctan2(y, x) / np.pi) / 2
+    return rho, phi
+
+
+def circle_generator():
+    image = np.zeros((14, 28, 3))
+    while True:
+        for i in range(180):
+            for x in range(28):
+                for y in range(14):
+                    rho, phi = cart2pol(x / 28 - 0.5, y / 14 - 0.5)
+                    image[y, x, :] = from_hsv((phi + i / 180) % 1.0, 1-rho, i / 90 if i < 91 else 1 - (i - 90) / 90)
+            yield image
+
+
+circle = circle_generator()
+
+
+def callback():
+    return next(circle)
+
+
+if __name__ == '__main__':
+    p = Pyghthouse(UNAME, TOKEN, image_callback=callback)
+    print("Starting... use CTRL+C to stop.")
+    p.start()

examples/movingdot.py

@@ -0,0 +1,37 @@
+from pyghthouse import Pyghthouse, VerbosityLevel
+from config import UNAME, TOKEN
+
+
+def clip(val, min_val, max_val):
+    if val < min_val:
+        return min_val
+    if val > max_val:
+        return max_val
+    return val
+
+
+def main_loop():
+    x = 0
+    y = 0
+    p = Pyghthouse(UNAME, TOKEN, verbosity=VerbosityLevel.NONE)
+    p.start()
+    while True:
+        img = p.empty_image()
+        img[y][x] = [255, 255, 255]
+        p.set_image(img)
+        s = input()
+        for c in s.upper():
+            if c == 'A':
+                x -= 1
+            elif c == 'D':
+                x += 1
+            elif c == 'W':
+                y -= 1
+            elif c == 'S':
+                y += 1
+        x = clip(x, 0, 27)
+        y = clip(y, 0, 13)
+
+
+if __name__ == '__main__':
+    main_loop()

examples/noisefill.py

@@ -0,0 +1,26 @@
+import numpy as np
+from random import random
+
+from config import UNAME, TOKEN
+from pyghthouse import Pyghthouse
+
+
+def image_gen():
+    image = np.zeros((14, 28, 3))
+    yield image
+    while True:
+        for y in range(28):
+            for j in range(3):
+                image[0, y, j] = int(random() * 255)
+            yield image
+        image = np.roll(image, 1, 0)
+        image *= 0.85
+        yield image
+
+
+g = image_gen()
+
+if __name__ == '__main__':
+    p = Pyghthouse(UNAME, TOKEN, image_callback=g.__next__, frame_rate=60)
+    print("Starting... use CTRL+C to stop.")
+    p.start()

examples/rainbow.py

@@ -0,0 +1,23 @@
+from pyghthouse import Pyghthouse
+from pyghthouse.utils._color import from_hsv
+from config import UNAME, TOKEN
+
+
+def rainbow_generator():
+    while True:
+        for i in range(180):
+            yield [from_hsv((i / 180 + j / (14 * 28)) % 1.0, 1.0, 1.0) for j in range(14 * 28)]
+
+
+rainbow = rainbow_generator()
+
+
+def callback():
+    return next(rainbow)
+
+
+if __name__ == '__main__':
+    p = Pyghthouse(UNAME, TOKEN, image_callback=callback)
+    print("Starting... use CTRL+C to stop.")
+    p.start()
+

examples/rgbfill.py

@@ -0,0 +1,28 @@
+import numpy as np
+
+from config import UNAME, TOKEN
+from pyghthouse import Pyghthouse
+
+
+def image_gen():
+    image = np.zeros((14, 28, 3))
+    yield image
+    while True:
+        for x in range(14):
+            for y in range(28):
+                for j in range(3):
+                    image[x, y, j] = 255
+                    yield image
+        for y in range(28):
+            for x in range(14):
+                for j in range(3):
+                    image[x, y, j] = 0
+                    yield image
+
+
+g = image_gen()
+
+if __name__ == '__main__':
+    p = Pyghthouse(UNAME, TOKEN, image_callback=g.__next__, frame_rate=60)
+    print("Starting... use CTRL+C to stop.")
+    p.start()

examples/rgbscan.py

@@ -0,0 +1,24 @@
+import numpy as np
+
+from config import UNAME, TOKEN
+from pyghthouse import Pyghthouse
+
+
+def image_gen():
+    image = np.zeros((14, 28, 3))
+    yield image
+    while True:
+        for j in range(3):
+            for x in range(14):
+                for y in range(28):
+                    image[x, y, j] = 255
+                    yield image
+                    image[x, y, j] = 0
+
+
+g = image_gen()
+
+if __name__ == '__main__':
+    p = Pyghthouse(UNAME, TOKEN, image_callback=g.__next__, frame_rate=60)
+    print("Starting... use CTRL+C to stop.")
+    p.start()

examples/twopoints.py

@@ -0,0 +1,64 @@
+from random import random
+
+import numpy as np
+from PIL import Image, ImageDraw
+
+from pyghthouse.utils._color import from_hsv
+from pyghthouse import Pyghthouse
+
+from config import UNAME, TOKEN
+
+SPEED_FACTOR = 0.02
+COLOR_SPEED = 0.01
+
+
+class BouncyPoint:
+
+    def __init__(self):
+        self.x = random()
+        self.y = random()
+        self.vx = SPEED_FACTOR * (random() + 0.5)
+        self.vy = SPEED_FACTOR * (random() + 0.5)
+
+    def update(self):
+        self.x += self.vx
+        self.y += self.vy
+
+        if self.x < 0:
+            self.vx *= -1
+            self.x *= -1
+        elif self.x > 1:
+            self.vx *= -1
+            self.x = 2 - self.x
+        if self.y < 0:
+            self.vy *= -1
+            self.y *= -1
+        elif self.y > 1:
+            self.vy *= -1
+            self.y = 2 - self.y
+
+
+class ImageMaker:
+
+    def __init__(self):
+        self.p1 = BouncyPoint()
+        self.p2 = BouncyPoint()
+        self.img = Image.new('RGB', (280, 140))
+        self.draw = ImageDraw.Draw(self.img)
+        self.hue = 0.0
+
+    def callback(self):
+        self.p1.update()
+        self.p2.update()
+        self.hue += COLOR_SPEED
+        self.draw.line([(self.p1.x * 280, self.p1.y * 140), (self.p2.x * 280, self.p2.y * 140)], width=10,
+                       fill=tuple(from_hsv(self.hue, 1, 1)))
+        output = self.img.copy()
+        output.thumbnail((28, 14))
+        return np.asarray(output)
+
+
+if __name__ == '__main__':
+    i = ImageMaker()
+    p = Pyghthouse(UNAME, TOKEN, image_callback=i.callback, frame_rate=60)
+    p.start()