1 #include "../../StdAfx.h"
\r
3 #include "image_mixer.h"
\r
4 #include "image_kernel.h"
\r
6 #include "../gpu/ogl_device.h"
\r
7 #include "../gpu/host_buffer.h"
\r
8 #include "../gpu/device_buffer.h"
\r
10 #include <common/exception/exceptions.h>
\r
11 #include <common/gl/gl_check.h>
\r
12 #include <common/concurrency/executor.h>
\r
15 #include <SFML/Window/Context.hpp>
\r
16 #include <unordered_map>
\r
18 namespace caspar { namespace core {
\r
20 image_transform::image_transform()
\r
23 , mode_(video_mode::invalid)
\r
25 std::fill(pos_.begin(), pos_.end(), 0.0);
\r
26 std::fill(size_.begin(), size_.end(), 1.0);
\r
27 std::fill(uv_.begin(), uv_.end(), 0.0);
\r
30 void image_transform::set_opacity(double value)
\r
32 tbb::spin_mutex::scoped_lock(mutex_);
\r
36 double image_transform::get_opacity() const
\r
38 tbb::spin_mutex::scoped_lock(mutex_);
\r
42 void image_transform::set_gain(double value)
\r
44 tbb::spin_mutex::scoped_lock(mutex_);
\r
48 double image_transform::get_gain() const
\r
50 tbb::spin_mutex::scoped_lock(mutex_);
\r
54 void image_transform::set_position(double x, double y)
\r
56 std::array<double, 2> value = {x, y};
\r
57 tbb::spin_mutex::scoped_lock(mutex_);
\r
61 std::array<double, 2> image_transform::get_position() const
\r
63 tbb::spin_mutex::scoped_lock(mutex_);
\r
67 void image_transform::set_size(double x, double y)
\r
69 std::array<double, 2> value = {x, y};
\r
70 tbb::spin_mutex::scoped_lock(mutex_);
\r
74 std::array<double, 2> image_transform::get_size() const
\r
76 tbb::spin_mutex::scoped_lock(mutex_);
\r
80 void image_transform::set_uv(double left, double top, double right, double bottom)
\r
82 std::array<double, 4> value = {left, top, right, bottom};
\r
83 tbb::spin_mutex::scoped_lock(mutex_);
\r
87 std::array<double, 4> image_transform::get_uv() const
\r
89 tbb::spin_mutex::scoped_lock(mutex_);
\r
93 void image_transform::set_mode(video_mode::type mode)
\r
95 tbb::spin_mutex::scoped_lock(mutex_);
\r
99 video_mode::type image_transform::get_mode() const
\r
101 tbb::spin_mutex::scoped_lock(mutex_);
\r
106 image_transform& image_transform::operator*=(const image_transform &other)
\r
108 tbb::spin_mutex::scoped_lock(mutex_);
\r
109 opacity_ *= other.opacity_;
\r
110 mode_ = other.mode_;
\r
111 gain_ *= other.gain_;
\r
112 uv_[0] += other.uv_[0];
\r
113 uv_[1] += other.uv_[1];
\r
114 uv_[2] += other.uv_[2];
\r
115 uv_[3] += other.uv_[3];
\r
116 pos_[0] += other.pos_[0];
\r
117 pos_[1] += other.pos_[1];
\r
118 size_[0] *= other.size_[0];
\r
119 size_[1] *= other.size_[1];
\r
123 const image_transform image_transform::operator*(const image_transform &other) const
\r
125 return image_transform(*this) *= other;
\r
128 struct image_mixer::implementation : boost::noncopyable
\r
130 const video_format_desc format_desc_;
\r
132 std::stack<image_transform> transform_stack_;
\r
135 std::array<std::shared_ptr<device_buffer>, 2> render_targets_;
\r
137 std::shared_ptr<host_buffer> reading_;
\r
139 image_kernel kernel_;
\r
141 safe_ptr<ogl_device> context_;
\r
144 implementation(const video_format_desc& format_desc)
\r
145 : format_desc_(format_desc)
\r
146 , context_(ogl_device::create())
\r
148 context_->begin_invoke([=]
\r
150 transform_stack_.push(image_transform());
\r
151 transform_stack_.top().set_mode(video_mode::progressive);
\r
152 transform_stack_.top().set_uv(0.0, 1.0, 1.0, 0.0);
\r
154 GL(glEnable(GL_TEXTURE_2D));
\r
155 GL(glDisable(GL_DEPTH_TEST));
\r
157 render_targets_[0] = context_->create_device_buffer(format_desc.width, format_desc.height, 4);
\r
158 render_targets_[1] = context_->create_device_buffer(format_desc.width, format_desc.height, 4);
\r
160 GL(glGenFramebuffers(1, &fbo_));
\r
161 GL(glBindFramebuffer(GL_FRAMEBUFFER_EXT, fbo_));
\r
162 GL(glReadBuffer(GL_COLOR_ATTACHMENT0_EXT));
\r
164 reading_ = context_->create_host_buffer(format_desc_.size, host_buffer::read_only);
\r
170 glDeleteFramebuffersEXT(1, &fbo_);
\r
173 void begin(const image_transform& transform)
\r
175 transform_stack_.push(transform_stack_.top()*transform);
\r
178 void render(const pixel_format_desc& desc, std::vector<safe_ptr<host_buffer>>& buffers)
\r
180 auto transform = transform_stack_.top();
\r
181 context_->begin_invoke([=]
\r
183 GL(glColor4d(1.0, 1.0, 1.0, transform.get_opacity()));
\r
184 GL(glViewport(0, 0, format_desc_.width, format_desc_.height));
\r
185 kernel_.apply(desc.pix_fmt, transform);
\r
187 std::vector<safe_ptr<device_buffer>> device_buffers;
\r
188 for(size_t n = 0; n < buffers.size(); ++n)
\r
190 auto texture = context_->create_device_buffer(desc.planes[n].width, desc.planes[n].height, desc.planes[n].channels);
\r
191 texture->read(*buffers[n]);
\r
192 device_buffers.push_back(texture);
\r
195 for(size_t n = 0; n < buffers.size(); ++n)
\r
197 glActiveTexture(GL_TEXTURE0+n);
\r
198 device_buffers[n]->bind();
\r
201 auto& p = transform.get_position();
\r
202 auto& s = transform.get_size();
\r
203 auto& uv = transform.get_uv();
\r
206 glTexCoord2d(uv[0], uv[3]); glVertex2d( p[0]*2.0-1.0, p[1]*2.0-1.0);
\r
207 glTexCoord2d(uv[2], uv[3]); glVertex2d((p[0]+s[0])*2.0-1.0, p[1]*2.0-1.0);
\r
208 glTexCoord2d(uv[2], uv[1]); glVertex2d((p[0]+s[0])*2.0-1.0, (p[1]+s[1])*2.0-1.0);
\r
209 glTexCoord2d(uv[0], uv[1]); glVertex2d( p[0]*2.0-1.0, (p[1]+s[1])*2.0-1.0);
\r
216 transform_stack_.pop();
\r
219 boost::unique_future<safe_ptr<const host_buffer>> begin_pass()
\r
221 return context_->begin_invoke([=]() -> safe_ptr<const host_buffer>
\r
224 render_targets_[0]->attach(0);
\r
225 GL(glClear(GL_COLOR_BUFFER_BIT));
\r
226 return safe_ptr<const host_buffer>(reading_);
\r
232 context_->begin_invoke([=]
\r
234 reading_ = context_->create_host_buffer(format_desc_.size, host_buffer::read_only);
\r
235 render_targets_[0]->write(*reading_);
\r
236 std::rotate(render_targets_.begin(), render_targets_.begin() + 1, render_targets_.end());
\r
240 std::vector<safe_ptr<host_buffer>> create_buffers(const pixel_format_desc& format)
\r
242 std::vector<safe_ptr<host_buffer>> buffers;
\r
243 std::transform(format.planes.begin(), format.planes.end(), std::back_inserter(buffers), [&](const pixel_format_desc::plane& plane) -> safe_ptr<host_buffer>
\r
245 return context_->create_host_buffer(plane.size, host_buffer::write_only);
\r
251 image_mixer::image_mixer(const video_format_desc& format_desc) : impl_(new implementation(format_desc)){}
\r
252 void image_mixer::begin(const image_transform& transform) { impl_->begin(transform);}
\r
253 void image_mixer::process(const pixel_format_desc& desc, std::vector<safe_ptr<host_buffer>>& buffers){ impl_->render(desc, buffers);}
\r
254 void image_mixer::end(){impl_->end();}
\r
255 boost::unique_future<safe_ptr<const host_buffer>> image_mixer::begin_pass(){ return impl_->begin_pass();}
\r
256 void image_mixer::end_pass(){impl_->end_pass();}
\r
257 std::vector<safe_ptr<host_buffer>> image_mixer::create_buffers(const pixel_format_desc& format){return impl_->create_buffers(format);}
\r