virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-12-23 12:43:46 +02:00
Code Issues Releases Activity

avformat/av1: Improve filtering AV1 OBUs

Browse Source 
Both ISOBMFF as well as Matroska require certain OBUs to be stripped
before muxing them. There are two functions for this purpose; one writes
directly into an AVIOContext, the other returns a freshly allocated
buffer with the undesired units stripped away.

The latter one actually relies on the former by means of a dynamic
buffer. This has several drawbacks: The underlying buffer might have to
be reallocated multiple times; the buffer will eventually be
overallocated; the data will not be directly copied into the final
buffer, but rather first in the write buffer (in chunks of 1024 byte)
and then written in these chunks. Moreover, the API for dynamic buffers
is defective wrt error checking and as a consequence, the earlier code
would indicate a length of -AV_INPUT_BUFFER_PADDING_SIZE on allocation
failure, but it would not return an error; there would also be no error
in case the arbitrary limit of INT_MAX/2 that is currently imposed on
dynamic buffers is hit.

This commit changes this: The buffer is now parsed twice, once to get
the precise length which will then be allocated; and once to actually
write the data.

For a 22.7mb/s file with average framesize 113 kB this improved the time
for the calls to ff_av1_filter_obus_buf() when writing Matroska from
753662 decicycles to 313319 decicycles (based upon 50 runs a 2048 frames
each); for another 1.5mb/s file (with average framesize of 7.3 kB) it
improved from 79270 decicycles to 34539 decicycles (based upon 50 runs a
4096 frames).

Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@gmail.com>
Signed-off-by: James Almer <jamrial@gmail.com>
This commit is contained in:
Andreas Rheinhardt 2020-01-23 17:08:32 +01:00 committed by James Almer
parent e8518c2580
commit d4bbc0db01
2 changed files with 23 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
libavformat/av1.c
View File

@ -19,6 +19,7 @@
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
#include "libavutil/avassert.h"
#include "libavutil/mem.h"
#include "libavcodec/av1.h"
#include "libavcodec/av1_parse.h"
@ -48,7 +49,8 @@ int ff_av1_filter_obus(AVIOContext *pb, const uint8_t *buf, int size)
case AV1_OBU_PADDING:
break;
default:
avio_write(pb, buf, len);
if (pb)
avio_write(pb, buf, len);
size += len;
break;
}
@ -58,23 +60,31 @@ int ff_av1_filter_obus(AVIOContext *pb, const uint8_t *buf, int size)
return size;
}
int ff_av1_filter_obus_buf(const uint8_t *buf, uint8_t **out, int *size)
int ff_av1_filter_obus_buf(const uint8_t *in, uint8_t **out, int *size)
{
AVIOContext *pb;
int ret;
AVIOContext pb;
uint8_t *buf;
int len, ret;
ret = avio_open_dyn_buf(&pb);
if (ret < 0)
return ret;
ret = ff_av1_filter_obus(pb, buf, *size);
len = ret = ff_av1_filter_obus(NULL, in, *size);
if (ret < 0) {
ffio_free_dyn_buf(&pb);
return ret;
}
buf = av_malloc(len + AV_INPUT_BUFFER_PADDING_SIZE);
if (!buf)
return AVERROR(ENOMEM);
ffio_init_context(&pb, buf, len, 1, NULL, NULL, NULL, NULL);
ret = ff_av1_filter_obus(&pb, in, *size);
av_assert1(ret == len);
memset(buf + len, 0, AV_INPUT_BUFFER_PADDING_SIZE);
av_freep(out);
*size = avio_close_dyn_buf(pb, out);
*out = buf;
*size = len;
return 0;
}

4
libavformat/av1.h
View File

@ -61,7 +61,7 @@ int ff_av1_filter_obus(AVIOContext *pb, const uint8_t *buf, int size);
*
* @param pb pointer to the AVIOContext where the filtered bitstream shall be
* written
* @param buf input data buffer
* @param in input data buffer
* @param out pointer to pointer that will hold the allocated data buffer
* @param size size of the input data buffer. The size of the resulting output
data buffer will be written here
@ -69,7 +69,7 @@ int ff_av1_filter_obus(AVIOContext *pb, const uint8_t *buf, int size);
* @return 0 in case of success, a negative AVERROR code in case of failure.
* On failure, out and size are unchanged
*/
int ff_av1_filter_obus_buf(const uint8_t *buf, uint8_t **out, int *size);
int ff_av1_filter_obus_buf(const uint8_t *in, uint8_t **out, int *size);
/**
* Parses a Sequence Header from the the provided buffer.