George Knowlden
7 years ago
commit
100eebf6d5
@ -0,0 +1,2 @@
|
|||||||
|
venv/
|
||||||
|
__pycache__/
|
||||||
@ -0,0 +1,98 @@
|
|||||||
|
import tornado.httpserver
|
||||||
|
import tornado.ioloop
|
||||||
|
import tornado.options
|
||||||
|
import tornado.web
|
||||||
|
import tornado.websocket
|
||||||
|
|
||||||
|
# To handle requests
|
||||||
|
import json
|
||||||
|
|
||||||
|
# Get the host IP
|
||||||
|
import netifaces as ni
|
||||||
|
ni.ifaddresses("wlan0")
|
||||||
|
ip = ni.ifaddresses("wlan0")[ni.AF_INET][0]["addr"]
|
||||||
|
print("Host IP address:", ip)
|
||||||
|
|
||||||
|
# Get the host port
|
||||||
|
from tornado.options import define, options
|
||||||
|
define("port", default=8080, help="run on the given port", type=int)
|
||||||
|
|
||||||
|
# Our program for reading/writing to our output
|
||||||
|
import led
|
||||||
|
|
||||||
|
class IndexHandler(tornado.web.RequestHandler):
|
||||||
|
def get(self):
|
||||||
|
# TODO: render all the controls automatically depending on what we have connected
|
||||||
|
self.render("index.html", ip = ip, port = options.port)
|
||||||
|
|
||||||
|
ws_clients = []
|
||||||
|
|
||||||
|
class WebSocketHandler(tornado.websocket.WebSocketHandler):
|
||||||
|
def open(self):
|
||||||
|
if self not in ws_clients:
|
||||||
|
print("New connection")
|
||||||
|
ws_clients.append(self)
|
||||||
|
#self.write_message("Connected")
|
||||||
|
|
||||||
|
def on_message(self, message):
|
||||||
|
print("Message received:", message)
|
||||||
|
#self.write_message("Message received: " + message)
|
||||||
|
|
||||||
|
message = json.loads(message) #load the json
|
||||||
|
|
||||||
|
if message["message_type"] == "io_request":
|
||||||
|
# We have something that we need to do
|
||||||
|
|
||||||
|
if message["io_type"] == "led":
|
||||||
|
colour = message["led_colour"]
|
||||||
|
operation = message["operation"]
|
||||||
|
|
||||||
|
if operation == "toggle":
|
||||||
|
result = led.toggle(colour)
|
||||||
|
|
||||||
|
response = {"message_type": "io_response",
|
||||||
|
"io_type": "led",
|
||||||
|
"led_colour": colour,
|
||||||
|
"led_state": result}
|
||||||
|
|
||||||
|
send_to_all(json.dumps(response))
|
||||||
|
|
||||||
|
if operation == "read":
|
||||||
|
result = led.read(colour)
|
||||||
|
|
||||||
|
response = {"message_type": "io_response",
|
||||||
|
"io_type": "led",
|
||||||
|
"led_colour": colour,
|
||||||
|
"led_state": result}
|
||||||
|
|
||||||
|
send_to_all(json.dumps(response))
|
||||||
|
|
||||||
|
# if message == "BUTTON_PRESS":
|
||||||
|
# result = led.toggle()
|
||||||
|
# send_to_all("LED_STATE " + str(result))
|
||||||
|
|
||||||
|
# if message == "GET_STATES":
|
||||||
|
# states = led.read("R")
|
||||||
|
# send_to_all("LED_STATE" + str(states))
|
||||||
|
|
||||||
|
def on_close(self):
|
||||||
|
if self in ws_clients:
|
||||||
|
ws_clients.remove(self)
|
||||||
|
print("Connection closed")
|
||||||
|
|
||||||
|
def send_to_all(message):
|
||||||
|
for c in ws_clients:
|
||||||
|
c.write_message(message)
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
tornado.options.parse_command_line()
|
||||||
|
app = tornado.web.Application(
|
||||||
|
handlers=[
|
||||||
|
(r"/", IndexHandler),
|
||||||
|
(r"/ws", WebSocketHandler)
|
||||||
|
]
|
||||||
|
)
|
||||||
|
httpServer = tornado.httpserver.HTTPServer(app)
|
||||||
|
httpServer.listen(options.port)
|
||||||
|
print("Listening on port:", options.port)
|
||||||
|
tornado.ioloop.IOLoop.instance().start()
|
||||||
@ -0,0 +1,77 @@
|
|||||||
|
<!DOCTYPE HTML>
|
||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<style>
|
||||||
|
body { margin: 0px; padding: 0px; }
|
||||||
|
canvas { border: 1px solid #9C9898; }
|
||||||
|
</style>
|
||||||
|
<script>
|
||||||
|
|
||||||
|
var socket = new WebSocket("ws://{{ ip }}:{{ port }}/ws");
|
||||||
|
|
||||||
|
socket.onopen = function(){
|
||||||
|
console.log("Connected");
|
||||||
|
//get initial states
|
||||||
|
var leds = ["R", "G"];
|
||||||
|
for (var i = 0; i < leds.length; i++) {
|
||||||
|
request = {"message_type": "io_request",
|
||||||
|
"io_type": "led",
|
||||||
|
"led_colour": leds[i],
|
||||||
|
"operation": "read"};
|
||||||
|
sendMessage(request);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
socket.onmessage = function (message) {
|
||||||
|
console.log("receiving: " + message.data);
|
||||||
|
processInstruction(message.data);
|
||||||
|
};
|
||||||
|
|
||||||
|
function processInstruction(instruction) {
|
||||||
|
var instruction_json = JSON.parse(instruction);
|
||||||
|
console.log(instruction_json.message_type);
|
||||||
|
console.log(typeof instruction_json.message_type);
|
||||||
|
if (instruction_json.message_type == "io_response") {
|
||||||
|
console.log("here");
|
||||||
|
if (instruction_json.io_type == "led") {
|
||||||
|
console.log("here2");;
|
||||||
|
var colour = instruction_json.led_colour;
|
||||||
|
var state = instruction_json.led_state;
|
||||||
|
document.getElementById(colour).innerHTML = "LED state: " + state;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
socket.onclose = function(){
|
||||||
|
console.log("disconnected");
|
||||||
|
};
|
||||||
|
|
||||||
|
sendMessage = function(message) {
|
||||||
|
socket.send(JSON.stringify(message));
|
||||||
|
};
|
||||||
|
|
||||||
|
function buttonPress(colour) {
|
||||||
|
var request = {"message_type": "io_request",
|
||||||
|
"io_type": "led",
|
||||||
|
"led_colour": colour,
|
||||||
|
"operation": "toggle"};
|
||||||
|
sendMessage(request);
|
||||||
|
}
|
||||||
|
|
||||||
|
</script>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<div id="container"></div>
|
||||||
|
<h1>Controller</h1>
|
||||||
|
<table>
|
||||||
|
<tr>
|
||||||
|
<td><button onClick="buttonPress('R')">Toggle red LED</button></td>
|
||||||
|
<td><p id="R">LED state: ?</p></td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td><button onClick="buttonPress('G')">Toggle green LED</button></td>
|
||||||
|
<td><p id="G">LED state: ?</p></td>
|
||||||
|
</tr>
|
||||||
|
</table>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
@ -0,0 +1,35 @@
|
|||||||
|
import RPi.GPIO as GPIO
|
||||||
|
GPIO.setwarnings(False)
|
||||||
|
GPIO.setmode(GPIO.BCM)
|
||||||
|
|
||||||
|
pins = [19, 26]
|
||||||
|
|
||||||
|
GPIO.setup(pins, GPIO.OUT, initial=GPIO.LOW)
|
||||||
|
|
||||||
|
def colour_to_pin(colour):
|
||||||
|
if colour == "R":
|
||||||
|
return 19
|
||||||
|
if colour == "G":
|
||||||
|
return 26
|
||||||
|
else:
|
||||||
|
return False
|
||||||
|
|
||||||
|
def on(colour):
|
||||||
|
pin = colour_to_pin(colour)
|
||||||
|
GPIO.output(pin, GPIO.HIGH)
|
||||||
|
|
||||||
|
def off(colour):
|
||||||
|
pin = colour_to_pin(colour)
|
||||||
|
GPIO.output(pin, GPIO.LOW)
|
||||||
|
|
||||||
|
def read(colour):
|
||||||
|
pin = colour_to_pin(colour)
|
||||||
|
return GPIO.input(pin)
|
||||||
|
|
||||||
|
def toggle(colour):
|
||||||
|
if read(colour):
|
||||||
|
off(colour)
|
||||||
|
return 0
|
||||||
|
else:
|
||||||
|
on(colour)
|
||||||
|
return 1
|
||||||
@ -0,0 +1,4 @@
|
|||||||
|
netifaces==0.10.9
|
||||||
|
pkg-resources==0.0.0
|
||||||
|
RPi.GPIO==0.6.5
|
||||||
|
tornado==6.0.1
|
||||||
Loading…
Reference in new issue