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_kernel.h"
\r
23 #include "blending_glsl.h"
\r
24 #include "../gpu/shader.h"
\r
25 #include "../gpu/device_buffer.h"
\r
27 #include <common/exception/exceptions.h>
\r
28 #include <common/gl/gl_check.h>
\r
30 #include <core/video_format.h>
\r
31 #include <core/producer/frame/pixel_format.h>
\r
32 #include <core/producer/frame/image_transform.h>
\r
34 #include <boost/noncopyable.hpp>
\r
36 #include <unordered_map>
\r
38 namespace caspar { namespace core {
\r
40 GLubyte upper_pattern[] = {
\r
41 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00,
\r
42 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00,
\r
43 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00,
\r
44 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00};
\r
46 GLubyte lower_pattern[] = {
\r
47 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff,
\r
48 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff,
\r
49 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff,
\r
50 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff};
\r
52 struct image_kernel::implementation : boost::noncopyable
\r
54 std::unique_ptr<shader> shader_;
\r
56 std::string vertex_;
\r
57 std::string fragment_;
\r
59 core::video_mode::type last_mode_;
\r
61 size_t last_height_;
\r
64 : last_mode_(core::video_mode::progressive)
\r
71 " gl_TexCoord[0] = gl_MultiTexCoord0; \n"
\r
72 " gl_TexCoord[1] = gl_MultiTexCoord1; \n"
\r
73 " gl_FrontColor = gl_Color; \n"
\r
74 " gl_Position = ftransform(); \n"
\r
79 "uniform sampler2D background; \n"
\r
80 "uniform sampler2D plane[4]; \n"
\r
81 "uniform sampler2D local_key; \n"
\r
82 "uniform sampler2D layer_key; \n"
\r
84 "uniform bool is_hd; \n"
\r
85 "uniform bool has_local_key; \n"
\r
86 "uniform bool has_layer_key; \n"
\r
87 "uniform int blend_mode; \n"
\r
88 "uniform int alpha_mode; \n"
\r
89 "uniform int pixel_format; \n"
\r
91 "uniform bool levels; \n"
\r
92 "uniform float min_input; \n"
\r
93 "uniform float max_input; \n"
\r
94 "uniform float gamma; \n"
\r
95 "uniform float min_output; \n"
\r
96 "uniform float max_output; \n"
\r
98 "uniform bool csb; \n"
\r
99 "uniform float brt; \n"
\r
100 "uniform float sat; \n"
\r
101 "uniform float con; \n"
\r
110 "vec3 get_blend_color(vec3 back, vec3 fore) \n"
\r
112 " switch(blend_mode) \n"
\r
114 " case 0: return BlendNormal(back, fore); \n"
\r
115 " case 1: return BlendLighten(back, fore); \n"
\r
116 " case 2: return BlendDarken(back, fore); \n"
\r
117 " case 3: return BlendMultiply(back, fore); \n"
\r
118 " case 4: return BlendAverage(back, fore); \n"
\r
119 " case 5: return BlendAdd(back, fore); \n"
\r
120 " case 6: return BlendSubstract(back, fore); \n"
\r
121 " case 7: return BlendDifference(back, fore); \n"
\r
122 " case 8: return BlendNegation(back, fore); \n"
\r
123 " case 9: return BlendExclusion(back, fore); \n"
\r
124 " case 10: return BlendScreen(back, fore); \n"
\r
125 " case 11: return BlendOverlay(back, fore); \n"
\r
126 //" case 12: return BlendSoftLight(back, fore); \n"
\r
127 " case 13: return BlendHardLight(back, fore); \n"
\r
128 " case 14: return BlendColorDodge(back, fore); \n"
\r
129 " case 15: return BlendColorBurn(back, fore); \n"
\r
130 " case 16: return BlendLinearDodge(back, fore); \n"
\r
131 " case 17: return BlendLinearBurn(back, fore); \n"
\r
132 " case 18: return BlendLinearLight(back, fore); \n"
\r
133 " case 19: return BlendVividLight(back, fore); \n"
\r
134 " case 20: return BlendPinLight(back, fore); \n"
\r
135 " case 21: return BlendHardMix(back, fore); \n"
\r
136 " case 22: return BlendReflect(back, fore); \n"
\r
137 " case 23: return BlendGlow(back, fore); \n"
\r
138 " case 24: return BlendPhoenix(back, fore); \n"
\r
139 " case 25: return BlendHue(back, fore); \n"
\r
140 " case 26: return BlendSaturation(back, fore); \n"
\r
141 " case 27: return BlendColor(back, fore); \n"
\r
142 " case 28: return BlendLuminosity(back, fore); \n"
\r
145 " return BlendNormal(back, fore); \n"
\r
148 "vec4 blend_color(vec4 fore) \n"
\r
150 " vec4 back = texture2D(background, gl_TexCoord[1].st); \n"
\r
152 " fore.rgb = LevelsControl(fore.rgb, min_input, max_input, gamma, min_output, max_output); \n"
\r
154 " fore.rgb = ContrastSaturationBrightness(fore.rgb, brt, sat, con); \n"
\r
155 " fore.rgb = get_blend_color(back.bgr, fore.rgb); \n"
\r
157 " return vec4(mix(back.rgb, fore.rgb, fore.a), back.a + fore.a); \n"
\r
160 "// NOTE: YCbCr, ITU-R, http://www.intersil.com/data/an/an9717.pdf \n"
\r
161 "// TODO: Support for more yuv formats might be needed. \n"
\r
162 "vec4 ycbcra_to_rgba_sd(float y, float cb, float cr, float a) \n"
\r
166 " y = 1.164*(y-0.0625); \n"
\r
169 " color.r = y + 2.018 * cb; \n"
\r
170 " color.b = y + 1.596 * cr; \n"
\r
171 " color.g = y - 0.813 * cr - 0.391 * cb; \n"
\r
174 " return color; \n"
\r
177 "vec4 ycbcra_to_rgba_hd(float y, float cb, float cr, float a) \n"
\r
181 " y = 1.164*(y-0.0625); \n"
\r
184 " color.r = y + 2.115 * cb; \n"
\r
185 " color.b = y + 1.793 * cr; \n"
\r
186 " color.g = y - 0.534 * cr - 0.213 * cb; \n"
\r
189 " return color; \n"
\r
192 "vec4 ycbcra_to_rgba(float y, float cb, float cr, float a) \n"
\r
195 " return ycbcra_to_rgba_hd(y, cb, cr, a); \n"
\r
197 " return ycbcra_to_rgba_sd(y, cb, cr, a); \n"
\r
200 "vec4 get_rgba_color() \n"
\r
202 " switch(pixel_format) \n"
\r
204 " case 0: //gray \n"
\r
205 " return vec4(texture2D(plane[0], gl_TexCoord[0].st).rrr, 1.0); \n"
\r
206 " case 1: //bgra, \n"
\r
207 " return texture2D(plane[0], gl_TexCoord[0].st).bgra; \n"
\r
208 " case 2: //rgba, \n"
\r
209 " return texture2D(plane[0], gl_TexCoord[0].st).rgba; \n"
\r
210 " case 3: //argb, \n"
\r
211 " return texture2D(plane[0], gl_TexCoord[0].st).argb; \n"
\r
212 " case 4: //abgr, \n"
\r
213 " return texture2D(plane[0], gl_TexCoord[0].st).gbar; \n"
\r
214 " case 5: //ycbcr, \n"
\r
216 " float y = texture2D(plane[0], gl_TexCoord[0].st).r; \n"
\r
217 " float cb = texture2D(plane[1], gl_TexCoord[0].st).r; \n"
\r
218 " float cr = texture2D(plane[2], gl_TexCoord[0].st).r; \n"
\r
219 " return ycbcra_to_rgba(y, cb, cr, 1.0); \n"
\r
221 " case 6: //ycbcra \n"
\r
223 " float y = texture2D(plane[0], gl_TexCoord[0].st).r; \n"
\r
224 " float cb = texture2D(plane[1], gl_TexCoord[0].st).r; \n"
\r
225 " float cr = texture2D(plane[2], gl_TexCoord[0].st).r; \n"
\r
226 " float a = texture2D(plane[3], gl_TexCoord[0].st).r; \n"
\r
227 " return ycbcra_to_rgba(y, cb, cr, a); \n"
\r
230 " return vec4(0.0, 0.0, 0.0, 0.0); \n"
\r
235 " vec4 color = get_rgba_color(); \n"
\r
236 " if(has_local_key) \n"
\r
237 " color.a *= texture2D(local_key, gl_TexCoord[1].st).r; \n"
\r
238 " if(has_layer_key) \n"
\r
239 " color.a *= texture2D(layer_key, gl_TexCoord[1].st).r; \n"
\r
240 " gl_FragColor = blend_color(color.bgra * gl_Color); \n"
\r
245 void draw(const render_item& item,
\r
246 const safe_ptr<device_buffer>& background,
\r
247 const std::shared_ptr<device_buffer>& local_key,
\r
248 const std::shared_ptr<device_buffer>& layer_key)
\r
250 static const double epsilon = 0.001;
\r
252 CASPAR_ASSERT(item.pix_desc.planes.size() == item.textures.size());
\r
254 if(item.textures.empty())
\r
257 if(item.transform.get_opacity() < epsilon)
\r
262 shader_.reset(new shader(vertex_, fragment_));
\r
263 GL(glEnable(GL_TEXTURE_2D));
\r
266 if(last_mode_ != item.mode)
\r
268 last_mode_ = item.mode;
\r
270 if(item.mode == core::video_mode::progressive)
\r
271 GL(glDisable(GL_POLYGON_STIPPLE));
\r
274 GL(glEnable(GL_POLYGON_STIPPLE));
\r
276 if(item.mode == core::video_mode::upper)
\r
277 glPolygonStipple(upper_pattern);
\r
278 else if(item.mode == core::video_mode::lower)
\r
279 glPolygonStipple(lower_pattern);
\r
283 if(last_width_ != background->width() || last_height_ != background->height())
\r
285 last_width_ = background->width();
\r
286 last_height_ = background->height();
\r
287 GL(glViewport(0, 0, background->width(), background->height()));
\r
292 for(size_t n = 0; n < item.textures.size(); ++n)
\r
293 item.textures[n]->bind(n);
\r
296 local_key->bind(4);
\r
299 layer_key->bind(5);
\r
301 background->bind(6);
\r
307 shader_->set("plane[0]", 0);
\r
308 shader_->set("plane[1]", 1);
\r
309 shader_->set("plane[2]", 2);
\r
310 shader_->set("plane[3]", 3);
\r
311 shader_->set("local_key", 4);
\r
312 shader_->set("layer_key", 5);
\r
313 shader_->set("background", 6);
\r
314 shader_->set("is_hd", item.pix_desc.planes.at(0).height > 700 ? 1 : 0);
\r
315 shader_->set("has_local_key", local_key ? 1 : 0);
\r
316 shader_->set("has_layer_key", layer_key ? 1 : 0);
\r
317 shader_->set("blend_mode", item.transform.get_is_key() ? core::image_transform::blend_mode::normal : item.transform.get_blend_mode());
\r
318 shader_->set("alpha_mode", item.transform.get_alpha_mode());
\r
319 shader_->set("pixel_format", item.pix_desc.pix_fmt);
\r
321 auto levels = item.transform.get_levels();
\r
323 if(levels.min_input > epsilon ||
\r
324 levels.max_input < 1.0-epsilon ||
\r
325 levels.min_output > epsilon ||
\r
326 levels.max_output < 1.0-epsilon ||
\r
327 std::abs(levels.gamma - 1.0) > epsilon)
\r
329 shader_->set("levels", true);
\r
330 shader_->set("min_input", levels.min_input);
\r
331 shader_->set("max_input", levels.max_input);
\r
332 shader_->set("min_output", levels.min_output);
\r
333 shader_->set("max_output", levels.max_output);
\r
334 shader_->set("gamma", levels.gamma);
\r
337 shader_->set("levels", false);
\r
339 if(std::abs(item.transform.get_brightness() - 1.0) > epsilon ||
\r
340 std::abs(item.transform.get_saturation() - 1.0) > epsilon ||
\r
341 std::abs(item.transform.get_contrast() - 1.0) > epsilon)
\r
343 shader_->set("csb", true);
\r
345 shader_->set("brt", item.transform.get_brightness());
\r
346 shader_->set("sat", item.transform.get_saturation());
\r
347 shader_->set("con", item.transform.get_contrast());
\r
350 shader_->set("csb", false);
\r
352 // Setup drawing area
\r
354 GL(glColor4d(item.transform.get_gain(), item.transform.get_gain(), item.transform.get_gain(), item.transform.get_opacity()));
\r
356 auto m_p = item.transform.get_clip_translation();
\r
357 auto m_s = item.transform.get_clip_scale();
\r
358 double w = static_cast<double>(background->width());
\r
359 double h = static_cast<double>(background->height());
\r
361 GL(glEnable(GL_SCISSOR_TEST));
\r
362 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
364 auto f_p = item.transform.get_fill_translation();
\r
365 auto f_s = item.transform.get_fill_scale();
\r
370 glMultiTexCoord2d(GL_TEXTURE0, 0.0, 0.0); glMultiTexCoord2d(GL_TEXTURE1, f_p[0] , f_p[1] ); glVertex2d( f_p[0] *2.0-1.0, f_p[1] *2.0-1.0);
\r
371 glMultiTexCoord2d(GL_TEXTURE0, 1.0, 0.0); glMultiTexCoord2d(GL_TEXTURE1, (f_p[0]+f_s[0]), f_p[1] ); glVertex2d((f_p[0]+f_s[0])*2.0-1.0, f_p[1] *2.0-1.0);
\r
372 glMultiTexCoord2d(GL_TEXTURE0, 1.0, 1.0); glMultiTexCoord2d(GL_TEXTURE1, (f_p[0]+f_s[0]), (f_p[1]+f_s[1])); glVertex2d((f_p[0]+f_s[0])*2.0-1.0, (f_p[1]+f_s[1])*2.0-1.0);
\r
373 glMultiTexCoord2d(GL_TEXTURE0, 0.0, 1.0); glMultiTexCoord2d(GL_TEXTURE1, f_p[0] , (f_p[1]+f_s[1])); glVertex2d( f_p[0] *2.0-1.0, (f_p[1]+f_s[1])*2.0-1.0);
\r
376 GL(glDisable(GL_SCISSOR_TEST));
\r
380 image_kernel::image_kernel() : impl_(new implementation()){}
\r
381 void image_kernel::draw(const render_item& item, const safe_ptr<device_buffer>& background, const std::shared_ptr<device_buffer>& local_key, const std::shared_ptr<device_buffer>& layer_key)
\r
383 impl_->draw(item, background, local_key, layer_key);
\r
386 bool operator==(const render_item& lhs, const render_item& rhs)
\r
388 return lhs.textures == rhs.textures && lhs.transform == rhs.transform && lhs.tag == rhs.tag && lhs.mode == rhs.mode;
\r