You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

534 lines
22KB

  1. /*
  2. * DXVA2 H264 HW acceleration.
  3. *
  4. * copyright (c) 2009 Laurent Aimar
  5. *
  6. * This file is part of Libav.
  7. *
  8. * Libav is free software; you can redistribute it and/or
  9. * modify it under the terms of the GNU Lesser General Public
  10. * License as published by the Free Software Foundation; either
  11. * version 2.1 of the License, or (at your option) any later version.
  12. *
  13. * Libav is distributed in the hope that it will be useful,
  14. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16. * Lesser General Public License for more details.
  17. *
  18. * You should have received a copy of the GNU Lesser General Public
  19. * License along with Libav; if not, write to the Free Software
  20. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21. */
  22. #include "dxva2_internal.h"
  23. #include "h264.h"
  24. #include "h264data.h"
  25. #include "mpegutils.h"
  26. struct dxva2_picture_context {
  27. DXVA_PicParams_H264 pp;
  28. DXVA_Qmatrix_H264 qm;
  29. unsigned slice_count;
  30. DXVA_Slice_H264_Short slice_short[MAX_SLICES];
  31. DXVA_Slice_H264_Long slice_long[MAX_SLICES];
  32. const uint8_t *bitstream;
  33. unsigned bitstream_size;
  34. };
  35. static void fill_picture_entry(DXVA_PicEntry_H264 *pic,
  36. unsigned index, unsigned flag)
  37. {
  38. assert((index&0x7f) == index && (flag&0x01) == flag);
  39. pic->bPicEntry = index | (flag << 7);
  40. }
  41. static void fill_picture_parameters(const AVCodecContext *avctx, AVDXVAContext *ctx, const H264Context *h,
  42. DXVA_PicParams_H264 *pp)
  43. {
  44. const H264Picture *current_picture = h->cur_pic_ptr;
  45. int i, j;
  46. memset(pp, 0, sizeof(*pp));
  47. /* Configure current picture */
  48. fill_picture_entry(&pp->CurrPic,
  49. ff_dxva2_get_surface_index(avctx, ctx, current_picture->f),
  50. h->picture_structure == PICT_BOTTOM_FIELD);
  51. /* Configure the set of references */
  52. pp->UsedForReferenceFlags = 0;
  53. pp->NonExistingFrameFlags = 0;
  54. for (i = 0, j = 0; i < FF_ARRAY_ELEMS(pp->RefFrameList); i++) {
  55. const H264Picture *r;
  56. if (j < h->short_ref_count) {
  57. r = h->short_ref[j++];
  58. } else {
  59. r = NULL;
  60. while (!r && j < h->short_ref_count + 16)
  61. r = h->long_ref[j++ - h->short_ref_count];
  62. }
  63. if (r) {
  64. fill_picture_entry(&pp->RefFrameList[i],
  65. ff_dxva2_get_surface_index(avctx, ctx, r->f),
  66. r->long_ref != 0);
  67. if ((r->reference & PICT_TOP_FIELD) && r->field_poc[0] != INT_MAX)
  68. pp->FieldOrderCntList[i][0] = r->field_poc[0];
  69. if ((r->reference & PICT_BOTTOM_FIELD) && r->field_poc[1] != INT_MAX)
  70. pp->FieldOrderCntList[i][1] = r->field_poc[1];
  71. pp->FrameNumList[i] = r->long_ref ? r->pic_id : r->frame_num;
  72. if (r->reference & PICT_TOP_FIELD)
  73. pp->UsedForReferenceFlags |= 1 << (2*i + 0);
  74. if (r->reference & PICT_BOTTOM_FIELD)
  75. pp->UsedForReferenceFlags |= 1 << (2*i + 1);
  76. } else {
  77. pp->RefFrameList[i].bPicEntry = 0xff;
  78. pp->FieldOrderCntList[i][0] = 0;
  79. pp->FieldOrderCntList[i][1] = 0;
  80. pp->FrameNumList[i] = 0;
  81. }
  82. }
  83. pp->wFrameWidthInMbsMinus1 = h->mb_width - 1;
  84. pp->wFrameHeightInMbsMinus1 = h->mb_height - 1;
  85. pp->num_ref_frames = h->sps.ref_frame_count;
  86. pp->wBitFields = ((h->picture_structure != PICT_FRAME) << 0) |
  87. ((h->sps.mb_aff &&
  88. (h->picture_structure == PICT_FRAME)) << 1) |
  89. (h->sps.residual_color_transform_flag << 2) |
  90. /* sp_for_switch_flag (not implemented by Libav) */
  91. (0 << 3) |
  92. (h->sps.chroma_format_idc << 4) |
  93. ((h->nal_ref_idc != 0) << 6) |
  94. (h->pps.constrained_intra_pred << 7) |
  95. (h->pps.weighted_pred << 8) |
  96. (h->pps.weighted_bipred_idc << 9) |
  97. /* MbsConsecutiveFlag */
  98. (1 << 11) |
  99. (h->sps.frame_mbs_only_flag << 12) |
  100. (h->pps.transform_8x8_mode << 13) |
  101. ((h->sps.level_idc >= 31) << 14) |
  102. /* IntraPicFlag (Modified if we detect a non
  103. * intra slice in dxva2_h264_decode_slice) */
  104. (1 << 15);
  105. pp->bit_depth_luma_minus8 = h->sps.bit_depth_luma - 8;
  106. pp->bit_depth_chroma_minus8 = h->sps.bit_depth_chroma - 8;
  107. if (DXVA_CONTEXT_WORKAROUND(avctx, ctx) & FF_DXVA2_WORKAROUND_SCALING_LIST_ZIGZAG)
  108. pp->Reserved16Bits = 0;
  109. else if (DXVA_CONTEXT_WORKAROUND(avctx, ctx) & FF_DXVA2_WORKAROUND_INTEL_CLEARVIDEO)
  110. pp->Reserved16Bits = 0x34c;
  111. else
  112. pp->Reserved16Bits = 3; /* FIXME is there a way to detect the right mode ? */
  113. pp->StatusReportFeedbackNumber = 1 + DXVA_CONTEXT_REPORT_ID(avctx, ctx)++;
  114. pp->CurrFieldOrderCnt[0] = 0;
  115. if ((h->picture_structure & PICT_TOP_FIELD) &&
  116. current_picture->field_poc[0] != INT_MAX)
  117. pp->CurrFieldOrderCnt[0] = current_picture->field_poc[0];
  118. pp->CurrFieldOrderCnt[1] = 0;
  119. if ((h->picture_structure & PICT_BOTTOM_FIELD) &&
  120. current_picture->field_poc[1] != INT_MAX)
  121. pp->CurrFieldOrderCnt[1] = current_picture->field_poc[1];
  122. pp->pic_init_qs_minus26 = h->pps.init_qs - 26;
  123. pp->chroma_qp_index_offset = h->pps.chroma_qp_index_offset[0];
  124. pp->second_chroma_qp_index_offset = h->pps.chroma_qp_index_offset[1];
  125. pp->ContinuationFlag = 1;
  126. pp->pic_init_qp_minus26 = h->pps.init_qp - 26;
  127. pp->num_ref_idx_l0_active_minus1 = h->pps.ref_count[0] - 1;
  128. pp->num_ref_idx_l1_active_minus1 = h->pps.ref_count[1] - 1;
  129. pp->Reserved8BitsA = 0;
  130. pp->frame_num = h->frame_num;
  131. pp->log2_max_frame_num_minus4 = h->sps.log2_max_frame_num - 4;
  132. pp->pic_order_cnt_type = h->sps.poc_type;
  133. if (h->sps.poc_type == 0)
  134. pp->log2_max_pic_order_cnt_lsb_minus4 = h->sps.log2_max_poc_lsb - 4;
  135. else if (h->sps.poc_type == 1)
  136. pp->delta_pic_order_always_zero_flag = h->sps.delta_pic_order_always_zero_flag;
  137. pp->direct_8x8_inference_flag = h->sps.direct_8x8_inference_flag;
  138. pp->entropy_coding_mode_flag = h->pps.cabac;
  139. pp->pic_order_present_flag = h->pps.pic_order_present;
  140. pp->num_slice_groups_minus1 = h->pps.slice_group_count - 1;
  141. pp->slice_group_map_type = h->pps.mb_slice_group_map_type;
  142. pp->deblocking_filter_control_present_flag = h->pps.deblocking_filter_parameters_present;
  143. pp->redundant_pic_cnt_present_flag= h->pps.redundant_pic_cnt_present;
  144. pp->Reserved8BitsB = 0;
  145. pp->slice_group_change_rate_minus1= 0; /* XXX not implemented by Libav */
  146. //pp->SliceGroupMap[810]; /* XXX not implemented by Libav */
  147. }
  148. static void fill_scaling_lists(const AVCodecContext *avctx, AVDXVAContext *ctx, const H264Context *h, DXVA_Qmatrix_H264 *qm)
  149. {
  150. unsigned i, j;
  151. memset(qm, 0, sizeof(*qm));
  152. if (DXVA_CONTEXT_WORKAROUND(avctx, ctx) & FF_DXVA2_WORKAROUND_SCALING_LIST_ZIGZAG) {
  153. for (i = 0; i < 6; i++)
  154. for (j = 0; j < 16; j++)
  155. qm->bScalingLists4x4[i][j] = h->pps.scaling_matrix4[i][j];
  156. for (i = 0; i < 64; i++) {
  157. qm->bScalingLists8x8[0][i] = h->pps.scaling_matrix8[0][i];
  158. qm->bScalingLists8x8[1][i] = h->pps.scaling_matrix8[3][i];
  159. }
  160. } else {
  161. for (i = 0; i < 6; i++)
  162. for (j = 0; j < 16; j++)
  163. qm->bScalingLists4x4[i][j] = h->pps.scaling_matrix4[i][zigzag_scan[j]];
  164. for (i = 0; i < 64; i++) {
  165. qm->bScalingLists8x8[0][i] = h->pps.scaling_matrix8[0][ff_zigzag_direct[i]];
  166. qm->bScalingLists8x8[1][i] = h->pps.scaling_matrix8[3][ff_zigzag_direct[i]];
  167. }
  168. }
  169. }
  170. static int is_slice_short(const AVCodecContext *avctx, AVDXVAContext *ctx)
  171. {
  172. assert(DXVA_CONTEXT_CFG_BITSTREAM(avctx, ctx) == 1 ||
  173. DXVA_CONTEXT_CFG_BITSTREAM(avctx, ctx) == 2);
  174. return DXVA_CONTEXT_CFG_BITSTREAM(avctx, ctx) == 2;
  175. }
  176. static void fill_slice_short(DXVA_Slice_H264_Short *slice,
  177. unsigned position, unsigned size)
  178. {
  179. memset(slice, 0, sizeof(*slice));
  180. slice->BSNALunitDataLocation = position;
  181. slice->SliceBytesInBuffer = size;
  182. slice->wBadSliceChopping = 0;
  183. }
  184. static int get_refpic_index(const DXVA_PicParams_H264 *pp, int surface_index)
  185. {
  186. int i;
  187. for (i = 0; i < FF_ARRAY_ELEMS(pp->RefFrameList); i++) {
  188. if ((pp->RefFrameList[i].bPicEntry & 0x7f) == surface_index)
  189. return i;
  190. }
  191. return 0x7f;
  192. }
  193. static void fill_slice_long(AVCodecContext *avctx, DXVA_Slice_H264_Long *slice,
  194. const DXVA_PicParams_H264 *pp, unsigned position, unsigned size)
  195. {
  196. const H264Context *h = avctx->priv_data;
  197. H264SliceContext *sl = &h->slice_ctx[0];
  198. AVDXVAContext *ctx = avctx->hwaccel_context;
  199. unsigned list;
  200. memset(slice, 0, sizeof(*slice));
  201. slice->BSNALunitDataLocation = position;
  202. slice->SliceBytesInBuffer = size;
  203. slice->wBadSliceChopping = 0;
  204. slice->first_mb_in_slice = (sl->mb_y >> FIELD_OR_MBAFF_PICTURE(h)) * h->mb_width + sl->mb_x;
  205. slice->NumMbsForSlice = 0; /* XXX it is set once we have all slices */
  206. slice->BitOffsetToSliceData = get_bits_count(&sl->gb);
  207. slice->slice_type = ff_h264_get_slice_type(sl);
  208. if (sl->slice_type_fixed)
  209. slice->slice_type += 5;
  210. slice->luma_log2_weight_denom = sl->luma_log2_weight_denom;
  211. slice->chroma_log2_weight_denom = sl->chroma_log2_weight_denom;
  212. if (sl->list_count > 0)
  213. slice->num_ref_idx_l0_active_minus1 = sl->ref_count[0] - 1;
  214. if (sl->list_count > 1)
  215. slice->num_ref_idx_l1_active_minus1 = sl->ref_count[1] - 1;
  216. slice->slice_alpha_c0_offset_div2 = sl->slice_alpha_c0_offset / 2;
  217. slice->slice_beta_offset_div2 = sl->slice_beta_offset / 2;
  218. slice->Reserved8Bits = 0;
  219. for (list = 0; list < 2; list++) {
  220. unsigned i;
  221. for (i = 0; i < FF_ARRAY_ELEMS(slice->RefPicList[list]); i++) {
  222. if (list < sl->list_count && i < sl->ref_count[list]) {
  223. const H264Picture *r = sl->ref_list[list][i].parent;
  224. unsigned plane;
  225. unsigned index;
  226. if (DXVA_CONTEXT_WORKAROUND(avctx, ctx) & FF_DXVA2_WORKAROUND_INTEL_CLEARVIDEO)
  227. index = ff_dxva2_get_surface_index(avctx, ctx, r->f);
  228. else
  229. index = get_refpic_index(pp, ff_dxva2_get_surface_index(avctx, ctx, r->f));
  230. fill_picture_entry(&slice->RefPicList[list][i], index,
  231. r->reference == PICT_BOTTOM_FIELD);
  232. for (plane = 0; plane < 3; plane++) {
  233. int w, o;
  234. if (plane == 0 && sl->luma_weight_flag[list]) {
  235. w = sl->luma_weight[i][list][0];
  236. o = sl->luma_weight[i][list][1];
  237. } else if (plane >= 1 && sl->chroma_weight_flag[list]) {
  238. w = sl->chroma_weight[i][list][plane-1][0];
  239. o = sl->chroma_weight[i][list][plane-1][1];
  240. } else {
  241. w = 1 << (plane == 0 ? sl->luma_log2_weight_denom :
  242. sl->chroma_log2_weight_denom);
  243. o = 0;
  244. }
  245. slice->Weights[list][i][plane][0] = w;
  246. slice->Weights[list][i][plane][1] = o;
  247. }
  248. } else {
  249. unsigned plane;
  250. slice->RefPicList[list][i].bPicEntry = 0xff;
  251. for (plane = 0; plane < 3; plane++) {
  252. slice->Weights[list][i][plane][0] = 0;
  253. slice->Weights[list][i][plane][1] = 0;
  254. }
  255. }
  256. }
  257. }
  258. slice->slice_qs_delta = 0; /* XXX not implemented by Libav */
  259. slice->slice_qp_delta = sl->qscale - h->pps.init_qp;
  260. slice->redundant_pic_cnt = sl->redundant_pic_count;
  261. if (sl->slice_type == AV_PICTURE_TYPE_B)
  262. slice->direct_spatial_mv_pred_flag = sl->direct_spatial_mv_pred;
  263. slice->cabac_init_idc = h->pps.cabac ? sl->cabac_init_idc : 0;
  264. if (sl->deblocking_filter < 2)
  265. slice->disable_deblocking_filter_idc = 1 - sl->deblocking_filter;
  266. else
  267. slice->disable_deblocking_filter_idc = sl->deblocking_filter;
  268. slice->slice_id = h->current_slice - 1;
  269. }
  270. static int commit_bitstream_and_slice_buffer(AVCodecContext *avctx,
  271. DECODER_BUFFER_DESC *bs,
  272. DECODER_BUFFER_DESC *sc)
  273. {
  274. const H264Context *h = avctx->priv_data;
  275. const unsigned mb_count = h->mb_width * h->mb_height;
  276. AVDXVAContext *ctx = avctx->hwaccel_context;
  277. const H264Picture *current_picture = h->cur_pic_ptr;
  278. struct dxva2_picture_context *ctx_pic = current_picture->hwaccel_picture_private;
  279. DXVA_Slice_H264_Short *slice = NULL;
  280. void *dxva_data_ptr;
  281. uint8_t *dxva_data, *current, *end;
  282. unsigned dxva_size;
  283. void *slice_data;
  284. unsigned slice_size;
  285. unsigned padding;
  286. unsigned i;
  287. unsigned type;
  288. /* Create an annex B bitstream buffer with only slice NAL and finalize slice */
  289. #if CONFIG_D3D11VA
  290. if (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD) {
  291. type = D3D11_VIDEO_DECODER_BUFFER_BITSTREAM;
  292. if (FAILED(ID3D11VideoContext_GetDecoderBuffer(D3D11VA_CONTEXT(ctx)->video_context,
  293. D3D11VA_CONTEXT(ctx)->decoder,
  294. type,
  295. &dxva_size, &dxva_data_ptr)))
  296. return -1;
  297. }
  298. #endif
  299. #if CONFIG_DXVA2
  300. if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD) {
  301. type = DXVA2_BitStreamDateBufferType;
  302. if (FAILED(IDirectXVideoDecoder_GetBuffer(DXVA2_CONTEXT(ctx)->decoder,
  303. type,
  304. &dxva_data_ptr, &dxva_size)))
  305. return -1;
  306. }
  307. #endif
  308. dxva_data = dxva_data_ptr;
  309. current = dxva_data;
  310. end = dxva_data + dxva_size;
  311. for (i = 0; i < ctx_pic->slice_count; i++) {
  312. static const uint8_t start_code[] = { 0, 0, 1 };
  313. static const unsigned start_code_size = sizeof(start_code);
  314. unsigned position, size;
  315. assert(offsetof(DXVA_Slice_H264_Short, BSNALunitDataLocation) ==
  316. offsetof(DXVA_Slice_H264_Long, BSNALunitDataLocation));
  317. assert(offsetof(DXVA_Slice_H264_Short, SliceBytesInBuffer) ==
  318. offsetof(DXVA_Slice_H264_Long, SliceBytesInBuffer));
  319. if (is_slice_short(avctx, ctx))
  320. slice = &ctx_pic->slice_short[i];
  321. else
  322. slice = (DXVA_Slice_H264_Short*)&ctx_pic->slice_long[i];
  323. position = slice->BSNALunitDataLocation;
  324. size = slice->SliceBytesInBuffer;
  325. if (start_code_size + size > end - current) {
  326. av_log(avctx, AV_LOG_ERROR, "Failed to build bitstream");
  327. break;
  328. }
  329. slice->BSNALunitDataLocation = current - dxva_data;
  330. slice->SliceBytesInBuffer = start_code_size + size;
  331. if (!is_slice_short(avctx, ctx)) {
  332. DXVA_Slice_H264_Long *slice_long = (DXVA_Slice_H264_Long*)slice;
  333. if (i < ctx_pic->slice_count - 1)
  334. slice_long->NumMbsForSlice =
  335. slice_long[1].first_mb_in_slice - slice_long[0].first_mb_in_slice;
  336. else
  337. slice_long->NumMbsForSlice = mb_count - slice_long->first_mb_in_slice;
  338. }
  339. memcpy(current, start_code, start_code_size);
  340. current += start_code_size;
  341. memcpy(current, &ctx_pic->bitstream[position], size);
  342. current += size;
  343. }
  344. padding = FFMIN(128 - ((current - dxva_data) & 127), end - current);
  345. if (slice && padding > 0) {
  346. memset(current, 0, padding);
  347. current += padding;
  348. slice->SliceBytesInBuffer += padding;
  349. }
  350. #if CONFIG_D3D11VA
  351. if (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD)
  352. if (FAILED(ID3D11VideoContext_ReleaseDecoderBuffer(D3D11VA_CONTEXT(ctx)->video_context, D3D11VA_CONTEXT(ctx)->decoder, type)))
  353. return -1;
  354. #endif
  355. #if CONFIG_DXVA2
  356. if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD)
  357. if (FAILED(IDirectXVideoDecoder_ReleaseBuffer(DXVA2_CONTEXT(ctx)->decoder, type)))
  358. return -1;
  359. #endif
  360. if (i < ctx_pic->slice_count)
  361. return -1;
  362. #if CONFIG_D3D11VA
  363. if (avctx->pix_fmt == AV_PIX_FMT_D3D11VA_VLD) {
  364. D3D11_VIDEO_DECODER_BUFFER_DESC *dsc11 = bs;
  365. memset(dsc11, 0, sizeof(*dsc11));
  366. dsc11->BufferType = type;
  367. dsc11->DataSize = current - dxva_data;
  368. dsc11->NumMBsInBuffer = mb_count;
  369. type = D3D11_VIDEO_DECODER_BUFFER_SLICE_CONTROL;
  370. }
  371. #endif
  372. #if CONFIG_DXVA2
  373. if (avctx->pix_fmt == AV_PIX_FMT_DXVA2_VLD) {
  374. DXVA2_DecodeBufferDesc *dsc2 = bs;
  375. memset(dsc2, 0, sizeof(*dsc2));
  376. dsc2->CompressedBufferType = type;
  377. dsc2->DataSize = current - dxva_data;
  378. dsc2->NumMBsInBuffer = mb_count;
  379. type = DXVA2_SliceControlBufferType;
  380. }
  381. #endif
  382. if (is_slice_short(avctx, ctx)) {
  383. slice_data = ctx_pic->slice_short;
  384. slice_size = ctx_pic->slice_count * sizeof(*ctx_pic->slice_short);
  385. } else {
  386. slice_data = ctx_pic->slice_long;
  387. slice_size = ctx_pic->slice_count * sizeof(*ctx_pic->slice_long);
  388. }
  389. assert((bs->DataSize & 127) == 0);
  390. return ff_dxva2_commit_buffer(avctx, ctx, sc,
  391. type,
  392. slice_data, slice_size, mb_count);
  393. }
  394. static int dxva2_h264_start_frame(AVCodecContext *avctx,
  395. av_unused const uint8_t *buffer,
  396. av_unused uint32_t size)
  397. {
  398. const H264Context *h = avctx->priv_data;
  399. AVDXVAContext *ctx = avctx->hwaccel_context;
  400. struct dxva2_picture_context *ctx_pic = h->cur_pic_ptr->hwaccel_picture_private;
  401. if (DXVA_CONTEXT_DECODER(avctx, ctx) == NULL ||
  402. DXVA_CONTEXT_CFG(avctx, ctx) == NULL ||
  403. DXVA_CONTEXT_COUNT(avctx, ctx) <= 0)
  404. return -1;
  405. assert(ctx_pic);
  406. /* Fill up DXVA_PicParams_H264 */
  407. fill_picture_parameters(avctx, ctx, h, &ctx_pic->pp);
  408. /* Fill up DXVA_Qmatrix_H264 */
  409. fill_scaling_lists(avctx, ctx, h, &ctx_pic->qm);
  410. ctx_pic->slice_count = 0;
  411. ctx_pic->bitstream_size = 0;
  412. ctx_pic->bitstream = NULL;
  413. return 0;
  414. }
  415. static int dxva2_h264_decode_slice(AVCodecContext *avctx,
  416. const uint8_t *buffer,
  417. uint32_t size)
  418. {
  419. const H264Context *h = avctx->priv_data;
  420. const H264SliceContext *sl = &h->slice_ctx[0];
  421. AVDXVAContext *ctx = avctx->hwaccel_context;
  422. const H264Picture *current_picture = h->cur_pic_ptr;
  423. struct dxva2_picture_context *ctx_pic = current_picture->hwaccel_picture_private;
  424. unsigned position;
  425. if (ctx_pic->slice_count >= MAX_SLICES)
  426. return -1;
  427. if (!ctx_pic->bitstream)
  428. ctx_pic->bitstream = buffer;
  429. ctx_pic->bitstream_size += size;
  430. position = buffer - ctx_pic->bitstream;
  431. if (is_slice_short(avctx, ctx))
  432. fill_slice_short(&ctx_pic->slice_short[ctx_pic->slice_count],
  433. position, size);
  434. else
  435. fill_slice_long(avctx, &ctx_pic->slice_long[ctx_pic->slice_count],
  436. &ctx_pic->pp, position, size);
  437. ctx_pic->slice_count++;
  438. if (sl->slice_type != AV_PICTURE_TYPE_I && sl->slice_type != AV_PICTURE_TYPE_SI)
  439. ctx_pic->pp.wBitFields &= ~(1 << 15); /* Set IntraPicFlag to 0 */
  440. return 0;
  441. }
  442. static int dxva2_h264_end_frame(AVCodecContext *avctx)
  443. {
  444. H264Context *h = avctx->priv_data;
  445. H264SliceContext *sl = &h->slice_ctx[0];
  446. struct dxva2_picture_context *ctx_pic =
  447. h->cur_pic_ptr->hwaccel_picture_private;
  448. int ret;
  449. if (ctx_pic->slice_count <= 0 || ctx_pic->bitstream_size <= 0)
  450. return -1;
  451. ret = ff_dxva2_common_end_frame(avctx, h->cur_pic_ptr->f,
  452. &ctx_pic->pp, sizeof(ctx_pic->pp),
  453. &ctx_pic->qm, sizeof(ctx_pic->qm),
  454. commit_bitstream_and_slice_buffer);
  455. if (!ret)
  456. ff_h264_draw_horiz_band(h, sl, 0, h->avctx->height);
  457. return ret;
  458. }
  459. #if CONFIG_H264_DXVA2_HWACCEL
  460. AVHWAccel ff_h264_dxva2_hwaccel = {
  461. .name = "h264_dxva2",
  462. .type = AVMEDIA_TYPE_VIDEO,
  463. .id = AV_CODEC_ID_H264,
  464. .pix_fmt = AV_PIX_FMT_DXVA2_VLD,
  465. .start_frame = dxva2_h264_start_frame,
  466. .decode_slice = dxva2_h264_decode_slice,
  467. .end_frame = dxva2_h264_end_frame,
  468. .frame_priv_data_size = sizeof(struct dxva2_picture_context),
  469. };
  470. #endif
  471. #if CONFIG_H264_D3D11VA_HWACCEL
  472. AVHWAccel ff_h264_d3d11va_hwaccel = {
  473. .name = "h264_d3d11va",
  474. .type = AVMEDIA_TYPE_VIDEO,
  475. .id = AV_CODEC_ID_H264,
  476. .pix_fmt = AV_PIX_FMT_D3D11VA_VLD,
  477. .start_frame = dxva2_h264_start_frame,
  478. .decode_slice = dxva2_h264_decode_slice,
  479. .end_frame = dxva2_h264_end_frame,
  480. .frame_priv_data_size = sizeof(struct dxva2_picture_context),
  481. };
  482. #endif