1/*
2 * Filter layer - default implementations
3 * copyright (c) 2007 Bobby Bingham
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22#include "libavcodec/imgconvert.h"
23#include "avfilter.h"
24
25/* TODO: buffer pool.  see comment for avfilter_default_get_video_buffer() */
26void avfilter_default_free_video_buffer(AVFilterPic *pic)
27{
28    av_free(pic->data[0]);
29    av_free(pic);
30}
31
32#define ALIGN(a) do{ \
33                     (a) = ((a) + 15) & (~15); \
34                 } while(0);
35
36/* TODO: set the buffer's priv member to a context structure for the whole
37 * filter chain.  This will allow for a buffer pool instead of the constant
38 * alloc & free cycle currently implemented. */
39AVFilterPicRef *avfilter_default_get_video_buffer(AVFilterLink *link, int perms)
40{
41    AVFilterPic *pic = av_mallocz(sizeof(AVFilterPic));
42    AVFilterPicRef *ref = av_mallocz(sizeof(AVFilterPicRef));
43    int i, tempsize;
44    char *buf;
45
46    ref->pic   = pic;
47    ref->w     = link->w;
48    ref->h     = link->h;
49
50    /* make sure the buffer gets read permission or it's useless for output */
51    ref->perms = perms | AV_PERM_READ;
52
53    pic->refcount = 1;
54    pic->format   = link->format;
55    pic->free     = avfilter_default_free_video_buffer;
56    ff_fill_linesize((AVPicture *)pic, pic->format, ref->w);
57
58    for (i=0; i<4;i++)
59        ALIGN(pic->linesize[i]);
60
61    tempsize = ff_fill_pointer((AVPicture *)pic, NULL, pic->format, ref->h);
62    buf = av_malloc(tempsize);
63    ff_fill_pointer((AVPicture *)pic, buf, pic->format, ref->h);
64
65    memcpy(ref->data,     pic->data,     sizeof(pic->data));
66    memcpy(ref->linesize, pic->linesize, sizeof(pic->linesize));
67
68    return ref;
69}
70
71void avfilter_default_start_frame(AVFilterLink *link, AVFilterPicRef *picref)
72{
73    AVFilterLink *out = NULL;
74
75    if(link->dst->output_count)
76        out = link->dst->outputs[0];
77
78    if(out) {
79        out->outpic      = avfilter_get_video_buffer(out, AV_PERM_WRITE);
80        out->outpic->pts = picref->pts;
81        avfilter_start_frame(out, avfilter_ref_pic(out->outpic, ~0));
82    }
83}
84
85void avfilter_default_draw_slice(AVFilterLink *link, int y, int h)
86{
87    AVFilterLink *out = NULL;
88
89    if(link->dst->output_count)
90        out = link->dst->outputs[0];
91
92    if(out)
93        avfilter_draw_slice(out, y, h);
94}
95
96void avfilter_default_end_frame(AVFilterLink *link)
97{
98    AVFilterLink *out = NULL;
99
100    if(link->dst->output_count)
101        out = link->dst->outputs[0];
102
103    avfilter_unref_pic(link->cur_pic);
104    link->cur_pic = NULL;
105
106    if(out) {
107        if(out->outpic) {
108            avfilter_unref_pic(out->outpic);
109            out->outpic = NULL;
110        }
111        avfilter_end_frame(out);
112    }
113}
114
115/**
116 * default config_link() implementation for output video links to simplify
117 * the implementation of one input one output video filters */
118int avfilter_default_config_output_link(AVFilterLink *link)
119{
120    if(link->src->input_count && link->src->inputs[0]) {
121        link->w = link->src->inputs[0]->w;
122        link->h = link->src->inputs[0]->h;
123    } else {
124        /* XXX: any non-simple filter which would cause this branch to be taken
125         * really should implement its own config_props() for this link. */
126        return -1;
127    }
128
129    return 0;
130}
131
132/**
133 * A helper for query_formats() which sets all links to the same list of
134 * formats. If there are no links hooked to this filter, the list of formats is
135 * freed.
136 *
137 * FIXME: this will need changed for filters with a mix of pad types
138 * (video + audio, etc)
139 */
140void avfilter_set_common_formats(AVFilterContext *ctx, AVFilterFormats *formats)
141{
142    int count = 0, i;
143
144    for(i = 0; i < ctx->input_count; i ++) {
145        if(ctx->inputs[i]) {
146            avfilter_formats_ref(formats, &ctx->inputs[i]->out_formats);
147            count ++;
148        }
149    }
150    for(i = 0; i < ctx->output_count; i ++) {
151        if(ctx->outputs[i]) {
152            avfilter_formats_ref(formats, &ctx->outputs[i]->in_formats);
153            count ++;
154        }
155    }
156
157    if(!count) {
158        av_free(formats->formats);
159        av_free(formats->refs);
160        av_free(formats);
161    }
162}
163
164int avfilter_default_query_formats(AVFilterContext *ctx)
165{
166    avfilter_set_common_formats(ctx, avfilter_all_colorspaces());
167    return 0;
168}
169
170