mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-13 05:40:42 +01:00
95eeaa7aca
Bug: webrtc:14245 Change-Id: Ibd98d3a0c548443578953ef3e25aee9919eea3d3 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/267980 Commit-Queue: Danil Chapovalov <danilchap@webrtc.org> Reviewed-by: Erik Språng <sprang@webrtc.org> Cr-Commit-Position: refs/heads/main@{#37465}
124 lines
4.2 KiB
C++
124 lines
4.2 KiB
C++
/*
|
|
* Copyright (c) 2020 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.
|
|
*/
|
|
|
|
#include "video/rtp_video_stream_receiver_frame_transformer_delegate.h"
|
|
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#include "absl/memory/memory.h"
|
|
#include "modules/rtp_rtcp/source/rtp_descriptor_authentication.h"
|
|
#include "rtc_base/thread.h"
|
|
|
|
namespace webrtc {
|
|
|
|
namespace {
|
|
class TransformableVideoReceiverFrame
|
|
: public TransformableVideoFrameInterface {
|
|
public:
|
|
TransformableVideoReceiverFrame(std::unique_ptr<RtpFrameObject> frame,
|
|
uint32_t ssrc)
|
|
: frame_(std::move(frame)),
|
|
metadata_(frame_->GetRtpVideoHeader()),
|
|
ssrc_(ssrc) {}
|
|
~TransformableVideoReceiverFrame() override = default;
|
|
|
|
// Implements TransformableVideoFrameInterface.
|
|
rtc::ArrayView<const uint8_t> GetData() const override {
|
|
return *frame_->GetEncodedData();
|
|
}
|
|
|
|
void SetData(rtc::ArrayView<const uint8_t> data) override {
|
|
frame_->SetEncodedData(
|
|
EncodedImageBuffer::Create(data.data(), data.size()));
|
|
}
|
|
|
|
uint8_t GetPayloadType() const override { return frame_->PayloadType(); }
|
|
uint32_t GetSsrc() const override { return ssrc_; }
|
|
uint32_t GetTimestamp() const override { return frame_->Timestamp(); }
|
|
|
|
bool IsKeyFrame() const override {
|
|
return frame_->FrameType() == VideoFrameType::kVideoFrameKey;
|
|
}
|
|
|
|
std::vector<uint8_t> GetAdditionalData() const override {
|
|
return RtpDescriptorAuthentication(frame_->GetRtpVideoHeader());
|
|
}
|
|
|
|
const VideoFrameMetadata& GetMetadata() const override { return metadata_; }
|
|
|
|
std::unique_ptr<RtpFrameObject> ExtractFrame() && {
|
|
return std::move(frame_);
|
|
}
|
|
|
|
Direction GetDirection() const override { return Direction::kReceiver; }
|
|
|
|
private:
|
|
std::unique_ptr<RtpFrameObject> frame_;
|
|
const VideoFrameMetadata metadata_;
|
|
const uint32_t ssrc_;
|
|
};
|
|
} // namespace
|
|
|
|
RtpVideoStreamReceiverFrameTransformerDelegate::
|
|
RtpVideoStreamReceiverFrameTransformerDelegate(
|
|
RtpVideoFrameReceiver* receiver,
|
|
rtc::scoped_refptr<FrameTransformerInterface> frame_transformer,
|
|
rtc::Thread* network_thread,
|
|
uint32_t ssrc)
|
|
: receiver_(receiver),
|
|
frame_transformer_(std::move(frame_transformer)),
|
|
network_thread_(network_thread),
|
|
ssrc_(ssrc) {}
|
|
|
|
void RtpVideoStreamReceiverFrameTransformerDelegate::Init() {
|
|
RTC_DCHECK_RUN_ON(&network_sequence_checker_);
|
|
frame_transformer_->RegisterTransformedFrameSinkCallback(
|
|
rtc::scoped_refptr<TransformedFrameCallback>(this), ssrc_);
|
|
}
|
|
|
|
void RtpVideoStreamReceiverFrameTransformerDelegate::Reset() {
|
|
RTC_DCHECK_RUN_ON(&network_sequence_checker_);
|
|
frame_transformer_->UnregisterTransformedFrameSinkCallback(ssrc_);
|
|
frame_transformer_ = nullptr;
|
|
receiver_ = nullptr;
|
|
}
|
|
|
|
void RtpVideoStreamReceiverFrameTransformerDelegate::TransformFrame(
|
|
std::unique_ptr<RtpFrameObject> frame) {
|
|
RTC_DCHECK_RUN_ON(&network_sequence_checker_);
|
|
frame_transformer_->Transform(
|
|
std::make_unique<TransformableVideoReceiverFrame>(std::move(frame),
|
|
ssrc_));
|
|
}
|
|
|
|
void RtpVideoStreamReceiverFrameTransformerDelegate::OnTransformedFrame(
|
|
std::unique_ptr<TransformableFrameInterface> frame) {
|
|
rtc::scoped_refptr<RtpVideoStreamReceiverFrameTransformerDelegate> delegate(
|
|
this);
|
|
network_thread_->PostTask(
|
|
[delegate = std::move(delegate), frame = std::move(frame)]() mutable {
|
|
delegate->ManageFrame(std::move(frame));
|
|
});
|
|
}
|
|
|
|
void RtpVideoStreamReceiverFrameTransformerDelegate::ManageFrame(
|
|
std::unique_ptr<TransformableFrameInterface> frame) {
|
|
RTC_DCHECK_RUN_ON(&network_sequence_checker_);
|
|
RTC_CHECK_EQ(frame->GetDirection(),
|
|
TransformableFrameInterface::Direction::kReceiver);
|
|
if (!receiver_)
|
|
return;
|
|
auto transformed_frame = absl::WrapUnique(
|
|
static_cast<TransformableVideoReceiverFrame*>(frame.release()));
|
|
receiver_->ManageFrame(std::move(*transformed_frame).ExtractFrame());
|
|
}
|
|
|
|
} // namespace webrtc
|