Add SimpleEncode::GetCodingFrameNum()

Also add unit tests for GetCodingFrameNum() and EncodeFrame()

Change-Id: I3e7b65f47226be4660409481435f8f784db72a68

1 files changed, 8 insertions, 0 deletions

diff --git a/vp9/simple_encode.h b/vp9/simple_encode.h
index 6308ff5ea..bc8b70bab 100644
--- a/vp9/simple_encode.h
+++ b/vp9/simple_encode.h
@@ -17,14 +17,22 @@ class SimpleEncode {
   std::vector<std::vector<double>> ObserveFirstPassStats();
 
   // Initialize the encoder for actual encoding
+  // This funtion should be called after ComputeFirstPassStats()
   void StartEncode();
 
   // Free the encoder
+  // This funtion should be called after StartEncode() or EncodeFrame()
   void EndEncode();
 
   // Encode a frame
+  // This funtion should be called after StartEncode() before EndEncode()
   void EncodeFrame(char *cx_data, size_t *size, size_t max_size);
 
+  // Get the number of coding frames for the video. The coding frames include
+  // show frame and no show frame.
+  // This funtion should be called after ComputeFirstPassStats()
+  int GetCodingFrameNum();
+
  private:
   class impl;
   int frame_width;