formats.c
Go to the documentation of this file.
1 /*
2  * Filter layer - format negotiation
3  * Copyright (c) 2007 Bobby Bingham
4  *
5  * This file is part of Libav.
6  *
7  * Libav 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  * Libav 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 Libav; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 #include "libavutil/common.h"
23 #include "libavutil/pixdesc.h"
24 #include "avfilter.h"
25 #include "internal.h"
26 #include "formats.h"
27 
31 #define MERGE_REF(ret, a, fmts, type, fail) \
32 do { \
33  type ***tmp; \
34  int i; \
35  \
36  if (!(tmp = av_realloc(ret->refs, \
37  sizeof(*tmp) * (ret->refcount + a->refcount)))) \
38  goto fail; \
39  ret->refs = tmp; \
40  \
41  for (i = 0; i < a->refcount; i ++) { \
42  ret->refs[ret->refcount] = a->refs[i]; \
43  *ret->refs[ret->refcount++] = ret; \
44  } \
45  \
46  av_freep(&a->refs); \
47  av_freep(&a->fmts); \
48  av_freep(&a); \
49 } while (0)
50 
55 #define MERGE_FORMATS(ret, a, b, fmts, nb, type, fail) \
56 do { \
57  int i, j, k = 0, count = FFMIN(a->nb, b->nb); \
58  \
59  if (!(ret = av_mallocz(sizeof(*ret)))) \
60  goto fail; \
61  \
62  if (count) { \
63  if (!(ret->fmts = av_malloc(sizeof(*ret->fmts) * count))) \
64  goto fail; \
65  for (i = 0; i < a->nb; i++) \
66  for (j = 0; j < b->nb; j++) \
67  if (a->fmts[i] == b->fmts[j]) \
68  ret->fmts[k++] = a->fmts[i]; \
69  \
70  ret->nb = k; \
71  } \
72  /* check that there was at least one common format */ \
73  if (!ret->nb) \
74  goto fail; \
75  \
76  MERGE_REF(ret, a, fmts, type, fail); \
77  MERGE_REF(ret, b, fmts, type, fail); \
78 } while (0)
79 
81 {
82  AVFilterFormats *ret = NULL;
83 
84  if (a == b)
85  return a;
86 
87  MERGE_FORMATS(ret, a, b, formats, format_count, AVFilterFormats, fail);
88 
89  return ret;
90 fail:
91  if (ret) {
92  av_freep(&ret->refs);
93  av_freep(&ret->formats);
94  }
95  av_freep(&ret);
96  return NULL;
97 }
98 
101 {
102  AVFilterFormats *ret = NULL;
103 
104  if (a == b) return a;
105 
106  if (a->format_count && b->format_count) {
107  MERGE_FORMATS(ret, a, b, formats, format_count, AVFilterFormats, fail);
108  } else if (a->format_count) {
109  MERGE_REF(a, b, formats, AVFilterFormats, fail);
110  ret = a;
111  } else {
112  MERGE_REF(b, a, formats, AVFilterFormats, fail);
113  ret = b;
114  }
115 
116  return ret;
117 fail:
118  if (ret) {
119  av_freep(&ret->refs);
120  av_freep(&ret->formats);
121  }
122  av_freep(&ret);
123  return NULL;
124 }
125 
128 {
130 
131  if (a == b) return a;
132 
133  if (a->nb_channel_layouts && b->nb_channel_layouts) {
134  MERGE_FORMATS(ret, a, b, channel_layouts, nb_channel_layouts,
135  AVFilterChannelLayouts, fail);
136  } else if (a->nb_channel_layouts) {
137  MERGE_REF(a, b, channel_layouts, AVFilterChannelLayouts, fail);
138  ret = a;
139  } else {
140  MERGE_REF(b, a, channel_layouts, AVFilterChannelLayouts, fail);
141  ret = b;
142  }
143 
144  return ret;
145 fail:
146  if (ret) {
147  av_freep(&ret->refs);
148  av_freep(&ret->channel_layouts);
149  }
150  av_freep(&ret);
151  return NULL;
152 }
153 
154 int ff_fmt_is_in(int fmt, const int *fmts)
155 {
156  const int *p;
157 
158  for (p = fmts; *p != AV_PIX_FMT_NONE; p++) {
159  if (fmt == *p)
160  return 1;
161  }
162  return 0;
163 }
164 
166 {
168  int count;
169 
170  for (count = 0; fmts[count] != -1; count++)
171  ;
172 
173  formats = av_mallocz(sizeof(*formats));
174  if (count)
175  formats->formats = av_malloc(sizeof(*formats->formats) * count);
176  formats->format_count = count;
177  memcpy(formats->formats, fmts, sizeof(*formats->formats) * count);
178 
179  return formats;
180 }
181 
182 #define ADD_FORMAT(f, fmt, type, list, nb) \
183 do { \
184  type *fmts; \
185  \
186  if (!(*f) && !(*f = av_mallocz(sizeof(**f)))) \
187  return AVERROR(ENOMEM); \
188  \
189  fmts = av_realloc((*f)->list, \
190  sizeof(*(*f)->list) * ((*f)->nb + 1));\
191  if (!fmts) \
192  return AVERROR(ENOMEM); \
193  \
194  (*f)->list = fmts; \
195  (*f)->list[(*f)->nb++] = fmt; \
196  return 0; \
197 } while (0)
198 
199 int ff_add_format(AVFilterFormats **avff, int fmt)
200 {
201  ADD_FORMAT(avff, fmt, int, formats, format_count);
202 }
203 
204 int ff_add_channel_layout(AVFilterChannelLayouts **l, uint64_t channel_layout)
205 {
206  ADD_FORMAT(l, channel_layout, uint64_t, channel_layouts, nb_channel_layouts);
207 }
208 
210 {
211  AVFilterFormats *ret = NULL;
212  int fmt;
213  int num_formats = type == AVMEDIA_TYPE_VIDEO ? AV_PIX_FMT_NB :
214  type == AVMEDIA_TYPE_AUDIO ? AV_SAMPLE_FMT_NB : 0;
215 
216  for (fmt = 0; fmt < num_formats; fmt++) {
217  const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(fmt);
218  if ((type != AVMEDIA_TYPE_VIDEO) ||
219  (type == AVMEDIA_TYPE_VIDEO && !(desc->flags & PIX_FMT_HWACCEL)))
220  ff_add_format(&ret, fmt);
221  }
222 
223  return ret;
224 }
225 
227 {
228  AVFilterFormats *ret = NULL;
229  int fmt;
230 
231  for (fmt = 0; fmt < AV_SAMPLE_FMT_NB; fmt++)
232  if (av_sample_fmt_is_planar(fmt))
233  ff_add_format(&ret, fmt);
234 
235  return ret;
236 }
237 
239 {
240  AVFilterFormats *ret = av_mallocz(sizeof(*ret));
241  return ret;
242 }
243 
245 {
246  AVFilterChannelLayouts *ret = av_mallocz(sizeof(*ret));
247  return ret;
248 }
249 
250 #define FORMATS_REF(f, ref) \
251 do { \
252  *ref = f; \
253  f->refs = av_realloc(f->refs, sizeof(*f->refs) * ++f->refcount); \
254  f->refs[f->refcount-1] = ref; \
255 } while (0)
256 
258 {
259  FORMATS_REF(f, ref);
260 }
261 
263 {
264  FORMATS_REF(f, ref);
265 }
266 
267 #define FIND_REF_INDEX(ref, idx) \
268 do { \
269  int i; \
270  for (i = 0; i < (*ref)->refcount; i ++) \
271  if((*ref)->refs[i] == ref) { \
272  idx = i; \
273  break; \
274  } \
275 } while (0)
276 
277 #define FORMATS_UNREF(ref, list) \
278 do { \
279  int idx = -1; \
280  \
281  if (!*ref) \
282  return; \
283  \
284  FIND_REF_INDEX(ref, idx); \
285  \
286  if (idx >= 0) \
287  memmove((*ref)->refs + idx, (*ref)->refs + idx + 1, \
288  sizeof(*(*ref)->refs) * ((*ref)->refcount - idx - 1)); \
289  \
290  if(!--(*ref)->refcount) { \
291  av_free((*ref)->list); \
292  av_free((*ref)->refs); \
293  av_free(*ref); \
294  } \
295  *ref = NULL; \
296 } while (0)
297 
299 {
300  FORMATS_UNREF(ref, formats);
301 }
302 
304 {
305  FORMATS_UNREF(ref, channel_layouts);
306 }
307 
308 #define FORMATS_CHANGEREF(oldref, newref) \
309 do { \
310  int idx = -1; \
311  \
312  FIND_REF_INDEX(oldref, idx); \
313  \
314  if (idx >= 0) { \
315  (*oldref)->refs[idx] = newref; \
316  *newref = *oldref; \
317  *oldref = NULL; \
318  } \
319 } while (0)
320 
322  AVFilterChannelLayouts **newref)
323 {
324  FORMATS_CHANGEREF(oldref, newref);
325 }
326 
328 {
329  FORMATS_CHANGEREF(oldref, newref);
330 }
331 
332 #define SET_COMMON_FORMATS(ctx, fmts, in_fmts, out_fmts, ref, list) \
333 { \
334  int count = 0, i; \
335  \
336  for (i = 0; i < ctx->nb_inputs; i++) { \
337  if (ctx->inputs[i]) { \
338  ref(fmts, &ctx->inputs[i]->out_fmts); \
339  count++; \
340  } \
341  } \
342  for (i = 0; i < ctx->nb_outputs; i++) { \
343  if (ctx->outputs[i]) { \
344  ref(fmts, &ctx->outputs[i]->in_fmts); \
345  count++; \
346  } \
347  } \
348  \
349  if (!count) { \
350  av_freep(&fmts->list); \
351  av_freep(&fmts->refs); \
352  av_freep(&fmts); \
353  } \
354 }
355 
358 {
359  SET_COMMON_FORMATS(ctx, layouts, in_channel_layouts, out_channel_layouts,
360  ff_channel_layouts_ref, channel_layouts);
361 }
362 
364  AVFilterFormats *samplerates)
365 {
366  SET_COMMON_FORMATS(ctx, samplerates, in_samplerates, out_samplerates,
368 }
369 
376 {
377  SET_COMMON_FORMATS(ctx, formats, in_formats, out_formats,
378  ff_formats_ref, formats);
379 }
380 
382 {
383  enum AVMediaType type = ctx->inputs && ctx->inputs [0] ? ctx->inputs [0]->type :
384  ctx->outputs && ctx->outputs[0] ? ctx->outputs[0]->type :
386 
388  if (type == AVMEDIA_TYPE_AUDIO) {
391  }
392 
393  return 0;
394 }