4 * Copyright (C) 2008 Adam Williams <broadcast at earthling dot net>
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
23 #include "bccmodels.h"
27 #include "rgb601window.h"
33 REGISTER_PLUGIN(RGB601Main)
38 RGB601Config::RGB601Config()
43 RGB601Main::RGB601Main(PluginServer *server)
44 : PluginVClient(server)
49 RGB601Main::~RGB601Main()
54 const char* RGB601Main::plugin_title() { return N_("RGB - 601"); }
55 int RGB601Main::is_realtime() { return 1; }
58 NEW_WINDOW_MACRO(RGB601Main, RGB601Window)
61 void RGB601Main::update_gui()
66 thread->window->lock_window();
67 ((RGB601Window*)thread->window)->forward->update(config.direction == 1);
68 ((RGB601Window*)thread->window)->reverse->update(config.direction == 2);
69 thread->window->unlock_window();
74 int RGB601Main::load_configuration()
76 KeyFrame *prev_keyframe;
78 prev_keyframe = get_prev_keyframe(get_source_position());
79 // Must also switch between interpolation between keyframes and using first keyframe
80 read_data(prev_keyframe);
85 void RGB601Main::save_data(KeyFrame *keyframe)
89 // cause data to be stored directly in text
90 output.set_shared_output(keyframe->get_data(), MESSAGESIZE);
91 output.tag.set_title("RGB601");
92 output.tag.set_property("DIRECTION", config.direction);
94 output.tag.set_title("/RGB601");
96 output.append_newline();
97 output.terminate_string();
100 void RGB601Main::read_data(KeyFrame *keyframe)
104 input.set_shared_input(keyframe->get_data(), strlen(keyframe->get_data()));
110 result = input.read_tag();
114 if(input.tag.title_is("RGB601"))
116 config.direction = input.tag.get_property("DIRECTION", config.direction);
122 #define CREATE_TABLE(max) \
124 for(int i = 0; i < max; i++) \
126 int forward_output = (int)((double)0.8588 * i + max * 0.0627 + 0.5); \
127 int reverse_output = (int)((double)1.1644 * i - max * 0.0627 + 0.5); \
128 forward_table[i] = CLIP(forward_output, 0, max - 1); \
129 reverse_table[i] = CLIP(reverse_output, 0, max - 1); \
133 void RGB601Main::create_table(VFrame *input_ptr)
135 switch(input_ptr->get_color_model())
146 case BC_RGBA16161616:
147 case BC_YUVA16161616:
148 CREATE_TABLE(0x10000);
153 #define PROCESS(table, type, components, yuv) \
155 for(int i = 0; i < h; i++) \
157 type *in_row = (type*)input_ptr->get_rows()[i]; \
158 type *out_row = (type*)output_ptr->get_rows()[i]; \
163 for(int j = 0; j < w; j++) \
165 out_row[j * components] = table[(int)in_row[j * components]]; \
166 out_row[j * components + 1] = in_row[j * components + 1]; \
167 out_row[j * components + 2] = in_row[j * components + 2]; \
171 if(sizeof(type) == 4) \
173 for(int j = 0; j < w; j++) \
175 if(table == forward_table) \
177 out_row[j * components] = (type)(in_row[j * components] * 0.8588 + 0.0627); \
178 out_row[j * components + 1] = (type)(in_row[j * components + 1] * 0.8588 + 0.0627); \
179 out_row[j * components + 2] = (type)(in_row[j * components + 2] * 0.8588 + 0.0627); \
183 out_row[j * components] = (type)(in_row[j * components] * 1.1644 - 0.0627); \
184 out_row[j * components + 1] = (type)(in_row[j * components + 1] * 1.1644 - 0.0627); \
185 out_row[j * components + 2] = (type)(in_row[j * components + 2] * 1.1644 - 0.0627); \
191 for(int j = 0; j < w; j++) \
193 out_row[j * components] = table[(int)in_row[j * components]]; \
194 out_row[j * components + 1] = table[(int)in_row[j * components + 1]]; \
195 out_row[j * components + 2] = table[(int)in_row[j * components + 2]]; \
201 void RGB601Main::process(int *table, VFrame *input_ptr, VFrame *output_ptr)
203 int w = input_ptr->get_w();
204 int h = input_ptr->get_h();
206 if(config.direction == 1)
207 switch(input_ptr->get_color_model())
210 PROCESS(forward_table, unsigned char, 3, 1);
213 PROCESS(forward_table, unsigned char, 4, 1);
216 PROCESS(forward_table, u_int16_t, 3, 1);
218 case BC_YUVA16161616:
219 PROCESS(forward_table, u_int16_t, 4, 1);
222 PROCESS(forward_table, unsigned char, 3, 0);
225 PROCESS(forward_table, unsigned char, 4, 0);
228 PROCESS(forward_table, float, 3, 0);
231 PROCESS(forward_table, float, 4, 0);
234 PROCESS(forward_table, u_int16_t, 3, 0);
236 case BC_RGBA16161616:
237 PROCESS(forward_table, u_int16_t, 4, 0);
241 if(config.direction == 2)
242 switch(input_ptr->get_color_model())
245 PROCESS(reverse_table, unsigned char, 3, 1);
248 PROCESS(reverse_table, unsigned char, 4, 1);
251 PROCESS(reverse_table, u_int16_t, 3, 1);
253 case BC_YUVA16161616:
254 PROCESS(reverse_table, u_int16_t, 4, 1);
257 PROCESS(reverse_table, unsigned char, 3, 0);
260 PROCESS(reverse_table, unsigned char, 4, 0);
263 PROCESS(reverse_table, float, 3, 0);
266 PROCESS(reverse_table, float, 4, 0);
269 PROCESS(reverse_table, u_int16_t, 3, 0);
271 case BC_RGBA16161616:
272 PROCESS(reverse_table, u_int16_t, 4, 0);
277 int RGB601Main::process_buffer(VFrame *frame,
278 int64_t start_position,
281 load_configuration();
283 // Set parameters for aggregation with previous or next effect.
284 frame->get_params()->update("RGB601_DIRECTION", config.direction);
292 // Deinterlace effects may aggregate this one,
293 if(get_use_opengl() &&
294 (prev_effect_is(_("Frames to fields")) ||
295 next_effect_is(_("Frames to fields"))))
300 if(get_use_opengl() && config.direction)
309 if(config.direction == 1)
310 process(forward_table, frame, frame);
312 if(config.direction == 2)
313 process(reverse_table, frame, frame);
318 int RGB601Main::handle_opengl()
321 static const char *yuv_fwd_frag =
322 "uniform sampler2D tex;\n"
325 " gl_FragColor = texture2D(tex, gl_TexCoord[0].st);\n"
326 " gl_FragColor.r = gl_FragColor.r * 0.8588 + 0.0627;\n"
328 static const char *yuv_rev_frag =
329 "uniform sampler2D tex;\n"
332 " gl_FragColor = texture2D(tex, gl_TexCoord[0].st);\n"
333 " gl_FragColor.r = gl_FragColor.r * 1.1644 - 0.0627;\n"
335 static const char *rgb_fwd_frag =
336 "uniform sampler2D tex;\n"
339 " gl_FragColor = texture2D(tex, gl_TexCoord[0].st);\n"
340 " gl_FragColor.rgb = gl_FragColor.rgb * vec3(0.8588, 0.8588, 0.8588) + vec3(0.0627, 0.0627, 0.0627);\n"
342 static const char *rgb_rev_frag =
343 "uniform sampler2D tex;\n"
346 " gl_FragColor = texture2D(tex, gl_TexCoord[0].st);\n"
347 " gl_FragColor.rgb = gl_FragColor.rgb * vec3(1.1644, 1.1644, 1.1644) - vec3(0.0627, 0.0627, 0.0627);\n"
351 get_output()->to_texture();
352 get_output()->enable_opengl();
353 get_output()->bind_texture(0);
355 unsigned int frag_shader = 0;
356 switch( get_output()->get_color_model() ) {
359 frag_shader = VFrame::make_shader(0,
360 config.direction == 1 ? yuv_fwd_frag : yuv_rev_frag,
365 frag_shader = VFrame::make_shader(0,
366 config.direction == 1 ? rgb_fwd_frag : rgb_rev_frag,
371 if( frag_shader > 0 ) {
372 glUseProgram(frag_shader);
373 glUniform1i(glGetUniformLocation(frag_shader, "tex"), 0);
375 VFrame::init_screen(get_output()->get_w(), get_output()->get_h());
376 get_output()->draw_texture();
378 get_output()->set_opengl_state(VFrame::SCREEN);