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 "../gpu/gpu_write_frame.h"
\r
30 #include <common/concurrency/executor.h>
\r
31 #include <common/exception/exceptions.h>
\r
32 #include <common/gl/gl_check.h>
\r
34 #include <core/producer/frame/image_transform.h>
\r
35 #include <core/producer/frame/pixel_format.h>
\r
36 #include <core/video_format.h>
\r
38 #include <boost/foreach.hpp>
\r
41 #include <SFML/Window/Context.hpp>
\r
44 #include <unordered_map>
\r
46 namespace caspar { namespace mixer {
\r
48 struct image_mixer::implementation : boost::noncopyable
\r
52 core::pixel_format_desc desc;
\r
53 std::vector<safe_ptr<device_buffer>> textures;
\r
54 core::image_transform transform;
\r
57 const core::video_format_desc format_desc_;
\r
59 std::stack<core::image_transform> transform_stack_;
\r
63 safe_ptr<device_buffer> front_target_;
\r
64 safe_ptr<device_buffer> back_target_;
\r
65 safe_ptr<device_buffer> key_target_;
\r
67 safe_ptr<host_buffer> reading_;
\r
69 image_kernel kernel_;
\r
71 std::vector<render_item> waiting_queue_;
\r
72 std::vector<render_item> render_queue_;
\r
75 implementation(const core::video_format_desc& format_desc)
\r
76 : format_desc_(format_desc)
\r
78 , back_target_(ogl_device::create_device_buffer(format_desc.width, format_desc.height, 4))
\r
79 , front_target_(ogl_device::create_device_buffer(format_desc.width, format_desc.height, 4))
\r
80 , key_target_(ogl_device::create_device_buffer(format_desc.width, format_desc.height, 4))
\r
81 , reading_(ogl_device::create_host_buffer(format_desc_.size, host_buffer::read_only))
\r
83 transform_stack_.push(core::image_transform());
\r
86 void begin(const core::basic_frame& frame)
\r
88 transform_stack_.push(transform_stack_.top()*frame.get_image_transform());
\r
91 void visit(core::write_frame& frame)
\r
93 auto gpu_frame = dynamic_cast<gpu_write_frame*>(&frame);
\r
97 auto desc = gpu_frame->get_pixel_format_desc();
\r
98 auto buffers = gpu_frame->get_plane_buffers();
\r
99 auto transform = transform_stack_.top();
\r
101 ogl_device::begin_invoke([=]
\r
106 item.transform = transform;
\r
108 for(size_t n = 0; n < buffers.size(); ++n)
\r
110 GL(glActiveTexture(GL_TEXTURE0+n));
\r
111 auto texture = ogl_device::create_device_buffer(desc.planes[n].width, desc.planes[n].height, desc.planes[n].channels);
\r
112 texture->read(*buffers[n]);
\r
113 item.textures.push_back(texture);
\r
116 waiting_queue_.push_back(item);
\r
122 transform_stack_.pop();
\r
125 boost::unique_future<safe_ptr<const host_buffer>> render()
\r
127 auto result = ogl_device::begin_invoke([=]() -> safe_ptr<const host_buffer>
\r
129 reading_->map(); // Might block.
\r
133 ogl_device::begin_invoke([=]
\r
137 // Clear and bind frame-buffers.
\r
139 key_target_->attach();
\r
140 GL(glClear(GL_COLOR_BUFFER_BIT));
\r
142 front_target_->attach();
\r
143 GL(glClear(GL_COLOR_BUFFER_BIT));
\r
147 BOOST_FOREACH(auto item, render_queue_)
\r
150 // Move waiting items to queue.
\r
152 render_queue_ = std::move(waiting_queue_);
\r
154 // Start read-back.
\r
156 reading_ = ogl_device::create_host_buffer(format_desc_.size, host_buffer::read_only);
\r
157 front_target_->attach();
\r
158 front_target_->write(*reading_);
\r
159 std::swap(front_target_, back_target_);
\r
162 return std::move(result);
\r
165 void render(const render_item& item)
\r
167 const auto desc = item.desc;
\r
168 auto textures = item.textures;
\r
169 const auto transform = item.transform;
\r
173 for(size_t n = 0; n < textures.size(); ++n)
\r
175 GL(glActiveTexture(GL_TEXTURE0+n));
\r
176 textures[n]->bind();
\r
179 // Setup key and kernel
\r
181 bool has_separate_key = false;
\r
183 if(transform.get_is_key())
\r
187 key_target_->attach();
\r
189 GL(glClear(GL_COLOR_BUFFER_BIT));
\r
196 has_separate_key = true;
\r
199 front_target_->attach();
\r
200 GL(glActiveTexture(GL_TEXTURE0+3));
\r
201 key_target_->bind();
\r
207 kernel_.draw(format_desc_.width, format_desc_.height, desc, transform, has_separate_key);
\r
210 std::vector<safe_ptr<host_buffer>> create_buffers(const core::pixel_format_desc& format)
\r
212 std::vector<safe_ptr<host_buffer>> buffers;
\r
213 std::transform(format.planes.begin(), format.planes.end(), std::back_inserter(buffers), [&](const core::pixel_format_desc::plane& plane)
\r
215 return ogl_device::create_host_buffer(plane.size, host_buffer::write_only);
\r
221 image_mixer::image_mixer(const core::video_format_desc& format_desc) : impl_(new implementation(format_desc)){}
\r
222 void image_mixer::begin(const core::basic_frame& frame){impl_->begin(frame);}
\r
223 void image_mixer::visit(core::write_frame& frame){impl_->visit(frame);}
\r
224 void image_mixer::end(){impl_->end();}
\r
225 boost::unique_future<safe_ptr<const host_buffer>> image_mixer::render(){return impl_->render();}
\r
226 std::vector<safe_ptr<host_buffer>> image_mixer::create_buffers(const core::pixel_format_desc& format){return impl_->create_buffers(format);}
\r