]> git.sesse.net Git - casparcg/blobdiff - modules/ffmpeg/ffmpeg.cpp
[ffmpeg] Never log AV_LOG_TRACE. Not even when we have trace set.
[casparcg] / modules / ffmpeg / ffmpeg.cpp
index 02d302e937b83b240be4130837dcbaf944c9c825..0b9f8611f1a120aa42c82a2deca83c971e1f82d2 100644 (file)
-/*\r
-* Copyright (c) 2011 Sveriges Television AB <info@casparcg.com>\r
-*\r
-* This file is part of CasparCG (www.casparcg.com).\r
-*\r
-* CasparCG is free software: you can redistribute it and/or modify\r
-* it under the terms of the GNU General Public License as published by\r
-* the Free Software Foundation, either version 3 of the License, or\r
-* (at your option) any later version.\r
-*\r
-* CasparCG is distributed in the hope that it will be useful,\r
-* but WITHOUT ANY WARRANTY; without even the implied warranty of\r
-* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
-* GNU General Public License for more details.\r
-*\r
-* You should have received a copy of the GNU General Public License\r
-* along with CasparCG. If not, see <http://www.gnu.org/licenses/>.\r
-*\r
-* Author: Robert Nagy, ronag89@gmail.com\r
-*/\r
-\r
-#include "StdAfx.h"\r
-\r
-#include "consumer/ffmpeg_consumer.h"\r
-#include "producer/ffmpeg_producer.h"\r
-\r
-#include <common/log/log.h>\r
-\r
-#include <core/consumer/frame_consumer.h>\r
-#include <core/producer/frame_producer.h>\r
-\r
-#include <tbb/recursive_mutex.h>\r
-\r
-#if defined(_MSC_VER)\r
-#pragma warning (disable : 4244)\r
-#pragma warning (disable : 4603)\r
-#pragma warning (disable : 4996)\r
-#endif\r
-\r
-extern "C" \r
-{\r
-       #define __STDC_CONSTANT_MACROS\r
-       #define __STDC_LIMIT_MACROS\r
-       #include <libavformat/avformat.h>\r
-       #include <libswscale/swscale.h>\r
-       #include <libavutil/avutil.h>\r
-       #include <libavfilter/avfilter.h>\r
-}\r
-\r
-namespace caspar { namespace ffmpeg {\r
-       \r
-int ffmpeg_lock_callback(void **mutex, enum AVLockOp op) \r
-{ \r
-       if(!mutex)\r
-               return 0;\r
-\r
-       auto my_mutex = reinterpret_cast<tbb::recursive_mutex*>(*mutex);\r
-       \r
-       switch(op) \r
-       { \r
-               case AV_LOCK_CREATE: \r
-               { \r
-                       *mutex = new tbb::recursive_mutex(); \r
-                       break; \r
-               } \r
-               case AV_LOCK_OBTAIN: \r
-               { \r
-                       if(my_mutex)\r
-                               my_mutex->lock(); \r
-                       break; \r
-               } \r
-               case AV_LOCK_RELEASE: \r
-               { \r
-                       if(my_mutex)\r
-                               my_mutex->unlock(); \r
-                       break; \r
-               } \r
-               case AV_LOCK_DESTROY: \r
-               { \r
-                       delete my_mutex;\r
-                       *mutex = nullptr;\r
-                       break; \r
-               } \r
-       } \r
-       return 0; \r
-} \r
-\r
-static void sanitize(uint8_t *line)\r
-{\r
-    while(*line)\r
-       {\r
-        if(*line < 0x08 || (*line > 0x0D && *line < 0x20))\r
-            *line='?';\r
-        line++;\r
-    }\r
-}\r
-\r
-void log_callback(void* ptr, int level, const char* fmt, va_list vl)\r
-{\r
-    static int print_prefix=1;\r
-    static int count;\r
-    static char prev[1024];\r
-    char line[8192];\r
-    static int is_atty;\r
-    AVClass* avc= ptr ? *(AVClass**)ptr : NULL;\r
-    if(level > av_log_get_level())\r
-        return;\r
-    line[0]=0;\r
-       \r
-#undef fprintf\r
-    if(print_prefix && avc) \r
-       {\r
-        if (avc->parent_log_context_offset) \r
-               {\r
-            AVClass** parent= *(AVClass***)(((uint8_t*)ptr) + avc->parent_log_context_offset);\r
-            if(parent && *parent)\r
-                std::sprintf(line, "[%s @ %p] ", (*parent)->item_name(parent), parent);            \r
-        }\r
-        std::sprintf(line + strlen(line), "[%s @ %p] ", avc->item_name(ptr), ptr);\r
-    }\r
-\r
-    std::vsprintf(line + strlen(line), fmt, vl);\r
-\r
-    print_prefix = strlen(line) && line[strlen(line)-1] == '\n';\r
-       \r
-    //if(print_prefix && !strcmp(line, prev)){\r
-    //    count++;\r
-    //    if(is_atty==1)\r
-    //        fprintf(stderr, "    Last message repeated %d times\r", count);\r
-    //    return;\r
-    //}\r
-    //if(count>0){\r
-    //    fprintf(stderr, "    Last message repeated %d times\n", count);\r
-    //    count=0;\r
-    //}\r
-    strcpy(prev, line);\r
-    sanitize((uint8_t*)line);\r
-       \r
-       if(level == AV_LOG_DEBUG)\r
-               CASPAR_LOG(debug) << L"[ffmpeg] " << line;\r
-       else if(level == AV_LOG_INFO)\r
-               CASPAR_LOG(info) << L"[ffmpeg] " << line;\r
-       else if(level == AV_LOG_WARNING)\r
-               CASPAR_LOG(warning) << L"[ffmpeg] " << line;\r
-       else if(level == AV_LOG_ERROR)\r
-               CASPAR_LOG(error) << L"[ffmpeg] " << line;\r
-       else if(level == AV_LOG_FATAL)\r
-               CASPAR_LOG(fatal) << L"[ffmpeg] " << line;\r
-       else\r
-               CASPAR_LOG(trace) << L"[ffmpeg] " << line;\r
-\r
-    //colored_fputs(av_clip(level>>3, 0, 6), line);\r
-}\r
-\r
-//static int query_yadif_formats(AVFilterContext *ctx)\r
-//{\r
-//    static const int pix_fmts[] = {\r
-//        PIX_FMT_YUV444P,\r
-//        PIX_FMT_YUV422P,\r
-//        PIX_FMT_YUV420P,\r
-//        PIX_FMT_YUV410P,\r
-//        PIX_FMT_YUV411P,\r
-//        PIX_FMT_GRAY8,\r
-//        PIX_FMT_YUVJ444P,\r
-//        PIX_FMT_YUVJ422P,\r
-//        PIX_FMT_YUVJ420P,\r
-//        AV_NE( PIX_FMT_GRAY16BE, PIX_FMT_GRAY16LE ),\r
-//        PIX_FMT_YUV440P,\r
-//        PIX_FMT_YUVJ440P,\r
-//        AV_NE( PIX_FMT_YUV444P16BE, PIX_FMT_YUV444P16LE ),\r
-//        AV_NE( PIX_FMT_YUV422P16BE, PIX_FMT_YUV422P16LE ),\r
-//        AV_NE( PIX_FMT_YUV420P16BE, PIX_FMT_YUV420P16LE ),\r
-//        PIX_FMT_YUVA420P,\r
-//        PIX_FMT_NONE\r
-//    };\r
-//    avfilter_set_common_pixel_formats(ctx, avfilter_make_format_list(pix_fmts));\r
-//\r
-//    return 0;\r
-//}\r
-//\r
-//#pragma warning (push)\r
-//#pragma warning (disable : 4706)\r
-//void fix_yadif_filter_format_query()\r
-//{\r
-//     AVFilter** filter = nullptr;\r
-//    while((filter = av_filter_next(filter)) && *filter)\r
-//     {\r
-//             if(strstr((*filter)->name, "yadif") != 0)\r
-//                     (*filter)->query_formats = query_yadif_formats;\r
-//     }\r
-//}\r
-//#pragma warning (pop)\r
-\r
-void init()\r
-{\r
-       av_lockmgr_register(ffmpeg_lock_callback);\r
-       av_log_set_callback(log_callback);\r
-    avfilter_register_all();\r
-       //fix_yadif_filter_format_query();\r
-       av_register_all();\r
-    avformat_network_init();\r
-       avcodec_init();\r
-    avcodec_register_all();\r
-       \r
-       core::register_consumer_factory([](const std::vector<std::wstring>& params){return create_consumer(params);});\r
-       core::register_producer_factory(create_producer);\r
-}\r
-\r
-void uninit()\r
-{\r
-       avfilter_uninit();\r
-    avformat_network_deinit();\r
-       av_lockmgr_register(nullptr);\r
-}\r
-\r
-std::wstring make_version(unsigned int ver)\r
-{\r
-       std::wstringstream str;\r
-       str << ((ver >> 16) & 0xFF) << L"." << ((ver >> 8) & 0xFF) << L"." << ((ver >> 0) & 0xFF);\r
-       return str.str();\r
-}\r
-\r
-std::wstring get_avcodec_version()\r
-{\r
-       return make_version(avcodec_version());\r
-}\r
-\r
-std::wstring get_avformat_version()\r
-{\r
-       return make_version(avformat_version());\r
-}\r
-\r
-std::wstring get_avutil_version()\r
-{\r
-       return make_version(avutil_version());\r
-}\r
-\r
-std::wstring get_avfilter_version()\r
-{\r
-       return make_version(avfilter_version());\r
-}\r
-\r
-std::wstring get_swscale_version()\r
-{\r
-       return make_version(swscale_version());\r
-}\r
-\r
-}}
\ No newline at end of file
+/*
+* Copyright (c) 2011 Sveriges Television AB <info@casparcg.com>
+*
+* This file is part of CasparCG (www.casparcg.com).
+*
+* CasparCG is free software: you can redistribute it and/or modify
+* it under the terms of the GNU General Public License as published by
+* the Free Software Foundation, either version 3 of the License, or
+* (at your option) any later version.
+*
+* CasparCG is distributed in the hope that it will be useful,
+* but WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+* GNU General Public License for more details.
+*
+* You should have received a copy of the GNU General Public License
+* along with CasparCG. If not, see <http://www.gnu.org/licenses/>.
+*
+* Author: Robert Nagy, ronag89@gmail.com
+*/
+
+#include "StdAfx.h"
+
+#include "ffmpeg.h"
+
+#include "consumer/ffmpeg_consumer.h"
+#include "consumer/streaming_consumer.h"
+#include "producer/ffmpeg_producer.h"
+#include "producer/util/util.h"
+
+#include <common/log.h>
+#include <common/os/general_protection_fault.h>
+
+#include <core/consumer/frame_consumer.h>
+#include <core/frame/draw_frame.h>
+#include <core/producer/frame_producer.h>
+#include <core/producer/media_info/media_info.h>
+#include <core/producer/media_info/media_info_repository.h>
+#include <core/system_info_provider.h>
+
+#include <boost/property_tree/ptree.hpp>
+#include <boost/thread/tss.hpp>
+#include <boost/bind.hpp>
+
+#include <tbb/recursive_mutex.h>
+
+#if defined(_MSC_VER)
+#pragma warning (disable : 4244)
+#pragma warning (disable : 4603)
+#pragma warning (disable : 4996)
+#endif
+
+extern "C"
+{
+       #define __STDC_CONSTANT_MACROS
+       #define __STDC_LIMIT_MACROS
+       #include <libavformat/avformat.h>
+       #include <libswscale/swscale.h>
+       #include <libavutil/avutil.h>
+       #include <libavfilter/avfilter.h>
+       #include <libavdevice/avdevice.h>
+}
+
+namespace caspar { namespace ffmpeg {
+int ffmpeg_lock_callback(void **mutex, enum AVLockOp op)
+{
+       if(!mutex)
+               return 0;
+
+       auto my_mutex = reinterpret_cast<tbb::recursive_mutex*>(*mutex);
+
+       switch(op)
+       {
+               case AV_LOCK_CREATE:
+               {
+                       *mutex = new tbb::recursive_mutex();
+                       break;
+               }
+               case AV_LOCK_OBTAIN:
+               {
+                       if(my_mutex)
+                               my_mutex->lock();
+                       break;
+               }
+               case AV_LOCK_RELEASE:
+               {
+                       if(my_mutex)
+                               my_mutex->unlock();
+                       break;
+               }
+               case AV_LOCK_DESTROY:
+               {
+                       delete my_mutex;
+                       *mutex = nullptr;
+                       break;
+               }
+       }
+       return 0;
+}
+
+static void sanitize(uint8_t *line)
+{
+       while(*line)
+       {
+               if(*line < 0x08 || (*line > 0x0D && *line < 0x20))
+                       *line='?';
+               line++;
+       }
+}
+
+void log_callback(void* ptr, int level, const char* fmt, va_list vl)
+{
+       static int print_prefix=1;
+       static char prev[1024];
+       char line[8192];
+       AVClass* avc= ptr ? *(AVClass**)ptr : NULL;
+       if (level > AV_LOG_DEBUG)
+               return;
+       line[0]=0;
+
+#undef fprintf
+       if(print_prefix && avc)
+       {
+               if (avc->parent_log_context_offset)
+               {
+                       AVClass** parent= *(AVClass***)(((uint8_t*)ptr) + avc->parent_log_context_offset);
+                       if(parent && *parent)
+                               std::sprintf(line, "[%s @ %p] ", (*parent)->item_name(parent), parent);
+               }
+               std::sprintf(line + strlen(line), "[%s @ %p] ", avc->item_name(ptr), ptr);
+       }
+
+       std::vsprintf(line + strlen(line), fmt, vl);
+
+       print_prefix = strlen(line) && line[strlen(line)-1] == '\n';
+
+       strcpy(prev, line);
+       sanitize((uint8_t*)line);
+
+       auto len = strlen(line);
+       if(len > 0)
+               line[len-1] = 0;
+
+       try
+       {
+               if (level == AV_LOG_VERBOSE)
+                       CASPAR_LOG(debug) << L"[ffmpeg] " << line;
+               else if (level == AV_LOG_INFO)
+                       CASPAR_LOG(info) << L"[ffmpeg] " << line;
+               else if (level == AV_LOG_WARNING)
+                       CASPAR_LOG(warning) << L"[ffmpeg] " << line;
+               else if (level == AV_LOG_ERROR)
+                       CASPAR_LOG(error) << L"[ffmpeg] " << line;
+               else if (level == AV_LOG_FATAL)
+                       CASPAR_LOG(fatal) << L"[ffmpeg] " << line;
+               else
+                       CASPAR_LOG(trace) << L"[ffmpeg] " << line;
+       }
+       catch (...)
+       {
+       }
+}
+
+std::wstring make_version(unsigned int ver)
+{
+       std::wstringstream str;
+       str << ((ver >> 16) & 0xFF) << L"." << ((ver >> 8) & 0xFF) << L"." << ((ver >> 0) & 0xFF);
+       return str.str();
+}
+
+std::wstring avcodec_version()
+{
+       return make_version(::avcodec_version());
+}
+
+std::wstring avformat_version()
+{
+       return make_version(::avformat_version());
+}
+
+std::wstring avutil_version()
+{
+       return make_version(::avutil_version());
+}
+
+std::wstring avfilter_version()
+{
+       return make_version(::avfilter_version());
+}
+
+std::wstring swscale_version()
+{
+       return make_version(::swscale_version());
+}
+bool& get_quiet_logging_for_thread()
+{
+       static boost::thread_specific_ptr<bool> quiet_logging_for_thread;
+
+       auto local = quiet_logging_for_thread.get();
+
+       if (!local)
+       {
+               local = new bool(false);
+               quiet_logging_for_thread.reset(local);
+       }
+
+       return *local;
+}
+
+void enable_quiet_logging_for_thread()
+{
+       get_quiet_logging_for_thread() = true;
+}
+
+bool is_logging_quiet_for_thread()
+{
+       return get_quiet_logging_for_thread();
+}
+
+std::shared_ptr<void> temporary_enable_quiet_logging_for_thread(bool enable)
+{
+       if (!enable || is_logging_quiet_for_thread())
+               return std::shared_ptr<void>();
+
+       get_quiet_logging_for_thread() = true;
+
+       return std::shared_ptr<void>(nullptr, [](void*)
+       {
+               get_quiet_logging_for_thread() = false; // Only works correctly if destructed in same thread as original caller.
+       });
+}
+
+void log_for_thread(void* ptr, int level, const char* fmt, va_list vl)
+{
+       ensure_gpf_handler_installed_for_thread("ffmpeg-thread");
+
+       int min_level = is_logging_quiet_for_thread() ? AV_LOG_DEBUG : AV_LOG_FATAL;
+
+       log_callback(ptr, std::max(level, min_level), fmt, vl);
+}
+
+void init(core::module_dependencies dependencies)
+{
+       av_lockmgr_register(ffmpeg_lock_callback);
+       av_log_set_callback(log_for_thread);
+
+       avfilter_register_all();
+       //fix_yadif_filter_format_query();
+       av_register_all();
+       avformat_network_init();
+       avcodec_register_all();
+       avdevice_register_all();
+
+       auto info_repo = dependencies.media_info_repo;
+
+       dependencies.consumer_registry->register_consumer_factory(L"FFmpeg Consumer", create_consumer, describe_consumer);
+       dependencies.consumer_registry->register_consumer_factory(L"Streaming Consumer", create_streaming_consumer, describe_streaming_consumer);
+       dependencies.consumer_registry->register_preconfigured_consumer_factory(L"file", create_preconfigured_consumer);
+       dependencies.consumer_registry->register_preconfigured_consumer_factory(L"stream", create_preconfigured_streaming_consumer);
+       dependencies.producer_registry->register_producer_factory(L"FFmpeg Producer", boost::bind(&create_producer, _1, _2, info_repo), describe_producer);
+       dependencies.producer_registry->register_thumbnail_producer(boost::bind(&create_thumbnail_frame, _1, _2, info_repo));
+
+       info_repo->register_extractor(
+                       [](const std::wstring& file, const std::wstring& extension, core::media_info& info) -> bool
+                       {
+                               auto quiet_logging = temporary_enable_quiet_logging_for_thread(true);
+                               if (extension == L".WAV" || extension == L".MP3")
+                               {
+                                       info.clip_type = L"AUDIO";
+                                       return true;
+                               }
+
+                               if (!is_valid_file(file, true))
+                                       return false;
+
+                               info.clip_type = L"MOVIE";
+
+                               return try_get_duration(file, info.duration, info.time_base);
+                       });
+       dependencies.system_info_provider_repo->register_system_info_provider([](boost::property_tree::wptree& info)
+       {
+               info.add(L"system.ffmpeg.avcodec", avcodec_version());
+               info.add(L"system.ffmpeg.avformat", avformat_version());
+               info.add(L"system.ffmpeg.avfilter", avfilter_version());
+               info.add(L"system.ffmpeg.avutil", avutil_version());
+               info.add(L"system.ffmpeg.swscale", swscale_version());
+       });
+}
+
+void uninit()
+{
+       avfilter_uninit();
+       avformat_network_deinit();
+       av_lockmgr_register(nullptr);
+}
+}}