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 "holowindow.h"
36 REGISTER_PLUGIN(HoloMain)
47 HoloConfig::HoloConfig()
57 HoloMain::HoloMain(PluginServer *server)
58 : PluginVClient(server)
83 const char* HoloMain::plugin_title() { return _("HolographicTV"); }
84 int HoloMain::is_realtime() { return 1; }
86 NEW_WINDOW_MACRO(HoloMain, HoloWindow)
88 int HoloMain::load_configuration()
94 void HoloMain::save_data(KeyFrame *keyframe)
98 void HoloMain::read_data(KeyFrame *keyframe)
102 void HoloMain::reconfigure()
106 effecttv->image_set_threshold_y(config.threshold);
110 #define ADD_FRAMES(type, components) \
112 type **input_rows = (type**)input->get_rows(); \
113 type **output_rows = (type**)output->get_rows(); \
114 int w = input->get_w(); \
115 int h = input->get_h(); \
117 for(int i = 0; i < h; i++) \
119 type *output_row = (type*)output_rows[i]; \
120 type *input_row = (type*)input_rows[i]; \
122 for(int j = 0; j < w; j++) \
124 for(int k = 0; k < 3; k++) \
126 if(sizeof(type) == 4) \
128 int in_temp = (int)(*input_row * 0xffff); \
129 int out_temp = (int)(*output_row * 0xffff); \
130 int temp = (in_temp & out_temp) + \
131 ((in_temp ^ out_temp) >> 1); \
132 *output_row = (type)temp / 0xffff; \
136 *output_row = ((uint16_t)*input_row & (uint16_t)*output_row) + \
137 (((uint16_t)*input_row ^ (uint16_t)*output_row) >> 1); \
143 if(components == 4) \
153 // Add input to output and store result in output
154 void HoloMain::add_frames(VFrame *output, VFrame *input)
156 switch(output->get_color_model())
160 ADD_FRAMES(uint8_t, 3);
163 ADD_FRAMES(float, 3);
166 ADD_FRAMES(float, 4);
170 ADD_FRAMES(uint8_t, 4);
174 ADD_FRAMES(uint16_t, 3);
176 case BC_RGBA16161616:
177 case BC_YUVA16161616:
178 ADD_FRAMES(uint16_t, 4);
183 void HoloMain::set_background()
186 * grab 4 frames and composite them to get a quality background image
189 * For Cinelerra, we make every frame a holograph and expect the user to
197 /* step 1: grab frame-1 to buffer-1 */
198 // tmp = new VFrame(0,
199 // input_ptr->get_w(),
200 // input_ptr->get_h(),
201 // project_color_model);
202 bgimage->copy_from(input_ptr);
206 /* step 2: add frame-2 to buffer-1 */
207 add_frames(bgimage, input_ptr);
211 /* step 3: grab frame-3 to buffer-2 */
212 tmp->copy_from(input_ptr);
216 /* step 4: add frame-4 to buffer-2 */
217 add_frames(tmp, input_ptr);
221 /* step 5: add buffer-3 to buffer-1 */
222 add_frames(bgimage, tmp);
224 effecttv->image_bgset_y(bgimage);
233 int HoloMain::process_realtime(VFrame *input_ptr, VFrame *output_ptr)
235 this->input_ptr = input_ptr;
236 this->output_ptr = output_ptr;
241 load_configuration();
249 effecttv = new EffectTV(input_ptr->get_w(), input_ptr->get_h());
250 bgimage = new VFrame(0,
254 input_ptr->get_color_model(),
257 for(int i = 0; i < 256; i++)
259 noisepattern[i] = (i * i * i / 40000)* i / 256;
262 holo_server = new HoloServer(this, 1, 1);
270 holo_server->process_packages();
273 if(total >= config.recycle * project_frame_rate)
283 HoloServer::HoloServer(HoloMain *plugin, int total_clients, int total_packages)
284 : LoadServer(total_clients, total_packages)
286 this->plugin = plugin;
290 LoadClient* HoloServer::new_client()
292 return new HoloClient(this);
298 LoadPackage* HoloServer::new_package()
300 return new HoloPackage;
305 void HoloServer::init_packages()
307 for(int i = 0; i < get_total_packages(); i++)
309 HoloPackage *package = (HoloPackage*)get_package(i);
310 package->row1 = plugin->input_ptr->get_h() * i / get_total_packages();
311 package->row2 = plugin->input_ptr->get_h() * (i + 1) / get_total_packages();
322 HoloClient::HoloClient(HoloServer *server)
325 this->plugin = server->plugin;
330 void HoloClient::process_package(LoadPackage *package)
333 HoloPackage *local_package = (HoloPackage*)package;
334 unsigned char **input_rows = plugin->input_ptr->get_rows() + local_package->row1;
335 unsigned char **output_rows = plugin->output_ptr->get_rows() + local_package->row1;
336 int width = plugin->input_ptr->get_w();
337 int height = local_package->row2 - local_package->row1;
342 diff = plugin->effecttv->image_diff_filter(plugin->effecttv->image_bgsubtract_y(input_rows,
343 plugin->input_ptr->get_color_model()));
350 // Convert discrete channels to a single 24 bit number
351 #define STORE_PIXEL(type, components, dest, src, is_yuv) \
352 if(sizeof(type) == 4) \
354 int r = (int)(src[0] * 0xff); \
355 int g = (int)(src[1] * 0xff); \
356 int b = (int)(src[2] * 0xff); \
360 dest = (r << 16) | (g << 8) | b; \
363 if(sizeof(type) == 2) \
367 int r = (int)src[0] >> 8; \
368 int g = (int)src[1] >> 8; \
369 int b = (int)src[2] >> 8; \
370 plugin->yuv->yuv_to_rgb_8(r, g, b); \
371 dest = (r << 16) | (g << 8) | b; \
375 dest = (((uint32_t)src[0] << 8) & 0xff0000) | \
376 ((uint32_t)src[1] & 0xff00) | \
377 ((uint32_t)src[2]) >> 8; \
384 int r = (int)src[0]; \
385 int g = (int)src[1]; \
386 int b = (int)src[2]; \
387 plugin->yuv->yuv_to_rgb_8(r, g, b); \
388 dest = (r << 16) | (g << 8) | b; \
392 dest = ((uint32_t)src[0] << 16) | \
393 ((uint32_t)src[1] << 8) | \
401 #define HOLO_CORE(type, components, is_yuv) \
402 for(y = 1; y < height - 1; y++) \
404 type *src = (type*)input_rows[y]; \
405 type *bg = (type*)plugin->bgimage->get_rows()[y]; \
406 type *dest = (type*)output_rows[y]; \
410 if(((y + phase) & 0x7f) < 0x58) \
412 for(x = 0 ; x < width; x++) \
416 STORE_PIXEL(type, components, s, src, is_yuv); \
419 ((s & 0xff00) >> 7) + \
420 ((s & 0xff0000) >> 16); \
421 t += plugin->noisepattern[EffectTV::fastrand() >> 24]; \
423 r = ((s & 0xff0000) >> 17) + t; \
424 g = ((s & 0xff00) >> 8) + t; \
425 b = (s & 0xff) + t; \
427 r = (r >> 1) - 100; \
428 g = (g >> 1) - 100; \
434 STORE_PIXEL(type, components, s, bg, is_yuv); \
436 r += (s & 0xff0000) >> 17; \
437 g += (s & 0xff00) >> 9; \
438 b += ((s & 0xff) >> 1) + 40; \
440 if(r > 255) r = 255; \
441 if(g > 255) g = 255; \
442 if(b > 255) b = 255; \
444 if(is_yuv) plugin->yuv->rgb_to_yuv_8(r, g, b); \
445 if(sizeof(type) == 4) \
447 dest[0] = (type)r / 0xff; \
448 dest[1] = (type)g / 0xff; \
449 dest[2] = (type)b / 0xff; \
452 if(sizeof(type) == 2) \
454 dest[0] = (r << 8) | r; \
455 dest[1] = (g << 8) | g; \
456 dest[2] = (b << 8) | b; \
474 dest += components; \
480 for(x = 0; x < width; x++) \
484 STORE_PIXEL(type, components, s, src, is_yuv); \
487 t = (s & 0xff) + ((s & 0xff00) >> 6) + ((s & 0xff0000) >> 16); \
488 t += plugin->noisepattern[EffectTV::fastrand() >> 24]; \
490 r = ((s & 0xff0000) >> 16) + t; \
491 g = ((s & 0xff00) >> 8) + t; \
492 b = (s & 0xff) + t; \
494 r = (r >> 1) - 100; \
495 g = (g >> 1) - 100; \
501 STORE_PIXEL(type, components, s, bg, is_yuv); \
503 r += ((s & 0xff0000) >> 17) + 10; \
504 g += ((s & 0xff00) >> 9) + 10; \
505 b += ((s & 0xff) >> 1) + 40; \
507 if(r > 255) r = 255; \
508 if(g > 255) g = 255; \
509 if(b > 255) b = 255; \
511 if(is_yuv) plugin->yuv->rgb_to_yuv_8(r, g, b); \
512 if(sizeof(type) == 4) \
514 dest[0] = (type)r / 0xff; \
515 dest[1] = (type)g / 0xff; \
516 dest[2] = (type)b / 0xff; \
519 if(sizeof(type) == 2) \
521 dest[0] = (r << 8) | r; \
522 dest[1] = (g << 8) | g; \
523 dest[2] = (b << 8) | b; \
541 dest += components; \
550 switch(plugin->input_ptr->get_color_model())
553 HOLO_CORE(uint8_t, 3, 0);
556 HOLO_CORE(float, 3, 0);
559 HOLO_CORE(uint8_t, 3, 1);
562 HOLO_CORE(float, 4, 0);
565 HOLO_CORE(uint8_t, 4, 0);
568 HOLO_CORE(uint8_t, 4, 1);
571 HOLO_CORE(uint16_t, 3, 0);
574 HOLO_CORE(uint16_t, 3, 1);
576 case BC_RGBA16161616:
577 HOLO_CORE(uint16_t, 4, 0);
579 case BC_YUVA16161616:
580 HOLO_CORE(uint16_t, 4, 1);
591 HoloPackage::HoloPackage()