--- /dev/null
+
+/*
+ * 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 "overlayframe.h"
+#include "pluginvclient.h"
+#include "vframe.h"
+
+#include <string.h>
+#include <stdint.h>
+
+
+class Overlay;
+class OverlayWindow;
+
+
+class OverlayConfig
+{
+public:
+ OverlayConfig();
+
+ static const char* mode_to_text(int mode);
+ int mode;
+
+ static const char* direction_to_text(int direction);
+ int direction;
+ enum
+ {
+ BOTTOM_FIRST,
+ TOP_FIRST
+ };
+
+ static const char* output_to_text(int output_layer);
+ int output_layer;
+ enum
+ {
+ TOP,
+ BOTTOM
+ };
+};
+
+class OverlayMode : public BC_PopupMenu
+{
+public:
+ OverlayMode(Overlay *plugin,
+ int x,
+ int y);
+ void create_objects();
+ int handle_event();
+ Overlay *plugin;
+};
+
+class OverlayDirection : public BC_PopupMenu
+{
+public:
+ OverlayDirection(Overlay *plugin,
+ int x,
+ int y);
+ void create_objects();
+ int handle_event();
+ Overlay *plugin;
+};
+
+class OverlayOutput : public BC_PopupMenu
+{
+public:
+ OverlayOutput(Overlay *plugin,
+ int x,
+ int y);
+ void create_objects();
+ int handle_event();
+ Overlay *plugin;
+};
+
+
+class OverlayWindow : public PluginClientWindow
+{
+public:
+ OverlayWindow(Overlay *plugin);
+ ~OverlayWindow();
+
+ void create_objects();
+
+
+ Overlay *plugin;
+ OverlayMode *mode;
+ OverlayDirection *direction;
+ OverlayOutput *output;
+};
+
+class Overlay : public PluginVClient
+{
+public:
+ Overlay(PluginServer *server);
+ ~Overlay();
+
+
+ PLUGIN_CLASS_MEMBERS(OverlayConfig);
+
+ int process_buffer(VFrame **frame,
+ int64_t start_position,
+ double frame_rate);
+ int is_realtime();
+ int is_multichannel();
+ int is_synthesis();
+ void save_data(KeyFrame *keyframe);
+ void read_data(KeyFrame *keyframe);
+ void update_gui();
+ int handle_opengl();
+
+ OverlayFrame *overlayer;
+ VFrame *temp;
+ int current_layer;
+ int output_layer;
+// Inclusive layer numbers
+ int input_layer1;
+ int input_layer2;
+};
+
+OverlayConfig::OverlayConfig()
+{
+ mode = TRANSFER_NORMAL;
+ direction = OverlayConfig::BOTTOM_FIRST;
+ output_layer = OverlayConfig::TOP;
+}
+
+const char* OverlayConfig::mode_to_text(int mode)
+{
+ switch(mode) {
+ case TRANSFER_NORMAL: return _("Normal");
+ case TRANSFER_ADDITION: return _("Addition");
+ case TRANSFER_SUBTRACT: return _("Subtract");
+ case TRANSFER_MULTIPLY: return _("Multiply");
+ case TRANSFER_DIVIDE: return _("Divide");
+ case TRANSFER_REPLACE: return _("Replace");
+ case TRANSFER_MAX: return _("Max");
+ case TRANSFER_MIN: return _("Min");
+ case TRANSFER_AVERAGE: return _("Average");
+ case TRANSFER_DARKEN: return _("Darken");
+ case TRANSFER_LIGHTEN: return _("Lighten");
+ case TRANSFER_DST: return _("Dst");
+ case TRANSFER_DST_ATOP: return _("DstAtop");
+ case TRANSFER_DST_IN: return _("DstIn");
+ case TRANSFER_DST_OUT: return _("DstOut");
+ case TRANSFER_DST_OVER: return _("DstOver");
+ case TRANSFER_SRC: return _("Src");
+ case TRANSFER_SRC_ATOP: return _("SrcAtop");
+ case TRANSFER_SRC_IN: return _("SrcIn");
+ case TRANSFER_SRC_OUT: return _("SrcOut");
+ case TRANSFER_SRC_OVER: return _("SrcOver");
+ case TRANSFER_OR: return _("Or");
+ case TRANSFER_XOR: return _("Xor");
+ default: break;
+ }
+ return _("Normal");
+}
+
+const char* OverlayConfig::direction_to_text(int direction)
+{
+ switch(direction)
+ {
+ case OverlayConfig::BOTTOM_FIRST: return _("Bottom first");
+ case OverlayConfig::TOP_FIRST: return _("Top first");
+ }
+ return "";
+}
+
+const char* OverlayConfig::output_to_text(int output_layer)
+{
+ switch(output_layer)
+ {
+ case OverlayConfig::TOP: return _("Top");
+ case OverlayConfig::BOTTOM: return _("Bottom");
+ }
+ return "";
+}
+
+
+
+
+
+
+
+
+
+OverlayWindow::OverlayWindow(Overlay *plugin)
+ : PluginClientWindow(plugin,
+ 300,
+ 160,
+ 300,
+ 160,
+ 0)
+{
+ this->plugin = plugin;
+}
+
+OverlayWindow::~OverlayWindow()
+{
+}
+
+void OverlayWindow::create_objects()
+{
+ int x = 10, y = 10;
+
+ BC_Title *title;
+ add_subwindow(title = new BC_Title(x, y, _("Mode:")));
+ add_subwindow(mode = new OverlayMode(plugin,
+ x + title->get_w() + 5,
+ y));
+ mode->create_objects();
+
+ y += 30;
+ add_subwindow(title = new BC_Title(x, y, _("Layer order:")));
+ add_subwindow(direction = new OverlayDirection(plugin,
+ x + title->get_w() + 5,
+ y));
+ direction->create_objects();
+
+ y += 30;
+ add_subwindow(title = new BC_Title(x, y, _("Output layer:")));
+ add_subwindow(output = new OverlayOutput(plugin,
+ x + title->get_w() + 5,
+ y));
+ output->create_objects();
+
+ show_window();
+ flush();
+}
+
+
+
+
+
+
+
+OverlayMode::OverlayMode(Overlay *plugin, int x, int y)
+ : BC_PopupMenu(x, y, 150,
+ OverlayConfig::mode_to_text(plugin->config.mode), 1)
+{
+ this->plugin = plugin;
+}
+
+void OverlayMode::create_objects()
+{
+ for(int i = 0; i < TRANSFER_TYPES; i++)
+ add_item(new BC_MenuItem(OverlayConfig::mode_to_text(i)));
+}
+
+int OverlayMode::handle_event()
+{
+ char *text = get_text();
+
+ for(int i = 0; i < TRANSFER_TYPES; i++)
+ {
+ if(!strcmp(text, OverlayConfig::mode_to_text(i)))
+ {
+ plugin->config.mode = i;
+ break;
+ }
+ }
+
+ plugin->send_configure_change();
+ return 1;
+}
+
+
+OverlayDirection::OverlayDirection(Overlay *plugin,
+ int x,
+ int y)
+ : BC_PopupMenu(x,
+ y,
+ 150,
+ OverlayConfig::direction_to_text(plugin->config.direction),
+ 1)
+{
+ this->plugin = plugin;
+}
+
+void OverlayDirection::create_objects()
+{
+ add_item(new BC_MenuItem(
+ OverlayConfig::direction_to_text(
+ OverlayConfig::TOP_FIRST)));
+ add_item(new BC_MenuItem(
+ OverlayConfig::direction_to_text(
+ OverlayConfig::BOTTOM_FIRST)));
+}
+
+int OverlayDirection::handle_event()
+{
+ char *text = get_text();
+
+ if(!strcmp(text,
+ OverlayConfig::direction_to_text(
+ OverlayConfig::TOP_FIRST)))
+ plugin->config.direction = OverlayConfig::TOP_FIRST;
+ else
+ if(!strcmp(text,
+ OverlayConfig::direction_to_text(
+ OverlayConfig::BOTTOM_FIRST)))
+ plugin->config.direction = OverlayConfig::BOTTOM_FIRST;
+
+ plugin->send_configure_change();
+ return 1;
+}
+
+
+OverlayOutput::OverlayOutput(Overlay *plugin,
+ int x,
+ int y)
+ : BC_PopupMenu(x,
+ y,
+ 100,
+ OverlayConfig::output_to_text(plugin->config.output_layer),
+ 1)
+{
+ this->plugin = plugin;
+}
+
+void OverlayOutput::create_objects()
+{
+ add_item(new BC_MenuItem(
+ OverlayConfig::output_to_text(
+ OverlayConfig::TOP)));
+ add_item(new BC_MenuItem(
+ OverlayConfig::output_to_text(
+ OverlayConfig::BOTTOM)));
+}
+
+int OverlayOutput::handle_event()
+{
+ char *text = get_text();
+
+ if(!strcmp(text,
+ OverlayConfig::output_to_text(
+ OverlayConfig::TOP)))
+ plugin->config.output_layer = OverlayConfig::TOP;
+ else
+ if(!strcmp(text,
+ OverlayConfig::output_to_text(
+ OverlayConfig::BOTTOM)))
+ plugin->config.output_layer = OverlayConfig::BOTTOM;
+
+ plugin->send_configure_change();
+ return 1;
+}
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+REGISTER_PLUGIN(Overlay)
+
+
+
+
+
+
+Overlay::Overlay(PluginServer *server)
+ : PluginVClient(server)
+{
+
+ overlayer = 0;
+ temp = 0;
+}
+
+
+Overlay::~Overlay()
+{
+
+ if(overlayer) delete overlayer;
+ if(temp) delete temp;
+}
+
+
+
+int Overlay::process_buffer(VFrame **frame,
+ int64_t start_position,
+ double frame_rate)
+{
+ load_configuration();
+
+
+ if(!temp) temp = new VFrame(0,
+ -1,
+ frame[0]->get_w(),
+ frame[0]->get_h(),
+ frame[0]->get_color_model(),
+ -1);
+
+ if(!overlayer)
+ overlayer = new OverlayFrame(get_project_smp() + 1);
+
+ int step;
+ VFrame *output;
+
+ if(config.direction == OverlayConfig::BOTTOM_FIRST)
+ {
+ input_layer1 = get_total_buffers() - 1;
+ input_layer2 = -1;
+ step = -1;
+ }
+ else
+ {
+ input_layer1 = 0;
+ input_layer2 = get_total_buffers();
+ step = 1;
+ }
+
+ if(config.output_layer == OverlayConfig::TOP)
+ {
+ output_layer = 0;
+ }
+ else
+ {
+ output_layer = get_total_buffers() - 1;
+ }
+
+
+
+// Direct copy the first layer
+ output = frame[output_layer];
+ read_frame(output,
+ input_layer1,
+ start_position,
+ frame_rate,
+ get_use_opengl());
+
+ if(get_total_buffers() == 1) return 0;
+
+
+
+ current_layer = input_layer1;
+ if(get_use_opengl())
+ run_opengl();
+
+ for(int i = input_layer1 + step; i != input_layer2; i += step)
+ {
+ read_frame(temp,
+ i,
+ start_position,
+ frame_rate,
+ get_use_opengl());
+
+// Call the opengl handler once for each layer
+ if(get_use_opengl())
+ {
+ current_layer = i;
+ run_opengl();
+ }
+ else
+ {
+ overlayer->overlay(output,
+ temp,
+ 0,
+ 0,
+ output->get_w(),
+ output->get_h(),
+ 0,
+ 0,
+ output->get_w(),
+ output->get_h(),
+ 1,
+ config.mode,
+ NEAREST_NEIGHBOR);
+ }
+ }
+
+
+ return 0;
+}
+
+int Overlay::handle_opengl()
+{
+#ifdef HAVE_GL
+ static const char *get_pixels_frag =
+ "uniform sampler2D src_tex;\n"
+ "uniform sampler2D dst_tex;\n"
+ "uniform vec2 dst_tex_dimensions;\n"
+ "uniform vec3 chroma_offset;\n"
+ "void main()\n"
+ "{\n"
+ " vec4 result_color;\n"
+ " vec4 dst_color = texture2D(dst_tex, gl_FragCoord.xy / dst_tex_dimensions);\n"
+ " vec4 src_color = texture2D(src_tex, gl_TexCoord[0].st);\n"
+ " src_color.rgb -= chroma_offset;\n"
+ " dst_color.rgb -= chroma_offset;\n";
+
+ static const char *put_pixels_frag =
+ " result_color.rgb += chroma_offset;\n"
+ " result_color.rgb = mix(dst_color.rgb, result_color.rgb, src_color.a);\n"
+ " result_color.a = max(src_color.a, dst_color.a);\n"
+ " gl_FragColor = result_color;\n"
+ "}\n";
+
+ static const char *blend_add_frag =
+ " result_color.rgb = dst_color.rgb + src_color.rgb;\n";
+
+ static const char *blend_max_frag =
+ " result_color.r = max(abs(dst_color.r, src_color.r);\n"
+ " result_color.g = max(abs(dst_color.g, src_color.g);\n"
+ " result_color.b = max(abs(dst_color.b, src_color.b);\n";
+
+ static const char *blend_min_frag =
+ " result_color.r = min(abs(dst_color.r, src_color.r);\n"
+ " result_color.g = min(abs(dst_color.g, src_color.g);\n"
+ " result_color.b = min(abs(dst_color.b, src_color.b);\n";
+
+ static const char *blend_subtract_frag =
+ " result_color.rgb = dst_color.rgb - src_color.rgb;\n";
+
+
+ static const char *blend_multiply_frag =
+ " result_color.rgb = dst_color.rgb * src_color.rgb;\n";
+
+ static const char *blend_divide_frag =
+ " result_color.rgb = dst_color.rgb / src_color.rgb;\n"
+ " if(src_color.r == 0.0) result_color.r = 1.0;\n"
+ " if(src_color.g == 0.0) result_color.g = 1.0;\n"
+ " if(src_color.b == 0.0) result_color.b = 1.0;\n";
+
+
+ VFrame *src = temp;
+ VFrame *dst = get_output(output_layer);
+
+ dst->enable_opengl();
+ dst->init_screen();
+
+ const char *shader_stack[] = { 0, 0, 0 };
+ int current_shader = 0;
+
+
+
+
+// Direct copy layer
+ if(config.mode == TRANSFER_REPLACE)
+ {
+ src->to_texture();
+ src->bind_texture(0);
+ dst->enable_opengl();
+ dst->init_screen();
+
+// Multiply alpha
+ glDisable(GL_BLEND);
+ src->draw_texture();
+ }
+ else
+ if(config.mode == TRANSFER_NORMAL)
+ {
+ dst->enable_opengl();
+ dst->init_screen();
+
+// Move destination to screen
+ if(dst->get_opengl_state() != VFrame::SCREEN)
+ {
+ dst->to_texture();
+ dst->bind_texture(0);
+ dst->draw_texture();
+ }
+
+ src->to_texture();
+ src->bind_texture(0);
+ dst->enable_opengl();
+ dst->init_screen();
+
+ glEnable(GL_BLEND);
+ glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
+ src->draw_texture();
+ }
+ else
+ {
+// Read destination back to texture
+ dst->to_texture();
+
+ src->enable_opengl();
+ src->init_screen();
+ src->to_texture();
+
+ dst->enable_opengl();
+ dst->init_screen();
+ src->bind_texture(0);
+ dst->bind_texture(1);
+
+
+ shader_stack[current_shader++] = get_pixels_frag;
+
+ switch(config.mode)
+ {
+ case TRANSFER_ADDITION:
+ shader_stack[current_shader++] = blend_add_frag;
+ break;
+ case TRANSFER_SUBTRACT:
+ shader_stack[current_shader++] = blend_subtract_frag;
+ break;
+ case TRANSFER_MULTIPLY:
+ shader_stack[current_shader++] = blend_multiply_frag;
+ break;
+ case TRANSFER_DIVIDE:
+ shader_stack[current_shader++] = blend_divide_frag;
+ break;
+ case TRANSFER_MAX:
+ shader_stack[current_shader++] = blend_max_frag;
+ break;
+ case TRANSFER_MIN:
+ shader_stack[current_shader++] = blend_min_frag;
+ break;
+ }
+
+ shader_stack[current_shader++] = put_pixels_frag;
+
+ unsigned int shader_id = 0;
+ shader_id = VFrame::make_shader(0,
+ shader_stack[0],
+ shader_stack[1],
+ shader_stack[2],
+ 0);
+
+ glUseProgram(shader_id);
+ glUniform1i(glGetUniformLocation(shader_id, "src_tex"), 0);
+ glUniform1i(glGetUniformLocation(shader_id, "dst_tex"), 1);
+ if(BC_CModels::is_yuv(dst->get_color_model()))
+ glUniform3f(glGetUniformLocation(shader_id, "chroma_offset"), 0.0, 0.5, 0.5);
+ else
+ glUniform3f(glGetUniformLocation(shader_id, "chroma_offset"), 0.0, 0.0, 0.0);
+ glUniform2f(glGetUniformLocation(shader_id, "dst_tex_dimensions"),
+ (float)dst->get_texture_w(),
+ (float)dst->get_texture_h());
+
+ glDisable(GL_BLEND);
+ src->draw_texture();
+ glUseProgram(0);
+ }
+
+ glDisable(GL_BLEND);
+ glActiveTexture(GL_TEXTURE1);
+ glDisable(GL_TEXTURE_2D);
+ glActiveTexture(GL_TEXTURE0);
+ glDisable(GL_TEXTURE_2D);
+
+ dst->set_opengl_state(VFrame::SCREEN);
+#endif
+ return 0;
+}
+
+
+const char* Overlay::plugin_title() { return _("Overlay"); }
+int Overlay::is_realtime() { return 1; }
+int Overlay::is_multichannel() { return 1; }
+int Overlay::is_synthesis() { return 1; }
+
+
+
+NEW_WINDOW_MACRO(Overlay, OverlayWindow)
+
+
+
+int Overlay::load_configuration()
+{
+ KeyFrame *prev_keyframe;
+ prev_keyframe = get_prev_keyframe(get_source_position());
+ read_data(prev_keyframe);
+ return 0;
+}
+
+
+void Overlay::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("OVERLAY");
+ output.tag.set_property("MODE", config.mode);
+ output.tag.set_property("DIRECTION", config.direction);
+ output.tag.set_property("OUTPUT_LAYER", config.output_layer);
+ output.append_tag();
+ output.tag.set_title("/OVERLAY");
+ output.append_tag();
+ output.terminate_string();
+}
+
+void Overlay::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("OVERLAY"))
+ {
+ config.mode = input.tag.get_property("MODE", config.mode);
+ config.direction = input.tag.get_property("DIRECTION", config.direction);
+ config.output_layer = input.tag.get_property("OUTPUT_LAYER", config.output_layer);
+ }
+ }
+}
+
+void Overlay::update_gui()
+{
+ if(thread)
+ {
+ thread->window->lock_window("Overlay::update_gui");
+ ((OverlayWindow*)thread->window)->mode->set_text(OverlayConfig::mode_to_text(config.mode));
+ ((OverlayWindow*)thread->window)->direction->set_text(OverlayConfig::direction_to_text(config.direction));
+ ((OverlayWindow*)thread->window)->output->set_text(OverlayConfig::output_to_text(config.output_layer));
+ thread->window->unlock_window();
+ }
+}
+
+
+
+
+