]> git.sesse.net Git - casparcg/blobdiff - modules/flash/producer/flash_producer.cpp
2.0. flash_producer: Fixed invalid fields in last_frame.
[casparcg] / modules / flash / producer / flash_producer.cpp
index 5880daac13516380211e30000338a9d9269d6d07..c285ba9fcc876ab55a9843bed79fa7e899958dda 100644 (file)
@@ -32,7 +32,7 @@
 \r
 #include <core/producer/frame/basic_frame.h>\r
 #include <core/producer/frame/frame_factory.h>\r
-#include <core/producer/frame/write_frame.h>\r
+#include <core/mixer/write_frame.h>\r
 \r
 #include <common/env.h>\r
 #include <common/concurrency/com_context.h>\r
@@ -47,6 +47,8 @@
 \r
 #include <functional>\r
 \r
+#include <tbb/spin_mutex.h>\r
+\r
 namespace caspar {\r
                \r
 class bitmap\r
@@ -67,6 +69,9 @@ public:
 \r
                bmp_.reset(CreateDIBSection(static_cast<HDC>(hdc_.get()), &info, DIB_RGB_COLORS, reinterpret_cast<void**>(&bmp_data_), 0, 0), DeleteObject);\r
                SelectObject(static_cast<HDC>(hdc_.get()), bmp_.get()); \r
+\r
+               if(!bmp_data_)\r
+                       BOOST_THROW_EXCEPTION(std::bad_alloc());\r
        }\r
 \r
        operator HDC() {return static_cast<HDC>(hdc_.get());}\r
@@ -80,10 +85,48 @@ private:
        std::shared_ptr<void> bmp_;\r
 };\r
 \r
+struct template_host\r
+{\r
+       std::string  field_mode;\r
+       std::string  filename;\r
+       size_t           width;\r
+       size_t           height;\r
+};\r
+\r
+template_host get_template_host(const core::video_format_desc& desc)\r
+{\r
+       std::vector<template_host> template_hosts;\r
+       BOOST_FOREACH(auto& xml_mapping, env::properties().get_child("configuration.producers.template-hosts"))\r
+       {\r
+               try\r
+               {\r
+                       template_host template_host;\r
+                       template_host.field_mode                = xml_mapping.second.get("video-mode", narrow(desc.name));\r
+                       template_host.filename                  = xml_mapping.second.get("filename", "cg.fth");\r
+                       template_host.width                             = xml_mapping.second.get("width", desc.width);\r
+                       template_host.height                    = xml_mapping.second.get("height", desc.height);\r
+                       template_hosts.push_back(template_host);\r
+               }\r
+               catch(...){}\r
+       }\r
+\r
+       auto template_host_it = boost::find_if(template_hosts, [&](template_host template_host){return template_host.field_mode == narrow(desc.name);});\r
+       if(template_host_it == template_hosts.end())\r
+               template_host_it = boost::find_if(template_hosts, [&](template_host template_host){return template_host.field_mode == "";});\r
+\r
+       if(template_host_it != template_hosts.end())\r
+               return *template_host_it;\r
+       \r
+       template_host template_host;\r
+       template_host.filename = "cg.fth";\r
+       template_host.width = desc.width;\r
+       template_host.height = desc.height;\r
+       return template_host;\r
+}\r
+\r
 class flash_renderer\r
 {      \r
        const std::wstring filename_;\r
-       const core::video_format_desc format_desc_;\r
 \r
        const std::shared_ptr<core::frame_factory> frame_factory_;\r
        \r
@@ -96,23 +139,27 @@ class flash_renderer
        boost::timer tick_timer_;\r
 \r
        high_prec_timer timer_;\r
+\r
+       const size_t width_;\r
+       const size_t height_;\r
        \r
 public:\r
-       flash_renderer(const safe_ptr<diagnostics::graph>& graph, const std::shared_ptr<core::frame_factory>& frame_factory, const std::wstring& filename) \r
+       flash_renderer(const safe_ptr<diagnostics::graph>& graph, const std::shared_ptr<core::frame_factory>& frame_factory, const std::wstring& filename, int width, int height\r
                : graph_(graph)\r
                , filename_(filename)\r
-               , format_desc_(frame_factory->get_video_format_desc())\r
                , frame_factory_(frame_factory)\r
                , ax_(nullptr)\r
                , head_(core::basic_frame::empty())\r
-               , bmp_(format_desc_.width, format_desc_.height)\r
-       {\r
+               , bmp_(width, height)\r
+               , width_(width)\r
+               , height_(height)\r
+       {               \r
                graph_->add_guide("frame-time", 0.5f);\r
-               graph_->set_color("frame-time", diagnostics::color(1.0f, 0.0f, 0.0f));  \r
+               graph_->set_color("frame-time", diagnostics::color(0.1f, 1.0f, 0.1f));\r
                graph_->add_guide("tick-time", 0.5);\r
-               graph_->set_color("tick-time", diagnostics::color(0.1f, 0.7f, 0.8f));\r
+               graph_->set_color("tick-time", diagnostics::color(0.0f, 0.6f, 0.9f));\r
                graph_->set_color("param", diagnostics::color(1.0f, 0.5f, 0.0f));       \r
-               graph_->set_color("underflow", diagnostics::color(0.6f, 0.3f, 0.9f));                   \r
+               graph_->set_color("skip-sync", diagnostics::color(0.8f, 0.3f, 0.2f));                   \r
                \r
                if(FAILED(CComObject<caspar::flash::FlashAxContainer>::CreateInstance(&ax_)))\r
                        BOOST_THROW_EXCEPTION(caspar_exception() << msg_info(narrow(print()) + " Failed to create FlashAxContainer"));\r
@@ -135,9 +182,10 @@ public:
                if(FAILED(spFlash->put_ScaleMode(2)))  //Exact fit. Scale without respect to the aspect ratio.\r
                        BOOST_THROW_EXCEPTION(caspar_exception() << msg_info(narrow(print()) + " Failed to Set Scale Mode"));\r
                                                \r
-               ax_->SetFormat(format_desc_);           \r
+               ax_->SetSize(width_, height_);          \r
        \r
                CASPAR_LOG(info) << print() << L" Thread started.";\r
+               CASPAR_LOG(info) << print() << L" Successfully initialized with template-host: " << filename << L" width: " << width_ << L" height: " << height_ << L".";\r
        }\r
 \r
        ~flash_renderer()\r
@@ -153,11 +201,8 @@ public:
        void param(const std::wstring& param)\r
        {               \r
                if(!ax_->FlashCall(param))\r
-                       CASPAR_LOG(warning) << print() << " Flash function call failed. Param: " << param << ".";\r
+                       CASPAR_LOG(warning) << print() << L" Flash call failed:" << param;//BOOST_THROW_EXCEPTION(invalid_operation() << msg_info("Flash function call failed.") << arg_name_info("param") << arg_value_info(narrow(param)));\r
                graph_->add_tag("param");\r
-\r
-               if(abs(ax_->GetFPS() - format_desc_.fps) > 0.001 && abs(ax_->GetFPS()/2.0 - format_desc_.fps) > 0.001)\r
-                       CASPAR_LOG(warning) << print() << " Invalid frame-rate: " << ax_->GetFPS() << L". Should be either " << format_desc_.fps << L" or " << format_desc_.fps*2.0 << L".";\r
        }\r
        \r
        safe_ptr<core::basic_frame> render_frame(bool has_underflow)\r
@@ -173,18 +218,19 @@ public:
                if(!has_underflow)                      \r
                        timer_.tick(frame_time); // This will block the thread.\r
                else\r
-                       graph_->add_tag("underflow");\r
+                       graph_->add_tag("skip-sync");\r
                        \r
                frame_timer_.restart();\r
 \r
                ax_->Tick();\r
                if(ax_->InvalidRect())\r
                {                       \r
-                       fast_memclr(bmp_.data(),  format_desc_.size);\r
+                       fast_memclr(bmp_.data(), width_*height_*4);\r
                        ax_->DrawControl(bmp_);\r
                \r
-                       auto frame = frame_factory_->create_frame(this);\r
-                       fast_memcpy(frame->image_data().begin(), bmp_.data(), format_desc_.size);\r
+                       auto frame = frame_factory_->create_frame(this, width_, height_);\r
+                       fast_memcpy(frame->image_data().begin(), bmp_.data(), width_*height_*4);\r
+                       frame->commit();\r
                        head_ = frame;\r
                }               \r
                                \r
@@ -213,23 +259,33 @@ struct flash_producer : public core::frame_producer
        std::shared_ptr<diagnostics::graph> graph_;\r
 \r
        tbb::concurrent_bounded_queue<safe_ptr<core::basic_frame>> frame_buffer_;\r
+\r
+       mutable tbb::spin_mutex         last_frame_mutex_;\r
+       safe_ptr<core::basic_frame>     last_frame_;\r
                                \r
-       com_context<flash_renderer> context_;           \r
+       com_context<flash_renderer> context_;   \r
+\r
+       int width_;\r
+       int height_;\r
 public:\r
-       flash_producer(const safe_ptr<core::frame_factory>& frame_factory, const std::wstring& filename) \r
+       flash_producer(const safe_ptr<core::frame_factory>& frame_factory, const std::wstring& filename, size_t width, size_t height\r
                : filename_(filename)           \r
                , frame_factory_(frame_factory)\r
                , context_(L"flash_producer")\r
+               , last_frame_(core::basic_frame::empty())\r
+               , width_(width > 0 ? width : frame_factory->get_video_format_desc().width)\r
+               , height_(height > 0 ? height : frame_factory->get_video_format_desc().height)\r
        {       \r
                if(!boost::filesystem::exists(filename))\r
                        BOOST_THROW_EXCEPTION(file_not_found() << boost::errinfo_file_name(narrow(filename)));  \r
-                \r
+\r
                fps_ = 0;\r
 \r
                graph_ = diagnostics::create_graph([this]{return print();});\r
-               graph_->set_color("output-buffer", diagnostics::color(0.0f, 1.0f, 0.0f));\r
+               graph_->set_color("output-buffer-count", diagnostics::color(1.0f, 1.0f, 0.0f));          \r
+               graph_->set_color("underflow", diagnostics::color(0.6f, 0.3f, 0.9f));   \r
                \r
-               frame_buffer_.set_capacity(2);\r
+               frame_buffer_.set_capacity(frame_factory_->get_video_format_desc().fps > 30.0 ? 2 : 1);\r
 \r
                initialize();                           \r
        }\r
@@ -241,14 +297,22 @@ public:
 \r
        // frame_producer\r
                \r
-       virtual safe_ptr<core::basic_frame> receive()\r
+       virtual safe_ptr<core::basic_frame> receive(int)\r
        {                               \r
-               graph_->set_value("output-buffer", static_cast<float>(frame_buffer_.size())/static_cast<float>(frame_buffer_.capacity()));\r
+               graph_->set_value("output-buffer-count", static_cast<float>(frame_buffer_.size())/static_cast<float>(frame_buffer_.capacity()));\r
 \r
                auto frame = core::basic_frame::late();\r
-               frame_buffer_.try_pop(frame);           \r
+               if(!frame_buffer_.try_pop(frame))\r
+                       graph_->add_tag("underflow");\r
+\r
                return frame;\r
        }\r
+\r
+       virtual safe_ptr<core::basic_frame> last_frame() const\r
+       {\r
+               tbb::spin_mutex::scoped_lock lock(last_frame_mutex_);\r
+               return last_frame_;\r
+       }               \r
        \r
        virtual void param(const std::wstring& param) \r
        {       \r
@@ -260,6 +324,10 @@ public:
                        try\r
                        {\r
                                context_->param(param); \r
+\r
+                               //const auto& format_desc = frame_factory_->get_video_format_desc();\r
+                               //if(abs(context_->fps() - format_desc.fps) > 0.01 && abs(context_->fps()/2.0 - format_desc.fps) > 0.01)\r
+                               //      CASPAR_LOG(warning) << print() << " Invalid frame-rate: " << context_->fps() << L". Should be either " << format_desc.fps << L" or " << format_desc.fps*2.0 << L".";\r
                        }\r
                        catch(...)\r
                        {\r
@@ -272,19 +340,26 @@ public:
                \r
        virtual std::wstring print() const\r
        { \r
-               return L"flash[" + boost::filesystem::wpath(filename_).filename() + L", " + \r
-                                       boost::lexical_cast<std::wstring>(fps_) + L"]";         \r
+               return L"flash[" + boost::filesystem::wpath(filename_).filename() + L"|" + boost::lexical_cast<std::wstring>(fps_) + L"]";              \r
        }       \r
 \r
        // flash_producer\r
 \r
        void initialize()\r
        {\r
-               context_.reset([&]{return new flash_renderer(safe_ptr<diagnostics::graph>(graph_), frame_factory_, filename_);});\r
+               context_.reset([&]{return new flash_renderer(safe_ptr<diagnostics::graph>(graph_), frame_factory_, filename_, width_, height_);});\r
                while(frame_buffer_.try_push(core::basic_frame::empty())){}             \r
                render(context_.get());\r
        }\r
 \r
+       safe_ptr<core::basic_frame> render_frame()\r
+       {\r
+               auto frame = context_->render_frame(frame_buffer_.size() < frame_buffer_.capacity());           \r
+               tbb::spin_mutex::scoped_lock lock(last_frame_mutex_);\r
+               last_frame_ = make_safe<core::basic_frame>(frame);\r
+               return frame;\r
+       }\r
+\r
        void render(const flash_renderer* renderer)\r
        {               \r
                context_.begin_invoke([=]\r
@@ -295,28 +370,26 @@ public:
                        try\r
                        {               \r
                                const auto& format_desc = frame_factory_->get_video_format_desc();\r
-                               auto frame = core::basic_frame::empty();\r
 \r
-                               if(abs(context_->fps()/2.0 - format_desc.fps) < 0.01) // flash == 2 * format -> interlace\r
+                               if(abs(context_->fps()/2.0 - format_desc.fps) < 2.0) // flash == 2 * format -> interlace\r
                                {\r
-                                       auto frame1 = context_->render_frame(frame_buffer_.size() < frame_buffer_.capacity());\r
-                                       auto frame2 = context_->render_frame(frame_buffer_.size() < frame_buffer_.capacity());\r
-                                       frame_buffer_.push(core::basic_frame::interlace(frame1, frame2, format_desc.mode));\r
-                                       frame = frame2;\r
+                                       auto frame1 = render_frame();\r
+                                       auto frame2 = render_frame();\r
+                                       frame_buffer_.push(core::basic_frame::interlace(frame1, frame2, format_desc.field_mode));\r
                                }\r
-                               else if(abs(context_->fps()- format_desc.fps/2.0) < 0.01) // format == 2 * flash -> duplicate\r
+                               else if(abs(context_->fps()- format_desc.fps/2.0) < 2.0) // format == 2 * flash -> duplicate\r
                                {\r
-                                       frame = context_->render_frame(frame_buffer_.size() < frame_buffer_.capacity());\r
+                                       auto frame = render_frame();\r
                                        frame_buffer_.push(frame);\r
                                        frame_buffer_.push(frame);\r
                                }\r
                                else //if(abs(renderer_->fps() - format_desc_.fps) < 0.1) // format == flash -> simple\r
                                {\r
-                                       frame = context_->render_frame(frame_buffer_.size() < frame_buffer_.capacity());\r
+                                       auto frame = render_frame();\r
                                        frame_buffer_.push(frame);\r
                                }\r
 \r
-                               graph_->set_value("output-buffer", static_cast<float>(frame_buffer_.size())/static_cast<float>(frame_buffer_.capacity()));      \r
+                               graph_->set_value("output-buffer-count", static_cast<float>(frame_buffer_.size())/static_cast<float>(frame_buffer_.capacity()));        \r
                                fps_.fetch_and_store(static_cast<int>(context_->fps()*100.0));\r
 \r
                                render(renderer);\r
@@ -333,9 +406,9 @@ public:
 \r
 safe_ptr<core::frame_producer> create_flash_producer(const safe_ptr<core::frame_factory>& frame_factory, const std::vector<std::wstring>& params)\r
 {\r
-       std::wstring filename = env::template_folder() + L"\\" + params[0];\r
+       auto template_host = get_template_host(frame_factory->get_video_format_desc());\r
        \r
-       return make_safe<flash_producer>(frame_factory, filename);\r
+       return make_safe<flash_producer>(frame_factory, env::template_folder() + L"\\" + widen(template_host.filename), template_host.width, template_host.height);\r
 }\r
 \r
 std::wstring find_flash_template(const std::wstring& template_name)\r