Add interface for external arf indexes.

Pass in external arf indexes to encode command.

Cheng Chen committed 6 years ago

Commit 7e665ec1 parent 04848493 files · +22 −0Browse files

Unified · Split

3 files

 typedef struct ENCODE_COMMAND {
   int use_external_quantize_index;
   int external_quantize_index;
+  int use_external_arf;
+  // A list of binary flags set from the external controller.
+  // Each binary flag indicates whether the frame is an arf or not.
+  const int *external_arf_indexes;
 } ENCODE_COMMAND;
 typedef struct PARTITION_INFO {

   }
 }
+// Gets group of picture information from VP9's decision, and update
+// |group_of_picture| accordingly.
+// This is called at the starting of encoding of each group of picture.
 static void UpdateGroupOfPicture(const VP9_COMP *cpi, int start_coding_index,
                                  GroupOfPicture *group_of_picture) {
   int first_is_key_frame;
   num_frames_ = num_frames;
   frame_coding_index_ = 0;
   // TODO(angirbid): Should we keep a file pointer here or keep the file_path?
+  assert(infile_path != nullptr);
   in_file_ = fopen(infile_path, "r");
   if (outfile_path != nullptr) {
     out_file_ = fopen(outfile_path, "w");
   return output_stats;
 }
+void SimpleEncode::SetExternalGroupOfPicture(const bool use_external_arf,
+                                             const int *external_arf_indexes) {
+  impl_ptr_->cpi->encode_command.use_external_arf = use_external_arf;
+  impl_ptr_->cpi->encode_command.external_arf_indexes = external_arf_indexes;
+}
 void SimpleEncode::StartEncode() {
   assert(impl_ptr_->first_pass_stats.size() > 0);
   vpx_rational_t frame_rate =

@@ -272,6 +272,14 @@ class SimpleEncode { · 271 unmodified lines
272	// values. For details, please check FIRSTPASS_STATS in vp9_firstpass.h	272	// values. For details, please check FIRSTPASS_STATS in vp9_firstpass.h
273	std::vector<std::vector<double>> ObserveFirstPassStats();	273	std::vector<std::vector<double>> ObserveFirstPassStats();
274		274
		275	// Sets arf indexes for the video from external input.
		276	// The arf index determines whether a frame is arf or not.
		277	// Therefore it also determines the group of picture size.
		278	// If set, VP9 will use the external arf index to make decision.
		279	// This function is called only once before StartEncde().
		280	void SetExternalGroupOfPicture(bool use_external_arf,
		281	const int *external_arf_indexes);
		282
275	// Initializes the encoder for actual encoding.	283	// Initializes the encoder for actual encoding.
276	// This function should be called after ComputeFirstPassStats().	284	// This function should be called after ComputeFirstPassStats().
277	void StartEncode();	285	void StartEncode();

Keyboard shortcuts