2 * Copyright (c) 2011 Sveriges Television AB <info@casparcg.com>
4 * This file is part of CasparCG (www.casparcg.com).
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.
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.
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/>.
19 * Author: Robert Nagy, ronag89@gmail.com
22 #include "../StdAfx.h"
25 #pragma warning (disable : 4244)
30 #include "frame_consumer.h"
33 #include "../video_format.h"
34 #include "../frame/frame.h"
36 #include <common/assert.h>
37 #include <common/future.h>
38 #include <common/executor.h>
39 #include <common/diagnostics/graph.h>
40 #include <common/prec_timer.h>
41 #include <common/memshfl.h>
42 #include <common/env.h>
44 #include <boost/circular_buffer.hpp>
45 #include <boost/lexical_cast.hpp>
46 #include <boost/property_tree/ptree.hpp>
47 #include <boost/range/algorithm.hpp>
48 #include <boost/range/adaptors.hpp>
49 #include <boost/timer.hpp>
51 namespace caspar { namespace core {
55 spl::shared_ptr<diagnostics::graph> graph_;
56 spl::shared_ptr<monitor::subject> monitor_subject_;
57 const int channel_index_;
58 video_format_desc format_desc_;
59 std::map<int, port> ports_;
60 prec_timer sync_timer_;
61 boost::circular_buffer<const_frame> frames_;
64 impl(spl::shared_ptr<diagnostics::graph> graph, const video_format_desc& format_desc, int channel_index)
65 : graph_(std::move(graph))
66 , monitor_subject_(spl::make_shared<monitor::subject>("/output"))
67 , channel_index_(channel_index)
68 , format_desc_(format_desc)
69 , executor_(L"output")
71 graph_->set_color("consume-time", diagnostics::color(1.0f, 0.4f, 0.0f, 0.8));
74 void add(int index, spl::shared_ptr<frame_consumer> consumer)
78 consumer->initialize(format_desc_, channel_index_);
80 executor_.begin_invoke([this, index, consumer]
82 port p(index, channel_index_, std::move(consumer));
83 p.monitor_output().attach_parent(monitor_subject_);
84 ports_.insert(std::make_pair(index, std::move(p)));
85 }, task_priority::high_priority);
88 void add(const spl::shared_ptr<frame_consumer>& consumer)
90 add(consumer->index(), consumer);
93 void remove(int index)
95 executor_.begin_invoke([=]
97 auto it = ports_.find(index);
98 if(it != ports_.end())
100 }, task_priority::high_priority);
103 void remove(const spl::shared_ptr<frame_consumer>& consumer)
105 remove(consumer->index());
108 void video_format_desc(const core::video_format_desc& format_desc)
112 if(format_desc_ == format_desc)
115 auto it = ports_.begin();
116 while(it != ports_.end())
120 it->second.video_format_desc(format_desc);
125 CASPAR_LOG_CURRENT_EXCEPTION();
130 format_desc_ = format_desc;
135 std::pair<int, int> minmax_buffer_depth() const
138 return std::make_pair(0, 0);
140 auto buffer_depths = ports_ |
141 boost::adaptors::map_values |
142 boost::adaptors::transformed([](const port& p){return p.buffer_depth();});
145 return std::make_pair(*boost::range::min_element(buffer_depths), *boost::range::max_element(buffer_depths));
148 bool has_synchronization_clock() const
150 return boost::range::count_if(ports_ | boost::adaptors::map_values,
151 [](const port& p){return p.has_synchronization_clock();}) > 0;
154 void operator()(const_frame input_frame, const core::video_format_desc& format_desc)
156 boost::timer frame_timer;
158 video_format_desc(format_desc);
163 if(!has_synchronization_clock())
164 sync_timer_.tick(1.0/format_desc_.fps);
166 if(input_frame.size() != format_desc_.size)
168 CASPAR_LOG(debug) << print() << L" Invalid input frame dimension.";
172 auto minmax = minmax_buffer_depth();
174 frames_.set_capacity(std::max(2, minmax.second - minmax.first) + 1); // std::max(2, x) since we want to guarantee some pipeline depth for asycnhronous mixer read-back.
175 frames_.push_back(input_frame);
180 std::map<int, std::future<bool>> send_results;
183 for (auto it = ports_.begin(); it != ports_.end();)
185 auto& port = it->second;
186 auto& frame = frames_.at(port.buffer_depth()-minmax.first);
190 send_results.insert(std::make_pair(it->first, port.send(frame)));
195 CASPAR_LOG_CURRENT_EXCEPTION();
198 send_results.insert(std::make_pair(it->first, port.send(frame)));
203 CASPAR_LOG_CURRENT_EXCEPTION();
204 CASPAR_LOG(error) << "Failed to recover consumer: " << port.print() << L". Removing it.";
205 it = ports_.erase(it);
211 for (auto it = send_results.begin(); it != send_results.end(); ++it)
215 if (!it->second.get())
216 ports_.erase(it->first);
220 CASPAR_LOG_CURRENT_EXCEPTION();
221 ports_.erase(it->first);
226 graph_->set_value("consume-time", frame_timer.elapsed()*format_desc.fps*0.5);
229 std::wstring print() const
231 return L"output[" + boost::lexical_cast<std::wstring>(channel_index_) + L"]";
234 std::future<boost::property_tree::wptree> info()
236 return std::move(executor_.begin_invoke([&]() -> boost::property_tree::wptree
238 boost::property_tree::wptree info;
239 for (auto& port : ports_)
241 info.add_child(L"consumers.consumer", port.second.info())
242 .add(L"index", port.first);
245 }, task_priority::high_priority));
249 output::output(spl::shared_ptr<diagnostics::graph> graph, const video_format_desc& format_desc, int channel_index) : impl_(new impl(std::move(graph), format_desc, channel_index)){}
250 void output::add(int index, const spl::shared_ptr<frame_consumer>& consumer){impl_->add(index, consumer);}
251 void output::add(const spl::shared_ptr<frame_consumer>& consumer){impl_->add(consumer);}
252 void output::remove(int index){impl_->remove(index);}
253 void output::remove(const spl::shared_ptr<frame_consumer>& consumer){impl_->remove(consumer);}
254 std::future<boost::property_tree::wptree> output::info() const{return impl_->info();}
255 void output::operator()(const_frame frame, const video_format_desc& format_desc){(*impl_)(std::move(frame), format_desc);}
256 monitor::subject& output::monitor_output() {return *impl_->monitor_subject_;}