mirror of
https://github.com/PhotonVision/photonvision
synced 2026-06-19 00:41:41 +00:00
86 lines
2.9 KiB
Python
86 lines
2.9 KiB
Python
import tornado.websocket
|
|
import json
|
|
from ..classes.Exceptions import NoCameraConnectedException
|
|
from ..classes.SettingsManager import SettingsManager
|
|
|
|
|
|
class ChameleonWebSocket(tornado.websocket.WebSocketHandler):
|
|
actions = {}
|
|
|
|
set_this_camera_settings = ["exposure", "brightness", "video_mode"]
|
|
|
|
def __init__(self, application, request, **kwargs):
|
|
super().__init__(application, request, **kwargs)
|
|
self.settings_manager = SettingsManager()
|
|
self.init_actions()
|
|
|
|
def init_actions(self):
|
|
self.actions["change_pipeline_values"] = self.change_pipeline_values
|
|
self.actions["change_general_settings_values"] = self.settings_manager.change_general_settings_values
|
|
self.actions["change_cam"] = self.change_curr_camera
|
|
self.actions["change_pipeline"] = self.change_curr_pipeline
|
|
|
|
def open(self):
|
|
|
|
self.send_full_settings()
|
|
|
|
print("WebSocket opened")
|
|
|
|
def on_message(self, message):
|
|
|
|
message_dic = json.loads(message)
|
|
|
|
for key in message_dic:
|
|
self.actions.get(key, self.actions["change_pipeline_values"])(message_dic)
|
|
|
|
print(message)
|
|
|
|
def on_close(self):
|
|
self.settings_manager.save_settings()
|
|
|
|
print("WebSocket closed")
|
|
|
|
def check_origin(self, origin):
|
|
return True
|
|
|
|
def send_curr_pipeline(self):
|
|
try:
|
|
self.write_message(self.settings_manager.get_curr_pipeline())
|
|
except NoCameraConnectedException:
|
|
# TODO: return something if no camera connected
|
|
self.write_message("No camera connected")
|
|
|
|
def send_curr_cam(self):
|
|
try:
|
|
self.write_message(self.settings_manager.get_curr_cam())
|
|
except NoCameraConnectedException:
|
|
# TODO: return something if no camera connected
|
|
self.write_message("No camera connected")
|
|
|
|
def send_full_settings(self):
|
|
full_settings = self.settings_manager.general_settings.copy()
|
|
full_settings["cameraList"] = list(self.settings_manager.cams.copy().keys())
|
|
try:
|
|
full_settings.update(self.settings_manager.get_curr_pipeline())
|
|
except NoCameraConnectedException:
|
|
# TODO: return something if no camera connected
|
|
full_settings["data"] = None
|
|
|
|
self.write_message(full_settings)
|
|
|
|
def change_curr_camera(self, dic):
|
|
self.settings_manager.set_curr_camera(cam_name=dic["cam"])
|
|
self.send_curr_cam()
|
|
|
|
def change_curr_pipeline(self, dic):
|
|
self.settings_manager.set_curr_pipeline(pipe_name=dic["pipeline"])
|
|
self.send_curr_pipeline()
|
|
|
|
def change_pipeline_values(self, dic):
|
|
self.settings_manager.change_pipeline_values(dic)
|
|
|
|
for key in self.set_this_camera_settings:
|
|
if key in dic:
|
|
self.settings_manager.set_camera_settings(self.settings_manager.general_settings["curr_camera"],
|
|
dic)
|