Hu Chunming / vpt_ascend_arm

  #pragma once
  #include <memory>
  

  class AVFrame;
  class AVStream;
  class AVCodecContext;
  class AVFormatContext;
  class AVPacket;

  
  class FFRecoder
  {
  public:
  	FFRecoder();

  	~FFRecoder();

  	bool init(int w, int h, int fps, int bit_rate, const char* outfile_name);

  	void uninit();
  	bool write_image(const uint8_t* bgr);
  	bool write_yuv(const uint8_t* yuv_data);

  	bool write_frame(const AVFrame* frame);
  	bool write_pkt(AVPacket* pkt);
  	bool write_pkt_data(const uint8_t* data, int size);

  	bool flush();

  	bool close();

  
  private:
  	bool bgr_to_yuv420p(const uint8_t* const buf_bgr, uint8_t* const buf_420p);

  	void calc_pkt_ts(AVPacket* pkt, int frame_index);

  
  private:
  	int width_;
  	int height_;
  	int y_size_;
  	int uv_size_;
  	int pts_;
  	AVCodecContext* codec_ctx_;
  	AVFormatContext* fmt_ctx_;
  	AVStream* out_stream_;
  	AVFrame* yuv_frame_;
  

  	int m_fps{1};

  	int frame_nb{0};

  };