1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177
|
/*
* Copyright 2019 The WebRTC project authors. All Rights Reserved.
*
* Use of this source code is governed by a BSD-style license
* that can be found in the LICENSE file in the root of the source
* tree. An additional intellectual property rights grant can be found
* in the file PATENTS. All contributing project authors may
* be found in the AUTHORS file in the root of the source tree.
*/
#ifndef PC_VIDEO_RTP_RECEIVER_H_
#define PC_VIDEO_RTP_RECEIVER_H_
#include <stdint.h>
#include <optional>
#include <string>
#include <vector>
#include "api/crypto/frame_decryptor_interface.h"
#include "api/dtls_transport_interface.h"
#include "api/frame_transformer_interface.h"
#include "api/media_stream_interface.h"
#include "api/media_types.h"
#include "api/rtp_parameters.h"
#include "api/rtp_receiver_interface.h"
#include "api/scoped_refptr.h"
#include "api/sequence_checker.h"
#include "api/transport/rtp/rtp_source.h"
#include "api/video/video_frame.h"
#include "api/video/video_sink_interface.h"
#include "media/base/media_channel.h"
#include "pc/jitter_buffer_delay.h"
#include "pc/media_stream_track_proxy.h"
#include "pc/rtp_receiver.h"
#include "pc/video_rtp_track_source.h"
#include "pc/video_track.h"
#include "rtc_base/system/no_unique_address.h"
#include "rtc_base/thread.h"
#include "rtc_base/thread_annotations.h"
namespace webrtc {
class VideoRtpReceiver : public RtpReceiverInternal {
public:
// An SSRC of 0 will create a receiver that will match the first SSRC it
// sees. Must be called on signaling thread.
VideoRtpReceiver(Thread* worker_thread,
std::string receiver_id,
std::vector<std::string> streams_ids);
// TODO(hbos): Remove this when streams() is removed.
// https://crbug.com/webrtc/9480
VideoRtpReceiver(
Thread* worker_thread,
const std::string& receiver_id,
const std::vector<scoped_refptr<MediaStreamInterface>>& streams);
virtual ~VideoRtpReceiver();
scoped_refptr<VideoTrackInterface> video_track() const { return track_; }
// RtpReceiverInterface implementation
scoped_refptr<MediaStreamTrackInterface> track() const override {
return track_;
}
scoped_refptr<DtlsTransportInterface> dtls_transport() const override;
std::vector<std::string> stream_ids() const override;
std::vector<scoped_refptr<MediaStreamInterface>> streams() const override;
MediaType media_type() const override { return MediaType::VIDEO; }
std::string id() const override { return id_; }
RtpParameters GetParameters() const override;
void SetFrameDecryptor(
scoped_refptr<FrameDecryptorInterface> frame_decryptor) override;
scoped_refptr<FrameDecryptorInterface> GetFrameDecryptor() const override;
void SetFrameTransformer(
scoped_refptr<FrameTransformerInterface> frame_transformer) override;
// RtpReceiverInternal implementation.
void Stop() override;
void SetupMediaChannel(uint32_t ssrc) override;
void SetupUnsignaledMediaChannel() override;
std::optional<uint32_t> ssrc() const override;
void NotifyFirstPacketReceived() override;
void set_stream_ids(std::vector<std::string> stream_ids) override;
void set_transport(
scoped_refptr<DtlsTransportInterface> dtls_transport) override;
void SetStreams(
const std::vector<scoped_refptr<MediaStreamInterface>>& streams) override;
void SetObserver(RtpReceiverObserverInterface* observer) override;
void SetJitterBufferMinimumDelay(
std::optional<double> delay_seconds) override;
void SetMediaChannel(MediaReceiveChannelInterface* media_channel) override;
int AttachmentId() const override { return attachment_id_; }
std::vector<RtpSource> GetSources() const override;
// Combines SetMediaChannel, SetupMediaChannel and
// SetupUnsignaledMediaChannel.
void SetupMediaChannel(std::optional<uint32_t> ssrc,
MediaReceiveChannelInterface* media_channel);
private:
void RestartMediaChannel(std::optional<uint32_t> ssrc)
RTC_RUN_ON(&signaling_thread_checker_);
void RestartMediaChannel_w(std::optional<uint32_t> ssrc,
MediaSourceInterface::SourceState state)
RTC_RUN_ON(worker_thread_);
void SetSink(VideoSinkInterface<VideoFrame>* sink) RTC_RUN_ON(worker_thread_);
void SetMediaChannel_w(MediaReceiveChannelInterface* media_channel)
RTC_RUN_ON(worker_thread_);
// VideoRtpTrackSource::Callback
void OnGenerateKeyFrame();
void OnEncodedSinkEnabled(bool enable);
void SetEncodedSinkEnabled(bool enable) RTC_RUN_ON(worker_thread_);
class SourceCallback : public VideoRtpTrackSource::Callback {
public:
explicit SourceCallback(VideoRtpReceiver* receiver) : receiver_(receiver) {}
~SourceCallback() override = default;
private:
void OnGenerateKeyFrame() override { receiver_->OnGenerateKeyFrame(); }
void OnEncodedSinkEnabled(bool enable) override {
receiver_->OnEncodedSinkEnabled(enable);
}
VideoRtpReceiver* const receiver_;
} source_callback_{this};
RTC_NO_UNIQUE_ADDRESS SequenceChecker signaling_thread_checker_;
Thread* const worker_thread_;
const std::string id_;
VideoMediaReceiveChannelInterface* media_channel_
RTC_GUARDED_BY(worker_thread_) = nullptr;
std::optional<uint32_t> signaled_ssrc_ RTC_GUARDED_BY(worker_thread_);
// `source_` is held here to be able to change the state of the source when
// the VideoRtpReceiver is stopped.
const scoped_refptr<VideoRtpTrackSource> source_;
const scoped_refptr<VideoTrackProxyWithInternal<VideoTrack>> track_;
std::vector<scoped_refptr<MediaStreamInterface>> streams_
RTC_GUARDED_BY(&signaling_thread_checker_);
RtpReceiverObserverInterface* observer_
RTC_GUARDED_BY(&signaling_thread_checker_) = nullptr;
bool received_first_packet_ RTC_GUARDED_BY(&signaling_thread_checker_) =
false;
const int attachment_id_;
scoped_refptr<FrameDecryptorInterface> frame_decryptor_
RTC_GUARDED_BY(worker_thread_);
scoped_refptr<DtlsTransportInterface> dtls_transport_
RTC_GUARDED_BY(&signaling_thread_checker_);
scoped_refptr<FrameTransformerInterface> frame_transformer_
RTC_GUARDED_BY(worker_thread_);
// Stores the minimum jitter buffer delay. Handles caching cases
// if `SetJitterBufferMinimumDelay` is called before start.
JitterBufferDelay delay_ RTC_GUARDED_BY(worker_thread_);
// Records if we should generate a keyframe when `media_channel_` gets set up
// or switched.
bool saved_generate_keyframe_ RTC_GUARDED_BY(worker_thread_) = false;
bool saved_encoded_sink_enabled_ RTC_GUARDED_BY(worker_thread_) = false;
};
} // namespace webrtc
#endif // PC_VIDEO_RTP_RECEIVER_H_
|