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
26 REGISTER_PLUGIN(RGBShiftEffect)
33 RGBShiftConfig::RGBShiftConfig()
38 void RGBShiftConfig::reset(int clear)
41 case RESET_R_DX : r_dx = 0;
43 case RESET_R_DY : r_dy = 0;
45 case RESET_G_DX : g_dx = 0;
47 case RESET_G_DY : g_dy = 0;
49 case RESET_B_DX : b_dx = 0;
51 case RESET_B_DY : b_dy = 0;
62 void RGBShiftConfig::copy_from(RGBShiftConfig &src)
64 r_dx = src.r_dx; r_dy = src.r_dy;
65 g_dx = src.g_dx; g_dy = src.g_dy;
66 b_dx = src.b_dx; b_dy = src.b_dy;
69 int RGBShiftConfig::equivalent(RGBShiftConfig &src)
71 return EQUIV(r_dx, src.r_dx) && EQUIV(g_dx, src.g_dx) && EQUIV(b_dx, src.b_dx) &&
72 EQUIV(r_dy, src.r_dy) && EQUIV(g_dy, src.g_dy) && EQUIV(b_dy, src.b_dy);
75 void RGBShiftConfig::interpolate(RGBShiftConfig &prev,
81 double next_scale = (double)(current_frame - prev_frame) / (next_frame - prev_frame);
82 double prev_scale = (double)(next_frame - current_frame) / (next_frame - prev_frame);
84 r_dx = prev.r_dx * prev_scale + next.r_dx * next_scale;
85 r_dy = prev.r_dy * prev_scale + next.r_dy * next_scale;
86 g_dx = prev.g_dx * prev_scale + next.g_dx * next_scale;
87 g_dy = prev.g_dy * prev_scale + next.g_dy * next_scale;
88 b_dx = prev.b_dx * prev_scale + next.b_dx * next_scale;
89 b_dy = prev.b_dy * prev_scale + next.b_dy * next_scale;
99 RGBShiftLevel::RGBShiftLevel(RGBShiftEffect *plugin, int *output, int x, int y)
100 : BC_ISlider(x, y, 0, xS(200), yS(200), -MAXVALUE, MAXVALUE, *output)
102 this->plugin = plugin;
103 this->output = output;
106 int RGBShiftLevel::handle_event()
108 *output = get_value();
109 plugin->send_configure_change();
114 RGBShiftReset::RGBShiftReset(RGBShiftEffect *plugin, RGBShiftWindow *window, int x, int y)
115 : BC_GenericButton(x, y, _("Reset"))
117 this->plugin = plugin;
118 this->window = window;
120 RGBShiftReset::~RGBShiftReset()
123 int RGBShiftReset::handle_event()
125 plugin->config.reset(RESET_ALL);
126 window->update_gui(RESET_ALL);
127 plugin->send_configure_change();
132 RGBShiftSliderClr::RGBShiftSliderClr(RGBShiftEffect *plugin, RGBShiftWindow *window, int x, int y, int w, int clear)
133 : BC_Button(x, y, w, plugin->get_theme()->get_image_set("reset_button"))
135 this->plugin = plugin;
136 this->window = window;
139 RGBShiftSliderClr::~RGBShiftSliderClr()
142 int RGBShiftSliderClr::handle_event()
144 // clear==1 ==> r_dx slider --- clear==2 ==> r_dy slider
145 // clear==3 ==> g_dx slider --- clear==4 ==> g_dy slider
146 // clear==5 ==> b_dx slider --- clear==6 ==> b_dy slider
147 plugin->config.reset(clear);
148 window->update_gui(clear);
149 plugin->send_configure_change();
154 RGBShiftWindow::RGBShiftWindow(RGBShiftEffect *plugin)
155 : PluginClientWindow(plugin, xS(320), yS(230), xS(320), yS(230), 0)
157 this->plugin = plugin;
160 void RGBShiftWindow::create_objects()
162 int xs10 = xS(10), xs50 = xS(50);
163 int ys10 = yS(10), ys30 = yS(30), ys40 = yS(40);
164 int x = xs10, y = ys10, x1 = xs50;
165 int x2 = 0; int clrBtn_w = xs50;
167 add_subwindow(new BC_Title(x, y, _("R_dx:")));
168 add_subwindow(r_dx = new RGBShiftLevel(plugin, &plugin->config.r_dx, x1, y));
169 x2 = x1 + r_dx->get_w() + xs10;
170 add_subwindow(r_dxClr = new RGBShiftSliderClr(plugin, this, x2, y, clrBtn_w, RESET_R_DX));
173 add_subwindow(new BC_Title(x, y, _("R_dy:")));
174 add_subwindow(r_dy = new RGBShiftLevel(plugin, &plugin->config.r_dy, x1, y));
175 add_subwindow(r_dyClr = new RGBShiftSliderClr(plugin, this, x2, y, clrBtn_w, RESET_R_DY));
178 add_subwindow(new BC_Title(x, y, _("G_dx:")));
179 add_subwindow(g_dx = new RGBShiftLevel(plugin, &plugin->config.g_dx, x1, y));
180 add_subwindow(g_dxClr = new RGBShiftSliderClr(plugin, this, x2, y, clrBtn_w, RESET_G_DX));
183 add_subwindow(new BC_Title(x, y, _("G_dy:")));
184 add_subwindow(g_dy = new RGBShiftLevel(plugin, &plugin->config.g_dy, x1, y));
185 add_subwindow(g_dyClr = new RGBShiftSliderClr(plugin, this, x2, y, clrBtn_w, RESET_G_DY));
188 add_subwindow(new BC_Title(x, y, _("B_dx:")));
189 add_subwindow(b_dx = new RGBShiftLevel(plugin, &plugin->config.b_dx, x1, y));
190 add_subwindow(b_dxClr = new RGBShiftSliderClr(plugin, this, x2, y, clrBtn_w, RESET_B_DX));
193 add_subwindow(new BC_Title(x, y, _("B_dy:")));
194 add_subwindow(b_dy = new RGBShiftLevel(plugin, &plugin->config.b_dy, x1, y));
195 add_subwindow(b_dyClr = new RGBShiftSliderClr(plugin, this, x2, y, clrBtn_w, RESET_B_DY));
198 add_subwindow(reset = new RGBShiftReset(plugin, this, x, y));
206 void RGBShiftWindow::update_gui(int clear)
209 case RESET_R_DX : r_dx->update(plugin->config.r_dx);
211 case RESET_R_DY : r_dy->update(plugin->config.r_dy);
213 case RESET_G_DX : g_dx->update(plugin->config.g_dx);
215 case RESET_G_DY : g_dy->update(plugin->config.g_dy);
217 case RESET_B_DX : b_dx->update(plugin->config.b_dx);
219 case RESET_B_DY : b_dy->update(plugin->config.b_dy);
223 r_dx->update(plugin->config.r_dx);
224 r_dy->update(plugin->config.r_dy);
225 g_dx->update(plugin->config.g_dx);
226 g_dy->update(plugin->config.g_dy);
227 b_dx->update(plugin->config.b_dx);
228 b_dy->update(plugin->config.b_dy);
238 RGBShiftEffect::RGBShiftEffect(PluginServer *server)
239 : PluginVClient(server)
243 RGBShiftEffect::~RGBShiftEffect()
248 const char* RGBShiftEffect::plugin_title() { return N_("RGBShift"); }
249 int RGBShiftEffect::is_realtime() { return 1; }
252 NEW_WINDOW_MACRO(RGBShiftEffect, RGBShiftWindow)
253 LOAD_CONFIGURATION_MACRO(RGBShiftEffect, RGBShiftConfig)
255 void RGBShiftEffect::update_gui()
259 RGBShiftWindow *yuv_wdw = (RGBShiftWindow*)thread->window;
260 yuv_wdw->lock_window("RGBShiftEffect::update_gui");
261 load_configuration();
262 yuv_wdw->r_dx->update(config.r_dx);
263 yuv_wdw->r_dy->update(config.r_dy);
264 yuv_wdw->g_dx->update(config.g_dx);
265 yuv_wdw->g_dy->update(config.g_dy);
266 yuv_wdw->b_dx->update(config.b_dx);
267 yuv_wdw->b_dy->update(config.b_dy);
268 yuv_wdw->unlock_window();
272 void RGBShiftEffect::save_data(KeyFrame *keyframe)
275 output.set_shared_output(keyframe->xbuf);
276 output.tag.set_title("RGBSHIFT");
277 output.tag.set_property("R_DX", config.r_dx);
278 output.tag.set_property("R_DY", config.r_dy);
279 output.tag.set_property("G_DX", config.g_dx);
280 output.tag.set_property("G_DY", config.g_dy);
281 output.tag.set_property("B_DX", config.b_dx);
282 output.tag.set_property("B_DY", config.b_dy);
284 output.tag.set_title("/RGBSHIFT");
286 output.append_newline();
287 output.terminate_string();
290 void RGBShiftEffect::read_data(KeyFrame *keyframe)
293 input.set_shared_input(keyframe->xbuf);
294 while(!input.read_tag())
296 if(input.tag.title_is("RGBSHIFT"))
298 config.r_dx = input.tag.get_property("R_DX", config.r_dx);
299 config.r_dy = input.tag.get_property("R_DY", config.r_dy);
300 config.g_dx = input.tag.get_property("G_DX", config.g_dx);
301 config.g_dy = input.tag.get_property("G_DY", config.g_dy);
302 config.b_dx = input.tag.get_property("B_DX", config.b_dx);
303 config.b_dy = input.tag.get_property("B_DY", config.b_dy);
308 #define RGB_MACRO(type, temp_type, components) \
310 for(int i = 0; i < h; i++) { \
311 int ri = i + config.r_dy, gi = i + config.g_dy, bi = i + config.b_dy; \
312 type *in_r = ri >= 0 && ri < h ? (type *)frame->get_rows()[ri] : 0; \
313 type *in_g = gi >= 0 && gi < h ? (type *)frame->get_rows()[gi] : 0; \
314 type *in_b = bi >= 0 && bi < h ? (type *)frame->get_rows()[bi] : 0; \
315 type *out_row = (type *)output->get_rows()[i]; \
316 for(int j = 0; j < w; j++) { \
317 int rj = j + config.r_dx, gj = j + config.g_dx, bj = j + config.b_dx; \
318 type *rp = in_r && rj >= 0 && rj < w ? in_r + rj*components: 0; \
319 type *gp = in_g && gj >= 0 && gj < w ? in_g + gj*components: 0; \
320 type *bp = in_b && bj >= 0 && bj < w ? in_b + bj*components: 0; \
321 out_row[0] = rp ? rp[0] : 0; \
322 out_row[1] = gp ? gp[1] : 0; \
323 out_row[2] = bp ? bp[2] : 0; \
324 out_row += components; \
329 #define YUV_MACRO(type, temp_type, components) \
331 for(int i = 0; i < h; i++) { \
332 int ri = i + config.r_dy, gi = i + config.g_dy, bi = i + config.b_dy; \
333 uint8_t *in_r = ri >= 0 && ri < h ? (uint8_t *)frame->get_rows()[ri] : 0; \
334 uint8_t *in_g = gi >= 0 && gi < h ? (uint8_t *)frame->get_rows()[gi] : 0; \
335 uint8_t *in_b = bi >= 0 && bi < h ? (uint8_t *)frame->get_rows()[bi] : 0; \
336 type *out_row = (type *)output->get_rows()[i]; \
337 for(int j = 0; j < w; j++) { \
338 int rj = j + config.r_dx, gj = j + config.g_dx, bj = j + config.b_dx; \
339 uint8_t *rp = in_r && rj >= 0 && rj < w ? in_r + rj*3: 0; \
340 uint8_t *gp = in_g && gj >= 0 && gj < w ? in_g + gj*3: 0; \
341 uint8_t *bp = in_b && bj >= 0 && bj < w ? in_b + bj*3: 0; \
343 temp_type r = rp ? rp[0] : 0; \
344 temp_type g = gp ? gp[1] : 0; \
345 temp_type b = bp ? bp[2] : 0; \
346 if( sizeof(type) == 4 ) \
347 YUV::yuv.rgb_to_yuv_f(r, g, b, y, u, v); \
348 else if( sizeof(type) == 2 ) \
349 YUV::yuv.rgb_to_yuv_16(r, g, b, y, u, v); \
351 YUV::yuv.rgb_to_yuv_8(r, g, b, y, u, v); \
355 out_row += components; \
360 int RGBShiftEffect::process_realtime(VFrame *input, VFrame *output)
362 load_configuration();
364 if( EQUIV(config.r_dx, 0) && EQUIV(config.g_dx, 0) && EQUIV(config.b_dx, 0) &&
365 EQUIV(config.r_dy, 0) && EQUIV(config.g_dy, 0) && EQUIV(config.b_dy, 0) ) {
366 if(input->get_rows()[0] != output->get_rows()[0])
367 output->copy_from(input);
371 int w = input->get_w(), h = input->get_h();
372 int color_model = input->get_color_model();
373 int is_yuv = BC_CModels::is_yuv(color_model);
374 if( is_yuv ) color_model = BC_RGB888;
375 VFrame *frame = input;
376 if( input->get_rows()[0] == output->get_rows()[0] || !is_yuv ) {
377 if( temp_frame && ( temp_frame->get_color_model() != color_model ||
378 temp_frame->get_w() != w || temp_frame->get_h() != h ) ) {
379 delete temp_frame; temp_frame = 0;
382 temp_frame = new VFrame(w, h, color_model, 0);
384 if( color_model != input->get_color_model() )
385 BC_CModels::transfer(frame->get_rows(), input->get_rows(),
386 frame->get_y(), frame->get_u(), frame->get_v(),
387 input->get_y(), input->get_u(), input->get_v(),
388 0, 0, input->get_w(), input->get_h(),
389 0, 0, frame->get_w(), frame->get_h(),
390 input->get_color_model(), frame->get_color_model(), 0,
391 input->get_bytes_per_line(), w);
393 frame->copy_from(input);
396 switch( input->get_color_model() ) {
397 case BC_YUV888: YUV_MACRO(unsigned char, int, 3); break;
398 case BC_YUV161616: YUV_MACRO(uint16_t, int, 3); break;
399 case BC_YUVA8888: YUV_MACRO(unsigned char, int, 4); break;
400 case BC_YUVA16161616: YUV_MACRO(uint16_t, int, 4); break;
401 case BC_RGB_FLOAT: RGB_MACRO(float, float, 3); break;
402 case BC_RGB888: RGB_MACRO(unsigned char, int, 3); break;
403 case BC_RGB161616: RGB_MACRO(uint16_t, int, 3); break;
404 case BC_RGBA_FLOAT: RGB_MACRO(float, float, 4); break;
405 case BC_RGBA8888: RGB_MACRO(unsigned char, int, 4); break;
406 case BC_RGBA16161616: RGB_MACRO(uint16_t, int, 4); break;