1 files changed, 82 insertions, 0 deletions
diff --git a/third_party/libwebrtc/video/frame_encode_metadata_writer.h b/third_party/libwebrtc/video/frame_encode_metadata_writer.h
new file mode 100644
index 0000000000..afebca816c
--- /dev/null
+++ b/third_party/libwebrtc/video/frame_encode_metadata_writer.h
@@ -0,0 +1,82 @@
+/*
+ *  Copyright (c) 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 VIDEO_FRAME_ENCODE_METADATA_WRITER_H_
+#define VIDEO_FRAME_ENCODE_METADATA_WRITER_H_
+
+#include <list>
+#include <vector>
+
+#include "absl/types/optional.h"
+#include "api/video/encoded_image.h"
+#include "api/video_codecs/video_codec.h"
+#include "api/video_codecs/video_encoder.h"
+#include "modules/video_coding/include/video_codec_interface.h"
+#include "rtc_base/synchronization/mutex.h"
+
+namespace webrtc {
+
+class FrameEncodeMetadataWriter {
+ public:
+  explicit FrameEncodeMetadataWriter(EncodedImageCallback* frame_drop_callback);
+  ~FrameEncodeMetadataWriter();
+
+  void OnEncoderInit(const VideoCodec& codec);
+  void OnSetRates(const VideoBitrateAllocation& bitrate_allocation,
+                  uint32_t framerate_fps);
+
+  void OnEncodeStarted(const VideoFrame& frame);
+
+  void FillTimingInfo(size_t simulcast_svc_idx, EncodedImage* encoded_image);
+
+  void UpdateBitstream(const CodecSpecificInfo* codec_specific_info,
+                       EncodedImage* encoded_image);
+
+  void Reset();
+
+ private:
+  // For non-internal-source encoders, returns encode started time and fixes
+  // capture timestamp for the frame, if corrupted by the encoder.
+  absl::optional<int64_t> ExtractEncodeStartTimeAndFillMetadata(
+      size_t simulcast_svc_idx,
+      EncodedImage* encoded_image) RTC_EXCLUSIVE_LOCKS_REQUIRED(lock_);
+
+  struct FrameMetadata {
+    uint32_t rtp_timestamp;
+    int64_t encode_start_time_ms;
+    int64_t ntp_time_ms = 0;
+    int64_t timestamp_us = 0;
+    VideoRotation rotation = kVideoRotation_0;
+    absl::optional<ColorSpace> color_space;
+    RtpPacketInfos packet_infos;
+  };
+  struct TimingFramesLayerInfo {
+    TimingFramesLayerInfo();
+    ~TimingFramesLayerInfo();
+    size_t target_bitrate_bytes_per_sec = 0;
+    std::list<FrameMetadata> frames;
+  };
+
+  Mutex lock_;
+  EncodedImageCallback* const frame_drop_callback_;
+  VideoCodec codec_settings_ RTC_GUARDED_BY(&lock_);
+  uint32_t framerate_fps_ RTC_GUARDED_BY(&lock_);
+
+  size_t num_spatial_layers_ RTC_GUARDED_BY(&lock_);
+  // Separate instance for each simulcast stream or spatial layer.
+  std::vector<TimingFramesLayerInfo> timing_frames_info_ RTC_GUARDED_BY(&lock_);
+  int64_t last_timing_frame_time_ms_ RTC_GUARDED_BY(&lock_);
+  size_t reordered_frames_logged_messages_ RTC_GUARDED_BY(&lock_);
+  size_t stalled_encoder_logged_messages_ RTC_GUARDED_BY(&lock_);
+};
+
+}  // namespace webrtc
+
+#endif  // VIDEO_FRAME_ENCODE_METADATA_WRITER_H_