115 lines
2.8 KiB
Python
115 lines
2.8 KiB
Python
import psn
|
|
import time
|
|
import socket
|
|
import json
|
|
from aiohttp import web
|
|
import logging
|
|
from dataclasses import dataclass, make_dataclass
|
|
|
|
PSN_DEFAULT_UDP_PORT = 56565
|
|
PSN_DEFAULT_UDP_MCAST_ADDRESS = "236.10.10.10"
|
|
PORT = 8000
|
|
IP = "0.0.0.0"
|
|
NUM_TRACKERS = 3
|
|
|
|
@dataclass
|
|
class TrackerData:
|
|
id: int
|
|
x: float
|
|
y: float
|
|
|
|
|
|
trackers = {}
|
|
ws_clients = set()
|
|
|
|
for i in range(NUM_TRACKERS):
|
|
trackers[i] = psn.Tracker(i, f"Tracker {i}")
|
|
trackers[i].set_pos(psn.Float3(0, 0, 0))
|
|
|
|
def update_tracker(tracker_data_json: str):
|
|
global trackers
|
|
tracker = TrackerData(**json.loads(tracker_data_json))
|
|
trackers[tracker.id].set_pos(psn.Float3(tracker.x, tracker.y, 0))
|
|
|
|
def trackers_to_json():
|
|
return json.dumps([{"id": tracker.get_id(), "x": tracker.get_pos().x, "y": tracker.get_pos().y} for tracker in trackers.values()])
|
|
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
|
|
def get_time_ms():
|
|
return int(time.time() * 1000)
|
|
|
|
|
|
START_TIME = get_time_ms()
|
|
|
|
|
|
def get_elapsed_time_ms():
|
|
return get_time_ms() - START_TIME
|
|
|
|
|
|
def pic_to_scene_coords(x, y):
|
|
return x / 200, y / 200
|
|
|
|
async def update_all_clients():
|
|
for ws in ws_clients:
|
|
await ws.send_str(trackers_to_json())
|
|
|
|
def send_psn_positions():
|
|
encoder = psn.Encoder("Server 1")
|
|
|
|
packets = encoder.encode_data(trackers, get_elapsed_time_ms())
|
|
for packet in packets:
|
|
sock.sendto(packet, (PSN_DEFAULT_UDP_MCAST_ADDRESS, PSN_DEFAULT_UDP_PORT))
|
|
|
|
|
|
async def handle_websocket(request):
|
|
ws = web.WebSocketResponse()
|
|
logging.debug("Websocket connection starting")
|
|
await ws.prepare(request)
|
|
logging.debug("Websocket connection ready")
|
|
|
|
ws_clients.add(ws)
|
|
|
|
await ws.send_str(trackers_to_json())
|
|
|
|
try:
|
|
async for msg in ws:
|
|
logging.debug(f"Websocket data: {msg}")
|
|
if msg.type == web.WSMsgType.TEXT:
|
|
# Each message is a single tracker object
|
|
update_tracker(msg.data)
|
|
await update_all_clients()
|
|
|
|
elif msg.type == web.WSMsgType.ERROR:
|
|
logging.error("ws connection closed with exception %s" % ws.exception())
|
|
print("ws connection closed with exception %s" % ws.exception())
|
|
except Exception as e:
|
|
logging.error(f"Websocket exception: {e}")
|
|
|
|
finally:
|
|
logging.debug("Websocket connection closing")
|
|
await ws.close()
|
|
|
|
ws_clients.remove(ws)
|
|
|
|
return ws
|
|
|
|
|
|
|
|
|
|
async def handle_root(request):
|
|
return web.FileResponse("./static/index.html")
|
|
|
|
|
|
def create_app():
|
|
app = web.Application()
|
|
app.router.add_get("/", handle_root)
|
|
app.router.add_get("/ws", handle_websocket)
|
|
app.router.add_static("/", "./static")
|
|
return app
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
web.run_app(create_app(), host=IP, port=PORT)
|