Merge CV, ver=5.1; ops/methods from HV, and interface from CV where possible
[goodguy/history.git] / cinelerra-5.1 / plugins / wave / wave.C
diff --git a/cinelerra-5.1/plugins/wave/wave.C b/cinelerra-5.1/plugins/wave/wave.C
new file mode 100644 (file)
index 0000000..ce37965
--- /dev/null
@@ -0,0 +1,832 @@
+
+/*
+ * 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 "bcdisplayinfo.h"
+#include "clip.h"
+#include "bchash.h"
+#include "filexml.h"
+#include "guicast.h"
+#include "keyframe.h"
+#include "language.h"
+#include "loadbalance.h"
+#include "pluginvclient.h"
+#include "vframe.h"
+
+#include <math.h>
+#include <stdint.h>
+#include <string.h>
+
+
+
+
+#define SMEAR 0
+#define BLACKEN 1
+
+
+
+class WaveEffect;
+class WaveWindow;
+
+
+class WaveConfig
+{
+public:
+       WaveConfig();
+       void copy_from(WaveConfig &src);
+       int equivalent(WaveConfig &src);
+       void interpolate(WaveConfig &prev, 
+               WaveConfig &next, 
+               long prev_frame, 
+               long next_frame, 
+               long current_frame);
+       int mode;
+       int reflective;
+       float amplitude;
+       float phase;
+       float wavelength;
+};
+
+class WaveSmear : public BC_Radial
+{
+public:
+       WaveSmear(WaveEffect *plugin, WaveWindow *window, int x, int y);
+       int handle_event();
+       WaveEffect *plugin;
+       WaveWindow *window;
+};
+
+class WaveBlacken : public BC_Radial
+{
+public:
+       WaveBlacken(WaveEffect *plugin, WaveWindow *window, int x, int y);
+       int handle_event();
+       WaveEffect *plugin;
+       WaveWindow *window;
+};
+
+
+class WaveReflective : public BC_CheckBox
+{
+public:
+       WaveReflective(WaveEffect *plugin, int x, int y);
+       int handle_event();
+       WaveEffect *plugin;
+};
+
+class WaveAmplitude : public BC_FSlider
+{
+public:
+       WaveAmplitude(WaveEffect *plugin, int x, int y);
+       int handle_event();
+       WaveEffect *plugin;
+};
+
+class WavePhase : public BC_FSlider
+{
+public:
+       WavePhase(WaveEffect *plugin, int x, int y);
+       int handle_event();
+       WaveEffect *plugin;
+};
+
+class WaveLength : public BC_FSlider
+{
+public:
+       WaveLength(WaveEffect *plugin, int x, int y);
+       int handle_event();
+       WaveEffect *plugin;
+};
+
+
+
+
+
+
+
+
+
+class WaveWindow : public PluginClientWindow
+{
+public:
+       WaveWindow(WaveEffect *plugin);
+       ~WaveWindow();
+       void create_objects();
+       void update_mode();
+       WaveEffect *plugin;
+//     WaveSmear *smear;
+//     WaveBlacken *blacken;
+//     WaveReflective *reflective;
+       WaveAmplitude *amplitude;
+       WavePhase *phase;
+       WaveLength *wavelength;
+};
+
+
+
+
+
+
+
+class WaveServer : public LoadServer
+{
+public:
+       WaveServer(WaveEffect *plugin, int cpus);
+       void init_packages();
+       LoadClient* new_client();
+       LoadPackage* new_package();
+       WaveEffect *plugin;
+};
+
+class WavePackage : public LoadPackage
+{
+public:
+       WavePackage();
+       int row1, row2;
+};
+
+class WaveUnit : public LoadClient
+{
+public:
+       WaveUnit(WaveEffect *plugin, WaveServer *server);
+       void process_package(LoadPackage *package);
+       WaveEffect *plugin;
+};
+
+
+
+
+
+
+
+
+
+class WaveEffect : public PluginVClient
+{
+public:
+       WaveEffect(PluginServer *server);
+       ~WaveEffect();
+
+       PLUGIN_CLASS_MEMBERS(WaveConfig)
+       int process_realtime(VFrame *input, VFrame *output);
+       int is_realtime();
+       void save_data(KeyFrame *keyframe);
+       void read_data(KeyFrame *keyframe);
+       void update_gui();
+
+       VFrame *temp_frame;
+       VFrame *input, *output;
+       WaveServer *engine;
+};
+
+
+
+
+
+
+
+
+
+
+
+
+
+WaveConfig::WaveConfig()
+{
+       mode = SMEAR;
+       reflective = 0;
+       amplitude = 10;
+       phase = 0;
+       wavelength = 10;
+}
+
+void WaveConfig::copy_from(WaveConfig &src)
+{
+       this->mode = src.mode;
+       this->reflective = src.reflective;
+       this->amplitude = src.amplitude;
+       this->phase = src.phase;
+       this->wavelength = src.wavelength;
+}
+
+int WaveConfig::equivalent(WaveConfig &src)
+{
+       return 
+               (this->mode == src.mode) &&
+               EQUIV(this->reflective, src.reflective) &&
+               EQUIV(this->amplitude, src.amplitude) &&
+               EQUIV(this->phase, src.phase) &&
+               EQUIV(this->wavelength, src.wavelength);
+}
+
+void WaveConfig::interpolate(WaveConfig &prev, 
+               WaveConfig &next, 
+               long prev_frame, 
+               long next_frame, 
+               long current_frame)
+{
+       double next_scale = (double)(current_frame - prev_frame) / (next_frame - prev_frame);
+       double prev_scale = (double)(next_frame - current_frame) / (next_frame - prev_frame);
+
+       this->amplitude = prev.amplitude * prev_scale + next.amplitude * next_scale;
+       this->phase = prev.phase * prev_scale + next.phase * next_scale;
+       this->wavelength = prev.wavelength * prev_scale + next.wavelength * next_scale;
+       this->mode = prev.mode;
+       this->reflective = prev.reflective;
+}
+
+
+
+
+
+
+
+
+WaveSmear::WaveSmear(WaveEffect *plugin, WaveWindow *window, int x, int y)
+ : BC_Radial(x, y, plugin->config.mode == SMEAR, _("Smear"))
+{
+       this->plugin = plugin;
+       this->window = window;
+}
+int WaveSmear::handle_event()
+{
+       plugin->config.mode = SMEAR;
+       window->update_mode();
+       plugin->send_configure_change();
+       return 1;
+}
+
+
+
+
+WaveBlacken::WaveBlacken(WaveEffect *plugin, WaveWindow *window, int x, int y)
+ : BC_Radial(x, y, plugin->config.mode == BLACKEN, _("Blacken"))
+{
+       this->plugin = plugin;
+       this->window = window;
+}
+int WaveBlacken::handle_event()
+{
+       plugin->config.mode = BLACKEN;
+       window->update_mode();
+       plugin->send_configure_change();
+       return 1;
+}
+
+
+
+
+
+
+WaveReflective::WaveReflective(WaveEffect *plugin, int x, int y)
+ : BC_CheckBox(x, y, plugin->config.reflective, _("Reflective"))
+{
+       this->plugin = plugin;
+}
+int WaveReflective::handle_event()
+{
+       plugin->config.reflective = get_value();
+       plugin->send_configure_change();
+       return 1;
+}
+
+
+WaveAmplitude::WaveAmplitude(WaveEffect *plugin, int x, int y)
+ : BC_FSlider(x, 
+                       y,
+                       0,
+                       200, 
+                       200, 
+                       (float)0, 
+                       (float)100, 
+                       plugin->config.amplitude)
+{
+       this->plugin = plugin;
+}
+int WaveAmplitude::handle_event()
+{
+       plugin->config.amplitude = get_value();
+       plugin->send_configure_change();
+       return 1;
+}
+
+
+
+WavePhase::WavePhase(WaveEffect *plugin, int x, int y)
+ : BC_FSlider(x, 
+                       y,
+                       0,
+                       200, 
+                       200, 
+                       (float)0, 
+                       (float)360, 
+                       plugin->config.phase)
+{
+       this->plugin = plugin;
+}
+int WavePhase::handle_event()
+{
+       plugin->config.phase = get_value();
+       plugin->send_configure_change();
+       return 1;
+}
+
+WaveLength::WaveLength(WaveEffect *plugin, int x, int y)
+ : BC_FSlider(x, 
+                       y,
+                       0,
+                       200, 
+                       200, 
+                       (float)0, 
+                       (float)50, 
+                       plugin->config.wavelength)
+{
+       this->plugin = plugin;
+}
+int WaveLength::handle_event()
+{
+       plugin->config.wavelength = get_value();
+       plugin->send_configure_change();
+       return 1;
+}
+
+
+
+
+
+
+
+
+
+WaveWindow::WaveWindow(WaveEffect *plugin)
+ : PluginClientWindow(plugin, 
+       320, 
+       150, 
+       320, 
+       150, 
+       0)
+{
+       this->plugin = plugin;
+}
+
+WaveWindow::~WaveWindow()
+{
+}
+
+void WaveWindow::create_objects()
+{
+       int x = 10, y = 10, x1 = 100;
+
+//     add_subwindow(new BC_Title(x, y, _("Mode:")));
+//     add_subwindow(smear = new WaveSmear(plugin, this, x1, y));
+//     y += 20;
+//     add_subwindow(blacken = new WaveBlacken(plugin, this, x1, y));
+//     y += 30;
+//     add_subwindow(reflective = new WaveReflective(plugin, x1, y));
+//     y += 30;
+       add_subwindow(new BC_Title(x, y, _("Amplitude:")));
+       add_subwindow(amplitude = new WaveAmplitude(plugin, x1, y));
+       y += 30;
+       add_subwindow(new BC_Title(x, y, _("Phase:")));
+       add_subwindow(phase = new WavePhase(plugin, x1, y));
+       y += 30;
+       add_subwindow(new BC_Title(x, y, _("Wavelength:")));
+       add_subwindow(wavelength = new WaveLength(plugin, x1, y));
+
+       show_window();
+       flush();
+}
+
+void WaveWindow::update_mode()
+{
+//     smear->update(plugin->config.mode == SMEAR);
+//     blacken->update(plugin->config.mode == BLACKEN);
+}
+
+
+
+
+
+
+
+REGISTER_PLUGIN(WaveEffect)
+
+
+
+
+WaveEffect::WaveEffect(PluginServer *server)
+ : PluginVClient(server)
+{
+       temp_frame = 0;
+       engine = 0;
+       
+}
+
+WaveEffect::~WaveEffect()
+{
+       
+
+       if(temp_frame) delete temp_frame;
+       if(engine) delete engine;
+}
+
+
+const char* WaveEffect::plugin_title() { return _("Wave"); }
+int WaveEffect::is_realtime() { return 1; }
+
+
+NEW_WINDOW_MACRO(WaveEffect, WaveWindow)
+
+void WaveEffect::update_gui()
+{
+       if(thread)
+       {
+               thread->window->lock_window();
+               load_configuration();
+               ((WaveWindow*)thread->window)->update_mode();
+//             thread->window->reflective->update(config.reflective);
+               ((WaveWindow*)thread->window)->amplitude->update(config.amplitude);
+               ((WaveWindow*)thread->window)->phase->update(config.phase);
+               ((WaveWindow*)thread->window)->wavelength->update(config.wavelength);
+               thread->window->unlock_window();
+       }
+}
+
+
+LOAD_CONFIGURATION_MACRO(WaveEffect, WaveConfig)
+
+
+void WaveEffect::save_data(KeyFrame *keyframe)
+{
+       FileXML output;
+
+// cause data to be stored directly in text
+       output.set_shared_output(keyframe->get_data(), MESSAGESIZE);
+       output.tag.set_title("WAVE");
+       output.tag.set_property("MODE", config.mode);
+       output.tag.set_property("REFLECTIVE", config.reflective);
+       output.tag.set_property("AMPLITUDE", config.amplitude);
+       output.tag.set_property("PHASE", config.phase);
+       output.tag.set_property("WAVELENGTH", config.wavelength);
+       output.append_tag();
+       output.tag.set_title("/WAVE");
+       output.append_tag();
+       output.append_newline();
+       output.terminate_string();
+}
+
+void WaveEffect::read_data(KeyFrame *keyframe)
+{
+       FileXML input;
+
+       input.set_shared_input(keyframe->get_data(), strlen(keyframe->get_data()));
+
+       while(!input.read_tag())
+       {
+               if(input.tag.title_is("WAVE"))
+               {
+                       config.mode = input.tag.get_property("MODE", config.mode);
+                       config.reflective = input.tag.get_property("REFLECTIVE", config.reflective);
+                       config.amplitude = input.tag.get_property("AMPLITUDE", config.amplitude);
+                       config.phase = input.tag.get_property("PHASE", config.phase);
+                       config.wavelength = input.tag.get_property("WAVELENGTH", config.wavelength);
+               }
+       }
+}
+
+
+int WaveEffect::process_realtime(VFrame *input, VFrame *output)
+{
+       load_configuration();
+
+
+
+//printf("WaveEffect::process_realtime %f %d\n", config.radius, config.use_intensity);
+       this->input = input;
+       this->output = output;
+
+       if(EQUIV(config.amplitude, 0) || EQUIV(config.wavelength, 0))
+       {
+               if(input->get_rows()[0] != output->get_rows()[0])
+                       output->copy_from(input);
+       }
+       else
+       {
+               if(input->get_rows()[0] == output->get_rows()[0])
+               {
+                       if(!temp_frame) temp_frame = new VFrame(0,
+                               -1,
+                               input->get_w(),
+                               input->get_h(),
+                               input->get_color_model(),
+                               -1);
+                       temp_frame->copy_from(input);
+                       this->input = temp_frame;
+               }
+               
+               
+               if(!engine)
+               {
+                       engine = new WaveServer(this, (PluginClient::smp + 1));
+               }
+               
+               engine->process_packages();
+       }
+       
+       
+       
+       return 0;
+}
+
+
+
+
+
+
+WavePackage::WavePackage()
+ : LoadPackage()
+{
+}
+
+
+
+
+
+
+WaveUnit::WaveUnit(WaveEffect *plugin, WaveServer *server)
+ : LoadClient(server)
+{
+       this->plugin = plugin;
+}
+
+
+
+#define WITHIN(a, b, c) ((((a) <= (b)) && ((b) <= (c))) ? 1 : 0)
+
+static float bilinear(double  x,
+         double  y,
+         float  *v)
+{
+       double m0, m1;
+       x = fmod(x, 1.0);
+       y = fmod(y, 1.0);
+
+       if(x < 0)
+       x += 1.0;
+       if(y < 0)
+       y += 1.0;
+
+       m0 = (1.0 - x) * v[0] + x * v[1];
+       m1 = (1.0 - x) * v[2] + x * v[3];
+
+       return((1.0 - y) * m0 + y * m1);
+}
+
+void WaveUnit::process_package(LoadPackage *package)
+{
+       WavePackage *pkg = (WavePackage*)package;
+       int w = plugin->input->get_w();
+       int h = plugin->input->get_h();
+       double cen_x, cen_y;       /* Center of wave */
+       double xhsiz, yhsiz;       /* Half size of selection */
+       double radius;             /* Radius */
+       double amnt, d;
+       double needx, needy;
+       double dx, dy;
+       double xscale, yscale;
+       double wavelength;
+       int xi, yi;
+       float values[4];
+       float val;
+       int x1, y1, x2, y2;
+       int x1_in, y1_in, x2_in, y2_in;
+       double phase = plugin->config.phase * M_PI / 180;
+       
+       
+       x1 = y1 = 0;
+       x2 = w;
+       y2 = h;
+       cen_x = (double) (x2 - 1 + x1) / 2.0;
+       cen_y = (double) (y2 - 1 + y1) / 2.0;
+       xhsiz = (double) (x2 - x1) / 2.0;
+       yhsiz = (double) (y2 - y1) / 2.0;
+
+       if (xhsiz < yhsiz)
+    {
+       xscale = yhsiz / xhsiz;
+       yscale = 1.0;
+    }
+       else if (xhsiz > yhsiz)
+    {
+       xscale = 1.0;
+       yscale = xhsiz / yhsiz;
+    }
+       else
+    {
+       xscale = 1.0;
+       yscale = 1.0;
+    }
+
+       radius  = MAX(xhsiz, yhsiz);
+
+
+       wavelength = plugin->config.wavelength / 100 * radius;
+
+
+
+
+#define WAVE(type, components, chroma_offset) \
+{ \
+       int row_size = w * components; \
+       type **in_rows = (type**)plugin->input->get_rows(); \
+       for(int y = pkg->row1; y < pkg->row2; y++) \
+       { \
+               type *dest = (type*)plugin->output->get_rows()[y]; \
+ \
+               for(int x = x1; x < x2; x++) \
+               { \
+                       dx = (x - cen_x) * xscale; \
+                       dy = (y - cen_y) * yscale; \
+                   d = sqrt(dx * dx + dy * dy); \
+ \
+                       if(plugin->config.reflective) \
+                       { \
+                       amnt = plugin->config.amplitude *  \
+                                       fabs(sin(((d / wavelength) *  \
+                                               (2.0 * M_PI) + \
+                                               phase))); \
+ \
+                       needx = (amnt * dx) / xscale + cen_x; \
+                       needy = (amnt * dy) / yscale + cen_y; \
+                       } \
+                       else \
+                       { \
+                       amnt = plugin->config.amplitude *  \
+                                       sin(((d / wavelength) *  \
+                                               (2.0 * M_PI) + \
+                                       phase)); \
+ \
+                       needx = (amnt + dx) / xscale + cen_x; \
+                       needy = (amnt + dy) / yscale + cen_y; \
+                       } \
+ \
+                       xi = (int)needx; \
+                       yi = (int)needy; \
+ \
+                       if(plugin->config.mode == SMEAR) \
+               { \
+                       if(xi > w - 2) \
+                               { \
+                                       xi = w - 2; \
+                               } \
+                       else  \
+                               if(xi < 0) \
+                               { \
+                                       xi = 0; \
+                               } \
+ \
+                       if(yi > h - 2) \
+                               { \
+                                       yi = h - 2; \
+                               } \
+                       else  \
+                               if(yi < 0) \
+                               { \
+                                       yi = 0; \
+                               } \
+               } \
+ \
+                       type *p = in_rows[CLIP(yi, 0, h - 1)] + \
+                               CLIP(xi, 0, w - 1) * components; \
+                       x1_in = WITHIN(0, xi, w - 1); \
+                       y1_in = WITHIN(0, yi, h - 1); \
+                       x2_in = WITHIN(0, xi + 1, w - 1); \
+                       y2_in = WITHIN(0, yi + 1, h - 1); \
+ \
+ \
+                       for(int k = 0; k < components; k++) \
+               { \
+                       if (x1_in && y1_in) \
+                                       values[0] = *(p + k); \
+                       else \
+                                       values[0] = ((k == 1 || k == 2) ? 0 : chroma_offset); \
+ \
+                       if (x2_in && y1_in) \
+                                       values[1] = *(p + components + k); \
+                       else \
+                                       values[1] = ((k == 1 || k == 2) ? 0 : chroma_offset); \
+ \
+                       if (x1_in && y2_in) \
+                                       values[2] = *(p + row_size + k); \
+                       else \
+                                       values[2] = ((k == 1 || k == 2) ? 0 : chroma_offset); \
+ \
+                       if (x2_in) \
+                               { \
+                                       if (y2_in) \
+                                       values[3] = *(p + row_size + components + k); \
+                                       else \
+                                       values[3] = ((k == 1 || k == 2) ? 0 : chroma_offset); \
+                               } \
+                       else \
+                                       values[3] = ((k == 1 || k == 2) ? 0 : chroma_offset); \
+ \
+                       val = bilinear(needx, needy, values); \
+ \
+                       *dest++ = (type)val; \
+               } \
+               } \
+       } \
+}
+
+
+
+
+       switch(plugin->input->get_color_model())
+       {
+               case BC_RGB888:
+                       WAVE(unsigned char, 3, 0x0);
+                       break;
+               case BC_RGB_FLOAT:
+                       WAVE(float, 3, 0x0);
+                       break;
+               case BC_YUV888:
+                       WAVE(unsigned char, 3, 0x80);
+                       break;
+               case BC_RGB161616:
+                       WAVE(uint16_t, 3, 0x0);
+                       break;
+               case BC_YUV161616:
+                       WAVE(uint16_t, 3, 0x8000);
+                       break;
+               case BC_RGBA_FLOAT:
+                       WAVE(unsigned char, 4, 0x0);
+                       break;
+               case BC_RGBA8888:
+                       WAVE(unsigned char, 4, 0x0);
+                       break;
+               case BC_YUVA8888:
+                       WAVE(unsigned char, 4, 0x8000);
+                       break;
+               case BC_RGBA16161616:
+                       WAVE(uint16_t, 4, 0x0);
+                       break;
+               case BC_YUVA16161616:
+                       WAVE(uint16_t, 4, 0x8000);
+                       break;
+       }
+
+
+
+
+}
+
+
+
+
+
+
+WaveServer::WaveServer(WaveEffect *plugin, int cpus)
+ : LoadServer(cpus, cpus)
+{
+       this->plugin = plugin;
+}
+
+void WaveServer::init_packages()
+{
+       for(int i = 0; i < LoadServer::get_total_packages(); i++)
+       {
+               WavePackage *pkg = (WavePackage*)get_package(i);
+               pkg->row1 = plugin->input->get_h() * i / LoadServer::get_total_packages();
+               pkg->row2 = plugin->input->get_h() * (i + 1) / LoadServer::get_total_packages();
+       }
+}
+
+
+LoadClient* WaveServer::new_client()
+{
+       return new WaveUnit(plugin, this);
+}
+
+LoadPackage* WaveServer::new_package()
+{
+       return new WavePackage;
+}
+