Nagram/TMessagesProj/jni/libtgvoip3/EchoCanceller.h
2020-04-24 12:21:58 +03:00

102 lines
2.2 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_ECHOCANCELLER_H
#define LIBTGVOIP_ECHOCANCELLER_H
#include "threading.h"
#include "utils.h"
#include "BlockingQueue.h"
#include "Buffers.h"
#include "MediaStreamItf.h"
#include <cstdint>
namespace webrtc
{
class AudioProcessing;
class AudioFrame;
} // namespace webrtc
namespace tgvoip
{
class EchoCanceller
{
public:
TGVOIP_DISALLOW_COPY_AND_ASSIGN(EchoCanceller);
EchoCanceller(bool m_enableAEC, bool m_enableNS, bool m_enableAGC);
virtual ~EchoCanceller();
virtual void Start();
virtual void Stop();
void SpeakerOutCallback(std::uint8_t* data, std::size_t len);
void Enable(bool enabled);
void ProcessInput(std::int16_t* inOut, std::size_t numSamples, bool& hasVoice);
void SetAECStrength(int strength);
void SetVoiceDetectionEnabled(bool enabled);
private:
#ifndef TGVOIP_NO_DSP
BufferPool<960 * 2, 10> m_farendBufferPool;
BlockingQueue<Buffer>* m_farendQueue;
Thread* m_bufferFarendThread;
webrtc::AudioProcessing* m_apm = nullptr;
webrtc::AudioFrame* m_audioFrame = nullptr;
bool m_didBufferFarend;
bool m_running;
void RunBufferFarendThread();
#endif
bool m_enableAEC;
bool m_enableAGC;
bool m_enableNS;
bool m_enableVAD = false;
bool m_isOn;
};
namespace effects
{
class AudioEffect
{
public:
virtual ~AudioEffect() = 0;
virtual void Process(std::int16_t* inOut, std::size_t numSamples) const = 0;
virtual void SetPassThrough(bool m_passThrough);
protected:
[[nodiscard]] bool GetPassThrough() const;
private:
bool m_passThrough = false;
};
class Volume : public AudioEffect
{
public:
Volume();
~Volume() override;
void Process(std::int16_t* inOut, std::size_t numSamples) const override;
/**
* Level is (0.0, 2.0]
*/
void SetLevel(float m_level);
[[nodiscard]] float GetLevel() const;
private:
float m_level = 1.0f;
float m_multiplier = 1.0f;
};
} // namespace effects
} // namespace tgvoip
#endif // LIBTGVOIP_ECHOCANCELLER_H