Add video mode change support to VisionProcess

This commit is contained in:
Matt
2019-11-16 11:10:55 -08:00
parent f426552644
commit 58c1924c7e
4 changed files with 34 additions and 3 deletions

View File

@@ -1,5 +1,6 @@
package com.chameleonvision.classabstraction.camera;
import edu.wpi.cscore.VideoMode;
import org.apache.commons.lang3.tuple.Pair;
import org.opencv.core.Mat;
@@ -23,4 +24,10 @@ public interface CameraProcess {
* @param brightness the new brightness to set the camera to
*/
void setBrightness(int brightness);
/**
* Set the video mode (fps and resolution) of the camera
* @param mode the wanted mode
*/
void setVideoMode(VideoMode mode);
}

View File

@@ -3,6 +3,7 @@ package com.chameleonvision.classabstraction.camera;
import com.chameleonvision.vision.camera.StreamDivisor;
import com.chameleonvision.web.ServerHandler;
import edu.wpi.cscore.CvSource;
import edu.wpi.cscore.VideoMode;
import edu.wpi.first.cameraserver.CameraServer;
import org.opencv.core.CvType;
import org.opencv.core.Mat;
@@ -39,6 +40,12 @@ public class CameraStreamer {
ServerHandler.sendFullSettings();
}
public void setNewVideoMode(VideoMode newVideoMode) {
// Trick to update cvSource and streamBuffer to the new resolution
// Must change the cameraProcess resolution first
setDivisor(divisor);
}
public void runStream(Mat image) {
image.copyTo(streamBuffer);
if (divisor.value != 1) {

View File

@@ -13,7 +13,7 @@ public class USBCameraProcess implements CameraProcess {
private final UsbCamera baseCamera;
private final CvSink cvSink;
private Mat imageBuffer = new Mat();
public final CameraProperties properties;
public CameraProperties properties;
public USBCameraProcess(UsbCamera camera, CameraConfig config) {
baseCamera = camera;
@@ -53,4 +53,14 @@ public class USBCameraProcess implements CameraProcess {
System.err.println("Current camera does not support brightness change");
}
}
@Override
public void setVideoMode(VideoMode mode) {
try {
baseCamera.setVideoMode(mode);
properties = new CameraProperties(baseCamera, properties.FOV);
} catch (VideoException e) {
System.err.println("Current camera does not support resolution change");
}
}
}