added server file
This commit is contained in:
parent
65afc9d151
commit
60fbfe59a7
2 changed files with 109 additions and 1 deletions
108
Server/sockets_controller.py
Normal file
108
Server/sockets_controller.py
Normal file
|
@ -0,0 +1,108 @@
|
||||||
|
import RPi.GPIO as GPIO
|
||||||
|
import time
|
||||||
|
import sys
|
||||||
|
import websockets
|
||||||
|
import asyncio
|
||||||
|
|
||||||
|
pin_right = 13 # blue right
|
||||||
|
pin_left = 6 # purple left
|
||||||
|
|
||||||
|
pin_up = 20 # white up
|
||||||
|
pin_down = 21 # green down
|
||||||
|
|
||||||
|
|
||||||
|
GPIO.setmode(GPIO.BCM)
|
||||||
|
|
||||||
|
GPIO.setup(pin_up, GPIO.OUT)
|
||||||
|
GPIO.setup(pin_down, GPIO.OUT)
|
||||||
|
GPIO.setup(pin_left, GPIO.OUT)
|
||||||
|
GPIO.setup(pin_right, GPIO.OUT)
|
||||||
|
|
||||||
|
# PWM(pin, frequency_in_Hz)
|
||||||
|
pwm_pin_up = GPIO.PWM(pin_up, 150)
|
||||||
|
pwm_pin_down = GPIO.PWM(pin_down, 150)
|
||||||
|
|
||||||
|
pwm_pin_left = GPIO.PWM(pin_left, 150)
|
||||||
|
pwm_pin_right = GPIO.PWM(pin_right, 150)
|
||||||
|
|
||||||
|
socket_connections = {}
|
||||||
|
|
||||||
|
async def handle_client(websocket, path):
|
||||||
|
global socket_connections
|
||||||
|
|
||||||
|
print("packet connected")
|
||||||
|
socket_connections[websocket] = ""
|
||||||
|
|
||||||
|
try:
|
||||||
|
while True:
|
||||||
|
data = await websocket.recv()
|
||||||
|
for command in data.split(":"):
|
||||||
|
print(f"data is: {command}")
|
||||||
|
await update_from_packet(command)
|
||||||
|
except websockets.exceptions.ConnectionClosed:
|
||||||
|
print("Connection closed with: ", websocket.remote_address)
|
||||||
|
finally:
|
||||||
|
print("closing websocket")
|
||||||
|
del socket_connections[websocket]
|
||||||
|
|
||||||
|
axis_x = 0
|
||||||
|
axis_y = 0
|
||||||
|
async def update_from_packet(data):
|
||||||
|
global axis_x, axis_y
|
||||||
|
|
||||||
|
if data[0] == 'L':
|
||||||
|
axis_x = -int(data[1:])
|
||||||
|
if data[0] == 'R':
|
||||||
|
axis_x = int(data[1:])
|
||||||
|
if data[0] == 'U':
|
||||||
|
axis_y = int(data[1:])
|
||||||
|
if data[0] == 'D':
|
||||||
|
axis_y = -int(data[1:])
|
||||||
|
|
||||||
|
# if speed_up > 0:
|
||||||
|
if axis_y > 0:
|
||||||
|
pwm_pin_down.stop()
|
||||||
|
pwm_pin_up.start(min(axis_y, 100))
|
||||||
|
GPIO.output(pin_down, GPIO.LOW)
|
||||||
|
|
||||||
|
# if speed_down > 0:
|
||||||
|
if axis_y < 0:
|
||||||
|
pwm_pin_up.stop()
|
||||||
|
pwm_pin_down.start(min(abs(axis_y), 100))
|
||||||
|
GPIO.output(pin_up, GPIO.LOW)
|
||||||
|
|
||||||
|
#if speed_left > 0:
|
||||||
|
if axis_x < 0:
|
||||||
|
pwm_pin_right.stop()
|
||||||
|
pwm_pin_left.start(min(abs(axis_x), 100))
|
||||||
|
GPIO.output(pin_right, GPIO.LOW)
|
||||||
|
|
||||||
|
# if speed_right > 0:
|
||||||
|
if axis_x > 0:
|
||||||
|
pwm_pin_left.stop()
|
||||||
|
pwm_pin_right.start(min(axis_x, 100))
|
||||||
|
GPIO.output(pin_left, GPIO.LOW)
|
||||||
|
|
||||||
|
if axis_y == 0:
|
||||||
|
pwm_pin_up.stop()
|
||||||
|
pwm_pin_down.stop()
|
||||||
|
if axis_x == 0:
|
||||||
|
pwm_pin_left.stop()
|
||||||
|
pwm_pin_right.stop()
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
server = await websockets.serve(handle_client, "0.0.0.0", 8765)
|
||||||
|
|
||||||
|
print("server started")
|
||||||
|
|
||||||
|
await server.wait_closed()
|
||||||
|
|
||||||
|
try:
|
||||||
|
asyncio.run(main())
|
||||||
|
finally:
|
||||||
|
pwm_pin_up.stop()
|
||||||
|
pwm_pin_down.stop()
|
||||||
|
pwm_pin_left.stop()
|
||||||
|
pwm_pin_right.stop()
|
||||||
|
|
||||||
|
GPIO.cleanup()
|
|
@ -1 +1 @@
|
||||||
Subproject commit 02cfda62a2ada1de4a71251a1077750da44801df
|
Subproject commit 3609b9a848c56ea0d014e4de1a009fc9cc8d5734
|
Loading…
Reference in a new issue