X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=modules%2Fbluefish%2Fconsumer%2Fbluefish_consumer.cpp;h=77786ca62b24644b46316ba133e323c6ae2ffc23;hb=24af64b72bf309e46f808eabd7e46dc38f9bc222;hp=1b4a874bcbeb9646b4bed13b7993a3fa502c03aa;hpb=b0d570055364583701c46ce41fbdddf7d7780f2b;p=casparcg diff --git a/modules/bluefish/consumer/bluefish_consumer.cpp b/modules/bluefish/consumer/bluefish_consumer.cpp index 1b4a874bc..77786ca62 100644 --- a/modules/bluefish/consumer/bluefish_consumer.cpp +++ b/modules/bluefish/consumer/bluefish_consumer.cpp @@ -24,11 +24,12 @@ #include "../util/util.h" #include "../util/memory.h" -#include +#include #include #include #include +#include #include #include @@ -81,104 +82,89 @@ void blue_initialize() blue_velvet_initialize(); blue_hanc_initialize(); } + +safe_ptr create_blue(size_t device_index) +{ + if(!BlueVelvetFactory4 || !encode_hanc_frame || !encode_hanc_frame) + BOOST_THROW_EXCEPTION(caspar_exception() << msg_info("Bluefish drivers not found.")); + + auto blue = safe_ptr(BlueVelvetFactory4()); + + if(BLUE_FAIL(blue->device_attach(device_index, FALSE))) + BOOST_THROW_EXCEPTION(bluefish_exception() << msg_info("Failed to attach device.")); + + return blue; +} + +EVideoMode get_video_mode(CBlueVelvet4& blue, const core::video_format_desc& format_desc) +{ + EVideoMode vid_fmt = VID_FMT_INVALID; + auto desiredVideoFormat = vid_fmt_from_video_format(format_desc.format); + int videoModeCount = blue.count_video_mode(); + for(int videoModeIndex = 1; videoModeIndex <= videoModeCount; ++videoModeIndex) + { + EVideoMode videoMode = blue.enum_video_mode(videoModeIndex); + if(videoMode == desiredVideoFormat) + vid_fmt = videoMode; + } + if(vid_fmt == VID_FMT_INVALID) + BOOST_THROW_EXCEPTION(bluefish_exception() << msg_info("Failed get videomode.") << arg_value_info(narrow(format_desc.name))); + + return vid_fmt; +} -struct bluefish_consumer::implementation : boost::noncopyable +struct bluefish_consumer : boost::noncopyable { - std::wstring model_name_; - const unsigned int device_index_; + safe_ptr blue_; + const unsigned int device_index_; + const core::video_format_desc format_desc_; - std::shared_ptr graph_; - boost::timer perf_timer_; + const std::wstring model_name_; - boost::unique_future active_; - - std::shared_ptr blue_; - - core::video_format_desc format_desc_; + std::shared_ptr graph_; + boost::timer frame_timer_; + boost::timer tick_timer_; + boost::timer sync_timer_; - unsigned long mem_fmt_; - unsigned long upd_fmt_; - EVideoMode vid_fmt_; - unsigned long res_fmt_; - unsigned long engine_mode_; + const EVideoMode vid_fmt_; + const EMemoryFormat mem_fmt_; + const EUpdateMethod upd_fmt_; + const EResoFormat res_fmt_; + EEngineMode engine_mode_; - std::array reserved_frames_; + std::array reserved_frames_; + tbb::concurrent_bounded_queue> frame_buffer_; - const bool embed_audio_; + int preroll_count_; + const bool embedded_audio_; + executor executor_; public: - implementation::implementation(unsigned int device_index, bool embed_audio) - : model_name_(L"BLUEFISH") - , device_index_(device_index) + bluefish_consumer(const core::video_format_desc& format_desc, unsigned int device_index, bool embedded_audio) + : blue_(create_blue(device_index)) + , device_index_(device_index) + , format_desc_(format_desc) + , model_name_(get_card_desc(blue_->has_video_cardtype())) + , vid_fmt_(get_video_mode(*blue_, format_desc)) , mem_fmt_(MEM_FMT_ARGB_PC) , upd_fmt_(UPD_FMT_FRAME) - , vid_fmt_(VID_FMT_INVALID) , res_fmt_(RES_FMT_NORMAL) - , engine_mode_(VIDEO_ENGINE_FRAMESTORE) - , embed_audio_(embed_audio) + , engine_mode_(VIDEO_ENGINE_FRAMESTORE) + , preroll_count_(0) + , embedded_audio_(embedded_audio) , executor_(print()) { - if(!BlueVelvetFactory4 || (embed_audio_ && (!encode_hanc_frame || !encode_hanc_frame))) - BOOST_THROW_EXCEPTION(bluefish_exception() << msg_info("Bluefish drivers not found.")); - } - - ~implementation() - { - if(executor_.is_running()) - { - executor_.invoke([&] - { - disable_video_output(); - - if(blue_) - blue_->device_detach(); - }); - } - - CASPAR_LOG(info) << print() << L" Shutting down."; - } - - void initialize(const core::video_format_desc& format_desc) - { - format_desc_ = format_desc; - - blue_.reset(BlueVelvetFactory4()); - - if(BLUE_FAIL(blue_->device_attach(device_index_, FALSE))) - BOOST_THROW_EXCEPTION(bluefish_exception() << msg_info(narrow(print()) + " Failed to attach device.")); - - int videoCardType = blue_->has_video_cardtype(); - model_name_ = get_card_desc(videoCardType); + executor_.set_capacity(CONSUMER_BUFFER_DEPTH); graph_ = diagnostics::create_graph(narrow(print())); graph_->add_guide("tick-time", 0.5); graph_->set_color("tick-time", diagnostics::color(0.1f, 0.7f, 0.8f)); + graph_->add_guide("frame-time", 0.5f); + graph_->set_color("frame-time", diagnostics::color(1.0f, 0.0f, 0.0f)); + graph_->set_color("sync-time", diagnostics::color(0.5f, 1.0f, 0.2f)); + graph_->set_color("input-buffer", diagnostics::color(1.0f, 1.0f, 0.0f)); - //void* pBlueDevice = blue_attach_to_device(1); - //EBlueConnectorPropertySetting video_routing[1]; - //auto channel = BLUE_VIDEO_OUTPUT_CHANNEL_A; - //video_routing[0].channel = channel; - //video_routing[0].propType = BLUE_CONNECTOR_PROP_SINGLE_LINK; - //video_routing[0].connector = channel == BLUE_VIDEO_OUTPUT_CHANNEL_A ? BLUE_CONNECTOR_SDI_OUTPUT_A : BLUE_CONNECTOR_SDI_OUTPUT_B; - //blue_set_connector_property(pBlueDevice, 1, video_routing); - //blue_detach_from_device(&pBlueDevice); - - auto desiredVideoFormat = vid_fmt_from_video_format(format_desc_.format); - int videoModeCount = blue_->count_video_mode(); - for(int videoModeIndex = 1; videoModeIndex <= videoModeCount; ++videoModeIndex) - { - EVideoMode videoMode = blue_->enum_video_mode(videoModeIndex); - if(videoMode == desiredVideoFormat) - vid_fmt_ = videoMode; - } - if(vid_fmt_ == VID_FMT_INVALID) - BOOST_THROW_EXCEPTION(bluefish_exception() << msg_info(narrow(print()) + " Failed to set videomode.")); - - // Set default video output channel - //if(BLUE_FAIL(set_card_property(blue_, DEFAULT_VIDEO_OUTPUT_CHANNEL, channel))) - // CASPAR_LOG(error) << TEXT("BLUECARD ERROR: Failed to set default channel. (device ") << device_index_ << TEXT(")"); - //Setting output Video mode if(BLUE_FAIL(set_card_property(blue_, VIDEO_MODE, vid_fmt_))) BOOST_THROW_EXCEPTION(bluefish_exception() << msg_info(narrow(print()) + " Failed to set videomode.")); @@ -211,7 +197,7 @@ public: if(BLUE_FAIL(set_card_property(blue_, VIDEO_PREDEFINED_COLOR_MATRIX, vid_fmt_ == VID_FMT_PAL ? MATRIX_601_CGR : MATRIX_709_CGR))) CASPAR_LOG(warning) << print() << TEXT(" Failed to set colormatrix to ") << (vid_fmt_ == VID_FMT_PAL ? TEXT("601 CGR") : TEXT("709 CGR")) << TEXT("."); - if(!embed_audio_) + if(!embedded_audio_) { if(BLUE_FAIL(set_card_property(blue_, EMBEDEDDED_AUDIO_OUTPUT, 0))) CASPAR_LOG(warning) << TEXT("BLUECARD ERROR: Failed to disable embedded audio."); @@ -233,20 +219,33 @@ public: if(blue_->GetHDCardType(device_index_) != CRD_HD_INVALID) blue_->Set_DownConverterSignalType(vid_fmt_ == VID_FMT_PAL ? SD_SDI : HD_SDI); - if(BLUE_FAIL(blue_->set_video_engine(engine_mode_))) + if(BLUE_FAIL(blue_->set_video_engine(*reinterpret_cast(&engine_mode_)))) BOOST_THROW_EXCEPTION(bluefish_exception() << msg_info(narrow(print()) + " Failed to set video engine.")); enable_video_output(); for(size_t n = 0; n < reserved_frames_.size(); ++n) - reserved_frames_[n] = std::make_shared(format_desc_.size, n); - - executor_.start(); - active_ = executor_.begin_invoke([]{}); - - CASPAR_LOG(info) << print() << TEXT(" Successfully initialized for ") << format_desc_ << TEXT("."); + reserved_frames_[n] = std::make_shared(format_desc_.size, n); + + CASPAR_LOG(info) << print() << L" Successfully Initialized."; } + + ~bluefish_consumer() + { + executor_.invoke([&] + { + disable_video_output(); + blue_->device_detach(); + }); + CASPAR_LOG(info) << print() << L" Shutting down."; + } + + const core::video_format_desc& get_video_format_desc() const + { + return format_desc_; + } + void enable_video_output() { if(!BLUE_PASS(set_card_property(blue_, VIDEO_BLACKGENERATOR, 0))) @@ -258,22 +257,42 @@ public: if(!BLUE_PASS(set_card_property(blue_, VIDEO_BLACKGENERATOR, 1))) CASPAR_LOG(error)<< print() << TEXT(" Failed to disable video output."); } - - virtual void send(const safe_ptr& frame) - { - static std::vector silence(MAX_HANC_BUFFER_SIZE, 0); + + void send(const safe_ptr& frame) + { + if(preroll_count_ < executor_.capacity()) + { + while(preroll_count_++ < executor_.capacity()) + schedule_next_video(make_safe()); + } - size_t audio_samples = static_cast(48000.0 / format_desc_.fps); - size_t audio_nchannels = 2; + schedule_next_video(frame); + } + + void schedule_next_video(const safe_ptr& frame) + { + static std::vector silence(MAX_HANC_BUFFER_SIZE, 0); - active_.get(); - active_ = executor_.begin_invoke([=] + executor_.begin_invoke([=] { try { - fast_memcpy(reserved_frames_.front()->image_data(), frame->image_data().begin(), frame->image_data().size()); + const size_t audio_samples = format_desc_.audio_samples_per_frame; + const size_t audio_nchannels = format_desc_.audio_channels; + + frame_timer_.restart(); - if(embed_audio_) + if(!frame->image_data().empty()) + fast_memcpy(reserved_frames_.front()->image_data(), frame->image_data().begin(), frame->image_data().size()); + else + fast_memclr(reserved_frames_.front()->image_data(), reserved_frames_.front()->image_size()); + + sync_timer_.restart(); + unsigned long n_field = 0; + blue_->wait_output_video_synch(UPD_FMT_FRAME, n_field); + graph_->update_value("sync-time", static_cast(sync_timer_.elapsed()*format_desc_.fps*0.5)); + + if(embedded_audio_) { auto frame_audio_data = frame->audio_data().empty() ? silence.data() : const_cast(frame->audio_data().begin()); @@ -303,22 +322,22 @@ public: CASPAR_LOG(warning) << print() << TEXT(" render_buffer_update failed."); } - unsigned long n_field = 0; - blue_->wait_output_video_synch(UPD_FMT_FRAME, n_field); - std::rotate(reserved_frames_.begin(), reserved_frames_.begin() + 1, reserved_frames_.end()); - graph_->update_value("tick-time", static_cast(perf_timer_.elapsed()/format_desc_.interval*0.5)); - perf_timer_.restart(); + + graph_->update_value("frame-time", static_cast(frame_timer_.elapsed()*format_desc_.fps*0.5)); + + graph_->update_value("tick-time", static_cast(tick_timer_.elapsed()*format_desc_.fps*0.5)); + tick_timer_.restart(); } catch(...) { CASPAR_LOG_CURRENT_EXCEPTION(); } + graph_->set_value("input-buffer", static_cast(executor_.size())/static_cast(executor_.capacity())); }); + graph_->set_value("input-buffer", static_cast(executor_.size())/static_cast(executor_.capacity())); } - virtual size_t buffer_depth() const{return 1;} - void encode_hanc(BLUE_UINT32* hanc_data, void* audio_data, size_t audio_samples, size_t audio_nchannels) { auto card_type = blue_->has_video_cardtype(); @@ -344,21 +363,48 @@ public: std::wstring print() const { - return model_name_ + L" [" + boost::lexical_cast(device_index_) + L"]"; + return model_name_ + L" [" + boost::lexical_cast(device_index_) + L"|" + format_desc_.name + L"]"; } }; -bluefish_consumer::bluefish_consumer(unsigned int device_index, bool embed_audio) : impl_(new implementation(device_index, embed_audio)){} -bluefish_consumer::bluefish_consumer(bluefish_consumer&& other) : impl_(std::move(other.impl_)){} -void bluefish_consumer::initialize(const core::video_format_desc& format_desc) +struct bluefish_consumer_proxy : public core::frame_consumer { - // TODO: Ugly - impl_.reset(new implementation(impl_->device_index_, impl_->embed_audio_)); - impl_->initialize(format_desc); -} -void bluefish_consumer::send(const safe_ptr& frame){impl_->send(frame);} -size_t bluefish_consumer::buffer_depth() const{return impl_->buffer_depth();} -std::wstring bluefish_consumer::print() const {return impl_->print();} + std::unique_ptr consumer_; + const size_t device_index_; + const bool embedded_audio_; + bool key_only_; +public: + + bluefish_consumer_proxy(size_t device_index, bool embedded_audio, bool key_only) + : device_index_(device_index) + , embedded_audio_(embedded_audio) + , key_only_(key_only){} + + virtual void initialize(const core::video_format_desc& format_desc) + { + consumer_.reset(new bluefish_consumer(format_desc, device_index_, embedded_audio_)); + } + + virtual void send(const safe_ptr& frame) + { + consumer_->send(frame); + } + + virtual const core::video_format_desc& get_video_format_desc() const + { + return consumer_->get_video_format_desc(); + } + + virtual std::wstring print() const + { + return consumer_->print(); + } + + virtual bool key_only() const + { + return key_only_; + } +}; std::wstring get_bluefish_version() { @@ -389,7 +435,7 @@ std::vector get_bluefish_device_list() for(int n = 1; BLUE_PASS(blue->device_attach(n, FALSE)); ++n) { - devices.push_back(L"[" + boost::lexical_cast(n) + L"] " + get_card_desc(blue->has_video_cardtype())); + devices.push_back(std::wstring(get_card_desc(blue->has_video_cardtype())) + L" [" + boost::lexical_cast(n) + L"]"); blue->device_detach(); } } @@ -404,15 +450,22 @@ safe_ptr create_bluefish_consumer(const std::vector 1) - device_index = lexical_cast_or_default(params[2]); + device_index = lexical_cast_or_default(params[1], 1); + + bool embedded_audio = std::find(params.begin(), params.end(), L"EMBEDDED_AUDIO") != params.end(); + bool key_only = std::find(params.begin(), params.end(), L"KEY_ONLY") != params.end(); + + return make_safe(device_index, embedded_audio, key_only); +} - if(params.size() > 2) - embed_audio = lexical_cast_or_default(params[3]); +safe_ptr create_bluefish_consumer(const boost::property_tree::ptree& ptree) +{ + auto device_index = ptree.get("device", 1); + auto embedded_audio = ptree.get("embedded-audio", false); + bool key_only = ptree.get("key-only", false); - return make_safe(device_index, embed_audio); + return make_safe(device_index, embedded_audio, key_only); } } \ No newline at end of file