add rainbow and theatre chaser pattern

flask_pixel_server/flask-app.py

@@ -37,6 +37,14 @@ def set_all():
         pixelservice.set_all(int(r), int(g), int(b))
     return "OK"
 
+@app.route('/setpattern', methods=['POST'])
+def set_pattern():
+    name = request.args.get('name')
+    if not name:
+        return 'invalid pattern name', 400
+    pixelservice.set_pattern(name)
+    return "OK"
+
 
 if __name__ == '__main__':
     args = ArgParser(DESCRIPTION).parse_args()

flask_pixel_server/pixel_service.py

@@ -4,6 +4,7 @@
 sys.path.append('..')
 
 from presenter import Pixel
+import pixel_patterns.pixel_strip_patterns as patterns
 
 
 class PixelService:
@@ -30,6 +31,10 @@ def set_all(self, r, g, b):
         self.generator = ConstantValueGenerator(self.num_pixels)
         self.generator.set_all(r, g, b)
 
+    def set_pattern(self, name):
+        self.generator = PatternGenerator(self.num_pixels)
+        self.generator.set_pattern(name)
+
     def _generate(self):
         return self.generator.generate()
 
@@ -46,3 +51,24 @@ def set_all(self, r, g, b):
             p.rgb[0] = r
             p.rgb[1] = g
             p.rgb[2] = b
+
+
+class PatternGenerator:
+    def __init__(self, num_pixels):
+        self.num_pixels = num_pixels
+        self.patterns = {
+            'rainbow': lambda: patterns.Rainbow(self.num_pixels),
+            'colorwipe': lambda: patterns.ColorWiper(self.num_pixels, Pixel(100, 0, 100), 50),
+            'rainbow_cycle': lambda: patterns.RainbowCycle(self.num_pixels),
+            'theatre': lambda: patterns.TheatreChaser(self.num_pixels, Pixel(100, 0, 100), 50)
+        }
+        self.generator = self._get_generator('rainbow')
+
+    def set_pattern(self, name):
+        self.generator = self._get_generator(name)
+
+    def generate(self):
+        return self.generator.generate()
+
+    def _get_generator(self, name):
+        return self.patterns[name]()

flask_pixel_server/static/index.html

@@ -16,6 +16,8 @@ <h1>Pi Lights</h1>
 
     <button id="btn-off" type="button" class="btn btn-primary btn-lg btn-black">OFF</button>
     <button id="btn-red" type="button" class="btn btn-primary btn-lg btn-red">All red</button>
+    <button id="btn-rainbow" type="button" class="btn btn-primary btn-lg">Rainbow</button>
+    <button id="btn-theatre" type="button" class="btn btn-primary btn-lg">Theatre</button>
 
     <div>
         <label for="rrange">Red</label>
@@ -44,6 +46,12 @@ <h1>Pi Lights</h1>
     $('#btn-red').on('click', function(e) {
         $.post('setall?color=red');
     });
+    $('#btn-rainbow').on('click', function(e) {
+        $.post('setpattern?name=rainbow');
+    });
+    $('#btn-theatre').on('click', function(e) {
+        $.post('setpattern?name=theatre');
+    });
     $('input[type=range]').on('input', function () {
         postSliderValues();
     });