OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "content/common/gpu/media/vaapi_video_decode_accelerator.h" | 5 #include "content/common/gpu/media/vaapi_video_decode_accelerator.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/logging.h" | 8 #include "base/logging.h" |
9 #include "base/metrics/histogram.h" | 9 #include "base/metrics/histogram.h" |
10 #include "base/stl_util.h" | 10 #include "base/stl_util.h" |
11 #include "base/strings/string_util.h" | 11 #include "base/strings/string_util.h" |
12 #include "base/synchronization/waitable_event.h" | 12 #include "base/synchronization/waitable_event.h" |
13 #include "base/trace_event/trace_event.h" | 13 #include "base/trace_event/trace_event.h" |
14 #include "content/common/gpu/gpu_channel.h" | 14 #include "content/common/gpu/gpu_channel.h" |
15 #include "content/common/gpu/media/accelerated_video_decoder.h" | 15 #include "content/common/gpu/media/accelerated_video_decoder.h" |
16 #include "content/common/gpu/media/h264_decoder.h" | 16 #include "content/common/gpu/media/h264_decoder.h" |
17 #include "content/common/gpu/media/vaapi_picture.h" | 17 #include "content/common/gpu/media/vaapi_picture.h" |
18 #include "content/common/gpu/media/vp8_decoder.h" | 18 #include "content/common/gpu/media/vp8_decoder.h" |
| 19 #include "content/common/gpu/media/vp9_decoder.h" |
19 #include "media/base/bind_to_current_loop.h" | 20 #include "media/base/bind_to_current_loop.h" |
20 #include "media/video/picture.h" | 21 #include "media/video/picture.h" |
21 #include "third_party/libva/va/va_dec_vp8.h" | 22 #include "third_party/libva/va/va_dec_vp8.h" |
22 #include "ui/gl/gl_bindings.h" | 23 #include "ui/gl/gl_bindings.h" |
23 #include "ui/gl/gl_image.h" | 24 #include "ui/gl/gl_image.h" |
24 | 25 |
25 namespace content { | 26 namespace content { |
26 | 27 |
27 namespace { | 28 namespace { |
28 // UMA errors that the VaapiVideoDecodeAccelerator class reports. | 29 // UMA errors that the VaapiVideoDecodeAccelerator class reports. |
(...skipping 162 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
191 private: | 192 private: |
192 scoped_refptr<VaapiDecodeSurface> VP8PictureToVaapiDecodeSurface( | 193 scoped_refptr<VaapiDecodeSurface> VP8PictureToVaapiDecodeSurface( |
193 const scoped_refptr<VP8Picture>& pic); | 194 const scoped_refptr<VP8Picture>& pic); |
194 | 195 |
195 VaapiWrapper* vaapi_wrapper_; | 196 VaapiWrapper* vaapi_wrapper_; |
196 VaapiVideoDecodeAccelerator* vaapi_dec_; | 197 VaapiVideoDecodeAccelerator* vaapi_dec_; |
197 | 198 |
198 DISALLOW_COPY_AND_ASSIGN(VaapiVP8Accelerator); | 199 DISALLOW_COPY_AND_ASSIGN(VaapiVP8Accelerator); |
199 }; | 200 }; |
200 | 201 |
| 202 class VaapiVP9Picture : public VP9Picture { |
| 203 public: |
| 204 VaapiVP9Picture( |
| 205 const scoped_refptr<VaapiVideoDecodeAccelerator::VaapiDecodeSurface>& |
| 206 dec_surface); |
| 207 |
| 208 VaapiVP9Picture* AsVaapiVP9Picture() override { return this; } |
| 209 scoped_refptr<VaapiVideoDecodeAccelerator::VaapiDecodeSurface> dec_surface() { |
| 210 return dec_surface_; |
| 211 } |
| 212 |
| 213 private: |
| 214 ~VaapiVP9Picture() override; |
| 215 |
| 216 scoped_refptr<VaapiVideoDecodeAccelerator::VaapiDecodeSurface> dec_surface_; |
| 217 |
| 218 DISALLOW_COPY_AND_ASSIGN(VaapiVP9Picture); |
| 219 }; |
| 220 |
| 221 VaapiVP9Picture::VaapiVP9Picture( |
| 222 const scoped_refptr<VaapiVideoDecodeAccelerator::VaapiDecodeSurface>& |
| 223 dec_surface) |
| 224 : dec_surface_(dec_surface) {} |
| 225 |
| 226 VaapiVP9Picture::~VaapiVP9Picture() {} |
| 227 |
| 228 class VaapiVideoDecodeAccelerator::VaapiVP9Accelerator |
| 229 : public VP9Decoder::VP9Accelerator { |
| 230 public: |
| 231 VaapiVP9Accelerator(VaapiVideoDecodeAccelerator* vaapi_dec, |
| 232 VaapiWrapper* vaapi_wrapper); |
| 233 ~VaapiVP9Accelerator() override; |
| 234 |
| 235 // VP9Decoder::VP9Accelerator implementation. |
| 236 scoped_refptr<VP9Picture> CreateVP9Picture() override; |
| 237 |
| 238 bool SubmitDecode( |
| 239 const scoped_refptr<VP9Picture>& pic, |
| 240 const media::Vp9Segmentation& seg, |
| 241 const media::Vp9LoopFilter& lf, |
| 242 const std::vector<scoped_refptr<VP9Picture>>& ref_pictures) override; |
| 243 |
| 244 bool OutputPicture(const scoped_refptr<VP9Picture>& pic) override; |
| 245 |
| 246 private: |
| 247 scoped_refptr<VaapiDecodeSurface> VP9PictureToVaapiDecodeSurface( |
| 248 const scoped_refptr<VP9Picture>& pic); |
| 249 |
| 250 VaapiWrapper* vaapi_wrapper_; |
| 251 VaapiVideoDecodeAccelerator* vaapi_dec_; |
| 252 |
| 253 DISALLOW_COPY_AND_ASSIGN(VaapiVP9Accelerator); |
| 254 }; |
| 255 |
201 VaapiVideoDecodeAccelerator::InputBuffer::InputBuffer() : id(0), size(0) { | 256 VaapiVideoDecodeAccelerator::InputBuffer::InputBuffer() : id(0), size(0) { |
202 } | 257 } |
203 | 258 |
204 VaapiVideoDecodeAccelerator::InputBuffer::~InputBuffer() { | 259 VaapiVideoDecodeAccelerator::InputBuffer::~InputBuffer() { |
205 } | 260 } |
206 | 261 |
207 void VaapiVideoDecodeAccelerator::NotifyError(Error error) { | 262 void VaapiVideoDecodeAccelerator::NotifyError(Error error) { |
208 if (message_loop_ != base::MessageLoop::current()) { | 263 if (message_loop_ != base::MessageLoop::current()) { |
209 DCHECK(decoder_thread_task_runner_->BelongsToCurrentThread()); | 264 DCHECK(decoder_thread_task_runner_->BelongsToCurrentThread()); |
210 message_loop_->PostTask(FROM_HERE, base::Bind( | 265 message_loop_->PostTask(FROM_HERE, base::Bind( |
(...skipping 83 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
294 } | 349 } |
295 | 350 |
296 if (profile >= media::H264PROFILE_MIN && profile <= media::H264PROFILE_MAX) { | 351 if (profile >= media::H264PROFILE_MIN && profile <= media::H264PROFILE_MAX) { |
297 h264_accelerator_.reset( | 352 h264_accelerator_.reset( |
298 new VaapiH264Accelerator(this, vaapi_wrapper_.get())); | 353 new VaapiH264Accelerator(this, vaapi_wrapper_.get())); |
299 decoder_.reset(new H264Decoder(h264_accelerator_.get())); | 354 decoder_.reset(new H264Decoder(h264_accelerator_.get())); |
300 } else if (profile >= media::VP8PROFILE_MIN && | 355 } else if (profile >= media::VP8PROFILE_MIN && |
301 profile <= media::VP8PROFILE_MAX) { | 356 profile <= media::VP8PROFILE_MAX) { |
302 vp8_accelerator_.reset(new VaapiVP8Accelerator(this, vaapi_wrapper_.get())); | 357 vp8_accelerator_.reset(new VaapiVP8Accelerator(this, vaapi_wrapper_.get())); |
303 decoder_.reset(new VP8Decoder(vp8_accelerator_.get())); | 358 decoder_.reset(new VP8Decoder(vp8_accelerator_.get())); |
| 359 } else if (profile >= media::VP9PROFILE_MIN && |
| 360 profile <= media::VP9PROFILE_MAX) { |
| 361 vp9_accelerator_.reset(new VaapiVP9Accelerator(this, vaapi_wrapper_.get())); |
| 362 decoder_.reset(new VP9Decoder(vp9_accelerator_.get())); |
304 } else { | 363 } else { |
305 DLOG(ERROR) << "Unsupported profile " << profile; | 364 DLOG(ERROR) << "Unsupported profile " << profile; |
306 return false; | 365 return false; |
307 } | 366 } |
308 | 367 |
309 CHECK(decoder_thread_.Start()); | 368 CHECK(decoder_thread_.Start()); |
310 decoder_thread_task_runner_ = decoder_thread_.task_runner(); | 369 decoder_thread_task_runner_ = decoder_thread_.task_runner(); |
311 | 370 |
312 state_ = kIdle; | 371 state_ = kIdle; |
313 return true; | 372 return true; |
(...skipping 666 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
980 const media::H264SPS* sps, | 1039 const media::H264SPS* sps, |
981 const media::H264PPS* pps, | 1040 const media::H264PPS* pps, |
982 const H264DPB& dpb, | 1041 const H264DPB& dpb, |
983 const H264Picture::Vector& ref_pic_listp0, | 1042 const H264Picture::Vector& ref_pic_listp0, |
984 const H264Picture::Vector& ref_pic_listb0, | 1043 const H264Picture::Vector& ref_pic_listb0, |
985 const H264Picture::Vector& ref_pic_listb1, | 1044 const H264Picture::Vector& ref_pic_listb1, |
986 const scoped_refptr<H264Picture>& pic) { | 1045 const scoped_refptr<H264Picture>& pic) { |
987 VAPictureParameterBufferH264 pic_param; | 1046 VAPictureParameterBufferH264 pic_param; |
988 memset(&pic_param, 0, sizeof(pic_param)); | 1047 memset(&pic_param, 0, sizeof(pic_param)); |
989 | 1048 |
990 #define FROM_SPS_TO_PP(a) pic_param.a = sps->a; | 1049 #define FROM_SPS_TO_PP(a) pic_param.a = sps->a |
991 #define FROM_SPS_TO_PP2(a, b) pic_param.b = sps->a; | 1050 #define FROM_SPS_TO_PP2(a, b) pic_param.b = sps->a |
992 FROM_SPS_TO_PP2(pic_width_in_mbs_minus1, picture_width_in_mbs_minus1); | 1051 FROM_SPS_TO_PP2(pic_width_in_mbs_minus1, picture_width_in_mbs_minus1); |
993 // This assumes non-interlaced video | 1052 // This assumes non-interlaced video |
994 FROM_SPS_TO_PP2(pic_height_in_map_units_minus1, picture_height_in_mbs_minus1); | 1053 FROM_SPS_TO_PP2(pic_height_in_map_units_minus1, picture_height_in_mbs_minus1); |
995 FROM_SPS_TO_PP(bit_depth_luma_minus8); | 1054 FROM_SPS_TO_PP(bit_depth_luma_minus8); |
996 FROM_SPS_TO_PP(bit_depth_chroma_minus8); | 1055 FROM_SPS_TO_PP(bit_depth_chroma_minus8); |
997 #undef FROM_SPS_TO_PP | 1056 #undef FROM_SPS_TO_PP |
998 #undef FROM_SPS_TO_PP2 | 1057 #undef FROM_SPS_TO_PP2 |
999 | 1058 |
1000 #define FROM_SPS_TO_PP_SF(a) pic_param.seq_fields.bits.a = sps->a; | 1059 #define FROM_SPS_TO_PP_SF(a) pic_param.seq_fields.bits.a = sps->a |
1001 #define FROM_SPS_TO_PP_SF2(a, b) pic_param.seq_fields.bits.b = sps->a; | 1060 #define FROM_SPS_TO_PP_SF2(a, b) pic_param.seq_fields.bits.b = sps->a |
1002 FROM_SPS_TO_PP_SF(chroma_format_idc); | 1061 FROM_SPS_TO_PP_SF(chroma_format_idc); |
1003 FROM_SPS_TO_PP_SF2(separate_colour_plane_flag, | 1062 FROM_SPS_TO_PP_SF2(separate_colour_plane_flag, |
1004 residual_colour_transform_flag); | 1063 residual_colour_transform_flag); |
1005 FROM_SPS_TO_PP_SF(gaps_in_frame_num_value_allowed_flag); | 1064 FROM_SPS_TO_PP_SF(gaps_in_frame_num_value_allowed_flag); |
1006 FROM_SPS_TO_PP_SF(frame_mbs_only_flag); | 1065 FROM_SPS_TO_PP_SF(frame_mbs_only_flag); |
1007 FROM_SPS_TO_PP_SF(mb_adaptive_frame_field_flag); | 1066 FROM_SPS_TO_PP_SF(mb_adaptive_frame_field_flag); |
1008 FROM_SPS_TO_PP_SF(direct_8x8_inference_flag); | 1067 FROM_SPS_TO_PP_SF(direct_8x8_inference_flag); |
1009 pic_param.seq_fields.bits.MinLumaBiPredSize8x8 = (sps->level_idc >= 31); | 1068 pic_param.seq_fields.bits.MinLumaBiPredSize8x8 = (sps->level_idc >= 31); |
1010 FROM_SPS_TO_PP_SF(log2_max_frame_num_minus4); | 1069 FROM_SPS_TO_PP_SF(log2_max_frame_num_minus4); |
1011 FROM_SPS_TO_PP_SF(pic_order_cnt_type); | 1070 FROM_SPS_TO_PP_SF(pic_order_cnt_type); |
1012 FROM_SPS_TO_PP_SF(log2_max_pic_order_cnt_lsb_minus4); | 1071 FROM_SPS_TO_PP_SF(log2_max_pic_order_cnt_lsb_minus4); |
1013 FROM_SPS_TO_PP_SF(delta_pic_order_always_zero_flag); | 1072 FROM_SPS_TO_PP_SF(delta_pic_order_always_zero_flag); |
1014 #undef FROM_SPS_TO_PP_SF | 1073 #undef FROM_SPS_TO_PP_SF |
1015 #undef FROM_SPS_TO_PP_SF2 | 1074 #undef FROM_SPS_TO_PP_SF2 |
1016 | 1075 |
1017 #define FROM_PPS_TO_PP(a) pic_param.a = pps->a; | 1076 #define FROM_PPS_TO_PP(a) pic_param.a = pps->a |
1018 FROM_PPS_TO_PP(num_slice_groups_minus1); | 1077 FROM_PPS_TO_PP(num_slice_groups_minus1); |
1019 pic_param.slice_group_map_type = 0; | 1078 pic_param.slice_group_map_type = 0; |
1020 pic_param.slice_group_change_rate_minus1 = 0; | 1079 pic_param.slice_group_change_rate_minus1 = 0; |
1021 FROM_PPS_TO_PP(pic_init_qp_minus26); | 1080 FROM_PPS_TO_PP(pic_init_qp_minus26); |
1022 FROM_PPS_TO_PP(pic_init_qs_minus26); | 1081 FROM_PPS_TO_PP(pic_init_qs_minus26); |
1023 FROM_PPS_TO_PP(chroma_qp_index_offset); | 1082 FROM_PPS_TO_PP(chroma_qp_index_offset); |
1024 FROM_PPS_TO_PP(second_chroma_qp_index_offset); | 1083 FROM_PPS_TO_PP(second_chroma_qp_index_offset); |
1025 #undef FROM_PPS_TO_PP | 1084 #undef FROM_PPS_TO_PP |
1026 | 1085 |
1027 #define FROM_PPS_TO_PP_PF(a) pic_param.pic_fields.bits.a = pps->a; | 1086 #define FROM_PPS_TO_PP_PF(a) pic_param.pic_fields.bits.a = pps->a |
1028 #define FROM_PPS_TO_PP_PF2(a, b) pic_param.pic_fields.bits.b = pps->a; | 1087 #define FROM_PPS_TO_PP_PF2(a, b) pic_param.pic_fields.bits.b = pps->a |
1029 FROM_PPS_TO_PP_PF(entropy_coding_mode_flag); | 1088 FROM_PPS_TO_PP_PF(entropy_coding_mode_flag); |
1030 FROM_PPS_TO_PP_PF(weighted_pred_flag); | 1089 FROM_PPS_TO_PP_PF(weighted_pred_flag); |
1031 FROM_PPS_TO_PP_PF(weighted_bipred_idc); | 1090 FROM_PPS_TO_PP_PF(weighted_bipred_idc); |
1032 FROM_PPS_TO_PP_PF(transform_8x8_mode_flag); | 1091 FROM_PPS_TO_PP_PF(transform_8x8_mode_flag); |
1033 | 1092 |
1034 pic_param.pic_fields.bits.field_pic_flag = 0; | 1093 pic_param.pic_fields.bits.field_pic_flag = 0; |
1035 FROM_PPS_TO_PP_PF(constrained_intra_pred_flag); | 1094 FROM_PPS_TO_PP_PF(constrained_intra_pred_flag); |
1036 FROM_PPS_TO_PP_PF2(bottom_field_pic_order_in_frame_present_flag, | 1095 FROM_PPS_TO_PP_PF2(bottom_field_pic_order_in_frame_present_flag, |
1037 pic_order_present_flag); | 1096 pic_order_present_flag); |
1038 FROM_PPS_TO_PP_PF(deblocking_filter_control_present_flag); | 1097 FROM_PPS_TO_PP_PF(deblocking_filter_control_present_flag); |
(...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1100 const uint8_t* data, | 1159 const uint8_t* data, |
1101 size_t size) { | 1160 size_t size) { |
1102 VASliceParameterBufferH264 slice_param; | 1161 VASliceParameterBufferH264 slice_param; |
1103 memset(&slice_param, 0, sizeof(slice_param)); | 1162 memset(&slice_param, 0, sizeof(slice_param)); |
1104 | 1163 |
1105 slice_param.slice_data_size = slice_hdr->nalu_size; | 1164 slice_param.slice_data_size = slice_hdr->nalu_size; |
1106 slice_param.slice_data_offset = 0; | 1165 slice_param.slice_data_offset = 0; |
1107 slice_param.slice_data_flag = VA_SLICE_DATA_FLAG_ALL; | 1166 slice_param.slice_data_flag = VA_SLICE_DATA_FLAG_ALL; |
1108 slice_param.slice_data_bit_offset = slice_hdr->header_bit_size; | 1167 slice_param.slice_data_bit_offset = slice_hdr->header_bit_size; |
1109 | 1168 |
1110 #define SHDRToSP(a) slice_param.a = slice_hdr->a; | 1169 #define SHDRToSP(a) slice_param.a = slice_hdr->a |
1111 SHDRToSP(first_mb_in_slice); | 1170 SHDRToSP(first_mb_in_slice); |
1112 slice_param.slice_type = slice_hdr->slice_type % 5; | 1171 slice_param.slice_type = slice_hdr->slice_type % 5; |
1113 SHDRToSP(direct_spatial_mv_pred_flag); | 1172 SHDRToSP(direct_spatial_mv_pred_flag); |
1114 | 1173 |
1115 // TODO posciak: make sure parser sets those even when override flags | 1174 // TODO posciak: make sure parser sets those even when override flags |
1116 // in slice header is off. | 1175 // in slice header is off. |
1117 SHDRToSP(num_ref_idx_l0_active_minus1); | 1176 SHDRToSP(num_ref_idx_l0_active_minus1); |
1118 SHDRToSP(num_ref_idx_l1_active_minus1); | 1177 SHDRToSP(num_ref_idx_l1_active_minus1); |
1119 SHDRToSP(cabac_init_idc); | 1178 SHDRToSP(cabac_init_idc); |
1120 SHDRToSP(slice_qp_delta); | 1179 SHDRToSP(slice_qp_delta); |
(...skipping 262 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1383 VP8PictureToVaapiDecodeSurface(alt_frame); | 1442 VP8PictureToVaapiDecodeSurface(alt_frame); |
1384 pic_param.alt_ref_frame = alt_frame_surface->va_surface()->id(); | 1443 pic_param.alt_ref_frame = alt_frame_surface->va_surface()->id(); |
1385 } else { | 1444 } else { |
1386 pic_param.alt_ref_frame = VA_INVALID_SURFACE; | 1445 pic_param.alt_ref_frame = VA_INVALID_SURFACE; |
1387 } | 1446 } |
1388 | 1447 |
1389 pic_param.out_of_loop_frame = VA_INVALID_SURFACE; | 1448 pic_param.out_of_loop_frame = VA_INVALID_SURFACE; |
1390 | 1449 |
1391 const media::Vp8LoopFilterHeader& lf_hdr = frame_hdr->loopfilter_hdr; | 1450 const media::Vp8LoopFilterHeader& lf_hdr = frame_hdr->loopfilter_hdr; |
1392 | 1451 |
1393 #define FHDR_TO_PP_PF(a, b) pic_param.pic_fields.bits.a = (b); | 1452 #define FHDR_TO_PP_PF(a, b) pic_param.pic_fields.bits.a = (b) |
1394 FHDR_TO_PP_PF(key_frame, frame_hdr->IsKeyframe() ? 0 : 1); | 1453 FHDR_TO_PP_PF(key_frame, frame_hdr->IsKeyframe() ? 0 : 1); |
1395 FHDR_TO_PP_PF(version, frame_hdr->version); | 1454 FHDR_TO_PP_PF(version, frame_hdr->version); |
1396 FHDR_TO_PP_PF(segmentation_enabled, sgmnt_hdr.segmentation_enabled); | 1455 FHDR_TO_PP_PF(segmentation_enabled, sgmnt_hdr.segmentation_enabled); |
1397 FHDR_TO_PP_PF(update_mb_segmentation_map, | 1456 FHDR_TO_PP_PF(update_mb_segmentation_map, |
1398 sgmnt_hdr.update_mb_segmentation_map); | 1457 sgmnt_hdr.update_mb_segmentation_map); |
1399 FHDR_TO_PP_PF(update_segment_feature_data, | 1458 FHDR_TO_PP_PF(update_segment_feature_data, |
1400 sgmnt_hdr.update_segment_feature_data); | 1459 sgmnt_hdr.update_segment_feature_data); |
1401 FHDR_TO_PP_PF(filter_type, lf_hdr.type); | 1460 FHDR_TO_PP_PF(filter_type, lf_hdr.type); |
1402 FHDR_TO_PP_PF(sharpness_level, lf_hdr.sharpness_level); | 1461 FHDR_TO_PP_PF(sharpness_level, lf_hdr.sharpness_level); |
1403 FHDR_TO_PP_PF(loop_filter_adj_enable, lf_hdr.loop_filter_adj_enable); | 1462 FHDR_TO_PP_PF(loop_filter_adj_enable, lf_hdr.loop_filter_adj_enable); |
(...skipping 29 matching lines...) Expand all Loading... |
1433 arraysize(lf_hdr.mb_mode_delta) == | 1492 arraysize(lf_hdr.mb_mode_delta) == |
1434 arraysize(pic_param.loop_filter_deltas_mode) && | 1493 arraysize(pic_param.loop_filter_deltas_mode) && |
1435 arraysize(lf_hdr.ref_frame_delta) == | 1494 arraysize(lf_hdr.ref_frame_delta) == |
1436 arraysize(lf_hdr.mb_mode_delta), | 1495 arraysize(lf_hdr.mb_mode_delta), |
1437 "loop filter deltas arrays size mismatch"); | 1496 "loop filter deltas arrays size mismatch"); |
1438 for (size_t i = 0; i < arraysize(lf_hdr.ref_frame_delta); ++i) { | 1497 for (size_t i = 0; i < arraysize(lf_hdr.ref_frame_delta); ++i) { |
1439 pic_param.loop_filter_deltas_ref_frame[i] = lf_hdr.ref_frame_delta[i]; | 1498 pic_param.loop_filter_deltas_ref_frame[i] = lf_hdr.ref_frame_delta[i]; |
1440 pic_param.loop_filter_deltas_mode[i] = lf_hdr.mb_mode_delta[i]; | 1499 pic_param.loop_filter_deltas_mode[i] = lf_hdr.mb_mode_delta[i]; |
1441 } | 1500 } |
1442 | 1501 |
1443 #define FHDR_TO_PP(a) pic_param.a = frame_hdr->a; | 1502 #define FHDR_TO_PP(a) pic_param.a = frame_hdr->a |
1444 FHDR_TO_PP(prob_skip_false); | 1503 FHDR_TO_PP(prob_skip_false); |
1445 FHDR_TO_PP(prob_intra); | 1504 FHDR_TO_PP(prob_intra); |
1446 FHDR_TO_PP(prob_last); | 1505 FHDR_TO_PP(prob_last); |
1447 FHDR_TO_PP(prob_gf); | 1506 FHDR_TO_PP(prob_gf); |
1448 #undef FHDR_TO_PP | 1507 #undef FHDR_TO_PP |
1449 | 1508 |
1450 ARRAY_MEMCPY_CHECKED(pic_param.y_mode_probs, entr_hdr.y_mode_probs); | 1509 ARRAY_MEMCPY_CHECKED(pic_param.y_mode_probs, entr_hdr.y_mode_probs); |
1451 ARRAY_MEMCPY_CHECKED(pic_param.uv_mode_probs, entr_hdr.uv_mode_probs); | 1510 ARRAY_MEMCPY_CHECKED(pic_param.uv_mode_probs, entr_hdr.uv_mode_probs); |
1452 ARRAY_MEMCPY_CHECKED(pic_param.mv_probs, entr_hdr.mv_probs); | 1511 ARRAY_MEMCPY_CHECKED(pic_param.mv_probs, entr_hdr.mv_probs); |
1453 | 1512 |
1454 pic_param.bool_coder_ctx.range = frame_hdr->bool_dec_range; | 1513 pic_param.bool_coder_ctx.range = frame_hdr->bool_dec_range; |
1455 pic_param.bool_coder_ctx.value = frame_hdr->bool_dec_value; | 1514 pic_param.bool_coder_ctx.value = frame_hdr->bool_dec_value; |
1456 pic_param.bool_coder_ctx.count = frame_hdr->bool_dec_count; | 1515 pic_param.bool_coder_ctx.count = frame_hdr->bool_dec_count; |
1457 | 1516 |
1458 if (!vaapi_wrapper_->SubmitBuffer(VAPictureParameterBufferType, | 1517 if (!vaapi_wrapper_->SubmitBuffer(VAPictureParameterBufferType, |
1459 sizeof(VAPictureParameterBufferVP8), | 1518 sizeof(pic_param), &pic_param)) |
1460 &pic_param)) | |
1461 return false; | 1519 return false; |
1462 | 1520 |
1463 VASliceParameterBufferVP8 slice_param; | 1521 VASliceParameterBufferVP8 slice_param; |
1464 memset(&slice_param, 0, sizeof(VASliceParameterBufferVP8)); | 1522 memset(&slice_param, 0, sizeof(slice_param)); |
1465 slice_param.slice_data_size = frame_hdr->frame_size; | 1523 slice_param.slice_data_size = frame_hdr->frame_size; |
1466 slice_param.slice_data_offset = frame_hdr->first_part_offset; | 1524 slice_param.slice_data_offset = frame_hdr->first_part_offset; |
1467 slice_param.slice_data_flag = VA_SLICE_DATA_FLAG_ALL; | 1525 slice_param.slice_data_flag = VA_SLICE_DATA_FLAG_ALL; |
1468 slice_param.macroblock_offset = frame_hdr->macroblock_bit_offset; | 1526 slice_param.macroblock_offset = frame_hdr->macroblock_bit_offset; |
1469 // Number of DCT partitions plus control partition. | 1527 // Number of DCT partitions plus control partition. |
1470 slice_param.num_of_partitions = frame_hdr->num_of_dct_partitions + 1; | 1528 slice_param.num_of_partitions = frame_hdr->num_of_dct_partitions + 1; |
1471 | 1529 |
1472 // Per VAAPI, this size only includes the size of the macroblock data in | 1530 // Per VAAPI, this size only includes the size of the macroblock data in |
1473 // the first partition (in bytes), so we have to subtract the header size. | 1531 // the first partition (in bytes), so we have to subtract the header size. |
1474 slice_param.partition_size[0] = | 1532 slice_param.partition_size[0] = |
(...skipping 29 matching lines...) Expand all Loading... |
1504 } | 1562 } |
1505 | 1563 |
1506 scoped_refptr<VaapiVideoDecodeAccelerator::VaapiDecodeSurface> | 1564 scoped_refptr<VaapiVideoDecodeAccelerator::VaapiDecodeSurface> |
1507 VaapiVideoDecodeAccelerator::VaapiVP8Accelerator:: | 1565 VaapiVideoDecodeAccelerator::VaapiVP8Accelerator:: |
1508 VP8PictureToVaapiDecodeSurface(const scoped_refptr<VP8Picture>& pic) { | 1566 VP8PictureToVaapiDecodeSurface(const scoped_refptr<VP8Picture>& pic) { |
1509 VaapiVP8Picture* vaapi_pic = pic->AsVaapiVP8Picture(); | 1567 VaapiVP8Picture* vaapi_pic = pic->AsVaapiVP8Picture(); |
1510 CHECK(vaapi_pic); | 1568 CHECK(vaapi_pic); |
1511 return vaapi_pic->dec_surface(); | 1569 return vaapi_pic->dec_surface(); |
1512 } | 1570 } |
1513 | 1571 |
| 1572 VaapiVideoDecodeAccelerator::VaapiVP9Accelerator::VaapiVP9Accelerator( |
| 1573 VaapiVideoDecodeAccelerator* vaapi_dec, |
| 1574 VaapiWrapper* vaapi_wrapper) |
| 1575 : vaapi_wrapper_(vaapi_wrapper), vaapi_dec_(vaapi_dec) { |
| 1576 DCHECK(vaapi_wrapper_); |
| 1577 DCHECK(vaapi_dec_); |
| 1578 } |
| 1579 |
| 1580 VaapiVideoDecodeAccelerator::VaapiVP9Accelerator::~VaapiVP9Accelerator() {} |
| 1581 |
| 1582 scoped_refptr<VP9Picture> |
| 1583 VaapiVideoDecodeAccelerator::VaapiVP9Accelerator::CreateVP9Picture() { |
| 1584 scoped_refptr<VaapiDecodeSurface> va_surface = vaapi_dec_->CreateSurface(); |
| 1585 if (!va_surface) |
| 1586 return nullptr; |
| 1587 |
| 1588 return new VaapiVP9Picture(va_surface); |
| 1589 } |
| 1590 |
| 1591 bool VaapiVideoDecodeAccelerator::VaapiVP9Accelerator::SubmitDecode( |
| 1592 const scoped_refptr<VP9Picture>& pic, |
| 1593 const media::Vp9Segmentation& seg, |
| 1594 const media::Vp9LoopFilter& lf, |
| 1595 const std::vector<scoped_refptr<VP9Picture>>& ref_pictures) { |
| 1596 VADecPictureParameterBufferVP9 pic_param; |
| 1597 memset(&pic_param, 0, sizeof(pic_param)); |
| 1598 |
| 1599 const media::Vp9FrameHeader* frame_hdr = pic->frame_hdr.get(); |
| 1600 DCHECK(frame_hdr); |
| 1601 |
| 1602 if (frame_hdr->profile != 0) { |
| 1603 DVLOG(1) << "Unsupported profile" << frame_hdr->profile; |
| 1604 return false; |
| 1605 } |
| 1606 |
| 1607 pic_param.frame_width = base::checked_cast<uint16_t>(frame_hdr->width); |
| 1608 pic_param.frame_height = base::checked_cast<uint16_t>(frame_hdr->height); |
| 1609 |
| 1610 CHECK_EQ(ref_pictures.size(), arraysize(pic_param.reference_frames)); |
| 1611 for (size_t i = 0; i < arraysize(pic_param.reference_frames); ++i) { |
| 1612 VASurfaceID va_surface_id; |
| 1613 if (ref_pictures[i]) { |
| 1614 scoped_refptr<VaapiDecodeSurface> surface = |
| 1615 VP9PictureToVaapiDecodeSurface(ref_pictures[i]); |
| 1616 va_surface_id = surface->va_surface()->id(); |
| 1617 } else { |
| 1618 va_surface_id = VA_INVALID_SURFACE; |
| 1619 } |
| 1620 |
| 1621 pic_param.reference_frames[i] = va_surface_id; |
| 1622 } |
| 1623 |
| 1624 #define FHDR_TO_PP_PF1(a) pic_param.pic_fields.bits.a = frame_hdr->a |
| 1625 #define FHDR_TO_PP_PF2(a, b) pic_param.pic_fields.bits.a = b |
| 1626 FHDR_TO_PP_PF2(subsampling_x, frame_hdr->subsampling_x == 1); |
| 1627 FHDR_TO_PP_PF2(subsampling_y, frame_hdr->subsampling_y == 1); |
| 1628 FHDR_TO_PP_PF2(frame_type, frame_hdr->IsKeyframe() ? 0 : 1); |
| 1629 FHDR_TO_PP_PF1(show_frame); |
| 1630 FHDR_TO_PP_PF1(error_resilient_mode); |
| 1631 FHDR_TO_PP_PF1(intra_only); |
| 1632 FHDR_TO_PP_PF1(allow_high_precision_mv); |
| 1633 FHDR_TO_PP_PF2(mcomp_filter_type, frame_hdr->interp_filter); |
| 1634 FHDR_TO_PP_PF1(frame_parallel_decoding_mode); |
| 1635 FHDR_TO_PP_PF2(reset_frame_context, frame_hdr->reset_context); |
| 1636 FHDR_TO_PP_PF1(refresh_frame_context); |
| 1637 FHDR_TO_PP_PF1(frame_context_idx); |
| 1638 FHDR_TO_PP_PF2(segmentation_enabled, seg.enabled); |
| 1639 FHDR_TO_PP_PF2(segmentation_temporal_update, seg.temporal_update); |
| 1640 FHDR_TO_PP_PF2(segmentation_update_map, seg.update_map); |
| 1641 FHDR_TO_PP_PF2(last_ref_frame, frame_hdr->frame_refs[0]); |
| 1642 FHDR_TO_PP_PF2(last_ref_frame_sign_bias, frame_hdr->ref_sign_biases[0]); |
| 1643 FHDR_TO_PP_PF2(golden_ref_frame, frame_hdr->frame_refs[1]); |
| 1644 FHDR_TO_PP_PF2(golden_ref_frame_sign_bias, frame_hdr->ref_sign_biases[1]); |
| 1645 FHDR_TO_PP_PF2(alt_ref_frame, frame_hdr->frame_refs[2]); |
| 1646 FHDR_TO_PP_PF2(alt_ref_frame_sign_bias, frame_hdr->ref_sign_biases[2]); |
| 1647 FHDR_TO_PP_PF2(lossless_flag, frame_hdr->quant_params.IsLossless()); |
| 1648 #undef FHDR_TO_PP_PF2 |
| 1649 #undef FHDR_TO_PP_PF1 |
| 1650 |
| 1651 pic_param.filter_level = lf.filter_level; |
| 1652 pic_param.sharpness_level = lf.sharpness_level; |
| 1653 pic_param.log2_tile_rows = frame_hdr->log2_tile_rows; |
| 1654 pic_param.log2_tile_columns = frame_hdr->log2_tile_cols; |
| 1655 pic_param.frame_header_length_in_bytes = frame_hdr->uncompressed_header_size; |
| 1656 pic_param.first_partition_size = frame_hdr->first_partition_size; |
| 1657 |
| 1658 ARRAY_MEMCPY_CHECKED(pic_param.mb_segment_tree_probs, seg.tree_probs); |
| 1659 ARRAY_MEMCPY_CHECKED(pic_param.segment_pred_probs, seg.pred_probs); |
| 1660 |
| 1661 pic_param.profile = frame_hdr->profile; |
| 1662 |
| 1663 if (!vaapi_wrapper_->SubmitBuffer(VAPictureParameterBufferType, |
| 1664 sizeof(pic_param), &pic_param)) |
| 1665 return false; |
| 1666 |
| 1667 VASliceParameterBufferVP9 slice_param; |
| 1668 memset(&slice_param, 0, sizeof(slice_param)); |
| 1669 slice_param.slice_data_size = frame_hdr->frame_size; |
| 1670 slice_param.slice_data_offset = 0; |
| 1671 slice_param.slice_data_flag = VA_SLICE_DATA_FLAG_ALL; |
| 1672 |
| 1673 static_assert(arraysize(media::Vp9Segmentation::feature_enabled) == |
| 1674 arraysize(slice_param.seg_param), |
| 1675 "seg_param array of incorrect size"); |
| 1676 for (size_t i = 0; i < arraysize(slice_param.seg_param); ++i) { |
| 1677 VASegmentParameterVP9& seg_param = slice_param.seg_param[i]; |
| 1678 #define SEG_TO_SP_SF(a, b) seg_param.segment_flags.fields.a = b |
| 1679 SEG_TO_SP_SF( |
| 1680 segment_reference_enabled, |
| 1681 seg.FeatureEnabled(i, media::Vp9Segmentation::SEG_LVL_REF_FRAME)); |
| 1682 SEG_TO_SP_SF(segment_reference, |
| 1683 seg.FeatureData(i, media::Vp9Segmentation::SEG_LVL_REF_FRAME)); |
| 1684 SEG_TO_SP_SF(segment_reference_skipped, |
| 1685 seg.FeatureEnabled(i, media::Vp9Segmentation::SEG_LVL_SKIP)); |
| 1686 #undef SEG_TO_SP_SF |
| 1687 |
| 1688 ARRAY_MEMCPY_CHECKED(seg_param.filter_level, lf.lvl[i]); |
| 1689 |
| 1690 seg_param.luma_dc_quant_scale = seg.y_dequant[i][0]; |
| 1691 seg_param.luma_ac_quant_scale = seg.y_dequant[i][1]; |
| 1692 seg_param.chroma_dc_quant_scale = seg.uv_dequant[i][0]; |
| 1693 seg_param.chroma_ac_quant_scale = seg.uv_dequant[i][1]; |
| 1694 } |
| 1695 |
| 1696 if (!vaapi_wrapper_->SubmitBuffer(VASliceParameterBufferType, |
| 1697 sizeof(slice_param), &slice_param)) |
| 1698 return false; |
| 1699 |
| 1700 void* non_const_ptr = const_cast<uint8*>(frame_hdr->data); |
| 1701 if (!vaapi_wrapper_->SubmitBuffer(VASliceDataBufferType, |
| 1702 frame_hdr->frame_size, non_const_ptr)) |
| 1703 return false; |
| 1704 |
| 1705 scoped_refptr<VaapiDecodeSurface> dec_surface = |
| 1706 VP9PictureToVaapiDecodeSurface(pic); |
| 1707 |
| 1708 return vaapi_dec_->DecodeSurface(dec_surface); |
| 1709 } |
| 1710 |
| 1711 bool VaapiVideoDecodeAccelerator::VaapiVP9Accelerator::OutputPicture( |
| 1712 const scoped_refptr<VP9Picture>& pic) { |
| 1713 scoped_refptr<VaapiDecodeSurface> dec_surface = |
| 1714 VP9PictureToVaapiDecodeSurface(pic); |
| 1715 |
| 1716 vaapi_dec_->SurfaceReady(dec_surface); |
| 1717 return true; |
| 1718 } |
| 1719 |
| 1720 scoped_refptr<VaapiVideoDecodeAccelerator::VaapiDecodeSurface> |
| 1721 VaapiVideoDecodeAccelerator::VaapiVP9Accelerator:: |
| 1722 VP9PictureToVaapiDecodeSurface(const scoped_refptr<VP9Picture>& pic) { |
| 1723 VaapiVP9Picture* vaapi_pic = pic->AsVaapiVP9Picture(); |
| 1724 CHECK(vaapi_pic); |
| 1725 return vaapi_pic->dec_surface(); |
| 1726 } |
| 1727 |
1514 // static | 1728 // static |
1515 media::VideoDecodeAccelerator::SupportedProfiles | 1729 media::VideoDecodeAccelerator::SupportedProfiles |
1516 VaapiVideoDecodeAccelerator::GetSupportedProfiles() { | 1730 VaapiVideoDecodeAccelerator::GetSupportedProfiles() { |
1517 return VaapiWrapper::GetSupportedDecodeProfiles(); | 1731 return VaapiWrapper::GetSupportedDecodeProfiles(); |
1518 } | 1732 } |
1519 | 1733 |
1520 } // namespace content | 1734 } // namespace content |
OLD | NEW |