From 5dd11b748e5409f8376376f51d0a18d111cbcfdc Mon Sep 17 00:00:00 2001 From: Good Guy Date: Mon, 10 Jun 2024 18:46:59 -0600 Subject: [PATCH] Credit SGE conversion of Adam-s plugins ChromakeyAvid/Color Swatch + updated ContextManual.pl for index.html variations --- cinelerra-5.1/doc/ContextManual.pl | 53 +- cinelerra-5.1/expanders.es | 2 + cinelerra-5.1/expanders.fr | 2 + cinelerra-5.1/expanders.txt | 2 + cinelerra-5.1/info/plugins.txt | 7 +- cinelerra-5.1/plugin_defs | 2 + cinelerra-5.1/plugins/Makefile | 2 + cinelerra-5.1/plugins/chromakeyavid/Makefile | 13 + .../plugins/chromakeyavid/chromakey.C | 1738 +++++++++++++++++ .../plugins/chromakeyavid/chromakey.h | 385 ++++ .../plugins/chromakeyavid/chromakeyavid.sl | 230 +++ cinelerra-5.1/plugins/swatch/Makefile | 8 + cinelerra-5.1/plugins/swatch/swatch.C | 760 +++++++ cinelerra-5.1/plugins/swatch/swatch.h | 186 ++ 14 files changed, 3385 insertions(+), 5 deletions(-) create mode 100644 cinelerra-5.1/plugins/chromakeyavid/Makefile create mode 100644 cinelerra-5.1/plugins/chromakeyavid/chromakey.C create mode 100644 cinelerra-5.1/plugins/chromakeyavid/chromakey.h create mode 100644 cinelerra-5.1/plugins/chromakeyavid/chromakeyavid.sl create mode 100644 cinelerra-5.1/plugins/swatch/Makefile create mode 100644 cinelerra-5.1/plugins/swatch/swatch.C create mode 100644 cinelerra-5.1/plugins/swatch/swatch.h diff --git a/cinelerra-5.1/doc/ContextManual.pl b/cinelerra-5.1/doc/ContextManual.pl index f3e25cdb..45d00dc8 100755 --- a/cinelerra-5.1/doc/ContextManual.pl +++ b/cinelerra-5.1/doc/ContextManual.pl @@ -17,7 +17,7 @@ # Several important definitions # ContextManual.pl script API version. Must not be changed ! -$cin_cm_api = 1; +$cin_cm_api = 2; # Web browser executable, can be redefined on user's demand $cin_browser = $ENV{'CIN_BROWSER'}; @@ -34,8 +34,9 @@ $cin_browser = 'firefox' if $cin_browser eq ''; # The node with the manual contents $contents_node = 'Contents.html'; -# The node with the manual index -$index_node = 'Index.html'; +# The node with the actual index if needed will be found after parsing contents +$index_node = ''; +$index = ''; # Several special plugin names necessary to rewrite %rewrite = ( @@ -80,7 +81,6 @@ $cin_dat = '.' if $cin_dat eq ''; # Cinelerra HTML manual must reside here $cin_man = "$cin_dat/doc/CinelerraGG_Manual"; $contents = $cin_man.'/'.$contents_node; -$index = $cin_man.'/'.$index_node; #print "ContextManual: using contents $contents\n"; # Cinelerra user's config directory @@ -134,6 +134,21 @@ if ($help_key eq 'TOC') # Show index on this special request if ($help_key eq 'IDX') { + # Index node will be needed now, find it + if ($index_node eq '') + { + $node = ''; + open CONTENTS, $contents or die "Cannot open $contents: $!"; + while () + { + chomp; + last if ($node) = /^\s*HREF=\"(.+?\.html)\">\s*Index\s*<\/A>(?:<\/B>)?$/; + } + close CONTENTS; + $index_node = $node if $node ne ''; + $index_node = 'Index.html' if $index_node eq ''; + $index = $cin_man.'/'.$index_node; + } system "$cin_browser \"file://$index\" &"; exit 0; } @@ -158,6 +173,21 @@ if ($help_key eq 'TOC') # Show index on this special request if ($help_key eq 'IDX') { + # Index node will be needed now, find it + if ($index_node eq '') + { + $node = ''; + open CONTENTS, $contents or die "Cannot open $contents: $!"; + while () + { + chomp; + last if ($node) = /^\s*HREF=\"(.+?\.html)\">\s*Index\s*<\/A>(?:<\/B>)?$/; + } + close CONTENTS; + $index_node = $node if $node ne ''; + $index_node = 'Index.html' if $index_node eq ''; + $index = $cin_man.'/'.$index_node; + } system "$cin_browser \"file://$index\" &"; exit 0; } @@ -190,6 +220,21 @@ if ($node eq '') } } +# Index node will be needed now, find it +if ($node eq '' && $index_node eq '') +{ + seek CONTENTS, 0, 0; + while () + { + chomp; + last if ($node) = /^\s*HREF=\"(.+?\.html)\">\s*Index\s*<\/A>(?:<\/B>)?$/; + } + $index_node = $node if $node ne ''; + $index_node = 'Index.html' if $index_node eq ''; + $index = $cin_man.'/'.$index_node; + $node = ''; +} + # If not found, search index for the exact key if ($node eq '') { diff --git a/cinelerra-5.1/expanders.es b/cinelerra-5.1/expanders.es index 99c58389..5e5ada63 100644 --- a/cinelerra-5.1/expanders.es +++ b/cinelerra-5.1/expanders.es @@ -123,6 +123,8 @@ Video Effects Blue Banana Chroma key Chroma key (HSV) + Chroma key (Avid) + Color Swatch CriKey Difference key F_chromakey diff --git a/cinelerra-5.1/expanders.fr b/cinelerra-5.1/expanders.fr index 25ed8883..fa092c21 100644 --- a/cinelerra-5.1/expanders.fr +++ b/cinelerra-5.1/expanders.fr @@ -123,6 +123,8 @@ Video Effects Blue Banana Chroma key Chroma key (HSV) + Chroma key (Avid) + Color Swatch CriKey Difference key F_chromakey diff --git a/cinelerra-5.1/expanders.txt b/cinelerra-5.1/expanders.txt index 0e3ae841..898ade19 100644 --- a/cinelerra-5.1/expanders.txt +++ b/cinelerra-5.1/expanders.txt @@ -124,6 +124,8 @@ Video Effects Blue Banana Chroma key Chroma key (HSV) + Chroma key (Avid) + Color Swatch CriKey Difference key F_chromakey diff --git a/cinelerra-5.1/info/plugins.txt b/cinelerra-5.1/info/plugins.txt index daf71caf..145924a0 100644 --- a/cinelerra-5.1/info/plugins.txt +++ b/cinelerra-5.1/info/plugins.txt @@ -42,12 +42,17 @@ Chroma key: Erases pixels which match chosen color; channel and transparency if there is alpha. Chroma key (HSV): Replaces a color with another color or transparency using HSV variables. +Chroma key (Avid): ChromkayHSV version based on the Avid + SpectraMatte algorithm. First attach the Color + Swatch plugin to help visualize the effect. Color 3 Way: Modify color of Shadows, Midtones, and Highlights as you specify. Color Balance: Modify RGB colors or white balance to compensate for errors in video such as low lighting. ColorSpace: Tweak colors from one color space/range to another - space=BT601/BT709/BT2020 range=MPEG/JPEG + space=BT601/BT709/BT2020 range=MPEG/JPEG. +Color Swatch: Draws the color space for saturation or brightness. + Provides an input to assist in ChromaKey adjustments. CriKey: Regionally based chroma key with interpolation; useful when you only want a specific zone defined. . diff --git a/cinelerra-5.1/plugin_defs b/cinelerra-5.1/plugin_defs index cb3c700a..ded3fd70 100644 --- a/cinelerra-5.1/plugin_defs +++ b/cinelerra-5.1/plugin_defs @@ -34,6 +34,7 @@ video := \ C41 \ chromakey \ chromakeyhsv \ + chromakeyavid \ color3way \ colorbalance \ colorspace \ @@ -112,6 +113,7 @@ video := \ svg \ swapchannels \ swapframes \ + swatch \ threshold \ timeavg \ timeblur \ diff --git a/cinelerra-5.1/plugins/Makefile b/cinelerra-5.1/plugins/Makefile index 22aaf433..89f16466 100644 --- a/cinelerra-5.1/plugins/Makefile +++ b/cinelerra-5.1/plugins/Makefile @@ -40,6 +40,7 @@ DIRS = $(OPENCV_OBJS) \ chorus \ chromakey \ chromakeyhsv \ + chromakeyavid \ color3way \ colorbalance \ colorspace \ @@ -139,6 +140,7 @@ DIRS = $(OPENCV_OBJS) \ svg \ swapchannels \ swapframes \ + swatch \ synthesizer \ threshold \ timeavg \ diff --git a/cinelerra-5.1/plugins/chromakeyavid/Makefile b/cinelerra-5.1/plugins/chromakeyavid/Makefile new file mode 100644 index 00000000..934f5fda --- /dev/null +++ b/cinelerra-5.1/plugins/chromakeyavid/Makefile @@ -0,0 +1,13 @@ +include ../../plugin_defs + +PLUGIN = chromakeyavid +OUTPUT_BINS := $(OBJDIR)/chromakeyavid_sl.o +OBJS := $(OBJDIR)/chromakey.o + +include ../../plugin_config + +$(OBJDIR)/chromakey.o: chromakey.C +$(OUTPUT_BINS): chromakeyavid.sl + cd $(OBJDIR) && \ + ../../../guicast/$(OBJDIR)/bootstrap -s chromakeyavid_sl.o ../chromakeyavid.sl + diff --git a/cinelerra-5.1/plugins/chromakeyavid/chromakey.C b/cinelerra-5.1/plugins/chromakeyavid/chromakey.C new file mode 100644 index 00000000..a1324bd7 --- /dev/null +++ b/cinelerra-5.1/plugins/chromakeyavid/chromakey.C @@ -0,0 +1,1738 @@ +/* + * CINELERRA + * Copyright (C) 2012-2024 Adam Williams + * + * 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 + * + */ + +// original chromakey HSV author: +// https://www.mail-archive.com/cinelerra@skolelinux.no/msg00379.html +// http://jcornet.free.fr/linux/chromakey.html + +// He was trying to replicate the Avid SpectraMatte: +// https://resources.avid.com/SupportFiles/attach/Symphony_Effects_and_CC_Guide_v5.5.pdf +// page 691 +// but the problem seems to be harder than he thought + +// A rewrite got a slightly different spill light processing +// but still not very useful: +// https://github.com/vanakala/cinelerra-cve/blob/master/plugins/chromakeyhsv/chromakey.C + +// The current implementation is most of the Spectramatte algorithm. +// The only way to test it is to use the color swatch plugin. +// Fix brightness to test the saturation wedge. +// Fix saturation to test the brightness wedge. + +// There are boundary artifacts caused by the color swatch showing a slice +// of a cube. +// If out slope > 0, constant 0% saturation or 0% brightness is all wedge. +// If out slope == 0, constant 0% saturation or constant 0% brightness has no wedge +// The general idea is if it's acting weird, switch between constant brightness +// & constant saturation. + +// TODO: integrated color swatch & alpha blur, but it takes a lot of space in the GUI. + +#include "bcdisplayinfo.h" +#include "bcsignals.h" +#include "chromakey.h" +#include "clip.h" +#include "bchash.h" +#include "filexml.h" +#include "guicast.h" +#include "keyframe.h" +#include "language.h" +#include "loadbalance.h" +#include "playback3d.h" +#include "bccolors.h" +#include "pluginvclient.h" +#include "vframe.h" + +#include +#include +#include + +#define WINDOW_W xS(550) +#define WINDOW_H yS(600) +#define SLIDER_W xS(250) +#define MAX_SLOPE 100.0 +#define MAX_VALUE 100.0 + +ChromaKeyConfig::ChromaKeyConfig () +{ + red = 0.0; + green = 1.0; + blue = 0.0; + + min_brightness = 50.0; + max_brightness = 100.0; + tolerance = 15.0; + saturation_start = 0.0; + saturation_line = 50.0; + + in_slope = 2; + out_slope = 2; + alpha_offset = 0; + + spill_saturation = 0.0; + spill_angle = 0.0; + desaturate_only = 0; + + show_mask = 0; + + undo_red = red; + undo_green = green; + undo_blue = blue; + undo_min_brightness = min_brightness; + undo_max_brightness = max_brightness; + undo_tolerance = tolerance; + undo_saturation_start = saturation_start; + undo_saturation_line = saturation_line; + undo_in_slope = in_slope; + undo_out_slope = out_slope; + undo_alpha_offset = alpha_offset; + undo_spill_saturation = spill_saturation; + undo_spill_angle = spill_angle; + undo_desaturate_only = desaturate_only; + undo_show_mask = show_mask; + + store_red = red; + store_green = green; + store_blue = blue; + store_min_brightness = min_brightness; + store_max_brightness = max_brightness; + store_tolerance = tolerance; + store_saturation_start = saturation_start; + store_saturation_line = saturation_line; + store_in_slope = in_slope; + store_out_slope = out_slope; + store_alpha_offset = alpha_offset; + store_spill_saturation = spill_saturation; + store_spill_angle = spill_angle; + store_desaturate_only = desaturate_only; + store_show_mask = show_mask; +} + +void ChromaKeyConfig::reset(int clear) +{ + switch(clear) + { + case RESET_RGB: + red = 0.0; + green = 1.0; + blue = 0.0; + break; + case RESET_MIN_BRIGHTNESS: + min_brightness = 50.0; + break; + case RESET_MAX_BRIGHTNESS: + max_brightness = 100.0; + break; + case RESET_TOLERANCE: + tolerance = 15.0; + break; + case RESET_SATURATION_START: + saturation_start = 0.0; + break; + case RESET_SATURATION_LINE: + saturation_line = 50.0; + break; + case RESET_IN_SLOPE: + in_slope = 2; + break; + case RESET_OUT_SLOPE: + out_slope = 2; + break; + case RESET_ALPHA_OFFSET: + alpha_offset = 0; + break; + case RESET_SPILL_SATURATION: + spill_saturation = 0.0; + break; + case RESET_SPILL_ANGLE: + spill_angle = 0.0; + break; + case RESET_ALL: + case RESET_DEFAULT_SETTINGS: + default: + undo_red = red; + undo_green = green; + undo_blue = blue; + undo_min_brightness = min_brightness; + undo_max_brightness = max_brightness; + undo_tolerance = tolerance; + undo_saturation_start = saturation_start; + undo_saturation_line = saturation_line; + undo_in_slope = in_slope; + undo_out_slope = out_slope; + undo_alpha_offset = alpha_offset; + undo_spill_saturation = spill_saturation; + undo_spill_angle = spill_angle; + undo_desaturate_only = desaturate_only; + undo_show_mask = show_mask; + + red = 0.0; + green = 1.0; + blue = 0.0; + + min_brightness = 50.0; + max_brightness = 100.0; + tolerance = 15.0; + saturation_start = 0.0; + saturation_line = 50.0; + + in_slope = 2; + out_slope = 2; + alpha_offset = 0; + + spill_saturation = 0.0; + spill_angle = 0.0; + desaturate_only = 0; + + show_mask = 0; + break; + } +} + + +void +ChromaKeyConfig::copy_from (ChromaKeyConfig & src) +{ + red = src.red; + green = src.green; + blue = src.blue; + spill_saturation = src.spill_saturation; + spill_angle = src.spill_angle; + min_brightness = src.min_brightness; + max_brightness = src.max_brightness; + saturation_start = src.saturation_start; + saturation_line = src.saturation_line; + tolerance = src.tolerance; + in_slope = src.in_slope; + out_slope = src.out_slope; + alpha_offset = src.alpha_offset; + show_mask = src.show_mask; + desaturate_only = src.desaturate_only; + + undo_red = src.red; + undo_green = src.green; + undo_blue = src.blue; + undo_min_brightness = src.min_brightness; + undo_max_brightness = src.max_brightness; + undo_tolerance = src.tolerance; + undo_saturation_start = src.saturation_start; + undo_saturation_line = src.saturation_line; + undo_in_slope = src.in_slope; + undo_out_slope = src.out_slope; + undo_alpha_offset = src.alpha_offset; + undo_spill_saturation = src.spill_saturation; + undo_spill_angle = src.spill_angle; + undo_desaturate_only = src.desaturate_only; + undo_show_mask = src.show_mask; +} + +int +ChromaKeyConfig::equivalent (ChromaKeyConfig & src) +{ + return (EQUIV (red, src.red) && + EQUIV (green, src.green) && + EQUIV (blue, src.blue) && + EQUIV (spill_saturation, src.spill_saturation) && + EQUIV (spill_angle, src.spill_angle) && + EQUIV (min_brightness, src.min_brightness) && + EQUIV (max_brightness, src.max_brightness) && + EQUIV (saturation_start, src.saturation_start) && + EQUIV (saturation_line, src.saturation_line) && + EQUIV (tolerance, src.tolerance) && + EQUIV (in_slope, src.in_slope) && + EQUIV (out_slope, src.out_slope) && + EQUIV (alpha_offset, src.alpha_offset) && + show_mask == src.show_mask && + desaturate_only == src.desaturate_only); +} + +void +ChromaKeyConfig::interpolate (ChromaKeyConfig & prev, + ChromaKeyConfig & next, + int64_t prev_frame, + int64_t next_frame, + int64_t 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->red = prev.red * prev_scale + next.red * next_scale; + this->green = prev.green * prev_scale + next.green * next_scale; + this->blue = prev.blue * prev_scale + next.blue * next_scale; + this->spill_saturation = + prev.spill_saturation * prev_scale + next.spill_saturation * next_scale; + this->spill_angle = + prev.spill_angle * prev_scale + next.tolerance * next_scale; + this->min_brightness = + prev.min_brightness * prev_scale + next.min_brightness * next_scale; + this->max_brightness = + prev.max_brightness * prev_scale + next.max_brightness * next_scale; + this->saturation_start = + prev.saturation_start * prev_scale + next.saturation_start * next_scale; + this->saturation_line = + prev.saturation_line * prev_scale + next.saturation_line * next_scale; + this->tolerance = prev.tolerance * prev_scale + next.tolerance * next_scale; + this->in_slope = prev.in_slope * prev_scale + next.in_slope * next_scale; + this->out_slope = prev.out_slope * prev_scale + next.out_slope * next_scale; + this->alpha_offset = + prev.alpha_offset * prev_scale + next.alpha_offset * next_scale; + this->show_mask = prev.show_mask; + this->desaturate_only = prev.desaturate_only; + + this->undo_red = this->red; + this->undo_green = this->green; + this->undo_blue = this->blue; + this->undo_min_brightness = this->min_brightness; + this->undo_max_brightness = this->max_brightness; + this->undo_tolerance = this->tolerance; + this->undo_saturation_start = this->saturation_start; + this->undo_saturation_line = this->saturation_line; + this->undo_in_slope = this->in_slope; + this->undo_out_slope = this->out_slope; + this->undo_alpha_offset = this->alpha_offset; + this->undo_spill_saturation = this->spill_saturation; + this->undo_spill_angle = this->spill_angle; + this->undo_desaturate_only = this->desaturate_only; + this->undo_show_mask = this->show_mask; +} + +int +ChromaKeyConfig::get_color () +{ + int red = (int) (CLIP (this->red, 0, 1) * 0xff); + int green = (int) (CLIP (this->green, 0, 1) * 0xff); + int blue = (int) (CLIP (this->blue, 0, 1) * 0xff); + return (red << 16) | (green << 8) | blue; +} + + + +ChromaKeyWindow::ChromaKeyWindow (ChromaKeyAvid * plugin) + : PluginClientWindow(plugin, + WINDOW_W, + WINDOW_H, + WINDOW_W, + WINDOW_H, + 0) +{ + this->plugin = plugin; + color_thread = 0; +} + +ChromaKeyWindow::~ChromaKeyWindow () +{ + delete color_thread; +} + +void ChromaKeyWindow::create_objects () +{ + int xs10 = xS(10), xs20 = xS(20), xs100 = xS(100), wslid = SLIDER_W; + int ys10 = yS(10), ys20 = yS(20), ys30 = yS(30), ys40 = yS(40), ys50 = yS(50); + int y = ys10, x2 = xS(160), x3 = xS(260), xs5 = xS(5); + int x = xs10; + int clr_x = get_w()-x - xS(22); // note: clrBtn_w = 22 + + BC_Title *title; + BC_Bar *bar; + BC_TitleBar *title_bar; + +// Color section + add_subwindow (title_bar = + new BC_TitleBar(x, y, get_w()-2*x, xs20, xs10, _("Color"))); + y += ys20; + + add_subwindow (color = new ChromaKeyColor (plugin, this, x, y)); + // Info for Sample rectangle: x_slider w_slider w_sample + // \ | / y, w, h + add_subwindow (sample = new BC_SubWindow (x3 + wslid - xs100, y, xs100, ys50)); + y += ys30; + add_subwindow (use_colorpicker = new ChromaKeyUseColorPicker (plugin, this, x, y)); + y += ys30; + add_subwindow (show_mask = + new ChromaKeyToggle (plugin, x, y, &plugin->config.show_mask, + _("Show Mask"))); + +// Key parameters section + y += ys30; + add_subwindow (title_bar = + new BC_TitleBar (x, y, get_w()-2*x, xs20, xs10, + _("Key parameters"))); + y += ys20; + add_subwindow (title = new BC_Title (x, y, _("Hue Tolerance:"))); + tolerance_text = new ChromaKeyText (plugin, this, 0, x+x2, y, 0, MAX_VALUE, + &plugin->config.tolerance); + tolerance_text->create_objects(); + tolerance = new ChromaKeySlider (plugin, tolerance_text, x3, y, wslid, + 0, MAX_VALUE, &plugin->config.tolerance); + add_subwindow (tolerance); + tolerance_text->slider = tolerance; + clr_x = x3 + tolerance->get_w() + x; + add_subwindow (tolerance_clr = + new ChromaKeyClr (plugin, this, clr_x, y, RESET_TOLERANCE)); + y += ys30; + + add_subwindow (title = new BC_Title (x, y, _("Min. Brightness:"))); + min_brightness_text = new ChromaKeyText (plugin, this, 0, x+x2, y, + 0, MAX_VALUE, + &plugin->config.min_brightness); + min_brightness_text->create_objects(); + min_brightness = new ChromaKeySlider (plugin, min_brightness_text, + x3, y, wslid, 0, MAX_VALUE, + &plugin->config.min_brightness); + add_subwindow(min_brightness); + min_brightness_text->slider = min_brightness; + add_subwindow (min_brightness_clr = + new ChromaKeyClr (plugin, this, clr_x, y, + RESET_MIN_BRIGHTNESS)); + y += ys30; + + add_subwindow (title = new BC_Title (x, y, _("Max. Brightness:"))); + max_brightness_text = new ChromaKeyText (plugin, this, 0, x+x2, y, + 0, MAX_VALUE, + &plugin->config.max_brightness); + max_brightness_text->create_objects(); + max_brightness = new ChromaKeySlider (plugin, max_brightness_text, + x3, y, wslid, 0, MAX_VALUE, + &plugin->config.max_brightness); + add_subwindow(max_brightness); + max_brightness_text->slider = max_brightness; + add_subwindow (max_brightness_clr = + new ChromaKeyClr (plugin, this, clr_x, y, + RESET_MAX_BRIGHTNESS)); + y += ys30; + + add_subwindow (title = new BC_Title (x, y, _("Saturation Start:"))); + saturation_start_text = new ChromaKeyText (plugin, this, 0, x+x2, y, + 0, MAX_VALUE, + &plugin->config.saturation_start); + saturation_start_text->create_objects(); + saturation_start = new ChromaKeySlider (plugin, saturation_start_text, + x3, y, wslid, 0, MAX_VALUE, + &plugin->config.saturation_start); + add_subwindow(saturation_start); + saturation_start_text->slider = saturation_start; + add_subwindow (saturation_start_clr = + new ChromaKeyClr (plugin, this, clr_x, y, + RESET_SATURATION_START)); + y += ys30; + + add_subwindow (title = new BC_Title (x, y, _("Saturation Line:"))); + saturation_line_text = new ChromaKeyText (plugin, this, 0, x+x2, y, + 0, MAX_VALUE, + &plugin->config.saturation_line); + saturation_line_text->create_objects(); + saturation_line = new ChromaKeySlider (plugin, saturation_line_text, + x3, y, wslid, 0, MAX_VALUE, + &plugin->config.saturation_line); + add_subwindow(saturation_line); + saturation_line_text->slider = saturation_line; + add_subwindow (saturation_line_clr = + new ChromaKeyClr (plugin, this, clr_x, y, + RESET_SATURATION_LINE)); + y += ys40; + +// Mask tweaking section + add_subwindow (title_bar = + new BC_TitleBar (x, y, get_w()-2*x, xs20, xs10, + _("Mask tweaking"))); + y += ys20; + add_subwindow (title = new BC_Title (x, y, _("In Slope:"))); + in_slope_text = new ChromaKeyText (plugin, this, 0, x+x2, y, 0, MAX_SLOPE, + &plugin->config.in_slope); + in_slope_text->create_objects(); + in_slope = new ChromaKeySlider (plugin, in_slope_text, x3, y, wslid, + 0, MAX_SLOPE, &plugin->config.in_slope); + add_subwindow(in_slope); + in_slope_text->slider = in_slope; + add_subwindow (in_slope_clr = + new ChromaKeyClr (plugin, this, clr_x, y, RESET_IN_SLOPE)); + y += ys30; + + add_subwindow (title = new BC_Title (x, y, _("Out Slope:"))); + out_slope_text = new ChromaKeyText (plugin, this, 0, x+x2, y, 0, MAX_SLOPE, + &plugin->config.out_slope); + out_slope_text->create_objects(); + out_slope = new ChromaKeySlider (plugin, out_slope_text, x3, y, wslid, + 0, MAX_SLOPE, &plugin->config.out_slope); + add_subwindow(out_slope); + out_slope_text->slider = out_slope; + add_subwindow (out_slope_clr = + new ChromaKeyClr (plugin, this, clr_x, y, RESET_OUT_SLOPE)); + y += ys30; + + add_subwindow (title = new BC_Title (x, y, _("Alpha Offset:"))); + alpha_offset_text = new ChromaKeyText (plugin, this, 0, x+x2, y, + -MAX_VALUE, MAX_VALUE, + &plugin->config.alpha_offset); + alpha_offset_text->create_objects(); + alpha_offset = new ChromaKeySlider (plugin, alpha_offset_text, x3, y, wslid, + -MAX_VALUE, MAX_VALUE, + &plugin->config.alpha_offset); + add_subwindow(alpha_offset); + alpha_offset_text->slider = alpha_offset; + add_subwindow (alpha_offset_clr = + new ChromaKeyClr (plugin, this, clr_x, y, RESET_ALPHA_OFFSET)); + y += ys40; + +// Spill light control section + add_subwindow (title_bar = + new BC_TitleBar (x, y, get_w()-2*x, xs20, xs10, + _("Spill light control"))); + y += ys20; + add_subwindow (title = new BC_Title (x, y, _("Spill Saturation:"))); + spill_saturation_text = new ChromaKeyText (plugin, this, 0, x+x2, y, + 0, MAX_VALUE, + &plugin->config.spill_saturation); + spill_saturation_text->create_objects(); + spill_saturation = new ChromaKeySlider (plugin, spill_saturation_text, + x3, y, wslid, 0, MAX_VALUE, + &plugin->config.spill_saturation); + add_subwindow(spill_saturation); + spill_saturation_text->slider = spill_saturation; + add_subwindow (spill_saturation_clr = + new ChromaKeyClr (plugin, this, clr_x, y, + RESET_SPILL_SATURATION)); + y += ys30; + + add_subwindow (title = new BC_Title (x, y, _("Spill Angle:"))); + spill_angle_text = new ChromaKeyText (plugin, this, 0, x+x2, y, 0, MAX_VALUE, + &plugin->config.spill_angle); + spill_angle_text->create_objects(); + spill_angle = new ChromaKeySlider (plugin, spill_angle_text, x3, y, wslid, + 0, MAX_VALUE, &plugin->config.spill_angle); + add_subwindow(spill_angle); + spill_angle_text->slider = spill_angle; + add_subwindow (spill_angle_clr = + new ChromaKeyClr (plugin, this, clr_x, y, RESET_SPILL_ANGLE)); + y += ys30; + + add_subwindow(desaturate_only = + new ChromaKeyToggle(plugin, x, y, + &plugin->config.desaturate_only, + _("Desaturate Only"))); + y += ys40; + +// Global buttons section + add_subwindow (bar = new BC_Bar(x, y, get_w()-2*x)); + y += ys20; + add_subwindow (store = new ChromaKeyStore (plugin, this, x, y)); + x += store->get_w() + xs5; + add_subwindow (recall = new ChromaKeyRecall (plugin, this, x, y)); + x += recall->get_w() + xs5; + add_subwindow (exchange = new ChromaKeyExchange (plugin, this, x, y)); + x += exchange->get_w() + xs5; + add_subwindow (undo = new ChromaKeyUndo (plugin, this, x, y)); + x += undo->get_w() + xs5; + add_subwindow (reset = new ChromaKeyReset (plugin, this, x, y)); + + color_thread = new ChromaKeyColorThread (plugin, this); + + update_sample (); + show_window (); +} + +void +ChromaKeyWindow::update_sample () +{ + sample->set_color (plugin->config.get_color ()); + sample->draw_box (0, 0, sample->get_w (), sample->get_h ()); + sample->set_color (BLACK); + sample->draw_rectangle (0, 0, sample->get_w (), sample->get_h ()); + sample->flash (); +} + +void ChromaKeyWindow::done_event(int result) +{ + color_thread->close_window(); +} + + +ChromaKeyColor::ChromaKeyColor (ChromaKeyAvid * plugin, + ChromaKeyWindow * gui, int x, int y) + : BC_GenericButton (x, y, _("Color...")) +{ + this->plugin = plugin; + this->gui = gui; +} + +int +ChromaKeyColor::handle_event () +{ + gui->color_thread->start_window (plugin->config.get_color (), 0xff); + return 1; +} + + + +ChromaKeyText::ChromaKeyText(ChromaKeyAvid *plugin, ChromaKeyWindow *gui, + ChromaKeySlider *slider, int x, int y, + float min, float max, float *output) + : BC_TumbleTextBox(gui, *output, min, max, x, y, xS(60), 2) +{ + this->plugin = plugin; + this->gui = gui; + this->slider = slider; + this->min = min; + this->max = max; + this->output = output; + set_increment(0.01); +} + +ChromaKeyText::~ChromaKeyText() +{ +} + +int ChromaKeyText::handle_event() +{ + *output = atof(get_text()); + if(*output > max) *output = max; + if(*output < min) *output = min; + slider->update(*output); + plugin->send_configure_change(); + return 1; +} + +ChromaKeySlider::ChromaKeySlider(ChromaKeyAvid *plugin, ChromaKeyText *text, + int x, int y, int w, + float min, float max, float *output) + : BC_FSlider(x, y, 0, w, w, min, max, *output) +{ + this->plugin = plugin; + this->text = text; + this->output = output; + set_precision(0.01); + enable_show_value(0); // Hide caption +} + +ChromaKeySlider::~ChromaKeySlider() +{ +} + +int ChromaKeySlider::handle_event() +{ + *output = get_value(); + text->update(*output); + plugin->send_configure_change(); + return 1; +} + +ChromaKeyClr::ChromaKeyClr(ChromaKeyAvid *plugin, ChromaKeyWindow *gui, + int x, int y, int clear) + : BC_Button(x, y, plugin->get_theme()->get_image_set("reset_button")) +{ + this->plugin = plugin; + this->gui = gui; + this->clear = clear; +} + +ChromaKeyClr::~ChromaKeyClr() +{ +} + +int ChromaKeyClr::handle_event() +{ + plugin->config.reset(clear); + gui->update_gui(clear); + plugin->send_configure_change(); + return 1; +} + +ChromaKeyToggle::ChromaKeyToggle (ChromaKeyAvid * plugin, + int x, + int y, + bool *output, + const char *caption) + : BC_CheckBox (x, + y, + *output ? 1 : 0, + caption) +{ + this->plugin = plugin; + this->output = output; +} + +int +ChromaKeyToggle::handle_event () +{ + *output = get_value() ? true : false; + plugin->send_configure_change (); + return 1; +} + + +ChromaKeyReset::ChromaKeyReset (ChromaKeyAvid *plugin, ChromaKeyWindow *gui, + int x, int y) + : BC_GenericButton(x, y, _("Reset")) +{ + this->plugin = plugin; + this->gui = gui; +} + +int ChromaKeyReset::handle_event () +{ + plugin->config.reset(RESET_DEFAULT_SETTINGS); + gui->update_gui(RESET_DEFAULT_SETTINGS); + plugin->send_configure_change(); + return 1; +} + +ChromaKeyStore::ChromaKeyStore (ChromaKeyAvid *plugin, ChromaKeyWindow *gui, + int x, int y) + : BC_GenericButton(x, y, _("Store")) +{ + this->plugin = plugin; + this->gui = gui; +} + +int ChromaKeyStore::handle_event () +{ + plugin->config.store_red = plugin->config.red; + plugin->config.store_green = plugin->config.green; + plugin->config.store_blue = plugin->config.blue; + plugin->config.store_min_brightness = plugin->config.min_brightness; + plugin->config.store_max_brightness = plugin->config.max_brightness; + plugin->config.store_tolerance = plugin->config.tolerance; + plugin->config.store_saturation_start = plugin->config.saturation_start; + plugin->config.store_saturation_line = plugin->config.saturation_line; + plugin->config.store_in_slope = plugin->config.in_slope; + plugin->config.store_out_slope = plugin->config.out_slope; + plugin->config.store_alpha_offset = plugin->config.alpha_offset; + plugin->config.store_spill_saturation = plugin->config.spill_saturation; + plugin->config.store_spill_angle = plugin->config.spill_angle; + plugin->config.store_desaturate_only = plugin->config.desaturate_only; + plugin->config.store_show_mask = plugin->config.show_mask; + return 1; +} + +ChromaKeyRecall::ChromaKeyRecall (ChromaKeyAvid *plugin, ChromaKeyWindow *gui, + int x, int y) + : BC_GenericButton(x, y, _("Recall")) +{ + this->plugin = plugin; + this->gui = gui; +} + +int ChromaKeyRecall::handle_event () +{ + plugin->config.undo_red = plugin->config.red; + plugin->config.undo_green = plugin->config.green; + plugin->config.undo_blue = plugin->config.blue; + plugin->config.undo_min_brightness = plugin->config.min_brightness; + plugin->config.undo_max_brightness = plugin->config.max_brightness; + plugin->config.undo_tolerance = plugin->config.tolerance; + plugin->config.undo_saturation_start = plugin->config.saturation_start; + plugin->config.undo_saturation_line = plugin->config.saturation_line; + plugin->config.undo_in_slope = plugin->config.in_slope; + plugin->config.undo_out_slope = plugin->config.out_slope; + plugin->config.undo_alpha_offset = plugin->config.alpha_offset; + plugin->config.undo_spill_saturation = plugin->config.spill_saturation; + plugin->config.undo_spill_angle = plugin->config.spill_angle; + plugin->config.undo_desaturate_only = plugin->config.desaturate_only; + plugin->config.undo_show_mask = plugin->config.show_mask; + + plugin->config.red = plugin->config.store_red; + plugin->config.green = plugin->config.store_green; + plugin->config.blue = plugin->config.store_blue; + plugin->config.min_brightness = plugin->config.store_min_brightness; + plugin->config.max_brightness = plugin->config.store_max_brightness; + plugin->config.tolerance = plugin->config.store_tolerance; + plugin->config.saturation_start = plugin->config.store_saturation_start; + plugin->config.saturation_line = plugin->config.store_saturation_line; + plugin->config.in_slope = plugin->config.store_in_slope; + plugin->config.out_slope = plugin->config.store_out_slope; + plugin->config.alpha_offset = plugin->config.store_alpha_offset; + plugin->config.spill_saturation = plugin->config.store_spill_saturation; + plugin->config.spill_angle = plugin->config.store_spill_angle; + plugin->config.desaturate_only = plugin->config.store_desaturate_only; + plugin->config.show_mask = plugin->config.store_show_mask; + + gui->update_gui(RESET_DEFAULT_SETTINGS); + plugin->send_configure_change(); + return 1; +} + +ChromaKeyExchange::ChromaKeyExchange (ChromaKeyAvid *plugin, + ChromaKeyWindow *gui, int x, int y) + : BC_GenericButton(x, y, _("Exchange")) +{ + this->plugin = plugin; + this->gui = gui; +} + +#define CHROMAKEY_SWAP(what,typ,elem) \ +tmp_##typ = plugin->config.what##_##elem; \ +plugin->config.what##_##elem = plugin->config.elem; \ +plugin->config.elem = tmp_##typ; + +int ChromaKeyExchange::handle_event () +{ + float tmp_flt; + bool tmp_bool; + + plugin->config.undo_red = plugin->config.red; + plugin->config.undo_green = plugin->config.green; + plugin->config.undo_blue = plugin->config.blue; + plugin->config.undo_min_brightness = plugin->config.min_brightness; + plugin->config.undo_max_brightness = plugin->config.max_brightness; + plugin->config.undo_tolerance = plugin->config.tolerance; + plugin->config.undo_saturation_start = plugin->config.saturation_start; + plugin->config.undo_saturation_line = plugin->config.saturation_line; + plugin->config.undo_in_slope = plugin->config.in_slope; + plugin->config.undo_out_slope = plugin->config.out_slope; + plugin->config.undo_alpha_offset = plugin->config.alpha_offset; + plugin->config.undo_spill_saturation = plugin->config.spill_saturation; + plugin->config.undo_spill_angle = plugin->config.spill_angle; + plugin->config.undo_desaturate_only = plugin->config.desaturate_only; + plugin->config.undo_show_mask = plugin->config.show_mask; + + CHROMAKEY_SWAP (store, flt, red) + CHROMAKEY_SWAP (store, flt, green) + CHROMAKEY_SWAP (store, flt, blue) + CHROMAKEY_SWAP (store, flt, min_brightness) + CHROMAKEY_SWAP (store, flt, max_brightness) + CHROMAKEY_SWAP (store, flt, tolerance) + CHROMAKEY_SWAP (store, flt, saturation_start) + CHROMAKEY_SWAP (store, flt, saturation_line) + CHROMAKEY_SWAP (store, flt, in_slope) + CHROMAKEY_SWAP (store, flt, out_slope) + CHROMAKEY_SWAP (store, flt, alpha_offset) + CHROMAKEY_SWAP (store, flt, spill_saturation) + CHROMAKEY_SWAP (store, flt, spill_angle) + CHROMAKEY_SWAP (store, bool, desaturate_only) + CHROMAKEY_SWAP (store, bool, show_mask) + + gui->update_gui(RESET_DEFAULT_SETTINGS); + plugin->send_configure_change(); + return 1; +} + +ChromaKeyUndo::ChromaKeyUndo (ChromaKeyAvid *plugin, ChromaKeyWindow *gui, + int x, int y) + : BC_GenericButton(x, y, _("Undo")) +{ + this->plugin = plugin; + this->gui = gui; +} + +int ChromaKeyUndo::handle_event () +{ + //float tmp_flt; + //bool tmp_bool; + + //CHROMAKEY_SWAP (undo, flt, red) + //CHROMAKEY_SWAP (undo, flt, green) + //CHROMAKEY_SWAP (undo, flt, blue) + //CHROMAKEY_SWAP (undo, flt, min_brightness) + //CHROMAKEY_SWAP (undo, flt, max_brightness) + //CHROMAKEY_SWAP (undo, flt, tolerance) + //CHROMAKEY_SWAP (undo, flt, saturation_start) + //CHROMAKEY_SWAP (undo, flt, saturation_line) + //CHROMAKEY_SWAP (undo, flt, in_slope) + //CHROMAKEY_SWAP (undo, flt, out_slope) + //CHROMAKEY_SWAP (undo, flt, alpha_offset) + //CHROMAKEY_SWAP (undo, flt, spill_saturation) + //CHROMAKEY_SWAP (undo, flt, spill_angle) + //CHROMAKEY_SWAP (undo, bool, desaturate_only) + //CHROMAKEY_SWAP (undo, bool, show_mask) + + // better not to swap but restore undo values + + plugin->config.red = plugin->config.undo_red; + plugin->config.green = plugin->config.undo_green; + plugin->config.blue = plugin->config.undo_blue; + plugin->config.min_brightness = plugin->config.undo_min_brightness; + plugin->config.max_brightness = plugin->config.undo_max_brightness; + plugin->config.tolerance = plugin->config.undo_tolerance; + plugin->config.saturation_start = plugin->config.undo_saturation_start; + plugin->config.saturation_line = plugin->config.undo_saturation_line; + plugin->config.in_slope = plugin->config.undo_in_slope; + plugin->config.out_slope = plugin->config.undo_out_slope; + plugin->config.alpha_offset = plugin->config.undo_alpha_offset; + plugin->config.spill_saturation = plugin->config.undo_spill_saturation; + plugin->config.spill_angle = plugin->config.undo_spill_angle; + plugin->config.desaturate_only = plugin->config.undo_desaturate_only; + plugin->config.show_mask = plugin->config.undo_show_mask; + + gui->update_gui(RESET_DEFAULT_SETTINGS); + plugin->send_configure_change(); + return 1; +} + +ChromaKeyUseColorPicker::ChromaKeyUseColorPicker (ChromaKeyAvid * plugin, + ChromaKeyWindow * gui, + int x, int y) + : BC_GenericButton (x, y, _("Use color picker")) +{ + this->plugin = plugin; + this->gui = gui; +} + +int +ChromaKeyUseColorPicker::handle_event () +{ + plugin->config.red = plugin->get_red (); + plugin->config.green = plugin->get_green (); + plugin->config.blue = plugin->get_blue (); + + gui->update_sample (); + + plugin->send_configure_change (); + return 1; +} + +ChromaKeyColorThread::ChromaKeyColorThread (ChromaKeyAvid * plugin, + ChromaKeyWindow * gui) + : ColorPicker (0, _("Color")) +{ + this->plugin = plugin; + this->gui = gui; +} + +int +ChromaKeyColorThread::handle_new_color (int output, int alpha) +{ + plugin->config.red = (float) (output & 0xff0000) / 0xff0000; + plugin->config.green = (float) (output & 0xff00) / 0xff00; + plugin->config.blue = (float) (output & 0xff) / 0xff; + + get_gui()->unlock_window(); + gui->lock_window("ChromaKeyColorThread::handle_new_color"); + gui->update_sample(); + gui->unlock_window(); + get_gui()->lock_window("ChromaKeyColorThread::handle_new_color"); + + plugin->send_configure_change (); + return 1; +} + + + +ChromaKeyServer::ChromaKeyServer (ChromaKeyAvid * plugin) + : LoadServer (plugin->PluginClient::smp + 1, + plugin->PluginClient::smp + 1) +// DEBUG +// : LoadServer (1, 1) +{ + this->plugin = plugin; +} + +void +ChromaKeyServer::init_packages () +{ + for (int i = 0; i < get_total_packages (); i++) + { + ChromaKeyPackage *pkg = (ChromaKeyPackage *) get_package (i); + pkg->y1 = plugin->input->get_h () * i / get_total_packages (); + pkg->y2 = plugin->input->get_h () * (i + 1) / get_total_packages (); + } +} + +LoadClient * +ChromaKeyServer::new_client () +{ + return new ChromaKeyUnit (plugin, this); +} + +LoadPackage * +ChromaKeyServer::new_package () +{ + return new ChromaKeyPackage; +} + +ChromaKeyPackage::ChromaKeyPackage ():LoadPackage () +{ +} + +ChromaKeyUnit::ChromaKeyUnit (ChromaKeyAvid * plugin, ChromaKeyServer * server) + : LoadClient (server) +{ + this->plugin = plugin; +} + + + +#define ABS(a) (((a) < 0) ? -(a) : (a)) + + + +// Compute the same values in the opengl version +#define COMMON_VARIABLES \ + float red = plugin->config.red; \ + float green = plugin->config.green; \ + float blue = plugin->config.blue; \ + \ + float in_slope = plugin->config.in_slope / MAX_SLOPE; \ + float out_slope = plugin->config.out_slope / MAX_SLOPE; \ + \ +/* Convert RGB key to HSV key */ \ + float hue_key, saturation_key, value_key; \ + HSV::rgb_to_hsv(red, \ + green, \ + blue, \ + hue_key, \ + saturation_key, \ + value_key); \ + \ +/* hue range */ \ + float tolerance = plugin->config.tolerance * 180 / MAX_VALUE; \ + float tolerance_in = tolerance - in_slope * 180; \ + tolerance_in = MAX(tolerance_in, 0); \ + float tolerance_out = tolerance + out_slope * 180; \ + tolerance_out = MIN(tolerance_out, 180); \ + \ +/* distance of wedge point from center */ \ + float sat_distance = plugin->config.saturation_start / MAX_VALUE; \ +/* XY shift of input color to get wedge point */ \ + float sat_x = -cos(TO_RAD(hue_key)) * sat_distance; \ + float sat_y = -sin(TO_RAD(hue_key)) * sat_distance; \ +/* minimum saturation after wedge point */ \ + float min_s = plugin->config.saturation_line / MAX_VALUE; \ + float min_s_in = min_s + in_slope; \ + float min_s_out = min_s - out_slope; \ + \ + float min_v = plugin->config.min_brightness / MAX_VALUE; \ + float min_v_in = min_v + in_slope; \ + float min_v_out = min_v - out_slope; \ +/* ignore min_brightness 0 */ \ + if(plugin->config.min_brightness == 0) min_v_in = 0; \ + \ + float max_v = plugin->config.max_brightness / MAX_VALUE; \ + float max_v_in = max_v - in_slope; \ + float max_v_out = max_v + out_slope; \ +/* handle wedge boundaries crossing over */ \ + if(max_v_in < min_v_in) max_v_in = min_v_in = (max_v_in + min_v_in) / 2; \ +/* ignore max_brightness 100% */ \ + if(plugin->config.max_brightness == MAX_VALUE) max_v_in = 1.0; \ + \ +/* distance of spill wedge point from center */ \ + float spill_distance = sat_distance * (1.0 - plugin->config.spill_saturation / MAX_VALUE); \ +/* XY shift of input color to get spill wedge point */ \ + float spill_x = -cos(TO_RAD(hue_key)) * spill_distance; \ + float spill_y = -sin(TO_RAD(hue_key)) * spill_distance; \ +/* tolerance of the spill wedge */ \ + float spill_tolerance = tolerance + plugin->config.spill_angle * 90.0 / MAX_VALUE; \ +/* borders of the spill wedge */ \ + float min_h = hue_key - spill_tolerance; \ + float max_h = hue_key + spill_tolerance; \ + int desaturate_only = plugin->config.desaturate_only; \ + \ + float alpha_offset = plugin->config.alpha_offset / MAX_VALUE; + +// shortest distance between 2 hues +static float hue_distance(float h1, float h2) +{ + float result = h1 - h2; + if(result < -180) result += 360; + else if(result > 180) result -= 360; + return result; +} + +// shift H & S based on an X & Y offset +static void shift_hs(float &h_shifted, + float &s_shifted, + float h, + float s, + float x_offset, + float y_offset) +{ + float h_rad = TO_RAD(h); + float x = cos(h_rad) * s; + float y = sin(h_rad) * s; + x += x_offset; + y += y_offset; + h_shifted = TO_DEG(atan2(y, x)); + s_shifted = hypot(x, y); +} + +template +void ChromaKeyUnit::process_chromakey(int components, + component_type max, + bool use_yuv, + ChromaKeyPackage *pkg) +{ + COMMON_VARIABLES + + int w = plugin->input->get_w(); + +// printf("ChromaKeyUnit::process_chromakey %d hue_key=%f min_h=%f max_h=%f\n", +// __LINE__, +// hue_key, +// min_h, +// max_h); +// printf("ChromaKeyUnit::process_chromakey %d tolerance_in=%f tolerance=%f tolerance_out=%f\n", +// __LINE__, +// tolerance_in, +// tolerance, +// tolerance_out); +// printf("ChromaKeyUnit::process_chromakey %d min_s_in=%f min_s=%f min_s_out=%f\n", +// __LINE__, +// min_s_in, +// min_s, +// min_s_out); +// printf("ChromaKeyUnit::process_chromakey %d min_v_in=%f min_v=%f min_v_out=%f\n", +// __LINE__, +// min_v_in, +// min_v, +// min_v_out); +// printf("ChromaKeyUnit::process_chromakey %d max_v_in=%f max_v=%f max_v_out=%f\n", +// __LINE__, +// max_v_in, +// max_v, +// max_v_out); + + for (int i = pkg->y1; i < pkg->y2; i++) + { + component_type *row = (component_type *) plugin->input->get_rows ()[i]; + + for (int j = 0; j < w; j++) + { + float r, g, b, a = 1; + if (!use_yuv) + { + r = (float) row[0] / max; + g = (float) row[1] / max; + b = (float) row[2] / max; + } + else /* Convert pixel to RGB float */ + YUV::yuv.yuv_to_rgb_f (r, g, b, row[0], row[1], row[2]); + +// the input color + float h, s, v; + +// alpha contribution of each component of the HSV space + float ah = 1, as = 1, av = 1; + + HSV::rgb_to_hsv (r, g, b, h, s, v); + +// shift the input color in XY to get the wedge point + float h_shifted, s_shifted; + if(!EQUIV(plugin->config.saturation_start, 0)) + { + shift_hs(h_shifted, + s_shifted, + h, + s, + sat_x, + sat_y); + } + else + { + h_shifted = h; + s_shifted = s; + } + +/* Get the difference between the shifted input color & the unshifted wedge */ + float h_diff = hue_distance(h_shifted, hue_key); + h_diff = ABS(h_diff); + +// alpha contribution from hue difference +// outside wedge < tolerance_out < tolerance_in < inside wedge < tolerance_in < tolerance_out < outside wedge + if (tolerance_out > 0) + { +// completely inside the wedge + if (h_diff < tolerance_in) + ah = 0; + else +// between the outer & inner slope + if(h_diff < tolerance_out) + ah = (h_diff - tolerance_in) / (tolerance_out - tolerance_in); + if(ah > 1) ah = 1; + } + +// alpha contribution from saturation +// outside wedge < min_s_out < min_s_in < inside wedge + if(s_shifted > min_s_out) + { +// saturation with offset applied +// completely inside the wedge + if(s_shifted > min_s_in) + as = 0; +// inside the gradient + if(s_shifted >= min_s_out) + as = (min_s_in - s_shifted) / (min_s_in - min_s_out); + } + +// alpha contribution from brightness range +// brightness range is defined by 4 in/out variables +// outside wedge < min_v_out < min_v_in < inside wedge < max_v_in < max_v_out < outside wedge + if(v > min_v_out) + { + if(v < min_v_in || max_v_in >= 1.0) + av = (min_v_in - v) / (min_v_in - min_v_out); + else if(v <= max_v_in) + av = 0; + else if(v <= max_v_out) + av = (v - max_v_in) / (max_v_out - max_v_in); + } + +// combine the alpha contribution of every component into a single alpha + a = MAX(as, ah); + a = MAX(a, av); + +// Spill light processing + if(spill_tolerance > 0) + { +// get the difference between the shifted input color to the unshifted spill wedge + if(!EQUIV(spill_distance, 0)) + { + shift_hs(h_shifted, + s_shifted, + h, + s, + spill_x, + spill_y); + } + else + { + h_shifted = h; + s_shifted = s; + } + +// Difference between the shifted hue & the unshifted hue key + h_diff = hue_distance(h_shifted, hue_key); + +// inside the wedge + if(ABS(h_diff) < spill_tolerance) + { + if(!desaturate_only) + { +// the shifted input color in the unshifted wedge +// gives 2 unshifted border colors & the weighting + float blend = 0.5 + h_diff / spill_tolerance / 2; +// shift the 2 border colors to the output wedge + float min_h_shifted; + float min_s_shifted; + shift_hs(min_h_shifted, + min_s_shifted, + min_h, + s_shifted, + -spill_x, + -spill_y); + float max_h_shifted; + float max_s_shifted; + shift_hs(max_h_shifted, + max_s_shifted, + max_h, + s_shifted, + -spill_x, + -spill_y); + +// blend the shifted border colors using the unshifted weighting +// the only thing which doesn't restore the key color & doesn't make an edge is +// fading the saturation to 0 in the middle + if(blend > 0.5) + { + h = max_h_shifted; + s = max_s_shifted; + // s = max_s_shifted * (blend - 0.5) / 0.5; + // DEBUG + //h = 0; + //s = blend; + } + else + { + h = min_h_shifted; + s = min_s_shifted; + // s = min_s_shifted * (0.5 - blend) / 0.5; + // DEBUG + //h = 0; + //s = blend; + } + } + else // !desaturate_only + { + +// fade the saturation to 0 in the middle + s *= ABS(h_diff) / spill_tolerance; + } + + if(h < 0) h += 360; + HSV::hsv_to_rgb (r, g, b, h, s, v); + +// store new color + if (!use_yuv) + { + row[0] = (component_type) ((float) r * max); + row[1] = (component_type) ((float) g * max); + row[2] = (component_type) ((float) b * max); + } + else + YUV::yuv.rgb_to_yuv_f(r, g, b, row[0], row[1], row[2]); + } + } + + a += alpha_offset; + CLAMP (a, 0.0, 1.0); + + if (plugin->config.show_mask) + { + if (use_yuv) + { + row[0] = (component_type) ((float) a * max); + row[1] = (component_type) ((float) max / 2); + row[2] = (component_type) ((float) max / 2); + } + else + { + row[0] = (component_type) ((float) a * max); + row[1] = (component_type) ((float) a * max); + row[2] = (component_type) ((float) a * max); + } + } + + /* Multiply alpha and put back in frame */ + if (components == 4) + { + row[3] = MIN ((component_type) (a * max), row[3]); + } + else if (use_yuv) + { + row[0] = (component_type) ((float) a * row[0]); + row[1] = (component_type) ((float) a * (row[1] - (max / 2 + 1)) + + max / 2 + 1); + row[2] = (component_type) ((float) a * (row[2] - (max / 2 + 1)) + + max / 2 + 1); + } + else + { + row[0] = (component_type) ((float) a * row[0]); + row[1] = (component_type) ((float) a * row[1]); + row[2] = (component_type) ((float) a * row[2]); + } + + row += components; + } + } +} + + + +void ChromaKeyUnit::process_package(LoadPackage *package) +{ + ChromaKeyPackage *pkg = (ChromaKeyPackage*)package; + + switch(plugin->input->get_color_model()) + { + case BC_RGB_FLOAT: + process_chromakey (3, 1.0, 0, pkg); + break; + case BC_RGBA_FLOAT: + process_chromakey ( 4, 1.0, 0, pkg); + break; + case BC_RGB888: + process_chromakey ( 3, 0xff, 0, pkg); + break; + case BC_RGBA8888: + process_chromakey ( 4, 0xff, 0, pkg); + break; + case BC_YUV888: + process_chromakey ( 3, 0xff, 1, pkg); + break; + case BC_YUVA8888: + process_chromakey ( 4, 0xff, 1, pkg); + break; + case BC_YUV161616: + process_chromakey (3, 0xffff, 1, pkg); + break; + case BC_YUVA16161616: + process_chromakey (4, 0xffff, 1, pkg); + break; + } +} + + + +REGISTER_PLUGIN(ChromaKeyAvid) + + + +ChromaKeyAvid::ChromaKeyAvid(PluginServer *server) + : PluginVClient(server) +{ + engine = 0; +} + +ChromaKeyAvid::~ChromaKeyAvid() +{ + if(engine) delete engine; +} + + +int ChromaKeyAvid::process_buffer(VFrame *frame, + int64_t start_position, + double frame_rate) +{ + load_configuration(); + this->input = frame; + this->output = frame; + + read_frame(frame, + 0, + start_position, + frame_rate, + get_use_opengl()); + if(get_use_opengl()) return run_opengl(); + + if(!engine) engine = new ChromaKeyServer(this); + engine->process_packages(); + + return 0; +} + +const char* ChromaKeyAvid::plugin_title() { return N_("Chroma key (Avid)"); } +int ChromaKeyAvid::is_realtime() { return 1; } + + +LOAD_CONFIGURATION_MACRO(ChromaKeyAvid, ChromaKeyConfig) + + +void ChromaKeyAvid::save_data(KeyFrame * keyframe) +{ + FileXML output; + output.set_shared_output(keyframe->xbuf); + output.tag.set_title("CHROMAKEY_AVID"); + output.tag.set_property("RED", config.red); + output.tag.set_property("GREEN", config.green); + output.tag.set_property("BLUE", config.blue); + output.tag.set_property("MIN_BRIGHTNESS", config.min_brightness); + output.tag.set_property("MAX_BRIGHTNESS", config.max_brightness); + output.tag.set_property("SATURATION_START", config.saturation_start); + output.tag.set_property("SATURATION_LINE", config.saturation_line); + output.tag.set_property("TOLERANCE", config.tolerance); + output.tag.set_property("IN_SLOPE", config.in_slope); + output.tag.set_property("OUT_SLOPE", config.out_slope); + output.tag.set_property("ALPHA_OFFSET", config.alpha_offset); + output.tag.set_property("SPILL_SATURATION", config.spill_saturation); + output.tag.set_property("SPILL_ANGLE", config.spill_angle); + output.tag.set_property("DESATURATE_ONLY", config.desaturate_only); + output.tag.set_property("SHOW_MASK", config.show_mask); + output.append_tag(); + output.tag.set_title("/CHROMAKEY_AVID"); + output.append_tag(); + output.append_newline(); + output.terminate_string(); +} + +void ChromaKeyAvid::read_data(KeyFrame * keyframe) +{ + FileXML input; + + input.set_shared_input(keyframe->xbuf); + + while( !input.read_tag() ) + { + if( input.tag.title_is("CHROMAKEY_AVID") ) + { + config.red = input.tag.get_property("RED", config.red); + config.green = input.tag.get_property("GREEN", config.green); + config.blue = input.tag.get_property("BLUE", config.blue); + config.min_brightness = + input.tag.get_property("MIN_BRIGHTNESS", config.min_brightness); + config.max_brightness = + input.tag.get_property("MAX_BRIGHTNESS", config.max_brightness); + config.saturation_start = + input.tag.get_property("SATURATION_START", config.saturation_start); + config.saturation_line = + input.tag.get_property("SATURATION_LINE", config.saturation_line); + config.tolerance = + input.tag.get_property("TOLERANCE", config.tolerance); + config.in_slope = + input.tag.get_property("IN_SLOPE", config.in_slope); + config.out_slope = + input.tag.get_property("OUT_SLOPE", config.out_slope); + config.alpha_offset = + input.tag.get_property("ALPHA_OFFSET", config.alpha_offset); + config.spill_saturation = + input.tag.get_property("SPILL_SATURATION", config.spill_saturation); + config.spill_angle = + input.tag.get_property("SPILL_ANGLE", config.spill_angle); + config.desaturate_only = + input.tag.get_property("DESATURATE_ONLY", config.desaturate_only); + config.show_mask = + input.tag.get_property("SHOW_MASK", config.show_mask); + + config.undo_red = config.red; + config.undo_green = config.green; + config.undo_blue = config.blue; + config.undo_min_brightness = config.min_brightness; + config.undo_max_brightness = config.max_brightness; + config.undo_tolerance = config.tolerance; + config.undo_saturation_start = config.saturation_start; + config.undo_saturation_line = config.saturation_line; + config.undo_in_slope = config.in_slope; + config.undo_out_slope = config.out_slope; + config.undo_alpha_offset = config.alpha_offset; + config.undo_spill_saturation = config.spill_saturation; + config.undo_spill_angle = config.spill_angle; + config.undo_desaturate_only = config.desaturate_only; + config.undo_show_mask = config.show_mask; + } + } +} + + +NEW_WINDOW_MACRO(ChromaKeyAvid, ChromaKeyWindow) + + +void ChromaKeyAvid::update_gui() +{ + if( thread ) + { + load_configuration(); + ChromaKeyWindow *window = (ChromaKeyWindow*)thread->window; + window->lock_window(); + window->update_gui(RESET_ALL); + window->unlock_window(); + } +} + +void ChromaKeyWindow::update_gui(int clear) +{ + ChromaKeyConfig &config = plugin->config; + switch(clear) + { + case RESET_RGB: + update_sample(); + break; + case RESET_MIN_BRIGHTNESS: + min_brightness->update(config.min_brightness); + min_brightness_text->update(config.min_brightness); + break; + case RESET_MAX_BRIGHTNESS: + max_brightness->update(config.max_brightness); + max_brightness_text->update(config.max_brightness); + break; + case RESET_TOLERANCE: + tolerance->update(config.tolerance); + tolerance_text->update(config.tolerance); + break; + case RESET_SATURATION_START: + saturation_start->update(config.saturation_start); + saturation_start_text->update(config.saturation_start); + break; + case RESET_SATURATION_LINE: + saturation_line->update(config.saturation_line); + saturation_line_text->update(config.saturation_line); + break; + case RESET_IN_SLOPE: + in_slope->update(config.in_slope); + in_slope_text->update(config.in_slope); + break; + case RESET_OUT_SLOPE: + out_slope->update(config.out_slope); + out_slope_text->update(config.out_slope); + break; + case RESET_ALPHA_OFFSET: + alpha_offset->update(config.alpha_offset); + alpha_offset_text->update(config.alpha_offset); + break; + case RESET_SPILL_SATURATION: + spill_saturation->update(config.spill_saturation); + spill_saturation_text->update(config.spill_saturation); + break; + case RESET_SPILL_ANGLE: + spill_angle->update(config.spill_angle); + spill_angle_text->update(config.spill_angle); + break; + case RESET_ALL: + case RESET_DEFAULT_SETTINGS: + default: + min_brightness->update(config.min_brightness); + min_brightness_text->update(config.min_brightness); + max_brightness->update(config.max_brightness); + max_brightness_text->update(config.max_brightness); + saturation_start->update(config.saturation_start); + saturation_start_text->update(config.saturation_start); + saturation_line->update(config.saturation_line); + saturation_line_text->update(config.saturation_line); + tolerance->update(config.tolerance); + tolerance_text->update(config.tolerance); + in_slope->update(config.in_slope); + in_slope_text->update(config.in_slope); + out_slope->update(config.out_slope); + out_slope_text->update(config.out_slope); + alpha_offset->update(config.alpha_offset); + alpha_offset_text->update(config.alpha_offset); + spill_saturation->update(config.spill_saturation); + spill_saturation_text->update(config.spill_saturation); + spill_angle->update(config.spill_angle); + spill_angle_text->update(config.spill_angle); + desaturate_only->update(config.desaturate_only); + show_mask->update(config.show_mask); + update_sample(); + break; + } +} + + + +int ChromaKeyAvid::handle_opengl() +{ +#ifdef HAVE_GL +// For macro + ChromaKeyAvid *plugin = this; + COMMON_VARIABLES + + static const char *yuv_shader = + "const vec3 black = vec3(0.0, 0.5, 0.5);\n" + "uniform mat3 yuv_to_rgb_matrix;\n" + "uniform mat3 rgb_to_yuv_matrix;\n" + "uniform float yminf;\n" + "\n" + "vec4 yuv_to_rgb(vec4 color)\n" + "{\n" + " color.rgb -= vec3(yminf, 0.5, 0.5);\n" + " color.rgb = yuv_to_rgb_matrix * color.rgb;\n" + " return color;\n" + "}\n" + "\n" + "vec4 rgb_to_yuv(vec4 color)\n" + "{\n" + " color.rgb = rgb_to_yuv_matrix * color.rgb;\n" + " color.rgb += vec3(yminf, 0.5, 0.5);\n" + " return color;\n" + "}\n"; + + static const char *rgb_shader = + "const vec3 black = vec3(0.0, 0.0, 0.0);\n" + "\n" + "vec4 yuv_to_rgb(vec4 color)\n" + "{\n" + " return color;\n" + "}\n" + "vec4 rgb_to_yuv(vec4 color)\n" + "{\n" + " return color;\n" + "}\n"; + + static const char *hsv_shader = + "vec4 rgb_to_hsv(vec4 color)\n" + "{\n" + RGB_TO_HSV_FRAG("color") + " return color;\n" + "}\n" + "\n" + "vec4 hsv_to_rgb(vec4 color)\n" + "{\n" + HSV_TO_RGB_FRAG("color") + " return color;\n" + "}\n" + "\n"; + + static const char *show_rgbmask_shader = + "vec4 show_mask(vec4 color, vec4 color2)\n" + "{\n" + " return vec4(1.0, 1.0, 1.0, min(color.a, color2.a));" + "}\n"; + + static const char *show_yuvmask_shader = + "vec4 show_mask(vec4 color, vec4 color2)\n" + "{\n" + " return vec4(1.0, 0.5, 0.5, min(color.a, color2.a));" + "}\n"; + + static const char *nomask_shader = + "vec4 show_mask(vec4 color, vec4 color2)\n" + "{\n" + " return vec4(color.rgb, min(color.a, color2.a));" + "}\n"; + + get_output()->to_texture(); + get_output()->enable_opengl(); + get_output()->init_screen(); + + const char *shader_stack[16]; + memset(shader_stack,0, sizeof(shader_stack)); + int current_shader = 0; + + shader_stack[current_shader++] = \ + !BC_CModels::is_yuv(get_output()->get_color_model()) ? + rgb_shader : yuv_shader; + shader_stack[current_shader++] = hsv_shader; + shader_stack[current_shader++] = !config.show_mask ? nomask_shader : + !BC_CModels::is_yuv(get_output()->get_color_model()) ? + show_rgbmask_shader : show_yuvmask_shader ; + extern unsigned char _binary_chromakeyavid_sl_start[]; + static const char *shader_frag = (char*)_binary_chromakeyavid_sl_start; + shader_stack[current_shader++] = shader_frag; + + shader_stack[current_shader] = 0; + unsigned int shader = VFrame::make_shader(shader_stack); + if( shader > 0 ) { + glUseProgram(shader); + glUniform1i(glGetUniformLocation(shader, "tex"), 0); + glUniform1f(glGetUniformLocation(shader, "red"), red); + glUniform1f(glGetUniformLocation(shader, "green"), green); + glUniform1f(glGetUniformLocation(shader, "blue"), blue); + glUniform1f(glGetUniformLocation(shader, "in_slope"), in_slope); + glUniform1f(glGetUniformLocation(shader, "out_slope"), out_slope); + glUniform1f(glGetUniformLocation(shader, "tolerance"), tolerance); + glUniform1f(glGetUniformLocation(shader, "tolerance_in"), tolerance_in); + glUniform1f(glGetUniformLocation(shader, "tolerance_out"), tolerance_out); + glUniform1f(glGetUniformLocation(shader, "sat_x"), sat_x); + glUniform1f(glGetUniformLocation(shader, "sat_y"), sat_y); + glUniform1f(glGetUniformLocation(shader, "min_s"), min_s); + glUniform1f(glGetUniformLocation(shader, "min_s_in"), min_s_in); + glUniform1f(glGetUniformLocation(shader, "min_s_out"), min_s_out); + glUniform1f(glGetUniformLocation(shader, "min_v"), min_v); + glUniform1f(glGetUniformLocation(shader, "min_v_in"), min_v_in); + glUniform1f(glGetUniformLocation(shader, "min_v_out"), min_v_out); + glUniform1f(glGetUniformLocation(shader, "max_v"), max_v); + glUniform1f(glGetUniformLocation(shader, "max_v_in"), max_v_in); + glUniform1f(glGetUniformLocation(shader, "max_v_out"), max_v_out); + glUniform1f(glGetUniformLocation(shader, "spill_distance"), spill_distance); + glUniform1f(glGetUniformLocation(shader, "spill_x"), spill_x); + glUniform1f(glGetUniformLocation(shader, "spill_y"), spill_y); + glUniform1f(glGetUniformLocation(shader, "spill_tolerance"), spill_tolerance); + glUniform1f(glGetUniformLocation(shader, "min_h"), min_h); + glUniform1f(glGetUniformLocation(shader, "max_h"), max_h); + glUniform1i(glGetUniformLocation(shader, "desaturate_only"), desaturate_only); + glUniform1f(glGetUniformLocation(shader, "alpha_offset"), alpha_offset); + glUniform1f(glGetUniformLocation(shader, "hue_key"), hue_key); + glUniform1f(glGetUniformLocation(shader, "saturation_key"), saturation_key); + glUniform1f(glGetUniformLocation(shader, "value_key"), value_key); + if( BC_CModels::is_yuv(get_output()->get_color_model()) ) + BC_GL_COLORS(shader); + } + + get_output()->bind_texture(0); + glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST); + glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST); + + if(BC_CModels::components(get_output()->get_color_model()) == 3) + { + glEnable(GL_BLEND); + glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA); + get_output()->clear_pbuffer(); + } + get_output()->draw_texture(); + + glUseProgram(0); + get_output()->set_opengl_state(VFrame::SCREEN); + glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR); + glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR); + glDisable(GL_BLEND); + + +#endif + return 0; +} diff --git a/cinelerra-5.1/plugins/chromakeyavid/chromakey.h b/cinelerra-5.1/plugins/chromakeyavid/chromakey.h new file mode 100644 index 00000000..e39426ed --- /dev/null +++ b/cinelerra-5.1/plugins/chromakeyavid/chromakey.h @@ -0,0 +1,385 @@ + +/* + * CINELERRA + * Copyright (C) 2008 Adam Williams + * + * 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 + * + */ + +#ifndef CHROMAKEY_H +#define CHROMAKEY_H + +#include "colorpicker.h" +#include "guicast.h" +#include "theme.h" +#include "loadbalance.h" +#include "pluginvclient.h" + +#define RESET_DEFAULT_SETTINGS 20 +#define RESET_ALL 0 +#define RESET_RGB 1 +#define RESET_MIN_BRIGHTNESS 2 +#define RESET_MAX_BRIGHTNESS 3 +#define RESET_TOLERANCE 4 +#define RESET_SATURATION_START 5 +#define RESET_SATURATION_LINE 6 +#define RESET_IN_SLOPE 7 +#define RESET_OUT_SLOPE 8 +#define RESET_ALPHA_OFFSET 9 +#define RESET_SPILL_SATURATION 10 +#define RESET_SPILL_ANGLE 11 + +class ChromaKeyAvid; +class ChromaKeyWindow; +class ChromaKeyText; +class ChromaKeySlider; +class ChromaKeyClr; +class ChromaKeyReset; +class ChromaKeyStore; +class ChromaKeyRecall; +class ChromaKeyExchange; +class ChromaKeyUndo; + +enum { + CHROMAKEY_POSTPROCESS_NONE, + CHROMAKEY_POSTPROCESS_BLUR, + CHROMAKEY_POSTPROCESS_DILATE +}; + + +class ChromaKeyConfig +{ +public: + ChromaKeyConfig(); + void reset(int clear); + void copy_from(ChromaKeyConfig &src); + int equivalent(ChromaKeyConfig &src); + void interpolate(ChromaKeyConfig &prev, + ChromaKeyConfig &next, + int64_t prev_frame, + int64_t next_frame, + int64_t current_frame); + int get_color(); + + // Output mode + bool show_mask; + // Key color definition + float red; + float green; + float blue; + // Key shade definition + float min_brightness; + float max_brightness; + // distance of wedge point from the center + float saturation_start; + // minimum saturation after wedge point + float saturation_line; + // hue range + float tolerance; + // Mask feathering + float in_slope; + float out_slope; + float alpha_offset; + // Spill light compensation + float spill_saturation; + float spill_angle; + bool desaturate_only; + // Instances for Store + bool store_show_mask; + float store_red; + float store_green; + float store_blue; + float store_min_brightness; + float store_max_brightness; + float store_saturation_start; + float store_saturation_line; + float store_tolerance; + float store_in_slope; + float store_out_slope; + float store_alpha_offset; + float store_spill_saturation; + float store_spill_angle; + bool store_desaturate_only; + // Instances for Undo + bool undo_show_mask; + float undo_red; + float undo_green; + float undo_blue; + float undo_min_brightness; + float undo_max_brightness; + float undo_saturation_start; + float undo_saturation_line; + float undo_tolerance; + float undo_in_slope; + float undo_out_slope; + float undo_alpha_offset; + float undo_spill_saturation; + float undo_spill_angle; + bool undo_desaturate_only; +}; + + +class ChromaKeyColor : public BC_GenericButton +{ +public: + ChromaKeyColor(ChromaKeyAvid *plugin, + ChromaKeyWindow *gui, + int x, + int y); + + int handle_event(); + + ChromaKeyAvid *plugin; + ChromaKeyWindow *gui; +}; + +class ChromaKeyReset : public BC_GenericButton +{ +public: + ChromaKeyReset(ChromaKeyAvid *plugin, ChromaKeyWindow *gui, int x, int y); + int handle_event(); + ChromaKeyAvid *plugin; + ChromaKeyWindow *gui; +}; + +class ChromaKeyStore : public BC_GenericButton +{ +public: + ChromaKeyStore(ChromaKeyAvid *plugin, ChromaKeyWindow *gui, int x, int y); + int handle_event(); + ChromaKeyAvid *plugin; + ChromaKeyWindow *gui; +}; + +class ChromaKeyRecall : public BC_GenericButton +{ +public: + ChromaKeyRecall(ChromaKeyAvid *plugin, ChromaKeyWindow *gui, int x, int y); + int handle_event(); + ChromaKeyAvid *plugin; + ChromaKeyWindow *gui; +}; + +class ChromaKeyExchange : public BC_GenericButton +{ +public: + ChromaKeyExchange(ChromaKeyAvid *plugin, ChromaKeyWindow *gui, int x, int y); + int handle_event(); + ChromaKeyAvid *plugin; + ChromaKeyWindow *gui; +}; + +class ChromaKeyUndo : public BC_GenericButton +{ +public: + ChromaKeyUndo(ChromaKeyAvid *plugin, ChromaKeyWindow *gui, int x, int y); + int handle_event(); + ChromaKeyAvid *plugin; + ChromaKeyWindow *gui; +}; + +class ChromaKeyText : public BC_TumbleTextBox +{ +public: + ChromaKeyText(ChromaKeyAvid *plugin, ChromaKeyWindow *gui, + ChromaKeySlider *slider, int x, int y, + float min, float max, float *output); + ~ChromaKeyText(); + int handle_event(); + ChromaKeyAvid *plugin; + ChromaKeyWindow *gui; + ChromaKeySlider *slider; + float *output; + float min, max; +}; + +class ChromaKeySlider : public BC_FSlider +{ +public: + ChromaKeySlider(ChromaKeyAvid *plugin, ChromaKeyText *text, + int x, int y, int w, float min, float max, float *output); + ~ChromaKeySlider(); + int handle_event(); + ChromaKeyAvid *plugin; + ChromaKeyText *text; + float *output; +}; + +class ChromaKeyClr : public BC_Button +{ +public: + ChromaKeyClr(ChromaKeyAvid *plugin, ChromaKeyWindow *gui, + int x, int y, int clear); + ~ChromaKeyClr(); + int handle_event(); + ChromaKeyAvid *plugin; + ChromaKeyWindow *gui; + int clear; +}; + + +class ChromaKeyUseColorPicker : public BC_GenericButton +{ +public: + ChromaKeyUseColorPicker(ChromaKeyAvid *plugin, + ChromaKeyWindow *gui, int x, int y); + int handle_event(); + ChromaKeyAvid *plugin; + ChromaKeyWindow *gui; +}; + +class ChromaKeyColorThread : public ColorPicker +{ +public: + ChromaKeyColorThread(ChromaKeyAvid *plugin, ChromaKeyWindow *gui); + int handle_new_color(int output, int alpha); + ChromaKeyAvid *plugin; + ChromaKeyWindow *gui; +}; + +class ChromaKeyToggle : public BC_CheckBox +{ +public: + ChromaKeyToggle(ChromaKeyAvid *plugin, + int x, + int y, + bool *output, + const char *caption); + int handle_event(); + ChromaKeyAvid *plugin; + bool *output; +}; + + + +class ChromaKeyWindow : public PluginClientWindow +{ +public: + ChromaKeyWindow(ChromaKeyAvid *plugin); + ~ChromaKeyWindow(); + + void create_objects(); + void update_sample(); + void update_gui(int clear); + void done_event(int result); + + ChromaKeyColor *color; + ChromaKeyUseColorPicker *use_colorpicker; + + ChromaKeyText *min_brightness_text; + ChromaKeySlider *min_brightness; + ChromaKeyClr *min_brightness_clr; + + ChromaKeyText *max_brightness_text; + ChromaKeySlider *max_brightness; + ChromaKeyClr *max_brightness_clr; + + ChromaKeyText *saturation_start_text; + ChromaKeySlider *saturation_start; + ChromaKeyClr *saturation_start_clr; + + ChromaKeyText *saturation_line_text; + ChromaKeySlider *saturation_line; + ChromaKeyClr *saturation_line_clr; + + ChromaKeyText *tolerance_text; + ChromaKeySlider *tolerance; + ChromaKeyClr *tolerance_clr; + + ChromaKeyText *in_slope_text; + ChromaKeySlider *in_slope; + ChromaKeyClr *in_slope_clr; + + ChromaKeyText *out_slope_text; + ChromaKeySlider *out_slope; + ChromaKeyClr *out_slope_clr; + + ChromaKeyText *alpha_offset_text; + ChromaKeySlider *alpha_offset; + ChromaKeyClr *alpha_offset_clr; + + ChromaKeyText *spill_saturation_text; + ChromaKeySlider *spill_saturation; + ChromaKeyClr *spill_saturation_clr; + + ChromaKeyText *spill_angle_text; + ChromaKeySlider *spill_angle; + ChromaKeyClr *spill_angle_clr; + + ChromaKeyToggle *desaturate_only; + ChromaKeyToggle *show_mask; + + ChromaKeyReset *reset; + ChromaKeyStore *store; + ChromaKeyRecall *recall; + ChromaKeyExchange *exchange; + ChromaKeyUndo *undo; + + BC_SubWindow *sample; + ChromaKeyAvid *plugin; + ChromaKeyColorThread *color_thread; +}; + +class ChromaKeyServer : public LoadServer +{ +public: + ChromaKeyServer(ChromaKeyAvid *plugin); + void init_packages(); + LoadClient* new_client(); + LoadPackage* new_package(); + + ChromaKeyAvid *plugin; +}; + +class ChromaKeyPackage : public LoadPackage +{ +public: + ChromaKeyPackage(); + int y1, y2; +}; + +class ChromaKeyUnit : public LoadClient +{ +public: + ChromaKeyUnit(ChromaKeyAvid *plugin, ChromaKeyServer *server); + void process_package(LoadPackage *package); + template void process_chromakey(int components, component_type max, bool use_yuv, ChromaKeyPackage *pkg); + bool is_same_color(float r, float g, float b, float rk,float bk,float gk, + float color_threshold, float light_threshold, + int key_main_component); + + ChromaKeyAvid *plugin; +}; + +class ChromaKeyAvid : public PluginVClient +{ +public: + ChromaKeyAvid(PluginServer *server); + ~ChromaKeyAvid(); + + PLUGIN_CLASS_MEMBERS(ChromaKeyConfig); + int process_buffer(VFrame *frame, int64_t start_position, double frame_rate); + int handle_opengl(); + int is_realtime(); + void save_data(KeyFrame *keyframe); + void read_data(KeyFrame *keyframe); + void update_gui(); + + VFrame *input, *output; + ChromaKeyServer *engine; +}; + +#endif diff --git a/cinelerra-5.1/plugins/chromakeyavid/chromakeyavid.sl b/cinelerra-5.1/plugins/chromakeyavid/chromakeyavid.sl new file mode 100644 index 00000000..2fd86da4 --- /dev/null +++ b/cinelerra-5.1/plugins/chromakeyavid/chromakeyavid.sl @@ -0,0 +1,230 @@ +uniform sampler2D tex; +uniform float red; +uniform float green; +uniform float blue; +uniform float in_slope; +uniform float out_slope; +uniform float tolerance; +uniform float tolerance_in; +uniform float tolerance_out; +uniform float sat_x; +uniform float sat_y; +uniform float min_s; +uniform float min_s_in; +uniform float min_s_out; +uniform float min_v; +uniform float min_v_in; +uniform float min_v_out; +uniform float max_v; +uniform float max_v_in; +uniform float max_v_out; +uniform float spill_distance; +uniform float spill_x; +uniform float spill_y; +uniform float spill_tolerance; +uniform float min_h; +uniform float max_h; +uniform bool desaturate_only; +uniform float alpha_offset; +uniform float hue_key; +uniform float saturation_key; +uniform float value_key; + + + +// shortest distance between 2 hues +float hue_distance(float h1, float h2) +{ + float result = h1 - h2; + if(result < -180.0) result += 360.0; + else + if(result > 180.0) result -= 360.0; + return result; +} + +// shift H & S based on an X & Y offset +void shift_hs(out float h_shifted, + out float s_shifted, + float h, + float s, + float x_offset, + float y_offset) +{ + float h_rad = radians(h); + float x = cos(h_rad) * s; + float y = sin(h_rad) * s; + x += x_offset; + y += y_offset; + h_shifted = degrees(atan(y, x)); + s_shifted = length(vec2(x, y)); +} + + +void main() +{ + vec4 color = texture2D(tex, gl_TexCoord[0].st); +/* Contribution to alpha from each component */ + float ah = 1.0; + float as = 1.0; + float av = 1.0; + float a = 1.0; + vec4 color2; + +/* Convert to HSV */ + color2 = yuv_to_rgb(color); + color2 = rgb_to_hsv(color2); + float h = color2.r; + float s = color2.g; + float v = color2.b; + +// shift the color in XY to shift the wedge point + float h_shifted, s_shifted; + shift_hs(h_shifted, + s_shifted, + h, + s, + sat_x, + sat_y); + +/* Get the difference between the current hue & the hue key */ + float h_diff = abs(hue_distance(h_shifted, hue_key)); + +// alpha contribution from hue difference +// outside wedge < tolerance_out < tolerance_in < inside wedge < tolerance_in < tolerance_out < outside wedge + if (tolerance_out > 0.0) + { +// completely inside the wedge + if (h_diff < tolerance_in) + ah = 0.0; + else +// between the outer & inner slope + if(h_diff < tolerance_out) + ah = (h_diff - tolerance_in) / (tolerance_out - tolerance_in); + if(ah > 1.0) ah = 1.0; + } + +// alpha contribution from saturation +// outside wedge < min_s_out < min_s_in < inside wedge + if(s_shifted > min_s_out) + { +// saturation with offset applied +// completely inside the wedge + if(s_shifted > min_s_in) + as = 0.0; +// inside the gradient + if(s_shifted >= min_s_out) + as = (min_s_in - s_shifted) / (min_s_in - min_s_out); + } + + +// alpha contribution from brightness range +// brightness range is defined by 4 in/out variables +// outside wedge < min_v_out < min_v_in < inside wedge < max_v_in < max_v_out < outside wedge + if(v > min_v_out) + { + if(v < min_v_in || max_v_in >= 1.0) + av = (min_v_in - v) / (min_v_in - min_v_out); + else + if(v <= max_v_in) + av = 0.0; + else + if(v <= max_v_out) + av = (v - max_v_in) / (max_v_out - max_v_in); + } + +// combine the alpha contribution of every component into a single alpha + a = max(as, ah); + a = max(a, av); + +// Spill light processing + if(spill_tolerance > 0.0) + { +// get the difference between the shifted input color to the unshifted spill wedge + if(spill_distance != 0.0) + { + shift_hs(h_shifted, + s_shifted, + h, + s, + spill_x, + spill_y); + } + else + { + h_shifted = h; + s_shifted = s; + } + +// Difference between the shifted hue & the unshifted hue key + h_diff = hue_distance(h_shifted, hue_key); + +// inside the wedge + if(abs(h_diff) < spill_tolerance) + { + if(!desaturate_only) + { +// the shifted input color in the unshifted wedge +// gives 2 unshifted border colors & the weighting + float blend = 0.5 + h_diff / spill_tolerance / 2.0; +// shift the 2 border colors to the output wedge + float min_h_shifted; + float min_s_shifted; + shift_hs(min_h_shifted, + min_s_shifted, + min_h, + s_shifted, + -spill_x, + -spill_y); + float max_h_shifted; + float max_s_shifted; + shift_hs(max_h_shifted, + max_s_shifted, + max_h, + s_shifted, + -spill_x, + -spill_y); + + +// blend the shifted border colors using the unshifted weighting +// the only thing which doesn't restore the key color & doesn't make an edge is +// fading the saturation to 0 in the middle + if(blend > 0.5) + { + h = max_h_shifted; + s = max_s_shifted; + } + else + { + h = min_h_shifted; + s = min_s_shifted; + } + } + else // !desaturate_only + { + +// fade the saturation to 0 in the middle + s *= abs(h_diff) / spill_tolerance; + } + + + if(h < 0.0) h += 360.0; + +// store new color + color2.r = h; + color2.g = s; + color2.b = v; + color2 = hsv_to_rgb(color2); + color.rgb = rgb_to_yuv(color2).rgb; + } + } + + + + a += alpha_offset; + a = min(a, 1.0); + color.a = a; +/* Convert mask into image */ + gl_FragColor = show_mask(color, color2); +} + + diff --git a/cinelerra-5.1/plugins/swatch/Makefile b/cinelerra-5.1/plugins/swatch/Makefile new file mode 100644 index 00000000..4f1cb99b --- /dev/null +++ b/cinelerra-5.1/plugins/swatch/Makefile @@ -0,0 +1,8 @@ +include ../../plugin_defs + +PLUGIN = swatch +OBJS = $(OBJDIR)/swatch.o + +include ../../plugin_config + +$(OBJDIR)/swatch.o: swatch.C diff --git a/cinelerra-5.1/plugins/swatch/swatch.C b/cinelerra-5.1/plugins/swatch/swatch.C new file mode 100644 index 00000000..2e9142eb --- /dev/null +++ b/cinelerra-5.1/plugins/swatch/swatch.C @@ -0,0 +1,760 @@ +/* + * CINELERRA + * Copyright (C) 2024 Adam Williams + * + * 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 + * + */ + + +// draw a color swatch for a given brightness or saturation +// does not visualize but draws output to be processed + +#include +#include +#include + +#include "bcdisplayinfo.h" +#include "bccolors.h" +#include "clip.h" +#include "bchash.h" +#include "filexml.h" +#include "keyframe.h" +#include "language.h" +#include "playback3d.h" +#include "swatch.h" +#include "vframe.h" + + + + +REGISTER_PLUGIN(SwatchMain) + + +#define MAX_VALUE 100 + + + +SwatchConfig::SwatchConfig() +{ + brightness = MAX_VALUE; + saturation = MAX_VALUE; + fix_brightness = 0; + angle = 0; + draw_src = 0; +} + +int SwatchConfig::equivalent(SwatchConfig &that) +{ + return brightness == that.brightness && + saturation == that.saturation && + fix_brightness == that.fix_brightness && + angle == that.angle && + draw_src == that.draw_src; +} + +void SwatchConfig::copy_from(SwatchConfig &that) +{ + brightness = that.brightness; + saturation = that.saturation; + fix_brightness = that.fix_brightness; + angle = that.angle; + draw_src = that.draw_src; +} + +void SwatchConfig::interpolate(SwatchConfig &prev, + SwatchConfig &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->brightness = (int)(prev.brightness * prev_scale + next.brightness * next_scale); + this->saturation = (int)(prev.saturation * prev_scale + next.saturation * next_scale); + this->angle = (int)(prev.angle * prev_scale + next.angle * next_scale); + fix_brightness = prev.fix_brightness; + draw_src = prev.draw_src; +} + + + + +SwatchSlider::SwatchSlider(SwatchMain *plugin, + SwatchWindow *gui, + int x, + int y, + int min, + int max, + int *output) + : BC_ISlider(x, + y, + 0, + gui->get_w() - xS(10) - x, + gui->get_w() - xS(10) - x, + min, + max, + *output) +{ + this->plugin = plugin; + this->output = output; +} + +int SwatchSlider::handle_event () +{ + *output = get_value(); + plugin->send_configure_change(); + return 1; +} + +SwatchRadial::SwatchRadial(SwatchMain *plugin, + SwatchWindow *gui, + int x, + int y, + const char *text, + int fix_brightness) + : BC_Radial(x, + y, + (plugin->config.fix_brightness && fix_brightness), + text) +{ + this->plugin = plugin; + this->gui = gui; + this->fix_brightness = fix_brightness; +} + +int SwatchRadial::handle_event() +{ + plugin->config.fix_brightness = fix_brightness; + gui->update_fixed(); + plugin->send_configure_change(); + return 1; +} + + +SwatchCheck::SwatchCheck(SwatchMain *plugin, + int x, + int y, + const char *text, + int *output) + : BC_CheckBox(x, + y, + *output, + text) +{ + this->plugin = plugin; + this->output = output; +} + +int SwatchCheck::handle_event() +{ + *output = get_value(); + plugin->send_configure_change(); + return 1; +} + + + + +SwatchWindow::SwatchWindow(SwatchMain *plugin) + : PluginClientWindow(plugin, + xS(350), + yS(300), + xS(350), + yS(300), + 0) +{ + this->plugin = plugin; +} + +SwatchWindow::~SwatchWindow() +{ +} + +void SwatchWindow::create_objects() +{ + int margin = yS(10); + int x = margin, y = margin; + BC_Title *title; + + add_subwindow(brightness_title = new BC_Title(x, y, _("Brightness:"))); + y += brightness_title->get_h() + margin; + add_subwindow (brightness = new SwatchSlider(plugin, this, x, y, 0, MAX_VALUE, &plugin->config.brightness)); + y += brightness->get_h() + margin; + + add_subwindow(saturation_title = new BC_Title(x, y, _("Saturation:"))); + y += saturation_title->get_h() + margin; + add_subwindow (saturation = new SwatchSlider(plugin, this, x, y, 0, MAX_VALUE, &plugin->config.saturation)); + y += saturation->get_h() + margin; + + add_subwindow(title = new BC_Title(x, y, _("Angle:"))); + y += title->get_h() + margin; + add_subwindow (angle = new SwatchSlider(plugin, this, x, y, -180, 180, &plugin->config.angle)); + y += saturation->get_h() + margin; + + add_subwindow(fix_brightness = new SwatchRadial(plugin, + this, + x, + y, + _("Constant Brightness"), + 1)); + y += fix_brightness->get_h() + margin; + add_subwindow(fix_saturation = new SwatchRadial(plugin, + this, + x, + y, + _("Constant Saturation"), + 0)); + y += fix_saturation->get_h() + margin; + update_fixed(); + + add_subwindow(draw_src = new SwatchCheck(plugin, + x, + y, + _("Draw source"), + &plugin->config.draw_src)); + + + show_window(); +} + +void SwatchWindow::update_fixed() +{ + fix_brightness->update(plugin->config.fix_brightness); + fix_saturation->update(!plugin->config.fix_brightness); + if(plugin->config.fix_brightness) + { + saturation_title->set_color(BC_WindowBase::get_resources()->disabled_text_color); + brightness_title->set_color(BC_WindowBase::get_resources()->default_text_color); + saturation->disable(); + brightness->enable(); + } + else + { + saturation_title->set_color(BC_WindowBase::get_resources()->default_text_color); + brightness_title->set_color(BC_WindowBase::get_resources()->disabled_text_color); + saturation->enable(); + brightness->disable(); + } +} + + +SwatchMain::SwatchMain(PluginServer *server) + : PluginVClient(server) +{ + need_reconfigure = 1; + engine = 0; + temp = 0; + src_temp = 0; +} + +SwatchMain::~SwatchMain() +{ + if(engine) delete engine; + if(temp) delete temp; + if(src_temp) delete src_temp; +} + +const char* SwatchMain::plugin_title() { return N_("Color Swatch"); } +int SwatchMain::is_realtime() { return 1; } + + +NEW_WINDOW_MACRO(SwatchMain, SwatchWindow) + +LOAD_CONFIGURATION_MACRO(SwatchMain, SwatchConfig) + +int SwatchMain::is_synthesis() +{ + return 1; +} + +int SwatchMain::process_buffer(VFrame *frame, + int64_t start_position, + double frame_rate) +{ + need_reconfigure |= load_configuration(); + int use_opengl = get_use_opengl(); + +// have to draw output pixels out of order + if(config.draw_src) use_opengl = 0; + + if(use_opengl) return run_opengl(); + + if(!engine) engine = new SwatchEngine(this, + get_project_smp() + 1, + get_project_smp() + 1); + if(config.draw_src) + { + if(!src_temp) + src_temp = new VFrame(0, + -1, + frame->get_w(), + frame->get_h(), + frame->get_color_model(), + -1); + + read_frame(src_temp, + 0, + start_position, + frame_rate, + use_opengl); + } + + + if(!temp) temp = new VFrame(0, + -1, + frame->get_w(), + frame->get_h(), + frame->get_color_model(), + -1); + +// draw pattern once + if(need_reconfigure) + engine->draw_pattern(); + +// draw the pattern on the output + frame->copy_from(temp); +// draw input on the pattern + if(config.draw_src) + engine->draw_src(); + +//printf("SwatchMain::process_buffer %d %d\n", __LINE__, config.draw_src); + return 0; +} + + +void SwatchMain::update_gui() +{ + if(thread) + { + if(load_configuration()) + { + ((SwatchWindow*)thread->window)->lock_window("SwatchMain::update_gui"); + ((SwatchWindow*)thread->window)->brightness->update(config.brightness); + ((SwatchWindow*)thread->window)->saturation->update(config.saturation); + ((SwatchWindow*)thread->window)->angle->update(config.angle); + ((SwatchWindow*)thread->window)->draw_src->update(config.draw_src); + ((SwatchWindow*)thread->window)->update_fixed(); + ((SwatchWindow*)thread->window)->unlock_window(); + } + } +} + + + + + +void SwatchMain::save_data(KeyFrame *keyframe) +{ + FileXML output; + +// cause data to be stored directly in text + output.set_shared_output(keyframe->xbuf); + output.tag.set_title("SWATCH"); + + output.tag.set_property("BRIGHTNESS", config.brightness); + output.tag.set_property("SATURATION", config.saturation); + output.tag.set_property("ANGLE", config.angle); + output.tag.set_property("FIX_BRIGHTNESS", config.fix_brightness); + output.tag.set_property("DRAW_SRC", config.draw_src); + output.append_tag(); + output.tag.set_title("/SWATCH"); + output.append_tag(); + output.append_newline(); + output.terminate_string(); +//printf("SwatchMain::save_data %d %s\n", __LINE__, output.string); +} + +void SwatchMain::read_data(KeyFrame *keyframe) +{ + FileXML input; + + input.set_shared_input(keyframe->xbuf); + + while( !input.read_tag() ) + { + if(input.tag.title_is("SWATCH")) + { + config.brightness = + input.tag.get_property("BRIGHTNESS", config.brightness); + config.saturation = + input.tag.get_property("SATURATION", config.saturation); + config.angle = + input.tag.get_property("ANGLE", config.angle); + config.fix_brightness = + input.tag.get_property("FIX_BRIGHTNESS", config.fix_brightness); + config.draw_src = + input.tag.get_property("DRAW_SRC", config.draw_src); + } + } +} + +int SwatchMain::handle_opengl() +{ +#ifdef HAVE_GL + const char *head_frag = + "uniform mat3 yuv_to_rgb_matrix;\n" + "uniform mat3 rgb_to_yuv_matrix;\n" + "uniform float yminf;\n" + "uniform sampler2D tex;\n" + "uniform vec2 texture_extents;\n" + "uniform vec2 center_coord;\n" + "uniform float value;\n" + "uniform float saturation;\n" + "uniform float angle;\n" + "uniform bool fix_value;\n" + "\n" + "void main()\n" + "{\n" + " vec2 out_coord = gl_TexCoord[0].st * texture_extents;\n" + " vec2 in_coord = vec2(out_coord.x - center_coord.x, out_coord.y - center_coord.y);\n" + " float max_s = center_coord.x;\n" + " if(center_coord.y < max_s) max_s = center_coord.y;\n" + " vec4 pixel;\n" + " pixel.a = 1.0;\n" + " pixel.r = atan(in_coord.x, in_coord.y) / 2.0 / 3.14159 * 360.0 + angle; // hue\n" + " if(pixel.r < 0.0) pixel.r += 360.0;\n" + " if(fix_value)\n" + " {\n" + " pixel.g = length(vec2(in_coord.x, in_coord.y)) / max_s; // saturation\n" + " if(pixel.g > 1.0) pixel.g = 1.0; \n" + " pixel.b = value;\n" + " }\n" + " else\n" + " {\n" + " pixel.g = saturation;\n" + " pixel.b = length(vec2(in_coord.x, in_coord.y)) / max_s; // value\n" + " if(pixel.b > 1.0) pixel.b = 1.0; \n" + " }\n" + HSV_TO_RGB_FRAG("pixel"); + + static const char *put_yuv_frag = + RGB_TO_YUV_FRAG("pixel") + " gl_FragColor = pixel;\n" + "}\n"; + + static const char *put_rgb_frag = + " gl_FragColor = pixel;\n" + "}\n"; + + + + + const char *shader_stack[5] = { 0, 0, 0, 0, 0 }; + shader_stack[0] = head_frag; + if(BC_CModels::is_yuv(get_output()->get_color_model())) + shader_stack[1] = put_yuv_frag; + else + shader_stack[1] = put_rgb_frag; + + get_output()->set_opengl_state(VFrame::TEXTURE); + get_output()->to_texture(); + get_output()->enable_opengl(); + get_output()->init_screen(); + get_output()->bind_texture(0); + + unsigned int frag = VFrame::make_shader(0, + shader_stack[0], + shader_stack[1], + 0); + + if(frag) + { + glUseProgram(frag); + float w = get_output()->get_w(); + float h = get_output()->get_h(); + float texture_w = get_output()->get_texture_w(); + float texture_h = get_output()->get_texture_h(); + glUniform1i(glGetUniformLocation(frag, "tex"), 0); + glUniform2f(glGetUniformLocation(frag, "center_coord"), + (GLfloat)w / 2, + (GLfloat)h / 2); + glUniform2f(glGetUniformLocation(frag, "texture_extents"), + (GLfloat)texture_w, + (GLfloat)texture_h); + + glUniform1f(glGetUniformLocation(frag, "value"), (float)config.brightness / MAX_VALUE); + glUniform1f(glGetUniformLocation(frag, "saturation"), (float)config.saturation / MAX_VALUE); + glUniform1f(glGetUniformLocation(frag, "angle"), (float)config.angle); + glUniform1i(glGetUniformLocation(frag, "fix_value"), config.fix_brightness); + if(BC_CModels::is_yuv(get_output()->get_color_model())) + BC_GL_COLORS(frag); + } + + get_output()->draw_texture(); + glUseProgram(0); + get_output()->set_opengl_state(VFrame::SCREEN); + +#endif + return 0; +} + + + + + + + + + + + +SwatchPackage::SwatchPackage() + : LoadPackage() +{ +} + + + + +SwatchUnit::SwatchUnit(SwatchEngine *server, SwatchMain *plugin) + : LoadClient(server) +{ + this->plugin = plugin; + this->server = server; +} + + + + +#define CREATE_SWATCH(type, components, max, is_yuv) \ +{ \ + for(int i = pkg->y1; i < pkg->y2; i++) \ + { \ + type *out_row = (type*)plugin->temp->get_rows()[i]; \ + for(int j = 0; j < w; j++) \ + { \ + float hue = atan2(j - center_x, i - center_y) * 360 / 2 / M_PI + angle; \ + if(fix_brightness) \ + { \ + saturation = hypot(j - center_x, i - center_y) / max_s; \ + if(saturation > 1) saturation = 1; \ + } \ + else \ + { \ + value = hypot(j - center_x, i - center_y) / max_s; \ + if(value > 1) value = 1; \ + } \ + if(hue < 0) hue += 360; \ + if(is_yuv) \ + { \ + int y, u, v; \ + HSV::hsv_to_yuv(y, u, v, hue, saturation, value, max); \ + out_row[0] = y; \ + out_row[1] = u; \ + out_row[2] = v; \ + } \ + else \ + { \ + float r, g, b; \ + HSV::hsv_to_rgb(r, g, b, hue, saturation, value); \ + out_row[0] = (type)(r * max); \ + out_row[1] = (type)(g * max); \ + out_row[2] = (type)(b * max); \ + } \ + \ +/* the alpha */ \ + if(components == 4) out_row[3] = max; \ + out_row += components; \ + } \ + } \ +} + + +#define DRAW_SRC(type, components, max, is_yuv) \ +{ \ + type **dst_rows = (type**)plugin->get_output()->get_rows(); \ + for(int i = pkg->y1; i < pkg->y2; i++) \ + { \ + type *src_row = (type*)plugin->src_temp->get_rows()[i]; \ + for(int j = 0; j < w; j++) \ + { \ +/* the source values */ \ + type r, g, b; \ + float r2, g2, b2; \ + if(is_yuv) \ + { \ + YUV::yuv.yuv_to_rgb_f (r2, g2, b2, src_row[0], src_row[1], src_row[2]); \ + } \ + else \ + { \ + r = src_row[0]; \ + g = src_row[1]; \ + b = src_row[2]; \ + r2 = (float)r / max; \ + g2 = (float)g / max; \ + b2 = (float)b / max; \ + } \ + float hue, s, value; \ + HSV::rgb_to_hsv(r2, g2, b2, hue, s, value); \ + float h_rad = TO_RAD(hue - angle); \ +/* get coordinate of color in output */ \ + int x_out, y_out; \ + if(fix_brightness) \ + { \ + x_out = center_x + (int)(sin(h_rad) * s * max_s); \ + y_out = center_y + (int)(cos(h_rad) * s * max_s); \ + } \ + else \ + { \ + x_out = center_x + (int)(sin(h_rad) * value * max_s); \ + y_out = center_y + (int)(cos(h_rad) * value * max_s); \ + } \ + if(x_out >= 0 && x_out < w && y_out >= 0 && y_out < h) \ + { \ + type *dst = dst_rows[y_out] + x_out * components; \ + if(is_yuv) \ + { \ + dst[0] = src_row[0]; \ + dst[1] = src_row[1]; \ + dst[2] = src_row[2]; \ + } \ + else \ + { \ + dst[0] = r; \ + dst[1] = g; \ + dst[2] = b; \ + } \ + } \ + \ + src_row += components; \ + } \ + } \ +} + +#define DRAW_PATTERN_MODE 0 +#define DRAW_SRC_MODE 1 + +void SwatchUnit::process_package(LoadPackage *package) +{ + SwatchPackage *pkg = (SwatchPackage*)package; + int h = plugin->temp->get_h(); + int w = plugin->temp->get_w(); + int center_x = w / 2; + int center_y = h / 2; + int cmodel = plugin->temp->get_color_model(); +// maximum saturation + float max_s = center_x; + if(center_y < max_s) max_s = center_y; + int fix_brightness = plugin->config.fix_brightness; + float saturation = (float)plugin->config.saturation / MAX_VALUE; + float value = (float)plugin->config.brightness / MAX_VALUE; + float angle = plugin->config.angle; + + if(server->mode == DRAW_PATTERN_MODE) + { + switch(cmodel) + { + case BC_RGB888: + CREATE_SWATCH(unsigned char, 3, 0xff, 0) + break; + case BC_RGBA8888: + CREATE_SWATCH(unsigned char, 4, 0xff, 0) + break; + case BC_RGB_FLOAT: + CREATE_SWATCH(float, 3, 1.0, 0) + break; + case BC_RGBA_FLOAT: + CREATE_SWATCH(float, 4, 1.0, 0) + break; + case BC_YUV888: + CREATE_SWATCH(unsigned char, 3, 0xff, 1) + break; + case BC_YUVA8888: + CREATE_SWATCH(unsigned char, 4, 0xff, 1) + break; + } + } + else + { + switch(cmodel) + { + case BC_RGB888: + DRAW_SRC(unsigned char, 3, 0xff, 0) + break; + case BC_RGBA8888: + DRAW_SRC(unsigned char, 4, 0xff, 0) + break; + case BC_RGB_FLOAT: + DRAW_SRC(float, 3, 1.0, 0) + break; + case BC_RGBA_FLOAT: + DRAW_SRC(float, 4, 1.0, 0) + break; + case BC_YUV888: + DRAW_SRC(unsigned char, 3, 0xff, 1) + break; + case BC_YUVA8888: + DRAW_SRC(unsigned char, 4, 0xff, 1) + break; + } + } +} + + + + + + +SwatchEngine::SwatchEngine(SwatchMain *plugin, + int total_clients, + int total_packages) + : LoadServer(total_clients, total_packages) +{ + this->plugin = plugin; +} + +void SwatchEngine::draw_pattern() +{ + mode = DRAW_PATTERN_MODE; + process_packages(); +} + +void SwatchEngine::draw_src() +{ + mode = DRAW_SRC_MODE; + process_packages(); +} + +void SwatchEngine::init_packages() +{ + for(int i = 0; i < get_total_packages(); i++) + { + SwatchPackage *package = (SwatchPackage*)get_package(i); + package->y1 = plugin->temp->get_h() * + i / + get_total_packages(); + package->y2 = plugin->temp->get_h() * + (i + 1) / + get_total_packages(); + } +} + +LoadClient* SwatchEngine::new_client() +{ + return new SwatchUnit(this, plugin); +} + +LoadPackage* SwatchEngine::new_package() +{ + return new SwatchPackage; +} + + + + + diff --git a/cinelerra-5.1/plugins/swatch/swatch.h b/cinelerra-5.1/plugins/swatch/swatch.h new file mode 100644 index 00000000..d4551c99 --- /dev/null +++ b/cinelerra-5.1/plugins/swatch/swatch.h @@ -0,0 +1,186 @@ +/* + * CINELERRA + * Copyright (C) 2024 Adam Williams + * + * 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 + * + */ + +#ifndef SWATCH_H +#define SWATCH_H + +class SwatchMain; +class SwatchEngine; +class SwatchWindow; +class SwatchEngine; + + + +#include "guicast.h" +#include "loadbalance.h" +#include "pluginvclient.h" + +class SwatchConfig +{ +public: + SwatchConfig(); + + int equivalent(SwatchConfig &that); + void copy_from(SwatchConfig &that); + void interpolate(SwatchConfig &prev, + SwatchConfig &next, + long prev_frame, + long next_frame, + long current_frame); + + int brightness; + int saturation; + int fix_brightness; + int draw_src; + int angle; +}; + + +class SwatchSlider : public BC_ISlider +{ +public: + SwatchSlider(SwatchMain *plugin, + SwatchWindow *gui, + int x, + int y, + int min, + int max, + int *output); + int handle_event(); + SwatchMain *plugin; + int *output; +}; + +class SwatchRadial : public BC_Radial +{ +public: + SwatchRadial(SwatchMain *plugin, + SwatchWindow *gui, + int x, + int y, + const char *text, + int fix_brightness); + int handle_event(); + SwatchMain *plugin; + SwatchWindow *gui; + int fix_brightness; +}; + +class SwatchCheck : public BC_CheckBox +{ +public: + SwatchCheck(SwatchMain *plugin, + int x, + int y, + const char *text, + int *output); + int handle_event(); + SwatchMain *plugin; + SwatchWindow *gui; + int *output; +}; + +class SwatchWindow : public PluginClientWindow +{ +public: + SwatchWindow(SwatchMain *plugin); + ~SwatchWindow(); + + void create_objects(); + void update_fixed(); + + SwatchMain *plugin; + SwatchSlider *brightness; + SwatchSlider *saturation; + SwatchSlider *angle; + SwatchRadial *fix_brightness; + SwatchRadial *fix_saturation; + SwatchCheck *draw_src; + BC_Title *brightness_title; + BC_Title *saturation_title; +}; + + + + + + + +class SwatchMain : public PluginVClient +{ +public: + SwatchMain(PluginServer *server); + ~SwatchMain(); + + int process_buffer(VFrame *frame, + int64_t start_position, + double frame_rate); + int is_realtime(); + void save_data(KeyFrame *keyframe); + void read_data(KeyFrame *keyframe); + void update_gui(); + int is_synthesis(); + int handle_opengl(); + + PLUGIN_CLASS_MEMBERS(SwatchConfig) + + int need_reconfigure; + + VFrame *temp; + VFrame *src_temp; + SwatchEngine *engine; +}; + +class SwatchPackage : public LoadPackage +{ +public: + SwatchPackage(); + int y1; + int y2; +}; + +class SwatchUnit : public LoadClient +{ +public: + SwatchUnit(SwatchEngine *server, SwatchMain *plugin); + void process_package(LoadPackage *package); + SwatchEngine *server; + SwatchMain *plugin; + YUV yuv; +}; + +class SwatchEngine : public LoadServer +{ +public: + SwatchEngine(SwatchMain *plugin, int total_clients, int total_packages); + void init_packages(); + + void draw_pattern(); + void draw_src(); + + LoadClient* new_client(); + LoadPackage* new_package(); + SwatchMain *plugin; + int mode; +}; + + + +#endif -- 2.26.2