git.sesse.net Git - casparcg/blob - shell/server.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 #include "stdafx.h"
  22
  23 #include "server.h"
  24
  25 #include <accelerator/accelerator.h>
  26
  27 #include <common/env.h>
  28 #include <common/except.h>
  29 #include <common/utf.h>
  30 #include <common/memory.h>
  31 #include <common/polling_filesystem_monitor.h>
  32
  33 #include <core/video_channel.h>
  34 #include <core/video_format.h>
  35 #include <core/producer/stage.h>
  36 #include <core/producer/frame_producer.h>
  37 #include <core/producer/scene/scene_producer.h>
  38 #include <core/producer/scene/xml_scene_producer.h>
  39 #include <core/producer/text/text_producer.h>
  40 #include <core/consumer/output.h>
  41 #include <core/thumbnail_generator.h>
  42
  43 #include <modules/bluefish/bluefish.h>
  44 #include <modules/decklink/decklink.h>
  45 #include <modules/ffmpeg/ffmpeg.h>
  46 #include <modules/flash/flash.h>
  47 #include <modules/oal/oal.h>
  48 #include <modules/screen/screen.h>
  49 #include <modules/image/image.h>
  50 #include <modules/image/consumer/image_consumer.h>
  51 #include <modules/psd/psd_scene_producer.h>
  52
  53 #include <modules/oal/consumer/oal_consumer.h>
  54 #include <modules/bluefish/consumer/bluefish_consumer.h>
  55 #include <modules/decklink/consumer/decklink_consumer.h>
  56 #include <modules/screen/consumer/screen_consumer.h>
  57 #include <modules/ffmpeg/consumer/ffmpeg_consumer.h>
  58
  59 #include <protocol/amcp/AMCPProtocolStrategy.h>
  60 #include <protocol/cii/CIIProtocolStrategy.h>
  61 #include <protocol/CLK/CLKProtocolStrategy.h>
  62 #include <protocol/util/AsyncEventServer.h>
  63 #include <protocol/util/strategy_adapters.h>
  64
  65 #include <boost/algorithm/string.hpp>
  66 #include <boost/thread.hpp>
  67 #include <boost/lexical_cast.hpp>
  68 #include <boost/foreach.hpp>
  69 #include <boost/property_tree/ptree.hpp>
  70 #include <boost/property_tree/xml_parser.hpp>
  71
  72 namespace caspar {
  73
  74 using namespace core;
  75 using namespace protocol;
  76
  77 struct server::impl : boost::noncopyable
  78 {
  79         monitor::basic_subject                                                          event_subject_;
  80         accelerator::accelerator                                                        accelerator_;
  81         std::vector<spl::shared_ptr<IO::AsyncEventServer>>      async_servers_;
  82         std::vector<spl::shared_ptr<video_channel>>                     channels_;
  83         std::shared_ptr<thumbnail_generator>                            thumbnail_generator_;
  84
  85         impl()
  86                 : accelerator_(env::properties().get(L"configuration.accelerator", L"auto"))
  87         {
  88
  89                 ffmpeg::init();
  90                 CASPAR_LOG(info) << L"Initialized ffmpeg module.";
  91
  92                 bluefish::init();
  93                 CASPAR_LOG(info) << L"Initialized bluefish module.";
  94
  95                 decklink::init();
  96                 CASPAR_LOG(info) << L"Initialized decklink module.";
  97
  98                 oal::init();
  99                 CASPAR_LOG(info) << L"Initialized oal module.";
 100
 101                 screen::init();
 102                 CASPAR_LOG(info) << L"Initialized ogl module.";
 103
 104                 image::init();
 105                 CASPAR_LOG(info) << L"Initialized image module.";
 106
 107                 flash::init();
 108                 CASPAR_LOG(info) << L"Initialized flash module.";
 109
 110                 psd::init();
 111                 CASPAR_LOG(info) << L"Initialized psd module.";
 112
 113                 core::text::init();
 114
 115                 register_producer_factory(&core::scene::create_dummy_scene_producer);
 116                 register_producer_factory(&core::scene::create_xml_scene_producer);
 117
 118                 setup_channels(env::properties());
 119                 CASPAR_LOG(info) << L"Initialized channels.";
 120
 121                 setup_thumbnail_generation(env::properties());
 122                 CASPAR_LOG(info) << L"Initialized thumbnail generator.";
 123
 124                 setup_controllers(env::properties());
 125                 CASPAR_LOG(info) << L"Initialized controllers.";
 126         }
 127
 128         ~impl()
 129         {
 130                 async_servers_.clear();
 131                 channels_.clear();
 132
 133                 boost::this_thread::sleep(boost::posix_time::milliseconds(500));
 134                 //Sleep(500); // HACK: Wait for asynchronous destruction of producers and consumers.
 135
 136                 image::uninit();
 137                 ffmpeg::uninit();
 138         }
 139
 140         void setup_channels(const boost::property_tree::wptree& pt)
 141         {
 142                 using boost::property_tree::wptree;
 143                 BOOST_FOREACH(auto& xml_channel, pt.get_child(L"configuration.channels"))
 144                 {
 145                         auto format_desc = video_format_desc(xml_channel.second.get(L"video-mode", L"PAL"));
 146                         if(format_desc.format == video_format::invalid)
 147                                 CASPAR_THROW_EXCEPTION(caspar_exception() << msg_info("Invalid video-mode."));
 148
 149                         auto channel = spl::make_shared<video_channel>(static_cast<int>(channels_.size()+1), format_desc, accelerator_.create_image_mixer());
 150
 151                         BOOST_FOREACH(auto& xml_consumer, xml_channel.second.get_child(L"consumers"))
 152                         {
 153                                 try
 154                                 {
 155                                         auto name = xml_consumer.first;
 156                                         if(name == L"screen")
 157                                                 channel->output().add(caspar::screen::create_consumer(xml_consumer.second, &channel->stage()));
 158                                         else if(name == L"bluefish")
 159                                                 channel->output().add(bluefish::create_consumer(xml_consumer.second));
 160                                         else if(name == L"decklink")
 161                                                 channel->output().add(decklink::create_consumer(xml_consumer.second));
 162                                         else if(name == L"file")
 163                                                 channel->output().add(ffmpeg::create_consumer(xml_consumer.second));
 164                                         else if(name == L"system-audio")
 165                                                 channel->output().add(oal::create_consumer());
 166                                         else if(name != L"<xmlcomment>")
 167                                                 CASPAR_LOG(warning) << "Invalid consumer: " << name;
 168                                 }
 169                                 catch(...)
 170                                 {
 171                                         CASPAR_LOG_CURRENT_EXCEPTION();
 172                                 }
 173                         }
 174
 175                     channel->subscribe(monitor::observable::observer_ptr(event_subject_));
 176                         channels_.push_back(channel);
 177                 }
 178
 179                 // Dummy diagnostics channel
 180                 if(env::properties().get(L"configuration.channel-grid", false))
 181                         channels_.push_back(spl::make_shared<video_channel>(static_cast<int>(channels_.size()+1), core::video_format_desc(core::video_format::x576p2500), accelerator_.create_image_mixer()));
 182         }
 183
 184         void setup_thumbnail_generation(const boost::property_tree::wptree& pt)
 185         {
 186                 if (!pt.get(L"configuration.thumbnails.generate-thumbnails", true))
 187                         return;
 188
 189                 auto scan_interval_millis = pt.get(L"configuration.thumbnails.scan-interval-millis", 5000);
 190
 191                 polling_filesystem_monitor_factory monitor_factory(scan_interval_millis);
 192                 thumbnail_generator_.reset(new thumbnail_generator(
 193                         monitor_factory,
 194                         env::media_folder(),
 195                         env::thumbnails_folder(),
 196                         pt.get(L"configuration.thumbnails.width", 256),
 197                         pt.get(L"configuration.thumbnails.height", 144),
 198                         core::video_format_desc(pt.get(L"configuration.thumbnails.video-mode", L"720p2500")),
 199                         accelerator_.create_image_mixer(),
 200                         pt.get(L"configuration.thumbnails.generate-delay-millis", 2000),
 201                         &image::write_cropped_png));
 202
 203                 CASPAR_LOG(info) << L"Initialized thumbnail generator.";
 204         }
 205
 206         void setup_controllers(const boost::property_tree::wptree& pt)
 207         {
 208                 using boost::property_tree::wptree;
 209                 BOOST_FOREACH(auto& xml_controller, pt.get_child(L"configuration.controllers"))
 210                 {
 211                         try
 212                         {
 213                                 auto name = xml_controller.first;
 214                                 auto protocol = xml_controller.second.get<std::wstring>(L"protocol");
 215
 216                                 if(name == L"tcp")
 217                                 {
 218                                         unsigned int port = xml_controller.second.get(L"port", 5250);
 219                                         auto asyncbootstrapper = spl::make_shared<IO::AsyncEventServer>(create_protocol(protocol), port);
 220                                         async_servers_.push_back(asyncbootstrapper);
 221
 222                                         //TODO: remove - test
 223                                         asyncbootstrapper->add_client_lifecycle_object_factory([=] (const std::string& ipv4_address) {
 224                                                                                                                                                                         return std::pair<std::wstring, std::shared_ptr<void>>(L"log", std::shared_ptr<void>(nullptr, [] (void*)
 225                                                                                                                                                                         { CASPAR_LOG(info) << "Client disconnect (lifecycle)"; }));
 226                                                                                                                                                                 });
 227                                 }
 228                                 else
 229                                         CASPAR_LOG(warning) << "Invalid controller: " << name;
 230                         }
 231                         catch(...)
 232                         {
 233                                 CASPAR_LOG_CURRENT_EXCEPTION();
 234                         }
 235                 }
 236         }
 237
 238         IO::protocol_strategy_factory<char>::ptr create_protocol(const std::wstring& name) const
 239         {
 240                 using namespace IO;
 241
 242                 if(boost::iequals(name, L"AMCP"))
 243                         return wrap_legacy_protocol("\r\n", spl::make_shared<amcp::AMCPProtocolStrategy>(channels_, thumbnail_generator_));
 244                 else if(boost::iequals(name, L"CII"))
 245                         return wrap_legacy_protocol("\r\n", spl::make_shared<cii::CIIProtocolStrategy>(channels_));
 246                 else if(boost::iequals(name, L"CLOCK"))
 247                         return spl::make_shared<to_unicode_adapter_factory>(
 248                                         "ISO-8859-1",
 249                                         spl::make_shared<CLK::clk_protocol_strategy_factory>(channels_));
 250
 251                 CASPAR_THROW_EXCEPTION(caspar_exception() << arg_name_info(L"name") << arg_value_info(name) << msg_info(L"Invalid protocol"));
 252         }
 253
 254 };
 255
 256 server::server() : impl_(new impl()){}
 257
 258 const std::vector<spl::shared_ptr<video_channel>> server::channels() const
 259 {
 260         return impl_->channels_;
 261 }
 262 std::shared_ptr<core::thumbnail_generator> server::get_thumbnail_generator() const {return impl_->thumbnail_generator_; }
 263 void server::subscribe(const monitor::observable::observer_ptr& o){impl_->event_subject_.subscribe(o);}
 264 void server::unsubscribe(const monitor::observable::observer_ptr& o){impl_->event_subject_.unsubscribe(o);}
 265
 266 }