]> git.sesse.net Git - casparcg/blobdiff - core/mixer/mixer.cpp
set svn:eol-style native on .h and .cpp files
[casparcg] / core / mixer / mixer.cpp
index 804ba41e5e1bfc9a2bc62e0ed403fbff3f17c3c2..c518255da3b7d2fc739404d4837d72cfb03411df 100644 (file)
-/*\r
-* copyright (c) 2010 Sveriges Television AB <info@casparcg.com>\r
-*\r
-*  This file is part of CasparCG.\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
-*/\r
-#include "../StdAfx.h"\r
-\r
-#include "mixer.h"\r
-\r
-#include "read_frame.h"\r
-#include "write_frame.h"\r
-\r
-#include "audio/audio_mixer.h"\r
-#include "image/image_mixer.h"\r
-\r
-#include "../video_channel_context.h"\r
-\r
-#include <common/exception/exceptions.h>\r
-#include <common/concurrency/executor.h>\r
-#include <common/utility/tweener.h>\r
-#include <common/env.h>\r
-#include <common/gl/gl_check.h>\r
-\r
-#include <core/mixer/read_frame.h>\r
-#include <core/mixer/write_frame.h>\r
-#include <core/producer/frame/basic_frame.h>\r
-#include <core/producer/frame/frame_factory.h>\r
-#include <core/producer/frame/pixel_format.h>\r
-#include <core/producer/frame/frame_transform.h>\r
-\r
-#include <core/video_format.h>\r
-\r
-#include <boost/fusion/container/map.hpp>\r
-#include <boost/fusion/include/at_key.hpp>\r
-#include <boost/foreach.hpp>\r
-\r
-#include <tbb/parallel_invoke.h>\r
-\r
-#include <unordered_map>\r
-\r
-namespace caspar { namespace core {\r
-               \r
-template<typename T>\r
-class tweened_transform\r
-{\r
-       T source_;\r
-       T dest_;\r
-       int duration_;\r
-       int time_;\r
-       tweener_t tweener_;\r
-public:        \r
-       tweened_transform()\r
-               : duration_(0)\r
-               , time_(0)\r
-               , tweener_(get_tweener(L"linear")){}\r
-       tweened_transform(const T& source, const T& dest, int duration, const std::wstring& tween = L"linear")\r
-               : source_(source)\r
-               , dest_(dest)\r
-               , duration_(duration)\r
-               , time_(0)\r
-               , tweener_(get_tweener(tween)){}\r
-       \r
-       T fetch()\r
-       {\r
-               return time_ == duration_ ? dest_ : tween(static_cast<double>(time_), source_, dest_, static_cast<double>(duration_), tweener_);\r
-       }\r
-\r
-       T fetch_and_tick(int num)\r
-       {                                               \r
-               time_ = std::min(time_+num, duration_);\r
-               return fetch();\r
-       }\r
-};\r
-\r
-struct mixer::implementation : boost::noncopyable\r
-{              \r
-       video_channel_context& channel_;\r
-       \r
-       audio_mixer     audio_mixer_;\r
-       image_mixer image_mixer_;\r
-       \r
-       std::unordered_map<int, tweened_transform<core::frame_transform>> transforms_;  \r
-       std::unordered_map<int, blend_mode::type> blend_modes_;\r
-\r
-       std::queue<std::pair<boost::unique_future<safe_ptr<host_buffer>>, std::vector<int16_t>>> buffer_;\r
-       \r
-       const size_t buffer_size_;\r
-\r
-public:\r
-       implementation(video_channel_context& video_channel) \r
-               : channel_(video_channel)\r
-               , audio_mixer_(channel_.get_format_desc())\r
-               , image_mixer_(channel_)\r
-               , buffer_size_(env::properties().get("configuration.producers.buffer-depth", 1))\r
-       {       \r
-               CASPAR_LOG(info) << print() << L" Successfully initialized . Buffer-depth: " << buffer_size_;   \r
-       }\r
-                       \r
-       safe_ptr<read_frame> execute(const std::map<int, safe_ptr<core::basic_frame>>& frames)\r
-       {                       \r
-               try\r
-               {\r
-                       BOOST_FOREACH(auto& frame, frames)\r
-                       {\r
-                               auto blend_it = blend_modes_.find(frame.first);\r
-                               image_mixer_.begin_layer(blend_it != blend_modes_.end() ? blend_it->second : blend_mode::normal);\r
-                               \r
-                               auto frame1 = make_safe<core::basic_frame>(frame.second);\r
-                               frame1->get_frame_transform() = transforms_[frame.first].fetch_and_tick(1);\r
-\r
-                               if(channel_.get_format_desc().field_mode != core::field_mode::progressive)\r
-                               {                               \r
-                                       auto frame2 = make_safe<core::basic_frame>(frame.second);\r
-                                       frame2->get_frame_transform() = transforms_[frame.first].fetch_and_tick(1);\r
-                                               \r
-                                       if(frame1->get_frame_transform() != frame2->get_frame_transform())\r
-                                               frame1 = core::basic_frame::interlace(frame1, frame2, channel_.get_format_desc().field_mode);\r
-                               }\r
-                                                                       \r
-                               frame1->accept(audio_mixer_);                                   \r
-                               frame1->accept(image_mixer_);\r
-\r
-                               image_mixer_.end_layer();\r
-                       }\r
-\r
-                       auto image = image_mixer_.render();\r
-                       auto audio = audio_mixer_.mix();\r
-                       \r
-                       buffer_.push(std::make_pair(std::move(image), audio));\r
-\r
-                       if(buffer_.size()-1 < buffer_size_)                     \r
-                               return make_safe<read_frame>();\r
-               \r
-                       auto res = std::move(buffer_.front());\r
-                       buffer_.pop();\r
-\r
-                       return make_safe<read_frame>(channel_.ogl(), channel_.get_format_desc().size, std::move(res.first.get()), std::move(res.second));       \r
-               }\r
-               catch(...)\r
-               {\r
-                       CASPAR_LOG(error) << L"[mixer] Error detected.";\r
-                       throw;\r
-               }                               \r
-       }\r
-                                       \r
-       safe_ptr<core::write_frame> create_frame(const void* tag, const core::pixel_format_desc& desc)\r
-       {               \r
-               return image_mixer_.create_frame(tag, desc);\r
-       }\r
-               \r
-       void set_transform(int index, const frame_transform& transform, unsigned int mix_duration, const std::wstring& tween)\r
-       {\r
-               channel_.execution().invoke([&]\r
-               {\r
-                       auto src = transforms_[index].fetch();\r
-                       auto dst = transform;\r
-                       transforms_[index] = tweened_transform<frame_transform>(src, dst, mix_duration, tween);\r
-               });\r
-       }\r
-                               \r
-       void apply_transform(int index, const std::function<frame_transform(frame_transform)>& transform, unsigned int mix_duration, const std::wstring& tween)\r
-       {\r
-               channel_.execution().invoke([&]\r
-               {\r
-                       auto src = transforms_[index].fetch();\r
-                       auto dst = transform(src);\r
-                       transforms_[index] = tweened_transform<frame_transform>(src, dst, mix_duration, tween);\r
-               });\r
-       }\r
-\r
-       void clear_transforms()\r
-       {\r
-               channel_.execution().invoke([&]\r
-               {\r
-                       transforms_.clear();\r
-                       blend_modes_.clear();\r
-               });\r
-       }\r
-               \r
-       void set_blend_mode(int index, blend_mode::type value)\r
-       {\r
-               blend_modes_[index] = value;\r
-       }\r
-\r
-       std::wstring print() const\r
-       {\r
-               return L"mixer";\r
-       }\r
-};\r
-       \r
-mixer::mixer(video_channel_context& video_channel) : impl_(new implementation(video_channel)){}\r
-safe_ptr<core::read_frame> mixer::execute(const std::map<int, safe_ptr<core::basic_frame>>& frames){ return impl_->execute(frames);}\r
-core::video_format_desc mixer::get_video_format_desc() const { return impl_->channel_.get_format_desc(); }\r
-safe_ptr<core::write_frame> mixer::create_frame(const void* tag, const core::pixel_format_desc& desc){ return impl_->create_frame(tag, desc); }                \r
-safe_ptr<core::write_frame> mixer::create_frame(const void* tag, size_t width, size_t height, core::pixel_format::type pix_fmt)\r
-{\r
-       // Create bgra frame\r
-       core::pixel_format_desc desc;\r
-       desc.pix_fmt = pix_fmt;\r
-       desc.planes.push_back( core::pixel_format_desc::plane(width, height, 4));\r
-       return create_frame(tag, desc);\r
-}\r
-void mixer::set_frame_transform(int index, const core::frame_transform& transform, unsigned int mix_duration, const std::wstring& tween){impl_->set_transform(index, transform, mix_duration, tween);}\r
-void mixer::apply_frame_transform(int index, const std::function<core::frame_transform(core::frame_transform)>& transform, unsigned int mix_duration, const std::wstring& tween){impl_->apply_transform(index, transform, mix_duration, tween);}\r
-void mixer::clear_transforms(){impl_->clear_transforms();}\r
-void mixer::set_blend_mode(int index, blend_mode::type value){impl_->set_blend_mode(index, value);}\r
+/*
+* 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 "mixer.h"
+
+#include "../frame/frame.h"
+
+#include "audio/audio_mixer.h"
+#include "image/image_mixer.h"
+
+#include <common/env.h>
+#include <common/executor.h>
+#include <common/diagnostics/graph.h>
+#include <common/except.h>
+#include <common/gl/gl_check.h>
+
+#include <core/frame/draw_frame.h>
+#include <core/frame/frame_factory.h>
+#include <core/frame/frame_transform.h>
+#include <core/frame/pixel_format.h>
+#include <core/video_format.h>
+
+#include <boost/foreach.hpp>
+#include <boost/timer.hpp>
+#include <boost/property_tree/ptree.hpp>
+#include <boost/range/algorithm_ext.hpp>
+
+#include <tbb/concurrent_queue.h>
+#include <tbb/spin_mutex.h>
+
+#include <unordered_map>
+#include <vector>
+
+namespace caspar { namespace core {
+
+struct mixer::impl : boost::noncopyable
+{                              
+       spl::shared_ptr<diagnostics::graph> graph_;
+       audio_mixer                                                     audio_mixer_;
+       spl::shared_ptr<image_mixer>            image_mixer_;
+       
+       std::unordered_map<int, blend_mode>     blend_modes_;
+                       
+       executor executor_;
+
+public:
+       impl(spl::shared_ptr<diagnostics::graph> graph, spl::shared_ptr<image_mixer> image_mixer) 
+               : graph_(std::move(graph))
+               , audio_mixer_()
+               , image_mixer_(std::move(image_mixer))
+               , executor_(L"mixer")
+       {                       
+               graph_->set_color("mix-time", diagnostics::color(1.0f, 0.0f, 0.9f, 0.8));
+       }       
+       
+       const_frame operator()(std::map<int, draw_frame> frames, const video_format_desc& format_desc)
+       {               
+               boost::timer frame_timer;
+
+               auto frame = executor_.invoke([=]() mutable -> const_frame
+               {               
+                       try
+                       {       
+                               BOOST_FOREACH(auto& frame, frames)
+                               {
+                                       auto blend_it = blend_modes_.find(frame.first);
+                                       image_mixer_->begin_layer(blend_it != blend_modes_.end() ? blend_it->second : blend_mode::normal);
+                                                                                                       
+                                       frame.second.accept(audio_mixer_);                                      
+                                       frame.second.accept(*image_mixer_);
+
+                                       image_mixer_->end_layer();
+                               }
+                               
+                               auto image = (*image_mixer_)(format_desc);
+                               auto audio = audio_mixer_(format_desc);
+
+                               auto desc = core::pixel_format_desc(core::pixel_format::bgra);
+                               desc.planes.push_back(core::pixel_format_desc::plane(format_desc.width, format_desc.height, 4));
+                               return const_frame(std::move(image), std::move(audio), this, desc);     
+                       }
+                       catch(...)
+                       {
+                               CASPAR_LOG_CURRENT_EXCEPTION();
+                               return const_frame::empty();
+                       }       
+               });             
+                               
+               graph_->set_value("mix-time", frame_timer.elapsed()*format_desc.fps*0.5);
+
+               return frame;
+       }
+                                       
+       void set_blend_mode(int index, blend_mode value)
+       {
+               executor_.begin_invoke([=]
+               {
+                       auto it = blend_modes_.find(index);
+                       if(it == blend_modes_.end())
+                               blend_modes_.insert(std::make_pair(index, value));
+                       else
+                               it->second = value;
+               }, task_priority::high_priority);
+       }
+       
+       boost::unique_future<boost::property_tree::wptree> info() const
+       {
+               boost::promise<boost::property_tree::wptree> info;
+               info.set_value(boost::property_tree::wptree());
+               return info.get_future();
+       }
+};
+       
+mixer::mixer(spl::shared_ptr<diagnostics::graph> graph, spl::shared_ptr<image_mixer> image_mixer) 
+       : impl_(new impl(std::move(graph), std::move(image_mixer))){}
+void mixer::set_blend_mode(int index, blend_mode value){impl_->set_blend_mode(index, value);}
+boost::unique_future<boost::property_tree::wptree> mixer::info() const{return impl_->info();}
+const_frame mixer::operator()(std::map<int, draw_frame> frames, const struct video_format_desc& format_desc){return (*impl_)(std::move(frames), format_desc);}
+mutable_frame mixer::create_frame(const void* tag, const core::pixel_format_desc& desc) {return impl_->image_mixer_->create_frame(tag, desc);}
 }}
\ No newline at end of file