mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-14 14:20:45 +01:00
a09f21b207
This CL propagates capture_time_identifier introduced in webrtc::VideoFrame and propagates it to EncodedImage. For use cases involving EncodedTransforms, this identifier is further propagated to TransformableVideoSenderFrame. VideoEncoder::Encode function is overriden by each encoder. Each of these overriden functions needs to be changed so that they can handle this new identifier and propagate its value in the created EncodedImage. Change-Id: I5bea4c5a3fe714f1198e497a4bcb5fd059afe516 Bug: webrtc:14878 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/291800 Reviewed-by: Tony Herre <herre@google.com> Reviewed-by: Harald Alvestrand <hta@webrtc.org> Commit-Queue: Palak Agarwal <agpalak@google.com> Cr-Commit-Position: refs/heads/main@{#39374}
78 lines
2.7 KiB
C++
78 lines
2.7 KiB
C++
/*
|
|
* Copyright 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 "modules/video_coding/codecs/test/encoded_video_frame_producer.h"
|
|
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#include "api/test/create_frame_generator.h"
|
|
#include "api/test/frame_generator_interface.h"
|
|
#include "api/transport/rtp/dependency_descriptor.h"
|
|
#include "api/video/video_frame.h"
|
|
#include "api/video/video_frame_type.h"
|
|
#include "api/video_codecs/video_encoder.h"
|
|
#include "modules/video_coding/include/video_codec_interface.h"
|
|
#include "modules/video_coding/include/video_error_codes.h"
|
|
#include "rtc_base/checks.h"
|
|
|
|
namespace webrtc {
|
|
namespace {
|
|
|
|
class EncoderCallback : public EncodedImageCallback {
|
|
public:
|
|
explicit EncoderCallback(
|
|
std::vector<EncodedVideoFrameProducer::EncodedFrame>& output_frames)
|
|
: output_frames_(output_frames) {}
|
|
|
|
private:
|
|
Result OnEncodedImage(const EncodedImage& encoded_image,
|
|
const CodecSpecificInfo* codec_specific_info) override {
|
|
output_frames_.push_back({encoded_image, *codec_specific_info});
|
|
return Result(Result::Error::OK);
|
|
}
|
|
|
|
std::vector<EncodedVideoFrameProducer::EncodedFrame>& output_frames_;
|
|
};
|
|
|
|
} // namespace
|
|
|
|
std::vector<EncodedVideoFrameProducer::EncodedFrame>
|
|
EncodedVideoFrameProducer::Encode() {
|
|
std::unique_ptr<test::FrameGeneratorInterface> frame_buffer_generator =
|
|
test::CreateSquareFrameGenerator(
|
|
resolution_.Width(), resolution_.Height(),
|
|
test::FrameGeneratorInterface::OutputType::kI420, absl::nullopt);
|
|
|
|
std::vector<EncodedFrame> encoded_frames;
|
|
EncoderCallback encoder_callback(encoded_frames);
|
|
RTC_CHECK_EQ(encoder_.RegisterEncodeCompleteCallback(&encoder_callback),
|
|
WEBRTC_VIDEO_CODEC_OK);
|
|
|
|
uint32_t rtp_tick = 90000 / framerate_fps_;
|
|
for (int i = 0; i < num_input_frames_; ++i) {
|
|
VideoFrame frame =
|
|
VideoFrame::Builder()
|
|
.set_video_frame_buffer(frame_buffer_generator->NextFrame().buffer)
|
|
.set_timestamp_rtp(rtp_timestamp_)
|
|
.set_capture_time_identifier(capture_time_identifier_)
|
|
.build();
|
|
rtp_timestamp_ += rtp_tick;
|
|
RTC_CHECK_EQ(encoder_.Encode(frame, &next_frame_type_),
|
|
WEBRTC_VIDEO_CODEC_OK);
|
|
next_frame_type_[0] = VideoFrameType::kVideoFrameDelta;
|
|
}
|
|
|
|
RTC_CHECK_EQ(encoder_.RegisterEncodeCompleteCallback(nullptr),
|
|
WEBRTC_VIDEO_CODEC_OK);
|
|
return encoded_frames;
|
|
}
|
|
|
|
} // namespace webrtc
|