CameraFrameSourceESP32.h

Go to the documentation of this file.

#pragma once
#include "AudioTools/Video/Video.h"
#include "esp_camera.h"
 
namespace audio_tools {
 
class CameraFrameSource : public VideoFrameSource {
 public:
  CameraFrameSource() = default;
  CameraFrameSource(VideoInfo info) : info(info) {}
 
  void setCameraConfig(camera_config_t config) { cfg = config; }
 
  void setVideoInfo(VideoInfo info) { this->info = info; }
 
  bool begin() { return esp_camera_init(&cfg) == ESP_OK; }
 
  void end() {
    releaseFrame();
    esp_camera_deinit();
  }
 
  const uint8_t *nextFrame(size_t &len) override {
    releaseFrame();
    fb = esp_camera_fb_get();
    if (fb == nullptr) {
      len = 0;
      return nullptr;
    }
    len = fb->len;
    return fb->buf;
  }
 
  VideoInfo videoInfo() override { return info; }
 
 protected:
  camera_config_t cfg{};
  camera_fb_t *fb = nullptr;
  VideoInfo info;
 
  void releaseFrame() {
    if (fb != nullptr) {
      esp_camera_fb_return(fb);
      fb = nullptr;
    }
  }
};
 
using CameraFrameSource = CameraFrameSourceESP32;
 
}  // namespace audio_tools