Add GetNextEncodeFrameInfo ObserveGroupOfPicture

GetNextEncodeFrameInfo()
Gets encode_frame_info for the next coding frame.

ObserveGroupOfPicture()
Provides the group of pictures that the next coding frame is in.

Change-Id: Idbc437d32c392f25b06efb2d4e1ec01347d678f2

1 files changed, 20 insertions, 0 deletions

diff --git a/vp9/simple_encode.h b/vp9/simple_encode.h
index 471b4e7a8..b8085298b 100644
--- a/vp9/simple_encode.h
+++ b/vp9/simple_encode.h
@@ -25,6 +25,11 @@ enum FrameType {
   kAlternateReference,
 };
 
+struct EncodeFrameInfo {
+  int show_idx;
+  FrameType frame_type;
+};
+
 struct EncodeFrameResult {
   int show_idx;
   FrameType frame_type;
@@ -38,6 +43,11 @@ struct EncodeFrameResult {
   int quantize_index;
 };
 
+struct GroupOfPicture {
+  std::vector<EncodeFrameInfo> encode_frame_list;
+  int encode_frame_index;
+};
+
 class SimpleEncode {
  public:
   SimpleEncode(int frame_width, int frame_height, int frame_rate_num,
@@ -72,6 +82,14 @@ class SimpleEncode {
   // counted.
   int GetKeyFrameGroupSize(int key_frame_index) const;
 
+  // Provides the group of pictures that the next coding frame is in.
+  // Only call this function between StartEncode() and EndEncode()
+  GroupOfPicture ObserveGroupOfPicture() const;
+
+  // Gets encode_frame_info for the next coding frame.
+  // Only call this function between StartEncode() and EndEncode()
+  EncodeFrameInfo GetNextEncodeFrameInfo() const;
+
   // Encodes a frame
   // This function should be called after StartEncode() and before EndEncode().
   void EncodeFrame(EncodeFrameResult *encode_frame_result);
@@ -97,6 +115,8 @@ class SimpleEncode {
   int num_frames_;
   std::FILE *file_;
   std::unique_ptr<EncodeImpl> impl_ptr_;
+
+  GroupOfPicture group_of_picture_;
 };
 
 }  // namespace vp9