Source repositorynv12_zero_device.h

Go to the documentation of this file.

#pragma once
 
// local includes
#include "src/platform/common.h"
 
struct AVFrame;
 
#ifdef __OBJC__
@class AVVideo;
#else
struct AVVideo;
#endif
 
namespace platf {
  void free_frame(AVFrame *frame);

  class nv12_zero_device: public avcodec_encode_device_t {
    AVVideo *display = nullptr;  
 
  public:
    // this function is used to set the resolution on an av_video object that we cannot
    // call directly because of namespace collisions between AVFoundation and FFMPEG
    using resolution_fn_t = std::function<void(AVVideo *display, int width, int height)>;
    resolution_fn_t resolution_fn;  
    using pixel_format_fn_t = std::function<void(AVVideo *display, int pixelFormat)>;

    int init(AVVideo *capture_display, pix_fmt_e pix_fmt, resolution_fn_t set_resolution, const pixel_format_fn_t &pixel_format_fn);

    int convert(img_t &img) override;
    int set_frame(AVFrame *new_frame, AVBufferRef *hw_frames_ctx) override;
 
  private:
    util::safe_ptr<AVFrame, &free_frame> av_frame;
  };
 
}  // namespace platf