git.sesse.net Git - casparcg/blob - core/consumer/frame_consumer.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 "frame_consumer.h"
  25
  26 #include <common/except.h>
  27 #include <common/future.h>
  28
  29 #include <core/video_format.h>
  30 #include <core/frame/frame.h>
  31
  32 #include <boost/thread.hpp>
  33
  34 namespace caspar { namespace core {
  35
  36 std::vector<const consumer_factory_t> g_factories;
  37
  38 void register_consumer_factory(const consumer_factory_t& factory)
  39 {
  40         g_factories.push_back(factory);
  41 }
  42
  43 class destroy_consumer_proxy : public frame_consumer
  44 {
  45         std::shared_ptr<frame_consumer> consumer_;
  46 public:
  47         destroy_consumer_proxy(spl::shared_ptr<frame_consumer>&& consumer)
  48                 : consumer_(std::move(consumer))
  49         {
  50         }
  51
  52         ~destroy_consumer_proxy()
  53         {
  54                 static tbb::atomic<int> counter = tbb::atomic<int>();
  55
  56                 ++counter;
  57                 CASPAR_VERIFY(counter < 8);
  58
  59                 auto consumer = new std::shared_ptr<frame_consumer>(std::move(consumer_));
  60                 boost::thread([=]
  61                 {
  62                         std::unique_ptr<std::shared_ptr<frame_consumer>> pointer_guard(consumer);
  63
  64                         auto str = (*consumer)->print();
  65                         try
  66                         {
  67                                 if(!consumer->unique())
  68                                         CASPAR_LOG(trace) << str << L" Not destroyed on asynchronous destruction thread: " << consumer->use_count();
  69                                 else
  70                                         CASPAR_LOG(trace) << str << L" Destroying on asynchronous destruction thread.";
  71                         }
  72                         catch(...){}
  73
  74                         pointer_guard.reset();
  75
  76                         --counter;
  77                 }).detach();
  78         }
  79
  80         boost::unique_future<bool> send(const_frame frame) override                                                                                     {return consumer_->send(std::move(frame));}
  81         virtual void initialize(const struct video_format_desc& format_desc, int channel_index) override        {return consumer_->initialize(format_desc, channel_index);}
  82         std::wstring print() const override                                                                                                                                     {return consumer_->print();}
  83         std::wstring name() const override                                                                                                                                      {return consumer_->name();}
  84         boost::property_tree::wptree info() const override                                                                                                      {return consumer_->info();}
  85         bool has_synchronization_clock() const override                                                                                                         {return consumer_->has_synchronization_clock();}
  86         int buffer_depth() const override                                                                                                                                       {return consumer_->buffer_depth();}
  87         int index() const override                                                                                                                                                      {return consumer_->index();}
  88         void subscribe(const monitor::observable::observer_ptr& o) override                                                                     {consumer_->subscribe(o);}
  89         void unsubscribe(const monitor::observable::observer_ptr& o) override                                                           {consumer_->unsubscribe(o);}
  90 };
  91
  92 class print_consumer_proxy : public frame_consumer
  93 {
  94         std::shared_ptr<frame_consumer> consumer_;
  95 public:
  96         print_consumer_proxy(spl::shared_ptr<frame_consumer>&& consumer)
  97                 : consumer_(std::move(consumer))
  98         {
  99                 CASPAR_LOG(info) << consumer_->print() << L" Initialized.";
 100         }
 101
 102         ~print_consumer_proxy()
 103         {
 104                 auto str = consumer_->print();
 105                 CASPAR_LOG(trace) << str << L" Uninitializing.";
 106                 consumer_.reset();
 107                 CASPAR_LOG(info) << str << L" Uninitialized.";
 108         }
 109
 110         boost::unique_future<bool> send(const_frame frame) override                                                                     {return consumer_->send(std::move(frame));}
 111         virtual void initialize(const struct video_format_desc& format_desc, int channel_index) override        {return consumer_->initialize(format_desc, channel_index);}
 112         std::wstring print() const override                                                                                                                     {return consumer_->print();}
 113         std::wstring name() const override                                                                                                                      {return consumer_->name();}
 114         boost::property_tree::wptree info() const override                                                                                      {return consumer_->info();}
 115         bool has_synchronization_clock() const override                                                                                         {return consumer_->has_synchronization_clock();}
 116         int buffer_depth() const override                                                                                                                       {return consumer_->buffer_depth();}
 117         int index() const override                                                                                                                                      {return consumer_->index();}
 118         void subscribe(const monitor::observable::observer_ptr& o) override                                                                     {consumer_->subscribe(o);}
 119         void unsubscribe(const monitor::observable::observer_ptr& o) override                                                           {consumer_->unsubscribe(o);}
 120 };
 121
 122 class recover_consumer_proxy : public frame_consumer
 123 {
 124         std::shared_ptr<frame_consumer> consumer_;
 125         int                                                             channel_index_;
 126         video_format_desc                               format_desc_;
 127 public:
 128         recover_consumer_proxy(spl::shared_ptr<frame_consumer>&& consumer)
 129                 : consumer_(std::move(consumer))
 130         {
 131         }
 132
 133         virtual boost::unique_future<bool> send(const_frame frame)
 134         {
 135                 try
 136                 {
 137                         return consumer_->send(frame);
 138                 }
 139                 catch(...)
 140                 {
 141                         CASPAR_LOG_CURRENT_EXCEPTION();
 142                         try
 143                         {
 144                                 consumer_->initialize(format_desc_, channel_index_);
 145                                 return consumer_->send(frame);
 146                         }
 147                         catch(...)
 148                         {
 149                                 CASPAR_LOG_CURRENT_EXCEPTION();
 150                                 CASPAR_LOG(error) << print() << " Failed to recover consumer.";
 151                                 return wrap_as_future(false);
 152                         }
 153                 }
 154         }
 155
 156         virtual void initialize(const struct video_format_desc& format_desc, int channel_index)
 157         {
 158                 format_desc_    = format_desc;
 159                 channel_index_  = channel_index;
 160                 return consumer_->initialize(format_desc, channel_index);
 161         }
 162
 163         std::wstring print() const override                                                                             {return consumer_->print();}
 164         std::wstring name() const override                                                                              {return consumer_->name();}
 165         boost::property_tree::wptree info() const override                                              {return consumer_->info();}
 166         bool has_synchronization_clock() const override                                                 {return consumer_->has_synchronization_clock();}
 167         int buffer_depth() const override                                                                               {return consumer_->buffer_depth();}
 168         int index() const override                                                                                              {return consumer_->index();}
 169         void subscribe(const monitor::observable::observer_ptr& o) override             {consumer_->subscribe(o);}
 170         void unsubscribe(const monitor::observable::observer_ptr& o) override   {consumer_->unsubscribe(o);}
 171 };
 172
 173 // This class is used to guarantee that audio cadence is correct. This is important for NTSC audio.
 174 class cadence_guard : public frame_consumer
 175 {
 176         spl::shared_ptr<frame_consumer>         consumer_;
 177         std::vector<int>                                        audio_cadence_;
 178         boost::circular_buffer<std::size_t>     sync_buffer_;
 179 public:
 180         cadence_guard(const spl::shared_ptr<frame_consumer>& consumer)
 181                 : consumer_(consumer)
 182         {
 183         }
 184
 185         void initialize(const video_format_desc& format_desc, int channel_index) override
 186         {
 187                 audio_cadence_  = format_desc.audio_cadence;
 188                 sync_buffer_    = boost::circular_buffer<std::size_t>(format_desc.audio_cadence.size());
 189                 consumer_->initialize(format_desc, channel_index);
 190         }
 191
 192         boost::unique_future<bool> send(const_frame frame) override
 193         {
 194                 if(audio_cadence_.size() == 1)
 195                         return consumer_->send(frame);
 196
 197                 boost::unique_future<bool> result = wrap_as_future(true);
 198
 199                 if(boost::range::equal(sync_buffer_, audio_cadence_) && audio_cadence_.front() == static_cast<int>(frame.audio_data().size()))
 200                 {
 201                         // Audio sent so far is in sync, now we can send the next chunk.
 202                         result = consumer_->send(frame);
 203                         boost::range::rotate(audio_cadence_, std::begin(audio_cadence_)+1);
 204                 }
 205                 else
 206                         CASPAR_LOG(trace) << print() << L" Syncing audio.";
 207
 208                 sync_buffer_.push_back(static_cast<int>(frame.audio_data().size()));
 209
 210                 return std::move(result);
 211         }
 212
 213         std::wstring print() const override                                                                             {return consumer_->print();}
 214         std::wstring name() const override                                                                              {return consumer_->name();}
 215         boost::property_tree::wptree info() const override                                              {return consumer_->info();}
 216         bool has_synchronization_clock() const override                                                 {return consumer_->has_synchronization_clock();}
 217         int buffer_depth() const override                                                                               {return consumer_->buffer_depth();}
 218         int index() const override                                                                                              {return consumer_->index();}
 219         void subscribe(const monitor::observable::observer_ptr& o) override             {consumer_->subscribe(o);}
 220         void unsubscribe(const monitor::observable::observer_ptr& o) override   {consumer_->unsubscribe(o);}
 221 };
 222
 223 spl::shared_ptr<core::frame_consumer> create_consumer(const std::vector<std::wstring>& params)
 224 {
 225         if(params.empty())
 226                 CASPAR_THROW_EXCEPTION(invalid_argument() << arg_name_info("params") << arg_value_info(""));
 227
 228         auto consumer = frame_consumer::empty();
 229         std::any_of(g_factories.begin(), g_factories.end(), [&](const consumer_factory_t& factory) -> bool
 230                 {
 231                         try
 232                         {
 233                                 consumer = factory(params);
 234                         }
 235                         catch(...)
 236                         {
 237                                 CASPAR_LOG_CURRENT_EXCEPTION();
 238                         }
 239                         return consumer != frame_consumer::empty();
 240                 });
 241
 242         if(consumer == frame_consumer::empty())
 243                 CASPAR_THROW_EXCEPTION(file_not_found() << msg_info("No match found for supplied commands. Check syntax."));
 244
 245         return spl::make_shared<destroy_consumer_proxy>(
 246                         spl::make_shared<print_consumer_proxy>(
 247                          spl::make_shared<recover_consumer_proxy>(
 248                           spl::make_shared<cadence_guard>(
 249                            std::move(consumer)))));
 250 }
 251
 252 const spl::shared_ptr<frame_consumer>& frame_consumer::empty()
 253 {
 254         class empty_frame_consumer : public frame_consumer
 255         {
 256         public:
 257                 boost::unique_future<bool> send(const_frame) override {return wrap_as_future(false);}
 258                 void initialize(const video_format_desc&, int) override{}
 259                 std::wstring print() const override {return L"empty";}
 260                 std::wstring name() const override {return L"empty";}
 261                 bool has_synchronization_clock() const override {return false;}
 262                 int buffer_depth() const override {return 0;};
 263                 virtual int index() const{return -1;}
 264                 void subscribe(const monitor::observable::observer_ptr& o) override{}
 265                 void unsubscribe(const monitor::observable::observer_ptr& o) override{}
 266                 boost::property_tree::wptree info() const override
 267                 {
 268                         boost::property_tree::wptree info;
 269                         info.add(L"type", L"empty");
 270                         return info;
 271                 }
 272         };
 273         static spl::shared_ptr<frame_consumer> consumer = spl::make_shared<empty_frame_consumer>();
 274         return consumer;
 275 }
 276
 277 }}