]> git.sesse.net Git - casparcg/blob - core/mixer/image/image_kernel.cpp
2.0. ffmpeg_producer: Fixed integer overflow when looping.
[casparcg] / core / mixer / image / image_kernel.cpp
1 /*\r
2 * copyright (c) 2010 Sveriges Television AB <info@casparcg.com>\r
3 *\r
4 *  This file is part of CasparCG.\r
5 *\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
10 *\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
15 \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
18 *\r
19 */\r
20 #include "../../stdafx.h"\r
21 \r
22 #include "image_kernel.h"\r
23 \r
24 #include "image_shader.h"\r
25 \r
26 #include "blending_glsl.h"\r
27 #include "../gpu/shader.h"\r
28 #include "../gpu/device_buffer.h"\r
29 #include "../gpu/ogl_device.h"\r
30 \r
31 #include <common/exception/exceptions.h>\r
32 #include <common/gl/gl_check.h>\r
33 #include <common/env.h>\r
34 \r
35 #include <core/video_format.h>\r
36 #include <core/producer/frame/pixel_format.h>\r
37 #include <core/producer/frame/frame_transform.h>\r
38 \r
39 #include <GL/glew.h>\r
40 \r
41 #include <boost/noncopyable.hpp>\r
42 \r
43 #include <unordered_map>\r
44 \r
45 namespace caspar { namespace core {\r
46         \r
47 GLubyte upper_pattern[] = {\r
48         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
49         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
50         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
51         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
52                 \r
53 GLubyte lower_pattern[] = {\r
54         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
55         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
56         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
57         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
58 \r
59 struct image_kernel::implementation : boost::noncopyable\r
60 {       \r
61         std::shared_ptr<shader> shader_;\r
62         bool                                    blend_modes_;\r
63                                                         \r
64         void draw(ogl_device& ogl, draw_params&& params)\r
65         {\r
66                 static const double epsilon = 0.001;\r
67 \r
68                 CASPAR_ASSERT(params.pix_desc.planes.size() == params.textures.size());\r
69 \r
70                 if(params.textures.empty() || !params.background)\r
71                         return;\r
72 \r
73                 if(params.transform.opacity < epsilon)\r
74                         return;\r
75                 \r
76                 if(!std::all_of(params.textures.begin(), params.textures.end(), std::mem_fn(&device_buffer::ready)))\r
77                 {\r
78                         CASPAR_LOG(warning) << L"[image_mixer] Performance warning. Host to device transfer not complete, GPU will be stalled";\r
79                         ogl.yield(); // Try to give it some more time.\r
80                 }               \r
81                 \r
82                 // Bind textures\r
83 \r
84                 for(size_t n = 0; n < params.textures.size(); ++n)\r
85                         params.textures[n]->bind(n);\r
86 \r
87                 if(params.local_key)\r
88                         params.local_key->bind(texture_id::local_key);\r
89                 \r
90                 if(params.layer_key)\r
91                         params.layer_key->bind(texture_id::layer_key);\r
92                         \r
93                 // Setup shader\r
94 \r
95                 if(!shader_)\r
96                         shader_ = get_image_shader(ogl, blend_modes_);\r
97                                                 \r
98                 ogl.use(*shader_);\r
99 \r
100                 shader_->set("plane[0]",                texture_id::plane0);\r
101                 shader_->set("plane[1]",                texture_id::plane1);\r
102                 shader_->set("plane[2]",                texture_id::plane2);\r
103                 shader_->set("plane[3]",                texture_id::plane3);\r
104                 shader_->set("local_key",               texture_id::local_key);\r
105                 shader_->set("layer_key",               texture_id::layer_key);\r
106                 shader_->set("is_hd",                   params.pix_desc.planes.at(0).height > 700 ? 1 : 0);\r
107                 shader_->set("has_local_key",   params.local_key);\r
108                 shader_->set("has_layer_key",   params.layer_key);\r
109                 shader_->set("pixel_format",    params.pix_desc.pix_fmt);       \r
110                 shader_->set("opacity",                 params.transform.is_key ? 1.0 : params.transform.opacity);      \r
111                 \r
112                 // Setup blend_func\r
113                 \r
114                 if(params.transform.is_key)\r
115                         params.blend_mode = blend_mode::normal;\r
116 \r
117                 if(blend_modes_)\r
118                 {\r
119                         params.background->bind(6);\r
120 \r
121                         shader_->set("background",      texture_id::background);\r
122                         shader_->set("blend_mode",      params.blend_mode);\r
123                 }\r
124                 else\r
125                 {\r
126                         ogl.blend_func_separate(GL_ONE, GL_ONE_MINUS_SRC_ALPHA, GL_ONE, GL_ONE_MINUS_SRC_ALPHA);                        \r
127                 }\r
128 \r
129                 // Setup image-adjustements\r
130                 \r
131                 if(params.transform.levels.min_input  > epsilon         ||\r
132                    params.transform.levels.max_input  < 1.0-epsilon     ||\r
133                    params.transform.levels.min_output > epsilon         ||\r
134                    params.transform.levels.max_output < 1.0-epsilon     ||\r
135                    std::abs(params.transform.levels.gamma - 1.0) > epsilon)\r
136                 {\r
137                         shader_->set("levels", true);   \r
138                         shader_->set("min_input",       params.transform.levels.min_input);     \r
139                         shader_->set("max_input",       params.transform.levels.max_input);\r
140                         shader_->set("min_output",      params.transform.levels.min_output);\r
141                         shader_->set("max_output",      params.transform.levels.max_output);\r
142                         shader_->set("gamma",           params.transform.levels.gamma);\r
143                 }\r
144                 else\r
145                         shader_->set("levels", false);  \r
146 \r
147                 if(std::abs(params.transform.brightness - 1.0) > epsilon ||\r
148                    std::abs(params.transform.saturation - 1.0) > epsilon ||\r
149                    std::abs(params.transform.contrast - 1.0)   > epsilon)\r
150                 {\r
151                         shader_->set("csb",     true);  \r
152                         \r
153                         shader_->set("brt", params.transform.brightness);       \r
154                         shader_->set("sat", params.transform.saturation);\r
155                         shader_->set("con", params.transform.contrast);\r
156                 }\r
157                 else\r
158                         shader_->set("csb",     false); \r
159                 \r
160                 // Setup interlacing\r
161 \r
162                 if(params.transform.field_mode == core::field_mode::progressive)                        \r
163                         ogl.disable(GL_POLYGON_STIPPLE);                        \r
164                 else                    \r
165                 {\r
166                         ogl.enable(GL_POLYGON_STIPPLE);\r
167 \r
168                         if(params.transform.field_mode == core::field_mode::upper)\r
169                                 ogl.stipple_pattern(upper_pattern);\r
170                         else if(params.transform.field_mode == core::field_mode::lower)\r
171                                 ogl.stipple_pattern(lower_pattern);\r
172                 }\r
173 \r
174                 // Setup drawing area\r
175                 \r
176                 ogl.viewport(0, 0, params.background->width(), params.background->height());\r
177                                                                 \r
178                 auto m_p = params.transform.clip_translation;\r
179                 auto m_s = params.transform.clip_scale;\r
180 \r
181                 bool scissor = m_p[0] > std::numeric_limits<double>::epsilon()                  || m_p[1] > std::numeric_limits<double>::epsilon() ||\r
182                                            m_s[0] < (1.0 - std::numeric_limits<double>::epsilon())      || m_s[1] < (1.0 - std::numeric_limits<double>::epsilon());\r
183 \r
184                 if(scissor)\r
185                 {\r
186                         double w = static_cast<double>(params.background->width());\r
187                         double h = static_cast<double>(params.background->height());\r
188                 \r
189                         ogl.enable(GL_SCISSOR_TEST);\r
190                         ogl.scissor(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
191                 }\r
192 \r
193                 auto f_p = params.transform.fill_translation;\r
194                 auto f_s = params.transform.fill_scale;\r
195                 \r
196                 // Set render target\r
197                 \r
198                 ogl.attach(*params.background);\r
199                 \r
200                 // Draw\r
201 \r
202                 glBegin(GL_QUADS);\r
203                         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
204                         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
205                         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
206                         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
207                 glEnd();\r
208 \r
209                 // Cleanup\r
210 \r
211                 ogl.disable(GL_SCISSOR_TEST);   \r
212                                 \r
213                 params.textures.clear();\r
214                 ogl.yield(); // Return resources to pool as early as possible.\r
215 \r
216                 if(blend_modes_)\r
217                 {\r
218                         // http://www.opengl.org/registry/specs/NV/texture_barrier.txt\r
219                         // This allows us to use framebuffer (background) both as source and target while blending.\r
220                         glTextureBarrierNV(); \r
221                 }\r
222         }\r
223 };\r
224 \r
225 image_kernel::image_kernel() : impl_(new implementation()){}\r
226 void image_kernel::draw(ogl_device& ogl, draw_params&& params)\r
227 {\r
228         impl_->draw(ogl, std::move(params));\r
229 }\r
230 \r
231 }}