libtgvoip/OpusEncoder.h

//
// libtgvoip is free and unencumbered public domain software.
// For more information, see http://unlicense.org or the UNLICENSE file
// you should have received with this source code distribution.
//

#ifndef LIBTGVOIP_OPUSENCODER_H
#define LIBTGVOIP_OPUSENCODER_H


#include "MediaStreamItf.h"
#include "threading.h"
#include "BlockingQueue.h"
#include "Buffers.h"
#include "EchoCanceller.h"
#include "utils.h"

#include <stdint.h>
#include <atomic>

struct OpusEncoder;

namespace tgvoip{
class OpusEncoder{
public:
	TGVOIP_DISALLOW_COPY_AND_ASSIGN(OpusEncoder);
	OpusEncoder(MediaStreamItf* source, bool needSecondary);
	virtual ~OpusEncoder();
	virtual void Start();
	virtual void Stop();
	void SetBitrate(uint32_t bitrate);
	void SetEchoCanceller(EchoCanceller* aec);
	void SetOutputFrameDuration(uint32_t duration);
	void SetPacketLoss(int percent);
	int GetPacketLoss();
	uint32_t GetBitrate();
	void SetDTX(bool enable);
	void SetLevelMeter(AudioLevelMeter* levelMeter);
	void SetCallback(std::function <void(unsigned char*, size_t, unsigned char*, size_t)> callback);
	void SetSecondaryEncoderEnabled(bool enabled);
	void SetVadMode(bool vad);
	void AddAudioEffect(effects::AudioEffect* effect);
	void RemoveAudioEffect(effects::AudioEffect* effect);
	int GetComplexity(){
		return complexity;
	}

private:
	static size_t Callback(unsigned char* data, size_t len, void* param);
	void RunThread();
	void Encode(int16_t* data, size_t len);
	void InvokeCallback(unsigned char* data, size_t length, unsigned char* secondaryData, size_t secondaryLength);
	MediaStreamItf* source;
	::OpusEncoder* enc;
	::OpusEncoder* secondaryEncoder;
	unsigned char buffer[4096];
	std::atomic<uint32_t> requestedBitrate;
	uint32_t currentBitrate;
	Thread* thread;
	BlockingQueue<Buffer> queue;
	BufferPool<960*2, 10> bufferPool;
	EchoCanceller* echoCanceller;
	std::atomic<int> complexity;
	std::atomic<bool> running;
	uint32_t frameDuration;
	int packetLossPercent;
	AudioLevelMeter* levelMeter;
    std::atomic<bool> secondaryEncoderEnabled;
	bool vadMode=false;
	uint32_t vadNoVoiceBitrate;
	std::vector<effects::AudioEffect*> postProcEffects;
	int secondaryEnabledBandwidth;
	int vadModeVoiceBandwidth;
	int vadModeNoVoiceBandwidth;

	bool wasSecondaryEncoderEnabled=false;

	std::function <void(unsigned char*, size_t, unsigned char*, size_t)> callback;
};
}

#endif //LIBTGVOIP_OPUSENCODER_H