Move allocation of imbuf from array to allocimbuf.
[blender.git] / source / blender / imbuf / intern / util.c
index 9e819e56da5223e58db4d19c9f668e4cc86eef88..32100aa2288bc9ff033b5c45510d79f1e32349cb 100644 (file)
 
 
 #ifdef _WIN32
-#include <io.h>
-#define open _open
-#define read _read
-#define close _close
+#  include <io.h>
 #endif
 
-#include "BLI_blenlib.h"
+#include <stdlib.h>
+
+#include "BLI_utildefines.h"
+#include "BLI_path_util.h"
+#include "BLI_fileops.h"
+#include "BLI_string.h"
 
-#include "DNA_userdef_types.h"
 #include "BKE_global.h"
 
 #include "imbuf.h"
@@ -78,6 +79,7 @@ const char *imb_ext_image[] = {
 #endif
 #ifdef WITH_OPENJPEG
        ".jp2",
+       ".j2c",
 #endif
 #ifdef WITH_HDR
        ".hdr",
@@ -92,7 +94,18 @@ const char *imb_ext_image[] = {
 #ifdef WITH_OPENEXR
        ".exr",
 #endif
-       NULL};
+#ifdef WITH_OPENIMAGEIO
+       ".psd", ".pdd", ".psb",
+#endif
+       NULL
+};
+
+const char *imb_ext_image_filepath_only[] = {
+#ifdef WITH_OPENIMAGEIO
+       ".psd", ".pdd", ".psb",
+#endif
+       NULL
+};
 
 const char *imb_ext_image_qt[] = {
        ".gif",
@@ -100,7 +113,21 @@ const char *imb_ext_image_qt[] = {
        ".pct", ".pict",
        ".pntg",
        ".qtif",
-       NULL};
+       NULL
+};
+
+#if 0  /* UNUSED */
+const char *imb_ext_movie_qt[] = {
+       ".avi",   
+       ".flc",   
+       ".dv",    
+       ".r3d",   
+       ".mov",   
+       ".movie", 
+       ".mv",
+       NULL
+};
+#endif
 
 const char *imb_ext_movie[] = {
        ".avi",
@@ -113,10 +140,13 @@ const char *imb_ext_movie[] = {
        ".m2t",
        ".m2ts",
        ".mts",
+       ".ts",
        ".mv",
        ".avs",
        ".wmv",
        ".ogv",
+       ".ogg",
+       ".r3d",
        ".dv",
        ".mpeg",
        ".mpg",
@@ -127,7 +157,9 @@ const char *imb_ext_movie[] = {
        ".divx",
        ".xvid",
        ".mxf",
-       NULL};
+       ".webm",
+       NULL
+};
 
 /* sort of wrong being here... */
 const char *imb_ext_audio[] = {
@@ -144,180 +176,205 @@ const char *imb_ext_audio[] = {
        ".aif",
        ".aiff",
        ".m4a",
-       NULL};
+       ".mka",
+       NULL
+};
 
-static int IMB_ispic_name(const char *name)
+int IMB_ispic_type(const char *name)
 {
+       /* increased from 32 to 64 because of the bitmaps header size */
+#define HEADER_SIZE 64
+
+       unsigned char buf[HEADER_SIZE];
        ImFileType *type;
-       struct stat st;
-       int fp, buf[10];
+       BLI_stat_t st;
+       int fp;
 
        if (UTIL_DEBUG) printf("IMB_ispic_name: loading %s\n", name);
        
-       if (stat(name,&st) == -1)
-               return FALSE;
+       if (BLI_stat(name, &st) == -1)
+               return false;
        if (((st.st_mode) & S_IFMT) != S_IFREG)
-               return FALSE;
+               return false;
 
-       if ((fp = BLI_open(name,O_BINARY|O_RDONLY, 0)) < 0)
-               return FALSE;
+       if ((fp = BLI_open(name, O_BINARY | O_RDONLY, 0)) == -1)
+               return false;
 
-       if (read(fp, buf, 32) != 32) {
+       memset(buf, 0, sizeof(buf));
+       if (read(fp, buf, HEADER_SIZE) <= 0) {
                close(fp);
-               return FALSE;
+               return false;
        }
 
        close(fp);
 
        /* XXX move this exception */
-       if ((BIG_LONG(buf[0]) & 0xfffffff0) == 0xffd8ffe0)
+       if ((BIG_LONG(((int *)buf)[0]) & 0xfffffff0) == 0xffd8ffe0)
                return JPG;
 
-       for (type=IMB_FILE_TYPES; type->is_a; type++)
-               if (type->is_a((uchar*)buf))
-                       return type->filetype;
-
-       return FALSE;
-}
-
-int IMB_ispic(const char *filename)
-{
-       if (U.uiflag & USER_FILTERFILEEXTS) {
-               if (    (BLI_testextensie_array(filename, imb_ext_image)) ||
-                       (G.have_quicktime && BLI_testextensie_array(filename, imb_ext_image_qt))
-               ) {
-                       return IMB_ispic_name(filename);
+       for (type = IMB_FILE_TYPES; type < IMB_FILE_TYPES_LAST; type++) {
+               if (type->is_a) {
+                       if (type->is_a(buf)) {
+                               return type->filetype;
+                       }
                }
-               else {
-                       return FALSE;
+               else if (type->is_a_filepath) {
+                       if (type->is_a_filepath(name)) {
+                               return type->filetype;
+                       }
                }
        }
-       else { /* no FILTERFILEEXTS */
-               return IMB_ispic_name(filename);
-       }
-}
 
+       return 0;
 
+#undef HEADER_SIZE
+}
 
-static int isavi (const char *name)
+bool IMB_ispic(const char *name)
 {
-       return AVI_is_avi (name);
+       return (IMB_ispic_type(name) != 0);
+}
+
+static int isavi(const char *name)
+{
+#ifdef WITH_AVI
+       return AVI_is_avi(name);
+#else
+       (void)name;
+       return false;
+#endif
 }
 
 #ifdef WITH_QUICKTIME
-static int isqtime (const char *name)
+static int isqtime(const char *name)
 {
-       return anim_is_quicktime (name);
+       return anim_is_quicktime(name);
 }
 #endif
 
 #ifdef WITH_FFMPEG
 
-void silence_log_ffmpeg(int quiet)
+/* BLI_vsnprintf in ffmpeg_log_callback() causes invalid warning */
+#ifdef __GNUC__
+#  pragma GCC diagnostic push
+#  pragma GCC diagnostic ignored "-Wmissing-format-attribute"
+#endif
+
+static char ffmpeg_last_error[1024];
+
+static void ffmpeg_log_callback(void *ptr, int level, const char *format, va_list arg)
 {
-       if (quiet) {
-               av_log_set_level(AV_LOG_QUIET);
+       if (ELEM(level, AV_LOG_FATAL, AV_LOG_ERROR)) {
+               size_t n;
+               va_list args_cpy;
+
+               va_copy(args_cpy, arg);
+               n = BLI_vsnprintf(ffmpeg_last_error, sizeof(ffmpeg_last_error), format, args_cpy);
+               va_end(args_cpy);
+
+               /* strip trailing \n */
+               ffmpeg_last_error[n - 1] = '\0';
        }
-       else {
-               av_log_set_level(AV_LOG_DEBUG);
+
+       if (G.debug & G_DEBUG_FFMPEG) {
+               /* call default logger to print all message to console */
+               av_log_default_callback(ptr, level, format, arg);
        }
 }
 
-extern void do_init_ffmpeg(void);
-void do_init_ffmpeg(void)
+#ifdef __GNUC__
+#  pragma GCC diagnostic pop
+#endif
+
+void IMB_ffmpeg_init(void)
 {
-       static int ffmpeg_init = 0;
-       if (!ffmpeg_init) {
-               ffmpeg_init = 1;
-               av_register_all();
-               avdevice_register_all();
-               if ((G.debug & G_DEBUG_FFMPEG) == 0) {
-                       silence_log_ffmpeg(1);
-               }
-               else {
-                       silence_log_ffmpeg(0);
-               }
-       }
+       av_register_all();
+       avdevice_register_all();
+
+       ffmpeg_last_error[0] = '\0';
+
+       if (G.debug & G_DEBUG_FFMPEG)
+               av_log_set_level(AV_LOG_DEBUG);
+
+       /* set own callback which could store last error to report to UI */
+       av_log_set_callback(ffmpeg_log_callback);
 }
 
-static int isffmpeg (const char *filename)
+const char *IMB_ffmpeg_last_error(void)
 {
-       AVFormatContext *pFormatCtx;
+       return ffmpeg_last_error;
+}
+
+static int isffmpeg(const char *filename)
+{
+       AVFormatContext *pFormatCtx = NULL;
        unsigned int i;
        int videoStream;
        AVCodec *pCodec;
        AVCodecContext *pCodecCtx;
 
-       do_init_ffmpeg();
-
-       if (BLI_testextensie(filename, ".swf") ||
-               BLI_testextensie(filename, ".jpg") ||
-               BLI_testextensie(filename, ".png") ||
-               BLI_testextensie(filename, ".dds") ||
-               BLI_testextensie(filename, ".tga") ||
-               BLI_testextensie(filename, ".bmp") ||
-               BLI_testextensie(filename, ".exr") ||
-               BLI_testextensie(filename, ".cin") ||
-           BLI_testextensie(filename, ".wav"))
+       if (BLI_testextensie_n(
+               filename,
+               ".swf", ".jpg", ".png", ".dds", ".tga", ".bmp", ".tif", ".exr", ".cin", ".wav", NULL))
        {
                return 0;
        }
 
-       if (av_open_input_file(&pFormatCtx, filename, NULL, 0, NULL)!=0) {
+       if (avformat_open_input(&pFormatCtx, filename, NULL, NULL) != 0) {
                if (UTIL_DEBUG) fprintf(stderr, "isffmpeg: av_open_input_file failed\n");
                return 0;
        }
 
-       if (av_find_stream_info(pFormatCtx)<0) {
-               if (UTIL_DEBUG) fprintf(stderr, "isffmpeg: av_find_stream_info failed\n");
-               av_close_input_file(pFormatCtx);
+       if (avformat_find_stream_info(pFormatCtx, NULL) < 0) {
+               if (UTIL_DEBUG) fprintf(stderr, "isffmpeg: avformat_find_stream_info failed\n");
+               avformat_close_input(&pFormatCtx);
                return 0;
        }
 
        if (UTIL_DEBUG) av_dump_format(pFormatCtx, 0, filename, 0);
 
 
-               /* Find the first video stream */
-       videoStream=-1;
-       for (i=0; i<pFormatCtx->nb_streams; i++)
+       /* Find the first video stream */
+       videoStream = -1;
+       for (i = 0; i < pFormatCtx->nb_streams; i++)
                if (pFormatCtx->streams[i] &&
-                  pFormatCtx->streams[i]->codec && 
-                 (pFormatCtx->streams[i]->codec->codec_type==AVMEDIA_TYPE_VIDEO))
+                   pFormatCtx->streams[i]->codec &&
+                   (pFormatCtx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO))
                {
-                       videoStream=i;
+                       videoStream = i;
                        break;
                }
 
-       if (videoStream==-1) {
-               av_close_input_file(pFormatCtx);
+       if (videoStream == -1) {
+               avformat_close_input(&pFormatCtx);
                return 0;
        }
 
        pCodecCtx = pFormatCtx->streams[videoStream]->codec;
 
-               /* Find the decoder for the video stream */
-       pCodec=avcodec_find_decoder(pCodecCtx->codec_id);
-       if (pCodec==NULL) {
-               av_close_input_file(pFormatCtx);
+       /* Find the decoder for the video stream */
+       pCodec = avcodec_find_decoder(pCodecCtx->codec_id);
+       if (pCodec == NULL) {
+               avformat_close_input(&pFormatCtx);
                return 0;
        }
 
-       if (avcodec_open(pCodecCtx, pCodec)<0) {
-               av_close_input_file(pFormatCtx);
+       if (avcodec_open2(pCodecCtx, pCodec, NULL) < 0) {
+               avformat_close_input(&pFormatCtx);
                return 0;
        }
 
        avcodec_close(pCodecCtx);
-       av_close_input_file(pFormatCtx);
+       avformat_close_input(&pFormatCtx);
 
        return 1;
 }
 #endif
 
 #ifdef WITH_REDCODE
-static int isredcode(const char * filename)
+static int isredcode(const char *filename)
 {
-       struct redcode_handle * h = redcode_open(filename);
+       struct redcode_handle *h = redcode_open(filename);
        if (!h) {
                return 0;
        }
@@ -327,38 +384,40 @@ static int isredcode(const char * filename)
 
 #endif
 
-int imb_get_anim_type(const char * name)
+int imb_get_anim_type(const char *name)
 {
        int type;
-       struct stat st;
+       BLI_stat_t st;
 
        if (UTIL_DEBUG) printf("in getanimtype: %s\n", name);
 
 #ifndef _WIN32
-#      ifdef WITH_QUICKTIME
+#   ifdef WITH_QUICKTIME
        if (isqtime(name)) return (ANIM_QTIME);
-#      endif
-#      ifdef WITH_FFMPEG
+#   endif
+#   ifdef WITH_FFMPEG
        /* stat test below fails on large files > 4GB */
        if (isffmpeg(name)) return (ANIM_FFMPEG);
-#      endif
-       if (stat(name,&st) == -1) return(0);
+#   endif
+       if (BLI_stat(name, &st) == -1) return(0);
        if (((st.st_mode) & S_IFMT) != S_IFREG) return(0);
 
        if (isavi(name)) return (ANIM_AVI);
 
        if (ismovie(name)) return (ANIM_MOVIE);
 #else
-       if (stat(name,&st) == -1) return(0);
+       if (BLI_stat(name, &st) == -1) return(0);
        if (((st.st_mode) & S_IFMT) != S_IFREG) return(0);
 
        if (ismovie(name)) return (ANIM_MOVIE);
-#      ifdef WITH_QUICKTIME
+#   ifdef WITH_QUICKTIME
        if (isqtime(name)) return (ANIM_QTIME);
-#      endif
-#      ifdef WITH_FFMPEG
+#   endif
+#   ifdef WITH_FFMPEG
        if (isffmpeg(name)) return (ANIM_FFMPEG);
-#      endif
+#   endif
+
+
        if (isavi(name)) return (ANIM_AVI);
 #endif
 #ifdef WITH_REDCODE
@@ -372,42 +431,11 @@ int imb_get_anim_type(const char * name)
        return ANIM_NONE;
 }
  
-int IMB_isanim(const char *filename)
+bool IMB_isanim(const char *filename)
 {
        int type;
        
-       if (U.uiflag & USER_FILTERFILEEXTS) {
-               if (G.have_quicktime) {
-                       if (BLI_testextensie(filename, ".avi")   ||
-                           BLI_testextensie(filename, ".flc")   ||
-                           BLI_testextensie(filename, ".dv")    ||
-                           BLI_testextensie(filename, ".r3d")   ||
-                           BLI_testextensie(filename, ".mov")   ||
-                           BLI_testextensie(filename, ".movie") ||
-                           BLI_testextensie(filename, ".mv"))
-                       {
-                               type = imb_get_anim_type(filename);
-                       }
-                       else {
-                               return(FALSE);                  
-                       }
-               }
-               else { /* no quicktime */
-                       if (BLI_testextensie(filename, ".avi") ||
-                           BLI_testextensie(filename, ".dv")  ||
-                           BLI_testextensie(filename, ".r3d") ||
-                           BLI_testextensie(filename, ".mv"))
-                       {
-                               type = imb_get_anim_type(filename);
-                       }
-                       else {
-                               return(FALSE);
-                       }
-               }
-       }
-       else { /* no FILTERFILEEXTS */
-               type = imb_get_anim_type(filename);
-       }
+       type = imb_get_anim_type(filename);
        
-       return (type && type!=ANIM_SEQUENCE);
+       return (type && type != ANIM_SEQUENCE);
 }