copy pts and format props between lavfi buffer and frame.
[ffmpeg.git] / libavfilter / avcodec.c
1 /*
2  * This file is part of FFmpeg.
3  *
4  * FFmpeg is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * FFmpeg is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with FFmpeg; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18
19 /**
20  * @file
21  * libavcodec/libavfilter gluing utilities
22  */
23
24 #include "avcodec.h"
25
26 int avfilter_copy_frame_props(AVFilterBufferRef *dst, const AVFrame *src)
27 {
28     dst->pts    = src->pts;
29     dst->pos    = src->pkt_pos;
30     dst->format = src->format;
31
32     switch (dst->type) {
33     case AVMEDIA_TYPE_VIDEO:
34         dst->video->w                   = src->width;
35         dst->video->h                   = src->height;
36         dst->video->sample_aspect_ratio = src->sample_aspect_ratio;
37         dst->video->interlaced          = src->interlaced_frame;
38         dst->video->top_field_first     = src->top_field_first;
39         dst->video->key_frame           = src->key_frame;
40         dst->video->pict_type           = src->pict_type;
41     }
42
43     return 0;
44 }
45
46 AVFilterBufferRef *avfilter_get_video_buffer_ref_from_frame(const AVFrame *frame,
47                                                             int perms)
48 {
49     AVFilterBufferRef *picref =
50         avfilter_get_video_buffer_ref_from_arrays(frame->data, frame->linesize, perms,
51                                                   frame->width, frame->height,
52                                                   frame->format);
53     if (!picref)
54         return NULL;
55     avfilter_copy_frame_props(picref, frame);
56     return picref;
57 }
58
59 int avfilter_fill_frame_from_audio_buffer_ref(AVFrame *frame,
60                                               const AVFilterBufferRef *samplesref)
61 {
62     if (!samplesref || !samplesref->audio || !frame)
63         return AVERROR(EINVAL);
64
65     memcpy(frame->data, samplesref->data, sizeof(frame->data));
66     frame->pkt_pos    = samplesref->pos;
67     frame->format     = samplesref->format;
68     frame->nb_samples = samplesref->audio->nb_samples;
69     frame->pts        = samplesref->pts;
70
71     return 0;
72 }
73
74 int avfilter_fill_frame_from_video_buffer_ref(AVFrame *frame,
75                                               const AVFilterBufferRef *picref)
76 {
77     if (!picref || !picref->video || !frame)
78         return AVERROR(EINVAL);
79
80     memcpy(frame->data,     picref->data,     sizeof(frame->data));
81     memcpy(frame->linesize, picref->linesize, sizeof(frame->linesize));
82     frame->pkt_pos          = picref->pos;
83     frame->interlaced_frame = picref->video->interlaced;
84     frame->top_field_first  = picref->video->top_field_first;
85     frame->key_frame        = picref->video->key_frame;
86     frame->pict_type        = picref->video->pict_type;
87     frame->sample_aspect_ratio = picref->video->sample_aspect_ratio;
88     frame->width            = picref->video->w;
89     frame->height           = picref->video->h;
90     frame->format           = picref->format;
91     frame->pts              = picref->pts;
92
93     return 0;
94 }
95
96 int avfilter_fill_frame_from_buffer_ref(AVFrame *frame,
97                                         const AVFilterBufferRef *ref)
98 {
99     if (!ref)
100         return AVERROR(EINVAL);
101     return ref->video ? avfilter_fill_frame_from_video_buffer_ref(frame, ref)
102                       : avfilter_fill_frame_from_audio_buffer_ref(frame, ref);
103 }