Add psnr and sse to EncodeFrameResult

angiebird committed 7 years ago

Commit 56735c31 parent 6956e395 files · +48 −12Browse files

Unified · Split

5 files

test/simple_encode_test.cc +5 −2
vp9/encoder/vp9_encoder.c +37 −9
vp9/encoder/vp9_encoder.h +2 −0
vp9/simple_encode.cc +2 −0
vp9/simple_encode.h +2 −1

   int h = 288;
   int frame_rate_num = 30;
   int frame_rate_den = 1;
-  int target_bitrate = 200;
+  int target_bitrate = 1000;
   int num_frames = 17;
   // TODO(angiebird): Figure out how to upload test video to our codebase
   FILE *file = fopen("bus_352x288_420_f20_b8.yuv", "r");
     EXPECT_LT(encode_frame_result.show_idx, num_frames);
     if (i == num_coding_frames - 1) {
       EXPECT_EQ(encode_frame_result.show_idx, num_frames - 1)
-          << "The last coding frame should should be the last display order";
+          << "The last coding frame should be the last display order";
     }
+    EXPECT_GE(encode_frame_result.psnr, 34)
+        << "The psnr is supposed to be greater than 34 given the "
+           "target_bitrate 1000 kbps";
   }
   EXPECT_EQ(num_alternate_refereces, ref_num_alternate_refereces);
   simple_encode.EndEncode();

vp9/encoder/vp9_encoder.c+37−9

   mismatch_move_frame_idx_w();
 #endif
   encode_frame_to_data_rate(cpi, size, dest, frame_flags);
-  vp9_twopass_postencode_update(cpi);
 }
 #endif  // !CONFIG_REALTIME_ONLY
   encode_frame_result->show_idx = -1;  // Actual encoding deosn't happen.
 }
+#if !CONFIG_REALTIME_ONLY
 static void update_encode_frame_result(ENCODE_FRAME_RESULT *encode_frame_result,
                                        int show_idx,
-                                       FRAME_UPDATE_TYPE update_type) {
+                                       FRAME_UPDATE_TYPE update_type,
+                                       const YV12_BUFFER_CONFIG *source_frame,
+                                       const YV12_BUFFER_CONFIG *coded_frame,
+                                       uint32_t bit_depth,
+                                       uint32_t input_bit_depth) {
+  PSNR_STATS psnr;
+#if CONFIG_VP9_HIGHBITDEPTH
+  vpx_calc_highbd_psnr(source_frame, coded_frame, &psnr, bit_depth,
+                       input_bit_depth);
+#else
+  (void)bit_depth;
+  (void)input_bit_depth;
+  vpx_calc_psnr(source_frame, coded_frame, &psnr);
+#endif
+  encode_frame_result->psnr = psnr.psnr[0];
+  encode_frame_result->sse = psnr.sse[0];
   encode_frame_result->show_idx = show_idx;
   encode_frame_result->update_type = update_type;
 }
+#endif  // !CONFIG_REALTIME_ONLY
 int vp9_get_compressed_data(VP9_COMP *cpi, unsigned int *frame_flags,
                             size_t *size, uint8_t *dest, int64_t *time_stamp,
   bitstream_queue_set_frame_write(cm->current_video_frame * 2 + cm->show_frame);
 #endif
-  if (oxcf->pass != 1) {
-    update_encode_frame_result(
-        encode_frame_result, source->show_idx,
-        cpi->twopass.gf_group.update_type[cpi->twopass.gf_group.index]);
-  }
   cpi->td.mb.fp_src_pred = 0;
 #if CONFIG_REALTIME_ONLY
   if (cpi->use_svc) {
     vp9_first_pass(cpi, source);
   } else if (oxcf->pass == 2 && !cpi->use_svc) {
     Pass2Encode(cpi, size, dest, frame_flags);
+    // update_encode_frame_result() depends on twopass.gf_group.index and
+    // cm->new_fb_idx and cpi->Source are updated for current properly and have
+    // not been updated for the next frame yet.
+    // The update locations are as follows.
+    // 1) twopass.gf_group.index is initialized at define_gf_group by vp9_zero()
+    // for the first frame in the gf_group and is updated for the next frame at
+    // vp9_twopass_postencode_update().
+    // 2) cpi->Source is updated at the beginging of this function, i.e.
+    // vp9_get_compressed_data()
+    // 3) cm->new_fb_idx is updated at the beginging of this function by
+    // get_free_fb(cm)
+    // TODO(angiebird): Improve the codebase to make the update of frame
+    // dependent variables more robust.
+    update_encode_frame_result(
+        encode_frame_result, source->show_idx,
+        cpi->twopass.gf_group.update_type[cpi->twopass.gf_group.index],
+        cpi->Source, get_frame_new_buffer(cm), cpi->oxcf.input_bit_depth,
+        cm->bit_depth);
+    vp9_twopass_postencode_update(cpi);
   } else if (cpi->use_svc) {
     SvcEncode(cpi, size, dest, frame_flags);
   } else {

vp9/encoder/vp9_encoder.h+2−0

@@ -825,6 +825,8 @@ typedef struct VP9_COMP { · 824 unmodified lines
825	typedef struct ENCODE_FRAME_RESULT {	825	typedef struct ENCODE_FRAME_RESULT {
826	int show_idx;	826	int show_idx;
827	FRAME_UPDATE_TYPE update_type;	827	FRAME_UPDATE_TYPE update_type;
		828	double psnr;
		829	uint64_t sse;
828	} ENCODE_FRAME_RESULT;	830	} ENCODE_FRAME_RESULT;
829		831
830	void vp9_initialize_enc(void);	832	void vp9_initialize_enc(void);

vp9/simple_encode.cc+2−0

@@ -90,6 +90,8 @@ static void update_encode_frame_result( · 89 unmodified lines
90	encode_frame_result->show_idx = encode_frame_info->show_idx;	90	encode_frame_result->show_idx = encode_frame_info->show_idx;
91	encode_frame_result->frame_type =	91	encode_frame_result->frame_type =
92	get_frame_type_from_update_type(encode_frame_info->update_type);	92	get_frame_type_from_update_type(encode_frame_info->update_type);
		93	encode_frame_result->psnr = encode_frame_info->psnr;
		94	encode_frame_result->sse = encode_frame_info->sse;
93	}	95	}
94		96
95	SimpleEncode::SimpleEncode(int frame_width, int frame_height,	97	SimpleEncode::SimpleEncode(int frame_width, int frame_height,

vp9/simple_encode.h+2−1

@@ -14,7 +14,8 @@ struct EncodeFrameResult { · 13 unmodified lines
14	// The EncodeFrame will allocate a buffer, write the coding data into the	14	// The EncodeFrame will allocate a buffer, write the coding data into the
15	// buffer and give the ownership of the buffer to coding_data	15	// buffer and give the ownership of the buffer to coding_data
16	std::unique_ptr<unsigned char[]> coding_data;	16	std::unique_ptr<unsigned char[]> coding_data;
17	// TODO(angiebird): double psnr;	17	double psnr;
		18	uint64_t sse;
18	// TODO(angiebird): int quantize_index ;	19	// TODO(angiebird): int quantize_index ;
19	};	20	};
20		21

Add psnr and sse to EncodeFrameResult

Keyboard shortcuts

General

Issues and pull requests

Issues, pull requests and Inbox

Repository code