X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=core%2Fmixer%2Fimage%2Fimage_mixer.cpp;h=393c8a7140f6383df22c486dadb3951f72bc459d;hb=116e09a218cf056e4de868454b0fe26dc4db1413;hp=95130999120835fcfe5ca5fde6017ebf1e77fb13;hpb=59ccfd709a69e69226f014a2971e68dd93da1b99;p=casparcg

diff --git a/core/mixer/image/image_mixer.cpp b/core/mixer/image/image_mixer.cpp
index 951309991..393c8a714 100644
--- a/core/mixer/image/image_mixer.cpp
+++ b/core/mixer/image/image_mixer.cpp
@@ -20,198 +20,278 @@
 #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 <common/concurrency/executor.h>
 #include <common/exception/exceptions.h>
 #include <common/gl/gl_check.h>
+#include <common/utility/move_on_copy.h>
 
-#include <core/producer/frame/image_transform.h>
+#include <core/producer/frame/frame_transform.h>
 #include <core/producer/frame/pixel_format.h>
 #include <core/video_format.h>
 
+#include <gl/glew.h>
+
 #include <boost/foreach.hpp>
-#include <boost/range.hpp>
+#include <boost/range/algorithm_ext/erase.hpp>
 
 #include <algorithm>
-#include <array>
-#include <unordered_map>
+#include <deque>
 
-namespace caspar { namespace core {
-		
-struct image_mixer::implementation : boost::noncopyable
-{		
-	typedef std::deque<render_item>					layer;
-
-	video_channel_context&							channel_;
+using namespace boost::assign;
 
-	std::vector<image_transform>					transform_stack_;
-	std::vector<video_mode::type>					mode_stack_;
-
-	std::deque<std::deque<render_item>>				layers_; // layer/stream/items
+namespace caspar { namespace core {
 	
-	image_kernel									kernel_;
-		
-	std::array<std::shared_ptr<device_buffer>,2>	draw_buffer_;
-	std::shared_ptr<device_buffer>					write_buffer_;
+struct item
+{
+	pixel_format_desc						pix_desc;
+	std::vector<safe_ptr<device_buffer>>	textures;
+	frame_transform							transform;
+};
 
-	std::array<std::shared_ptr<device_buffer>,2>	local_key_buffer_;
-	std::shared_ptr<device_buffer>					layer_key_buffer_;
-	
-public:
-	implementation(video_channel_context& video_channel) 
-		: channel_(video_channel)
-		, transform_stack_(1)
-		, mode_stack_(1, video_mode::progressive)
-	{
-		initialize_buffers();
-	}
+struct layer
+{
+	std::vector<item>						items;
+	blend_mode::type						blend_mode;
 
-	~implementation()
+	layer(blend_mode::type blend_mode) : blend_mode(blend_mode)
 	{
-		channel_.ogl().gc();
 	}
+};
 
-	void initialize_buffers()
+class image_renderer
+{
+	video_channel_context&					channel_;	
+	image_kernel							kernel_;	
+	std::shared_ptr<device_buffer>			active_buffer_;
+public:
+	image_renderer(video_channel_context& channel)
+		: channel_(channel)
 	{
-		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);
-		local_key_buffer_[0]	= channel_.ogl().create_device_buffer(channel_.get_format_desc().width, channel_.get_format_desc().height, 1);
-		local_key_buffer_[1]	= channel_.ogl().create_device_buffer(channel_.get_format_desc().width, channel_.get_format_desc().height, 1);
-		channel_.ogl().gc();
 	}
-
-	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());
+	
+	boost::unique_future<safe_ptr<host_buffer>> render(std::vector<layer>&& layers)
+	{		
+		auto layers2 = make_move_on_copy(std::move(layers));
+		return channel_.ogl().begin_invoke([=]
+		{
+			return do_render(std::move(layers2.value));
+		});
 	}
-		
-	void visit(core::write_frame& frame)
-	{	
-		CASPAR_ASSERT(!layers_.empty());
+	
+private:
+	safe_ptr<host_buffer> do_render(std::vector<layer>&& layers)
+	{
+		auto draw_buffer = create_device_buffer(4);
 
-		// 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())
-			return;
-		
-		core::render_item item = {frame.get_pixel_format_desc(), frame.get_textures(), transform_stack_.back(), mode_stack_.back(), frame.tag()};	
+		if(channel_.get_format_desc().field_mode != field_mode::progressive)
+		{
+			auto upper = layers;
+			auto lower = std::move(layers);
+
+			BOOST_FOREACH(auto& layer, upper)
+			{
+				boost::remove_erase_if(layer.items, [](const item& item){return !(item.transform.field_mode & field_mode::upper);});
+				BOOST_FOREACH(auto& item, layer.items)
+					item.transform.field_mode = field_mode::upper;
+			}
+
+			BOOST_FOREACH(auto& layer, lower)
+			{
+				boost::remove_erase_if(layer.items, [](const item& item){return !(item.transform.field_mode & field_mode::lower);});
+				BOOST_FOREACH(auto& item, layer.items)
+					item.transform.field_mode = field_mode::lower;
+			}
+
+			draw(std::move(upper), draw_buffer);
+			draw(std::move(lower), draw_buffer);
+		}
+		else
+		{
+			draw(std::move(layers), draw_buffer);
+		}
 
-		auto& layer = layers_.back();
+		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;
 
-		auto it = boost::range::find(layer, item);
-		if(it == layer.end())
-			layer.push_back(item);
+		channel_.ogl().flush(); // NOTE: This is important, otherwise fences will deadlock.
+			
+		return host_buffer;
 	}
 
-	void end()
+	void draw(std::vector<layer>&&		layers, 
+			  safe_ptr<device_buffer>&	draw_buffer)
 	{
-		transform_stack_.pop_back();
-		mode_stack_.pop_back();
-	}
+		std::shared_ptr<device_buffer> layer_key_buffer;
 
-	void begin_layer()
-	{
-		layers_.push_back(layer());
+		BOOST_FOREACH(auto& layer, layers)
+			draw_layer(std::move(layer), draw_buffer, layer_key_buffer);
 	}
 
-	void end_layer()
-	{
-	}
-	
-	boost::unique_future<safe_ptr<host_buffer>> render()
-	{		
-		auto layers = std::move(layers_);
-		return channel_.ogl().begin_invoke([=]()mutable{return render(std::move(layers));});
-	}
-	
-	safe_ptr<host_buffer> render(std::deque<layer>&& layers)
-	{
-		if(channel_.get_format_desc().width != write_buffer_->width() || channel_.get_format_desc().height != write_buffer_->height())
-			initialize_buffers();
+	void draw_layer(layer&&							layer, 
+					safe_ptr<device_buffer>&		draw_buffer,
+					std::shared_ptr<device_buffer>& layer_key_buffer)
+	{				
+		if(layer.items.empty())
+			return;
 
-		auto read_buffer = channel_.ogl().create_host_buffer(channel_.get_format_desc().size, host_buffer::read_only);
+		std::shared_ptr<device_buffer> local_key_buffer;
+		std::shared_ptr<device_buffer> local_mix_buffer;
+				
+		if(layer.blend_mode != blend_mode::normal && layer.items.size() > 1)
+		{
+			auto layer_draw_buffer = create_device_buffer(4);
 
-		layer_key_buffer_->clear();
-		draw_buffer_[0]->clear();
-		draw_buffer_[1]->clear();
-		local_key_buffer_[0]->clear();
-		local_key_buffer_[1]->clear();
+			BOOST_FOREACH(auto& item, layer.items)
+				draw_item(std::move(item), layer_draw_buffer, layer_key_buffer, local_key_buffer, local_mix_buffer);	
+		
+			draw_device_buffer(layer_draw_buffer, std::move(local_mix_buffer), blend_mode::normal);							
+			draw_device_buffer(draw_buffer, std::move(layer_draw_buffer), layer.blend_mode);
+		}
+		else // fast path
+		{
+			BOOST_FOREACH(auto& item, layer.items)		
+				draw_item(std::move(item), draw_buffer, layer_key_buffer, local_key_buffer, local_mix_buffer);		
+					
+			draw_device_buffer(draw_buffer, std::move(local_mix_buffer), blend_mode::normal);
+		}					
 
-		bool local_key = false;
-		bool layer_key = false;
+		std::swap(local_key_buffer, layer_key_buffer);
+	}
 
-		BOOST_FOREACH(auto& layer, layers)
-			draw(std::move(layer), local_key, layer_key);
+	void draw_item(item&&							item, 
+				   safe_ptr<device_buffer>&			draw_buffer, 
+				   std::shared_ptr<device_buffer>&	layer_key_buffer, 
+				   std::shared_ptr<device_buffer>&	local_key_buffer, 
+				   std::shared_ptr<device_buffer>&	local_mix_buffer)
+	{			
+		draw_params draw_params;
+		draw_params.pix_desc				= std::move(item.pix_desc);
+		draw_params.textures				= std::move(item.textures);
+		draw_params.transform				= std::move(item.transform);
+		draw_params.blend_mode				= blend_mode::normal;
 
-		std::swap(draw_buffer_[0], write_buffer_);
+		if(item.transform.is_key)
+		{
+			local_key_buffer = local_key_buffer ? local_key_buffer : create_device_buffer(1);
 
-		// device -> host.			
-		read_buffer->begin_read(*write_buffer_);
+			draw_params.background			= local_key_buffer;
+			draw_params.local_key			= nullptr;
+			draw_params.layer_key			= nullptr;
 
-		return read_buffer;
-	}
+			kernel_.draw(channel_.ogl(), std::move(draw_params));
+		}
+		else if(item.transform.is_mix)
+		{
+			local_mix_buffer = local_mix_buffer ? local_mix_buffer : create_device_buffer(4);
 
-	void draw(layer&& layer, bool& local_key, bool& layer_key)
-	{			
-		local_key_buffer_[0]->clear();
+			draw_params.background			= local_mix_buffer;
+			draw_params.local_key			= std::move(local_key_buffer);
+			draw_params.layer_key			= layer_key_buffer;
 
-		BOOST_FOREACH(auto& item, layer)
-			draw(std::move(item), local_key, layer_key);
-		
-		layer_key = local_key;
-		local_key = false;
-		std::swap(local_key_buffer_[0], layer_key_buffer_);
-	}
+			draw_params.blend_mode			= blend_mode::mix;
 
-	void draw(render_item&& item, bool& local_key, bool& layer_key)
-	{											
-		if(item.transform.get_is_key())
-		{
-			draw(local_key_buffer_, std::move(item), nullptr, nullptr);
-			local_key = true;
+			kernel_.draw(channel_.ogl(), std::move(draw_params));
 		}
 		else
 		{
-			draw(draw_buffer_, std::move(item), local_key ? local_key_buffer_[0] : nullptr, layer_key ? layer_key_buffer_ : nullptr);	
-			local_key_buffer_[0]->clear();
-			local_key = false;
-		}
-		channel_.ogl().yield(); // Return resources to pool as early as possible.
+			draw_device_buffer(draw_buffer, std::move(local_mix_buffer), blend_mode::normal);
+			
+			draw_params.background			= draw_buffer;
+			draw_params.local_key			= std::move(local_key_buffer);
+			draw_params.layer_key			= layer_key_buffer;
+
+			kernel_.draw(channel_.ogl(), std::move(draw_params));
+		}	
 	}
-	
-	void draw(std::array<std::shared_ptr<device_buffer>,2>& targets, render_item&& item, const std::shared_ptr<device_buffer>& local_key, const std::shared_ptr<device_buffer>& layer_key)
+
+	void draw_device_buffer(safe_ptr<device_buffer>& draw_buffer, std::shared_ptr<device_buffer>&& source_buffer, blend_mode::type blend_mode = blend_mode::normal)
 	{
-		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.
-		}		
+		if(!source_buffer)
+			return;
 
-		targets[1]->attach();
+		draw_params draw_params;
+		draw_params.pix_desc.pix_fmt	= pixel_format::bgra;
+		draw_params.pix_desc.planes		= list_of(pixel_format_desc::plane(source_buffer->width(), source_buffer->height(), 4));
+		draw_params.textures			= list_of(source_buffer);
+		draw_params.transform			= frame_transform();
+		draw_params.blend_mode			= blend_mode;
+		draw_params.background			= draw_buffer;
+
+		kernel_.draw(channel_.ogl(), std::move(draw_params));
+	}
 			
-		kernel_.draw(item, make_safe(targets[0]), local_key, layer_key);
-		item.textures.clear();
+	safe_ptr<device_buffer> create_device_buffer(size_t stride)
+	{
+		auto buffer = channel_.ogl().create_device_buffer(channel_.get_format_desc().width, channel_.get_format_desc().height, stride);
+		channel_.ogl().clear(*buffer);
+		return buffer;
+	}
+};
 		
-		targets[0]->bind();
+struct image_mixer::implementation : boost::noncopyable
+{	
+	ogl_device&						ogl_;
+	image_renderer					renderer_;
+	std::vector<frame_transform>	transform_stack_;
+	std::vector<layer>				layers_; // layer/stream/items
+public:
+	implementation(video_channel_context& video_channel) 
+		: ogl_(video_channel.ogl())
+		, renderer_(video_channel)
+		, transform_stack_(1)	
+	{
+	}
 
-		glCopyTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, 0, 0, targets[0]->width(), targets[0]->height());
+	void begin_layer(blend_mode::type blend_mode)
+	{
+		layers_.push_back(layer(blend_mode));
+	}
 		
-		std::swap(targets[0], targets[1]);
+	void begin(core::basic_frame& frame)
+	{
+		transform_stack_.push_back(transform_stack_.back()*frame.get_frame_transform());
 	}
-				
+		
+	void visit(core::write_frame& frame)
+	{			
+		if(frame.get_frame_transform().field_mode == field_mode::empty)
+			return;
+
+		item item;
+		item.pix_desc	= frame.get_pixel_format_desc();
+		item.textures	= frame.get_textures();
+		item.transform	= transform_stack_.back();
+
+		layers_.back().items.push_back(item);
+	}
+
+	void end()
+	{
+		transform_stack_.pop_back();
+	}
+
+	void end_layer()
+	{		
+	}
+	
+	boost::unique_future<safe_ptr<host_buffer>> render()
+	{
+		return renderer_.render(std::move(layers_));
+	}
+
 	safe_ptr<write_frame> create_frame(const void* tag, const core::pixel_format_desc& desc)
 	{
-		return make_safe<write_frame>(channel_.ogl(), tag, desc);
+		return make_safe<write_frame>(ogl_, tag, desc);
 	}
 };
 
@@ -221,7 +301,7 @@ void image_mixer::visit(core::write_frame& frame){impl_->visit(frame);}
 void image_mixer::end(){impl_->end();}
 boost::unique_future<safe_ptr<host_buffer>> image_mixer::render(){return impl_->render();}
 safe_ptr<write_frame> 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)
 {