2 * copyright (c) 2010 Sveriges Television AB <info@casparcg.com>
\r
4 * This file is part of CasparCG.
\r
6 * CasparCG is free software: you can redistribute it and/or modify
\r
7 * it under the terms of the GNU General Public License as published by
\r
8 * the Free Software Foundation, either version 3 of the License, or
\r
9 * (at your option) any later version.
\r
11 * CasparCG is distributed in the hope that it will be useful,
\r
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
\r
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
\r
14 * GNU General Public License for more details.
\r
16 * You should have received a copy of the GNU General Public License
\r
17 * along with CasparCG. If not, see <http://www.gnu.org/licenses/>.
\r
20 #include "../../stdafx.h"
\r
22 #include "image_mixer.h"
\r
23 #include "image_kernel.h"
\r
25 #include "../gpu/ogl_device.h"
\r
26 #include "../gpu/host_buffer.h"
\r
27 #include "../gpu/device_buffer.h"
\r
28 #include "../write_frame.h"
\r
30 #include "../../video_channel_context.h"
\r
32 #include <common/concurrency/executor.h>
\r
33 #include <common/exception/exceptions.h>
\r
34 #include <common/gl/gl_check.h>
\r
36 #include <core/producer/frame/image_transform.h>
\r
37 #include <core/producer/frame/pixel_format.h>
\r
38 #include <core/video_format.h>
\r
40 #include <boost/foreach.hpp>
\r
43 #include <unordered_map>
\r
45 namespace caspar { namespace core {
\r
47 struct image_mixer::implementation : boost::noncopyable
\r
49 static const size_t LOCAL_KEY_INDEX = 3;
\r
50 static const size_t LAYER_KEY_INDEX = 4;
\r
54 pixel_format_desc desc;
\r
55 std::vector<safe_ptr<device_buffer>> textures;
\r
57 core::image_transform transform;
\r
60 video_channel_context& channel_;
\r
62 std::stack<core::image_transform> transform_stack_;
\r
63 std::queue<std::queue<render_item>> render_queue_;
\r
65 image_kernel kernel_;
\r
67 safe_ptr<host_buffer> read_buffer_;
\r
68 safe_ptr<device_buffer> draw_buffer_;
\r
69 safe_ptr<device_buffer> write_buffer_;
\r
71 safe_ptr<device_buffer> local_key_buffer_;
\r
72 safe_ptr<device_buffer> layer_key_buffer_;
\r
78 implementation(video_channel_context& video_channel)
\r
79 : channel_(video_channel)
\r
80 , read_buffer_(video_channel.ogl().create_host_buffer(video_channel.get_format_desc().size, host_buffer::read_only))
\r
81 , draw_buffer_(video_channel.ogl().create_device_buffer(video_channel.get_format_desc().width, channel_.get_format_desc().height, 4))
\r
82 , write_buffer_ (video_channel.ogl().create_device_buffer(video_channel.get_format_desc().width, channel_.get_format_desc().height, 4))
\r
83 , local_key_buffer_(video_channel.ogl().create_device_buffer(video_channel.get_format_desc().width, channel_.get_format_desc().height, 1))
\r
84 , layer_key_buffer_(video_channel.ogl().create_device_buffer(video_channel.get_format_desc().width, channel_.get_format_desc().height, 1))
\r
88 transform_stack_.push(core::image_transform());
\r
90 channel_.ogl().invoke([=]
\r
92 if(!GLEE_VERSION_3_0)
\r
93 CASPAR_LOG(warning) << "Missing OpenGL 3.0 support.";//BOOST_THROW_EXCEPTION(not_supported() << msg_info("Missing OpenGL 3.0 support."));
\r
97 void begin(const core::basic_frame& frame)
\r
99 transform_stack_.push(transform_stack_.top()*frame.get_image_transform());
\r
102 void visit(core::write_frame& frame)
\r
104 render_item item = {frame.get_pixel_format_desc(), frame.get_textures(), frame.tag(), transform_stack_.top()*frame.get_image_transform()};
\r
105 render_queue_.back().push(item);
\r
110 transform_stack_.pop();
\r
115 render_queue_.push(std::queue<render_item>());
\r
122 void reinitialize_buffers()
\r
124 read_buffer_ = channel_.ogl().create_host_buffer(channel_.get_format_desc().size, host_buffer::read_only);
\r
125 draw_buffer_ = channel_.ogl().create_device_buffer(channel_.get_format_desc().width, channel_.get_format_desc().height, 4);
\r
126 write_buffer_ = channel_.ogl().create_device_buffer(channel_.get_format_desc().width, channel_.get_format_desc().height, 4);
\r
127 local_key_buffer_ = channel_.ogl().create_device_buffer(channel_.get_format_desc().width, channel_.get_format_desc().height, 1);
\r
128 layer_key_buffer_ = channel_.ogl().create_device_buffer(channel_.get_format_desc().width, channel_.get_format_desc().height, 1);
\r
129 channel_.ogl().gc();
\r
132 safe_ptr<host_buffer> render()
\r
134 auto read_buffer = read_buffer_;
\r
135 auto result = channel_.ogl().begin_invoke([=]() -> safe_ptr<host_buffer>
\r
137 read_buffer->map();
\r
138 return read_buffer;
\r
141 auto render_queue = std::move(render_queue_);
\r
143 channel_.ogl().begin_invoke([=]() mutable
\r
145 if(draw_buffer_->width() != channel_.get_format_desc().width || draw_buffer_->height() != channel_.get_format_desc().height)
\r
146 reinitialize_buffers();
\r
148 local_key_ = false;
\r
149 layer_key_ = false;
\r
152 local_key_buffer_->clear();
\r
153 layer_key_buffer_->clear();
\r
154 draw_buffer_->clear();
\r
156 // Draw items in device.
\r
158 while(!render_queue.empty())
\r
160 auto layer = render_queue.front();
\r
161 render_queue.pop();
\r
163 draw_buffer_->attach();
\r
165 while(!layer.empty())
\r
167 // Split layer into streams
\r
168 std::vector<render_item> stream;
\r
171 stream.push_back(layer.front());
\r
174 while(!layer.empty() && layer.front().tag == stream.front().tag);
\r
178 channel_.ogl().yield(); // Allow quick buffer allocation to execute.
\r
181 layer_key_ = local_key_; // If there was only key in last layer then use it as key for the entire next layer.
\r
182 local_key_ = false;
\r
184 std::swap(local_key_buffer_, layer_key_buffer_);
\r
187 std::swap(draw_buffer_, write_buffer_);
\r
189 // Start transfer from device to host.
\r
190 read_buffer_ = channel_.ogl().create_host_buffer(channel_.get_format_desc().size, host_buffer::read_only);
\r
191 write_buffer_->write(*read_buffer_);
\r
194 return std::move(result.get());
\r
197 void draw(const std::vector<render_item>& stream)
\r
202 BOOST_FOREACH(auto item, stream)
\r
204 bool local_key = false;
\r
205 bool layer_key = false;
\r
207 // Setup key and kernel
\r
209 if(item.transform.get_is_key()) // This is a key frame, render it to the local_key buffer for later use.
\r
211 if(!local_key_) // Initialize local-key if it is not active.
\r
213 local_key_buffer_->clear();
\r
214 local_key_buffer_->attach();
\r
218 else // This is a normal frame. Use key buffers if they are active.
\r
220 local_key = local_key_;
\r
221 layer_key = layer_key_;
\r
223 if(local_key_) // Use local key if we have it.
\r
225 local_key_buffer_->bind(LOCAL_KEY_INDEX);
\r
226 draw_buffer_->attach();
\r
229 if(layer_key_) // Use layer key if we have it.
\r
230 layer_key_buffer_->bind(LAYER_KEY_INDEX);
\r
235 for(size_t n = 0; n < item.textures.size(); ++n)
\r
236 item.textures[n]->bind(n);
\r
240 kernel_.draw(channel_.get_format_desc().width, channel_.get_format_desc().height, item.desc, item.transform, local_key, layer_key);
\r
243 local_key_ = stream.back().transform.get_is_key(); // Keep for next layer if last frame is key
\r
246 safe_ptr<write_frame> create_frame(const void* tag, const core::pixel_format_desc& desc)
\r
248 return make_safe<write_frame>(channel_.ogl(), reinterpret_cast<int>(tag), desc);
\r
252 image_mixer::image_mixer(video_channel_context& video_channel) : impl_(new implementation(video_channel)){}
\r
253 void image_mixer::begin(const core::basic_frame& frame){impl_->begin(frame);}
\r
254 void image_mixer::visit(core::write_frame& frame){impl_->visit(frame);}
\r
255 void image_mixer::end(){impl_->end();}
\r
256 safe_ptr<host_buffer> image_mixer::render(){return impl_->render();}
\r
257 safe_ptr<write_frame> image_mixer::create_frame(const void* tag, const core::pixel_format_desc& desc){return impl_->create_frame(tag, desc);}
\r
258 void image_mixer::begin_layer(){impl_->begin_layer();}
\r
259 void image_mixer::end_layer(){impl_->end_layer();}
\r
260 image_mixer& image_mixer::operator=(image_mixer&& other)
\r
262 impl_ = std::move(other.impl_);
\r