mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-14 22:30:40 +01:00
0fa90887c5
Instead, add rtp_timestamp and set_rtp_timestamp. Bug: webrtc:13756 Change-Id: Ic4266394003e0d49e525d71f4d830f5e518299cc Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/342781 Commit-Queue: Per Kjellander <perkj@webrtc.org> Reviewed-by: Magnus Jedvert <magjed@webrtc.org> Reviewed-by: Erik Språng <sprang@webrtc.org> Reviewed-by: Markus Handell <handellm@webrtc.org> Cr-Commit-Position: refs/heads/main@{#41894}
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_rtp_timestamp(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
|