You've already forked FFmpeg
mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2025-06-14 22:15:12 +02:00
lavf/async: switch to new FIFO API
This commit is contained in:
committed by
Andreas Rheinhardt
parent
70a1774f1a
commit
36117968ad
@ -47,7 +47,7 @@
|
|||||||
|
|
||||||
typedef struct RingBuffer
|
typedef struct RingBuffer
|
||||||
{
|
{
|
||||||
AVFifoBuffer *fifo;
|
AVFifo *fifo;
|
||||||
int read_back_capacity;
|
int read_back_capacity;
|
||||||
|
|
||||||
int read_pos;
|
int read_pos;
|
||||||
@ -83,7 +83,7 @@ typedef struct Context {
|
|||||||
static int ring_init(RingBuffer *ring, unsigned int capacity, int read_back_capacity)
|
static int ring_init(RingBuffer *ring, unsigned int capacity, int read_back_capacity)
|
||||||
{
|
{
|
||||||
memset(ring, 0, sizeof(RingBuffer));
|
memset(ring, 0, sizeof(RingBuffer));
|
||||||
ring->fifo = av_fifo_alloc(capacity + read_back_capacity);
|
ring->fifo = av_fifo_alloc2(capacity + read_back_capacity, 1, 0);
|
||||||
if (!ring->fifo)
|
if (!ring->fifo)
|
||||||
return AVERROR(ENOMEM);
|
return AVERROR(ENOMEM);
|
||||||
|
|
||||||
@ -93,45 +93,59 @@ static int ring_init(RingBuffer *ring, unsigned int capacity, int read_back_capa
|
|||||||
|
|
||||||
static void ring_destroy(RingBuffer *ring)
|
static void ring_destroy(RingBuffer *ring)
|
||||||
{
|
{
|
||||||
av_fifo_freep(&ring->fifo);
|
av_fifo_freep2(&ring->fifo);
|
||||||
}
|
}
|
||||||
|
|
||||||
static void ring_reset(RingBuffer *ring)
|
static void ring_reset(RingBuffer *ring)
|
||||||
{
|
{
|
||||||
av_fifo_reset(ring->fifo);
|
av_fifo_reset2(ring->fifo);
|
||||||
ring->read_pos = 0;
|
ring->read_pos = 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int ring_size(RingBuffer *ring)
|
static int ring_size(RingBuffer *ring)
|
||||||
{
|
{
|
||||||
return av_fifo_size(ring->fifo) - ring->read_pos;
|
return av_fifo_can_read(ring->fifo) - ring->read_pos;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int ring_space(RingBuffer *ring)
|
static int ring_space(RingBuffer *ring)
|
||||||
{
|
{
|
||||||
return av_fifo_space(ring->fifo);
|
return av_fifo_can_write(ring->fifo);
|
||||||
}
|
}
|
||||||
|
|
||||||
static int ring_generic_read(RingBuffer *ring, void *dest, int buf_size, void (*func)(void*, void*, int))
|
static int ring_read(RingBuffer *ring, void *dest, int buf_size)
|
||||||
{
|
{
|
||||||
int ret;
|
int ret = 0;
|
||||||
|
|
||||||
av_assert2(buf_size <= ring_size(ring));
|
av_assert2(buf_size <= ring_size(ring));
|
||||||
ret = av_fifo_generic_peek_at(ring->fifo, dest, ring->read_pos, buf_size, func);
|
if (dest)
|
||||||
|
ret = av_fifo_peek(ring->fifo, dest, buf_size, ring->read_pos);
|
||||||
ring->read_pos += buf_size;
|
ring->read_pos += buf_size;
|
||||||
|
|
||||||
if (ring->read_pos > ring->read_back_capacity) {
|
if (ring->read_pos > ring->read_back_capacity) {
|
||||||
av_fifo_drain(ring->fifo, ring->read_pos - ring->read_back_capacity);
|
av_fifo_drain2(ring->fifo, ring->read_pos - ring->read_back_capacity);
|
||||||
ring->read_pos = ring->read_back_capacity;
|
ring->read_pos = ring->read_back_capacity;
|
||||||
}
|
}
|
||||||
|
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int ring_generic_write(RingBuffer *ring, void *src, int size, int (*func)(void*, void*, int))
|
static int wrapped_url_read(void *src, void *dst, size_t *size)
|
||||||
|
{
|
||||||
|
URLContext *h = src;
|
||||||
|
Context *c = h->priv_data;
|
||||||
|
int ret;
|
||||||
|
|
||||||
|
ret = ffurl_read(c->inner, dst, *size);
|
||||||
|
*size = ret > 0 ? ret : 0;
|
||||||
|
c->inner_io_error = ret < 0 ? ret : 0;
|
||||||
|
|
||||||
|
return c->inner_io_error;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int ring_write(RingBuffer *ring, URLContext *h, size_t size)
|
||||||
{
|
{
|
||||||
av_assert2(size <= ring_space(ring));
|
av_assert2(size <= ring_space(ring));
|
||||||
return av_fifo_generic_write(ring->fifo, src, size, func);
|
return av_fifo_write_from_cb(ring->fifo, wrapped_url_read, h, &size);
|
||||||
}
|
}
|
||||||
|
|
||||||
static int ring_size_of_read_back(RingBuffer *ring)
|
static int ring_size_of_read_back(RingBuffer *ring)
|
||||||
@ -161,18 +175,6 @@ static int async_check_interrupt(void *arg)
|
|||||||
return c->abort_request;
|
return c->abort_request;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int wrapped_url_read(void *src, void *dst, int size)
|
|
||||||
{
|
|
||||||
URLContext *h = src;
|
|
||||||
Context *c = h->priv_data;
|
|
||||||
int ret;
|
|
||||||
|
|
||||||
ret = ffurl_read(c->inner, dst, size);
|
|
||||||
c->inner_io_error = ret < 0 ? ret : 0;
|
|
||||||
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void *async_buffer_task(void *arg)
|
static void *async_buffer_task(void *arg)
|
||||||
{
|
{
|
||||||
URLContext *h = arg;
|
URLContext *h = arg;
|
||||||
@ -221,7 +223,7 @@ static void *async_buffer_task(void *arg)
|
|||||||
pthread_mutex_unlock(&c->mutex);
|
pthread_mutex_unlock(&c->mutex);
|
||||||
|
|
||||||
to_copy = FFMIN(4096, fifo_space);
|
to_copy = FFMIN(4096, fifo_space);
|
||||||
ret = ring_generic_write(ring, (void *)h, to_copy, wrapped_url_read);
|
ret = ring_write(ring, h, to_copy);
|
||||||
|
|
||||||
pthread_mutex_lock(&c->mutex);
|
pthread_mutex_lock(&c->mutex);
|
||||||
if (ret <= 0) {
|
if (ret <= 0) {
|
||||||
@ -327,11 +329,11 @@ static int async_close(URLContext *h)
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int async_read_internal(URLContext *h, void *dest, int size, int read_complete,
|
static int async_read_internal(URLContext *h, void *dest, int size)
|
||||||
void (*func)(void*, void*, int))
|
|
||||||
{
|
{
|
||||||
Context *c = h->priv_data;
|
Context *c = h->priv_data;
|
||||||
RingBuffer *ring = &c->ring;
|
RingBuffer *ring = &c->ring;
|
||||||
|
int read_complete = !dest;
|
||||||
int to_read = size;
|
int to_read = size;
|
||||||
int ret = 0;
|
int ret = 0;
|
||||||
|
|
||||||
@ -346,8 +348,8 @@ static int async_read_internal(URLContext *h, void *dest, int size, int read_com
|
|||||||
fifo_size = ring_size(ring);
|
fifo_size = ring_size(ring);
|
||||||
to_copy = FFMIN(to_read, fifo_size);
|
to_copy = FFMIN(to_read, fifo_size);
|
||||||
if (to_copy > 0) {
|
if (to_copy > 0) {
|
||||||
ring_generic_read(ring, dest, to_copy, func);
|
ring_read(ring, dest, to_copy);
|
||||||
if (!func)
|
if (dest)
|
||||||
dest = (uint8_t *)dest + to_copy;
|
dest = (uint8_t *)dest + to_copy;
|
||||||
c->logical_pos += to_copy;
|
c->logical_pos += to_copy;
|
||||||
to_read -= to_copy;
|
to_read -= to_copy;
|
||||||
@ -376,11 +378,7 @@ static int async_read_internal(URLContext *h, void *dest, int size, int read_com
|
|||||||
|
|
||||||
static int async_read(URLContext *h, unsigned char *buf, int size)
|
static int async_read(URLContext *h, unsigned char *buf, int size)
|
||||||
{
|
{
|
||||||
return async_read_internal(h, buf, size, 0, NULL);
|
return async_read_internal(h, buf, size);
|
||||||
}
|
|
||||||
|
|
||||||
static void fifo_do_not_copy_func(void* dest, void* src, int size) {
|
|
||||||
// do not copy
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static int64_t async_seek(URLContext *h, int64_t pos, int whence)
|
static int64_t async_seek(URLContext *h, int64_t pos, int whence)
|
||||||
@ -422,7 +420,7 @@ static int64_t async_seek(URLContext *h, int64_t pos, int whence)
|
|||||||
|
|
||||||
if (pos_delta > 0) {
|
if (pos_delta > 0) {
|
||||||
// fast seek forwards
|
// fast seek forwards
|
||||||
async_read_internal(h, NULL, pos_delta, 1, fifo_do_not_copy_func);
|
async_read_internal(h, NULL, pos_delta);
|
||||||
} else {
|
} else {
|
||||||
// fast seek backwards
|
// fast seek backwards
|
||||||
ring_drain(ring, pos_delta);
|
ring_drain(ring, pos_delta);
|
||||||
|
Reference in New Issue
Block a user