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.

1161 lines
39KB

  1. /*
  2. * Intel MediaSDK QSV encoder utility functions
  3. *
  4. * copyright (c) 2013 Yukinori Yamazoe
  5. * copyright (c) 2015 Anton Khirnov
  6. *
  7. * This file is part of Libav.
  8. *
  9. * Libav is free software; you can redistribute it and/or
  10. * modify it under the terms of the GNU Lesser General Public
  11. * License as published by the Free Software Foundation; either
  12. * version 2.1 of the License, or (at your option) any later version.
  13. *
  14. * Libav is distributed in the hope that it will be useful,
  15. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  17. * Lesser General Public License for more details.
  18. *
  19. * You should have received a copy of the GNU Lesser General Public
  20. * License along with Libav; if not, write to the Free Software
  21. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  22. */
  23. #include <string.h>
  24. #include <sys/types.h>
  25. #include <mfx/mfxvideo.h>
  26. #include "libavutil/common.h"
  27. #include "libavutil/hwcontext.h"
  28. #include "libavutil/hwcontext_qsv.h"
  29. #include "libavutil/mem.h"
  30. #include "libavutil/log.h"
  31. #include "libavutil/time.h"
  32. #include "libavutil/imgutils.h"
  33. #include "avcodec.h"
  34. #include "internal.h"
  35. #include "qsv.h"
  36. #include "qsv_internal.h"
  37. #include "qsvenc.h"
  38. static const struct {
  39. mfxU16 profile;
  40. const char *name;
  41. } profile_names[] = {
  42. { MFX_PROFILE_AVC_BASELINE, "baseline" },
  43. { MFX_PROFILE_AVC_MAIN, "main" },
  44. { MFX_PROFILE_AVC_EXTENDED, "extended" },
  45. { MFX_PROFILE_AVC_HIGH, "high" },
  46. #if QSV_VERSION_ATLEAST(1, 15)
  47. { MFX_PROFILE_AVC_HIGH_422, "high 422" },
  48. #endif
  49. #if QSV_VERSION_ATLEAST(1, 4)
  50. { MFX_PROFILE_AVC_CONSTRAINED_BASELINE, "constrained baseline" },
  51. { MFX_PROFILE_AVC_CONSTRAINED_HIGH, "constrained high" },
  52. { MFX_PROFILE_AVC_PROGRESSIVE_HIGH, "progressive high" },
  53. #endif
  54. { MFX_PROFILE_MPEG2_SIMPLE, "simple" },
  55. { MFX_PROFILE_MPEG2_MAIN, "main" },
  56. { MFX_PROFILE_MPEG2_HIGH, "high" },
  57. { MFX_PROFILE_VC1_SIMPLE, "simple" },
  58. { MFX_PROFILE_VC1_MAIN, "main" },
  59. { MFX_PROFILE_VC1_ADVANCED, "advanced" },
  60. #if QSV_VERSION_ATLEAST(1, 8)
  61. { MFX_PROFILE_HEVC_MAIN, "main" },
  62. { MFX_PROFILE_HEVC_MAIN10, "main10" },
  63. { MFX_PROFILE_HEVC_MAINSP, "mainsp" },
  64. #endif
  65. };
  66. static const char *print_profile(mfxU16 profile)
  67. {
  68. int i;
  69. for (i = 0; i < FF_ARRAY_ELEMS(profile_names); i++)
  70. if (profile == profile_names[i].profile)
  71. return profile_names[i].name;
  72. return "unknown";
  73. }
  74. static const struct {
  75. mfxU16 rc_mode;
  76. const char *name;
  77. } rc_names[] = {
  78. { MFX_RATECONTROL_CBR, "CBR" },
  79. { MFX_RATECONTROL_VBR, "VBR" },
  80. { MFX_RATECONTROL_CQP, "CQP" },
  81. { MFX_RATECONTROL_AVBR, "AVBR" },
  82. #if QSV_HAVE_LA
  83. { MFX_RATECONTROL_LA, "LA" },
  84. #endif
  85. #if QSV_HAVE_ICQ
  86. { MFX_RATECONTROL_ICQ, "ICQ" },
  87. { MFX_RATECONTROL_LA_ICQ, "LA_ICQ" },
  88. #endif
  89. #if QSV_HAVE_VCM
  90. { MFX_RATECONTROL_VCM, "VCM" },
  91. #endif
  92. #if QSV_VERSION_ATLEAST(1, 10)
  93. { MFX_RATECONTROL_LA_EXT, "LA_EXT" },
  94. #endif
  95. #if QSV_HAVE_LA_HRD
  96. { MFX_RATECONTROL_LA_HRD, "LA_HRD" },
  97. #endif
  98. #if QSV_HAVE_QVBR
  99. { MFX_RATECONTROL_QVBR, "QVBR" },
  100. #endif
  101. };
  102. static const char *print_ratecontrol(mfxU16 rc_mode)
  103. {
  104. int i;
  105. for (i = 0; i < FF_ARRAY_ELEMS(rc_names); i++)
  106. if (rc_mode == rc_names[i].rc_mode)
  107. return rc_names[i].name;
  108. return "unknown";
  109. }
  110. static const char *print_threestate(mfxU16 val)
  111. {
  112. if (val == MFX_CODINGOPTION_ON)
  113. return "ON";
  114. else if (val == MFX_CODINGOPTION_OFF)
  115. return "OFF";
  116. return "unknown";
  117. }
  118. static void dump_video_param(AVCodecContext *avctx, QSVEncContext *q,
  119. mfxExtBuffer **coding_opts)
  120. {
  121. mfxInfoMFX *info = &q->param.mfx;
  122. mfxExtCodingOption *co = (mfxExtCodingOption*)coding_opts[0];
  123. #if QSV_HAVE_CO2
  124. mfxExtCodingOption2 *co2 = (mfxExtCodingOption2*)coding_opts[1];
  125. #endif
  126. #if QSV_HAVE_CO3
  127. mfxExtCodingOption3 *co3 = (mfxExtCodingOption3*)coding_opts[2];
  128. #endif
  129. av_log(avctx, AV_LOG_VERBOSE, "profile: %s; level: %"PRIu16"\n",
  130. print_profile(info->CodecProfile), info->CodecLevel);
  131. av_log(avctx, AV_LOG_VERBOSE, "GopPicSize: %"PRIu16"; GopRefDist: %"PRIu16"; GopOptFlag: ",
  132. info->GopPicSize, info->GopRefDist);
  133. if (info->GopOptFlag & MFX_GOP_CLOSED)
  134. av_log(avctx, AV_LOG_VERBOSE, "closed ");
  135. if (info->GopOptFlag & MFX_GOP_STRICT)
  136. av_log(avctx, AV_LOG_VERBOSE, "strict ");
  137. av_log(avctx, AV_LOG_VERBOSE, "; IdrInterval: %"PRIu16"\n", info->IdrInterval);
  138. av_log(avctx, AV_LOG_VERBOSE, "TargetUsage: %"PRIu16"; RateControlMethod: %s\n",
  139. info->TargetUsage, print_ratecontrol(info->RateControlMethod));
  140. if (info->RateControlMethod == MFX_RATECONTROL_CBR ||
  141. info->RateControlMethod == MFX_RATECONTROL_VBR
  142. #if QSV_HAVE_VCM
  143. || info->RateControlMethod == MFX_RATECONTROL_VCM
  144. #endif
  145. ) {
  146. av_log(avctx, AV_LOG_VERBOSE,
  147. "InitialDelayInKB: %"PRIu16"; TargetKbps: %"PRIu16"; MaxKbps: %"PRIu16"\n",
  148. info->InitialDelayInKB, info->TargetKbps, info->MaxKbps);
  149. } else if (info->RateControlMethod == MFX_RATECONTROL_CQP) {
  150. av_log(avctx, AV_LOG_VERBOSE, "QPI: %"PRIu16"; QPP: %"PRIu16"; QPB: %"PRIu16"\n",
  151. info->QPI, info->QPP, info->QPB);
  152. } else if (info->RateControlMethod == MFX_RATECONTROL_AVBR) {
  153. av_log(avctx, AV_LOG_VERBOSE,
  154. "TargetKbps: %"PRIu16"; Accuracy: %"PRIu16"; Convergence: %"PRIu16"\n",
  155. info->TargetKbps, info->Accuracy, info->Convergence);
  156. }
  157. #if QSV_HAVE_LA
  158. else if (info->RateControlMethod == MFX_RATECONTROL_LA
  159. #if QSV_HAVE_LA_HRD
  160. || info->RateControlMethod == MFX_RATECONTROL_LA_HRD
  161. #endif
  162. ) {
  163. av_log(avctx, AV_LOG_VERBOSE,
  164. "TargetKbps: %"PRIu16"; LookAheadDepth: %"PRIu16"\n",
  165. info->TargetKbps, co2->LookAheadDepth);
  166. }
  167. #endif
  168. #if QSV_HAVE_ICQ
  169. else if (info->RateControlMethod == MFX_RATECONTROL_ICQ) {
  170. av_log(avctx, AV_LOG_VERBOSE, "ICQQuality: %"PRIu16"\n", info->ICQQuality);
  171. } else if (info->RateControlMethod == MFX_RATECONTROL_LA_ICQ) {
  172. av_log(avctx, AV_LOG_VERBOSE, "ICQQuality: %"PRIu16"; LookAheadDepth: %"PRIu16"\n",
  173. info->ICQQuality, co2->LookAheadDepth);
  174. }
  175. #endif
  176. #if QSV_HAVE_QVBR
  177. else if (info->RateControlMethod == MFX_RATECONTROL_QVBR) {
  178. av_log(avctx, AV_LOG_VERBOSE, "QVBRQuality: %"PRIu16"\n",
  179. co3->QVBRQuality);
  180. }
  181. #endif
  182. av_log(avctx, AV_LOG_VERBOSE, "NumSlice: %"PRIu16"; NumRefFrame: %"PRIu16"\n",
  183. info->NumSlice, info->NumRefFrame);
  184. av_log(avctx, AV_LOG_VERBOSE, "RateDistortionOpt: %s\n",
  185. print_threestate(co->RateDistortionOpt));
  186. #if QSV_HAVE_CO2
  187. av_log(avctx, AV_LOG_VERBOSE,
  188. "RecoveryPointSEI: %s IntRefType: %"PRIu16"; IntRefCycleSize: %"PRIu16"; IntRefQPDelta: %"PRId16"\n",
  189. print_threestate(co->RecoveryPointSEI), co2->IntRefType, co2->IntRefCycleSize, co2->IntRefQPDelta);
  190. av_log(avctx, AV_LOG_VERBOSE, "MaxFrameSize: %"PRIu16"; ", co2->MaxFrameSize);
  191. #if QSV_HAVE_MAX_SLICE_SIZE
  192. av_log(avctx, AV_LOG_VERBOSE, "MaxSliceSize: %"PRIu16"; ", co2->MaxSliceSize);
  193. #endif
  194. av_log(avctx, AV_LOG_VERBOSE, "\n");
  195. av_log(avctx, AV_LOG_VERBOSE,
  196. "BitrateLimit: %s; MBBRC: %s; ExtBRC: %s\n",
  197. print_threestate(co2->BitrateLimit), print_threestate(co2->MBBRC),
  198. print_threestate(co2->ExtBRC));
  199. #if QSV_HAVE_TRELLIS
  200. av_log(avctx, AV_LOG_VERBOSE, "Trellis: ");
  201. if (co2->Trellis & MFX_TRELLIS_OFF) {
  202. av_log(avctx, AV_LOG_VERBOSE, "off");
  203. } else if (!co2->Trellis) {
  204. av_log(avctx, AV_LOG_VERBOSE, "auto");
  205. } else {
  206. if (co2->Trellis & MFX_TRELLIS_I) av_log(avctx, AV_LOG_VERBOSE, "I");
  207. if (co2->Trellis & MFX_TRELLIS_P) av_log(avctx, AV_LOG_VERBOSE, "P");
  208. if (co2->Trellis & MFX_TRELLIS_B) av_log(avctx, AV_LOG_VERBOSE, "B");
  209. }
  210. av_log(avctx, AV_LOG_VERBOSE, "\n");
  211. #endif
  212. #if QSV_VERSION_ATLEAST(1, 8)
  213. av_log(avctx, AV_LOG_VERBOSE,
  214. "RepeatPPS: %s; NumMbPerSlice: %"PRIu16"; LookAheadDS: ",
  215. print_threestate(co2->RepeatPPS), co2->NumMbPerSlice);
  216. switch (co2->LookAheadDS) {
  217. case MFX_LOOKAHEAD_DS_OFF: av_log(avctx, AV_LOG_VERBOSE, "off"); break;
  218. case MFX_LOOKAHEAD_DS_2x: av_log(avctx, AV_LOG_VERBOSE, "2x"); break;
  219. case MFX_LOOKAHEAD_DS_4x: av_log(avctx, AV_LOG_VERBOSE, "4x"); break;
  220. default: av_log(avctx, AV_LOG_VERBOSE, "unknown"); break;
  221. }
  222. av_log(avctx, AV_LOG_VERBOSE, "\n");
  223. av_log(avctx, AV_LOG_VERBOSE, "AdaptiveI: %s; AdaptiveB: %s; BRefType: ",
  224. print_threestate(co2->AdaptiveI), print_threestate(co2->AdaptiveB));
  225. switch (co2->BRefType) {
  226. case MFX_B_REF_OFF: av_log(avctx, AV_LOG_VERBOSE, "off"); break;
  227. case MFX_B_REF_PYRAMID: av_log(avctx, AV_LOG_VERBOSE, "pyramid"); break;
  228. default: av_log(avctx, AV_LOG_VERBOSE, "auto"); break;
  229. }
  230. av_log(avctx, AV_LOG_VERBOSE, "\n");
  231. #endif
  232. #if QSV_VERSION_ATLEAST(1, 9)
  233. av_log(avctx, AV_LOG_VERBOSE,
  234. "MinQPI: %"PRIu8"; MaxQPI: %"PRIu8"; MinQPP: %"PRIu8"; MaxQPP: %"PRIu8"; MinQPB: %"PRIu8"; MaxQPB: %"PRIu8"\n",
  235. co2->MinQPI, co2->MaxQPI, co2->MinQPP, co2->MaxQPP, co2->MinQPB, co2->MaxQPB);
  236. #endif
  237. #endif
  238. if (avctx->codec_id == AV_CODEC_ID_H264) {
  239. av_log(avctx, AV_LOG_VERBOSE, "Entropy coding: %s; MaxDecFrameBuffering: %"PRIu16"\n",
  240. co->CAVLC == MFX_CODINGOPTION_ON ? "CAVLC" : "CABAC", co->MaxDecFrameBuffering);
  241. av_log(avctx, AV_LOG_VERBOSE,
  242. "NalHrdConformance: %s; SingleSeiNalUnit: %s; VuiVclHrdParameters: %s VuiNalHrdParameters: %s\n",
  243. print_threestate(co->NalHrdConformance), print_threestate(co->SingleSeiNalUnit),
  244. print_threestate(co->VuiVclHrdParameters), print_threestate(co->VuiNalHrdParameters));
  245. }
  246. }
  247. static int select_rc_mode(AVCodecContext *avctx, QSVEncContext *q)
  248. {
  249. const char *rc_desc;
  250. mfxU16 rc_mode;
  251. int want_la = q->la_depth >= 10;
  252. int want_qscale = !!(avctx->flags & AV_CODEC_FLAG_QSCALE);
  253. int want_vcm = q->vcm;
  254. if (want_la && !QSV_HAVE_LA) {
  255. av_log(avctx, AV_LOG_ERROR,
  256. "Lookahead ratecontrol mode requested, but is not supported by this SDK version\n");
  257. return AVERROR(ENOSYS);
  258. }
  259. if (want_vcm && !QSV_HAVE_VCM) {
  260. av_log(avctx, AV_LOG_ERROR,
  261. "VCM ratecontrol mode requested, but is not supported by this SDK version\n");
  262. return AVERROR(ENOSYS);
  263. }
  264. if (want_la + want_qscale + want_vcm > 1) {
  265. av_log(avctx, AV_LOG_ERROR,
  266. "More than one of: { constant qscale, lookahead, VCM } requested, "
  267. "only one of them can be used at a time.\n");
  268. return AVERROR(EINVAL);
  269. }
  270. if (want_qscale) {
  271. rc_mode = MFX_RATECONTROL_CQP;
  272. rc_desc = "constant quantization parameter (CQP)";
  273. }
  274. #if QSV_HAVE_VCM
  275. else if (want_vcm) {
  276. rc_mode = MFX_RATECONTROL_VCM;
  277. rc_desc = "video conferencing mode (VCM)";
  278. }
  279. #endif
  280. #if QSV_HAVE_LA
  281. else if (want_la) {
  282. rc_mode = MFX_RATECONTROL_LA;
  283. rc_desc = "VBR with lookahead (LA)";
  284. #if QSV_HAVE_ICQ
  285. if (avctx->global_quality > 0) {
  286. rc_mode = MFX_RATECONTROL_LA_ICQ;
  287. rc_desc = "intelligent constant quality with lookahead (LA_ICQ)";
  288. }
  289. #endif
  290. }
  291. #endif
  292. #if QSV_HAVE_ICQ
  293. else if (avctx->global_quality > 0) {
  294. rc_mode = MFX_RATECONTROL_ICQ;
  295. rc_desc = "intelligent constant quality (ICQ)";
  296. }
  297. #endif
  298. else if (avctx->rc_max_rate == avctx->bit_rate) {
  299. rc_mode = MFX_RATECONTROL_CBR;
  300. rc_desc = "constant bitrate (CBR)";
  301. } else if (!avctx->rc_max_rate) {
  302. rc_mode = MFX_RATECONTROL_AVBR;
  303. rc_desc = "average variable bitrate (AVBR)";
  304. } else {
  305. rc_mode = MFX_RATECONTROL_VBR;
  306. rc_desc = "variable bitrate (VBR)";
  307. }
  308. q->param.mfx.RateControlMethod = rc_mode;
  309. av_log(avctx, AV_LOG_VERBOSE, "Using the %s ratecontrol method\n", rc_desc);
  310. return 0;
  311. }
  312. static int rc_supported(QSVEncContext *q)
  313. {
  314. mfxVideoParam param_out = { .mfx.CodecId = q->param.mfx.CodecId };
  315. mfxStatus ret;
  316. ret = MFXVideoENCODE_Query(q->session, &q->param, &param_out);
  317. if (ret < 0 ||
  318. param_out.mfx.RateControlMethod != q->param.mfx.RateControlMethod)
  319. return 0;
  320. return 1;
  321. }
  322. static int init_video_param(AVCodecContext *avctx, QSVEncContext *q)
  323. {
  324. enum AVPixelFormat sw_format = avctx->pix_fmt == AV_PIX_FMT_QSV ?
  325. avctx->sw_pix_fmt : avctx->pix_fmt;
  326. const AVPixFmtDescriptor *desc;
  327. float quant;
  328. int ret;
  329. ret = ff_qsv_codec_id_to_mfx(avctx->codec_id);
  330. if (ret < 0)
  331. return AVERROR_BUG;
  332. q->param.mfx.CodecId = ret;
  333. if (avctx->level > 0)
  334. q->param.mfx.CodecLevel = avctx->level;
  335. q->param.mfx.CodecProfile = q->profile;
  336. q->param.mfx.TargetUsage = q->preset;
  337. q->param.mfx.GopPicSize = FFMAX(0, avctx->gop_size);
  338. q->param.mfx.GopRefDist = FFMAX(-1, avctx->max_b_frames) + 1;
  339. q->param.mfx.GopOptFlag = avctx->flags & AV_CODEC_FLAG_CLOSED_GOP ?
  340. MFX_GOP_CLOSED : 0;
  341. q->param.mfx.IdrInterval = q->idr_interval;
  342. q->param.mfx.NumSlice = avctx->slices;
  343. q->param.mfx.NumRefFrame = FFMAX(0, avctx->refs);
  344. q->param.mfx.EncodedOrder = 0;
  345. q->param.mfx.BufferSizeInKB = 0;
  346. desc = av_pix_fmt_desc_get(sw_format);
  347. if (!desc)
  348. return AVERROR_BUG;
  349. ff_qsv_map_pixfmt(sw_format, &q->param.mfx.FrameInfo.FourCC);
  350. q->param.mfx.FrameInfo.CropX = 0;
  351. q->param.mfx.FrameInfo.CropY = 0;
  352. q->param.mfx.FrameInfo.CropW = avctx->width;
  353. q->param.mfx.FrameInfo.CropH = avctx->height;
  354. q->param.mfx.FrameInfo.AspectRatioW = avctx->sample_aspect_ratio.num;
  355. q->param.mfx.FrameInfo.AspectRatioH = avctx->sample_aspect_ratio.den;
  356. q->param.mfx.FrameInfo.ChromaFormat = MFX_CHROMAFORMAT_YUV420;
  357. q->param.mfx.FrameInfo.BitDepthLuma = desc->comp[0].depth;
  358. q->param.mfx.FrameInfo.BitDepthChroma = desc->comp[0].depth;
  359. q->param.mfx.FrameInfo.Shift = desc->comp[0].depth > 8;
  360. // TODO: detect version of MFX--if the minor version is greater than
  361. // or equal to 19, then can use the same alignment settings as H.264
  362. // for HEVC
  363. q->width_align = avctx->codec_id == AV_CODEC_ID_HEVC ? 32 : 16;
  364. q->param.mfx.FrameInfo.Width = FFALIGN(avctx->width, q->width_align);
  365. if (avctx->flags & AV_CODEC_FLAG_INTERLACED_DCT) {
  366. // it is important that PicStruct be setup correctly from the
  367. // start--otherwise, encoding doesn't work and results in a bunch
  368. // of incompatible video parameter errors
  369. q->param.mfx.FrameInfo.PicStruct = MFX_PICSTRUCT_FIELD_TFF;
  370. // height alignment always must be 32 for interlaced video
  371. q->height_align = 32;
  372. } else {
  373. q->param.mfx.FrameInfo.PicStruct = MFX_PICSTRUCT_PROGRESSIVE;
  374. // for progressive video, the height should be aligned to 16 for
  375. // H.264. For HEVC, depending on the version of MFX, it should be
  376. // either 32 or 16. The lower number is better if possible.
  377. q->height_align = avctx->codec_id == AV_CODEC_ID_HEVC ? 32 : 16;
  378. }
  379. q->param.mfx.FrameInfo.Height = FFALIGN(avctx->height, q->height_align);
  380. if (avctx->hw_frames_ctx) {
  381. AVHWFramesContext *frames_ctx = (AVHWFramesContext*)avctx->hw_frames_ctx->data;
  382. AVQSVFramesContext *frames_hwctx = frames_ctx->hwctx;
  383. q->param.mfx.FrameInfo.Width = frames_hwctx->surfaces[0].Info.Width;
  384. q->param.mfx.FrameInfo.Height = frames_hwctx->surfaces[0].Info.Height;
  385. }
  386. if (avctx->framerate.den > 0 && avctx->framerate.num > 0) {
  387. q->param.mfx.FrameInfo.FrameRateExtN = avctx->framerate.num;
  388. q->param.mfx.FrameInfo.FrameRateExtD = avctx->framerate.den;
  389. } else {
  390. q->param.mfx.FrameInfo.FrameRateExtN = avctx->time_base.den;
  391. q->param.mfx.FrameInfo.FrameRateExtD = avctx->time_base.num;
  392. }
  393. ret = select_rc_mode(avctx, q);
  394. if (ret < 0)
  395. return ret;
  396. switch (q->param.mfx.RateControlMethod) {
  397. case MFX_RATECONTROL_CBR:
  398. case MFX_RATECONTROL_VBR:
  399. #if QSV_HAVE_VCM
  400. case MFX_RATECONTROL_VCM:
  401. #endif
  402. q->param.mfx.InitialDelayInKB = avctx->rc_initial_buffer_occupancy / 1000;
  403. q->param.mfx.TargetKbps = avctx->bit_rate / 1000;
  404. q->param.mfx.MaxKbps = avctx->rc_max_rate / 1000;
  405. break;
  406. case MFX_RATECONTROL_CQP:
  407. quant = avctx->global_quality / FF_QP2LAMBDA;
  408. q->param.mfx.QPI = av_clip(quant * fabs(avctx->i_quant_factor) + avctx->i_quant_offset, 0, 51);
  409. q->param.mfx.QPP = av_clip(quant, 0, 51);
  410. q->param.mfx.QPB = av_clip(quant * fabs(avctx->b_quant_factor) + avctx->b_quant_offset, 0, 51);
  411. break;
  412. case MFX_RATECONTROL_AVBR:
  413. q->param.mfx.TargetKbps = avctx->bit_rate / 1000;
  414. q->param.mfx.Convergence = q->avbr_convergence;
  415. q->param.mfx.Accuracy = q->avbr_accuracy;
  416. break;
  417. #if QSV_HAVE_LA
  418. case MFX_RATECONTROL_LA:
  419. q->param.mfx.TargetKbps = avctx->bit_rate / 1000;
  420. q->extco2.LookAheadDepth = q->la_depth;
  421. break;
  422. #if QSV_HAVE_ICQ
  423. case MFX_RATECONTROL_LA_ICQ:
  424. q->extco2.LookAheadDepth = q->la_depth;
  425. case MFX_RATECONTROL_ICQ:
  426. q->param.mfx.ICQQuality = avctx->global_quality;
  427. break;
  428. #endif
  429. #endif
  430. }
  431. // the HEVC encoder plugin currently fails if coding options
  432. // are provided
  433. if (avctx->codec_id != AV_CODEC_ID_HEVC) {
  434. q->extco.Header.BufferId = MFX_EXTBUFF_CODING_OPTION;
  435. q->extco.Header.BufferSz = sizeof(q->extco);
  436. #if FF_API_CODER_TYPE
  437. FF_DISABLE_DEPRECATION_WARNINGS
  438. if (avctx->coder_type != 0)
  439. q->cavlc = avctx->coder_type == FF_CODER_TYPE_VLC;
  440. FF_ENABLE_DEPRECATION_WARNINGS
  441. #endif
  442. q->extco.CAVLC = q->cavlc ? MFX_CODINGOPTION_ON
  443. : MFX_CODINGOPTION_UNKNOWN;
  444. if (q->rdo >= 0)
  445. q->extco.RateDistortionOpt = q->rdo > 0 ? MFX_CODINGOPTION_ON : MFX_CODINGOPTION_OFF;
  446. if (avctx->codec_id == AV_CODEC_ID_H264) {
  447. if (avctx->strict_std_compliance != FF_COMPLIANCE_NORMAL)
  448. q->extco.NalHrdConformance = avctx->strict_std_compliance > FF_COMPLIANCE_NORMAL ?
  449. MFX_CODINGOPTION_ON : MFX_CODINGOPTION_OFF;
  450. if (q->single_sei_nal_unit >= 0)
  451. q->extco.SingleSeiNalUnit = q->single_sei_nal_unit ? MFX_CODINGOPTION_ON : MFX_CODINGOPTION_OFF;
  452. if (q->recovery_point_sei >= 0)
  453. q->extco.RecoveryPointSEI = q->recovery_point_sei ? MFX_CODINGOPTION_ON : MFX_CODINGOPTION_OFF;
  454. q->extco.MaxDecFrameBuffering = q->max_dec_frame_buffering;
  455. }
  456. q->extparam_internal[q->nb_extparam_internal++] = (mfxExtBuffer *)&q->extco;
  457. #if QSV_HAVE_CO2
  458. if (avctx->codec_id == AV_CODEC_ID_H264) {
  459. q->extco2.Header.BufferId = MFX_EXTBUFF_CODING_OPTION2;
  460. q->extco2.Header.BufferSz = sizeof(q->extco2);
  461. if (q->int_ref_type >= 0)
  462. q->extco2.IntRefType = q->int_ref_type;
  463. if (q->int_ref_cycle_size >= 0)
  464. q->extco2.IntRefCycleSize = q->int_ref_cycle_size;
  465. if (q->int_ref_qp_delta != INT16_MIN)
  466. q->extco2.IntRefQPDelta = q->int_ref_qp_delta;
  467. if (q->bitrate_limit >= 0)
  468. q->extco2.BitrateLimit = q->bitrate_limit ? MFX_CODINGOPTION_ON : MFX_CODINGOPTION_OFF;
  469. if (q->mbbrc >= 0)
  470. q->extco2.MBBRC = q->mbbrc ? MFX_CODINGOPTION_ON : MFX_CODINGOPTION_OFF;
  471. if (q->extbrc >= 0)
  472. q->extco2.ExtBRC = q->extbrc ? MFX_CODINGOPTION_ON : MFX_CODINGOPTION_OFF;
  473. if (q->max_frame_size >= 0)
  474. q->extco2.MaxFrameSize = q->max_frame_size;
  475. #if QSV_HAVE_MAX_SLICE_SIZE
  476. if (q->max_slice_size >= 0)
  477. q->extco2.MaxSliceSize = q->max_slice_size;
  478. #endif
  479. #if QSV_HAVE_TRELLIS
  480. q->extco2.Trellis = q->trellis;
  481. #endif
  482. #if QSV_HAVE_BREF_TYPE
  483. #if FF_API_PRIVATE_OPT
  484. FF_DISABLE_DEPRECATION_WARNINGS
  485. if (avctx->b_frame_strategy >= 0)
  486. q->b_strategy = avctx->b_frame_strategy;
  487. FF_ENABLE_DEPRECATION_WARNINGS
  488. #endif
  489. if (q->b_strategy >= 0)
  490. q->extco2.BRefType = q->b_strategy ? MFX_B_REF_PYRAMID : MFX_B_REF_OFF;
  491. if (q->adaptive_i >= 0)
  492. q->extco2.AdaptiveI = q->adaptive_i ? MFX_CODINGOPTION_ON : MFX_CODINGOPTION_OFF;
  493. if (q->adaptive_b >= 0)
  494. q->extco2.AdaptiveB = q->adaptive_b ? MFX_CODINGOPTION_ON : MFX_CODINGOPTION_OFF;
  495. #endif
  496. q->extparam_internal[q->nb_extparam_internal++] = (mfxExtBuffer *)&q->extco2;
  497. }
  498. #endif
  499. }
  500. if (!rc_supported(q)) {
  501. av_log(avctx, AV_LOG_ERROR,
  502. "Selected ratecontrol mode is not supported by the QSV "
  503. "runtime. Choose a different mode.\n");
  504. return AVERROR(ENOSYS);
  505. }
  506. return 0;
  507. }
  508. static int qsv_retrieve_enc_params(AVCodecContext *avctx, QSVEncContext *q)
  509. {
  510. AVCPBProperties *cpb_props;
  511. uint8_t sps_buf[128];
  512. uint8_t pps_buf[128];
  513. mfxExtCodingOptionSPSPPS extradata = {
  514. .Header.BufferId = MFX_EXTBUFF_CODING_OPTION_SPSPPS,
  515. .Header.BufferSz = sizeof(extradata),
  516. .SPSBuffer = sps_buf, .SPSBufSize = sizeof(sps_buf),
  517. .PPSBuffer = pps_buf, .PPSBufSize = sizeof(pps_buf)
  518. };
  519. mfxExtCodingOption co = {
  520. .Header.BufferId = MFX_EXTBUFF_CODING_OPTION,
  521. .Header.BufferSz = sizeof(co),
  522. };
  523. #if QSV_HAVE_CO2
  524. mfxExtCodingOption2 co2 = {
  525. .Header.BufferId = MFX_EXTBUFF_CODING_OPTION2,
  526. .Header.BufferSz = sizeof(co2),
  527. };
  528. #endif
  529. #if QSV_HAVE_CO3
  530. mfxExtCodingOption3 co3 = {
  531. .Header.BufferId = MFX_EXTBUFF_CODING_OPTION3,
  532. .Header.BufferSz = sizeof(co3),
  533. };
  534. #endif
  535. mfxExtBuffer *ext_buffers[] = {
  536. (mfxExtBuffer*)&extradata,
  537. (mfxExtBuffer*)&co,
  538. #if QSV_HAVE_CO2
  539. (mfxExtBuffer*)&co2,
  540. #endif
  541. #if QSV_HAVE_CO3
  542. (mfxExtBuffer*)&co3,
  543. #endif
  544. };
  545. int need_pps = avctx->codec_id != AV_CODEC_ID_MPEG2VIDEO;
  546. int ret;
  547. q->param.ExtParam = ext_buffers;
  548. q->param.NumExtParam = FF_ARRAY_ELEMS(ext_buffers);
  549. ret = MFXVideoENCODE_GetVideoParam(q->session, &q->param);
  550. if (ret < 0)
  551. return ff_qsv_print_error(avctx, ret,
  552. "Error calling GetVideoParam");
  553. q->packet_size = q->param.mfx.BufferSizeInKB * 1000;
  554. if (!extradata.SPSBufSize || (need_pps && !extradata.PPSBufSize)) {
  555. av_log(avctx, AV_LOG_ERROR, "No extradata returned from libmfx.\n");
  556. return AVERROR_UNKNOWN;
  557. }
  558. avctx->extradata = av_malloc(extradata.SPSBufSize + need_pps * extradata.PPSBufSize +
  559. AV_INPUT_BUFFER_PADDING_SIZE);
  560. if (!avctx->extradata)
  561. return AVERROR(ENOMEM);
  562. memcpy(avctx->extradata, sps_buf, extradata.SPSBufSize);
  563. if (need_pps)
  564. memcpy(avctx->extradata + extradata.SPSBufSize, pps_buf, extradata.PPSBufSize);
  565. avctx->extradata_size = extradata.SPSBufSize + need_pps * extradata.PPSBufSize;
  566. memset(avctx->extradata + avctx->extradata_size, 0, AV_INPUT_BUFFER_PADDING_SIZE);
  567. cpb_props = ff_add_cpb_side_data(avctx);
  568. if (!cpb_props)
  569. return AVERROR(ENOMEM);
  570. cpb_props->max_bitrate = avctx->rc_max_rate;
  571. cpb_props->min_bitrate = avctx->rc_min_rate;
  572. cpb_props->avg_bitrate = avctx->bit_rate;
  573. cpb_props->buffer_size = avctx->rc_buffer_size;
  574. dump_video_param(avctx, q, ext_buffers + 1);
  575. return 0;
  576. }
  577. static int qsv_init_opaque_alloc(AVCodecContext *avctx, QSVEncContext *q)
  578. {
  579. AVQSVContext *qsv = avctx->hwaccel_context;
  580. mfxFrameSurface1 *surfaces;
  581. int nb_surfaces, i;
  582. nb_surfaces = qsv->nb_opaque_surfaces + q->req.NumFrameSuggested + q->async_depth;
  583. q->opaque_alloc_buf = av_buffer_allocz(sizeof(*surfaces) * nb_surfaces);
  584. if (!q->opaque_alloc_buf)
  585. return AVERROR(ENOMEM);
  586. q->opaque_surfaces = av_malloc_array(nb_surfaces, sizeof(*q->opaque_surfaces));
  587. if (!q->opaque_surfaces)
  588. return AVERROR(ENOMEM);
  589. surfaces = (mfxFrameSurface1*)q->opaque_alloc_buf->data;
  590. for (i = 0; i < nb_surfaces; i++) {
  591. surfaces[i].Info = q->req.Info;
  592. q->opaque_surfaces[i] = surfaces + i;
  593. }
  594. q->opaque_alloc.Header.BufferId = MFX_EXTBUFF_OPAQUE_SURFACE_ALLOCATION;
  595. q->opaque_alloc.Header.BufferSz = sizeof(q->opaque_alloc);
  596. q->opaque_alloc.In.Surfaces = q->opaque_surfaces;
  597. q->opaque_alloc.In.NumSurface = nb_surfaces;
  598. q->opaque_alloc.In.Type = q->req.Type;
  599. q->extparam_internal[q->nb_extparam_internal++] = (mfxExtBuffer *)&q->opaque_alloc;
  600. qsv->nb_opaque_surfaces = nb_surfaces;
  601. qsv->opaque_surfaces = q->opaque_alloc_buf;
  602. qsv->opaque_alloc_type = q->req.Type;
  603. return 0;
  604. }
  605. static int qsvenc_init_session(AVCodecContext *avctx, QSVEncContext *q)
  606. {
  607. int ret;
  608. if (avctx->hwaccel_context) {
  609. AVQSVContext *qsv = avctx->hwaccel_context;
  610. q->session = qsv->session;
  611. } else if (avctx->hw_frames_ctx) {
  612. q->frames_ctx.hw_frames_ctx = av_buffer_ref(avctx->hw_frames_ctx);
  613. if (!q->frames_ctx.hw_frames_ctx)
  614. return AVERROR(ENOMEM);
  615. ret = ff_qsv_init_session_frames(avctx, &q->internal_session,
  616. &q->frames_ctx, q->load_plugins,
  617. q->param.IOPattern == MFX_IOPATTERN_IN_OPAQUE_MEMORY);
  618. if (ret < 0) {
  619. av_buffer_unref(&q->frames_ctx.hw_frames_ctx);
  620. return ret;
  621. }
  622. q->session = q->internal_session;
  623. } else if (avctx->hw_device_ctx) {
  624. ret = ff_qsv_init_session_device(avctx, &q->internal_session,
  625. avctx->hw_device_ctx, q->load_plugins);
  626. if (ret < 0)
  627. return ret;
  628. q->session = q->internal_session;
  629. } else {
  630. ret = ff_qsv_init_internal_session(avctx, &q->internal_session,
  631. q->load_plugins);
  632. if (ret < 0)
  633. return ret;
  634. q->session = q->internal_session;
  635. }
  636. return 0;
  637. }
  638. int ff_qsv_enc_init(AVCodecContext *avctx, QSVEncContext *q)
  639. {
  640. int iopattern = 0;
  641. int opaque_alloc = 0;
  642. int ret;
  643. q->param.AsyncDepth = q->async_depth;
  644. q->async_fifo = av_fifo_alloc((1 + q->async_depth) *
  645. (sizeof(AVPacket) + sizeof(mfxSyncPoint*) + sizeof(mfxBitstream*)));
  646. if (!q->async_fifo)
  647. return AVERROR(ENOMEM);
  648. if (avctx->hwaccel_context) {
  649. AVQSVContext *qsv = avctx->hwaccel_context;
  650. iopattern = qsv->iopattern;
  651. opaque_alloc = qsv->opaque_alloc;
  652. }
  653. if (avctx->hw_frames_ctx) {
  654. AVHWFramesContext *frames_ctx = (AVHWFramesContext*)avctx->hw_frames_ctx->data;
  655. AVQSVFramesContext *frames_hwctx = frames_ctx->hwctx;
  656. if (!iopattern) {
  657. if (frames_hwctx->frame_type & MFX_MEMTYPE_OPAQUE_FRAME)
  658. iopattern = MFX_IOPATTERN_IN_OPAQUE_MEMORY;
  659. else if (frames_hwctx->frame_type &
  660. (MFX_MEMTYPE_VIDEO_MEMORY_DECODER_TARGET | MFX_MEMTYPE_VIDEO_MEMORY_PROCESSOR_TARGET))
  661. iopattern = MFX_IOPATTERN_IN_VIDEO_MEMORY;
  662. }
  663. }
  664. if (!iopattern)
  665. iopattern = MFX_IOPATTERN_IN_SYSTEM_MEMORY;
  666. q->param.IOPattern = iopattern;
  667. ret = qsvenc_init_session(avctx, q);
  668. if (ret < 0)
  669. return ret;
  670. ret = init_video_param(avctx, q);
  671. if (ret < 0)
  672. return ret;
  673. ret = MFXVideoENCODE_Query(q->session, &q->param, &q->param);
  674. if (ret == MFX_WRN_PARTIAL_ACCELERATION) {
  675. av_log(avctx, AV_LOG_WARNING, "Encoder will work with partial HW acceleration\n");
  676. } else if (ret < 0) {
  677. return ff_qsv_print_error(avctx, ret,
  678. "Error querying encoder params");
  679. }
  680. ret = MFXVideoENCODE_QueryIOSurf(q->session, &q->param, &q->req);
  681. if (ret < 0)
  682. return ff_qsv_print_error(avctx, ret,
  683. "Error querying (IOSurf) the encoding parameters");
  684. if (opaque_alloc) {
  685. ret = qsv_init_opaque_alloc(avctx, q);
  686. if (ret < 0)
  687. return ret;
  688. }
  689. if (avctx->hwaccel_context) {
  690. AVQSVContext *qsv = avctx->hwaccel_context;
  691. int i, j;
  692. q->extparam = av_mallocz_array(qsv->nb_ext_buffers + q->nb_extparam_internal,
  693. sizeof(*q->extparam));
  694. if (!q->extparam)
  695. return AVERROR(ENOMEM);
  696. q->param.ExtParam = q->extparam;
  697. for (i = 0; i < qsv->nb_ext_buffers; i++)
  698. q->param.ExtParam[i] = qsv->ext_buffers[i];
  699. q->param.NumExtParam = qsv->nb_ext_buffers;
  700. for (i = 0; i < q->nb_extparam_internal; i++) {
  701. for (j = 0; j < qsv->nb_ext_buffers; j++) {
  702. if (qsv->ext_buffers[j]->BufferId == q->extparam_internal[i]->BufferId)
  703. break;
  704. }
  705. if (j < qsv->nb_ext_buffers)
  706. continue;
  707. q->param.ExtParam[q->param.NumExtParam++] = q->extparam_internal[i];
  708. }
  709. } else {
  710. q->param.ExtParam = q->extparam_internal;
  711. q->param.NumExtParam = q->nb_extparam_internal;
  712. }
  713. ret = MFXVideoENCODE_Init(q->session, &q->param);
  714. if (ret < 0)
  715. return ff_qsv_print_error(avctx, ret,
  716. "Error initializing the encoder");
  717. else if (ret > 0)
  718. ff_qsv_print_warning(avctx, ret,
  719. "Warning in encoder initialization");
  720. ret = qsv_retrieve_enc_params(avctx, q);
  721. if (ret < 0) {
  722. av_log(avctx, AV_LOG_ERROR, "Error retrieving encoding parameters.\n");
  723. return ret;
  724. }
  725. q->avctx = avctx;
  726. return 0;
  727. }
  728. static void clear_unused_frames(QSVEncContext *q)
  729. {
  730. QSVFrame *cur = q->work_frames;
  731. while (cur) {
  732. if (cur->used && !cur->surface.Data.Locked) {
  733. av_frame_unref(cur->frame);
  734. cur->used = 0;
  735. }
  736. cur = cur->next;
  737. }
  738. }
  739. static int get_free_frame(QSVEncContext *q, QSVFrame **f)
  740. {
  741. QSVFrame *frame, **last;
  742. clear_unused_frames(q);
  743. frame = q->work_frames;
  744. last = &q->work_frames;
  745. while (frame) {
  746. if (!frame->used) {
  747. *f = frame;
  748. frame->used = 1;
  749. return 0;
  750. }
  751. last = &frame->next;
  752. frame = frame->next;
  753. }
  754. frame = av_mallocz(sizeof(*frame));
  755. if (!frame)
  756. return AVERROR(ENOMEM);
  757. frame->frame = av_frame_alloc();
  758. if (!frame->frame) {
  759. av_freep(&frame);
  760. return AVERROR(ENOMEM);
  761. }
  762. *last = frame;
  763. *f = frame;
  764. frame->used = 1;
  765. return 0;
  766. }
  767. static int submit_frame(QSVEncContext *q, const AVFrame *frame,
  768. mfxFrameSurface1 **surface)
  769. {
  770. QSVFrame *qf;
  771. int ret;
  772. ret = get_free_frame(q, &qf);
  773. if (ret < 0)
  774. return ret;
  775. if (frame->format == AV_PIX_FMT_QSV) {
  776. ret = av_frame_ref(qf->frame, frame);
  777. if (ret < 0)
  778. return ret;
  779. qf->surface = *(mfxFrameSurface1*)qf->frame->data[3];
  780. if (q->frames_ctx.mids) {
  781. ret = ff_qsv_find_surface_idx(&q->frames_ctx, qf);
  782. if (ret < 0)
  783. return ret;
  784. qf->surface.Data.MemId = &q->frames_ctx.mids[ret];
  785. }
  786. } else {
  787. /* make a copy if the input is not padded as libmfx requires */
  788. if (frame->height & 31 || frame->linesize[0] & (q->width_align - 1)) {
  789. qf->frame->height = FFALIGN(frame->height, q->height_align);
  790. qf->frame->width = FFALIGN(frame->width, q->width_align);
  791. ret = ff_get_buffer(q->avctx, qf->frame, AV_GET_BUFFER_FLAG_REF);
  792. if (ret < 0)
  793. return ret;
  794. qf->frame->height = frame->height;
  795. qf->frame->width = frame->width;
  796. ret = av_frame_copy(qf->frame, frame);
  797. if (ret < 0) {
  798. av_frame_unref(qf->frame);
  799. return ret;
  800. }
  801. } else {
  802. ret = av_frame_ref(qf->frame, frame);
  803. if (ret < 0)
  804. return ret;
  805. }
  806. qf->surface.Info = q->param.mfx.FrameInfo;
  807. qf->surface.Info.PicStruct =
  808. !frame->interlaced_frame ? MFX_PICSTRUCT_PROGRESSIVE :
  809. frame->top_field_first ? MFX_PICSTRUCT_FIELD_TFF :
  810. MFX_PICSTRUCT_FIELD_BFF;
  811. if (frame->repeat_pict == 1)
  812. qf->surface.Info.PicStruct |= MFX_PICSTRUCT_FIELD_REPEATED;
  813. else if (frame->repeat_pict == 2)
  814. qf->surface.Info.PicStruct |= MFX_PICSTRUCT_FRAME_DOUBLING;
  815. else if (frame->repeat_pict == 4)
  816. qf->surface.Info.PicStruct |= MFX_PICSTRUCT_FRAME_TRIPLING;
  817. qf->surface.Data.PitchLow = qf->frame->linesize[0];
  818. qf->surface.Data.Y = qf->frame->data[0];
  819. qf->surface.Data.UV = qf->frame->data[1];
  820. }
  821. qf->surface.Data.TimeStamp = av_rescale_q(frame->pts, q->avctx->time_base, (AVRational){1, 90000});
  822. *surface = &qf->surface;
  823. return 0;
  824. }
  825. static void print_interlace_msg(AVCodecContext *avctx, QSVEncContext *q)
  826. {
  827. if (q->param.mfx.CodecId == MFX_CODEC_AVC) {
  828. if (q->param.mfx.CodecProfile == MFX_PROFILE_AVC_BASELINE ||
  829. q->param.mfx.CodecLevel < MFX_LEVEL_AVC_21 ||
  830. q->param.mfx.CodecLevel > MFX_LEVEL_AVC_41)
  831. av_log(avctx, AV_LOG_WARNING,
  832. "Interlaced coding is supported"
  833. " at Main/High Profile Level 2.1-4.1\n");
  834. }
  835. }
  836. static int encode_frame(AVCodecContext *avctx, QSVEncContext *q,
  837. const AVFrame *frame)
  838. {
  839. AVPacket new_pkt = { 0 };
  840. mfxBitstream *bs;
  841. mfxFrameSurface1 *surf = NULL;
  842. mfxSyncPoint *sync = NULL;
  843. int ret;
  844. if (frame) {
  845. ret = submit_frame(q, frame, &surf);
  846. if (ret < 0) {
  847. av_log(avctx, AV_LOG_ERROR, "Error submitting the frame for encoding.\n");
  848. return ret;
  849. }
  850. }
  851. ret = av_new_packet(&new_pkt, q->packet_size);
  852. if (ret < 0) {
  853. av_log(avctx, AV_LOG_ERROR, "Error allocating the output packet\n");
  854. return ret;
  855. }
  856. bs = av_mallocz(sizeof(*bs));
  857. if (!bs) {
  858. av_packet_unref(&new_pkt);
  859. return AVERROR(ENOMEM);
  860. }
  861. bs->Data = new_pkt.data;
  862. bs->MaxLength = new_pkt.size;
  863. sync = av_mallocz(sizeof(*sync));
  864. if (!sync) {
  865. av_freep(&bs);
  866. av_packet_unref(&new_pkt);
  867. return AVERROR(ENOMEM);
  868. }
  869. do {
  870. ret = MFXVideoENCODE_EncodeFrameAsync(q->session, NULL, surf, bs, sync);
  871. if (ret == MFX_WRN_DEVICE_BUSY)
  872. av_usleep(1);
  873. } while (ret == MFX_WRN_DEVICE_BUSY || ret == MFX_WRN_IN_EXECUTION);
  874. if (ret > 0)
  875. ff_qsv_print_warning(avctx, ret, "Warning during encoding");
  876. if (ret < 0) {
  877. av_packet_unref(&new_pkt);
  878. av_freep(&bs);
  879. av_freep(&sync);
  880. return (ret == MFX_ERR_MORE_DATA) ?
  881. 0 : ff_qsv_print_error(avctx, ret, "Error during encoding");
  882. }
  883. if (ret == MFX_WRN_INCOMPATIBLE_VIDEO_PARAM && frame->interlaced_frame)
  884. print_interlace_msg(avctx, q);
  885. if (*sync) {
  886. av_fifo_generic_write(q->async_fifo, &new_pkt, sizeof(new_pkt), NULL);
  887. av_fifo_generic_write(q->async_fifo, &sync, sizeof(sync), NULL);
  888. av_fifo_generic_write(q->async_fifo, &bs, sizeof(bs), NULL);
  889. } else {
  890. av_freep(&sync);
  891. av_packet_unref(&new_pkt);
  892. av_freep(&bs);
  893. }
  894. return 0;
  895. }
  896. int ff_qsv_encode(AVCodecContext *avctx, QSVEncContext *q,
  897. AVPacket *pkt, const AVFrame *frame, int *got_packet)
  898. {
  899. int ret;
  900. ret = encode_frame(avctx, q, frame);
  901. if (ret < 0)
  902. return ret;
  903. if (!av_fifo_space(q->async_fifo) ||
  904. (!frame && av_fifo_size(q->async_fifo))) {
  905. AVPacket new_pkt;
  906. mfxBitstream *bs;
  907. mfxSyncPoint *sync;
  908. av_fifo_generic_read(q->async_fifo, &new_pkt, sizeof(new_pkt), NULL);
  909. av_fifo_generic_read(q->async_fifo, &sync, sizeof(sync), NULL);
  910. av_fifo_generic_read(q->async_fifo, &bs, sizeof(bs), NULL);
  911. do {
  912. ret = MFXVideoCORE_SyncOperation(q->session, *sync, 1000);
  913. } while (ret == MFX_WRN_IN_EXECUTION);
  914. new_pkt.dts = av_rescale_q(bs->DecodeTimeStamp, (AVRational){1, 90000}, avctx->time_base);
  915. new_pkt.pts = av_rescale_q(bs->TimeStamp, (AVRational){1, 90000}, avctx->time_base);
  916. new_pkt.size = bs->DataLength;
  917. if (bs->FrameType & MFX_FRAMETYPE_IDR ||
  918. bs->FrameType & MFX_FRAMETYPE_xIDR)
  919. new_pkt.flags |= AV_PKT_FLAG_KEY;
  920. #if FF_API_CODED_FRAME
  921. FF_DISABLE_DEPRECATION_WARNINGS
  922. if (bs->FrameType & MFX_FRAMETYPE_I || bs->FrameType & MFX_FRAMETYPE_xI)
  923. avctx->coded_frame->pict_type = AV_PICTURE_TYPE_I;
  924. else if (bs->FrameType & MFX_FRAMETYPE_P || bs->FrameType & MFX_FRAMETYPE_xP)
  925. avctx->coded_frame->pict_type = AV_PICTURE_TYPE_P;
  926. else if (bs->FrameType & MFX_FRAMETYPE_B || bs->FrameType & MFX_FRAMETYPE_xB)
  927. avctx->coded_frame->pict_type = AV_PICTURE_TYPE_B;
  928. FF_ENABLE_DEPRECATION_WARNINGS
  929. #endif
  930. av_freep(&bs);
  931. av_freep(&sync);
  932. if (pkt->data) {
  933. if (pkt->size < new_pkt.size) {
  934. av_log(avctx, AV_LOG_ERROR, "Submitted buffer not large enough: %d < %d\n",
  935. pkt->size, new_pkt.size);
  936. av_packet_unref(&new_pkt);
  937. return AVERROR(EINVAL);
  938. }
  939. memcpy(pkt->data, new_pkt.data, new_pkt.size);
  940. pkt->size = new_pkt.size;
  941. ret = av_packet_copy_props(pkt, &new_pkt);
  942. av_packet_unref(&new_pkt);
  943. if (ret < 0)
  944. return ret;
  945. } else
  946. *pkt = new_pkt;
  947. *got_packet = 1;
  948. }
  949. return 0;
  950. }
  951. int ff_qsv_enc_close(AVCodecContext *avctx, QSVEncContext *q)
  952. {
  953. QSVFrame *cur;
  954. if (q->session)
  955. MFXVideoENCODE_Close(q->session);
  956. if (q->internal_session)
  957. MFXClose(q->internal_session);
  958. q->session = NULL;
  959. q->internal_session = NULL;
  960. av_buffer_unref(&q->frames_ctx.hw_frames_ctx);
  961. av_buffer_unref(&q->frames_ctx.mids_buf);
  962. cur = q->work_frames;
  963. while (cur) {
  964. q->work_frames = cur->next;
  965. av_frame_free(&cur->frame);
  966. av_freep(&cur);
  967. cur = q->work_frames;
  968. }
  969. while (q->async_fifo && av_fifo_size(q->async_fifo)) {
  970. AVPacket pkt;
  971. mfxSyncPoint *sync;
  972. mfxBitstream *bs;
  973. av_fifo_generic_read(q->async_fifo, &pkt, sizeof(pkt), NULL);
  974. av_fifo_generic_read(q->async_fifo, &sync, sizeof(sync), NULL);
  975. av_fifo_generic_read(q->async_fifo, &bs, sizeof(bs), NULL);
  976. av_freep(&sync);
  977. av_freep(&bs);
  978. av_packet_unref(&pkt);
  979. }
  980. av_fifo_free(q->async_fifo);
  981. q->async_fifo = NULL;
  982. av_freep(&q->opaque_surfaces);
  983. av_buffer_unref(&q->opaque_alloc_buf);
  984. av_freep(&q->extparam);
  985. return 0;
  986. }