Team2890/PhotonVision
4
0
Fork 0
mirror of https://github.com/PhotonVision/photonvision synced 2026-06-20 00:51:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Websockets almost done TODO ports, testing

Browse Source
This commit is contained in:
Omer
2019-09-16 04:10:26 +03:00
parent d04096f2c8
commit 00552c751a
5 changed files with 235 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

141
Main/src/main/java/com/chameleonvision/web/Server.java
View File

@@ -1,13 +1,15 @@
package com.chameleonvision.web;
import com.chameleonvision.NoCameraException;
import com.chameleonvision.settings.SettingsManager;
import com.chameleonvision.vision.GeneralSettings;
import com.chameleonvision.vision.Pipeline;
import com.google.gson.JsonObject;
import io.javalin.Javalin;
import io.javalin.websocket.WsContext;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;
import java.util.*;
import org.json.JSONArray;
import org.json.JSONObject;
@@ -16,13 +18,15 @@ import org.springframework.beans.BeanUtils;
public class Server {
private static List<WsContext> users = new ArrayList<WsContext>();
public static void main(int port) {
Javalin app = Javalin.create();
app.config.addStaticFiles("web");
app.ws("/websocket", ws ->{
app.ws("/websocket", ws -> {
ws.onConnect(ctx -> {
users.add(ctx);
System.out.println("Socket Connected");
sendFullSettings();
});
ws.onClose(ctx -> {
users.remove(ctx);
@@ -30,36 +34,133 @@ public class Server {
SettingsManager.getInstance().SaveSettings();
});
ws.onMessage(ctx -> {
// System.out.println(SettingsManager.getInstance().GetCurrentPipeline().);
broadcastMessage(ctx, ctx.message());
JSONObject jsonObject = new JSONObject(ctx.message());
String key =jsonObject.keySet().toArray()[0].toString();
String key = jsonObject.keySet().toArray()[0].toString();
Object value = jsonObject.get(key);
System.out.println("Key: "+key+" Value: "+value);
Field[] fields = Pipeline.class.getFields();
for (Field f : fields)
{//TODO: check calibration in output tab for crashes
if(f.getName().equals(key))
{
if(BeanUtils.isSimpleValueType(value.getClass()))
f.set(SettingsManager.getInstance().GetCurrentPipeline(),value);
else
if(value.getClass()==JSONArray.class){
f.set(SettingsManager.getInstance().GetCurrentPipeline(),((JSONArray)value).toList());
}
if (!setField(SettingsManager.getInstance().GetCurrentPipeline(), key, value)) {
//If field not in pipeline
switch (key) {
case "change_general_settings_values":
JSONObject newSettings = (JSONObject) value;
setFields(SettingsManager.getInstance().GeneralSettings, newSettings);
break;
case "curr_camera":
SettingsManager.getInstance().SetCurrentCamera((String) value);
//broadcastMessage((Map<String, Object>) new HashMap<String, Object>(){}.put("port",SettingsManager.CameraPorts.get(SettingsManager.GeneralSettings.curr_camera)));
//broadcastMessage(SettingsManager.getInstance().GetCurrentCamera());//TODO CHECK JSON FOR CAMERA CHANGE
break;
case "curr_pipeline":
System.out.println("change pipeline");
SettingsManager.getInstance().SetCurrentPipeline((String) value);
SettingsManager.CamerasCurrentPipeline.put(SettingsManager.GeneralSettings.curr_camera, (String) value);
// broadcastMessage(SettingsManager.getInstance().GetCurrentPipeline());//TODO CHECK JSON FOR PIPELINE CHANGE
break;
case "resolution":
System.out.println("change res");
SettingsManager.getInstance().GetCurrentCamera().resolution = (int) value;
SettingsManager.getInstance().SetCameraSettings(SettingsManager.GeneralSettings.curr_camera, "resolution", value);
SettingsManager.getInstance().SaveSettings();
break;
case "fov":
System.out.println("change fov");
SettingsManager.getInstance().GetCurrentCamera().FOV = (double) value;
SettingsManager.getInstance().SaveSettings();
break;
default:
System.out.println("Unexpected value");
break;
}
}
});
});
app.start(port);
}
private static void broadcastMessage(WsContext sendingUser, String message){
for (var user : users)
{
if (user != sendingUser){
public static boolean setField(Object obj, String fieldName, Object value) {
boolean successful = false;
try {
Field[] fields = obj.getClass().getFields();
for (Field f : fields) {
if (f.getName().equals(fieldName)) {
successful = true;
if (BeanUtils.isSimpleValueType(value.getClass())) {
f.set(obj, value);
} else if (value.getClass() == JSONArray.class) {
f.set(obj, ((JSONArray) value).toList());
}
}
}
} catch (IllegalAccessException e) {
return false;
}
return successful;
}
public static boolean setFields(Object obj, JSONObject data) {
boolean successful = false;
try {
Field[] fields = obj.getClass().getFields();
for (Field f : fields) {
if (data.has(f.getName())) {
Object value = data.get(f.getName());
if (BeanUtils.isSimpleValueType(value.getClass())) {
successful = true;
f.set(obj, value);
}
}
}
} catch (IllegalAccessException e) {
return false;
}
return successful;
}
private static void broadcastMessage(WsContext sendingUser, String message) {
for (var user : users) {
if (user != sendingUser) {
user.send(message);
}
}
}
private static void broadcastMessage(Map<String, Object> map) {
for (var user : users) {
user.send(new JSONObject(map).toString());
}
}
private static void addAllFieldsToMap(Map<String, Object> map, Object obj) {
try {
Field[] fields = obj.getClass().getFields();
for (Field field : fields)
map.put(field.getName(), field.get(obj));
} catch (IllegalAccessException e) {
System.err.println("Illegal Access error:" + e.getStackTrace().toString());
}
}
private static void sendFullSettings() {
Map<String, Object> fullSettings = new HashMap<>();
//General settings
addAllFieldsToMap(fullSettings, SettingsManager.GeneralSettings);
fullSettings.put("cameraList", SettingsManager.Cameras.keySet());
try {
addAllFieldsToMap(fullSettings, SettingsManager.getInstance().GetCurrentPipeline());
fullSettings.put("pipelineList", SettingsManager.getInstance().GetCurrentCamera().pipelines.keySet());
fullSettings.put("resolutionList", SettingsManager.getInstance().GetResolutionList());
fullSettings.put("resolution", SettingsManager.getInstance().GetCurrentCamera().resolution);
fullSettings.put("FOV", SettingsManager.getInstance().GetCurrentCamera().FOV);
// fullSettings.put("port", SettingsManager.CameraPorts.get(SettingsManager.GeneralSettings.curr_camera));
} catch (NoCameraException e) {
System.err.println("No camera found!");
//TODO: add message to ui to inform that there are no cameras
}
broadcastMessage(fullSettings);
}
}