git.sesse.net Git - casparcg/blob - shell/server.cpp

   1 /*\r
   2 * Copyright (c) 2011 Sveriges Television AB <info@casparcg.com>\r
   3 *\r
   4 * This file is part of CasparCG (www.casparcg.com).\r
   5 *\r
   6 * CasparCG is free software: you can redistribute it and/or modify\r
   7 * it under the terms of the GNU General Public License as published by\r
   8 * the Free Software Foundation, either version 3 of the License, or\r
   9 * (at your option) any later version.\r
  10 *\r
  11 * CasparCG is distributed in the hope that it will be useful,\r
  12 * but WITHOUT ANY WARRANTY; without even the implied warranty of\r
  13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
  14 * GNU General Public License for more details.\r
  15 *\r
  16 * You should have received a copy of the GNU General Public License\r
  17 * along with CasparCG. If not, see <http://www.gnu.org/licenses/>.\r
  18 *\r
  19 * Author: Robert Nagy, ronag89@gmail.com\r
  20 */\r
  21 \r
  22 \r
  23 #include "server.h"\r
  24 \r
  25 #include <memory>\r
  26 \r
  27 #include <common/env.h>\r
  28 #include <common/exception/exceptions.h>\r
  29 #include <common/utility/string.h>\r
  30 #include <common/filesystem/polling_filesystem_monitor.h>\r
  31 \r
  32 #include <core/mixer/gpu/ogl_device.h>\r
  33 #include <core/mixer/audio/audio_util.h>\r
  34 #include <core/video_channel.h>\r
  35 #include <core/producer/stage.h>\r
  36 #include <core/consumer/output.h>\r
  37 #include <core/consumer/synchronizing/synchronizing_consumer.h>\r
  38 #include <core/thumbnail_generator.h>\r
  39 \r
  40 #include <modules/bluefish/bluefish.h>\r
  41 #include <modules/decklink/decklink.h>\r
  42 #include <modules/ffmpeg/ffmpeg.h>\r
  43 #include <modules/flash/flash.h>\r
  44 #include <modules/oal/oal.h>\r
  45 #include <modules/ogl/ogl.h>\r
  46 #include <modules/silverlight/silverlight.h>\r
  47 #include <modules/image/image.h>\r
  48 #include <modules/image/consumer/image_consumer.h>\r
  49 \r
  50 #include <modules/oal/consumer/oal_consumer.h>\r
  51 #include <modules/bluefish/consumer/bluefish_consumer.h>\r
  52 #include <modules/decklink/consumer/decklink_consumer.h>\r
  53 #include <modules/ogl/consumer/ogl_consumer.h>\r
  54 #include <modules/ffmpeg/consumer/ffmpeg_consumer.h>\r
  55 \r
  56 #include <protocol/amcp/AMCPProtocolStrategy.h>\r
  57 #include <protocol/cii/CIIProtocolStrategy.h>\r
  58 #include <protocol/CLK/CLKProtocolStrategy.h>\r
  59 #include <protocol/util/AsyncEventServer.h>\r
  60 #include <protocol/util/stateful_protocol_strategy_wrapper.h>\r
  61 #include <protocol/osc/server.h>\r
  62 \r
  63 #include <boost/algorithm/string.hpp>\r
  64 #include <boost/lexical_cast.hpp>\r
  65 #include <boost/foreach.hpp>\r
  66 #include <boost/property_tree/ptree.hpp>\r
  67 #include <boost/property_tree/xml_parser.hpp>\r
  68 \r
  69 namespace caspar {\r
  70 \r
  71 using namespace core;\r
  72 using namespace protocol;\r
  73 \r
  74 struct server::implementation : boost::noncopyable\r
  75 {\r
  76         core::monitor::subject                                          monitor_subject_;\r
  77         boost::promise<bool>&                                           shutdown_server_now_;\r
  78         safe_ptr<ogl_device>                                            ogl_;\r
  79         std::vector<safe_ptr<IO::AsyncEventServer>> async_servers_;     \r
  80         std::vector<osc::server>                                        osc_servers_;\r
  81         std::vector<safe_ptr<video_channel>>            channels_;\r
  82         std::shared_ptr<thumbnail_generator>            thumbnail_generator_;\r
  83 \r
  84         implementation(boost::promise<bool>& shutdown_server_now)\r
  85                 : shutdown_server_now_(shutdown_server_now)\r
  86                 , ogl_(ogl_device::create())\r
  87         {                       \r
  88                 setup_audio(env::properties());\r
  89 \r
  90                 ffmpeg::init();\r
  91                 CASPAR_LOG(info) << L"Initialized ffmpeg module.";\r
  92                                                           \r
  93                 bluefish::init();         \r
  94                 CASPAR_LOG(info) << L"Initialized bluefish module.";\r
  95                                                           \r
  96                 decklink::init();         \r
  97                 CASPAR_LOG(info) << L"Initialized decklink module.";\r
  98                                                                                                                   \r
  99                 oal::init();              \r
 100                 CASPAR_LOG(info) << L"Initialized oal module.";\r
 101                                                           \r
 102                 ogl::init();              \r
 103                 CASPAR_LOG(info) << L"Initialized ogl module.";\r
 104 \r
 105                 image::init();            \r
 106                 CASPAR_LOG(info) << L"Initialized image module.";\r
 107 \r
 108                 flash::init();            \r
 109                 CASPAR_LOG(info) << L"Initialized flash module.";\r
 110 \r
 111                 setup_channels(env::properties());\r
 112                 CASPAR_LOG(info) << L"Initialized channels.";\r
 113 \r
 114                 setup_thumbnail_generation(env::properties());\r
 115 \r
 116                 setup_controllers(env::properties());\r
 117                 CASPAR_LOG(info) << L"Initialized controllers.";\r
 118         }\r
 119 \r
 120         ~implementation()\r
 121         {               \r
 122                 ffmpeg::uninit();\r
 123 \r
 124                 async_servers_.clear();\r
 125                 channels_.clear();\r
 126         }\r
 127 \r
 128         void setup_audio(const boost::property_tree::wptree& pt)\r
 129         {\r
 130                 register_default_channel_layouts(default_channel_layout_repository());\r
 131                 register_default_mix_configs(default_mix_config_repository());\r
 132                 parse_channel_layouts(\r
 133                                 default_channel_layout_repository(),\r
 134                                 pt.get_child(L"configuration.audio.channel-layouts"));\r
 135                 parse_mix_configs(\r
 136                                 default_mix_config_repository(),\r
 137                                 pt.get_child(L"configuration.audio.mix-configs"));\r
 138         }\r
 139                                 \r
 140         void setup_channels(const boost::property_tree::wptree& pt)\r
 141         {   \r
 142                 using boost::property_tree::wptree;\r
 143                 BOOST_FOREACH(auto& xml_channel, pt.get_child(L"configuration.channels"))\r
 144                 {               \r
 145                         auto format_desc = video_format_desc::get(widen(xml_channel.second.get(L"video-mode", L"PAL")));                \r
 146                         if(format_desc.format == video_format::invalid)\r
 147                                 BOOST_THROW_EXCEPTION(caspar_exception() << msg_info("Invalid video-mode."));\r
 148                         auto audio_channel_layout = default_channel_layout_repository().get_by_name(\r
 149                                         boost::to_upper_copy(xml_channel.second.get(L"channel-layout", L"STEREO")));\r
 150                         \r
 151                         channels_.push_back(make_safe<video_channel>(channels_.size()+1, format_desc, ogl_, audio_channel_layout));\r
 152                         \r
 153                         channels_.back()->monitor_output().link_target(&monitor_subject_);\r
 154 \r
 155                         create_consumers(\r
 156                                 xml_channel.second.get_child(L"consumers"),\r
 157                                 [&] (const safe_ptr<core::frame_consumer>& consumer)\r
 158                                 {\r
 159                                         channels_.back()->output()->add(consumer);\r
 160                                 });\r
 161 \r
 162                         // Add all consumers before starting channel.\r
 163                         channels_.back()->start_channel();\r
 164                 }\r
 165 \r
 166                 // Dummy diagnostics channel\r
 167                 if(env::properties().get(L"configuration.channel-grid", false))\r
 168                 {\r
 169                         channels_.push_back(make_safe<video_channel>(channels_.size()+1, core::video_format_desc::get(core::video_format::x576p2500), ogl_, default_channel_layout_repository().get_by_name(L"STEREO")));\r
 170                         channels_.back()->start_channel();\r
 171                 }\r
 172         }\r
 173 \r
 174         template<typename Base>\r
 175         std::vector<safe_ptr<Base>> create_consumers(const boost::property_tree::wptree& pt)\r
 176         {\r
 177                 std::vector<safe_ptr<Base>> consumers;\r
 178 \r
 179                 create_consumers(pt, [&] (const safe_ptr<core::frame_consumer>& consumer)\r
 180                 {\r
 181                         consumers.push_back(dynamic_pointer_cast<Base>(consumer));\r
 182                 });\r
 183 \r
 184                 return consumers;\r
 185         }\r
 186 \r
 187         template<class Func>\r
 188         void create_consumers(const boost::property_tree::wptree& pt, const Func& on_consumer)\r
 189         {\r
 190                 BOOST_FOREACH(auto& xml_consumer, pt)\r
 191                 {\r
 192                         try\r
 193                         {\r
 194                                 auto name = xml_consumer.first;\r
 195                                 if (name == L"screen")\r
 196                                         on_consumer(ogl::create_consumer(xml_consumer.second));\r
 197                                 else if (name == L"bluefish")                                   \r
 198                                         on_consumer(bluefish::create_consumer(xml_consumer.second));                                    \r
 199                                 else if (name == L"decklink")                                   \r
 200                                         on_consumer(decklink::create_consumer(xml_consumer.second));                            \r
 201                                 else if (name == L"file")                                       \r
 202                                         on_consumer(ffmpeg::create_consumer(xml_consumer.second));                                              \r
 203                                 else if (name == L"system-audio")\r
 204                                         on_consumer(oal::create_consumer());\r
 205                                 else if (name == L"synchronizing")\r
 206                                         on_consumer(make_safe<core::synchronizing_consumer>(create_consumers<core::synchronizable_consumer>(xml_consumer.second)));\r
 207                                 else if (name != L"<xmlcomment>")\r
 208                                         CASPAR_LOG(warning) << "Invalid consumer: " << widen(name);     \r
 209                         }\r
 210                         catch(...)\r
 211                         {\r
 212                                 CASPAR_LOG_CURRENT_EXCEPTION();\r
 213                         }\r
 214                 }\r
 215         }\r
 216                 \r
 217         void setup_controllers(const boost::property_tree::wptree& pt)\r
 218         {               \r
 219                 using boost::property_tree::wptree;\r
 220                 BOOST_FOREACH(auto& xml_controller, pt.get_child(L"configuration.controllers"))\r
 221                 {\r
 222                         try\r
 223                         {\r
 224                                 auto name = xml_controller.first;\r
 225                                 auto protocol = xml_controller.second.get<std::wstring>(L"protocol");   \r
 226 \r
 227                                 if(name == L"tcp")\r
 228                                 {                                       \r
 229                                         unsigned int port = xml_controller.second.get(L"port", 5250);\r
 230                                         auto asyncbootstrapper = make_safe<IO::AsyncEventServer>(create_protocol(protocol), port);\r
 231                                         asyncbootstrapper->Start();\r
 232                                         async_servers_.push_back(asyncbootstrapper);\r
 233                                 }\r
 234                                 else if(name == L"udp")\r
 235                                 {                                       \r
 236                                         const auto address = xml_controller.second.get(L"address", L"127.0.0.1");\r
 237                                         const auto port = xml_controller.second.get<unsigned short>(L"port", 6250);\r
 238 \r
 239                                         osc_servers_.push_back(osc::server(boost::asio::ip::udp::endpoint(boost::asio::ip::address_v4::from_string(narrow(address)), port), monitor_subject_));\r
 240                                 }\r
 241                                 else\r
 242                                         CASPAR_LOG(warning) << "Invalid controller: " << widen(name);   \r
 243                         }\r
 244                         catch(...)\r
 245                         {\r
 246                                 CASPAR_LOG_CURRENT_EXCEPTION();\r
 247                         }\r
 248                 }\r
 249         }\r
 250 \r
 251         void setup_thumbnail_generation(const boost::property_tree::wptree& pt)\r
 252         {\r
 253                 if (!pt.get(L"configuration.thumbnails.generate-thumbnails", true))\r
 254                         return;\r
 255 \r
 256                 auto scan_interval_millis = pt.get(L"configuration.thumbnails.scan-interval-millis", 5000);\r
 257 \r
 258                 polling_filesystem_monitor_factory monitor_factory(scan_interval_millis);\r
 259                 thumbnail_generator_.reset(new thumbnail_generator(\r
 260                                 monitor_factory, \r
 261                                 env::media_folder(),\r
 262                                 env::thumbnails_folder(),\r
 263                                 pt.get(L"configuration.thumbnails.width", 256),\r
 264                                 pt.get(L"configuration.thumbnails.height", 144),\r
 265                                 core::video_format_desc::get(pt.get(L"configuration.thumbnails.video-mode", L"720p2500")),\r
 266                                 ogl_,\r
 267                                 pt.get(L"configuration.thumbnails.generate-delay-millis", 2000),\r
 268                                 &image::write_cropped_png));\r
 269 \r
 270                 CASPAR_LOG(info) << L"Initialized thumbnail generator.";\r
 271         }\r
 272 \r
 273         safe_ptr<IO::IProtocolStrategy> create_protocol(const std::wstring& name) const\r
 274         {\r
 275                 if(boost::iequals(name, L"AMCP"))\r
 276                         return make_safe<amcp::AMCPProtocolStrategy>(channels_, thumbnail_generator_, shutdown_server_now_);\r
 277                 else if(boost::iequals(name, L"CII"))\r
 278                         return make_safe<cii::CIIProtocolStrategy>(channels_);\r
 279                 else if(boost::iequals(name, L"CLOCK"))\r
 280                         //return make_safe<CLK::CLKProtocolStrategy>(channels_);\r
 281                         return make_safe<IO::stateful_protocol_strategy_wrapper>([=]\r
 282                         {\r
 283                                 return std::make_shared<CLK::CLKProtocolStrategy>(channels_);\r
 284                         });\r
 285                 \r
 286                 BOOST_THROW_EXCEPTION(caspar_exception() << arg_name_info("name") << arg_value_info(narrow(name)) << msg_info("Invalid protocol"));\r
 287         }\r
 288 };\r
 289 \r
 290 server::server(boost::promise<bool>& shutdown_server_now) : impl_(new implementation(shutdown_server_now)){}\r
 291 \r
 292 const std::vector<safe_ptr<video_channel>> server::get_channels() const\r
 293 {\r
 294         return impl_->channels_;\r
 295 }\r
 296 \r
 297 std::shared_ptr<thumbnail_generator> server::get_thumbnail_generator() const\r
 298 {\r
 299         return impl_->thumbnail_generator_;\r
 300 }\r
 301 \r
 302 core::monitor::source& server::monitor_output()\r
 303 {\r
 304         return impl_->monitor_subject_;\r
 305 }\r
 306 \r
 307 }