]> git.sesse.net Git - casparcg/blobdiff - modules/ffmpeg/ffmpeg.cpp
[ffmpeg] Ported 2.0.7 ffmpeg producer to 2.1.0 while still keeping the usage of the...
[casparcg] / modules / ffmpeg / ffmpeg.cpp
index 7a151f05b62879d6754560a5e736c08c307f6e84..ab500c3e050240558ef1f511dd7689283cfaa1dc 100644 (file)
 #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>
 
@@ -46,7 +50,7 @@
 #pragma warning (disable : 4996)
 #endif
 
-extern "C" 
+extern "C"
 {
        #define __STDC_CONSTANT_MACROS
        #define __STDC_LIMIT_MACROS
@@ -54,45 +58,45 @@ extern "C"
        #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) 
-{ 
+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: 
-               { 
+
+       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: 
-               { 
+                               my_mutex->lock();
+                       break;
+               }
+               case AV_LOCK_RELEASE:
+               {
                        if(my_mutex)
-                               my_mutex->unlock(); 
-                       break; 
-               } 
-               case AV_LOCK_DESTROY: 
-               { 
+                               my_mutex->unlock();
+                       break;
+               }
+               case AV_LOCK_DESTROY:
+               {
                        delete my_mutex;
                        *mutex = nullptr;
-                       break; 
-               } 
-       } 
-       return 0; 
-} 
+                       break;
+               }
+       }
+       return 0;
+}
 
 static void sanitize(uint8_t *line)
 {
@@ -115,15 +119,15 @@ void log_callback(void* ptr, int level, const char* fmt, va_list vl)
     if(level > av_log_get_level())
         return;
     line[0]=0;
-       
+
 #undef fprintf
-    if(print_prefix && avc) 
+    if(print_prefix && avc)
        {
-        if (avc->parent_log_context_offset) 
+        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, "[%s @ %p] ", (*parent)->item_name(parent), parent);
         }
         std::sprintf(line + strlen(line), "[%s @ %p] ", avc->item_name(ptr), ptr);
     }
@@ -131,7 +135,7 @@ void log_callback(void* ptr, int level, const char* fmt, va_list vl)
     std::vsprintf(line + strlen(line), fmt, vl);
 
     print_prefix = strlen(line) && line[strlen(line)-1] == '\n';
-       
+
     //if(print_prefix && !strcmp(line, prev)){
     //    count++;
     //    if(is_atty==1)
@@ -148,7 +152,7 @@ void log_callback(void* ptr, int level, const char* fmt, va_list vl)
        auto len = strlen(line);
        if(len > 0)
                line[len-1] = 0;
-       
+
        if(level == AV_LOG_DEBUG)
                CASPAR_LOG(debug) << L"[ffmpeg] " << line;
        else if(level == AV_LOG_INFO)
@@ -235,36 +239,85 @@ 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_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();
-       
-       dependencies.consumer_registry->register_consumer_factory(L"FFMpeg Consumer", create_consumer, describe_consumer);
+       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", create_producer, describe_producer);
-       
-       dependencies.media_info_repo->register_extractor(
+       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
                        {
-                               // TODO: merge thumbnail generation from 2.0
-                               //auto disable_logging = temporary_disable_logging_for_thread(true);
+                               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))
+                               if (!is_valid_file(file, true))
                                        return false;
 
                                info.clip_type = L"MOVIE";
@@ -287,5 +340,4 @@ void uninit()
     avformat_network_deinit();
        av_lockmgr_register(nullptr);
 }
-
-}}
\ No newline at end of file
+}}