mirror of https://github.com/NekoX-Dev/NekoX.git
92 lines
2.7 KiB
C++
92 lines
2.7 KiB
C++
//
|
|
// 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 "threading.h"
|
|
#include "utils.h"
|
|
#include "BlockingQueue.h"
|
|
#include "Buffers.h"
|
|
#include "EchoCanceller.h"
|
|
#include "MediaStreamItf.h"
|
|
|
|
#include <atomic>
|
|
#include <cstdint>
|
|
#include <vector>
|
|
|
|
struct OpusEncoder;
|
|
|
|
namespace tgvoip
|
|
{
|
|
|
|
class OpusEncoder
|
|
{
|
|
public:
|
|
TGVOIP_DISALLOW_COPY_AND_ASSIGN(OpusEncoder);
|
|
OpusEncoder(MediaStreamItf* m_source, bool needSecondary);
|
|
virtual ~OpusEncoder();
|
|
virtual void Start();
|
|
virtual void Stop();
|
|
|
|
using CallbackType = std::function<void(std::uint8_t* data, std::size_t length, std::uint8_t* secondaryData, std::size_t secondaryLength)>;
|
|
void SetCallback(CallbackType callback);
|
|
|
|
void SetBitrate(std::uint32_t bitrate);
|
|
void SetEchoCanceller(EchoCanceller* aec);
|
|
void SetOutputFrameDuration(std::uint32_t duration);
|
|
void SetPacketLoss(int percent);
|
|
int GetPacketLoss() const;
|
|
std::uint32_t GetBitrate() const;
|
|
void SetDTX(bool enable);
|
|
void SetLevelMeter(AudioLevelMeter* m_levelMeter);
|
|
void SetSecondaryEncoderEnabled(bool enabled);
|
|
void SetVadMode(bool vad);
|
|
void AddAudioEffect(effects::AudioEffect* effect);
|
|
void RemoveAudioEffect(effects::AudioEffect* effect);
|
|
int GetComplexity() const;
|
|
|
|
private:
|
|
std::vector<effects::AudioEffect*> m_postProcEffects;
|
|
BlockingQueue<Buffer> m_queue;
|
|
BufferPool<960 * 2, 10> m_bufferPool;
|
|
|
|
MediaStreamItf* m_source;
|
|
::OpusEncoder* m_enc;
|
|
::OpusEncoder* m_secondaryEncoder;
|
|
EchoCanceller* m_echoCanceller;
|
|
AudioLevelMeter* m_levelMeter;
|
|
Thread* m_thread;
|
|
|
|
CallbackType m_callback;
|
|
|
|
std::atomic<std::uint32_t> m_requestedBitrate;
|
|
std::uint32_t m_currentBitrate;
|
|
std::uint32_t m_frameDuration;
|
|
std::uint32_t m_vadNoVoiceBitrate;
|
|
std::atomic<int> m_complexity;
|
|
int m_packetLossPercent;
|
|
int m_secondaryEnabledBandwidth;
|
|
int m_vadModeVoiceBandwidth;
|
|
int m_vadModeNoVoiceBandwidth;
|
|
|
|
unsigned char m_buffer[4096];
|
|
|
|
std::atomic<bool> m_running;
|
|
std::atomic<bool> m_secondaryEncoderEnabled;
|
|
bool m_wasSecondaryEncoderEnabled = false;
|
|
bool m_vadMode = false;
|
|
|
|
static std::size_t Callback(std::uint8_t* data, std::size_t len, void* param);
|
|
void RunThread();
|
|
void Encode(std::int16_t* data, std::size_t len);
|
|
void InvokeCallback(std::uint8_t* data, std::size_t length, std::uint8_t* secondaryData, std::size_t secondaryLength);
|
|
};
|
|
|
|
} // namespace tgvoip
|
|
|
|
#endif // LIBTGVOIP_OPUSENCODER_H
|