git.sesse.net Git - casparcg/blob - core/mixer/mixer.cpp

   1 /*
   2 * Copyright (c) 2011 Sveriges Television AB <info@casparcg.com>
   3 *
   4 * This file is part of CasparCG (www.casparcg.com).
   5 *
   6 * CasparCG is free software: you can redistribute it and/or modify
   7 * it under the terms of the GNU General Public License as published by
   8 * the Free Software Foundation, either version 3 of the License, or
   9 * (at your option) any later version.
  10 *
  11 * CasparCG is distributed in the hope that it will be useful,
  12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 * GNU General Public License for more details.
  15 *
  16 * You should have received a copy of the GNU General Public License
  17 * along with CasparCG. If not, see <http://www.gnu.org/licenses/>.
  18 *
  19 * Author: Robert Nagy, ronag89@gmail.com
  20 */
  21
  22 #include "../StdAfx.h"
  23
  24 #include "mixer.h"
  25
  26 #include "../frame/frame.h"
  27
  28 #include "audio/audio_mixer.h"
  29 #include "image/image_mixer.h"
  30
  31 #include <common/env.h>
  32 #include <common/executor.h>
  33 #include <common/diagnostics/graph.h>
  34 #include <common/except.h>
  35 #include <common/gl/gl_check.h>
  36
  37 #include <core/frame/draw_frame.h>
  38 #include <core/frame/frame_factory.h>
  39 #include <core/frame/frame_transform.h>
  40 #include <core/frame/pixel_format.h>
  41 #include <core/video_format.h>
  42
  43 #include <boost/foreach.hpp>
  44 #include <boost/timer.hpp>
  45 #include <boost/property_tree/ptree.hpp>
  46 #include <boost/range/algorithm_ext.hpp>
  47
  48 #include <tbb/concurrent_queue.h>
  49 #include <tbb/spin_mutex.h>
  50
  51 #include <unordered_map>
  52 #include <vector>
  53
  54 namespace caspar { namespace core {
  55
  56 struct mixer::impl : boost::noncopyable
  57 {
  58         spl::shared_ptr<diagnostics::graph> graph_;
  59         audio_mixer                                                     audio_mixer_;
  60         spl::shared_ptr<image_mixer>            image_mixer_;
  61
  62         std::unordered_map<int, blend_mode>     blend_modes_;
  63
  64         executor executor_;
  65
  66 public:
  67         impl(spl::shared_ptr<diagnostics::graph> graph, spl::shared_ptr<image_mixer> image_mixer)
  68                 : graph_(std::move(graph))
  69                 , audio_mixer_()
  70                 , image_mixer_(std::move(image_mixer))
  71                 , executor_(L"mixer")
  72         {
  73                 graph_->set_color("mix-time", diagnostics::color(1.0f, 0.0f, 0.9f, 0.8));
  74         }
  75
  76         const_frame operator()(std::map<int, draw_frame> frames, const video_format_desc& format_desc)
  77         {
  78                 boost::timer frame_timer;
  79
  80                 auto frame = executor_.invoke([=]() mutable -> const_frame
  81                 {
  82                         try
  83                         {
  84                                 BOOST_FOREACH(auto& frame, frames)
  85                                 {
  86                                         auto blend_it = blend_modes_.find(frame.first);
  87                                         image_mixer_->begin_layer(blend_it != blend_modes_.end() ? blend_it->second : blend_mode::normal);
  88
  89                                         frame.second.accept(audio_mixer_);
  90                                         frame.second.accept(*image_mixer_);
  91
  92                                         image_mixer_->end_layer();
  93                                 }
  94
  95                                 auto image = (*image_mixer_)(format_desc);
  96                                 auto audio = audio_mixer_(format_desc);
  97
  98                                 auto desc = core::pixel_format_desc(core::pixel_format::bgra);
  99                                 desc.planes.push_back(core::pixel_format_desc::plane(format_desc.width, format_desc.height, 4));
 100                                 return const_frame(std::move(image), std::move(audio), this, desc);
 101                         }
 102                         catch(...)
 103                         {
 104                                 CASPAR_LOG_CURRENT_EXCEPTION();
 105                                 return const_frame::empty();
 106                         }
 107                 });
 108
 109                 graph_->set_value("mix-time", frame_timer.elapsed()*format_desc.fps*0.5);
 110
 111                 return frame;
 112         }
 113
 114         void set_blend_mode(int index, blend_mode value)
 115         {
 116                 executor_.begin_invoke([=]
 117                 {
 118                         auto it = blend_modes_.find(index);
 119                         if(it == blend_modes_.end())
 120                                 blend_modes_.insert(std::make_pair(index, value));
 121                         else
 122                                 it->second = value;
 123                 }, task_priority::high_priority);
 124         }
 125
 126         boost::unique_future<boost::property_tree::wptree> info() const
 127         {
 128                 boost::promise<boost::property_tree::wptree> info;
 129                 info.set_value(boost::property_tree::wptree());
 130                 return info.get_future();
 131         }
 132 };
 133
 134 mixer::mixer(spl::shared_ptr<diagnostics::graph> graph, spl::shared_ptr<image_mixer> image_mixer)
 135         : impl_(new impl(std::move(graph), std::move(image_mixer))){}
 136 void mixer::set_blend_mode(int index, blend_mode value){impl_->set_blend_mode(index, value);}
 137 boost::unique_future<boost::property_tree::wptree> mixer::info() const{return impl_->info();}
 138 const_frame mixer::operator()(std::map<int, draw_frame> frames, const struct video_format_desc& format_desc){return (*impl_)(std::move(frames), format_desc);}
 139 mutable_frame mixer::create_frame(const void* tag, const core::pixel_format_desc& desc) {return impl_->image_mixer_->create_frame(tag, desc);}
 140 }}