xref: /aosp_15_r20/external/webrtc/modules/video_coding/frame_helpers.cc (revision d9f758449e529ab9291ac668be2861e7a55c2422)
1*d9f75844SAndroid Build Coastguard Worker /*
2*d9f75844SAndroid Build Coastguard Worker  *  Copyright (c) 2021 The WebRTC project authors. All Rights Reserved.
3*d9f75844SAndroid Build Coastguard Worker  *
4*d9f75844SAndroid Build Coastguard Worker  *  Use of this source code is governed by a BSD-style license
5*d9f75844SAndroid Build Coastguard Worker  *  that can be found in the LICENSE file in the root of the source
6*d9f75844SAndroid Build Coastguard Worker  *  tree. An additional intellectual property rights grant can be found
7*d9f75844SAndroid Build Coastguard Worker  *  in the file PATENTS.  All contributing project authors may
8*d9f75844SAndroid Build Coastguard Worker  *  be found in the AUTHORS file in the root of the source tree.
9*d9f75844SAndroid Build Coastguard Worker  */
10*d9f75844SAndroid Build Coastguard Worker 
11*d9f75844SAndroid Build Coastguard Worker #include "modules/video_coding/frame_helpers.h"
12*d9f75844SAndroid Build Coastguard Worker 
13*d9f75844SAndroid Build Coastguard Worker #include <utility>
14*d9f75844SAndroid Build Coastguard Worker 
15*d9f75844SAndroid Build Coastguard Worker #include "rtc_base/logging.h"
16*d9f75844SAndroid Build Coastguard Worker 
17*d9f75844SAndroid Build Coastguard Worker namespace webrtc {
18*d9f75844SAndroid Build Coastguard Worker 
19*d9f75844SAndroid Build Coastguard Worker namespace {
20*d9f75844SAndroid Build Coastguard Worker constexpr TimeDelta kMaxVideoDelay = TimeDelta::Millis(10000);
21*d9f75844SAndroid Build Coastguard Worker }
22*d9f75844SAndroid Build Coastguard Worker 
FrameHasBadRenderTiming(Timestamp render_time,Timestamp now)23*d9f75844SAndroid Build Coastguard Worker bool FrameHasBadRenderTiming(Timestamp render_time, Timestamp now) {
24*d9f75844SAndroid Build Coastguard Worker   // Zero render time means render immediately.
25*d9f75844SAndroid Build Coastguard Worker   if (render_time.IsZero()) {
26*d9f75844SAndroid Build Coastguard Worker     return false;
27*d9f75844SAndroid Build Coastguard Worker   }
28*d9f75844SAndroid Build Coastguard Worker   if (render_time < Timestamp::Zero()) {
29*d9f75844SAndroid Build Coastguard Worker     return true;
30*d9f75844SAndroid Build Coastguard Worker   }
31*d9f75844SAndroid Build Coastguard Worker   TimeDelta frame_delay = render_time - now;
32*d9f75844SAndroid Build Coastguard Worker   if (frame_delay.Abs() > kMaxVideoDelay) {
33*d9f75844SAndroid Build Coastguard Worker     RTC_LOG(LS_WARNING) << "Frame has bad render timing because it is out of "
34*d9f75844SAndroid Build Coastguard Worker                            "the delay bounds (frame_delay_ms="
35*d9f75844SAndroid Build Coastguard Worker                         << frame_delay.ms()
36*d9f75844SAndroid Build Coastguard Worker                         << ", kMaxVideoDelay_ms=" << kMaxVideoDelay.ms() << ")";
37*d9f75844SAndroid Build Coastguard Worker     return true;
38*d9f75844SAndroid Build Coastguard Worker   }
39*d9f75844SAndroid Build Coastguard Worker   return false;
40*d9f75844SAndroid Build Coastguard Worker }
41*d9f75844SAndroid Build Coastguard Worker 
TargetVideoDelayIsTooLarge(TimeDelta target_video_delay)42*d9f75844SAndroid Build Coastguard Worker bool TargetVideoDelayIsTooLarge(TimeDelta target_video_delay) {
43*d9f75844SAndroid Build Coastguard Worker   if (target_video_delay > kMaxVideoDelay) {
44*d9f75844SAndroid Build Coastguard Worker     RTC_LOG(LS_WARNING)
45*d9f75844SAndroid Build Coastguard Worker         << "Target video delay is too large. (target_video_delay_ms="
46*d9f75844SAndroid Build Coastguard Worker         << target_video_delay.ms()
47*d9f75844SAndroid Build Coastguard Worker         << ", kMaxVideoDelay_ms=" << kMaxVideoDelay.ms() << ")";
48*d9f75844SAndroid Build Coastguard Worker     return true;
49*d9f75844SAndroid Build Coastguard Worker   }
50*d9f75844SAndroid Build Coastguard Worker   return false;
51*d9f75844SAndroid Build Coastguard Worker }
52*d9f75844SAndroid Build Coastguard Worker 
CombineAndDeleteFrames(absl::InlinedVector<std::unique_ptr<EncodedFrame>,4> frames)53*d9f75844SAndroid Build Coastguard Worker std::unique_ptr<EncodedFrame> CombineAndDeleteFrames(
54*d9f75844SAndroid Build Coastguard Worker     absl::InlinedVector<std::unique_ptr<EncodedFrame>, 4> frames) {
55*d9f75844SAndroid Build Coastguard Worker   RTC_DCHECK(!frames.empty());
56*d9f75844SAndroid Build Coastguard Worker 
57*d9f75844SAndroid Build Coastguard Worker   if (frames.size() == 1) {
58*d9f75844SAndroid Build Coastguard Worker     return std::move(frames[0]);
59*d9f75844SAndroid Build Coastguard Worker   }
60*d9f75844SAndroid Build Coastguard Worker 
61*d9f75844SAndroid Build Coastguard Worker   size_t total_length = 0;
62*d9f75844SAndroid Build Coastguard Worker   for (const auto& frame : frames) {
63*d9f75844SAndroid Build Coastguard Worker     total_length += frame->size();
64*d9f75844SAndroid Build Coastguard Worker   }
65*d9f75844SAndroid Build Coastguard Worker   const EncodedFrame& last_frame = *frames.back();
66*d9f75844SAndroid Build Coastguard Worker   std::unique_ptr<EncodedFrame> first_frame = std::move(frames[0]);
67*d9f75844SAndroid Build Coastguard Worker   auto encoded_image_buffer = EncodedImageBuffer::Create(total_length);
68*d9f75844SAndroid Build Coastguard Worker   uint8_t* buffer = encoded_image_buffer->data();
69*d9f75844SAndroid Build Coastguard Worker   first_frame->SetSpatialLayerFrameSize(first_frame->SpatialIndex().value_or(0),
70*d9f75844SAndroid Build Coastguard Worker                                         first_frame->size());
71*d9f75844SAndroid Build Coastguard Worker   memcpy(buffer, first_frame->data(), first_frame->size());
72*d9f75844SAndroid Build Coastguard Worker   buffer += first_frame->size();
73*d9f75844SAndroid Build Coastguard Worker 
74*d9f75844SAndroid Build Coastguard Worker   // Spatial index of combined frame is set equal to spatial index of its top
75*d9f75844SAndroid Build Coastguard Worker   // spatial layer.
76*d9f75844SAndroid Build Coastguard Worker   first_frame->SetSpatialIndex(last_frame.SpatialIndex().value_or(0));
77*d9f75844SAndroid Build Coastguard Worker 
78*d9f75844SAndroid Build Coastguard Worker   first_frame->video_timing_mutable()->network2_timestamp_ms =
79*d9f75844SAndroid Build Coastguard Worker       last_frame.video_timing().network2_timestamp_ms;
80*d9f75844SAndroid Build Coastguard Worker   first_frame->video_timing_mutable()->receive_finish_ms =
81*d9f75844SAndroid Build Coastguard Worker       last_frame.video_timing().receive_finish_ms;
82*d9f75844SAndroid Build Coastguard Worker 
83*d9f75844SAndroid Build Coastguard Worker   // Append all remaining frames to the first one.
84*d9f75844SAndroid Build Coastguard Worker   for (size_t i = 1; i < frames.size(); ++i) {
85*d9f75844SAndroid Build Coastguard Worker     // Let |next_frame| fall out of scope so it is deleted after copying.
86*d9f75844SAndroid Build Coastguard Worker     std::unique_ptr<EncodedFrame> next_frame = std::move(frames[i]);
87*d9f75844SAndroid Build Coastguard Worker     first_frame->SetSpatialLayerFrameSize(
88*d9f75844SAndroid Build Coastguard Worker         next_frame->SpatialIndex().value_or(0), next_frame->size());
89*d9f75844SAndroid Build Coastguard Worker     memcpy(buffer, next_frame->data(), next_frame->size());
90*d9f75844SAndroid Build Coastguard Worker     buffer += next_frame->size();
91*d9f75844SAndroid Build Coastguard Worker   }
92*d9f75844SAndroid Build Coastguard Worker   first_frame->SetEncodedData(encoded_image_buffer);
93*d9f75844SAndroid Build Coastguard Worker   return first_frame;
94*d9f75844SAndroid Build Coastguard Worker }
95*d9f75844SAndroid Build Coastguard Worker 
96*d9f75844SAndroid Build Coastguard Worker }  // namespace webrtc
97