2 * Copyright 2011 Stefano Sabatini | stefasab at gmail.com
4 * This file is part of FFmpeg.
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 * libavcodec/libavfilter gluing utilities
27 #include "libavutil/avassert.h"
28 #include "libavutil/opt.h"
30 int avfilter_copy_frame_props(AVFilterBufferRef *dst, const AVFrame *src)
33 dst->pos = av_frame_get_pkt_pos(src);
34 dst->format = src->format;
36 av_dict_free(&dst->metadata);
37 av_dict_copy(&dst->metadata, av_frame_get_metadata(src), 0);
40 case AVMEDIA_TYPE_VIDEO:
41 dst->video->w = src->width;
42 dst->video->h = src->height;
43 dst->video->sample_aspect_ratio = src->sample_aspect_ratio;
44 dst->video->interlaced = src->interlaced_frame;
45 dst->video->top_field_first = src->top_field_first;
46 dst->video->key_frame = src->key_frame;
47 dst->video->pict_type = src->pict_type;
48 av_freep(&dst->video->qp_table);
49 dst->video->qp_table_linesize = 0;
50 if (src->qscale_table) {
51 int qsize = src->qstride ? src->qstride * ((src->height+15)/16) : (src->width+15)/16;
52 dst->video->qp_table = av_malloc(qsize);
53 if (!dst->video->qp_table)
54 return AVERROR(ENOMEM);
55 dst->video->qp_table_linesize = src->qstride;
56 dst->video->qp_table_size = qsize;
57 memcpy(dst->video->qp_table, src->qscale_table, qsize);
60 case AVMEDIA_TYPE_AUDIO:
61 dst->audio->sample_rate = src->sample_rate;
62 dst->audio->channel_layout = src->channel_layout;
63 dst->audio->channels = src->channels;
64 if(src->channels < av_get_channel_layout_nb_channels(src->channel_layout)) {
65 av_log(0, AV_LOG_ERROR, "libavfilter does not support this channel layout\n");
66 return AVERROR(EINVAL);
70 return AVERROR(EINVAL);
76 AVFilterBufferRef *avfilter_get_video_buffer_ref_from_frame(const AVFrame *frame,
79 AVFilterBufferRef *picref =
80 avfilter_get_video_buffer_ref_from_arrays(frame->data, frame->linesize, perms,
81 frame->width, frame->height,
85 if (avfilter_copy_frame_props(picref, frame) < 0) {
86 picref->buf->data[0] = NULL;
87 avfilter_unref_bufferp(&picref);
92 AVFilterBufferRef *avfilter_get_audio_buffer_ref_from_frame(const AVFrame *frame,
95 AVFilterBufferRef *samplesref =
96 avfilter_get_audio_buffer_ref_from_arrays((uint8_t **)frame->data, frame->linesize[0], perms,
97 frame->nb_samples, frame->format,
98 av_frame_get_channel_layout(frame));
101 if (avfilter_copy_frame_props(samplesref, frame) < 0) {
102 samplesref->buf->data[0] = NULL;
103 avfilter_unref_bufferp(&samplesref);
108 AVFilterBufferRef *avfilter_get_buffer_ref_from_frame(enum AVMediaType type,
109 const AVFrame *frame,
113 case AVMEDIA_TYPE_VIDEO:
114 return avfilter_get_video_buffer_ref_from_frame(frame, perms);
115 case AVMEDIA_TYPE_AUDIO:
116 return avfilter_get_audio_buffer_ref_from_frame(frame, perms);
122 int avfilter_copy_buf_props(AVFrame *dst, const AVFilterBufferRef *src)
124 int planes, nb_channels;
127 return AVERROR(EINVAL);
128 /* abort in case the src is NULL and dst is not, avoid inconsistent state in dst */
131 memcpy(dst->data, src->data, sizeof(dst->data));
132 memcpy(dst->linesize, src->linesize, sizeof(dst->linesize));
135 dst->format = src->format;
136 av_frame_set_pkt_pos(dst, src->pos);
139 case AVMEDIA_TYPE_VIDEO:
140 av_assert0(src->video);
141 dst->width = src->video->w;
142 dst->height = src->video->h;
143 dst->sample_aspect_ratio = src->video->sample_aspect_ratio;
144 dst->interlaced_frame = src->video->interlaced;
145 dst->top_field_first = src->video->top_field_first;
146 dst->key_frame = src->video->key_frame;
147 dst->pict_type = src->video->pict_type;
149 case AVMEDIA_TYPE_AUDIO:
150 av_assert0(src->audio);
151 nb_channels = av_get_channel_layout_nb_channels(src->audio->channel_layout);
152 planes = av_sample_fmt_is_planar(src->format) ? nb_channels : 1;
154 if (planes > FF_ARRAY_ELEMS(dst->data)) {
155 dst->extended_data = av_mallocz(planes * sizeof(*dst->extended_data));
156 if (!dst->extended_data)
157 return AVERROR(ENOMEM);
158 memcpy(dst->extended_data, src->extended_data,
159 planes * sizeof(*dst->extended_data));
161 dst->extended_data = dst->data;
162 dst->nb_samples = src->audio->nb_samples;
163 av_frame_set_sample_rate (dst, src->audio->sample_rate);
164 av_frame_set_channel_layout(dst, src->audio->channel_layout);
165 av_frame_set_channels (dst, src->audio->channels);
168 return AVERROR(EINVAL);
174 #ifdef FF_API_FILL_FRAME
175 int avfilter_fill_frame_from_audio_buffer_ref(AVFrame *frame,
176 const AVFilterBufferRef *samplesref)
178 return avfilter_copy_buf_props(frame, samplesref);
181 int avfilter_fill_frame_from_video_buffer_ref(AVFrame *frame,
182 const AVFilterBufferRef *picref)
184 return avfilter_copy_buf_props(frame, picref);
187 int avfilter_fill_frame_from_buffer_ref(AVFrame *frame,
188 const AVFilterBufferRef *ref)
190 return avfilter_copy_buf_props(frame, ref);