X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=core%2Fmixer%2Fimage%2Fimage_mixer.cpp;h=d81c9f798156bafa03ee97777246e5998fb96895;hb=371df45d22eba16e04e14238d34c86162a36ce20;hp=a352578fd2de64f61f6baf92bc2a6eae466f2c0c;hpb=6173b6251068fb533170e85bcd46c6f3921394f0;p=casparcg

diff --git a/core/mixer/image/image_mixer.cpp b/core/mixer/image/image_mixer.cpp
index a352578fd..d81c9f798 100644
--- a/core/mixer/image/image_mixer.cpp
+++ b/core/mixer/image/image_mixer.cpp
@@ -20,238 +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_;
+namespace caspar { namespace core {
+	
+struct item
+{
+	pixel_format_desc						pix_desc;
+	std::vector<safe_ptr<device_buffer>>	textures;
+	frame_transform							transform;
+};
 
-	std::vector<image_transform>			transform_stack_;
-	std::vector<video_mode::type>			mode_stack_;
+typedef std::pair<blend_mode::type, std::vector<item>> layer;
 
-	std::deque<std::deque<render_item>>		layers_; // layer/stream/items
-	
-	image_kernel							kernel_;		
+class image_renderer
+{
+	ogl_device&								ogl_;
+	const video_format_desc					format_desc_;
+	image_kernel							kernel_;	
+	std::shared_ptr<device_buffer>			transferring_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>> operator()(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());
-	}
-		
-	void visit(core::write_frame& frame)
-	{	
-		CASPAR_ASSERT(!layers_.empty());
+		auto draw_buffer = create_mixer_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(format_desc_.field_mode != field_mode::progressive)
+		{
+			auto upper = layers;
+			auto lower = std::move(layers);
 
-		auto& layer = layers_.back();
+			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);
+			}
 
-		if(boost::range::find(layer, item) == layer.end())
-			layer.push_back(item);
-	}
+			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);
+			}
 
-	void end()
-	{
-		transform_stack_.pop_back();
-		mode_stack_.pop_back();
-	}
+			draw(std::move(upper), draw_buffer);
+			draw(std::move(lower), draw_buffer);
+		}
+		else
+		{
+			draw(std::move(layers), draw_buffer);
+		}
 
-	void begin_layer()
-	{
-		layers_.push_back(layer());
-	}
+		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()));
+		
+		transferring_buffer_ = std::move(draw_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));
-		});
+		ogl_.flush(); // NOTE: This is important, otherwise fences will deadlock.
+			
+		return host_buffer;
 	}
-	
-	safe_ptr<host_buffer> render(std::deque<layer>&& layers)
+
+	void draw(std::vector<layer>&&		layers, 
+			  safe_ptr<device_buffer>&	draw_buffer)
 	{
 		std::shared_ptr<device_buffer> layer_key_buffer;
 
-		auto draw_buffer = create_device_buffer(4);
-				
 		BOOST_FOREACH(auto& layer, layers)
-			draw(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()));
-		
-		GL(glFlush());
-		
-		return host_buffer;
+			draw_layer(std::move(layer), draw_buffer, layer_key_buffer);
 	}
 
-	// TODO: We might have more overlaps for opacity transitions
-	// TODO: What about blending modes, are they ok? Maybe only overlap detection is required for opacity?
-	void draw(layer&& layer, const safe_ptr<device_buffer>& draw_buffer, std::shared_ptr<device_buffer>& layer_key_buffer)
+	void draw_layer(layer&&							layer, 
+					safe_ptr<device_buffer>&		draw_buffer,
+					std::shared_ptr<device_buffer>& layer_key_buffer)
 	{				
-		if(layer.empty())
+		boost::remove_erase_if(layer.second, [](const item& item){return item.transform.field_mode == field_mode::empty;});
+
+		if(layer.second.empty())
 			return;
 
 		std::shared_ptr<device_buffer> local_key_buffer;
-					
-		if(has_overlapping_items(layer, layer.front().transform.get_blend_mode()))
+		std::shared_ptr<device_buffer> local_mix_buffer;
+				
+		if(layer.first != blend_mode::normal)
 		{
-			auto local_draw_buffer = create_device_buffer(4);	
-			
-			BOOST_FOREACH(auto& item, layer)		
-			{
-				item.transform.set_blend_mode(image_transform::blend_mode::normal);
-				draw_item(std::move(item), local_draw_buffer, local_key_buffer, layer_key_buffer);		
-			}
+			auto layer_draw_buffer = create_mixer_buffer(4);
 
-			kernel_.draw(channel_.ogl(), create_render_item(local_draw_buffer, layer.front().transform.get_blend_mode()), draw_buffer, nullptr, nullptr);
+			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)		
-				draw_item(std::move(item), draw_buffer, local_key_buffer, layer_key_buffer);		
+			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);
 		}					
 
-		std::swap(local_key_buffer, layer_key_buffer);
+		layer_key_buffer = std::move(local_key_buffer);
 	}
 
-	void draw_item(render_item&& item, const safe_ptr<device_buffer>& draw_buffer, std::shared_ptr<device_buffer>& local_key_buffer, std::shared_ptr<device_buffer>& layer_key_buffer)
-	{											
-		if(item.transform.get_is_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);
+
+		if(item.transform.is_key)
 		{
-			if(!local_key_buffer)
-				local_key_buffer = create_device_buffer(1);
+			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;
 
-			item.transform.set_opacity(1.0);
-			item.transform.set_blend_mode(image_transform::blend_mode::normal);
-			kernel_.draw(channel_.ogl(), std::move(item), make_safe(local_key_buffer), nullptr, nullptr);
+			kernel_.draw(ogl_, std::move(draw_params));
 		}
-		else
+		else if(item.transform.is_mix)
 		{
-			kernel_.draw(channel_.ogl(), std::move(item), draw_buffer, local_key_buffer, layer_key_buffer);
-			local_key_buffer.reset();
+			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));
+		}	
 	}
 
-	// TODO: Optimize
-	bool has_overlapping_items(const layer& layer, image_transform::blend_mode::type blend_mode)
+	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)
 	{
-		if(layer.empty())
-			return false;	
-		
-		implementation::layer fill;
+		if(!source_buffer)
+			return;
 
-		std::copy_if(layer.begin(), layer.end(), std::back_inserter(fill), [&](const render_item& item)
-		{
-			return !item.transform.get_is_key();
-		});
-			
-		if(blend_mode == image_transform::blend_mode::normal) // Only overlap if opacity
-		{
-			return std::any_of(fill.begin(), fill.end(), [&](const render_item& item)
-			{
-				return item.transform.get_opacity() < 1.0 - 0.001;
-			});
-		}
+		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;
 
-		// Simple solution, just check if we have differnt video streams / tags.
-		return std::any_of(fill.begin(), fill.end(), [&](const render_item& item)
-		{
-			return item.tag != fill.front().tag;
-		});
-	}			
+		kernel_.draw(ogl_, std::move(draw_params));
+	}
 			
-	render_item create_render_item(const safe_ptr<device_buffer>& buffer, image_transform::blend_mode::type blend_mode)
+	safe_ptr<device_buffer> create_mixer_buffer(size_t stride)
 	{
-		CASPAR_ASSERT(buffer->stride() == 4 && "Only used for bgra textures");
-
-		pixel_format_desc desc;
-		desc.pix_fmt = pixel_format::bgra;
-		desc.planes.push_back(pixel_format_desc::plane(channel_.get_format_desc().width, channel_.get_format_desc().height, 4));
+		auto buffer = ogl_.create_device_buffer(format_desc_.width, format_desc_.height, stride);
+		ogl_.clear(*buffer);
+		return buffer;
+	}
+};
+		
+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)	
+	{
+	}
 
-		std::vector<safe_ptr<device_buffer>> textures;
-		textures.push_back(buffer);
-				
-		image_transform transform;
-		transform.set_blend_mode(blend_mode);
+	void begin_layer(blend_mode::type blend_mode)
+	{
+		layers_.push_back(std::make_pair(blend_mode, std::vector<item>()));
+	}
+		
+	void begin(basic_frame& frame)
+	{
+		transform_stack_.push_back(transform_stack_.back()*frame.get_frame_transform());
+	}
+		
+	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);
+	}
 
-		return render_item(desc, std::move(textures), transform, video_mode::progressive, nullptr);		 
+	void end()
+	{
+		transform_stack_.pop_back();
 	}
 
-	safe_ptr<device_buffer> create_device_buffer(size_t stride)
+	void end_layer()
+	{		
+	}
+	
+	boost::unique_future<safe_ptr<host_buffer>> render()
 	{
-		auto buffer = channel_.ogl().create_device_buffer(channel_.get_format_desc().width, channel_.get_format_desc().height, stride);
-		channel_.ogl().clear(*buffer);
-		return buffer;
+		return renderer_(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)
 {