mpeg2enc: f_code[s][t] follows the picture coding type

Signed-off-by: Xiang, Haihao <haihao.xiang@intel.com>

diff --git a/test/encode/mpeg2enc.c b/test/encode/mpeg2enc.c
index a393053..9fcedc1 100644 (file)
--- a/test/encode/mpeg2enc.c
+++ b/test/encode/mpeg2enc.c
@@ -960,6 +960,25 @@ mpeg2enc_update_picture_parameter(struct mpeg2enc_context *ctx,
     pic_param->reconstructed_picture = surface_ids[SID_RECON_PICTURE];
     pic_param->forward_reference_picture = surface_ids[SID_REFERENCE_PICTURE_L0];
     pic_param->backward_reference_picture = surface_ids[SID_REFERENCE_PICTURE_L1];
+
+    if (pic_param->picture_type == VAEncPictureTypeIntra) {
+        pic_param->f_code[0][0] = 0xf;
+        pic_param->f_code[0][1] = 0xf;
+        pic_param->f_code[1][0] = 0xf;
+        pic_param->f_code[1][1] = 0xf;
+    } else if (pic_param->picture_type == VAEncPictureTypePredictive) {
+        pic_param->f_code[0][0] = 0x1;
+        pic_param->f_code[0][1] = 0x1;
+        pic_param->f_code[1][0] = 0xf;
+        pic_param->f_code[1][1] = 0xf;
+    } else if (pic_param->picture_type == VAEncPictureTypeBidirectional) {
+        pic_param->f_code[0][0] = 0x1;
+        pic_param->f_code[0][1] = 0x1;
+        pic_param->f_code[1][0] = 0x1;
+        pic_param->f_code[1][1] = 0x1;
+    } else {
+        assert(0);
+    }
 }
 
 static void