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(input_ptr->get_w(),input_ptr->get_h(),
199 // project_color_model, 0);
200 bgimage->copy_from(input_ptr);
204 /* step 2: add frame-2 to buffer-1 */
205 add_frames(bgimage, input_ptr);
209 /* step 3: grab frame-3 to buffer-2 */
210 tmp->copy_from(input_ptr);
214 /* step 4: add frame-4 to buffer-2 */
215 add_frames(tmp, input_ptr);
219 /* step 5: add buffer-3 to buffer-1 */
220 add_frames(bgimage, tmp);
222 effecttv->image_bgset_y(bgimage);
231 int HoloMain::process_realtime(VFrame *input_ptr, VFrame *output_ptr)
233 this->input_ptr = input_ptr;
234 this->output_ptr = output_ptr;
239 load_configuration();
247 effecttv = new EffectTV(input_ptr->get_w(), input_ptr->get_h());
248 bgimage = new VFrame(input_ptr->get_w(), input_ptr->get_h(),
249 input_ptr->get_color_model(), 0);
251 for(int i = 0; i < 256; i++)
253 noisepattern[i] = (i * i * i / 40000)* i / 256;
256 holo_server = new HoloServer(this, 1, 1);
264 holo_server->process_packages();
267 if(total >= config.recycle * project_frame_rate)
277 HoloServer::HoloServer(HoloMain *plugin, int total_clients, int total_packages)
278 : LoadServer(total_clients, total_packages)
280 this->plugin = plugin;
284 LoadClient* HoloServer::new_client()
286 return new HoloClient(this);
292 LoadPackage* HoloServer::new_package()
294 return new HoloPackage;
299 void HoloServer::init_packages()
301 for(int i = 0; i < get_total_packages(); i++)
303 HoloPackage *package = (HoloPackage*)get_package(i);
304 package->row1 = plugin->input_ptr->get_h() * i / get_total_packages();
305 package->row2 = plugin->input_ptr->get_h() * (i + 1) / get_total_packages();
316 HoloClient::HoloClient(HoloServer *server)
319 this->plugin = server->plugin;
324 void HoloClient::process_package(LoadPackage *package)
327 HoloPackage *local_package = (HoloPackage*)package;
328 unsigned char **input_rows = plugin->input_ptr->get_rows() + local_package->row1;
329 unsigned char **output_rows = plugin->output_ptr->get_rows() + local_package->row1;
330 int width = plugin->input_ptr->get_w();
331 int height = local_package->row2 - local_package->row1;
336 diff = plugin->effecttv->image_diff_filter(plugin->effecttv->image_bgsubtract_y(input_rows,
337 plugin->input_ptr->get_color_model()));
344 // Convert discrete channels to a single 24 bit number
345 #define STORE_PIXEL(type, components, dest, src, is_yuv) \
346 if(sizeof(type) == 4) \
348 int r = (int)(src[0] * 0xff); \
349 int g = (int)(src[1] * 0xff); \
350 int b = (int)(src[2] * 0xff); \
354 dest = (r << 16) | (g << 8) | b; \
357 if(sizeof(type) == 2) \
361 int r = (int)src[0] >> 8; \
362 int g = (int)src[1] >> 8; \
363 int b = (int)src[2] >> 8; \
364 plugin->yuv->yuv_to_rgb_8(r, g, b); \
365 dest = (r << 16) | (g << 8) | b; \
369 dest = (((uint32_t)src[0] << 8) & 0xff0000) | \
370 ((uint32_t)src[1] & 0xff00) | \
371 ((uint32_t)src[2]) >> 8; \
378 int r = (int)src[0]; \
379 int g = (int)src[1]; \
380 int b = (int)src[2]; \
381 plugin->yuv->yuv_to_rgb_8(r, g, b); \
382 dest = (r << 16) | (g << 8) | b; \
386 dest = ((uint32_t)src[0] << 16) | \
387 ((uint32_t)src[1] << 8) | \
395 #define HOLO_CORE(type, components, is_yuv) \
396 for(y = 1; y < height - 1; y++) \
398 type *src = (type*)input_rows[y]; \
399 type *bg = (type*)plugin->bgimage->get_rows()[y]; \
400 type *dest = (type*)output_rows[y]; \
404 if(((y + phase) & 0x7f) < 0x58) \
406 for(x = 0 ; x < width; x++) \
410 STORE_PIXEL(type, components, s, src, is_yuv); \
413 ((s & 0xff00) >> 7) + \
414 ((s & 0xff0000) >> 16); \
415 t += plugin->noisepattern[EffectTV::fastrand() >> 24]; \
417 r = ((s & 0xff0000) >> 17) + t; \
418 g = ((s & 0xff00) >> 8) + t; \
419 b = (s & 0xff) + t; \
421 r = (r >> 1) - 100; \
422 g = (g >> 1) - 100; \
428 STORE_PIXEL(type, components, s, bg, is_yuv); \
430 r += (s & 0xff0000) >> 17; \
431 g += (s & 0xff00) >> 9; \
432 b += ((s & 0xff) >> 1) + 40; \
434 if(r > 255) r = 255; \
435 if(g > 255) g = 255; \
436 if(b > 255) b = 255; \
438 if(is_yuv) plugin->yuv->rgb_to_yuv_8(r, g, b); \
439 if(sizeof(type) == 4) \
441 dest[0] = (type)r / 0xff; \
442 dest[1] = (type)g / 0xff; \
443 dest[2] = (type)b / 0xff; \
446 if(sizeof(type) == 2) \
448 dest[0] = (r << 8) | r; \
449 dest[1] = (g << 8) | g; \
450 dest[2] = (b << 8) | b; \
468 dest += components; \
474 for(x = 0; x < width; x++) \
478 STORE_PIXEL(type, components, s, src, is_yuv); \
481 t = (s & 0xff) + ((s & 0xff00) >> 6) + ((s & 0xff0000) >> 16); \
482 t += plugin->noisepattern[EffectTV::fastrand() >> 24]; \
484 r = ((s & 0xff0000) >> 16) + t; \
485 g = ((s & 0xff00) >> 8) + t; \
486 b = (s & 0xff) + t; \
488 r = (r >> 1) - 100; \
489 g = (g >> 1) - 100; \
495 STORE_PIXEL(type, components, s, bg, is_yuv); \
497 r += ((s & 0xff0000) >> 17) + 10; \
498 g += ((s & 0xff00) >> 9) + 10; \
499 b += ((s & 0xff) >> 1) + 40; \
501 if(r > 255) r = 255; \
502 if(g > 255) g = 255; \
503 if(b > 255) b = 255; \
505 if(is_yuv) plugin->yuv->rgb_to_yuv_8(r, g, b); \
506 if(sizeof(type) == 4) \
508 dest[0] = (type)r / 0xff; \
509 dest[1] = (type)g / 0xff; \
510 dest[2] = (type)b / 0xff; \
513 if(sizeof(type) == 2) \
515 dest[0] = (r << 8) | r; \
516 dest[1] = (g << 8) | g; \
517 dest[2] = (b << 8) | b; \
535 dest += components; \
544 switch(plugin->input_ptr->get_color_model())
547 HOLO_CORE(uint8_t, 3, 0);
550 HOLO_CORE(float, 3, 0);
553 HOLO_CORE(uint8_t, 3, 1);
556 HOLO_CORE(float, 4, 0);
559 HOLO_CORE(uint8_t, 4, 0);
562 HOLO_CORE(uint8_t, 4, 1);
565 HOLO_CORE(uint16_t, 3, 0);
568 HOLO_CORE(uint16_t, 3, 1);
570 case BC_RGBA16161616:
571 HOLO_CORE(uint16_t, 4, 0);
573 case BC_YUVA16161616:
574 HOLO_CORE(uint16_t, 4, 1);
585 HoloPackage::HoloPackage()