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