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/future.h>
  36 #include <common/timer.h>
  37
  38 #include <core/frame/draw_frame.h>
  39 #include <core/frame/frame_factory.h>
  40 #include <core/frame/frame_transform.h>
  41 #include <core/frame/pixel_format.h>
  42 #include <core/video_format.h>
  43
  44 #include <boost/property_tree/ptree.hpp>
  45
  46 #include <tbb/concurrent_queue.h>
  47 #include <tbb/spin_mutex.h>
  48
  49 #include <unordered_map>
  50 #include <vector>
  51
  52 namespace caspar { namespace core {
  53
  54 struct mixer::impl : boost::noncopyable
  55 {
  56         spl::shared_ptr<diagnostics::graph> graph_;
  57         audio_mixer                                                     audio_mixer_;
  58         spl::shared_ptr<image_mixer>            image_mixer_;
  59
  60         std::unordered_map<int, blend_mode>     blend_modes_;
  61
  62         executor executor_                                                                      { L"mixer" };
  63
  64 public:
  65         impl(spl::shared_ptr<diagnostics::graph> graph, spl::shared_ptr<image_mixer> image_mixer)
  66                 : graph_(std::move(graph))
  67                 , image_mixer_(std::move(image_mixer))
  68         {
  69                 graph_->set_color("mix-time", diagnostics::color(1.0f, 0.0f, 0.9f, 0.8f));
  70         }
  71
  72         const_frame operator()(std::map<int, draw_frame> frames, const video_format_desc& format_desc)
  73         {
  74                 caspar::timer frame_timer;
  75
  76                 auto frame = executor_.invoke([=]() mutable -> const_frame
  77                 {
  78                         try
  79                         {
  80                                 for (auto& frame : frames)
  81                                 {
  82                                         auto blend_it = blend_modes_.find(frame.first);
  83                                         image_mixer_->begin_layer(blend_it != blend_modes_.end() ? blend_it->second : blend_mode::normal);
  84
  85                                         frame.second.accept(audio_mixer_);
  86                                         frame.second.accept(*image_mixer_);
  87
  88                                         image_mixer_->end_layer();
  89                                 }
  90
  91                                 auto image = (*image_mixer_)(format_desc);
  92                                 auto audio = audio_mixer_(format_desc);
  93
  94                                 auto desc = core::pixel_format_desc(core::pixel_format::bgra);
  95                                 desc.planes.push_back(core::pixel_format_desc::plane(format_desc.width, format_desc.height, 4));
  96                                 return const_frame(std::move(image), std::move(audio), this, desc);
  97                         }
  98                         catch(...)
  99                         {
 100                                 CASPAR_LOG_CURRENT_EXCEPTION();
 101                                 return const_frame::empty();
 102                         }
 103                 });
 104
 105                 graph_->set_value("mix-time", frame_timer.elapsed()*format_desc.fps*0.5);
 106
 107                 return frame;
 108         }
 109
 110         void set_blend_mode(int index, blend_mode value)
 111         {
 112                 executor_.begin_invoke([=]
 113                 {
 114                         auto it = blend_modes_.find(index);
 115                         if(it == blend_modes_.end())
 116                                 blend_modes_.insert(std::make_pair(index, value));
 117                         else
 118                                 it->second = value;
 119                 }, task_priority::high_priority);
 120         }
 121
 122         void clear_blend_mode(int index)
 123         {
 124                 executor_.begin_invoke([=]
 125                 {
 126                         blend_modes_.erase(index);
 127                 }, task_priority::high_priority);
 128         }
 129
 130         void clear_blend_modes()
 131         {
 132                 executor_.begin_invoke([=]
 133                 {
 134                         blend_modes_.clear();
 135                 }, task_priority::high_priority);
 136         }
 137
 138         void set_master_volume(float volume)
 139         {
 140                 executor_.begin_invoke([=]
 141                 {
 142                         audio_mixer_.set_master_volume(volume);
 143                 }, task_priority::high_priority);
 144         }
 145
 146         std::future<boost::property_tree::wptree> info() const
 147         {
 148                 return make_ready_future(boost::property_tree::wptree());
 149         }
 150 };
 151
 152 mixer::mixer(spl::shared_ptr<diagnostics::graph> graph, spl::shared_ptr<image_mixer> image_mixer)
 153         : impl_(new impl(std::move(graph), std::move(image_mixer))){}
 154 void mixer::set_blend_mode(int index, blend_mode value){impl_->set_blend_mode(index, value);}
 155 void mixer::clear_blend_mode(int index) { impl_->clear_blend_mode(index); }
 156 void mixer::clear_blend_modes() { impl_->clear_blend_modes(); }
 157 void mixer::set_master_volume(float volume) { impl_->set_master_volume(volume); }
 158 std::future<boost::property_tree::wptree> mixer::info() const{return impl_->info();}
 159 const_frame mixer::operator()(std::map<int, draw_frame> frames, const struct video_format_desc& format_desc){return (*impl_)(std::move(frames), format_desc);}
 160 mutable_frame mixer::create_frame(const void* tag, const core::pixel_format_desc& desc) {return impl_->image_mixer_->create_frame(tag, desc);}
 161 }}