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/cast.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
50 const core::video_format_desc format_desc_;
\r
52 std::stack<core::image_transform> transform_stack_;
\r
55 std::array<std::shared_ptr<device_buffer>, 2> render_targets_;
\r
57 std::shared_ptr<host_buffer> reading_;
\r
59 image_kernel kernel_;
\r
63 std::shared_ptr<device_buffer> key;
\r
68 implementation(const core::video_format_desc& format_desc)
\r
69 : format_desc_(format_desc)
\r
71 ogl_device::invoke([]
\r
73 if(!GLEE_VERSION_3_0)
\r
74 BOOST_THROW_EXCEPTION(not_supported() << msg_info("Missing OpenGL 3.0 support."));
\r
77 ogl_device::begin_invoke([=]
\r
79 transform_stack_.push(core::image_transform());
\r
80 transform_stack_.top().set_mode(core::video_mode::progressive);
\r
82 GL(glEnable(GL_TEXTURE_2D));
\r
83 GL(glDisable(GL_DEPTH_TEST));
\r
85 render_targets_[0] = ogl_device::create_device_buffer(format_desc.width, format_desc.height, 4);
\r
86 render_targets_[1] = ogl_device::create_device_buffer(format_desc.width, format_desc.height, 4);
\r
88 GL(glGenFramebuffers(1, &fbo_));
\r
89 GL(glBindFramebuffer(GL_FRAMEBUFFER_EXT, fbo_));
\r
90 GL(glReadBuffer(GL_COLOR_ATTACHMENT0_EXT));
\r
92 reading_ = ogl_device::create_host_buffer(format_desc_.size, host_buffer::read_only);
\r
98 glDeleteFramebuffersEXT(1, &fbo_);
\r
101 void begin(const core::basic_frame& frame)
\r
103 transform_stack_.push(transform_stack_.top()*frame.get_image_transform());
\r
106 void visit(core::write_frame& frame)
\r
108 auto gpu_frame = dynamic_cast<gpu_write_frame*>(&frame);
\r
112 auto desc = gpu_frame->get_pixel_format_desc();
\r
113 auto buffers = gpu_frame->get_plane_buffers();
\r
114 auto transform = transform_stack_.top();
\r
116 if(buffers.empty())
\r
119 ogl_device::begin_invoke([=]
\r
121 if(transform.get_key_depth() > 0) // Its a key_frame just save buffer for next frame.
\r
123 auto texture = create_device_buffer(desc.planes[0]);
\r
124 texture->read(*buffers[0]);
\r
125 keyer_.key = texture;
\r
126 keyer_.depth = transform.get_key_depth();
\r
130 const bool has_separate_key = keyer_.depth > 0;
\r
131 const size_t max_index = has_separate_key ? 3 : 4;
\r
133 std::vector<safe_ptr<device_buffer>> device_buffers;
\r
134 for(size_t n = 0; n < std::min(max_index, buffers.size()); ++n)
\r
136 auto texture = create_device_buffer(desc.planes[n]);
\r
137 texture->read(*buffers[n]);
\r
138 device_buffers.push_back(texture);
\r
141 if(has_separate_key && keyer_.key)
\r
142 device_buffers.push_back(make_safe(keyer_.key));
\r
144 for(size_t n = 0; n < device_buffers.size(); ++n)
\r
146 GL(glActiveTexture(GL_TEXTURE0+n));
\r
147 device_buffers[n]->bind();
\r
150 GL(glColor4d(1.0, 1.0, 1.0, transform.get_opacity()));
\r
151 GL(glViewport(0, 0, format_desc_.width, format_desc_.height));
\r
152 kernel_.apply(desc, transform, has_separate_key);
\r
154 auto m_p = transform.get_key_translation();
\r
155 auto m_s = transform.get_key_scale();
\r
156 auto w = static_cast<double>(format_desc_.width);
\r
157 auto h = static_cast<double>(format_desc_.height);
\r
159 GL(glEnable(GL_SCISSOR_TEST));
\r
160 GL(glScissor(static_cast<size_t>(m_p[0]*w), static_cast<size_t>(m_p[1]*h), static_cast<size_t>(m_s[0]*w), static_cast<size_t>(m_s[1]*h)));
\r
162 auto f_p = transform.get_fill_translation();
\r
163 auto f_s = transform.get_fill_scale();
\r
166 glTexCoord2d(0.0, 0.0); glVertex2d( f_p[0] *2.0-1.0, f_p[1] *2.0-1.0);
\r
167 glTexCoord2d(1.0, 0.0); glVertex2d((f_p[0]+f_s[0])*2.0-1.0, f_p[1] *2.0-1.0);
\r
168 glTexCoord2d(1.0, 1.0); glVertex2d((f_p[0]+f_s[0])*2.0-1.0, (f_p[1]+f_s[1])*2.0-1.0);
\r
169 glTexCoord2d(0.0, 1.0); glVertex2d( f_p[0] *2.0-1.0, (f_p[1]+f_s[1])*2.0-1.0);
\r
171 GL(glDisable(GL_SCISSOR_TEST));
\r
173 if(keyer_.depth == 0)
\r
174 keyer_.key.reset();
\r
183 transform_stack_.pop();
\r
186 boost::unique_future<safe_ptr<const host_buffer>> begin_pass()
\r
188 return ogl_device::begin_invoke([=]() -> safe_ptr<const host_buffer>
\r
192 render_targets_[0]->attach(0);
\r
193 GL(glClear(GL_COLOR_BUFFER_BIT));
\r
194 return safe_ptr<const host_buffer>(reading_);
\r
200 ogl_device::begin_invoke([=]
\r
202 reading_ = ogl_device::create_host_buffer(format_desc_.size, host_buffer::read_only);
\r
203 render_targets_[0]->write(*reading_);
\r
204 std::rotate(render_targets_.begin(), render_targets_.begin() + 1, render_targets_.end());
\r
208 safe_ptr<device_buffer> create_device_buffer(const core::pixel_format_desc::plane& plane)
\r
210 return ogl_device::create_device_buffer(plane.width, plane.height, plane.channels);
\r
213 std::vector<safe_ptr<host_buffer>> create_buffers(const core::pixel_format_desc& format)
\r
215 std::vector<safe_ptr<host_buffer>> buffers;
\r
216 std::transform(format.planes.begin(), format.planes.end(), std::back_inserter(buffers), [&](const core::pixel_format_desc::plane& plane)
\r
218 return ogl_device::create_host_buffer(plane.size, host_buffer::write_only);
\r
224 image_mixer::image_mixer(const core::video_format_desc& format_desc) : impl_(new implementation(format_desc)){}
\r
225 void image_mixer::begin(const core::basic_frame& frame){impl_->begin(frame);}
\r
226 void image_mixer::visit(core::write_frame& frame){impl_->visit(frame);}
\r
227 void image_mixer::end(){impl_->end();}
\r
228 boost::unique_future<safe_ptr<const host_buffer>> image_mixer::begin_pass(){ return impl_->begin_pass();}
\r
229 void image_mixer::end_pass(){impl_->end_pass();}
\r
230 std::vector<safe_ptr<host_buffer>> image_mixer::create_buffers(const core::pixel_format_desc& format){return impl_->create_buffers(format);}
\r