Merge CV, ver=5.1; ops/methods from HV, and interface from CV where possible
[goodguy/history.git] / cinelerra-5.1 / plugins / burn / burn.C
diff --git a/cinelerra-5.1/plugins/burn/burn.C b/cinelerra-5.1/plugins/burn/burn.C
new file mode 100644 (file)
index 0000000..0adac78
--- /dev/null
@@ -0,0 +1,445 @@
+
+/*
+ * CINELERRA
+ * Copyright (C) 2008 Adam Williams <broadcast at earthling dot net>
+ * 
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ * 
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ * 
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ * 
+ */
+
+#include "clip.h"
+#include "bccmodels.h"
+#include "effecttv.h"
+#include "filexml.h"
+#include "language.h"
+#include "cicolors.h"
+#include "burn.h"
+#include "burnwindow.h"
+
+#include <stdint.h>
+#include <stdio.h>
+#include <string.h>
+
+
+
+REGISTER_PLUGIN(BurnMain)
+
+
+
+
+
+
+
+
+
+
+
+
+BurnConfig::BurnConfig()
+{
+       threshold = 50;
+       decay = 15;
+       recycle = 1.0;
+}
+
+BurnMain::BurnMain(PluginServer *server)
+ : PluginVClient(server)
+{
+       input_ptr = 0;
+       output_ptr = 0;
+       burn_server = 0;
+       buffer = 0;
+       effecttv = 0;
+       yuv = new YUV;
+       
+}
+
+BurnMain::~BurnMain()
+{
+       
+
+       if(buffer) delete [] buffer;
+       if(burn_server) delete burn_server;
+       if(effecttv) delete effecttv;
+       if(yuv) delete yuv;
+}
+
+const char* BurnMain::plugin_title() { return _("BurningTV"); }
+int BurnMain::is_realtime() { return 1; }
+
+
+NEW_WINDOW_MACRO(BurnMain, BurnWindow)
+
+
+int BurnMain::load_configuration()
+{
+       return 0;
+//printf("BurnMain::load_configuration %d\n", source_position);
+}
+
+
+void BurnMain::save_data(KeyFrame *keyframe)
+{
+}
+
+void BurnMain::read_data(KeyFrame *keyframe)
+{
+}
+
+
+
+#define MAXCOLOR 120
+
+void BurnMain::HSItoRGB(double H, 
+       double S, 
+       double I, 
+       int *r, 
+       int *g, 
+       int *b,
+       int color_model)
+{
+       double T, Rv, Gv, Bv;
+
+       T = H;
+       Rv = 1 + S * sin(T - 2 * M_PI / 3);
+       Gv = 1 + S * sin(T);
+       Bv = 1 + S * sin(T + 2  * M_PI / 3);
+       T = 255.999 * I / 2;
+
+       *r = (int)CLIP(Rv * T, 0, 255);
+       *g = (int)CLIP(Gv * T, 0, 255);
+       *b = (int)CLIP(Bv * T, 0, 255);
+}
+
+
+void BurnMain::make_palette(int color_model)
+{
+       int i, r, g, b;
+
+       for(i = 0; i < MAXCOLOR; i++)
+       {
+               HSItoRGB(4.6 - 1.5 * i / MAXCOLOR, 
+                       (double)i / MAXCOLOR, 
+                       (double)i / MAXCOLOR,  
+                       &r, 
+                       &g, 
+                       &b, 
+                       color_model);
+               palette[0][i] = r;
+               palette[1][i] = g;
+               palette[2][i] = b;
+//printf("BurnMain::make_palette %d %d %d %d\n", i, palette[0][i], palette[1][i], palette[2][i]);
+       }
+
+
+       for(i = MAXCOLOR; i < 256; i++)
+       {
+               if(r < 255) r++;
+               if(r < 255) r++;
+               if(r < 255) r++;
+               if(g < 255) g++;
+               if(g < 255) g++;
+               if(b < 255) b++;
+               if(b < 255) b++;
+               palette[0][i] = r;
+               palette[1][i] = g;
+               palette[2][i] = b;
+//printf("BurnMain::make_palette %d %d %d %d\n", i, palette[0][i], palette[1][i], palette[2][i]);
+       }
+}
+
+
+
+
+int BurnMain::process_realtime(VFrame *input_ptr, VFrame *output_ptr)
+{
+       this->input_ptr = input_ptr;
+       this->output_ptr = output_ptr;
+
+       load_configuration();
+
+       if(!burn_server)
+       {
+               effecttv = new EffectTV(input_ptr->get_w(), input_ptr->get_h());
+               buffer = (unsigned char *)new unsigned char[input_ptr->get_w() * input_ptr->get_h()];
+               make_palette(input_ptr->get_color_model());
+
+               effecttv->image_set_threshold_y(config.threshold);
+               total = 0;
+
+               burn_server = new BurnServer(this, 1, 1);
+       }
+
+       if(total == 0)
+       {
+               bzero(buffer, input_ptr->get_w() * input_ptr->get_h());
+               effecttv->image_bgset_y(input_ptr);
+       }
+       burn_server->process_packages();
+
+       total++;
+//     if(total >= config.recycle * project_frame_rate) total = 0;
+       return 0;
+}
+
+
+
+BurnServer::BurnServer(BurnMain *plugin, int total_clients, int total_packages)
+ : LoadServer(total_clients, total_packages)
+{
+       this->plugin = plugin;
+}
+
+
+LoadClient* BurnServer::new_client() 
+{
+       return new BurnClient(this);
+}
+
+
+
+
+LoadPackage* BurnServer::new_package() 
+{ 
+       return new BurnPackage; 
+}
+
+
+
+void BurnServer::init_packages()
+{
+       for(int i = 0; i < get_total_packages(); i++)
+       {
+               BurnPackage *package = (BurnPackage*)get_package(i);
+               package->row1 = plugin->input_ptr->get_h() * i / get_total_packages();
+               package->row2 = plugin->input_ptr->get_h() * (i + 1) / get_total_packages();
+       }
+}
+
+
+
+
+
+
+
+
+BurnClient::BurnClient(BurnServer *server)
+ : LoadClient(server)
+{
+       this->plugin = server->plugin;
+}
+
+
+
+
+
+
+
+
+
+#define BURN(type, components, is_yuv) \
+{ \
+       i = 1; \
+       type **rows = (type**)input_rows; \
+       for(y = 0; y < height; y++)  \
+       { \
+               for(x = 1; x < width - 1; x++)  \
+               { \
+                       if(sizeof(type) == 4) \
+                       { \
+                               a1 = (int)(rows[0][i * components] * 0xff); \
+                               a2 = (int)(rows[0][i * components + 1] * 0xff); \
+                               a3 = (int)(rows[0][i * components + 2] * 0xff); \
+                               CLAMP(a1, 0, 0xff); \
+                               CLAMP(a2, 0, 0xff); \
+                               CLAMP(a3, 0, 0xff); \
+                               b1 = plugin->palette[0][plugin->buffer[i]]; \
+                               b2 = plugin->palette[1][plugin->buffer[i]]; \
+                               b3 = plugin->palette[2][plugin->buffer[i]]; \
+                               a1 += b1; \
+                               a2 += b2; \
+                               a3 += b3; \
+                               b1 = a1 & 0x100; \
+                               b2 = a2 & 0x100; \
+                               b3 = a3 & 0x100; \
+                               rows[0][i * components] = (type)(a1 | (b1 - (b1 >> 8))) / 0xff; \
+                               rows[0][i * components + 1] = (type)(a2 | (b2 - (b2 >> 8))) / 0xff; \
+                               rows[0][i * components + 2] = (type)(a3 | (b3 - (b3 >> 8))) / 0xff; \
+                       } \
+                       else \
+                       if(sizeof(type) == 2) \
+                       { \
+                               a1 = ((int)rows[0][i * components + 0]) >> 8; \
+                               a2 = ((int)rows[0][i * components + 1]) >> 8; \
+                               a3 = ((int)rows[0][i * components + 2]) >> 8; \
+                               b1 = plugin->palette[0][plugin->buffer[i]]; \
+                               b2 = plugin->palette[1][plugin->buffer[i]]; \
+                               b3 = plugin->palette[2][plugin->buffer[i]]; \
+                               if(is_yuv) plugin->yuv->yuv_to_rgb_8(a1, a2, a3); \
+                               a1 += b1; \
+                               a2 += b2; \
+                               a3 += b3; \
+                               b1 = a1 & 0x100; \
+                               b2 = a2 & 0x100; \
+                               b3 = a3 & 0x100; \
+                               a1 = (a1 | (b1 - (b1 >> 8))); \
+                               a2 = (a2 | (b2 - (b2 >> 8))); \
+                               a3 = (a3 | (b3 - (b3 >> 8))); \
+                               if(is_yuv) \
+                               { \
+                                       CLAMP(a1, 0, 0xff); \
+                                       CLAMP(a2, 0, 0xff); \
+                                       CLAMP(a3, 0, 0xff); \
+                                       plugin->yuv->rgb_to_yuv_8(a1, a2, a3); \
+                               } \
+                               rows[0][i * components + 0] = a1 | (a1 << 8); \
+                               rows[0][i * components + 1] = a2 | (a2 << 8); \
+                               rows[0][i * components + 2] = a3 | (a3 << 8); \
+                       } \
+                       else \
+                       { \
+                               a1 = (int)rows[0][i * components + 0]; \
+                               a2 = (int)rows[0][i * components + 1]; \
+                               a3 = (int)rows[0][i * components + 2]; \
+                               b1 = plugin->palette[0][plugin->buffer[i]]; \
+                               b2 = plugin->palette[1][plugin->buffer[i]]; \
+                               b3 = plugin->palette[2][plugin->buffer[i]]; \
+                               if(is_yuv) plugin->yuv->yuv_to_rgb_8(a1, a2, a3); \
+                               a1 += b1; \
+                               a2 += b2; \
+                               a3 += b3; \
+                               b1 = a1 & 0x100; \
+                               b2 = a2 & 0x100; \
+                               b3 = a3 & 0x100; \
+                               a1 = (a1 | (b1 - (b1 >> 8))); \
+                               a2 = (a2 | (b2 - (b2 >> 8))); \
+                               a3 = (a3 | (b3 - (b3 >> 8))); \
+                               if(is_yuv) \
+                               { \
+                                       CLAMP(a1, 0, 0xff); \
+                                       CLAMP(a2, 0, 0xff); \
+                                       CLAMP(a3, 0, 0xff); \
+                                       plugin->yuv->rgb_to_yuv_8(a1, a2, a3); \
+                               } \
+                               rows[0][i * components + 0] = a1; \
+                               rows[0][i * components + 1] = a2; \
+                               rows[0][i * components + 2] = a3; \
+                       } \
+                       i++; \
+               } \
+               i += 2; \
+       } \
+}
+
+
+
+
+void BurnClient::process_package(LoadPackage *package)
+{
+       BurnPackage *local_package = (BurnPackage*)package;
+       unsigned char **input_rows = plugin->input_ptr->get_rows() + local_package->row1;
+       //unsigned char **output_rows = plugin->output_ptr->get_rows() + local_package->row1;
+       int width = plugin->input_ptr->get_w();
+       int height = local_package->row2 - local_package->row1;
+       unsigned char *diff;
+       //int pitch = width * plugin->input_ptr->get_bytes_per_pixel();
+       int i, x, y;
+       int a1, a2, a3;
+       int b1, b2, b3;
+
+       diff = plugin->effecttv->image_bgsubtract_y(input_rows, 
+               plugin->input_ptr->get_color_model());
+
+       for(x = 1; x < width - 1; x++)
+       {
+               unsigned int v = 0;
+               for(y = 0; y < height - 1; y++)
+               {
+                       unsigned int w = diff[y * width + x];
+                       plugin->buffer[y * width + x] |= v ^ w;
+                       v = w;
+               }
+       }
+
+       for(x = 1; x < width - 1; x++) 
+       {
+               i = width + x;
+
+               for(y = 1; y < height; y++) 
+               {
+                       int v = plugin->buffer[i];
+
+                       if(v < plugin->config.decay)
+                               plugin->buffer[i - width] = 0;
+                       else
+                               plugin->buffer[i - width + EffectTV::fastrand() % 3 - 1] = 
+                                       v - (EffectTV::fastrand() & plugin->config.decay);
+
+                       i += width;
+               }
+       }
+
+
+
+       switch(plugin->input_ptr->get_color_model())
+       {
+               case BC_RGB888:
+                       BURN(uint8_t, 3, 0);
+                       break;
+               case BC_YUV888:
+                       BURN(uint8_t, 3, 1);
+                       break;
+
+               case BC_RGB_FLOAT:
+                       BURN(float, 3, 0);
+                       break;
+
+               case BC_RGBA_FLOAT:
+                       BURN(float, 4, 0);
+                       break;
+
+               case BC_RGBA8888:
+                       BURN(uint8_t, 4, 0);
+                       break;
+               case BC_YUVA8888:
+                       BURN(uint8_t, 4, 1);
+                       break;
+
+               case BC_RGB161616:
+                       BURN(uint16_t, 3, 0);
+                       break;
+               case BC_YUV161616:
+                       BURN(uint16_t, 3, 1);
+                       break;
+
+               case BC_RGBA16161616:
+                       BURN(uint16_t, 4, 0);
+                       break;
+               case BC_YUVA16161616:
+                       BURN(uint16_t, 4, 1);
+                       break;
+       }
+
+
+}
+
+
+
+BurnPackage::BurnPackage()
+{
+}
+
+
+