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.

792 lines
24KB

  1. /*
  2. * This file is part of Libav.
  3. *
  4. * Libav is free software; you can redistribute it and/or
  5. * modify it under the terms of the GNU Lesser General Public
  6. * License as published by the Free Software Foundation; either
  7. * version 2.1 of the License, or (at your option) any later version.
  8. *
  9. * Libav is distributed in the hope that it will be useful,
  10. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  12. * Lesser General Public License for more details.
  13. *
  14. * You should have received a copy of the GNU Lesser General Public
  15. * License along with Libav; if not, write to the Free Software
  16. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  17. */
  18. /**
  19. * @file
  20. * Frame multithreading support functions
  21. * @see doc/multithreading.txt
  22. */
  23. #include "config.h"
  24. #include <stdint.h>
  25. #if HAVE_PTHREADS
  26. #include <pthread.h>
  27. #elif HAVE_W32THREADS
  28. #include "compat/w32pthreads.h"
  29. #endif
  30. #include "avcodec.h"
  31. #include "internal.h"
  32. #include "pthread_internal.h"
  33. #include "thread.h"
  34. #include "version.h"
  35. #include "libavutil/avassert.h"
  36. #include "libavutil/buffer.h"
  37. #include "libavutil/common.h"
  38. #include "libavutil/cpu.h"
  39. #include "libavutil/frame.h"
  40. #include "libavutil/internal.h"
  41. #include "libavutil/log.h"
  42. #include "libavutil/mem.h"
  43. /**
  44. * Context used by codec threads and stored in their AVCodecInternal thread_ctx.
  45. */
  46. typedef struct PerThreadContext {
  47. struct FrameThreadContext *parent;
  48. pthread_t thread;
  49. int thread_init;
  50. pthread_cond_t input_cond; ///< Used to wait for a new packet from the main thread.
  51. pthread_cond_t progress_cond; ///< Used by child threads to wait for progress to change.
  52. pthread_cond_t output_cond; ///< Used by the main thread to wait for frames to finish.
  53. pthread_mutex_t mutex; ///< Mutex used to protect the contents of the PerThreadContext.
  54. pthread_mutex_t progress_mutex; ///< Mutex used to protect frame progress values and progress_cond.
  55. AVCodecContext *avctx; ///< Context used to decode packets passed to this thread.
  56. AVPacket avpkt; ///< Input packet (for decoding) or output (for encoding).
  57. AVFrame *frame; ///< Output frame (for decoding) or input (for encoding).
  58. int got_frame; ///< The output of got_picture_ptr from the last avcodec_decode_video() call.
  59. int result; ///< The result of the last codec decode/encode() call.
  60. enum {
  61. STATE_INPUT_READY, ///< Set when the thread is awaiting a packet.
  62. STATE_SETTING_UP, ///< Set before the codec has called ff_thread_finish_setup().
  63. STATE_GET_BUFFER, /**<
  64. * Set when the codec calls get_buffer().
  65. * State is returned to STATE_SETTING_UP afterwards.
  66. */
  67. STATE_SETUP_FINISHED ///< Set after the codec has called ff_thread_finish_setup().
  68. } state;
  69. /**
  70. * Array of frames passed to ff_thread_release_buffer().
  71. * Frames are released after all threads referencing them are finished.
  72. */
  73. AVFrame *released_buffers;
  74. int num_released_buffers;
  75. int released_buffers_allocated;
  76. AVFrame *requested_frame; ///< AVFrame the codec passed to get_buffer()
  77. int requested_flags; ///< flags passed to get_buffer() for requested_frame
  78. } PerThreadContext;
  79. /**
  80. * Context stored in the client AVCodecInternal thread_ctx.
  81. */
  82. typedef struct FrameThreadContext {
  83. PerThreadContext *threads; ///< The contexts for each thread.
  84. PerThreadContext *prev_thread; ///< The last thread submit_packet() was called on.
  85. pthread_mutex_t buffer_mutex; ///< Mutex used to protect get/release_buffer().
  86. int next_decoding; ///< The next context to submit a packet to.
  87. int next_finished; ///< The next context to return output from.
  88. int delaying; /**<
  89. * Set for the first N packets, where N is the number of threads.
  90. * While it is set, ff_thread_en/decode_frame won't return any results.
  91. */
  92. int die; ///< Set when threads should exit.
  93. } FrameThreadContext;
  94. /**
  95. * Codec worker thread.
  96. *
  97. * Automatically calls ff_thread_finish_setup() if the codec does
  98. * not provide an update_thread_context method, or if the codec returns
  99. * before calling it.
  100. */
  101. static attribute_align_arg void *frame_worker_thread(void *arg)
  102. {
  103. PerThreadContext *p = arg;
  104. FrameThreadContext *fctx = p->parent;
  105. AVCodecContext *avctx = p->avctx;
  106. const AVCodec *codec = avctx->codec;
  107. while (1) {
  108. if (p->state == STATE_INPUT_READY && !fctx->die) {
  109. pthread_mutex_lock(&p->mutex);
  110. while (p->state == STATE_INPUT_READY && !fctx->die)
  111. pthread_cond_wait(&p->input_cond, &p->mutex);
  112. pthread_mutex_unlock(&p->mutex);
  113. }
  114. if (fctx->die) break;
  115. if (!codec->update_thread_context && avctx->thread_safe_callbacks)
  116. ff_thread_finish_setup(avctx);
  117. pthread_mutex_lock(&p->mutex);
  118. av_frame_unref(p->frame);
  119. p->got_frame = 0;
  120. p->result = codec->decode(avctx, p->frame, &p->got_frame, &p->avpkt);
  121. if ((p->result < 0 || !p->got_frame) && p->frame->buf[0]) {
  122. if (avctx->internal->allocate_progress)
  123. av_log(avctx, AV_LOG_ERROR, "A frame threaded decoder did not "
  124. "free the frame on failure. This is a bug, please report it.\n");
  125. av_frame_unref(p->frame);
  126. }
  127. if (p->state == STATE_SETTING_UP) ff_thread_finish_setup(avctx);
  128. p->state = STATE_INPUT_READY;
  129. pthread_mutex_lock(&p->progress_mutex);
  130. pthread_cond_signal(&p->output_cond);
  131. pthread_mutex_unlock(&p->progress_mutex);
  132. pthread_mutex_unlock(&p->mutex);
  133. }
  134. return NULL;
  135. }
  136. /**
  137. * Update the next thread's AVCodecContext with values from the reference thread's context.
  138. *
  139. * @param dst The destination context.
  140. * @param src The source context.
  141. * @param for_user 0 if the destination is a codec thread, 1 if the destination is the user's thread
  142. */
  143. static int update_context_from_thread(AVCodecContext *dst, AVCodecContext *src, int for_user)
  144. {
  145. int err = 0;
  146. if (dst != src) {
  147. dst->time_base = src->time_base;
  148. dst->framerate = src->framerate;
  149. dst->width = src->width;
  150. dst->height = src->height;
  151. dst->pix_fmt = src->pix_fmt;
  152. dst->coded_width = src->coded_width;
  153. dst->coded_height = src->coded_height;
  154. dst->has_b_frames = src->has_b_frames;
  155. dst->idct_algo = src->idct_algo;
  156. dst->bits_per_coded_sample = src->bits_per_coded_sample;
  157. dst->sample_aspect_ratio = src->sample_aspect_ratio;
  158. #if FF_API_AFD
  159. FF_DISABLE_DEPRECATION_WARNINGS
  160. dst->dtg_active_format = src->dtg_active_format;
  161. FF_ENABLE_DEPRECATION_WARNINGS
  162. #endif /* FF_API_AFD */
  163. dst->profile = src->profile;
  164. dst->level = src->level;
  165. dst->bits_per_raw_sample = src->bits_per_raw_sample;
  166. dst->ticks_per_frame = src->ticks_per_frame;
  167. dst->color_primaries = src->color_primaries;
  168. dst->color_trc = src->color_trc;
  169. dst->colorspace = src->colorspace;
  170. dst->color_range = src->color_range;
  171. dst->chroma_sample_location = src->chroma_sample_location;
  172. dst->hwaccel = src->hwaccel;
  173. dst->hwaccel_context = src->hwaccel_context;
  174. dst->internal->hwaccel_priv_data = src->internal->hwaccel_priv_data;
  175. }
  176. if (for_user) {
  177. #if FF_API_CODED_FRAME
  178. FF_DISABLE_DEPRECATION_WARNINGS
  179. dst->coded_frame = src->coded_frame;
  180. FF_ENABLE_DEPRECATION_WARNINGS
  181. #endif
  182. } else {
  183. if (dst->codec->update_thread_context)
  184. err = dst->codec->update_thread_context(dst, src);
  185. }
  186. return err;
  187. }
  188. /**
  189. * Update the next thread's AVCodecContext with values set by the user.
  190. *
  191. * @param dst The destination context.
  192. * @param src The source context.
  193. * @return 0 on success, negative error code on failure
  194. */
  195. static int update_context_from_user(AVCodecContext *dst, AVCodecContext *src)
  196. {
  197. #define copy_fields(s, e) memcpy(&dst->s, &src->s, (char*)&dst->e - (char*)&dst->s);
  198. dst->flags = src->flags;
  199. dst->draw_horiz_band= src->draw_horiz_band;
  200. dst->get_buffer2 = src->get_buffer2;
  201. dst->opaque = src->opaque;
  202. dst->debug = src->debug;
  203. dst->slice_flags = src->slice_flags;
  204. dst->flags2 = src->flags2;
  205. copy_fields(skip_loop_filter, subtitle_header);
  206. dst->frame_number = src->frame_number;
  207. dst->reordered_opaque = src->reordered_opaque;
  208. if (src->slice_count && src->slice_offset) {
  209. if (dst->slice_count < src->slice_count) {
  210. int *tmp = av_realloc(dst->slice_offset, src->slice_count *
  211. sizeof(*dst->slice_offset));
  212. if (!tmp) {
  213. av_free(dst->slice_offset);
  214. return AVERROR(ENOMEM);
  215. }
  216. dst->slice_offset = tmp;
  217. }
  218. memcpy(dst->slice_offset, src->slice_offset,
  219. src->slice_count * sizeof(*dst->slice_offset));
  220. }
  221. dst->slice_count = src->slice_count;
  222. return 0;
  223. #undef copy_fields
  224. }
  225. /// Releases the buffers that this decoding thread was the last user of.
  226. static void release_delayed_buffers(PerThreadContext *p)
  227. {
  228. FrameThreadContext *fctx = p->parent;
  229. while (p->num_released_buffers > 0) {
  230. AVFrame *f;
  231. pthread_mutex_lock(&fctx->buffer_mutex);
  232. // fix extended data in case the caller screwed it up
  233. av_assert0(p->avctx->codec_type == AVMEDIA_TYPE_VIDEO);
  234. f = &p->released_buffers[--p->num_released_buffers];
  235. f->extended_data = f->data;
  236. av_frame_unref(f);
  237. pthread_mutex_unlock(&fctx->buffer_mutex);
  238. }
  239. }
  240. static int submit_packet(PerThreadContext *p, AVPacket *avpkt)
  241. {
  242. FrameThreadContext *fctx = p->parent;
  243. PerThreadContext *prev_thread = fctx->prev_thread;
  244. const AVCodec *codec = p->avctx->codec;
  245. if (!avpkt->size && !(codec->capabilities & AV_CODEC_CAP_DELAY))
  246. return 0;
  247. pthread_mutex_lock(&p->mutex);
  248. release_delayed_buffers(p);
  249. if (prev_thread) {
  250. int err;
  251. if (prev_thread->state == STATE_SETTING_UP) {
  252. pthread_mutex_lock(&prev_thread->progress_mutex);
  253. while (prev_thread->state == STATE_SETTING_UP)
  254. pthread_cond_wait(&prev_thread->progress_cond, &prev_thread->progress_mutex);
  255. pthread_mutex_unlock(&prev_thread->progress_mutex);
  256. }
  257. err = update_context_from_thread(p->avctx, prev_thread->avctx, 0);
  258. if (err) {
  259. pthread_mutex_unlock(&p->mutex);
  260. return err;
  261. }
  262. }
  263. av_packet_unref(&p->avpkt);
  264. av_packet_ref(&p->avpkt, avpkt);
  265. p->state = STATE_SETTING_UP;
  266. pthread_cond_signal(&p->input_cond);
  267. pthread_mutex_unlock(&p->mutex);
  268. /*
  269. * If the client doesn't have a thread-safe get_buffer(),
  270. * then decoding threads call back to the main thread,
  271. * and it calls back to the client here.
  272. */
  273. if (!p->avctx->thread_safe_callbacks &&
  274. p->avctx->get_buffer2 != avcodec_default_get_buffer2) {
  275. while (p->state != STATE_SETUP_FINISHED && p->state != STATE_INPUT_READY) {
  276. pthread_mutex_lock(&p->progress_mutex);
  277. while (p->state == STATE_SETTING_UP)
  278. pthread_cond_wait(&p->progress_cond, &p->progress_mutex);
  279. if (p->state == STATE_GET_BUFFER) {
  280. p->result = ff_get_buffer(p->avctx, p->requested_frame, p->requested_flags);
  281. p->state = STATE_SETTING_UP;
  282. pthread_cond_signal(&p->progress_cond);
  283. }
  284. pthread_mutex_unlock(&p->progress_mutex);
  285. }
  286. }
  287. fctx->prev_thread = p;
  288. fctx->next_decoding++;
  289. return 0;
  290. }
  291. int ff_thread_decode_frame(AVCodecContext *avctx,
  292. AVFrame *picture, int *got_picture_ptr,
  293. AVPacket *avpkt)
  294. {
  295. FrameThreadContext *fctx = avctx->internal->thread_ctx;
  296. int finished = fctx->next_finished;
  297. PerThreadContext *p;
  298. int err;
  299. /*
  300. * Submit a packet to the next decoding thread.
  301. */
  302. p = &fctx->threads[fctx->next_decoding];
  303. err = update_context_from_user(p->avctx, avctx);
  304. if (err) return err;
  305. err = submit_packet(p, avpkt);
  306. if (err) return err;
  307. /*
  308. * If we're still receiving the initial packets, don't return a frame.
  309. */
  310. if (fctx->delaying) {
  311. if (fctx->next_decoding >= (avctx->thread_count-1)) fctx->delaying = 0;
  312. *got_picture_ptr=0;
  313. if (avpkt->size)
  314. return avpkt->size;
  315. }
  316. /*
  317. * Return the next available frame from the oldest thread.
  318. * If we're at the end of the stream, then we have to skip threads that
  319. * didn't output a frame, because we don't want to accidentally signal
  320. * EOF (avpkt->size == 0 && *got_picture_ptr == 0).
  321. */
  322. do {
  323. p = &fctx->threads[finished++];
  324. if (p->state != STATE_INPUT_READY) {
  325. pthread_mutex_lock(&p->progress_mutex);
  326. while (p->state != STATE_INPUT_READY)
  327. pthread_cond_wait(&p->output_cond, &p->progress_mutex);
  328. pthread_mutex_unlock(&p->progress_mutex);
  329. }
  330. av_frame_move_ref(picture, p->frame);
  331. *got_picture_ptr = p->got_frame;
  332. picture->pkt_dts = p->avpkt.dts;
  333. /*
  334. * A later call with avkpt->size == 0 may loop over all threads,
  335. * including this one, searching for a frame to return before being
  336. * stopped by the "finished != fctx->next_finished" condition.
  337. * Make sure we don't mistakenly return the same frame again.
  338. */
  339. p->got_frame = 0;
  340. if (finished >= avctx->thread_count) finished = 0;
  341. } while (!avpkt->size && !*got_picture_ptr && finished != fctx->next_finished);
  342. update_context_from_thread(avctx, p->avctx, 1);
  343. if (fctx->next_decoding >= avctx->thread_count) fctx->next_decoding = 0;
  344. fctx->next_finished = finished;
  345. /* return the size of the consumed packet if no error occurred */
  346. return (p->result >= 0) ? avpkt->size : p->result;
  347. }
  348. void ff_thread_report_progress(ThreadFrame *f, int n, int field)
  349. {
  350. PerThreadContext *p;
  351. int *progress = f->progress ? (int*)f->progress->data : NULL;
  352. if (!progress || progress[field] >= n) return;
  353. p = f->owner->internal->thread_ctx;
  354. if (f->owner->debug&FF_DEBUG_THREADS)
  355. av_log(f->owner, AV_LOG_DEBUG, "%p finished %d field %d\n", progress, n, field);
  356. pthread_mutex_lock(&p->progress_mutex);
  357. progress[field] = n;
  358. pthread_cond_broadcast(&p->progress_cond);
  359. pthread_mutex_unlock(&p->progress_mutex);
  360. }
  361. void ff_thread_await_progress(ThreadFrame *f, int n, int field)
  362. {
  363. PerThreadContext *p;
  364. int *progress = f->progress ? (int*)f->progress->data : NULL;
  365. if (!progress || progress[field] >= n) return;
  366. p = f->owner->internal->thread_ctx;
  367. if (f->owner->debug&FF_DEBUG_THREADS)
  368. av_log(f->owner, AV_LOG_DEBUG, "thread awaiting %d field %d from %p\n", n, field, progress);
  369. pthread_mutex_lock(&p->progress_mutex);
  370. while (progress[field] < n)
  371. pthread_cond_wait(&p->progress_cond, &p->progress_mutex);
  372. pthread_mutex_unlock(&p->progress_mutex);
  373. }
  374. void ff_thread_finish_setup(AVCodecContext *avctx) {
  375. PerThreadContext *p = avctx->internal->thread_ctx;
  376. if (!(avctx->active_thread_type&FF_THREAD_FRAME)) return;
  377. pthread_mutex_lock(&p->progress_mutex);
  378. p->state = STATE_SETUP_FINISHED;
  379. pthread_cond_broadcast(&p->progress_cond);
  380. pthread_mutex_unlock(&p->progress_mutex);
  381. }
  382. /// Waits for all threads to finish.
  383. static void park_frame_worker_threads(FrameThreadContext *fctx, int thread_count)
  384. {
  385. int i;
  386. for (i = 0; i < thread_count; i++) {
  387. PerThreadContext *p = &fctx->threads[i];
  388. if (p->state != STATE_INPUT_READY) {
  389. pthread_mutex_lock(&p->progress_mutex);
  390. while (p->state != STATE_INPUT_READY)
  391. pthread_cond_wait(&p->output_cond, &p->progress_mutex);
  392. pthread_mutex_unlock(&p->progress_mutex);
  393. }
  394. }
  395. }
  396. void ff_frame_thread_free(AVCodecContext *avctx, int thread_count)
  397. {
  398. FrameThreadContext *fctx = avctx->internal->thread_ctx;
  399. const AVCodec *codec = avctx->codec;
  400. int i;
  401. park_frame_worker_threads(fctx, thread_count);
  402. if (fctx->prev_thread && fctx->prev_thread != fctx->threads)
  403. update_context_from_thread(fctx->threads->avctx, fctx->prev_thread->avctx, 0);
  404. fctx->die = 1;
  405. for (i = 0; i < thread_count; i++) {
  406. PerThreadContext *p = &fctx->threads[i];
  407. pthread_mutex_lock(&p->mutex);
  408. pthread_cond_signal(&p->input_cond);
  409. pthread_mutex_unlock(&p->mutex);
  410. if (p->thread_init)
  411. pthread_join(p->thread, NULL);
  412. if (codec->close)
  413. codec->close(p->avctx);
  414. avctx->codec = NULL;
  415. release_delayed_buffers(p);
  416. av_frame_free(&p->frame);
  417. }
  418. for (i = 0; i < thread_count; i++) {
  419. PerThreadContext *p = &fctx->threads[i];
  420. pthread_mutex_destroy(&p->mutex);
  421. pthread_mutex_destroy(&p->progress_mutex);
  422. pthread_cond_destroy(&p->input_cond);
  423. pthread_cond_destroy(&p->progress_cond);
  424. pthread_cond_destroy(&p->output_cond);
  425. av_packet_unref(&p->avpkt);
  426. av_freep(&p->released_buffers);
  427. if (i) {
  428. av_freep(&p->avctx->priv_data);
  429. av_freep(&p->avctx->slice_offset);
  430. }
  431. av_freep(&p->avctx->internal);
  432. av_freep(&p->avctx);
  433. }
  434. av_freep(&fctx->threads);
  435. pthread_mutex_destroy(&fctx->buffer_mutex);
  436. av_freep(&avctx->internal->thread_ctx);
  437. }
  438. int ff_frame_thread_init(AVCodecContext *avctx)
  439. {
  440. int thread_count = avctx->thread_count;
  441. const AVCodec *codec = avctx->codec;
  442. AVCodecContext *src = avctx;
  443. FrameThreadContext *fctx;
  444. int i, err = 0;
  445. #if HAVE_W32THREADS
  446. w32thread_init();
  447. #endif
  448. if (!thread_count) {
  449. int nb_cpus = av_cpu_count();
  450. av_log(avctx, AV_LOG_DEBUG, "detected %d logical cores\n", nb_cpus);
  451. // use number of cores + 1 as thread count if there is more than one
  452. if (nb_cpus > 1)
  453. thread_count = avctx->thread_count = FFMIN(nb_cpus + 1, MAX_AUTO_THREADS);
  454. else
  455. thread_count = avctx->thread_count = 1;
  456. }
  457. if (thread_count <= 1) {
  458. avctx->active_thread_type = 0;
  459. return 0;
  460. }
  461. avctx->internal->thread_ctx = fctx = av_mallocz(sizeof(FrameThreadContext));
  462. if (!fctx)
  463. return AVERROR(ENOMEM);
  464. fctx->threads = av_mallocz(sizeof(PerThreadContext) * thread_count);
  465. if (!fctx->threads) {
  466. av_freep(&avctx->internal->thread_ctx);
  467. return AVERROR(ENOMEM);
  468. }
  469. pthread_mutex_init(&fctx->buffer_mutex, NULL);
  470. fctx->delaying = 1;
  471. for (i = 0; i < thread_count; i++) {
  472. AVCodecContext *copy = av_malloc(sizeof(AVCodecContext));
  473. PerThreadContext *p = &fctx->threads[i];
  474. pthread_mutex_init(&p->mutex, NULL);
  475. pthread_mutex_init(&p->progress_mutex, NULL);
  476. pthread_cond_init(&p->input_cond, NULL);
  477. pthread_cond_init(&p->progress_cond, NULL);
  478. pthread_cond_init(&p->output_cond, NULL);
  479. p->frame = av_frame_alloc();
  480. if (!p->frame) {
  481. av_freep(&copy);
  482. err = AVERROR(ENOMEM);
  483. goto error;
  484. }
  485. p->parent = fctx;
  486. p->avctx = copy;
  487. if (!copy) {
  488. err = AVERROR(ENOMEM);
  489. goto error;
  490. }
  491. *copy = *src;
  492. copy->internal = av_malloc(sizeof(AVCodecInternal));
  493. if (!copy->internal) {
  494. err = AVERROR(ENOMEM);
  495. goto error;
  496. }
  497. *copy->internal = *src->internal;
  498. copy->internal->thread_ctx = p;
  499. copy->internal->pkt = &p->avpkt;
  500. if (!i) {
  501. src = copy;
  502. if (codec->init)
  503. err = codec->init(copy);
  504. update_context_from_thread(avctx, copy, 1);
  505. } else {
  506. copy->priv_data = av_malloc(codec->priv_data_size);
  507. if (!copy->priv_data) {
  508. err = AVERROR(ENOMEM);
  509. goto error;
  510. }
  511. memcpy(copy->priv_data, src->priv_data, codec->priv_data_size);
  512. copy->internal->is_copy = 1;
  513. if (codec->init_thread_copy)
  514. err = codec->init_thread_copy(copy);
  515. }
  516. if (err) goto error;
  517. if (!pthread_create(&p->thread, NULL, frame_worker_thread, p))
  518. p->thread_init = 1;
  519. }
  520. return 0;
  521. error:
  522. ff_frame_thread_free(avctx, i+1);
  523. return err;
  524. }
  525. void ff_thread_flush(AVCodecContext *avctx)
  526. {
  527. int i;
  528. FrameThreadContext *fctx = avctx->internal->thread_ctx;
  529. if (!fctx) return;
  530. park_frame_worker_threads(fctx, avctx->thread_count);
  531. if (fctx->prev_thread) {
  532. if (fctx->prev_thread != &fctx->threads[0])
  533. update_context_from_thread(fctx->threads[0].avctx, fctx->prev_thread->avctx, 0);
  534. }
  535. fctx->next_decoding = fctx->next_finished = 0;
  536. fctx->delaying = 1;
  537. fctx->prev_thread = NULL;
  538. for (i = 0; i < avctx->thread_count; i++) {
  539. PerThreadContext *p = &fctx->threads[i];
  540. // Make sure decode flush calls with size=0 won't return old frames
  541. p->got_frame = 0;
  542. av_frame_unref(p->frame);
  543. release_delayed_buffers(p);
  544. if (avctx->codec->flush)
  545. avctx->codec->flush(p->avctx);
  546. }
  547. }
  548. int ff_thread_get_buffer(AVCodecContext *avctx, ThreadFrame *f, int flags)
  549. {
  550. PerThreadContext *p = avctx->internal->thread_ctx;
  551. int err;
  552. f->owner = avctx;
  553. if (!(avctx->active_thread_type & FF_THREAD_FRAME))
  554. return ff_get_buffer(avctx, f->f, flags);
  555. if (p->state != STATE_SETTING_UP &&
  556. (avctx->codec->update_thread_context || !avctx->thread_safe_callbacks)) {
  557. av_log(avctx, AV_LOG_ERROR, "get_buffer() cannot be called after ff_thread_finish_setup()\n");
  558. return -1;
  559. }
  560. if (avctx->internal->allocate_progress) {
  561. int *progress;
  562. f->progress = av_buffer_alloc(2 * sizeof(int));
  563. if (!f->progress) {
  564. return AVERROR(ENOMEM);
  565. }
  566. progress = (int*)f->progress->data;
  567. progress[0] = progress[1] = -1;
  568. }
  569. pthread_mutex_lock(&p->parent->buffer_mutex);
  570. if (avctx->thread_safe_callbacks ||
  571. avctx->get_buffer2 == avcodec_default_get_buffer2) {
  572. err = ff_get_buffer(avctx, f->f, flags);
  573. } else {
  574. p->requested_frame = f->f;
  575. p->requested_flags = flags;
  576. p->state = STATE_GET_BUFFER;
  577. pthread_mutex_lock(&p->progress_mutex);
  578. pthread_cond_signal(&p->progress_cond);
  579. while (p->state != STATE_SETTING_UP)
  580. pthread_cond_wait(&p->progress_cond, &p->progress_mutex);
  581. err = p->result;
  582. pthread_mutex_unlock(&p->progress_mutex);
  583. }
  584. if (!avctx->thread_safe_callbacks && !avctx->codec->update_thread_context)
  585. ff_thread_finish_setup(avctx);
  586. if (err)
  587. av_buffer_unref(&f->progress);
  588. pthread_mutex_unlock(&p->parent->buffer_mutex);
  589. return err;
  590. }
  591. void ff_thread_release_buffer(AVCodecContext *avctx, ThreadFrame *f)
  592. {
  593. PerThreadContext *p = avctx->internal->thread_ctx;
  594. FrameThreadContext *fctx;
  595. AVFrame *dst, *tmp;
  596. int can_direct_free = !(avctx->active_thread_type & FF_THREAD_FRAME) ||
  597. avctx->thread_safe_callbacks ||
  598. avctx->get_buffer2 == avcodec_default_get_buffer2;
  599. if (!f->f || !f->f->buf[0])
  600. return;
  601. if (avctx->debug & FF_DEBUG_BUFFERS)
  602. av_log(avctx, AV_LOG_DEBUG, "thread_release_buffer called on pic %p\n", f);
  603. av_buffer_unref(&f->progress);
  604. f->owner = NULL;
  605. if (can_direct_free) {
  606. av_frame_unref(f->f);
  607. return;
  608. }
  609. fctx = p->parent;
  610. pthread_mutex_lock(&fctx->buffer_mutex);
  611. if (p->num_released_buffers + 1 >= INT_MAX / sizeof(*p->released_buffers))
  612. goto fail;
  613. tmp = av_fast_realloc(p->released_buffers, &p->released_buffers_allocated,
  614. (p->num_released_buffers + 1) *
  615. sizeof(*p->released_buffers));
  616. if (!tmp)
  617. goto fail;
  618. p->released_buffers = tmp;
  619. dst = &p->released_buffers[p->num_released_buffers];
  620. av_frame_move_ref(dst, f->f);
  621. p->num_released_buffers++;
  622. fail:
  623. pthread_mutex_unlock(&fctx->buffer_mutex);
  624. }