lavfi/formats: use sizeof(var) instead of sizeof(type).
[ffmpeg.git] / libavfilter / formats.c
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/pixdesc.h"
23 #include "avfilter.h"
24 #include "internal.h"
25
26 /**
27  * Add all refs from a to ret and destroy a.
28  */
29 static void merge_ref(AVFilterFormats *ret, AVFilterFormats *a)
30 {
31     int i;
32
33     for(i = 0; i < a->refcount; i ++) {
34         ret->refs[ret->refcount] = a->refs[i];
35         *ret->refs[ret->refcount++] = ret;
36     }
37
38     av_free(a->refs);
39     av_free(a->formats);
40     av_free(a);
41 }
42
43 AVFilterFormats *avfilter_merge_formats(AVFilterFormats *a, AVFilterFormats *b)
44 {
45     AVFilterFormats *ret;
46     unsigned i, j, k = 0, m_count;
47
48     ret = av_mallocz(sizeof(*ret));
49
50     /* merge list of formats */
51     m_count = FFMIN(a->format_count, b->format_count);
52     if (m_count) {
53         ret->formats = av_malloc(sizeof(*ret->formats) * m_count);
54         for(i = 0; i < a->format_count; i ++)
55             for(j = 0; j < b->format_count; j ++)
56                 if(a->formats[i] == b->formats[j])
57                     ret->formats[k++] = a->formats[i];
58
59         ret->format_count = k;
60     }
61     /* check that there was at least one common format */
62     if(!ret->format_count) {
63         av_free(ret->formats);
64         av_free(ret);
65         return NULL;
66     }
67
68     ret->refs = av_malloc(sizeof(*ret->refs) * (a->refcount + b->refcount));
69
70     merge_ref(ret, a);
71     merge_ref(ret, b);
72
73     return ret;
74 }
75
76 int ff_fmt_is_in(int fmt, const int *fmts)
77 {
78     const int *p;
79
80     for (p = fmts; *p != PIX_FMT_NONE; p++) {
81         if (fmt == *p)
82             return 1;
83     }
84     return 0;
85 }
86
87 AVFilterFormats *avfilter_make_format_list(const int *fmts)
88 {
89     AVFilterFormats *formats;
90     int count;
91
92     for (count = 0; fmts[count] != -1; count++)
93         ;
94
95     formats               = av_mallocz(sizeof(*formats));
96     if (count)
97         formats->formats  = av_malloc(sizeof(*formats->formats) * count);
98     formats->format_count = count;
99     memcpy(formats->formats, fmts, sizeof(*formats->formats) * count);
100
101     return formats;
102 }
103
104 int avfilter_add_format(AVFilterFormats **avff, int fmt)
105 {
106     int *fmts;
107
108     if (!(*avff) && !(*avff = av_mallocz(sizeof(**avff))))
109         return AVERROR(ENOMEM);
110
111     fmts = av_realloc((*avff)->formats,
112                       sizeof(*(*avff)->formats) * ((*avff)->format_count+1));
113     if (!fmts)
114         return AVERROR(ENOMEM);
115
116     (*avff)->formats = fmts;
117     (*avff)->formats[(*avff)->format_count++] = fmt;
118     return 0;
119 }
120
121 AVFilterFormats *avfilter_all_formats(enum AVMediaType type)
122 {
123     AVFilterFormats *ret = NULL;
124     int fmt;
125     int num_formats = type == AVMEDIA_TYPE_VIDEO ? PIX_FMT_NB    :
126                       type == AVMEDIA_TYPE_AUDIO ? AV_SAMPLE_FMT_NB : 0;
127
128     for (fmt = 0; fmt < num_formats; fmt++)
129         if ((type != AVMEDIA_TYPE_VIDEO) ||
130             (type == AVMEDIA_TYPE_VIDEO && !(av_pix_fmt_descriptors[fmt].flags & PIX_FMT_HWACCEL)))
131             avfilter_add_format(&ret, fmt);
132
133     return ret;
134 }
135
136 void avfilter_formats_ref(AVFilterFormats *f, AVFilterFormats **ref)
137 {
138     *ref = f;
139     f->refs = av_realloc(f->refs, sizeof(*f->refs) * ++f->refcount);
140     f->refs[f->refcount-1] = ref;
141 }
142
143 static int find_ref_index(AVFilterFormats **ref)
144 {
145     int i;
146     for(i = 0; i < (*ref)->refcount; i ++)
147         if((*ref)->refs[i] == ref)
148             return i;
149     return -1;
150 }
151
152 void avfilter_formats_unref(AVFilterFormats **ref)
153 {
154     int idx;
155
156     if (!*ref)
157         return;
158
159     idx = find_ref_index(ref);
160
161     if(idx >= 0)
162         memmove((*ref)->refs + idx, (*ref)->refs + idx + 1,
163             sizeof(*(*ref)->refs) * ((*ref)->refcount - idx - 1));
164
165     if(!--(*ref)->refcount) {
166         av_free((*ref)->formats);
167         av_free((*ref)->refs);
168         av_free(*ref);
169     }
170     *ref = NULL;
171 }
172
173 void avfilter_formats_changeref(AVFilterFormats **oldref,
174                                 AVFilterFormats **newref)
175 {
176     int idx = find_ref_index(oldref);
177
178     if(idx >= 0) {
179         (*oldref)->refs[idx] = newref;
180         *newref = *oldref;
181         *oldref = NULL;
182     }
183 }