audiostream.h

Go to the documentation of this file.

/*
 *  Copyright (C) 2004-2025 Savoir-faire Linux Inc.
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program. If not, see <https://www.gnu.org/licenses/>.
 */
#pragma once
 
#include "noncopyable.h"
#include "pulselayer.h"
 
#include <pulse/pulseaudio.h>
#include <string>
 
namespace jami {
 
inline AVSampleFormat
sampleFormatFromPulse(pa_sample_format_t format) {
switch (format)
{
case PA_SAMPLE_S16LE:
case PA_SAMPLE_S16BE:
    return AV_SAMPLE_FMT_S16;
case PA_SAMPLE_FLOAT32LE:
case PA_SAMPLE_FLOAT32BE:
    return AV_SAMPLE_FMT_FLT;
case PA_SAMPLE_S32LE:
case PA_SAMPLE_S32BE:
    return AV_SAMPLE_FMT_S32;
default:
    return AV_SAMPLE_FMT_S16;
}
}
 
inline pa_sample_format_t
pulseSampleFormatFromAv(AVSampleFormat format) {
switch (format)
{
case AV_SAMPLE_FMT_S16:
    return PA_SAMPLE_S16LE;
case AV_SAMPLE_FMT_FLT:
    return PA_SAMPLE_FLOAT32LE;
case AV_SAMPLE_FMT_S32:
    return PA_SAMPLE_S32LE;
default:
    return PA_SAMPLE_S16LE;
}
}
 
class AudioStream
{
public:
    using OnReady = std::function<void()>;
    using OnData = std::function<void(size_t)>;
 
    AudioStream(pa_context*,
                pa_threaded_mainloop*,
                const char*,
                AudioDeviceType,
                unsigned,
                pa_sample_format_t,
                const PaDeviceInfos&,
                bool,
                OnReady onReady,
                OnData onData);
 
    ~AudioStream();
 
    void start();
    void stop();
 
    pa_stream* stream() { return audiostream_; }
 
    const pa_sample_spec* sampleSpec() const { return pa_stream_get_sample_spec(audiostream_); }
 
    inline size_t sampleSize() const { return pa_sample_size(sampleSpec()); }
    inline size_t frameSize() const { return pa_frame_size(sampleSpec()); }
 
    inline uint8_t channels() const { return sampleSpec()->channels; }
 
    inline AudioFormat format() const
    {
        auto s = sampleSpec();
        return AudioFormat(s->rate, s->channels, sampleFormatFromPulse(s->format));
    }
 
    inline std::string getDeviceName() const
    {
        auto res = pa_stream_get_device_name(audiostream_);
        if (res == reinterpret_cast<decltype(res)>(-PA_ERR_NOTSUPPORTED) or !res)
            return {};
        return res;
    }
 
    bool isReady();
 
    void setEchoCancelCb(std::function<void(bool)>&& cb) { echoCancelCb = cb; }
 
private:
    NON_COPYABLE(AudioStream);
 
    OnReady onReady_;
    OnData onData_;
 
    void stateChanged(pa_stream* s);
    void moved(pa_stream* s);
    void opEnded(pa_operation* s);
 
    pa_stream* audiostream_;
 
    pa_threaded_mainloop* mainloop_;
 
    AudioDeviceType audioType_;
 
    std::function<void(bool)> echoCancelCb;
 
    std::mutex mutex_;
    std::condition_variable cond_;
    std::set<pa_operation*> ongoing_ops;
};
 
} // namespace jami