Make avfilter_make_format_list() take in input a const argument.
[ffmpeg.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_fmt)
73 {
74     AVFilterFormats *formats;
75     int count;
76
77     for (count = 0; pix_fmt[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_fmt, sizeof(*formats->formats) * count);
84
85     return formats;
86 }
87
88 AVFilterFormats *avfilter_all_colorspaces(void)
89 {
90     AVFilterFormats *ret;
91     int i;
92
93     ret = av_mallocz(sizeof(AVFilterFormats));
94     ret->formats = av_malloc(sizeof(*ret->formats) * PIX_FMT_NB);
95     ret->format_count = PIX_FMT_NB;
96
97     for(i = 0; i < PIX_FMT_NB; i ++)
98         ret->formats[i] = i;
99
100     return ret;
101 }
102
103 void avfilter_formats_ref(AVFilterFormats *f, AVFilterFormats **ref)
104 {
105     *ref = f;
106     f->refs = av_realloc(f->refs, sizeof(AVFilterFormats**) * ++f->refcount);
107     f->refs[f->refcount-1] = ref;
108 }
109
110 static int find_ref_index(AVFilterFormats **ref)
111 {
112     int i;
113     for(i = 0; i < (*ref)->refcount; i ++)
114         if((*ref)->refs[i] == ref)
115             return i;
116     return -1;
117 }
118
119 void avfilter_formats_unref(AVFilterFormats **ref)
120 {
121     int idx = find_ref_index(ref);
122
123     if(idx >= 0)
124         memmove((*ref)->refs + idx, (*ref)->refs + idx+1,
125             sizeof(AVFilterFormats**) * ((*ref)->refcount-idx-1));
126
127     if(!--(*ref)->refcount) {
128         av_free((*ref)->formats);
129         av_free((*ref)->refs);
130         av_free(*ref);
131     }
132     *ref = NULL;
133 }
134
135 void avfilter_formats_changeref(AVFilterFormats **oldref,
136                                 AVFilterFormats **newref)
137 {
138     int idx = find_ref_index(oldref);
139
140     if(idx >= 0) {
141         (*oldref)->refs[idx] = newref;
142         *newref = *oldref;
143         *oldref = NULL;
144     }
145 }
146