#ifndef _cffmpeg_buz_recorder_hpp_
|
#define _cffmpeg_buz_recorder_hpp_
|
|
#include <stdint.h>
|
#include <string>
|
#include <list>
|
#include <memory>
|
|
#include <thread>
|
#include <atomic>
|
#include <mutex>
|
#include <condition_variable>
|
|
#include "../common/callback.hpp"
|
|
#include "../common.hpp"
|
|
struct AVPacket;
|
|
namespace ffwrapper{
|
class FormatIn;
|
class FormatOut;
|
}
|
|
namespace cffmpeg_wrap{
|
namespace buz{
|
|
class Recorder{
|
public:
|
Recorder(ffwrapper::FormatIn *in, const std::string &id);
|
~Recorder();
|
|
public:
|
int Run(const char* output, const int mind, const int maxd, const bool audio);
|
int PushPacket(const CPacket &pkt);
|
int PushPackets(std::list<CPacket> &lst);
|
int FireRecorder(const int64_t &id);
|
|
void SetCallback(FUNC_REC_INFO cb){
|
func_rec_info_ = cb;
|
}
|
|
const bool ErrorOcurred(){return error_occured_;}
|
const std::string& RecID()const{return id_;}
|
private:
|
void run_thread();
|
|
int init_writer(const bool audio);
|
int write_correctly(const CPacket &pkt);
|
void end_writer();
|
|
void maybe_dump_gop();
|
private:
|
ffwrapper::FormatIn *in_;
|
ffwrapper::FormatOut *out_;
|
|
int maxduration;
|
int minduration;
|
int end_frame;
|
int cur_frame;
|
int cur_frame_a;
|
|
std::list<CPacket> list_pkt_;
|
|
std::atomic_bool stop_recorder_;
|
std::mutex mutex_pkt_;
|
std::condition_variable cv_;
|
|
std::unique_ptr<std::thread> thrd_;
|
|
std::string dir_;
|
std::string id_;
|
|
int64_t id_frame_;
|
int id_frame_in_file_;
|
std::string file_path_;
|
FUNC_REC_INFO func_rec_info_;
|
|
bool error_occured_;
|
|
bool audio_;
|
};
|
}
|
}
|
|
#endif
|
