]> git.sesse.net Git - casparcg/blobdiff - core/producer/frame_producer.cpp
fixed problem with wrong alpha in thumbnails.
[casparcg] / core / producer / frame_producer.cpp
index 04400539c269a031de41aa792adf145e326395cb..066f7fce618b4764c2b4e39fb6d034adde925be1 100644 (file)
 /*\r
-* copyright (c) 2010 Sveriges Television AB <info@casparcg.com>\r
+* Copyright 2013 Sveriges Television AB http://casparcg.com/\r
 *\r
-*  This file is part of CasparCG.\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
+* 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
+* 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 "frame_producer.h"\r
 #include "frame/basic_frame.h"\r
-#include "frame/audio_transform.h"\r
+#include "frame/frame_transform.h"\r
+\r
+#include "../parameters/parameters.h"\r
 \r
 #include "color/color_producer.h"\r
 #include "separated/separated_producer.h"\r
 \r
 #include <common/memory/safe_ptr.h>\r
+#include <common/concurrency/executor.h>\r
 #include <common/exception/exceptions.h>\r
+#include <common/utility/move_on_copy.h>\r
 \r
 namespace caspar { namespace core {\r
        \r
 std::vector<const producer_factory_t> g_factories;\r
+std::vector<const producer_factory_t> g_thumbnail_factories;\r
+       \r
+class destroy_producer_proxy : public frame_producer\r
+{      \r
+       std::unique_ptr<std::shared_ptr<frame_producer>> producer_;\r
+public:\r
+       destroy_producer_proxy(safe_ptr<frame_producer>&& producer) \r
+               : producer_(new std::shared_ptr<frame_producer>(std::move(producer)))\r
+       {\r
+       }\r
 \r
-const safe_ptr<frame_producer>& frame_producer::empty() // nothrow\r
+       ~destroy_producer_proxy()\r
+       {               \r
+               static auto destroyers = std::make_shared<tbb::concurrent_bounded_queue<std::shared_ptr<executor>>>();\r
+               static tbb::atomic<int> destroyer_count;\r
+\r
+               try\r
+               {\r
+                       std::shared_ptr<executor> destroyer;\r
+                       if(!destroyers->try_pop(destroyer))\r
+                       {\r
+                               destroyer.reset(new executor(L"destroyer"));\r
+                               destroyer->set_priority_class(below_normal_priority_class);\r
+                               if(++destroyer_count > 16)\r
+                                       CASPAR_LOG(warning) << L"Potential destroyer dead-lock detected.";\r
+                               CASPAR_LOG(trace) << "Created destroyer: " << destroyer_count;\r
+                       }\r
+                               \r
+                       auto producer = producer_.release();\r
+                       auto pool         = destroyers;\r
+                       destroyer->begin_invoke([=]\r
+                       {\r
+                               std::unique_ptr<std::shared_ptr<frame_producer>> producer2(producer);\r
+\r
+                               auto str = (*producer2)->print();\r
+                               try\r
+                               {\r
+                                       if(!producer->unique())\r
+                                               CASPAR_LOG(trace) << str << L" Not destroyed on asynchronous destruction thread: " << producer->use_count();\r
+                                       else\r
+                                               CASPAR_LOG(trace) << str << L" Destroying on asynchronous destruction thread.";\r
+                               }\r
+                               catch(...){}\r
+                                                               \r
+                               producer2.reset();\r
+                               pool->push(destroyer);\r
+                       }); \r
+               }\r
+               catch(...)\r
+               {\r
+                       CASPAR_LOG_CURRENT_EXCEPTION();\r
+                       try\r
+                       {\r
+                               producer_.reset();\r
+                       }\r
+                       catch(...){}\r
+               }\r
+       }\r
+\r
+       virtual safe_ptr<basic_frame>                                                           receive(int hints) override                                                                                             {return (*producer_)->receive(hints);}\r
+       virtual safe_ptr<basic_frame>                                                           last_frame() const override                                                                                             {return (*producer_)->last_frame();}\r
+       virtual safe_ptr<basic_frame>                                                           create_thumbnail_frame() override                                                                               {return (*producer_)->create_thumbnail_frame();}\r
+       virtual std::wstring                                                                            print() const override                                                                                                  {return (*producer_)->print();}\r
+       virtual boost::property_tree::wptree                                            info() const override                                                                                                   {return (*producer_)->info();}\r
+       virtual boost::unique_future<std::wstring>                                      call(const std::wstring& str) override                                                                  {return (*producer_)->call(str);}\r
+       virtual safe_ptr<frame_producer>                                                        get_following_producer() const override                                                                 {return (*producer_)->get_following_producer();}\r
+       virtual void                                                                                            set_leading_producer(const safe_ptr<frame_producer>& producer) override {(*producer_)->set_leading_producer(producer);}\r
+       virtual uint32_t                                                                                        nb_frames() const override                                                                                              {return (*producer_)->nb_frames();}\r
+       virtual monitor::source&                                                                        monitor_output()                                                                                                                {return (*producer_)->monitor_output();}\r
+};\r
+\r
+safe_ptr<core::frame_producer> create_producer_destroy_proxy(safe_ptr<core::frame_producer> producer)\r
 {\r
-       struct empty_frame_producer : public frame_producer\r
+       return make_safe<destroy_producer_proxy>(std::move(producer));\r
+}\r
+\r
+class print_producer_proxy : public frame_producer\r
+{      \r
+       std::shared_ptr<frame_producer> producer_;\r
+public:\r
+       print_producer_proxy(safe_ptr<frame_producer>&& producer) \r
+               : producer_(std::move(producer))\r
+       {\r
+               CASPAR_LOG(info) << producer_->print() << L" Initialized.";\r
+       }\r
+\r
+       ~print_producer_proxy()\r
+       {               \r
+               auto str = producer_->print();\r
+               CASPAR_LOG(trace) << str << L" Uninitializing.";\r
+               producer_.reset();\r
+               CASPAR_LOG(info) << str << L" Uninitialized.";\r
+       }\r
+\r
+       virtual safe_ptr<basic_frame>                                                           receive(int hints) override                                                                                             {return (producer_)->receive(hints);}\r
+       virtual safe_ptr<basic_frame>                                                           last_frame() const override                                                                                             {return (producer_)->last_frame();}\r
+       virtual safe_ptr<basic_frame>                                                           create_thumbnail_frame() override                                                                               {return (producer_)->create_thumbnail_frame();}\r
+       virtual std::wstring                                                                            print() const override                                                                                                  {return (producer_)->print();}\r
+       virtual boost::property_tree::wptree                                            info() const override                                                                                                   {return (producer_)->info();}\r
+       virtual boost::unique_future<std::wstring>                                      call(const std::wstring& str) override                                                                  {return (producer_)->call(str);}\r
+       virtual safe_ptr<frame_producer>                                                        get_following_producer() const override                                                                 {return (producer_)->get_following_producer();}\r
+       virtual void                                                                                            set_leading_producer(const safe_ptr<frame_producer>& producer) override {(producer_)->set_leading_producer(producer);}\r
+       virtual uint32_t                                                                                        nb_frames() const override                                                                                              {return (producer_)->nb_frames();}\r
+       virtual monitor::source&                                                                        monitor_output()                                                                                                                {return (producer_)->monitor_output();}\r
+};\r
+\r
+safe_ptr<core::frame_producer> create_producer_print_proxy(safe_ptr<core::frame_producer> producer)\r
+{\r
+       return make_safe<print_producer_proxy>(std::move(producer));\r
+}\r
+\r
+class last_frame_producer : public frame_producer\r
+{\r
+       const std::wstring                      print_;\r
+       const safe_ptr<basic_frame>     frame_;\r
+       const uint32_t                          nb_frames_;\r
+public:\r
+       last_frame_producer(const safe_ptr<frame_producer>& producer) \r
+               : print_(producer->print())\r
+               , frame_(producer->last_frame() != basic_frame::eof() ? producer->last_frame() : basic_frame::empty())\r
+               , nb_frames_(producer->nb_frames())\r
        {\r
-               virtual safe_ptr<basic_frame> receive(int){return basic_frame::empty();}\r
-               virtual safe_ptr<basic_frame> last_frame() const{return basic_frame::empty();}\r
-               virtual void set_frame_factory(const safe_ptr<frame_factory>&){}\r
-               virtual int64_t nb_frames() const {return 0;}\r
-               virtual std::wstring print() const { return L"empty";}\r
-       };\r
+       }\r
+       \r
+       virtual safe_ptr<basic_frame> receive(int){return frame_;}\r
+       virtual safe_ptr<core::basic_frame> last_frame() const{return frame_;}\r
+       virtual safe_ptr<core::basic_frame> create_thumbnail_frame() {return frame_;}\r
+       virtual std::wstring print() const{return L"dummy[" + print_ + L"]";}\r
+       virtual uint32_t nb_frames() const {return nb_frames_;} \r
+       virtual boost::property_tree::wptree info() const override\r
+       {\r
+               boost::property_tree::wptree info;\r
+               info.add(L"type", L"last-frame-producer");\r
+               return info;\r
+       }\r
+       virtual monitor::source& monitor_output()\r
+       {\r
+               static monitor::subject monitor_subject("");\r
+               return monitor_subject;\r
+       }\r
+};\r
+\r
+struct empty_frame_producer : public frame_producer\r
+{\r
+       virtual safe_ptr<basic_frame> receive(int){return basic_frame::empty();}\r
+       virtual safe_ptr<basic_frame> last_frame() const{return basic_frame::empty();}\r
+       virtual void set_frame_factory(const safe_ptr<frame_factory>&){}\r
+       virtual uint32_t nb_frames() const {return 0;}\r
+       virtual std::wstring print() const { return L"empty";}\r
+       \r
+       virtual boost::property_tree::wptree info() const override\r
+       {\r
+               boost::property_tree::wptree info;\r
+               info.add(L"type", L"empty-producer");\r
+               return info;\r
+       }\r
+\r
+       virtual monitor::source& monitor_output()\r
+       {\r
+               static monitor::subject monitor_subject("");\r
+               return monitor_subject;\r
+       }\r
+};\r
+\r
+const safe_ptr<frame_producer>& frame_producer::empty() // nothrow\r
+{\r
        static safe_ptr<frame_producer> producer = make_safe<empty_frame_producer>();\r
        return producer;\r
-}      \r
+}\r
+\r
+safe_ptr<basic_frame> frame_producer::create_thumbnail_frame()\r
+{\r
+       return basic_frame::empty();\r
+}\r
 \r
 safe_ptr<basic_frame> receive_and_follow(safe_ptr<frame_producer>& producer, int hints)\r
 {      \r
@@ -55,11 +220,13 @@ safe_ptr<basic_frame> receive_and_follow(safe_ptr<frame_producer>& producer, int
        {\r
                CASPAR_LOG(info) << producer->print() << " End Of File.";\r
                auto following = producer->get_following_producer();\r
-               following->set_leading_producer(producer);\r
-               producer = std::move(following);                \r
-                               \r
-               if(producer == frame_producer::empty())\r
-                       return basic_frame::eof();\r
+               if(following != frame_producer::empty())\r
+               {\r
+                       following->set_leading_producer(producer);\r
+                       producer = std::move(following);\r
+               }\r
+               else\r
+                       producer = make_safe<last_frame_producer>(producer);\r
 \r
                return receive_and_follow(producer, hints);\r
        }\r
@@ -71,13 +238,18 @@ void register_producer_factory(const producer_factory_t& factory)
        g_factories.push_back(factory);\r
 }\r
 \r
-safe_ptr<core::frame_producer> do_create_producer(const safe_ptr<frame_factory>& my_frame_factory, const std::vector<std::wstring>& params)\r
+void register_thumbnail_producer_factory(const producer_factory_t& factory)\r
+{\r
+       g_thumbnail_factories.push_back(factory);\r
+}\r
+\r
+safe_ptr<core::frame_producer> do_create_producer(const safe_ptr<frame_factory>& my_frame_factory, const core::parameters& params, const std::vector<const producer_factory_t>& factories, bool throw_on_fail = false)\r
 {\r
        if(params.empty())\r
                BOOST_THROW_EXCEPTION(invalid_argument() << arg_name_info("params") << arg_value_info(""));\r
        \r
        auto producer = frame_producer::empty();\r
-       std::any_of(g_factories.begin(), g_factories.end(), [&](const producer_factory_t& factory) -> bool\r
+       std::any_of(factories.begin(), factories.end(), [&](const producer_factory_t& factory) -> bool\r
                {\r
                        try\r
                        {\r
@@ -85,46 +257,98 @@ safe_ptr<core::frame_producer> do_create_producer(const safe_ptr<frame_factory>&
                        }\r
                        catch(...)\r
                        {\r
-                               //CASPAR_LOG_CURRENT_EXCEPTION();\r
+                               if (throw_on_fail)\r
+                                       throw;\r
+                               else\r
+                                       CASPAR_LOG_CURRENT_EXCEPTION();\r
                        }\r
                        return producer != frame_producer::empty();\r
                });\r
 \r
        if(producer == frame_producer::empty())\r
                producer = create_color_producer(my_frame_factory, params);\r
+       return producer;\r
+}\r
+\r
+safe_ptr<core::frame_producer> create_producer(const safe_ptr<frame_factory>& my_frame_factory, const core::parameters& params)\r
+{      \r
+       auto producer = do_create_producer(my_frame_factory, params, g_factories);\r
+       auto key_producer = frame_producer::empty();\r
+       \r
+       std::wstring resource_name = L"";\r
+       auto tokens = parameters::protocol_split(params.at_original(0));\r
+       if (tokens[0].empty())\r
+       {\r
+               resource_name = params.at_original(0);\r
+       }\r
+\r
+       if(!resource_name.empty()) {\r
+       try // to find a key file.\r
+       {\r
+               auto params_copy = params;\r
+               if(params_copy.size() > 0)\r
+               {\r
+                       auto resource_name = params_copy.at_original(0);\r
+                       params_copy.set(0, resource_name + L"_A");\r
+                       key_producer = do_create_producer(my_frame_factory, params_copy, g_factories);                  \r
+                       if(key_producer == frame_producer::empty())\r
+                       {\r
+                               params_copy.set(0, resource_name + L"_ALPHA");\r
+                               key_producer = do_create_producer(my_frame_factory, params_copy, g_factories);  \r
+                       }\r
+               }\r
+       }\r
+       catch(...){}\r
+       }\r
 \r
+       if(producer != frame_producer::empty() && key_producer != frame_producer::empty())\r
+               return create_separated_producer(producer, key_producer);\r
+       \r
        if(producer == frame_producer::empty())\r
        {\r
-               std::wstring str;\r
-               BOOST_FOREACH(auto& param, params)\r
-                       str += param + L" ";\r
+               std::wstring str = params.get_original_string();\r
                BOOST_THROW_EXCEPTION(file_not_found() << msg_info("No match found for supplied commands. Check syntax.") << arg_value_info(narrow(str)));\r
        }\r
 \r
        return producer;\r
 }\r
 \r
+safe_ptr<core::frame_producer> create_thumbnail_producer(const safe_ptr<frame_factory>& my_frame_factory, const std::wstring& media_file)\r
+{\r
+       core::parameters params;\r
+       params.push_back(media_file);\r
 \r
-safe_ptr<core::frame_producer> create_producer(const safe_ptr<frame_factory>& my_frame_factory, const std::vector<std::wstring>& params)\r
-{      \r
-       auto producer = do_create_producer(my_frame_factory, params);\r
+       auto producer = do_create_producer(my_frame_factory, params, g_thumbnail_factories, true);\r
        auto key_producer = frame_producer::empty();\r
        \r
        try // to find a key file.\r
        {\r
                auto params_copy = params;\r
-               if(params_copy.size() > 0)\r
+               if (params_copy.size() > 0)\r
                {\r
-                       params_copy[0] += L"_A";\r
-                       key_producer = do_create_producer(my_frame_factory, params_copy);                       \r
+                       auto resource_name = params_copy.at_original(0);\r
+                       params_copy.set(0, resource_name + L"_A");\r
+                       key_producer = do_create_producer(my_frame_factory, params_copy, g_thumbnail_factories, true);\r
+                       if (key_producer == frame_producer::empty())\r
+                       {\r
+                               params_copy.set(0, resource_name + L"_ALPHA");\r
+                               key_producer = do_create_producer(my_frame_factory, params_copy, g_thumbnail_factories, true);\r
+                       }\r
                }\r
        }\r
        catch(...){}\r
 \r
-       if(key_producer != frame_producer::empty())\r
-               return create_separated_producer(producer, key_producer);\r
-\r
+       if (producer != frame_producer::empty() && key_producer != frame_producer::empty())\r
+               return create_separated_thumbnail_producer(producer, key_producer);\r
+       \r
        return producer;\r
 }\r
 \r
+safe_ptr<core::frame_producer> create_producer(const safe_ptr<frame_factory>& factory, const std::wstring& param)\r
+{\r
+       parameters params;\r
+       params.push_back(param);\r
+       return create_producer(factory, params);\r
+}\r
+\r
 }}
\ No newline at end of file