CameraFrameSourceTinyCamera.h

Go to the documentation of this file.

#pragma once
#include "AudioTools/Video/Video.h"
#include "TinyCamera.h"
 
namespace audio_tools {

class CameraFrameSourceTinyCamera : public VideoFrameSource {
 public:
  CameraFrameSourceTinyCamera() = default;

  bool begin(const camera_config_t &config, float fps) {
    info.fps = fps;
    if (!camera_.begin(config)) return false;
    tiny_camera::TinyCameraFrame probe = camera_.captureFrame();
    if (!probe) {
      LOGE("CameraFrameSourceTinyCamera: probe capture failed");
      return false;
    }
    info.width = probe.width();
    info.height = probe.height();
    info.format = toVideoFormat(probe.format());
    info.frame_size = videoFrameSizeBytes(info.format, info.width, info.height);
    LOGI("CameraFrameSourceTinyCamera: %dx%d, format %d", (int)info.width,
         (int)info.height, (int)info.format);
    return true;
  }

  void end() {
    frame_.release();
    camera_.end();
  }

  const uint8_t *nextFrame(size_t &len) override {
    frame_.release();
    frame_ = camera_.captureFrame();
    if (!frame_) {
      len = 0;
      return nullptr;
    }
    len = frame_.size();
    return frame_.data();
  }

  VideoInfo videoInfo() override { return info; }

  tiny_camera::TinyCamera &camera() { return camera_; }
 
 protected:
  tiny_camera::TinyCamera camera_;
  tiny_camera::TinyCameraFrame frame_;
  VideoInfo info;
 
  static VideoFormat toVideoFormat(pixformat_t format) {
    switch (format) {
      case PIXFORMAT_RGB565:
        return VideoFormat::RGB565;
      case PIXFORMAT_YUV422:
        return VideoFormat::YUV422;
      case PIXFORMAT_JPEG:
        return VideoFormat::MJPEG;
#if defined(TINY_CAMERA_ESP_VIDEO)
      case PIXFORMAT_RGB888:
        return VideoFormat::RGB888;
#endif
      default:
        return VideoFormat::UNKNOWN;
    }
  }
};
 
}  // namespace audio_tools