Merge branch 'master' of https://github.com/infrared5/massroute-pi

Author: Aaron Artessa

.behaverc

@@ -0,0 +1,4 @@
+[behave]
+show_source=yes
+stop=yes
+strict=yes

.gitignore

@@ -1,3 +1,4 @@
 .DS_Store
 *.sublime-*
-node_modules
+*.pyc
+massroute.log

Gruntfile.js

@@ -0,0 +1,20 @@
+The MIT License
+
+Copyright (C) 2013 Infrared5 Inc. and contributors.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of
+this software and associated documentation files (the "Software"), to deal in
+the Software without restriction, including without limitation the rights to
+use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
+of the Software, and to permit persons to whom the Software is furnished to do
+so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

LICENSE

@@ -1,3 +1,15 @@
-massroute-pi
-===
-> real-time bus route aggregation and display using a [raspberry pi](http://www.raspberrypi.org)
+Introduction
+---
+> An exploration of real-time bus data aggregation and display using Raspberry Pi
+
+###Schematic
+
+<img src="https://raw.github.com/infrared5/massroute-pi/master/notes/massroutepy_schematic.png" style="width:310px" alt="MassROUTE-Pi schematic">
+
+
+###Development
+
+```
+$> mkvirtualenv massroute
+$> pip install -r requirements.txt
+```

README.md

@@ -0,0 +1,2 @@
+def enum(**enums):
+    return type('Enum', (), enums)

app/__init__.py

@@ -0,0 +1,99 @@
+import random
+import logging
+import RPi.GPIO as GPIO
+
+from . import enum
+from settings import *
+from time import sleep
+from threading import Timer
+
+from component.shifter import Shifter
+from component.modifier.blinker import Blinker
+
+logger = logging.getLogger(__name__)
+States = enum(IDLE=0, LIVE=1, ERROR=2)
+
+class Board:
+  """ Raspberry Pi interface """
+  shifter = None
+  error_timer = None
+  state = States.IDLE
+
+  def __init__(self):
+    GPIO.setmode(GPIO.BCM)
+    self.shifter = Shifter(DATA_PIN, LATCH_PIN, CLOCK_PIN)
+    self.shifter.set_shift_register_count(NUM_REGISTERS)
+
+  def reset(self):
+    self.shifter.clear()
+    self.state = States.IDLE
+    if not self.error_timer == None:
+      self.error_timer.cancel()
+
+  def run_startup_seq(self):
+    logger.info('Running startup sequence...')
+    self.reset()
+    count = 0
+    while count < 3:
+      for index in range(0, NUM_REGISTERS*8):
+        self.shifter.set_pin(index, 1)
+        sleep(0.02)
+        self.shifter.write()
+      sleep(2)
+      for index in range(0, NUM_REGISTERS*8):
+        self.shifter.set_pin(index, 0)
+        sleep(0.02)
+        self.shifter.write()
+      sleep(2)
+      count = count + 1
+    logger.info('Done running startup sequence.')
+
+  def run_shutdown_seq(self):
+    logger.info('Running shutdown sequence...')
+    self.reset()
+    count = 0
+    for index in range(0, NUM_REGISTERS*8):
+      self.shifter.set_pin(index, 1)
+      sleep(0.02)
+      self.shifter.write()
+    while count < 3:
+     for index in range(0, NUM_REGISTERS*8):
+       self.shifter.set_pin(index, 0)
+     self.shifter.write()
+     sleep(0.1)
+     for index in range(0, NUM_REGISTERS*8):
+       self.shifter.set_pin(index, 1)
+     self.shifter.write()
+     sleep(0.1)     
+     count = count + 1
+    sleep(0.5)
+    self.shifter.clear()
+    logger.info('Done running shutdown sequence.')
+
+  def run_resource_error(self):
+    logger.info('Running error sequence...')
+    if not self.is_in_error_state():
+      pins = range(NUM_REGISTERS*8)
+      self.reset()
+      self.state = States.ERROR
+      self.show_error_sequence(pins)
+
+  def show_error_sequence(self, pins):
+    if self.is_in_error_state():
+      for pin in pins:
+        self.shifter.set_pin(pin, 1)
+        sleep(0.05)
+        self.shifter.write()
+        self.shifter.set_pin(pin, 0)
+        sleep(0.05)
+      self.shifter.clear()
+      self.error_timer = Timer(0.5, self.show_error_sequence, [pins])
+      self.error_timer.start()
+
+  def is_in_error_state(self):
+    return self.state == States.ERROR
+
+  def blinker_test(self):
+    blinker = Blinker(self.shifter, 0.5)
+    blinker.set_pins([10,11,12,13])
+    blinker.start()

app/board.py

@@ -0,0 +1,29 @@
+import json
+import logging
+
+from ..util.json_util import convert
+
+logger = logging.getLogger(__name__)
+
+class LED:
+
+  stop_id = 0
+  resolver = None
+
+  def __init__(self, stop_id, resolver):
+    self.stop_id = stop_id
+    self.resolver = resolver
+
+  def data(self, result):
+    data = convert(json.loads(result))
+    logger.info("LED:data(result) for stop %r" % self.stop_id)
+    if 'predictions' in data:
+      self.resolver.resolve(data['predictions'])
+    elif 'error' in data:
+      self.error("Error in getting predictions for %r: %r" % (self.stop_id, data['error']))
+    else:
+      self.error('Unkown error occurred.')
+
+  def error(self, error):
+    logger.error("Error for LED %r: %r" % (self.stop_id, error))
+    self.resolver.error()

app/component/__init__.py

@@ -0,0 +1,64 @@
+from threading import Timer
+
+class Blinker:
+  
+  flag = 0
+  pins = []
+  shifter = None
+  timer = None
+  active = False
+  delay = 0.5 # seconds
+
+  def __init__(self, shifter, delay=0.5):
+    self.flag = 0
+    self.shifter = shifter
+    self.delay = delay
+
+  def set_pins(self, pins):
+    self.pins = pins
+
+  def blink(self):
+    if self.flag % 2 == 0:
+      self.shifter.set_pins(self.pins, 1)
+    else:
+      self.shifter.set_pins(self.pins, 0)
+    
+    self.shifter.write()
+    self.timer = Timer(self.delay, self.blink)
+    self.timer.start()
+    self.flag = 0 if self.flag == 1 else 1
+
+  def start(self):
+    if self.active == False:
+      self.active = True
+      self.flag = 0
+      self.blink()
+
+  def stop(self):
+    self.active = False
+    if not self.timer == None:
+      self.timer.cancel()
+      self.shifter.set_pins(self.pins, 0)
+      self.shifter.write()
+
+class BlinkerPool:
+
+  pool = []
+
+  def __init__(self):
+    self.pool = []
+
+  def get(self, shifter, delay=0.5):
+    blinker = Blinker(shifter, delay)
+    self.pool.append(blinker)
+    return blinker
+
+  def stop_all(self):
+    for blinker in self.pool:
+      blinker.stop()
+
+  def drain(self):
+    self.stop_all()
+    del self.pool[:]
+
+blinker_pool = BlinkerPool()