Implement avfilter_make_format_list2(), which is going to replace
[ffmpeg.git] / libavfilter / formats.c
index b3873d7..3a44cfd 100644 (file)
@@ -21,7 +21,9 @@
 
 #include "avfilter.h"
 
-/** merge and update all the references */
+/**
+ * Add all refs from a to ret and destroy a.
+ */
 static void merge_ref(AVFilterFormats *ret, AVFilterFormats *a)
 {
     int i;
@@ -51,8 +53,9 @@ AVFilterFormats *avfilter_merge_formats(AVFilterFormats *a, AVFilterFormats *b)
             if(a->formats[i] == b->formats[j])
                 ret->formats[k++] = a->formats[i];
 
+    ret->format_count = k;
     /* check that there was at least one common format */
-    if(!(ret->format_count = k)) {
+    if(!ret->format_count) {
         av_free(ret->formats);
         av_free(ret);
         return NULL;
@@ -84,6 +87,22 @@ AVFilterFormats *avfilter_make_format_list(int len, ...)
     return ret;
 }
 
+AVFilterFormats *avfilter_make_format_list2(enum PixelFormat *pix_fmt)
+{
+    AVFilterFormats *formats;
+    int count;
+
+    for (count = 0; pix_fmt[count] != PIX_FMT_NONE; count++)
+        ;
+
+    formats               = av_mallocz(sizeof(AVFilterFormats));
+    formats->formats      = av_malloc(sizeof(*formats->formats) * count);
+    formats->format_count = count;
+    memcpy(formats->formats, pix_fmt, sizeof(*formats->formats) * count);
+
+    return formats;
+}
+
 AVFilterFormats *avfilter_all_colorspaces(void)
 {
     AVFilterFormats *ret;
@@ -117,9 +136,9 @@ static int find_ref_index(AVFilterFormats **ref)
 
 void avfilter_formats_unref(AVFilterFormats **ref)
 {
-    int idx;
+    int idx = find_ref_index(ref);
 
-    if((idx = find_ref_index(ref)) >= 0)
+    if(idx >= 0)
         memmove((*ref)->refs + idx, (*ref)->refs + idx+1,
             sizeof(AVFilterFormats**) * ((*ref)->refcount-idx-1));
 
@@ -134,9 +153,9 @@ void avfilter_formats_unref(AVFilterFormats **ref)
 void avfilter_formats_changeref(AVFilterFormats **oldref,
                                 AVFilterFormats **newref)
 {
-    int idx;
+    int idx = find_ref_index(oldref);
 
-    if((idx = find_ref_index(oldref)) >= 0) {
+    if(idx >= 0) {
         (*oldref)->refs[idx] = newref;
         *newref = *oldref;
         *oldref = NULL;