allwpilib/cscore/examples/usbcvstream/usbcvstream.cpp

// Copyright (c) FIRST and other WPILib contributors.
// Open Source Software; you can modify and/or share it under the terms of
// the WPILib BSD license file in the root directory of this project.

#include <opencv2/core/core.hpp>

#include "wpi/cs/CvSink.hpp"
#include "wpi/cs/CvSource.hpp"
#include "wpi/cs/MjpegServer.hpp"
#include "wpi/cs/UsbCamera.hpp"
#include "wpi/util/print.hpp"

int main() {
  wpi::cs::UsbCamera camera{"usbcam", 0};
  camera.SetVideoMode(wpi::util::PixelFormat::kMJPEG, 320, 240, 30);
  wpi::cs::MjpegServer mjpegServer{"httpserver", 8081};
  mjpegServer.SetSource(camera);
  wpi::cs::CvSink cvsink{"cvsink"};
  cvsink.SetSource(camera);
  wpi::cs::CvSource cvsource{"cvsource", wpi::util::PixelFormat::kMJPEG, 320,
                             240, 30};
  wpi::cs::MjpegServer cvMjpegServer{"cvhttpserver", 8082};
  cvMjpegServer.SetSource(cvsource);

  cv::Mat test;
  cv::Mat flip;
  for (;;) {
    uint64_t time = cvsink.GrabFrame(test);
    if (time == 0) {
      wpi::util::print("error: {}\n", cvsink.GetError());
      continue;
    }
    wpi::util::print("got frame at time {} size ({}, {})\n", time,
                     test.size().width, test.size().height);
    cv::flip(test, flip, 0);
    cvsource.PutFrame(flip);
  }
}