X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=core%2Fmixer%2Fimage%2Fimage_mixer.cpp;h=941e28d6c58fbbe18dd1c16af9671678bc07b915;hb=a486c25d5e6ce0ebe08e9a2d793a447ff3cb797a;hp=f7e403dd35e1167f35510562261af1cf193cb229;hpb=17a8853f08f9744a5cd0ac34053d1138824260ba;p=casparcg diff --git a/core/mixer/image/image_mixer.cpp b/core/mixer/image/image_mixer.cpp index f7e403dd3..941e28d6c 100644 --- a/core/mixer/image/image_mixer.cpp +++ b/core/mixer/image/image_mixer.cpp @@ -20,201 +20,225 @@ #include "../../stdafx.h" #include "image_mixer.h" -#include "image_kernel.h" +#include "image_kernel.h" +#include "../write_frame.h" #include "../gpu/ogl_device.h" #include "../gpu/host_buffer.h" #include "../gpu/device_buffer.h" -#include "../write_frame.h" - #include "../../video_channel_context.h" -#include #include #include +#include -#include +#include #include #include +#include + #include -#include #include -#include -#include +#include -namespace caspar { namespace core { - -struct image_mixer::implementation : boost::noncopyable -{ - typedef std::deque layer; +using namespace boost::assign; - video_channel_context& channel_; +namespace caspar { namespace core { + +struct layer +{ + std::vector items; + blend_mode::type blend_mode; - std::vector transform_stack_; - std::vector mode_stack_; + layer(blend_mode::type blend_mode) : blend_mode(blend_mode) + { + } +}; - std::queue> layers_; // layer/stream/items +class image_renderer +{ + video_channel_context& channel_; + image_kernel kernel_; + std::shared_ptr active_buffer_; +public: + image_renderer(video_channel_context& channel) + : channel_(channel) + { + } - image_kernel kernel_; - - std::array,2> draw_buffer_; - std::shared_ptr write_buffer_; - - std::array,2> stream_key_buffer_; - std::shared_ptr layer_key_buffer_; + boost::unique_future> render(std::vector&& layers) + { + auto layers2 = make_move_on_copy(std::move(layers)); + return channel_.ogl().begin_invoke([=] + { + return do_render(std::move(layers2.value)); + }); + } -public: - implementation(video_channel_context& video_channel) - : channel_(video_channel) - , transform_stack_(1) - , mode_stack_(1, video_mode::progressive) +private: + safe_ptr do_render(std::vector&& layers) { - initialize_buffers(); + std::shared_ptr layer_key_buffer; + + auto draw_buffer = create_device_buffer(4); + + BOOST_FOREACH(auto& layer, layers) + draw_layer(std::move(layer), draw_buffer, layer_key_buffer); + + auto host_buffer = channel_.ogl().create_host_buffer(channel_.get_format_desc().size, host_buffer::read_only); + channel_.ogl().attach(*draw_buffer); + host_buffer->begin_read(draw_buffer->width(), draw_buffer->height(), format(draw_buffer->stride())); + + active_buffer_ = draw_buffer; + + channel_.ogl().flush(); // NOTE: This is important, otherwise fences will deadlock. + + return host_buffer; + } + + void draw_layer(layer&& layer, const safe_ptr& draw_buffer, std::shared_ptr& layer_key_buffer) + { + if(layer.items.empty()) + return; + + std::pair> local_key_buffer = std::make_pair(0, nullptr); // int is fields flag + + if(layer.blend_mode != blend_mode::normal && has_overlapping_items(layer)) + { + auto layer_draw_buffer = create_device_buffer(4); // int is fields flag + + BOOST_FOREACH(auto& item, layer.items) + draw_item(std::move(item), *layer_draw_buffer, local_key_buffer, layer_key_buffer); + + render_item item; + item.pix_desc.pix_fmt = pixel_format::bgra; + item.pix_desc.planes = list_of(pixel_format_desc::plane(channel_.get_format_desc().width, channel_.get_format_desc().height, 4)); + item.textures = list_of(layer_draw_buffer); + item.transform = frame_transform(); + item.blend_mode = layer.blend_mode; + + kernel_.draw(channel_.ogl(), std::move(item), *draw_buffer, nullptr, nullptr); + } + else // fast path + { + BOOST_FOREACH(auto& item, layer.items) + draw_item(std::move(item), *draw_buffer, local_key_buffer, layer_key_buffer); + } + + CASPAR_ASSERT(local_key_buffer.first == 0 || local_key_buffer.first == core::field_mode::progressive); + + std::swap(local_key_buffer.second, layer_key_buffer); } - ~implementation() + void draw_item(render_item&& item, + device_buffer& draw_buffer, + std::pair>& local_key_buffer, + std::shared_ptr& layer_key_buffer) + { + if(item.transform.is_key) + { + if(!local_key_buffer.second) + { + local_key_buffer.first = 0; + local_key_buffer.second = create_device_buffer(1); + } + + local_key_buffer.first |= item.transform.field_mode; // Add field to flag. + kernel_.draw(channel_.ogl(), std::move(item), *local_key_buffer.second, nullptr, nullptr); + } + else + { + kernel_.draw(channel_.ogl(), std::move(item), draw_buffer, local_key_buffer.second, layer_key_buffer); + local_key_buffer.first ^= item.transform.field_mode; // Remove field from flag. + + if(local_key_buffer.first == 0) // If all fields from key has been used, reset it + { + local_key_buffer.first = 0; + local_key_buffer.second.reset(); + } + } + } + + bool has_overlapping_items(const layer& layer) + { + auto upper_count = boost::range::count_if(layer.items, [&](const render_item& item) + { + return !item.transform.is_key && (item.transform.field_mode & field_mode::upper); + }); + + auto lower_count = boost::range::count_if(layer.items, [&](const render_item& item) + { + return !item.transform.is_key && (item.transform.field_mode & field_mode::lower); + }); + + return upper_count > 1 || lower_count > 1; + } + + safe_ptr create_device_buffer(size_t stride) { - channel_.ogl().gc(); + auto buffer = channel_.ogl().create_device_buffer(channel_.get_format_desc().width, channel_.get_format_desc().height, stride); + channel_.ogl().clear(*buffer); + return buffer; } +}; - void initialize_buffers() + +struct image_mixer::implementation : boost::noncopyable +{ + ogl_device& ogl_; + image_renderer renderer_; + std::vector transform_stack_; + std::vector layers_; // layer/stream/items +public: + implementation(video_channel_context& video_channel) + : ogl_(video_channel.ogl()) + , renderer_(video_channel) + , transform_stack_(1) { - write_buffer_ = channel_.ogl().create_device_buffer(channel_.get_format_desc().width, channel_.get_format_desc().height, 4); - layer_key_buffer_ = channel_.ogl().create_device_buffer(channel_.get_format_desc().width, channel_.get_format_desc().height, 1); - draw_buffer_[0] = channel_.ogl().create_device_buffer(channel_.get_format_desc().width, channel_.get_format_desc().height, 4); - draw_buffer_[1] = channel_.ogl().create_device_buffer(channel_.get_format_desc().width, channel_.get_format_desc().height, 4); - stream_key_buffer_[0] = channel_.ogl().create_device_buffer(channel_.get_format_desc().width, channel_.get_format_desc().height, 1); - stream_key_buffer_[1] = channel_.ogl().create_device_buffer(channel_.get_format_desc().width, channel_.get_format_desc().height, 1); - channel_.ogl().gc(); } + void begin_layer(blend_mode::type blend_mode) + { + layers_.push_back(layer(blend_mode)); + } + void begin(core::basic_frame& frame) { - transform_stack_.push_back(transform_stack_.back()*frame.get_image_transform()); - mode_stack_.push_back(frame.get_mode() == video_mode::progressive ? mode_stack_.back() : frame.get_mode()); + transform_stack_.push_back(transform_stack_.back()*frame.get_frame_transform()); } void visit(core::write_frame& frame) { - CASPAR_ASSERT(!layers_.empty()); - - // Check if frame has been discarded by interlacing - if(boost::range::find(mode_stack_, video_mode::upper) != mode_stack_.end() && boost::range::find(mode_stack_, video_mode::lower) != mode_stack_.end()) + if(transform_stack_.back().field_mode == field_mode::empty) return; - core::render_item item = {frame.get_pixel_format_desc(), frame.get_textures(), transform_stack_.back(), mode_stack_.back(), frame.tag()}; - - auto& layer = layers_.back(); + core::render_item item; + item.pix_desc = frame.get_pixel_format_desc(); + item.textures = frame.get_textures(); + item.transform = transform_stack_.back(); - auto it = boost::range::find(layer, item); - if(it == layer.end()) - layer.push_back(item); + layers_.back().items.push_back(item); } void end() { transform_stack_.pop_back(); - mode_stack_.pop_back(); - } - - void begin_layer() - { - layers_.push(layer()); } void end_layer() - { - } - - boost::unique_future> render() { - auto layers = std::move(layers_); - - return channel_.ogl().begin_invoke([=]() mutable -> safe_ptr - { - if(channel_.get_format_desc().width != write_buffer_->width() || channel_.get_format_desc().height != write_buffer_->height()) - initialize_buffers(); - - return do_render(std::move(layers)); - }); } - safe_ptr do_render(std::queue&& layers) + boost::unique_future> render() { - auto read_buffer = channel_.ogl().create_host_buffer(channel_.get_format_desc().size, host_buffer::read_only); - - layer_key_buffer_->clear(); - draw_buffer_[0]->clear(); - draw_buffer_[1]->clear(); - stream_key_buffer_[0]->clear(); - stream_key_buffer_[1]->clear(); - - bool local_key = false; - bool layer_key = false; - - while(!layers.empty()) - { - stream_key_buffer_[0]->clear(); - - auto layer = std::move(layers.front()); - layers.pop(); - - while(!layer.empty()) - { - auto item = std::move(layer.front()); - layer.pop_front(); - - if(item.transform.get_is_key()) - { - render_item(stream_key_buffer_, std::move(item), nullptr, nullptr); - local_key = true; - } - else - { - render_item(draw_buffer_, std::move(item), local_key ? stream_key_buffer_[0] : nullptr, layer_key ? layer_key_buffer_ : nullptr); - stream_key_buffer_[0]->clear(); - local_key = false; - } - channel_.ogl().yield(); // Return resources to pool as early as possible. - } - - layer_key = local_key; - local_key = false; - std::swap(stream_key_buffer_[0], layer_key_buffer_); - } - - std::swap(draw_buffer_[0], write_buffer_); - - // device -> host. - read_buffer->begin_read(*write_buffer_); - - return read_buffer; + return renderer_.render(std::move(layers_)); } - - void render_item(std::array,2>& targets, render_item&& item, const std::shared_ptr& local_key, const std::shared_ptr& layer_key) - { - if(!std::all_of(item.textures.begin(), item.textures.end(), std::mem_fn(&device_buffer::ready))) - { - CASPAR_LOG(warning) << L"[image_mixer] Performance warning. Host to device transfer not complete, GPU will be stalled"; - channel_.ogl().yield(); // Try to give it some more time. - } - targets[1]->attach(); - - kernel_.draw(item, make_safe(targets[0]), local_key, layer_key); - - targets[0]->bind(); - - glCopyTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, 0, 0, targets[0]->width(), targets[0]->height()); - - std::swap(targets[0], targets[1]); - } - safe_ptr create_frame(const void* tag, const core::pixel_format_desc& desc) { - return make_safe(channel_.ogl(), tag, desc); + return make_safe(ogl_, tag, desc); } }; @@ -224,7 +248,7 @@ void image_mixer::visit(core::write_frame& frame){impl_->visit(frame);} void image_mixer::end(){impl_->end();} boost::unique_future> image_mixer::render(){return impl_->render();} safe_ptr image_mixer::create_frame(const void* tag, const core::pixel_format_desc& desc){return impl_->create_frame(tag, desc);} -void image_mixer::begin_layer(){impl_->begin_layer();} +void image_mixer::begin_layer(blend_mode::type blend_mode){impl_->begin_layer(blend_mode);} void image_mixer::end_layer(){impl_->end_layer();} image_mixer& image_mixer::operator=(image_mixer&& other) {