sha: add missing include

[ffmpeg] / libavfilter / formats.c

diff --git a/libavfilter/formats.c b/libavfilter/formats.c
index b1763d9ac05d2ed0b9686c3bd98f876584f345c2..2a9bdb0bd0600b7001651ad8b49be709a5cfe3c7 100644 (file)
--- a/libavfilter/formats.c
+++ b/libavfilter/formats.c
@@ -19,6 +19,7 @@
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  */
 
+#include "libavutil/pixdesc.h"
 #include "avfilter.h"
 
 /**
@@ -69,33 +70,47 @@ AVFilterFormats *avfilter_merge_formats(AVFilterFormats *a, AVFilterFormats *b)
     return ret;
 }
 
-AVFilterFormats *avfilter_make_format_list2(enum PixelFormat *pix_fmt)
+AVFilterFormats *avfilter_make_format_list(const enum PixelFormat *pix_fmts)
 {
     AVFilterFormats *formats;
     int count;
 
-    for (count = 0; pix_fmt[count] != PIX_FMT_NONE; count++)
+    for (count = 0; pix_fmts[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);
+    memcpy(formats->formats, pix_fmts, sizeof(*formats->formats) * count);
 
     return formats;
 }
 
-AVFilterFormats *avfilter_all_colorspaces(void)
+int avfilter_add_colorspace(AVFilterFormats **avff, enum PixelFormat pix_fmt)
 {
-    AVFilterFormats *ret;
-    int i;
+    enum PixelFormat *pix_fmts;
 
-    ret = av_mallocz(sizeof(AVFilterFormats));
-    ret->formats = av_malloc(sizeof(*ret->formats) * PIX_FMT_NB);
-    ret->format_count = PIX_FMT_NB;
+    if (!(*avff) && !(*avff = av_mallocz(sizeof(AVFilterFormats))))
+        return AVERROR(ENOMEM);
+
+    pix_fmts = av_realloc((*avff)->formats,
+                          sizeof((*avff)->formats) * ((*avff)->format_count+1));
+    if (!pix_fmts)
+        return AVERROR(ENOMEM);
+
+    (*avff)->formats = pix_fmts;
+    (*avff)->formats[(*avff)->format_count++] = pix_fmt;
+    return 0;
+}
+
+AVFilterFormats *avfilter_all_colorspaces(void)
+{
+    AVFilterFormats *ret = NULL;
+    enum PixelFormat pix_fmt;
 
-    for(i = 0; i < PIX_FMT_NB; i ++)
-        ret->formats[i] = i;
+    for (pix_fmt = 0; pix_fmt < PIX_FMT_NB; pix_fmt++)
+        if (!(av_pix_fmt_descriptors[pix_fmt].flags & PIX_FMT_HWACCEL))
+            avfilter_add_colorspace(&ret, pix_fmt);
 
     return ret;
 }
@@ -118,7 +133,12 @@ static int find_ref_index(AVFilterFormats **ref)
 
 void avfilter_formats_unref(AVFilterFormats **ref)
 {
-    int idx = find_ref_index(ref);
+    int idx;
+
+    if (!*ref)
+        return;
+
+    idx = find_ref_index(ref);
 
     if(idx >= 0)
         memmove((*ref)->refs + idx, (*ref)->refs + idx+1,