X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=shell%2Fserver.cpp;h=69fa7c083c0430e8e50c568f242b7f04e1d65ad2;hb=d8f53ad9888a04fb5c50551c566b85ffd25afb90;hp=3b3e9b50c346dcd0236cfe43654d1714bbfd47ea;hpb=4fe4692485eed8151acf9a731e3d49d1b84649c3;p=casparcg diff --git a/shell/server.cpp b/shell/server.cpp index 3b3e9b50c..69fa7c083 100644 --- a/shell/server.cpp +++ b/shell/server.cpp @@ -18,9 +18,12 @@ * * Author: Robert Nagy, ronag89@gmail.com */ + #include "stdafx.h" #include "server.h" +#include "included_modules.h" +#include "default_audio_config.h" #include @@ -28,208 +31,458 @@ #include #include #include +#include +#include #include #include +#include #include #include #include +#include #include +#include #include - -#include -#include -#include -#include -#include -#include -#include -#include - -#include -#include -#include -#include -#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include #include +#include +#include #include -#include +#include #include #include +#include +#include #include #include #include -#include #include #include +#include + +#include + +#include namespace caspar { using namespace core; using namespace protocol; -struct server::impl : boost::noncopyable +std::shared_ptr create_running_io_service() { - monitor::basic_subject event_subject_; - accelerator::accelerator accelerator_; - std::vector> async_servers_; - std::vector> channels_; - - impl() - : accelerator_(env::properties().get(L"configuration.accelerator", L"auto")) - { + auto service = std::make_shared(); + // To keep the io_service::run() running although no pending async + // operations are posted. + auto work = std::make_shared(*service); + auto weak_work = std::weak_ptr(work); + auto thread = std::make_shared([service, weak_work] + { + ensure_gpf_handler_installed_for_thread("asio-thread"); - ffmpeg::init(); - CASPAR_LOG(info) << L"Initialized ffmpeg module."; - - bluefish::init(); - CASPAR_LOG(info) << L"Initialized bluefish module."; - - decklink::init(); - CASPAR_LOG(info) << L"Initialized decklink module."; - - oal::init(); - CASPAR_LOG(info) << L"Initialized oal module."; - - screen::init(); - CASPAR_LOG(info) << L"Initialized ogl module."; + while (auto strong = weak_work.lock()) + { + try + { + service->run(); + } + catch (...) + { + CASPAR_LOG_CURRENT_EXCEPTION(); + } + } - image::init(); - CASPAR_LOG(info) << L"Initialized image module."; + CASPAR_LOG(info) << "[asio] Global io_service uninitialized."; + }); - flash::init(); - CASPAR_LOG(info) << L"Initialized flash module."; + return std::shared_ptr( + service.get(), + [service, work, thread](void*) mutable + { + CASPAR_LOG(info) << "[asio] Shutting down global io_service."; + work.reset(); + service->stop(); + if (thread->get_id() != boost::this_thread::get_id()) + thread->join(); + else + thread->detach(); + }); +} - psd::init(); - CASPAR_LOG(info) << L"Initialized psd module."; +struct server::impl : boost::noncopyable +{ + std::shared_ptr io_service_ = create_running_io_service(); + spl::shared_ptr monitor_subject_; + spl::shared_ptr diag_subject_ = core::diagnostics::get_or_create_subject(); + accelerator::accelerator accelerator_; + spl::shared_ptr help_repo_; + std::shared_ptr amcp_command_repo_; + std::vector> async_servers_; + std::shared_ptr primary_amcp_server_; + std::shared_ptr osc_client_ = std::make_shared(io_service_); + std::vector> predefined_osc_subscriptions_; + std::vector> channels_; + spl::shared_ptr media_info_repo_; + boost::thread initial_media_info_thread_; + spl::shared_ptr system_info_provider_repo_; + spl::shared_ptr cg_registry_; + spl::shared_ptr producer_registry_; + spl::shared_ptr consumer_registry_; + tbb::atomic running_; + std::shared_ptr thumbnail_generator_; + std::promise& shutdown_server_now_; + + explicit impl(std::promise& shutdown_server_now) + : accelerator_(env::properties().get(L"configuration.accelerator", L"auto")) + , media_info_repo_(create_in_memory_media_info_repository()) + , producer_registry_(spl::make_shared(help_repo_)) + , consumer_registry_(spl::make_shared(help_repo_)) + , shutdown_server_now_(shutdown_server_now) + { + running_ = false; + core::diagnostics::register_graph_to_log_sink(); + caspar::core::diagnostics::osd::register_sink(); + diag_subject_->attach_parent(monitor_subject_); + + module_dependencies dependencies( + system_info_provider_repo_, + cg_registry_, + media_info_repo_, + producer_registry_, + consumer_registry_); + + initialize_modules(dependencies); + core::text::init(dependencies); + core::scene::init(dependencies); + core::syncto::init(dependencies); + help_repo_->register_item({ L"producer" }, L"Color Producer", &core::describe_color_producer); + } - core::text::init(); + void start() + { + running_ = true; - register_producer_factory(&core::scene::create_dummy_scene_producer); - register_producer_factory(&core::create_text_producer); + setup_audio_config(env::properties()); + CASPAR_LOG(info) << L"Initialized audio config."; setup_channels(env::properties()); CASPAR_LOG(info) << L"Initialized channels."; + setup_thumbnail_generation(env::properties()); + CASPAR_LOG(info) << L"Initialized thumbnail generator."; + setup_controllers(env::properties()); CASPAR_LOG(info) << L"Initialized controllers."; + + setup_osc(env::properties()); + CASPAR_LOG(info) << L"Initialized osc."; + + start_initial_media_info_scan(); + CASPAR_LOG(info) << L"Started initial media information retrieval."; } ~impl() - { + { + if (running_) + { + running_ = false; + initial_media_info_thread_.join(); + } + + std::weak_ptr weak_io_service = io_service_; + io_service_.reset(); + osc_client_.reset(); + thumbnail_generator_.reset(); + amcp_command_repo_.reset(); + primary_amcp_server_.reset(); async_servers_.clear(); + destroy_producers_synchronously(); + destroy_consumers_synchronously(); channels_.clear(); - boost::this_thread::sleep(boost::posix_time::milliseconds(500)); - //Sleep(500); // HACK: Wait for asynchronous destruction of producers and consumers. + while (weak_io_service.lock()) + boost::this_thread::sleep_for(boost::chrono::milliseconds(100)); - image::uninit(); - ffmpeg::uninit(); + uninitialize_modules(); + core::diagnostics::osd::shutdown(); } - + + void setup_audio_config(const boost::property_tree::wptree& pt) + { + using boost::property_tree::wptree; + + auto default_config = get_default_audio_config(); + + // Start with the defaults + audio_channel_layout_repository::get_default()->register_all_layouts(default_config.get_child(L"audio.channel-layouts")); + audio_mix_config_repository::get_default()->register_all_configs(default_config.get_child(L"audio.mix-configs")); + + // Merge with user configuration (adds to or overwrites the defaults) + auto custom_channel_layouts = pt.get_child_optional(L"configuration.audio.channel-layouts"); + auto custom_mix_configs = pt.get_child_optional(L"configuration.audio.mix-configs"); + + if (custom_channel_layouts) + { + CASPAR_SCOPED_CONTEXT_MSG("/configuration/audio/channel-layouts"); + audio_channel_layout_repository::get_default()->register_all_layouts(*custom_channel_layouts); + } + + if (custom_mix_configs) + { + CASPAR_SCOPED_CONTEXT_MSG("/configuration/audio/mix-configs"); + audio_mix_config_repository::get_default()->register_all_configs(*custom_mix_configs); + } + } + void setup_channels(const boost::property_tree::wptree& pt) - { + { using boost::property_tree::wptree; - BOOST_FOREACH(auto& xml_channel, pt.get_child(L"configuration.channels")) - { - auto format_desc = video_format_desc(xml_channel.second.get(L"video-mode", L"PAL")); + + std::vector xml_channels; + + for (auto& xml_channel : pt | witerate_children(L"configuration.channels") | welement_context_iteration) + { + xml_channels.push_back(xml_channel.second); + ptree_verify_element_name(xml_channel, L"channel"); + + auto format_desc_str = xml_channel.second.get(L"video-mode", L"PAL"); + auto format_desc = video_format_desc(format_desc_str); if(format_desc.format == video_format::invalid) - CASPAR_THROW_EXCEPTION(caspar_exception() << msg_info("Invalid video-mode.")); - - auto channel = spl::make_shared(static_cast(channels_.size()+1), format_desc, accelerator_.create_image_mixer()); - - BOOST_FOREACH(auto& xml_consumer, xml_channel.second.get_child(L"consumers")) + CASPAR_THROW_EXCEPTION(user_error() << msg_info(L"Invalid video-mode: " + format_desc_str)); + + auto channel_layout_str = xml_channel.second.get(L"channel-layout", L"stereo"); + auto channel_layout = core::audio_channel_layout_repository::get_default()->get_layout(channel_layout_str); + if (!channel_layout) + CASPAR_THROW_EXCEPTION(user_error() << msg_info(L"Unknown channel-layout: " + channel_layout_str)); + + auto channel_id = static_cast(channels_.size() + 1); + auto channel = spl::make_shared(channel_id, format_desc, *channel_layout, accelerator_.create_image_mixer(channel_id)); + + channel->monitor_output().attach_parent(monitor_subject_); + channel->mixer().set_straight_alpha_output(xml_channel.second.get(L"straight-alpha-output", false)); + channels_.push_back(channel); + } + + for (auto& channel : channels_) + { + core::diagnostics::scoped_call_context save; + core::diagnostics::call_context::for_thread().video_channel = channel->index(); + + for (auto& xml_consumer : xml_channels.at(channel->index() - 1) | witerate_children(L"consumers") | welement_context_iteration) { + auto name = xml_consumer.first; + try { - auto name = xml_consumer.first; - if(name == L"screen") - channel->output().add(caspar::screen::create_consumer(xml_consumer.second, &channel->stage())); - else if(name == L"bluefish") - channel->output().add(bluefish::create_consumer(xml_consumer.second)); - else if(name == L"decklink") - channel->output().add(decklink::create_consumer(xml_consumer.second)); - else if(name == L"file") - channel->output().add(ffmpeg::create_consumer(xml_consumer.second)); - else if(name == L"system-audio") - channel->output().add(oal::create_consumer()); - else if(name != L"") - CASPAR_LOG(warning) << "Invalid consumer: " << name; + if (name != L"") + channel->output().add(consumer_registry_->create_consumer(name, xml_consumer.second, &channel->stage(), channels_)); + } + catch (const user_error& e) + { + CASPAR_LOG_CURRENT_EXCEPTION_AT_LEVEL(debug); + CASPAR_LOG(error) << get_message_and_context(e) << " Turn on log level debug for stacktrace."; } - catch(...) + catch (...) { CASPAR_LOG_CURRENT_EXCEPTION(); } - } - - channel->subscribe(monitor::observable::observer_ptr(event_subject_)); - channels_.push_back(channel); + } } // Dummy diagnostics channel - if(env::properties().get(L"configuration.channel-grid", false)) - channels_.push_back(spl::make_shared(static_cast(channels_.size()+1), core::video_format_desc(core::video_format::x576p2500), accelerator_.create_image_mixer())); + if (env::properties().get(L"configuration.channel-grid", false)) + { + auto channel_id = static_cast(channels_.size() + 1); + channels_.push_back(spl::make_shared( + channel_id, + core::video_format_desc(core::video_format::x576p2500), + *core::audio_channel_layout_repository::get_default()->get_layout(L"stereo"), + accelerator_.create_image_mixer(channel_id))); + channels_.back()->monitor_output().attach_parent(monitor_subject_); + } } - - void setup_controllers(const boost::property_tree::wptree& pt) - { + + void setup_osc(const boost::property_tree::wptree& pt) + { using boost::property_tree::wptree; - BOOST_FOREACH(auto& xml_controller, pt.get_child(L"configuration.controllers")) + using namespace boost::asio::ip; + + monitor_subject_->attach_parent(osc_client_->sink()); + + auto default_port = + pt.get(L"configuration.osc.default-port", 6250); + auto disable_send_to_amcp_clients = + pt.get(L"configuration.osc.disable-send-to-amcp-clients", false); + auto predefined_clients = + pt.get_child_optional(L"configuration.osc.predefined-clients"); + + if (predefined_clients) { - try + for (auto& predefined_client : pt | witerate_children(L"configuration.osc.predefined-clients") | welement_context_iteration) { - auto name = xml_controller.first; - auto protocol = xml_controller.second.get(L"protocol"); - - if(name == L"tcp") - { - unsigned int port = xml_controller.second.get(L"port", 5250); - auto asyncbootstrapper = spl::make_shared(create_protocol(protocol), port); - async_servers_.push_back(asyncbootstrapper); - - //TODO: remove - test - asyncbootstrapper->add_client_lifecycle_object_factory([=] (const std::string& ipv4_address) { - return std::pair>(L"log", std::shared_ptr(nullptr, [] (void*) - { CASPAR_LOG(info) << "Client disconnect (lifecycle)"; })); - }); - } - else - CASPAR_LOG(warning) << "Invalid controller: " << name; + ptree_verify_element_name(predefined_client, L"predefined-client"); + + const auto address = + ptree_get(predefined_client.second, L"address"); + const auto port = + ptree_get(predefined_client.second, L"port"); + predefined_osc_subscriptions_.push_back( + osc_client_->get_subscription_token(udp::endpoint( + address_v4::from_string(u8(address)), + port))); } - catch(...) + } + + if (!disable_send_to_amcp_clients && primary_amcp_server_) + primary_amcp_server_->add_client_lifecycle_object_factory( + [=] (const std::string& ipv4_address) + -> std::pair> + { + using namespace boost::asio::ip; + + return std::make_pair( + std::wstring(L"osc_subscribe"), + osc_client_->get_subscription_token( + udp::endpoint( + address_v4::from_string( + ipv4_address), + default_port))); + }); + } + + void setup_thumbnail_generation(const boost::property_tree::wptree& pt) + { + if (!pt.get(L"configuration.thumbnails.generate-thumbnails", true)) + return; + + auto scan_interval_millis = pt.get(L"configuration.thumbnails.scan-interval-millis", 5000); + + polling_filesystem_monitor_factory monitor_factory(io_service_, scan_interval_millis); + thumbnail_generator_.reset(new thumbnail_generator( + monitor_factory, + env::media_folder(), + env::thumbnails_folder(), + pt.get(L"configuration.thumbnails.width", 256), + pt.get(L"configuration.thumbnails.height", 144), + core::video_format_desc(pt.get(L"configuration.thumbnails.video-mode", L"720p2500")), + accelerator_.create_image_mixer(0), + pt.get(L"configuration.thumbnails.generate-delay-millis", 2000), + &image::write_cropped_png, + media_info_repo_, + producer_registry_, + pt.get(L"configuration.thumbnails.mipmap", true))); + } + + void setup_controllers(const boost::property_tree::wptree& pt) + { + amcp_command_repo_ = spl::make_shared( + channels_, + thumbnail_generator_, + media_info_repo_, + system_info_provider_repo_, + cg_registry_, + help_repo_, + producer_registry_, + consumer_registry_, + accelerator_.get_ogl_device(), + shutdown_server_now_); + amcp::register_commands(*amcp_command_repo_); + + using boost::property_tree::wptree; + for (auto& xml_controller : pt | witerate_children(L"configuration.controllers") | welement_context_iteration) + { + auto name = xml_controller.first; + auto protocol = ptree_get(xml_controller.second, L"protocol"); + + if(name == L"tcp") { - CASPAR_LOG_CURRENT_EXCEPTION(); + auto port = ptree_get(xml_controller.second, L"port"); + auto asyncbootstrapper = spl::make_shared( + io_service_, + create_protocol(protocol, L"TCP Port " + boost::lexical_cast(port)), + port); + async_servers_.push_back(asyncbootstrapper); + + if (!primary_amcp_server_ && boost::iequals(protocol, L"AMCP")) + primary_amcp_server_ = asyncbootstrapper; } + else + CASPAR_LOG(warning) << "Invalid controller: " << name; } } - IO::protocol_strategy_factory::ptr create_protocol(const std::wstring& name) const + IO::protocol_strategy_factory::ptr create_protocol(const std::wstring& name, const std::wstring& port_description) const { using namespace IO; if(boost::iequals(name, L"AMCP")) - return wrap_legacy_protocol("\r\n", spl::make_shared(channels_)); + return wrap_legacy_protocol("\r\n", spl::make_shared(port_description, spl::make_shared_ptr(amcp_command_repo_))); else if(boost::iequals(name, L"CII")) - return wrap_legacy_protocol("\r\n", spl::make_shared(channels_)); + return wrap_legacy_protocol("\r\n", spl::make_shared(channels_, cg_registry_, producer_registry_)); else if(boost::iequals(name, L"CLOCK")) return spl::make_shared( "ISO-8859-1", - spl::make_shared(channels_)); - - CASPAR_THROW_EXCEPTION(caspar_exception() << arg_name_info(L"name") << arg_value_info(name) << msg_info(L"Invalid protocol")); + spl::make_shared(channels_, cg_registry_, producer_registry_)); + else if (boost::iequals(name, L"LOG")) + return spl::make_shared(); + + CASPAR_THROW_EXCEPTION(user_error() << msg_info(L"Invalid protocol: " + name)); } + void start_initial_media_info_scan() + { + initial_media_info_thread_ = boost::thread([this] + { + try + { + ensure_gpf_handler_installed_for_thread("initial media scan"); + + for (boost::filesystem::wrecursive_directory_iterator iter(env::media_folder()), end; iter != end; ++iter) + { + if (running_) + { + if (boost::filesystem::is_regular_file(iter->path())) + media_info_repo_->get(iter->path().wstring()); + } + else + { + CASPAR_LOG(info) << L"Initial media information retrieval aborted."; + return; + } + } + + CASPAR_LOG(info) << L"Initial media information retrieval finished."; + } + catch (...) + { + CASPAR_LOG_CURRENT_EXCEPTION(); + } + }); + } }; -server::server() : impl_(new impl()){} +server::server(std::promise& shutdown_server_now) : impl_(new impl(shutdown_server_now)){} +void server::start() { impl_->start(); } +spl::shared_ptr server::get_system_info_provider_repo() const { return impl_->system_info_provider_repo_; } +spl::shared_ptr server::get_amcp_command_repository() const { return spl::make_shared_ptr(impl_->amcp_command_repo_); } +core::monitor::subject& server::monitor_output() { return *impl_->monitor_subject_; } -const std::vector> server::channels() const -{ - return impl_->channels_; } -void server::subscribe(const monitor::observable::observer_ptr& o){impl_->event_subject_.subscribe(o);} -void server::unsubscribe(const monitor::observable::observer_ptr& o){impl_->event_subject_.unsubscribe(o);} - -} \ No newline at end of file