media_recorder.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
 
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include "media_buffer.h"
#include "media_encoder.h"
#include "media_filter.h"
#include "media_stream.h"
#include "noncopyable.h"
#include "observer.h"
 
#include <map>
#include <memory>
#include <mutex>
#include <queue>
#include <stdexcept>
#include <string>
#include <utility>
#include <condition_variable>
#include <atomic>
 
namespace jami {
 
class MediaRecorder : public std::enable_shared_from_this<MediaRecorder>
{
public:
    MediaRecorder();
    ~MediaRecorder();
 
    bool isRecording() const;
 
    std::string getPath() const;
 
    void audioOnly(bool audioOnly);
 
    void setPath(const std::string& path);
 
    void setMetadata(const std::string& title, const std::string& desc);
 
    Observer<std::shared_ptr<MediaFrame>>* addStream(const MediaStream& ms);
 
    void removeStream(const MediaStream& ms);
 
    Observer<std::shared_ptr<MediaFrame>>* getStream(const std::string& name) const;
 
    int startRecording();
 
    void stopRecording();
 
private:
    NON_COPYABLE(MediaRecorder);
 
    struct StreamObserver;
 
    void onFrame(const std::string& name, const std::shared_ptr<MediaFrame>& frame);
 
    void flush();
    void reset();
 
    int initRecord();
    void setupVideoOutput();
    std::string buildVideoFilter(const std::vector<MediaStream>& peers,
                                 const MediaStream& local) const;
    void setupAudioOutput();
    std::mutex mutexStreamSetup_;
    std::string buildAudioFilter(const std::vector<MediaStream>& peers) const;
 
    std::mutex mutexFrameBuff_;
    std::mutex mutexFilterVideo_;
    std::mutex mutexFilterAudio_;
 
    std::map<std::string, std::unique_ptr<StreamObserver>> streams_;
 
    std::string path_;
    std::tm startTime_;
    int64_t startTimeStamp_;
    std::string title_;
    std::string description_;
 
    std::unique_ptr<MediaEncoder> encoder_;
    std::mutex encoderMtx_;
    std::unique_ptr<MediaFilter> outputVideoFilter_;
    std::unique_ptr<MediaFilter> outputAudioFilter_;
 
    std::unique_ptr<MediaFilter> videoFilter_;
    std::unique_ptr<MediaFilter> audioFilter_;
 
    int videoIdx_ = -1;
    int audioIdx_ = -1;
    bool isRecording_ = false;
    bool audioOnly_ = false;
    int lastVideoPts_ = 0;
 
    std::condition_variable cv_;
    std::atomic_bool interrupted_ {false};
 
    std::list<std::shared_ptr<MediaFrame>> frameBuff_;
};
 
}; // namespace jami