#include "framepool.h"
#include "libavutil/avassert.h"
+#include "libavutil/avutil.h"
#include "libavutil/buffer.h"
#include "libavutil/frame.h"
#include "libavutil/imgutils.h"
#include "libavutil/mem.h"
#include "libavutil/pixfmt.h"
-struct FFVideoFramePool {
+struct FFFramePool {
+ enum AVMediaType type;
+
+ /* video */
int width;
int height;
+
+ /* audio */
+ int planes;
+ int channels;
+ int nb_samples;
+
+ /* common */
int format;
int align;
int linesize[4];
};
-FFVideoFramePool *ff_video_frame_pool_init(AVBufferRef* (*alloc)(int size),
- int width,
- int height,
- enum AVPixelFormat format,
- int align)
+FFFramePool *ff_frame_pool_video_init(AVBufferRef* (*alloc)(int size),
+ int width,
+ int height,
+ enum AVPixelFormat format,
+ int align)
{
int i, ret;
- FFVideoFramePool *pool;
+ FFFramePool *pool;
const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(format);
if (!desc)
return NULL;
- pool = av_mallocz(sizeof(FFVideoFramePool));
+ pool = av_mallocz(sizeof(FFFramePool));
if (!pool)
return NULL;
+ pool->type = AVMEDIA_TYPE_VIDEO;
pool->width = width;
pool->height = height;
pool->format = format;
return pool;
fail:
- ff_video_frame_pool_uninit(&pool);
+ ff_frame_pool_uninit(&pool);
return NULL;
}
-int ff_video_frame_pool_get_config(FFVideoFramePool *pool,
+FFFramePool *ff_frame_pool_audio_init(AVBufferRef* (*alloc)(int size),
+ int channels,
+ int nb_samples,
+ enum AVSampleFormat format,
+ int align)
+{
+ int ret, planar;
+ FFFramePool *pool;
+
+ pool = av_mallocz(sizeof(FFFramePool));
+ if (!pool)
+ return NULL;
+
+ planar = av_sample_fmt_is_planar(format);
+
+ pool->type = AVMEDIA_TYPE_AUDIO;
+ pool->planes = planar ? channels : 1;
+ pool->channels = channels;
+ pool->nb_samples = nb_samples;
+ pool->format = format;
+ pool->align = align;
+
+ ret = av_samples_get_buffer_size(&pool->linesize[0], channels,
+ nb_samples, format, 0);
+ if (ret < 0)
+ goto fail;
+
+ pool->pools[0] = av_buffer_pool_init(pool->linesize[0], NULL);
+ if (!pool->pools[0])
+ goto fail;
+
+ return pool;
+
+fail:
+ ff_frame_pool_uninit(&pool);
+ return NULL;
+}
+
+int ff_frame_pool_get_video_config(FFFramePool *pool,
int *width,
int *height,
enum AVPixelFormat *format,
if (!pool)
return AVERROR(EINVAL);
+ av_assert0(pool->type == AVMEDIA_TYPE_VIDEO);
+
*width = pool->width;
*height = pool->height;
*format = pool->format;
return 0;
}
+int ff_frame_pool_get_audio_config(FFFramePool *pool,
+ int *channels,
+ int *nb_samples,
+ enum AVSampleFormat *format,
+ int *align)
+{
+ if (!pool)
+ return AVERROR(EINVAL);
+
+ av_assert0(pool->type == AVMEDIA_TYPE_AUDIO);
+
+ *channels = pool->channels;
+ *nb_samples = pool->nb_samples;
+ *format = pool->format;
+ *align = pool->align;
+
+ return 0;
+}
-AVFrame *ff_video_frame_pool_get(FFVideoFramePool *pool)
+AVFrame *ff_frame_pool_get(FFFramePool *pool)
{
int i;
AVFrame *frame;
return NULL;
}
- desc = av_pix_fmt_desc_get(pool->format);
- if (!desc) {
- goto fail;
- }
+ switch(pool->type) {
+ case AVMEDIA_TYPE_VIDEO:
+ desc = av_pix_fmt_desc_get(pool->format);
+ if (!desc) {
+ goto fail;
+ }
- frame->width = pool->width;
- frame->height = pool->height;
- frame->format = pool->format;
+ frame->width = pool->width;
+ frame->height = pool->height;
+ frame->format = pool->format;
- for (i = 0; i < 4; i++) {
- frame->linesize[i] = pool->linesize[i];
- if (!pool->pools[i])
- break;
+ for (i = 0; i < 4; i++) {
+ frame->linesize[i] = pool->linesize[i];
+ if (!pool->pools[i])
+ break;
- frame->buf[i] = av_buffer_pool_get(pool->pools[i]);
- if (!frame->buf[i]) {
- goto fail;
+ frame->buf[i] = av_buffer_pool_get(pool->pools[i]);
+ if (!frame->buf[i])
+ goto fail;
+
+ frame->data[i] = frame->buf[i]->data;
}
- frame->data[i] = frame->buf[i]->data;
- }
+ if (desc->flags & AV_PIX_FMT_FLAG_PAL ||
+ desc->flags & AV_PIX_FMT_FLAG_PSEUDOPAL) {
+ enum AVPixelFormat format =
+ pool->format == AV_PIX_FMT_PAL8 ? AV_PIX_FMT_BGR8 : pool->format;
- if (desc->flags & AV_PIX_FMT_FLAG_PAL ||
- desc->flags & AV_PIX_FMT_FLAG_PSEUDOPAL) {
- enum AVPixelFormat format =
- pool->format == AV_PIX_FMT_PAL8 ? AV_PIX_FMT_BGR8 : pool->format;
+ av_assert0(frame->data[1] != NULL);
+ if (avpriv_set_systematic_pal2((uint32_t *)frame->data[1], format) < 0)
+ goto fail;
+ }
- av_assert0(frame->data[1] != NULL);
- if (avpriv_set_systematic_pal2((uint32_t *)frame->data[1], format) < 0) {
- goto fail;
+ frame->extended_data = frame->data;
+ break;
+ case AVMEDIA_TYPE_AUDIO:
+ frame->nb_samples = pool->nb_samples;
+ av_frame_set_channels(frame, pool->channels);
+ frame->format = pool->format;
+ frame->linesize[0] = pool->linesize[0];
+
+ if (pool->planes > AV_NUM_DATA_POINTERS) {
+ frame->extended_data = av_mallocz_array(pool->planes,
+ sizeof(*frame->extended_data));
+ frame->nb_extended_buf = pool->planes - AV_NUM_DATA_POINTERS;
+ frame->extended_buf = av_mallocz_array(frame->nb_extended_buf,
+ sizeof(*frame->extended_buf));
+ if (!frame->extended_data || !frame->extended_buf)
+ goto fail;
+ } else {
+ frame->extended_data = frame->data;
+ av_assert0(frame->nb_extended_buf == 0);
}
- }
- frame->extended_data = frame->data;
+ for (i = 0; i < FFMIN(pool->planes, AV_NUM_DATA_POINTERS); i++) {
+ frame->buf[i] = av_buffer_pool_get(pool->pools[0]);
+ if (!frame->buf[i])
+ goto fail;
+ frame->extended_data[i] = frame->data[i] = frame->buf[i]->data;
+ }
+ for (i = 0; i < frame->nb_extended_buf; i++) {
+ frame->extended_buf[i] = av_buffer_pool_get(pool->pools[0]);
+ if (!frame->extended_buf[i])
+ goto fail;
+ frame->extended_data[i + AV_NUM_DATA_POINTERS] = frame->extended_buf[i]->data;
+ }
+
+ break;
+ default:
+ av_assert0(0);
+ }
return frame;
fail:
return NULL;
}
-void ff_video_frame_pool_uninit(FFVideoFramePool **pool)
+void ff_frame_pool_uninit(FFFramePool **pool)
{
int i;