git.sesse.net Git - casparcg/blob - modules/flash/producer/cg_producer.cpp

   1 /*\r
   2 * Copyright 2013 Sveriges Television AB http://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 #include "../StdAfx.h"\r
  23 \r
  24 #include "cg_producer.h"\r
  25 \r
  26 #include "flash_producer.h"\r
  27 \r
  28 #include <common/env.h>\r
  29 \r
  30 #include <core/parameters/parameters.h>\r
  31 #include <core/mixer/mixer.h>\r
  32 \r
  33 #include <boost/filesystem.hpp>\r
  34 #include <boost/format.hpp>\r
  35 #include <boost/algorithm/string.hpp>\r
  36 #include <boost/regex.hpp>\r
  37 #include <boost/property_tree/ptree.hpp>\r
  38                 \r
  39 namespace caspar { namespace flash {\r
  40         \r
  41 struct cg_producer::implementation : boost::noncopyable\r
  42 {\r
  43         safe_ptr<core::frame_producer> flash_producer_;\r
  44 public:\r
  45         implementation(const safe_ptr<core::frame_producer>& frame_producer) \r
  46                 : flash_producer_(frame_producer)\r
  47         {}\r
  48         \r
  49         boost::unique_future<std::wstring> add(int layer, std::wstring filename,  bool play_on_load, const std::wstring& label, const std::wstring& data)\r
  50         {\r
  51                 if(filename.size() > 0 && filename[0] == L'/')\r
  52                         filename = filename.substr(1, filename.size()-1);\r
  53 \r
  54                 if(boost::filesystem::wpath(filename).extension() == L"")\r
  55                         filename += L".ft";\r
  56                 \r
  57                 auto str = (boost::wformat(L"<invoke name=\"Add\" returntype=\"xml\"><arguments><number>%1%</number><string>%2%</string>%3%<string>%4%</string><string><![CDATA[%5%]]></string></arguments></invoke>") % layer % filename % (play_on_load?TEXT("<true/>"):TEXT("<false/>")) % label % data).str();\r
  58 \r
  59                 CASPAR_LOG(info) << flash_producer_->print() << " Invoking add-command: " << str;\r
  60                 return flash_producer_->call(str);\r
  61         }\r
  62 \r
  63         boost::unique_future<std::wstring> remove(int layer)\r
  64         {\r
  65                 auto str = (boost::wformat(L"<invoke name=\"Delete\" returntype=\"xml\"><arguments><array><property id=\"0\"><number>%1%</number></property></array></arguments></invoke>") % layer).str();\r
  66                 CASPAR_LOG(info) << flash_producer_->print() << " Invoking remove-command: " << str;\r
  67                 return flash_producer_->call(str);\r
  68         }\r
  69 \r
  70         boost::unique_future<std::wstring> play(int layer)\r
  71         {\r
  72                 auto str = (boost::wformat(L"<invoke name=\"Play\" returntype=\"xml\"><arguments><array><property id=\"0\"><number>%1%</number></property></array></arguments></invoke>") % layer).str();\r
  73                 CASPAR_LOG(info) << flash_producer_->print() << " Invoking play-command: " << str;\r
  74                 return flash_producer_->call(str);\r
  75         }\r
  76 \r
  77         boost::unique_future<std::wstring> stop(int layer, unsigned int)\r
  78         {\r
  79                 auto str = (boost::wformat(L"<invoke name=\"Stop\" returntype=\"xml\"><arguments><array><property id=\"0\"><number>%1%</number></property></array><number>0</number></arguments></invoke>") % layer).str();\r
  80                 CASPAR_LOG(info) << flash_producer_->print() << " Invoking stop-command: " << str;\r
  81                 return flash_producer_->call(str);\r
  82         }\r
  83 \r
  84         boost::unique_future<std::wstring> next(int layer)\r
  85         {\r
  86                 auto str = (boost::wformat(L"<invoke name=\"Next\" returntype=\"xml\"><arguments><array><property id=\"0\"><number>%1%</number></property></array></arguments></invoke>") % layer).str();\r
  87                 CASPAR_LOG(info) << flash_producer_->print() << " Invoking next-command: " << str;\r
  88                 return flash_producer_->call(str);\r
  89         }\r
  90 \r
  91         boost::unique_future<std::wstring> update(int layer, const std::wstring& data)\r
  92         {\r
  93                 auto str = (boost::wformat(L"<invoke name=\"SetData\" returntype=\"xml\"><arguments><array><property id=\"0\"><number>%1%</number></property></array><string><![CDATA[%2%]]></string></arguments></invoke>") % layer % data).str();\r
  94                 CASPAR_LOG(info) << flash_producer_->print() <<" Invoking update-command: " << str;\r
  95                 return flash_producer_->call(str);\r
  96         }\r
  97 \r
  98         boost::unique_future<std::wstring> invoke(int layer, const std::wstring& label)\r
  99         {\r
 100                 auto str = (boost::wformat(L"<invoke name=\"Invoke\" returntype=\"xml\"><arguments><array><property id=\"0\"><number>%1%</number></property></array><string>%2%</string></arguments></invoke>") % layer % label).str();\r
 101                 CASPAR_LOG(info) << flash_producer_->print() << " Invoking invoke-command: " << str;\r
 102                 return flash_producer_->call(str);\r
 103         }\r
 104 \r
 105         boost::unique_future<std::wstring> description(int layer)\r
 106         {\r
 107                 auto str = (boost::wformat(L"<invoke name=\"GetDescription\" returntype=\"xml\"><arguments><array><property id=\"0\"><number>%1%</number></property></array></arguments></invoke>") % layer).str();\r
 108                 CASPAR_LOG(info) << flash_producer_->print() << " Invoking description-command: " << str;\r
 109                 return flash_producer_->call(str);\r
 110         }\r
 111 \r
 112         boost::unique_future<std::wstring> template_host_info()\r
 113         {\r
 114                 auto str = (boost::wformat(L"<invoke name=\"GetInfo\" returntype=\"xml\"><arguments></arguments></invoke>")).str();\r
 115                 CASPAR_LOG(info) << flash_producer_->print() << " Invoking info-command: " << str;\r
 116                 return flash_producer_->call(str);\r
 117         }\r
 118 \r
 119         boost::unique_future<std::wstring> call(const std::wstring& str)\r
 120         {               \r
 121                 static const boost::wregex add_exp                      (L"ADD (?<LAYER>\\d+) (?<FILENAME>[^\\s]+) (?<PLAY_ON_LOAD>\\d)( (?<DATA>.*))?");\r
 122                 static const boost::wregex remove_exp           (L"REMOVE (?<LAYER>\\d+)");\r
 123                 static const boost::wregex play_exp                     (L"PLAY (?<LAYER>\\d+)");\r
 124                 static const boost::wregex stop_exp                     (L"STOP (?<LAYER>\\d+)");\r
 125                 static const boost::wregex next_exp                     (L"NEXT (?<LAYER>\\d+)");\r
 126                 static const boost::wregex update_exp           (L"UPDATE (?<LAYER>\\d+) (?<DATA>.+)");\r
 127                 static const boost::wregex invoke_exp           (L"INVOKE (?<LAYER>\\d+) (?<LABEL>.+)");\r
 128                 static const boost::wregex description_exp      (L"INFO (?<LAYER>\\d+)");\r
 129                 static const boost::wregex info_exp                     (L"INFO");\r
 130                 \r
 131                 boost::wsmatch what;\r
 132                 if(boost::regex_match(str, what, add_exp))\r
 133                         return add(boost::lexical_cast<int>(what["LAYER"].str()), flash::find_template(env::template_folder() + what["FILENAME"].str()), boost::lexical_cast<bool>(what["PLAY_ON_LOAD"].str()), L"", what["DATA"].str()); \r
 134                 else if(boost::regex_match(str, what, remove_exp))\r
 135                         return remove(boost::lexical_cast<int>(what["LAYER"].str())); \r
 136                 else if(boost::regex_match(str, what, stop_exp))\r
 137                         return stop(boost::lexical_cast<int>(what["LAYER"].str()), 0); \r
 138                 else if(boost::regex_match(str, what, next_exp))\r
 139                         return next(boost::lexical_cast<int>(what["LAYER"].str())); \r
 140                 else if(boost::regex_match(str, what, update_exp))\r
 141                         return update(boost::lexical_cast<int>(what["LAYER"].str()), what["DATA"].str()); \r
 142                 else if(boost::regex_match(str, what, next_exp))\r
 143                         return invoke(boost::lexical_cast<int>(what["LAYER"].str()), what["LABEL"].str()); \r
 144                 else if(boost::regex_match(str, what, description_exp))\r
 145                         return description(boost::lexical_cast<int>(what["LAYER"].str())); \r
 146                 else if(boost::regex_match(str, what, invoke_exp))\r
 147                         return template_host_info(); \r
 148 \r
 149                 return flash_producer_->call(str);\r
 150         }\r
 151 \r
 152         safe_ptr<core::basic_frame> receive(int hints)\r
 153         {\r
 154                 return flash_producer_->receive(hints);\r
 155         }\r
 156 \r
 157         safe_ptr<core::basic_frame> last_frame() const\r
 158         {\r
 159                 return flash_producer_->last_frame();\r
 160         }               \r
 161                         \r
 162         std::wstring print() const\r
 163         {\r
 164                 return flash_producer_->print();\r
 165         }\r
 166 \r
 167         boost::property_tree::wptree info() const\r
 168         {\r
 169                 boost::property_tree::wptree info;\r
 170                 info.add(L"type", L"cg-producer");\r
 171                 return info;\r
 172         }\r
 173 \r
 174         std::wstring timed_invoke(int layer, const std::wstring& label)\r
 175         {\r
 176                 auto result = invoke(layer, label);\r
 177                 if(result.timed_wait(boost::posix_time::seconds(2)))\r
 178                         return result.get();\r
 179                 return L"";\r
 180         }\r
 181         std::wstring timed_description(int layer)\r
 182         {\r
 183                 auto result = description(layer);\r
 184                 if(result.timed_wait(boost::posix_time::seconds(2)))\r
 185                         return result.get();\r
 186                 return L"";\r
 187         }\r
 188         std::wstring timed_template_host_info()\r
 189         {\r
 190                 auto result = template_host_info();\r
 191                 if(result.timed_wait(boost::posix_time::seconds(2)))\r
 192                         return result.get();\r
 193                 return L"";\r
 194         }\r
 195 \r
 196         core::monitor::source& monitor_output()\r
 197         {\r
 198                 return flash_producer_->monitor_output();\r
 199         }\r
 200 };\r
 201         \r
 202 safe_ptr<cg_producer> get_default_cg_producer(const safe_ptr<core::video_channel>& video_channel, int render_layer)\r
 203 {       \r
 204         auto flash_producer = video_channel->stage()->foreground(render_layer).get();\r
 205 \r
 206         try\r
 207         {\r
 208                 if(flash_producer->print().find(L"flash[") == std::string::npos) // UGLY hack\r
 209                 {\r
 210                         flash_producer = flash::create_producer(video_channel->mixer(), boost::assign::list_of<std::wstring>());        \r
 211                         video_channel->stage()->load(render_layer, flash_producer); \r
 212                         video_channel->stage()->play(render_layer);\r
 213                 }\r
 214         }\r
 215         catch(...)\r
 216         {\r
 217                 CASPAR_LOG_CURRENT_EXCEPTION();\r
 218                 throw;\r
 219         }\r
 220 \r
 221         return make_safe<cg_producer>(flash_producer);\r
 222 }\r
 223 \r
 224 safe_ptr<core::frame_producer> create_cg_producer_and_autoplay_file(\r
 225                 const safe_ptr<core::frame_factory>& frame_factory, \r
 226                 const std::vector<std::wstring>& params,\r
 227                 const std::wstring& filename) \r
 228 {\r
 229         if(!boost::filesystem::exists(filename))\r
 230                 return core::frame_producer::empty();\r
 231                 \r
 232         boost::filesystem2::wpath path(filename);\r
 233         path = boost::filesystem2::complete(path);\r
 234         auto filename2 = path.file_string();\r
 235 \r
 236         auto flash_producer = flash::create_producer(frame_factory, boost::assign::list_of<std::wstring>());    \r
 237         auto producer = make_safe<cg_producer>(flash_producer);\r
 238         producer->add(0, filename2, 1);\r
 239 \r
 240         return producer;\r
 241 }\r
 242 \r
 243 safe_ptr<core::frame_producer> create_ct_producer(\r
 244                 const safe_ptr<core::frame_factory>& frame_factory,\r
 245                 const core::parameters& params) \r
 246 {\r
 247         return create_cg_producer_and_autoplay_file(frame_factory, params, env::media_folder() + L"\\" + params[0] + L".ct");\r
 248 }\r
 249 \r
 250 safe_ptr<core::frame_producer> create_cg_producer(\r
 251                 const safe_ptr<core::frame_factory>& frame_factory,\r
 252                 const core::parameters& params) \r
 253 {\r
 254         if(params.empty() || params.at(0) != L"[CG]")\r
 255                 return core::frame_producer::empty();\r
 256 \r
 257         return make_safe<cg_producer>(flash::create_producer(frame_factory, boost::assign::list_of<std::wstring>()));   \r
 258 }\r
 259 \r
 260 cg_producer::cg_producer(const safe_ptr<core::frame_producer>& frame_producer) : impl_(new implementation(frame_producer)){}\r
 261 cg_producer::cg_producer(cg_producer&& other) : impl_(std::move(other.impl_)){}\r
 262 safe_ptr<core::basic_frame> cg_producer::receive(int hints){return impl_->receive(hints);}\r
 263 safe_ptr<core::basic_frame> cg_producer::last_frame() const{return impl_->last_frame();}\r
 264 void cg_producer::add(int layer, const std::wstring& template_name,  bool play_on_load, const std::wstring& startFromLabel, const std::wstring& data){impl_->add(layer, template_name, play_on_load, startFromLabel, data);}\r
 265 void cg_producer::remove(int layer){impl_->remove(layer);}\r
 266 void cg_producer::play(int layer){impl_->play(layer);}\r
 267 void cg_producer::stop(int layer, unsigned int mix_out_duration){impl_->stop(layer, mix_out_duration);}\r
 268 void cg_producer::next(int layer){impl_->next(layer);}\r
 269 void cg_producer::update(int layer, const std::wstring& data){impl_->update(layer, data);}\r
 270 std::wstring cg_producer::print() const{return impl_->print();}\r
 271 boost::unique_future<std::wstring> cg_producer::call(const std::wstring& str){return impl_->call(str);}\r
 272 std::wstring cg_producer::invoke(int layer, const std::wstring& label){return impl_->timed_invoke(layer, label);}\r
 273 std::wstring cg_producer::description(int layer){return impl_->timed_description(layer);}\r
 274 std::wstring cg_producer::template_host_info(){return impl_->timed_template_host_info();}\r
 275 boost::property_tree::wptree cg_producer::info() const{return impl_->info();}\r
 276 core::monitor::source& cg_producer::monitor_output(){return impl_->monitor_output();}\r
 277 }}