git.sesse.net Git - casparcg/blob - core/producer/frame_producer.cpp

   1 #include "../StdAfx.h"\r
   2 \r
   3 #include "frame_producer.h"\r
   4 #include "color/color_producer.h"\r
   5 \r
   6 #include <common/memory/safe_ptr.h>\r
   7 \r
   8 #include <tbb/spin_rw_mutex.h>\r
   9 \r
  10 namespace caspar { namespace core {\r
  11         \r
  12 std::vector<const producer_factory_t> p_factories;\r
  13 tbb::spin_rw_mutex p_factories_mutex;\r
  14 \r
  15 safe_ptr<basic_frame> receive_and_follow(safe_ptr<frame_producer>& producer)\r
  16 {       \r
  17         if(producer == frame_producer::empty())\r
  18                 return basic_frame::eof();\r
  19 \r
  20         auto frame = basic_frame::eof();\r
  21         try\r
  22         {\r
  23                 frame = producer->receive();\r
  24         }\r
  25         catch(...)\r
  26         {\r
  27                 try\r
  28                 {\r
  29                         // Producer will be removed since frame == basic_frame::eof.\r
  30                         CASPAR_LOG_CURRENT_EXCEPTION();\r
  31                         CASPAR_LOG(warning) << producer->print() << " Failed to receive frame. Removing producer.";\r
  32                 }\r
  33                 catch(...){}\r
  34         }\r
  35 \r
  36         if(frame == basic_frame::eof())\r
  37         {\r
  38                 auto following = producer->get_following_producer();\r
  39                 following->set_leading_producer(producer);\r
  40                 producer = std::move(following);                \r
  41 \r
  42                 return receive_and_follow(producer);\r
  43         }\r
  44         return frame;\r
  45 }\r
  46 \r
  47 std::wostream& operator<<(std::wostream& out, const frame_producer& producer)\r
  48 {\r
  49         out << producer.print().c_str();\r
  50         return out;\r
  51 }\r
  52 \r
  53 std::wostream& operator<<(std::wostream& out, const safe_ptr<const frame_producer>& producer)\r
  54 {\r
  55         out << producer->print().c_str();\r
  56         return out;\r
  57 }\r
  58 \r
  59 void register_producer_factory(const producer_factory_t& factory)\r
  60 {\r
  61         tbb::spin_rw_mutex::scoped_lock(p_factories_mutex, true);\r
  62         p_factories.push_back(factory);\r
  63 }\r
  64 \r
  65 safe_ptr<core::frame_producer> create_producer(const safe_ptr<frame_factory>& my_frame_factory, const std::vector<std::wstring>& params)\r
  66 {\r
  67         if(params.empty())\r
  68                 BOOST_THROW_EXCEPTION(invalid_argument() << arg_name_info("params") << arg_value_info(""));\r
  69         \r
  70         tbb::spin_rw_mutex::scoped_lock(p_factories_mutex, false);\r
  71         auto producer = frame_producer::empty();\r
  72         std::any_of(p_factories.begin(), p_factories.end(), [&](const producer_factory_t& factory) -> bool\r
  73                 {\r
  74                         try\r
  75                         {\r
  76                                 producer = factory(my_frame_factory, params);\r
  77                         }\r
  78                         catch(...)\r
  79                         {\r
  80                                 CASPAR_LOG_CURRENT_EXCEPTION();\r
  81                         }\r
  82                         return producer != frame_producer::empty();\r
  83                 });\r
  84 \r
  85         if(producer == frame_producer::empty())\r
  86                 producer = create_color_producer(my_frame_factory, params);\r
  87 \r
  88         if(producer == frame_producer::empty())\r
  89                 BOOST_THROW_EXCEPTION(file_not_found() << msg_info("No match found for supplied commands. Check syntax."));\r
  90 \r
  91         return producer;\r
  92 }\r
  93 \r
  94 }}