package com.chameleonvision.vision.camera;

import com.chameleonvision.vision.Pipeline;
import edu.wpi.cscore.*;
import edu.wpi.first.cameraserver.CameraServer;
import org.opencv.core.Mat;

import java.util.HashMap;
import java.util.stream.IntStream;

public class Camera {

	private static double defaultFOV = 60.8;

	public final String name;
	public final String path;

	public final UsbCamera UsbCam;
	private final UsbCameraInfo UsbCamInfo;
	private final VideoMode[] availableVideoModes;

	private final CameraServer cs = CameraServer.getInstance();
	private final CvSink cvSink;
	private CvSource cvSource;

	private double FOV;

	private CameraValues camVals;
	private CamVideoMode camVideoMode;

	private int currentPipelineIndex;
	private HashMap<Integer, Pipeline> pipelines = new HashMap<>();

	public Camera(String cameraName) {
		this(cameraName, defaultFOV);
	}

	public Camera(UsbCameraInfo usbCamInfo) {
		this(usbCamInfo, defaultFOV);
	}

	public Camera(String cameraName, double fov) {
		this(CameraManager.AllUsbCameraInfosByName.get(cameraName), fov);
	}

	public Camera(UsbCameraInfo usbCamInfo, double fov) {
		UsbCamInfo = usbCamInfo;
		FOV = fov;
		name = usbCamInfo.name;
		path = usbCamInfo.path;

		UsbCam = new UsbCamera(name, path);

		// set up video mode
		availableVideoModes = UsbCam.enumerateVideoModes();
		setCamVideoMode(new CamVideoMode(availableVideoModes[0]));

		cvSink = cs.getVideo(UsbCam);
		cvSource = cs.putVideo(name, camVals.ImageWidth, camVals.ImageHeight);
	}

	public void setCamVideoMode(int videoMode) {
		setCamVideoMode(UsbCam.enumerateVideoModes()[videoMode]);
	}

	private void setCamVideoMode(VideoMode videoMode) {
		setCamVideoMode(new CamVideoMode(videoMode));
	}

	private void setCamVideoMode(CamVideoMode newVideoMode) {
		var prevVideoMode = this.camVideoMode;
		this.camVideoMode = newVideoMode;
		UsbCam.setPixelFormat(newVideoMode.getActualPixelFormat());
		UsbCam.setFPS(newVideoMode.fps);
		UsbCam.setResolution(newVideoMode.width, newVideoMode.height);

		// update camera values
		camVals = new CameraValues(this);
		if ( prevVideoMode != null && prevVideoMode.width != newVideoMode.width && prevVideoMode.height != newVideoMode.height) { //  if resolution changed
			cvSource = cs.putVideo(name, newVideoMode.width, newVideoMode.height);
		}
	}

	public void addPipeline() {
		addPipeline(pipelines.size());
	}

	private void addPipeline(int pipelineNumber) {
		if (pipelines.containsKey(pipelineNumber)) return;
		pipelines.put(pipelineNumber, new Pipeline());
	}

	public Pipeline getCurrentPipeline() {
		return pipelines.get(currentPipelineIndex);
	}

	public int getCurrentPipelineIndex() {
		return currentPipelineIndex;
	}

	public void setCurrentPipelineIndex(int pipelineNumber) {
		if (pipelineNumber - 1 > pipelines.size()) return;
		currentPipelineIndex = pipelineNumber;
	}
	public HashMap<Integer, Pipeline> getPipelines() {
		return pipelines;
	}

	public CamVideoMode getVideoMode() {
		return camVideoMode;
	}

	public int getVideoModeIndex() {
		return IntStream.range(0, availableVideoModes.length)
				.filter(i -> camVideoMode.isEqualToVideoMode(availableVideoModes[i]))
				.findFirst()
				.orElse(-1);
	}

	public double getFOV() {
		return FOV;
	}

	public void setFOV(double fov) {
		FOV = fov;
		camVals = new CameraValues(this);
	}

	public int getBrightness() {
		return getCurrentPipeline().brightness;
	}

	public void setBrightness(int brightness) {
		getCurrentPipeline().brightness = brightness;
		UsbCam.setBrightness(brightness);
	}

	public void setExposure(int exposure) {
		getCurrentPipeline().exposure = exposure;
		UsbCam.setExposureManual(exposure);
	}

	public long grabFrame(Mat image) {
	    return cvSink.grabFrame(image);
    }

    public long grabFrame(Mat image, double timeout) {
	    return cvSink.grabFrame(image, timeout);
    }

    public void putFrame(Mat image) {
	    cvSource.putFrame(image);
    }
}