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

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