OpenteraAudioDeviceModule.h

#ifndef OPENTERA_WEBRTC_NATIVE_CLIENT_BLACK_HOLE_AUDIO_CAPTURE_MODULE_H
#define OPENTERA_WEBRTC_NATIVE_CLIENT_BLACK_HOLE_AUDIO_CAPTURE_MODULE_H
 
#include <OpenteraWebrtcNativeClient/Utils/ClassMacro.h>
#include <OpenteraWebrtcNativeClient/Sinks/AudioSink.h>
 
#include <modules/audio_device/include/audio_device.h>
 
#include <atomic>
#include <memory>
#include <mutex>
#include <thread>
 
namespace opentera
{
    class OpenteraAudioDeviceModule : public webrtc::AudioDeviceModule
    {
        AudioSinkCallback m_onMixedAudioFrameReceived;
 
        bool m_isPlayoutInitialized;
        bool m_isRecordingInitialized;
        bool m_isSpeakerInitialized;
        bool m_isMicrophoneInitialized;
 
        bool m_isPlaying;
        bool m_isRecording;
 
        std::atomic_bool m_playoutThreadStopped;
        std::unique_ptr<std::thread> m_thread;
        webrtc::AudioTransport* m_audioTransport;
        std::mutex m_audioTransportCaptureMutex;
 
        std::mutex m_setCallbackMutex;
 
    public:
        OpenteraAudioDeviceModule();
        ~OpenteraAudioDeviceModule() override;
 
        DECLARE_NOT_COPYABLE(OpenteraAudioDeviceModule);
        DECLARE_NOT_MOVABLE(OpenteraAudioDeviceModule);
 
        void setOnMixedAudioFrameReceived(const AudioSinkCallback& onMixedAudioFrameReceived);
        void sendFrame(
            const void* audioData,
            int bitsPerSample,
            int sampleRate,
            size_t numberOfChannels,
            size_t numberOfFrames,
            uint32_t audioDelayMs,
            bool isTyping);
 
        // Retrieve the currently utilized audio layer
        int32_t ActiveAudioLayer(AudioLayer* audioLayer) const override;
 
        // Full-duplex transportation of PCM audio
        int32_t RegisterAudioCallback(webrtc::AudioTransport* audioTransport) override;
 
        // Main initialization and termination
        int32_t Init() override;
        int32_t Terminate() override;
        [[nodiscard]] bool Initialized() const override;
 
        // Device enumeration
        int16_t PlayoutDevices() override;
        int16_t RecordingDevices() override;
        int32_t PlayoutDeviceName(
            uint16_t index,
            char name[webrtc::kAdmMaxDeviceNameSize],
            char guid[webrtc::kAdmMaxGuidSize]) override;
        int32_t RecordingDeviceName(
            uint16_t index,
            char name[webrtc::kAdmMaxDeviceNameSize],
            char guid[webrtc::kAdmMaxGuidSize]) override;
 
        // Device selection
        int32_t SetPlayoutDevice(uint16_t index) override;
        int32_t SetPlayoutDevice(WindowsDeviceType device) override;
        int32_t SetRecordingDevice(uint16_t index) override;
        int32_t SetRecordingDevice(WindowsDeviceType device) override;
 
        // Audio transport initialization
        int32_t PlayoutIsAvailable(bool* available) override;
        int32_t InitPlayout() override;
        [[nodiscard]] bool PlayoutIsInitialized() const override;
        int32_t RecordingIsAvailable(bool* available) override;
        int32_t InitRecording() override;
        [[nodiscard]] bool RecordingIsInitialized() const override;
 
        // Audio transport control
        int32_t StartPlayout() override;
        int32_t StopPlayout() override;
        [[nodiscard]] bool Playing() const override;
        int32_t StartRecording() override;
        int32_t StopRecording() override;
        [[nodiscard]] bool Recording() const override;
 
        // Audio mixer initialization
        int32_t InitSpeaker() override;
        [[nodiscard]] bool SpeakerIsInitialized() const override;
        int32_t InitMicrophone() override;
        [[nodiscard]] bool MicrophoneIsInitialized() const override;
 
        // Speaker volume controls
        int32_t SpeakerVolumeIsAvailable(bool* available) override;
        int32_t SetSpeakerVolume(uint32_t volume) override;
        int32_t SpeakerVolume(uint32_t* volume) const override;
        int32_t MaxSpeakerVolume(uint32_t* maxVolume) const override;
        int32_t MinSpeakerVolume(uint32_t* minVolume) const override;
 
        // Microphone volume controls
        int32_t MicrophoneVolumeIsAvailable(bool* available) override;
        int32_t SetMicrophoneVolume(uint32_t volume) override;
        int32_t MicrophoneVolume(uint32_t* volume) const override;
        int32_t MaxMicrophoneVolume(uint32_t* maxVolume) const override;
        int32_t MinMicrophoneVolume(uint32_t* minVolume) const override;
 
        // Speaker mute control
        int32_t SpeakerMuteIsAvailable(bool* available) override;
        int32_t SetSpeakerMute(bool enable) override;
        int32_t SpeakerMute(bool* enabled) const override;
 
        // Microphone mute control
        int32_t MicrophoneMuteIsAvailable(bool* available) override;
        int32_t SetMicrophoneMute(bool enable) override;
        int32_t MicrophoneMute(bool* enabled) const override;
 
        // Stereo support
        int32_t StereoPlayoutIsAvailable(bool* available) const override;
        int32_t SetStereoPlayout(bool enable) override;
        int32_t StereoPlayout(bool* enabled) const override;
        int32_t StereoRecordingIsAvailable(bool* available) const override;
        int32_t SetStereoRecording(bool enable) override;
        int32_t StereoRecording(bool* enabled) const override;
 
        // Playout delay
        int32_t PlayoutDelay(uint16_t* delayMS) const override;
 
        // Only supported on Android.
        [[nodiscard]] bool BuiltInAECIsAvailable() const override;
        [[nodiscard]] bool BuiltInAGCIsAvailable() const override;
        [[nodiscard]] bool BuiltInNSIsAvailable() const override;
 
        // Enables the built-in audio effects. Only supported on Android.
        int32_t EnableBuiltInAEC(bool enable) override;
        int32_t EnableBuiltInAGC(bool enable) override;
        int32_t EnableBuiltInNS(bool enable) override;
 
    private:
        void stopPlayoutThreadIfStarted();
        void startPlayoutThreadIfStoppedAndTransportValid();
 
        void run();
    };
}
 
#endif