FFmpeg
|
#include <stdatomic.h>
#include "avcodec.h"
#include "avcodec_internal.h"
#include "codec_desc.h"
#include "codec_internal.h"
#include "decode.h"
#include "hwaccel_internal.h"
#include "hwconfig.h"
#include "internal.h"
#include "packet_internal.h"
#include "pthread_internal.h"
#include "libavutil/refstruct.h"
#include "thread.h"
#include "threadframe.h"
#include "version_major.h"
#include "libavutil/avassert.h"
#include "libavutil/buffer.h"
#include "libavutil/common.h"
#include "libavutil/cpu.h"
#include "libavutil/frame.h"
#include "libavutil/internal.h"
#include "libavutil/log.h"
#include "libavutil/mem.h"
#include "libavutil/opt.h"
#include "libavutil/thread.h"
Go to the source code of this file.
Data Structures | |
struct | DecodedFrames |
struct | ThreadFrameProgress |
struct | PerThreadContext |
Context used by codec threads and stored in their AVCodecInternal thread_ctx. More... | |
struct | FrameThreadContext |
Context stored in the client AVCodecInternal thread_ctx. More... | |
Macros | |
#define | OFF(member) offsetof(FrameThreadContext, member) |
#define | OFF(member) offsetof(PerThreadContext, member) |
Enumerations | |
enum | { STATE_INPUT_READY, STATE_SETTING_UP, STATE_SETUP_FINISHED } |
enum | { UNINITIALIZED, NEEDS_CLOSE, INITIALIZED } |
Functions | |
static int | hwaccel_serial (const AVCodecContext *avctx) |
static void | async_lock (FrameThreadContext *fctx) |
static void | async_unlock (FrameThreadContext *fctx) |
static void | thread_set_name (PerThreadContext *p) |
static AVFrame * | decoded_frames_get_free (DecodedFrames *df) |
static void | decoded_frames_pop (DecodedFrames *df, AVFrame *dst) |
static void | decoded_frames_flush (DecodedFrames *df) |
static void | decoded_frames_free (DecodedFrames *df) |
static attribute_align_arg void * | frame_worker_thread (void *arg) |
Codec worker thread. More... | |
static int | update_context_from_thread (AVCodecContext *dst, const AVCodecContext *src, int for_user) |
Update the next thread's AVCodecContext with values from the reference thread's context. More... | |
static int | update_context_from_user (AVCodecContext *dst, const AVCodecContext *src) |
Update the next thread's AVCodecContext with values set by the user. More... | |
static int | submit_packet (PerThreadContext *p, AVCodecContext *user_avctx, AVPacket *in_pkt) |
int | ff_thread_receive_frame (AVCodecContext *avctx, AVFrame *frame) |
Submit available packets for decoding to worker threads, return a decoded frame if available. More... | |
void | ff_thread_report_progress (ThreadFrame *f, int n, int field) |
Notify later decoding threads when part of their reference picture is ready. More... | |
void | ff_thread_await_progress (const ThreadFrame *f, int n, int field) |
Wait for earlier decoding threads to finish reference pictures. More... | |
void | ff_thread_finish_setup (AVCodecContext *avctx) |
If the codec defines update_thread_context(), call this when they are ready for the next thread to start decoding the next frame. More... | |
static void | park_frame_worker_threads (FrameThreadContext *fctx, int thread_count) |
Waits for all threads to finish. More... | |
DEFINE_OFFSET_ARRAY (FrameThreadContext, thread_ctx, pthread_init_cnt,(OFF(buffer_mutex), OFF(hwaccel_mutex), OFF(async_mutex)),(OFF(async_cond))) | |
DEFINE_OFFSET_ARRAY (PerThreadContext, per_thread, pthread_init_cnt,(OFF(progress_mutex), OFF(mutex)),(OFF(input_cond), OFF(progress_cond), OFF(output_cond))) | |
void | ff_frame_thread_free (AVCodecContext *avctx, int thread_count) |
static av_cold int | init_thread (PerThreadContext *p, int *threads_to_free, FrameThreadContext *fctx, AVCodecContext *avctx, const FFCodec *codec, int first) |
int | ff_frame_thread_init (AVCodecContext *avctx) |
void | ff_thread_flush (AVCodecContext *avctx) |
Wait for decoding threads to finish and reset internal state. More... | |
int | ff_thread_can_start_frame (AVCodecContext *avctx) |
static int | thread_get_buffer_internal (AVCodecContext *avctx, AVFrame *f, int flags) |
int | ff_thread_get_buffer (AVCodecContext *avctx, AVFrame *f, int flags) |
Wrapper around get_buffer() for frame-multithreaded codecs. More... | |
int | ff_thread_get_ext_buffer (AVCodecContext *avctx, ThreadFrame *f, int flags) |
Wrapper around ff_get_buffer() for frame-multithreaded codecs. More... | |
void | ff_thread_release_ext_buffer (ThreadFrame *f) |
Unref a ThreadFrame. More... | |
enum ThreadingStatus | ff_thread_sync_ref (AVCodecContext *avctx, size_t offset) |
Allows to synchronize objects whose lifetime is the whole decoding process among all frame threads. More... | |
int | ff_thread_get_packet (AVCodecContext *avctx, AVPacket *pkt) |
Get a packet for decoding. More... | |
Frame multithreading support functions
Definition in file pthread_frame.c.
#define OFF | ( | member | ) | offsetof(FrameThreadContext, member) |
Definition at line 743 of file pthread_frame.c.
#define OFF | ( | member | ) | offsetof(PerThreadContext, member) |
Definition at line 743 of file pthread_frame.c.
anonymous enum |
Enumerator | |
---|---|
STATE_INPUT_READY | Set when the thread is awaiting a packet. |
STATE_SETTING_UP | Set before the codec has called ff_thread_finish_setup(). |
STATE_SETUP_FINISHED | Set after the codec has called ff_thread_finish_setup(). |
Definition at line 53 of file pthread_frame.c.
anonymous enum |
Enumerator | |
---|---|
UNINITIALIZED | Thread has not been created, AVCodec->close mustn't be called. |
NEEDS_CLOSE | FFCodec->close needs to be called. |
INITIALIZED | Thread has been properly set up. |
Definition at line 62 of file pthread_frame.c.
|
static |
Definition at line 161 of file pthread_frame.c.
Referenced by ff_thread_finish_setup(), and frame_worker_thread().
|
static |
Definition at line 166 of file pthread_frame.c.
Referenced by ff_thread_finish_setup(), ff_thread_receive_frame(), and park_frame_worker_threads().
|
static |
Definition at line 175 of file pthread_frame.c.
Referenced by ff_thread_receive_frame(), frame_worker_thread(), and park_frame_worker_threads().
|
static |
Definition at line 184 of file pthread_frame.c.
Referenced by frame_worker_thread().
|
static |
Definition at line 196 of file pthread_frame.c.
Referenced by frame_worker_thread().
|
static |
Definition at line 217 of file pthread_frame.c.
Referenced by ff_thread_receive_frame().
|
static |
Definition at line 225 of file pthread_frame.c.
Referenced by ff_thread_flush().
|
static |
Definition at line 232 of file pthread_frame.c.
Referenced by ff_frame_thread_free().
|
static |
Codec worker thread.
Automatically calls ff_thread_finish_setup() if the codec does not provide an update_thread_context method, or if the codec returns before calling it.
Definition at line 248 of file pthread_frame.c.
Referenced by init_thread().
|
static |
Update the next thread's AVCodecContext with values from the reference thread's context.
dst | The destination context. |
src | The source context. |
for_user | 0 if the destination is a codec thread, 1 if the destination is the user's thread |
Definition at line 350 of file pthread_frame.c.
Referenced by ff_thread_flush(), ff_thread_receive_frame(), init_thread(), and submit_packet().
|
static |
Update the next thread's AVCodecContext with values set by the user.
dst | The destination context. |
src | The source context. |
Definition at line 472 of file pthread_frame.c.
Referenced by submit_packet().
|
static |
Definition at line 502 of file pthread_frame.c.
Referenced by ff_thread_receive_frame().
int ff_thread_receive_frame | ( | struct AVCodecContext * | avctx, |
AVFrame * | frame | ||
) |
Submit available packets for decoding to worker threads, return a decoded frame if available.
Returns AVERROR(EAGAIN) if none is available.
Parameters are the same as FFCodec.receive_frame.
Definition at line 566 of file pthread_frame.c.
void ff_thread_report_progress | ( | ThreadFrame * | f, |
int | progress, | ||
int | field | ||
) |
Notify later decoding threads when part of their reference picture is ready.
Call this when some part of the picture is finished decoding. Later calls with lower values of progress have no effect.
f | The picture being decoded. |
progress | Value, in arbitrary units, of how much of the picture has decoded. |
field | The field being decoded, for field-picture codecs. 0 for top field or frame pictures, 1 for bottom field. |
Definition at line 627 of file pthread_frame.c.
Referenced by decode_finish_row(), decode_nal_units(), ff_h264_field_end(), ff_h264_queue_decode_slice(), and h264_field_start().
void ff_thread_await_progress | ( | const ThreadFrame * | f, |
int | progress, | ||
int | field | ||
) |
Wait for earlier decoding threads to finish reference pictures.
Call this before accessing some part of a picture, with a given value for progress, and it will return after the responsible decoding thread calls ff_thread_report_progress() with the same or higher value for progress.
f | The picture being referenced. |
progress | Value, in arbitrary units, to wait for. |
field | The field being referenced, for field-picture codecs. 0 for top field or frame pictures, 1 for bottom field. |
Definition at line 650 of file pthread_frame.c.
Referenced by await_reference_mb_row(), await_references(), guess_mv(), and h264_field_start().
void ff_thread_finish_setup | ( | AVCodecContext * | avctx | ) |
If the codec defines update_thread_context(), call this when they are ready for the next thread to start decoding the next frame.
After calling it, do not change any variables read by the update_thread_context() method, or call ff_thread_get_buffer().
avctx | The context. |
Definition at line 671 of file pthread_frame.c.
Referenced by frame_worker_thread().
|
static |
Waits for all threads to finish.
Definition at line 717 of file pthread_frame.c.
Referenced by ff_frame_thread_free(), and ff_thread_flush().
DEFINE_OFFSET_ARRAY | ( | FrameThreadContext | , |
thread_ctx | , | ||
pthread_init_cnt | , | ||
(OFF(buffer_mutex), OFF(hwaccel_mutex), OFF(async_mutex)) | , | ||
(OFF(async_cond)) | |||
) |
DEFINE_OFFSET_ARRAY | ( | PerThreadContext | , |
per_thread | , | ||
pthread_init_cnt | , | ||
(OFF(progress_mutex), OFF(mutex)) | , | ||
(OFF(input_cond), OFF(progress_cond), OFF(output_cond)) | |||
) |
void ff_frame_thread_free | ( | AVCodecContext * | avctx, |
int | thread_count | ||
) |
Definition at line 749 of file pthread_frame.c.
Referenced by ff_frame_thread_init(), and ff_thread_free().
|
static |
Definition at line 817 of file pthread_frame.c.
Referenced by ff_frame_thread_init().
int ff_frame_thread_init | ( | AVCodecContext * | avctx | ) |
Definition at line 921 of file pthread_frame.c.
Referenced by ff_thread_init().
void ff_thread_flush | ( | struct AVCodecContext * | avctx | ) |
Wait for decoding threads to finish and reset internal state.
Called by avcodec_flush_buffers().
avctx | The context. |
Definition at line 984 of file pthread_frame.c.
Referenced by avcodec_flush_buffers().
int ff_thread_can_start_frame | ( | AVCodecContext * | avctx | ) |
Definition at line 1013 of file pthread_frame.c.
Referenced by ff_mpv_frame_start(), and h264_frame_start().
|
static |
Definition at line 1026 of file pthread_frame.c.
Referenced by ff_thread_get_buffer().
int ff_thread_get_buffer | ( | AVCodecContext * | avctx, |
AVFrame * | f, | ||
int | flags | ||
) |
Wrapper around get_buffer() for frame-multithreaded codecs.
Call this function instead of ff_get_buffer(f). Cannot be called after the codec has called ff_thread_finish_setup().
avctx | The current context. |
f | The frame to write into. |
Definition at line 1049 of file pthread_frame.c.
Referenced by aic_decode_frame(), alloc_frame(), alloc_picture(), bitpacked_decode_yuv422p10(), cfhd_decode(), cllc_decode_frame(), cri_decode_frame(), decode_blocks(), decode_element(), decode_entropy_coded_image(), decode_frame(), decode_idat_chunk(), dnxhd_decode_frame(), dvvideo_decode_frame(), dxtory_decode_v1_410(), dxtory_decode_v1_420(), dxtory_decode_v1_444(), dxtory_decode_v1_rgb(), dxtory_decode_v2(), dxv_decode(), ff_progress_frame_get_buffer(), ff_thread_get_ext_buffer(), flac_decode_frame(), hap_decode(), hdr_decode_frame(), hevc_frame_start(), hqx_decode_frame(), init_image(), jpeg2000_decode_frame(), lag_decode_frame(), magy_decode_frame(), photocd_decode_frame(), pixlet_decode_frame(), qoi_decode_frame(), speedhq_decode_frame(), tak_decode_frame(), tta_decode_frame(), v410_decode_frame(), vble_decode_frame(), wavpack_decode_block(), and wbmp_decode_frame().
int ff_thread_get_ext_buffer | ( | AVCodecContext * | avctx, |
ThreadFrame * | f, | ||
int | flags | ||
) |
Wrapper around ff_get_buffer() for frame-multithreaded codecs.
Call this function instead of ff_get_buffer() if you might need to wait for progress on this frame. Cannot be called after the codec has called ff_thread_finish_setup().
avctx | The current context. |
f | The frame to write into. |
Definition at line 1057 of file pthread_frame.c.
Referenced by alloc_picture().
void ff_thread_release_ext_buffer | ( | ThreadFrame * | f | ) |
Unref a ThreadFrame.
This is basically a wrapper around av_frame_unref() and should be called instead of it.
avctx | The current context. |
f | The picture being released. |
Definition at line 1078 of file pthread_frame.c.
Referenced by ff_h264_unref_picture(), and h264_field_start().
enum ThreadingStatus ff_thread_sync_ref | ( | AVCodecContext * | avctx, |
size_t | offset | ||
) |
Allows to synchronize objects whose lifetime is the whole decoding process among all frame threads.
When called from a non-copy thread, do nothing. When called from another thread, place a new RefStruct reference at the given offset in the calling thread's private data from the RefStruct reference in the private data of the first decoding thread. The first thread must have a valid RefStruct reference at the given offset in its private data; the calling thread must not have a reference at this offset in its private data (must be NULL).
avctx | an AVCodecContext |
offset | offset of the RefStruct reference in avctx's private data |
FF_THREAD_IS_COPY | if frame-threading is in use and the calling thread is a copy; in this case, the RefStruct reference will be set. |
FF_THREAD_IS_MAIN_THREAD | if frame-threading is in use and the calling thread is the main thread. |
FF_THREAD_NO_FRAME_THREADING | if frame-threading is not in use. |
Definition at line 1086 of file pthread_frame.c.
int ff_thread_get_packet | ( | struct AVCodecContext * | avctx, |
AVPacket * | pkt | ||
) |
Get a packet for decoding.
This gets invoked by the worker threads.
Definition at line 1106 of file pthread_frame.c.