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 _("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.terminate_string();
97 void RGB601Main::read_data(KeyFrame *keyframe)
101 input.set_shared_input(keyframe->get_data(), strlen(keyframe->get_data()));
107 result = input.read_tag();
111 if(input.tag.title_is("RGB601"))
113 config.direction = input.tag.get_property("DIRECTION", config.direction);
120 ((RGB601Window*)thread->window)->update();
125 #define CREATE_TABLE(max) \
127 for(int i = 0; i < max; i++) \
129 int forward_output = (int)((double)0.8588 * i + max * 0.0627 + 0.5); \
130 int reverse_output = (int)((double)1.1644 * i - max * 0.0627 + 0.5); \
131 forward_table[i] = CLIP(forward_output, 0, max - 1); \
132 reverse_table[i] = CLIP(reverse_output, 0, max - 1); \
136 void RGB601Main::create_table(VFrame *input_ptr)
138 switch(input_ptr->get_color_model())
149 case BC_RGBA16161616:
150 case BC_YUVA16161616:
151 CREATE_TABLE(0x10000);
156 #define PROCESS(table, type, components, yuv) \
158 for(int i = 0; i < h; i++) \
160 type *in_row = (type*)input_ptr->get_rows()[i]; \
161 type *out_row = (type*)output_ptr->get_rows()[i]; \
166 for(int j = 0; j < w; j++) \
168 out_row[j * components] = table[(int)in_row[j * components]]; \
169 out_row[j * components + 1] = in_row[j * components + 1]; \
170 out_row[j * components + 2] = in_row[j * components + 2]; \
174 if(sizeof(type) == 4) \
176 for(int j = 0; j < w; j++) \
178 if(table == forward_table) \
180 out_row[j * components] = (type)(in_row[j * components] * 0.8588 + 0.0627); \
181 out_row[j * components + 1] = (type)(in_row[j * components + 1] * 0.8588 + 0.0627); \
182 out_row[j * components + 2] = (type)(in_row[j * components + 2] * 0.8588 + 0.0627); \
186 out_row[j * components] = (type)(in_row[j * components] * 1.1644 - 0.0627); \
187 out_row[j * components + 1] = (type)(in_row[j * components + 1] * 1.1644 - 0.0627); \
188 out_row[j * components + 2] = (type)(in_row[j * components + 2] * 1.1644 - 0.0627); \
194 for(int j = 0; j < w; j++) \
196 out_row[j * components] = table[(int)in_row[j * components]]; \
197 out_row[j * components + 1] = table[(int)in_row[j * components + 1]]; \
198 out_row[j * components + 2] = table[(int)in_row[j * components + 2]]; \
204 void RGB601Main::process(int *table, VFrame *input_ptr, VFrame *output_ptr)
206 int w = input_ptr->get_w();
207 int h = input_ptr->get_h();
209 if(config.direction == 1)
210 switch(input_ptr->get_color_model())
213 PROCESS(forward_table, unsigned char, 3, 1);
216 PROCESS(forward_table, unsigned char, 4, 1);
219 PROCESS(forward_table, u_int16_t, 3, 1);
221 case BC_YUVA16161616:
222 PROCESS(forward_table, u_int16_t, 4, 1);
225 PROCESS(forward_table, unsigned char, 3, 0);
228 PROCESS(forward_table, unsigned char, 4, 0);
231 PROCESS(forward_table, float, 3, 0);
234 PROCESS(forward_table, float, 4, 0);
237 PROCESS(forward_table, u_int16_t, 3, 0);
239 case BC_RGBA16161616:
240 PROCESS(forward_table, u_int16_t, 4, 0);
244 if(config.direction == 2)
245 switch(input_ptr->get_color_model())
248 PROCESS(reverse_table, unsigned char, 3, 1);
251 PROCESS(reverse_table, unsigned char, 4, 1);
254 PROCESS(reverse_table, u_int16_t, 3, 1);
256 case BC_YUVA16161616:
257 PROCESS(reverse_table, u_int16_t, 4, 1);
260 PROCESS(reverse_table, unsigned char, 3, 0);
263 PROCESS(reverse_table, unsigned char, 4, 0);
266 PROCESS(reverse_table, float, 3, 0);
269 PROCESS(reverse_table, float, 4, 0);
272 PROCESS(reverse_table, u_int16_t, 3, 0);
274 case BC_RGBA16161616:
275 PROCESS(reverse_table, u_int16_t, 4, 0);
280 int RGB601Main::process_buffer(VFrame *frame,
281 int64_t start_position,
284 load_configuration();
286 // Set parameters for aggregation with previous or next effect.
287 frame->get_params()->update("RGB601_DIRECTION", config.direction);
295 // Deinterlace effects may aggregate this one,
296 if(get_use_opengl() &&
297 (prev_effect_is(_("Frames to fields")) ||
298 next_effect_is(_("Frames to fields"))))
303 if(get_use_opengl() && config.direction)
312 if(config.direction == 1)
313 process(forward_table, frame, frame);
315 if(config.direction == 2)
316 process(reverse_table, frame, frame);
321 int RGB601Main::handle_opengl()
324 static const char *yuv_fwd_frag =
325 "uniform sampler2D tex;\n"
328 " gl_FragColor = texture2D(tex, gl_TexCoord[0].st);\n"
329 " gl_FragColor.r = gl_FragColor.r * 0.8588 + 0.0627;\n"
331 static const char *yuv_rev_frag =
332 "uniform sampler2D tex;\n"
335 " gl_FragColor = texture2D(tex, gl_TexCoord[0].st);\n"
336 " gl_FragColor.r = gl_FragColor.r * 1.1644 - 0.0627;\n"
338 static const char *rgb_fwd_frag =
339 "uniform sampler2D tex;\n"
342 " gl_FragColor = texture2D(tex, gl_TexCoord[0].st);\n"
343 " gl_FragColor.rgb = gl_FragColor.rgb * vec3(0.8588, 0.8588, 0.8588) + vec3(0.0627, 0.0627, 0.0627);\n"
345 static const char *rgb_rev_frag =
346 "uniform sampler2D tex;\n"
349 " gl_FragColor = texture2D(tex, gl_TexCoord[0].st);\n"
350 " gl_FragColor.rgb = gl_FragColor.rgb * vec3(1.1644, 1.1644, 1.1644) - vec3(0.0627, 0.0627, 0.0627);\n"
354 get_output()->to_texture();
355 get_output()->enable_opengl();
356 get_output()->bind_texture(0);
358 unsigned int frag_shader = 0;
359 switch(get_output()->get_color_model())
363 frag_shader = VFrame::make_shader(0,
364 config.direction == 1 ? yuv_fwd_frag : yuv_rev_frag,
369 frag_shader = VFrame::make_shader(0,
370 config.direction == 1 ? rgb_fwd_frag : rgb_rev_frag,
377 glUseProgram(frag_shader);
378 glUniform1i(glGetUniformLocation(frag_shader, "tex"), 0);
380 VFrame::init_screen(get_output()->get_w(), get_output()->get_h());
381 get_output()->draw_texture();
383 get_output()->set_opengl_state(VFrame::SCREEN);