2 * copyright (c) 2010 Sveriges Television AB <info@casparcg.com>
\r
4 * This file is part of CasparCG.
\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
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
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
21 #include "..\..\stdafx.h"
\r
23 #if defined(_MSC_VER)
\r
24 #pragma warning (disable : 4714) // marked as __forceinline not inlined
\r
27 #include "flash_producer.h"
\r
28 #include "FlashAxContainer.h"
\r
29 #include "TimerHelper.h"
\r
32 #include "../../frame/frame_format.h"
\r
33 #include "../../../common/utility/find_file.h"
\r
34 #include "../../server.h"
\r
35 #include "../../../common/concurrency/executor.h"
\r
36 #include "../../../common/concurrency/function_task.h"
\r
37 #include "../../../common/utility/memory.h"
\r
38 #include "../../../common/utility/scope_exit.h"
\r
40 #include "../../frame/gpu_frame.h"
\r
41 #include "../../frame/composite_gpu_frame.h"
\r
43 #include <boost/assign.hpp>
\r
44 #include <boost/filesystem.hpp>
\r
45 #include <boost/thread.hpp>
\r
47 #include <tbb/atomic.h>
\r
48 #include <tbb/concurrent_queue.h>
\r
50 #include <type_traits>
\r
52 namespace caspar { namespace flash {
\r
54 using namespace boost::assign;
\r
56 // This is needed in order to make CComObject work since this is not a real ATL project
\r
57 CComModule _AtlModule;
\r
58 extern __declspec(selectany) CAtlModule* _pAtlModule = &_AtlModule;
\r
60 struct flash_producer::implementation
\r
62 implementation(flash_producer* self, const std::wstring& filename, const frame_format_desc& format_desc)
\r
63 : flashax_container_(nullptr), filename_(filename), self_(self), format_desc_(format_desc),
\r
64 bitmap_pool_(new bitmap_pool), executor_([=]{run();}), invalid_count_(0)
\r
66 if(!boost::filesystem::exists(filename))
\r
67 BOOST_THROW_EXCEPTION(file_not_found() << boost::errinfo_file_name(common::narrow(filename)));
\r
69 frame_buffer_.set_capacity(flash_producer::DEFAULT_BUFFER_SIZE);
\r
84 executor_.stop(); // Restart if running
\r
86 executor_.invoke([=]()
\r
88 if(FAILED(CComObject<FlashAxContainer>::CreateInstance(&flashax_container_)) ||
\r
89 flashax_container_ == nullptr)
\r
90 BOOST_THROW_EXCEPTION(caspar_exception() << msg_info("Failed to create FlashAxContainer"));
\r
92 flashax_container_->pflash_producer_ = self_;
\r
93 CComPtr<IShockwaveFlash> spFlash;
\r
95 if(FAILED(flashax_container_->CreateAxControl()))
\r
96 BOOST_THROW_EXCEPTION(caspar_exception() << msg_info("Failed to Create FlashAxControl"));
\r
98 if(FAILED(flashax_container_->QueryControl(&spFlash)))
\r
99 BOOST_THROW_EXCEPTION(caspar_exception() << msg_info("Failed to Query FlashAxControl"));
\r
101 if(FAILED(spFlash->put_Playing(true)) )
\r
102 BOOST_THROW_EXCEPTION(caspar_exception() << msg_info("Failed to start playing Flash"));
\r
104 if(FAILED(spFlash->put_Movie(CComBSTR(filename_.c_str()))))
\r
105 BOOST_THROW_EXCEPTION(caspar_exception() << msg_info("Failed to Load Template Host"));
\r
107 //Exact fit. Scale without respect to the aspect ratio.
\r
108 if(FAILED(spFlash->put_ScaleMode(2)))
\r
109 BOOST_THROW_EXCEPTION(caspar_exception() << msg_info("Failed to Set Scale Mode"));
\r
112 if(FAILED(flashax_container_->SetFormat(format_desc_)))
\r
113 BOOST_THROW_EXCEPTION(caspar_exception() << msg_info("Failed to Set Format"));
\r
115 current_frame_ = nullptr; // Force re-render of current frame
\r
121 CASPAR_LOG_CURRENT_EXCEPTION();
\r
129 if(executor_.is_running())
\r
131 frame_buffer_.clear();
\r
136 void param(const std::wstring& param)
\r
138 if(!executor_.is_running())
\r
144 catch(caspar_exception& e)
\r
146 e << msg_info("Flashproducer failed to recover from failure.");
\r
151 executor_.invoke([&]()
\r
153 for(size_t retries = 0; !flashax_container_->CallFunction(param); ++retries)
\r
155 CASPAR_LOG(debug) << "Retrying. Count: " << retries;
\r
157 BOOST_THROW_EXCEPTION(operation_failed() << warg_name_info(L"param") << warg_value_info(param));
\r
159 is_empty_ = false;
\r
165 win32_exception::install_handler();
\r
166 CASPAR_LOG(info) << L"started flash_producer Thread";
\r
170 ::OleInitialize(nullptr);
\r
171 CASPAR_SCOPE_EXIT(::OleUninitialize);
\r
173 CASPAR_SCOPE_EXIT([=]
\r
175 if(flashax_container_ != nullptr)
\r
177 flashax_container_->DestroyAxControl();
\r
178 flashax_container_->Release();
\r
179 flashax_container_ = nullptr;
\r
183 CASPAR_SCOPE_EXIT([=]
\r
187 frame_buffer_.clear();
\r
188 frame_buffer_.try_push(nullptr); // EOF
\r
190 current_frame_ = nullptr;
\r
193 while(executor_.is_running())
\r
198 while(executor_.try_execute()){}
\r
202 executor_.execute();
\r
203 while(executor_.try_execute()){}
\r
209 CASPAR_LOG_CURRENT_EXCEPTION();
\r
212 CASPAR_LOG(info) << L"Ended flash_producer Thread";
\r
217 assert(flashax_container_);
\r
219 if(!is_empty_ || current_frame_ == nullptr)
\r
221 if(!flashax_container_->IsReadyToRender())
\r
223 CASPAR_LOG(trace) << "Flash Producer Underflow";
\r
224 boost::thread::yield();
\r
228 bool isProgressive = format_desc_.mode == video_mode::progressive || (flashax_container_->GetFPS() - format_desc_.fps/2 == 0);
\r
230 gpu_frame_ptr result;
\r
233 result = render_frame();
\r
236 auto result = std::make_shared<composite_gpu_frame>(format_desc_.width, format_desc_.height);
\r
237 auto frame1 = render_frame();
\r
238 auto frame2 = render_frame();
\r
239 result->add(frame1);
\r
240 result->add(frame2);
\r
241 if(format_desc_.mode == video_mode::upper)
\r
243 frame1->mode(video_mode::upper);
\r
244 frame2->mode(video_mode::lower);
\r
248 frame1->mode(video_mode::lower);
\r
249 frame2->mode(video_mode::upper);
\r
253 frame_buffer_.push(result);
\r
254 is_empty_ = flashax_container_->IsEmpty();
\r
258 gpu_frame_ptr render_frame()
\r
260 flashax_container_->Tick();
\r
261 invalid_count_ = !flashax_container_->InvalidRectangle() ? std::min(2, invalid_count_+1) : 0;
\r
262 if(current_frame_ == nullptr || invalid_count_ < 2)
\r
265 if(!bitmap_pool_->try_pop(frame))
\r
267 CASPAR_LOG(trace) << "Allocated bitmap";
\r
268 frame = std::make_shared<bitmap>(format_desc_.width, format_desc_.height);
\r
269 common::clear(frame->data(), frame->size());
\r
271 flashax_container_->DrawControl(frame->hdc());
\r
273 auto pool = bitmap_pool_;
\r
274 current_frame_.reset(frame.get(), [=](bitmap*)
\r
276 common::function_task::enqueue([=]
\r
278 if(pool->try_push(frame))
\r
279 common::clear(frame->data(), frame->size());
\r
284 auto frame = factory_->create_frame(format_desc_);
\r
285 auto bitmap = render_frame();
\r
286 common::copy(frame->data(), current_frame_->data(), current_frame_->size());
\r
291 gpu_frame_ptr get_frame()
\r
293 if(!frame_buffer_.try_pop(last_frame_) && is_empty_)
\r
294 return gpu_frame::null();
\r
296 return last_frame_;
\r
299 void initialize(const frame_factory_ptr& factory)
\r
301 factory_ = factory;
\r
304 typedef tbb::concurrent_bounded_queue<bitmap_ptr> bitmap_pool;
\r
305 std::shared_ptr<bitmap_pool> bitmap_pool_;
\r
306 frame_format_desc format_desc_;
\r
308 CComObject<caspar::flash::FlashAxContainer>* flashax_container_;
\r
310 tbb::concurrent_bounded_queue<gpu_frame_ptr> frame_buffer_;
\r
311 gpu_frame_ptr last_frame_;
\r
312 bitmap_ptr current_frame_;
\r
314 std::wstring filename_;
\r
315 flash_producer* self_;
\r
317 tbb::atomic<bool> is_empty_;
\r
318 common::executor executor_;
\r
319 int invalid_count_;
\r
321 frame_factory_ptr factory_;
\r
324 flash_producer::flash_producer(const std::wstring& filename, const frame_format_desc& format_desc) : impl_(new implementation(this, filename, format_desc)){}
\r
325 gpu_frame_ptr flash_producer::get_frame(){return impl_->get_frame();}
\r
326 void flash_producer::param(const std::wstring& param){impl_->param(param);}
\r
327 const frame_format_desc& flash_producer::get_frame_format_desc() const { return impl_->format_desc_; }
\r
328 void flash_producer::initialize(const frame_factory_ptr& factory) { impl_->initialize(factory);}
\r
330 std::wstring flash_producer::find_template(const std::wstring& template_name)
\r
332 if(boost::filesystem::exists(template_name + TEXT(".ft")))
\r
333 return template_name + TEXT(".ft");
\r
335 if(boost::filesystem::exists(template_name + TEXT(".ct")))
\r
336 return template_name + TEXT(".ct");
\r
341 flash_producer_ptr create_flash_producer(const std::vector<std::wstring>& params, const frame_format_desc& format_desc)
\r
343 // TODO: Check for flash support
\r
344 auto filename = params[0];
\r
345 std::wstring result_filename = common::find_file(server::media_folder() + filename,
\r
348 return result_filename.empty() ? nullptr : std::make_shared<flash_producer>(result_filename, format_desc);
\r