use std::io::Write;
use std::net::{SocketAddr, TcpListener};
use opencv::core::{Mat, Vector};
use opencv::imgcodecs::{imencode, IMWRITE_JPEG_QUALITY};
use opencv::prelude::*;
use opencv::videoio::{VideoCapture, VideoCaptureTraitConst, CAP_ANY};
const BASE_RESPONSE: &[u8] = b"HTTP/1.1 200 OK\r\nContent-Type: multipart/x-mixed-replace; boundary=frame\r\n\r\n";
fn main() -> Result<(), Box<dyn std::error::Error>> {
let mut cam = VideoCapture::new(0, CAP_ANY)?;
assert!(cam.is_opened()?, "Unable to open default camera!");
let address: SocketAddr = "127.0.0.1:8080".parse()?;
let listener = TcpListener::bind(address)?;
println!("Listening for connections at {address}");
let (mut stream, addr) = listener.accept()?;
println!("Client connected: {addr}");
stream.write_all(BASE_RESPONSE)?;
let encode_params = Vector::from_slice(&[IMWRITE_JPEG_QUALITY, 70]);
let mut buffer = Mat::default();
let mut frame = Vector::default();
loop {
cam.read(&mut buffer)?;
imencode(".jpg", &buffer, &mut frame, &encode_params)?;
let header = format!(
"--frame\r\nContent-Type: image/jpeg\r\nContent-Length: {}\r\n\r\n",
frame.len()
);
let packet = [header.as_bytes(), frame.as_slice()].concat();
stream.write_all(&packet)?;
}
}