X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=core%2Fmixer%2Fimage%2Fimage_mixer.cpp;h=b754e28e899eb63e70c9dfb3c1283d1910d189b3;hb=e427f7e2661d120c1152eaa7868cc18a3b8cdd7e;hp=ae8fec69ddf1bee20f56f712b42ce0434cba160a;hpb=2fcb2929a931f68cea258bf5ec1f1642b954801b;p=casparcg

diff --git a/core/mixer/image/image_mixer.cpp b/core/mixer/image/image_mixer.cpp
index ae8fec69d..b754e28e8 100644
--- a/core/mixer/image/image_mixer.cpp
+++ b/core/mixer/image/image_mixer.cpp
@@ -20,181 +20,280 @@
 #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/find.hpp>
+#include <boost/range/algorithm_ext/erase.hpp>
 
 #include <algorithm>
-#include <array>
 #include <deque>
-#include <unordered_map>
 
-namespace caspar { namespace core {
-		
-struct image_mixer::implementation : boost::noncopyable
-{		
-	typedef std::deque<render_item>			layer;
+using namespace boost::assign;
 
-	video_channel_context&					channel_;
-
-	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::shared_ptr<device_buffer>			draw_buffer_;
+struct item
+{
+	pixel_format_desc						pix_desc;
+	std::vector<safe_ptr<device_buffer>>	textures;
+	frame_transform							transform;
+};
 
-	std::shared_ptr<device_buffer>			local_key_buffer_;
-	std::shared_ptr<device_buffer>			layer_key_buffer_;
-		
+typedef std::pair<blend_mode::type, std::vector<item>> layer;
+
+class image_renderer
+{
+	ogl_device&								ogl_;
+	video_format_desc						format_desc_;
+	image_kernel							kernel_;	
+	std::shared_ptr<device_buffer>			active_buffer_;
 public:
-	implementation(video_channel_context& video_channel) 
-		: channel_(video_channel)
-		, transform_stack_(1)
-		, mode_stack_(1, video_mode::progressive)
+	image_renderer(ogl_device& ogl, const video_format_desc& format_desc)
+		: ogl_(ogl)
+		, format_desc_(format_desc)
 	{
 	}
-
-	~implementation()
-	{
-		channel_.ogl().gc();
+	
+	boost::unique_future<safe_ptr<host_buffer>> render(std::vector<layer>&& layers)
+	{		
+		auto layers2 = make_move_on_copy(std::move(layers));
+		return ogl_.begin_invoke([=]
+		{
+			return do_render(std::move(layers2.value));
+		});
 	}
 	
-	void begin(core::basic_frame& frame)
+private:
+	safe_ptr<host_buffer> do_render(std::vector<layer>&& layers)
 	{
-		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());
-	}
+		auto draw_buffer = create_mixer_buffer(4);
+
+		if(format_desc_.field_mode != field_mode::progressive)
+		{
+			auto upper = layers;
+			auto lower = std::move(layers);
+
+			BOOST_FOREACH(auto& layer, upper)
+			{
+				BOOST_FOREACH(auto& item, layer.second)
+					item.transform.field_mode = static_cast<field_mode::type>(item.transform.field_mode & field_mode::upper);
+			}
+
+			BOOST_FOREACH(auto& layer, lower)
+			{
+				BOOST_FOREACH(auto& item, layer.second)
+					item.transform.field_mode = static_cast<field_mode::type>(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 host_buffer = ogl_.create_host_buffer(format_desc_.size, host_buffer::read_only);
+		ogl_.attach(*draw_buffer);
+		host_buffer->begin_read(draw_buffer->width(), draw_buffer->height(), format(draw_buffer->stride()));
 		
-	void visit(core::write_frame& frame)
-	{	
-		CASPAR_ASSERT(!layers_.empty());
+		active_buffer_ = std::move(draw_buffer);
+
+		ogl_.flush(); // NOTE: This is important, otherwise fences will deadlock.
+			
+		return host_buffer;
+	}
+
+	void draw(std::vector<layer>&&		layers, 
+			  safe_ptr<device_buffer>&	draw_buffer)
+	{
+		std::shared_ptr<device_buffer> layer_key_buffer;
 
-		// 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())
+		BOOST_FOREACH(auto& layer, layers)
+			draw_layer(std::move(layer), draw_buffer, layer_key_buffer);
+	}
+
+	void draw_layer(layer&&							layer, 
+					safe_ptr<device_buffer>&		draw_buffer,
+					std::shared_ptr<device_buffer>& layer_key_buffer)
+	{				
+		boost::remove_erase_if(layer.second, [](const item& item){return item.transform.field_mode == field_mode::empty;});
+
+		if(layer.second.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();
+		std::shared_ptr<device_buffer> local_key_buffer;
+		std::shared_ptr<device_buffer> local_mix_buffer;
+				
+		if(layer.first != blend_mode::normal && layer.second.size() > 1)
+		{
+			auto layer_draw_buffer = create_mixer_buffer(4);
+
+			BOOST_FOREACH(auto& item, layer.second)
+				draw_item(std::move(item), layer_draw_buffer, layer_key_buffer, local_key_buffer, local_mix_buffer);	
+		
+			draw_mixer_buffer(layer_draw_buffer, std::move(local_mix_buffer), blend_mode::normal);							
+			draw_mixer_buffer(draw_buffer, std::move(layer_draw_buffer), layer.first);
+		}
+		else // fast path
+		{
+			BOOST_FOREACH(auto& item, layer.second)		
+				draw_item(std::move(item), draw_buffer, layer_key_buffer, local_key_buffer, local_mix_buffer);		
+					
+			draw_mixer_buffer(draw_buffer, std::move(local_mix_buffer), blend_mode::normal);
+		}					
 
-		if(boost::range::find(layer, item) == layer.end())
-			layer.push_back(item);
+		layer_key_buffer = std::move(local_key_buffer);
 	}
 
-	void end()
-	{
-		transform_stack_.pop_back();
-		mode_stack_.pop_back();
+	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);
+
+		if(item.transform.is_key)
+		{
+			local_key_buffer = local_key_buffer ? local_key_buffer : create_mixer_buffer(1);
+
+			draw_params.background			= local_key_buffer;
+			draw_params.local_key			= nullptr;
+			draw_params.layer_key			= nullptr;
+
+			kernel_.draw(ogl_, std::move(draw_params));
+		}
+		else if(item.transform.is_mix)
+		{
+			local_mix_buffer = local_mix_buffer ? local_mix_buffer : create_mixer_buffer(4);
+
+			draw_params.background			= local_mix_buffer;
+			draw_params.local_key			= std::move(local_key_buffer);
+			draw_params.layer_key			= layer_key_buffer;
+
+			draw_params.keyer				= keyer::additive;
+
+			kernel_.draw(ogl_, std::move(draw_params));
+		}
+		else
+		{
+			draw_mixer_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(ogl_, std::move(draw_params));
+		}	
 	}
 
-	void begin_layer()
+	void draw_mixer_buffer(safe_ptr<device_buffer>&			draw_buffer, 
+						   std::shared_ptr<device_buffer>&& source_buffer, 
+						   blend_mode::type					blend_mode = blend_mode::normal)
 	{
-		layers_.push_back(layer());
-	}
+		if(!source_buffer)
+			return;
 
-	void end_layer()
+		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(ogl_, std::move(draw_params));
+	}
+			
+	safe_ptr<device_buffer> create_mixer_buffer(size_t stride)
 	{
+		auto buffer = ogl_.create_device_buffer(format_desc_.width, format_desc_.height, stride);
+		ogl_.clear(*buffer);
+		return buffer;
 	}
-	
-	boost::unique_future<safe_ptr<host_buffer>> render()
-	{		
-		auto layers = std::move(layers_);
-		return channel_.ogl().begin_invoke([=]()mutable
-		{
-			return render(std::move(layers));
-		});
+};
+		
+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(ogl_device& ogl, const video_format_desc& format_desc) 
+		: ogl_(ogl)
+		, renderer_(ogl, format_desc)
+		, transform_stack_(1)	
+	{
 	}
-	
-	safe_ptr<host_buffer> render(std::deque<layer>&& layers)
+
+	void begin_layer(blend_mode::type blend_mode)
 	{
-		draw_buffer_ = channel_.ogl().create_device_buffer(channel_.get_format_desc().width, channel_.get_format_desc().height, 4);				
-		channel_.ogl().clear(*draw_buffer_);
-				
-		BOOST_FOREACH(auto& layer, layers)
-			draw(std::move(layer));
-				
-		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()));
-		
-		GL(glFlush());
+		layers_.push_back(std::make_pair(blend_mode, std::vector<item>()));
+	}
 		
-		return host_buffer;
+	void begin(basic_frame& frame)
+	{
+		transform_stack_.push_back(transform_stack_.back()*frame.get_frame_transform());
 	}
-
-	void draw(layer&& layer)
-	{					
-		BOOST_FOREACH(auto& item, layer)
-			draw(std::move(item));
 		
-		std::swap(local_key_buffer_, layer_key_buffer_);
+	void visit(write_frame& frame)
+	{			
+		item item;
+		item.pix_desc	= frame.get_pixel_format_desc();
+		item.textures	= frame.get_textures();
+		item.transform	= transform_stack_.back();
+
+		layers_.back().second.push_back(item);
 	}
 
-	void draw(render_item&& item)
-	{											
-		if(item.transform.get_is_key())
-		{
-			if(!local_key_buffer_)
-			{
-				local_key_buffer_ = channel_.ogl().create_device_buffer(channel_.get_format_desc().width, channel_.get_format_desc().height, 1);
-				channel_.ogl().clear(*local_key_buffer_);
-			}
+	void end()
+	{
+		transform_stack_.pop_back();
+	}
 
-			draw(local_key_buffer_, std::move(item), nullptr, nullptr);
-		}
-		else
-		{
-			draw(draw_buffer_, std::move(item), local_key_buffer_, layer_key_buffer_);	
-			local_key_buffer_.reset();
-		}
+	void end_layer()
+	{		
 	}
 	
-	void draw(std::shared_ptr<device_buffer>& target, render_item&& item, const std::shared_ptr<device_buffer>& local_key, const std::shared_ptr<device_buffer>& layer_key)
+	boost::unique_future<safe_ptr<host_buffer>> render()
 	{
-		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.
-		}		
-
-		kernel_.draw(channel_.ogl(), std::move(item), make_safe(target), local_key, layer_key);
+		return renderer_.render(std::move(layers_));
 	}
-				
-	safe_ptr<write_frame> create_frame(const void* tag, const core::pixel_format_desc& desc)
+
+	safe_ptr<write_frame> create_frame(const void* tag, const pixel_format_desc& desc)
 	{
-		return make_safe<write_frame>(channel_.ogl(), tag, desc);
+		return make_safe<write_frame>(ogl_, tag, desc);
 	}
 };
 
-image_mixer::image_mixer(video_channel_context& video_channel) : impl_(new implementation(video_channel)){}
-void image_mixer::begin(core::basic_frame& frame){impl_->begin(frame);}
-void image_mixer::visit(core::write_frame& frame){impl_->visit(frame);}
+image_mixer::image_mixer(ogl_device& ogl, const video_format_desc& format_desc) : impl_(new implementation(ogl, format_desc)){}
+void image_mixer::begin(basic_frame& frame){impl_->begin(frame);}
+void image_mixer::visit(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();}
+safe_ptr<write_frame> image_mixer::create_frame(const void* tag, const pixel_format_desc& desc){return impl_->create_frame(tag, desc);}
+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)
 {