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