0b0311087a35e61457d87221861e6e7548229b99
[libav.git] / libavfilter / formats.c
1 /*
2 * Filter layer - format negotiation
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 "avfilter.h"
23
24 /**
25 * Add all refs from a to ret and destroy a.
26 */
27 static void merge_ref(AVFilterFormats *ret, AVFilterFormats *a)
28 {
29 int i;
30
31 for(i = 0; i < a->refcount; i ++) {
32 ret->refs[ret->refcount] = a->refs[i];
33 *ret->refs[ret->refcount++] = ret;
34 }
35
36 av_free(a->refs);
37 av_free(a->formats);
38 av_free(a);
39 }
40
41 AVFilterFormats *avfilter_merge_formats(AVFilterFormats *a, AVFilterFormats *b)
42 {
43 AVFilterFormats *ret;
44 unsigned i, j, k = 0;
45
46 ret = av_mallocz(sizeof(AVFilterFormats));
47
48 /* merge list of formats */
49 ret->formats = av_malloc(sizeof(*ret->formats) * FFMIN(a->format_count,
50 b->format_count));
51 for(i = 0; i < a->format_count; i ++)
52 for(j = 0; j < b->format_count; j ++)
53 if(a->formats[i] == b->formats[j])
54 ret->formats[k++] = a->formats[i];
55
56 ret->format_count = k;
57 /* check that there was at least one common format */
58 if(!ret->format_count) {
59 av_free(ret->formats);
60 av_free(ret);
61 return NULL;
62 }
63
64 ret->refs = av_malloc(sizeof(AVFilterFormats**)*(a->refcount+b->refcount));
65
66 merge_ref(ret, a);
67 merge_ref(ret, b);
68
69 return ret;
70 }
71
72 AVFilterFormats *avfilter_make_format_list(const enum PixelFormat *pix_fmts)
73 {
74 AVFilterFormats *formats;
75 int count;
76
77 for (count = 0; pix_fmts[count] != PIX_FMT_NONE; count++)
78 ;
79
80 formats = av_mallocz(sizeof(AVFilterFormats));
81 formats->formats = av_malloc(sizeof(*formats->formats) * count);
82 formats->format_count = count;
83 memcpy(formats->formats, pix_fmts, sizeof(*formats->formats) * count);
84
85 return formats;
86 }
87
88 int avfilter_add_colorspace(AVFilterFormats *avff, enum PixelFormat pix_fmt)
89 {
90 enum PixelFormat *pix_fmts =
91 av_realloc(avff->formats, sizeof(avff->formats) * (avff->format_count+1));
92
93 if (!pix_fmts)
94 return AVERROR(ENOMEM);
95
96 avff->formats = pix_fmts;
97 avff->formats[avff->format_count++] = pix_fmt;
98 return 0;
99 }
100
101 AVFilterFormats *avfilter_all_colorspaces(void)
102 {
103 AVFilterFormats *ret;
104 int i;
105
106 ret = av_mallocz(sizeof(AVFilterFormats));
107 ret->formats = av_malloc(sizeof(*ret->formats) * PIX_FMT_NB);
108 ret->format_count = PIX_FMT_NB;
109
110 for(i = 0; i < PIX_FMT_NB; i ++)
111 ret->formats[i] = i;
112
113 return ret;
114 }
115
116 void avfilter_formats_ref(AVFilterFormats *f, AVFilterFormats **ref)
117 {
118 *ref = f;
119 f->refs = av_realloc(f->refs, sizeof(AVFilterFormats**) * ++f->refcount);
120 f->refs[f->refcount-1] = ref;
121 }
122
123 static int find_ref_index(AVFilterFormats **ref)
124 {
125 int i;
126 for(i = 0; i < (*ref)->refcount; i ++)
127 if((*ref)->refs[i] == ref)
128 return i;
129 return -1;
130 }
131
132 void avfilter_formats_unref(AVFilterFormats **ref)
133 {
134 int idx = find_ref_index(ref);
135
136 if(idx >= 0)
137 memmove((*ref)->refs + idx, (*ref)->refs + idx+1,
138 sizeof(AVFilterFormats**) * ((*ref)->refcount-idx-1));
139
140 if(!--(*ref)->refcount) {
141 av_free((*ref)->formats);
142 av_free((*ref)->refs);
143 av_free(*ref);
144 }
145 *ref = NULL;
146 }
147
148 void avfilter_formats_changeref(AVFilterFormats **oldref,
149 AVFilterFormats **newref)
150 {
151 int idx = find_ref_index(oldref);
152
153 if(idx >= 0) {
154 (*oldref)->refs[idx] = newref;
155 *newref = *oldref;
156 *oldref = NULL;
157 }
158 }
159