video-pc/libs/Recorder/record_audio.h

#ifndef RECORD_AUDIO
#define RECORD_AUDIO

#include "record_audio_define.h"

#include "headers_ffmpeg.h"

#include <atomic>
#include <thread>
#include <functional>
#include <string>

namespace am {
	typedef std::function<void(AVFrame *packet, int index)> cb_audio_data;
	typedef std::function<void(int,int)> cb_audio_error;

	class record_audio
	{
	public:
		record_audio();
		virtual ~record_audio();

		virtual int init(const std::string &device_name,
			const std::string &device_id,
			bool is_input) = 0;

		virtual int start() = 0;
		
		virtual int pause() = 0;
		
		virtual int resume() = 0;
		
		virtual int stop() = 0;

		virtual const AVRational get_time_base() = 0;

		virtual int64_t get_start_time() = 0;

	public:
		inline bool is_recording() { return _running; }
		
		inline int get_sample_rate() { return _sample_rate; }
		
		inline int get_bit_rate() { return _bit_rate; }

		inline int get_bit_per_sample() { return _bit_per_sample; }

		inline int get_channel_num() { return _channel_num; }

		inline int64_t av_get_channel_layout() { return _channel_layout; }
		
		inline AVSampleFormat get_fmt() { return _fmt; }

		inline const std::string & get_device_name() { return _device_name; }

		inline void registe_cb(
			cb_audio_data on_data,
			cb_audio_error on_error,
			int cb_extra_index = -1) {
			_on_data = on_data;
			_on_error = on_error;
			_cb_extra_index = cb_extra_index;
		}

	protected:
		std::atomic_bool _running;
		std::atomic_bool _inited;
		std::atomic_bool _paused;

		std::thread _thread;

		int _sample_rate;
		
		int _bit_rate;

		int _channel_num;

		int64_t _channel_layout;

		int _bit_per_sample;
		
		AVSampleFormat _fmt;
		
		std::string _device_name;
		std::string _device_id;

		bool _is_input;

		cb_audio_error _on_error;
		
		cb_audio_data _on_data;

		int _cb_extra_index;
	};
}

#endif // !RECORD_AUDIO