+++ /dev/null
-
-/*
- * CINELERRA
- * Copyright (C) 2008 Adam Williams <broadcast at earthling dot net>
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
- *
- */
-
-#include "bcdisplayinfo.h"
-#include "clip.h"
-#include "bchash.h"
-#include "bcsignals.h"
-#include "filexml.h"
-#include "keyframe.h"
-#include "language.h"
-#include "unsharp.h"
-#include "unsharpwindow.h"
-
-
-#include <errno.h>
-#include <unistd.h>
-
-REGISTER_PLUGIN(UnsharpMain)
-
-
-
-UnsharpConfig::UnsharpConfig()
-{
- radius = 5;
- amount = 0.5;
- threshold = 0;
-}
-
-int UnsharpConfig::equivalent(UnsharpConfig &that)
-{
- return EQUIV(radius, that.radius) &&
- EQUIV(amount, that.amount) &&
- threshold == that.threshold;
-}
-
-void UnsharpConfig::copy_from(UnsharpConfig &that)
-{
- radius = that.radius;
- amount = that.amount;
- threshold = that.threshold;
-}
-
-void UnsharpConfig::interpolate(UnsharpConfig &prev,
- UnsharpConfig &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->radius = prev.radius * prev_scale + next.radius * next_scale;
- this->amount = prev.amount * prev_scale + next.amount * next_scale;
- this->threshold = (int)(prev.threshold * prev_scale + next.threshold * next_scale);
-}
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-UnsharpMain::UnsharpMain(PluginServer *server)
- : PluginVClient(server)
-{
-
- engine = 0;
-}
-
-UnsharpMain::~UnsharpMain()
-{
-
- delete engine;
-}
-
-const char* UnsharpMain::plugin_title() { return _("Unsharp"); }
-int UnsharpMain::is_realtime() { return 1; }
-
-
-NEW_WINDOW_MACRO(UnsharpMain, UnsharpWindow)
-
-LOAD_CONFIGURATION_MACRO(UnsharpMain, UnsharpConfig)
-
-
-
-void UnsharpMain::update_gui()
-{
- if(thread)
- {
- if(load_configuration())
- {
- thread->window->lock_window("UnsharpMain::update_gui");
- ((UnsharpWindow*)thread->window)->update();
- thread->window->unlock_window();
- }
- }
-}
-
-
-
-
-void UnsharpMain::save_data(KeyFrame *keyframe)
-{
- FileXML output;
-
-// cause data to be stored directly in text
- output.set_shared_output(keyframe->get_data(), MESSAGESIZE);
- output.tag.set_title("UNSHARP");
-
- output.tag.set_property("RADIUS", config.radius);
- output.tag.set_property("AMOUNT", config.amount);
- output.tag.set_property("THRESHOLD", config.threshold);
- output.append_tag();
- output.tag.set_title("/UNSHARP");
- output.append_tag();
- output.append_newline();
- output.terminate_string();
-}
-
-void UnsharpMain::read_data(KeyFrame *keyframe)
-{
- FileXML input;
-
- input.set_shared_input(keyframe->get_data(), strlen(keyframe->get_data()));
-
- int result = 0;
-
- while(!result)
- {
- result = input.read_tag();
-
- if(!result)
- {
- if(input.tag.title_is("UNSHARP"))
- {
- config.radius = input.tag.get_property("RADIUS", config.radius);
- config.amount = input.tag.get_property("AMOUNT", config.amount);
- config.threshold = input.tag.get_property("THRESHOLD", config.threshold);
- }
- }
- }
-}
-
-
-
-
-
-
-
-int UnsharpMain::process_buffer(VFrame *frame,
- int64_t start_position,
- double frame_rate)
-{
- /*int need_reconfigure =*/ load_configuration();
-
- if(!engine) engine = new UnsharpEngine(this,
- get_project_smp() + 1, get_project_smp() + 1);
- read_frame(frame, 0, get_source_position(), get_framerate(), 0);
- engine->do_unsharp(frame);
- return 0;
-}
-
-
-
-
-
-
-
-
-
-
-
-
-UnsharpPackage::UnsharpPackage()
- : LoadPackage()
-{
-}
-
-
-
-
-
-
-UnsharpUnit::UnsharpUnit(UnsharpEngine *server,
- UnsharpMain *plugin)
- : LoadClient(server)
-{
- this->plugin = plugin;
- this->server = server;
- temp = 0;
-}
-
-UnsharpUnit::~UnsharpUnit()
-{
- delete temp;
-}
-
-
-// Derived from the Gimp.
-// In the original file it says
-//
-// Copyright (C) 1999 Winston Chang
-// <winstonc@cs.wisc.edu>
-// <winston@stdout.org>
-//
-// Adapted for Cinelerra by Heroine Virtual Ltd.
-
-static int calculate_convolution_matrix(double radius, double **cmatrix)
-{
- radius = fabs(radius) + 1.0;
- double std_dev = radius;
- radius = std_dev * 2;
- int matrix_length = (int)(2 * ceil(radius - 0.5) + 1);
- matrix_length = MAX(1, matrix_length);
-// int matrix_midpoint = matrix_length / 2 + 1;
- (*cmatrix) = new double[matrix_length];
-
-// Top right of matrix
- for(int i = matrix_length / 2 + 1; i < matrix_length; i++)
- {
- double base_x = i - floor(matrix_length / 2) - 0.5;
- double sum = 0;
- for(int j = 1; j <= 50; j++)
- {
- if(base_x + 0.02 * j <= radius)
- {
- sum += exp(-(base_x + 0.02 * j) *
- (base_x + 0.02 * j) /
- (2 * std_dev * std_dev));
- }
- }
- (*cmatrix)[i] = sum / 50;
- }
-
-// Top left of matrix
- for(int i = 0; i < matrix_length / 2; i++)
- {
- (*cmatrix)[i] = (*cmatrix)[matrix_length - 1 - i];
- }
-
-// Center value
- double sum = 0;
- for(int j = 0; j <= 50; j++)
- {
- sum += exp(-(0.5 + 0.02 * j) *
- (0.5 + 0.02 * j) /
- (2 * std_dev * std_dev));
- }
- (*cmatrix)[matrix_length / 2] = sum / 51;
-
-// Normalize
- sum = 0;
- for(int i = 0; i < matrix_length; i++)
- sum += (*cmatrix)[i];
- for(int i = 0; i < matrix_length; i++)
- (*cmatrix)[i] = (*cmatrix)[i] / sum;
-
- return matrix_length;
-}
-
-static double get_convolution(double *cmatrix,
- float input,
- int index)
-{
- return cmatrix[index] * input;
-}
-
-static void blur_pixels(double *cmatrix,
- int cmatrix_length,
- float *input,
- float *output,
- int pixels,
- int components)
-{
- if(cmatrix_length > pixels)
- {
- for(int pixel = 0; pixel < pixels; pixel++)
- {
- double scale = 0;
- for(int j = 0; j < pixels; j++)
- {
- if((j + cmatrix_length / 2 - pixel >= 0) &&
- (j + cmatrix_length / 2 - pixel < cmatrix_length))
- {
- scale += cmatrix[j + cmatrix_length / 2 - pixel];
- }
- }
-
- for(int i = 0; i < components; i++)
- {
- double sum = 0;
- for(int j = 0; j < pixels; j++)
- {
- if((j >= pixel - cmatrix_length / 2) &&
- (j <= pixel + cmatrix_length / 2))
- {
- sum += input[j * components + i] * cmatrix[i];
- }
- }
- output[pixel * components + i] = sum / scale;
- }
- }
- }
- else
- {
- int cmatrix_middle = cmatrix_length / 2;
- int pixel;
- for(pixel = 0; pixel < cmatrix_middle; pixel++)
- {
- double scale = 0;
- for(int j = cmatrix_middle - pixel; j < cmatrix_length;j++)
- {
- scale += cmatrix[j];
- }
-
- for(int i = 0; i < components; i++)
- {
- double sum = 0;
- for(int j = cmatrix_middle - pixel; j < cmatrix_length; j++)
- {
- sum += input[(pixel + j - cmatrix_middle) * components + i] *
- cmatrix[j];
- }
- output[pixel * components + i] = sum / scale;
- }
- }
-
- float *output_ptr = output + pixel * components;
- for( ; pixel < pixels - cmatrix_middle; pixel++)
- {
- float *input_ptr = input + (pixel - cmatrix_middle) * components;
- for(int i = 0; i < components; i++)
- {
- double sum = 0;
- float *input_ptr2 = input_ptr;
- for(int j = cmatrix_length; j > 0; j--)
- {
- sum += get_convolution(cmatrix,
- *input_ptr2,
- cmatrix_length - j);
- input_ptr2 += components;
- }
- input_ptr++;
- *output_ptr++ = sum;
- }
- }
-
- for( ; pixel < pixels; pixel++)
- {
- double scale = 0;
- for(int j = 0; j < pixels - pixel + cmatrix_middle; j++)
- {
- scale += cmatrix[j];
- }
-
- for(int i = 0; i < components; i++)
- {
- double sum = 0;
- for(int j = 0; j < pixels - pixel + cmatrix_middle; j++)
- {
- sum += input[(pixel + j - cmatrix_middle) * components + i] *
- cmatrix[j];
- }
- output[pixel * components + i] = sum / scale;
- }
- }
- }
-}
-
-#define GET_ROW(type, components) \
-{ \
- type *in_row = (type*)src->get_rows()[row]; \
- int pixels = src->get_w() * components; \
- for(int i = 0; i < pixels; i++) \
- { \
- dst[i] = in_row[i]; \
- } \
-}
-
-static void get_row(float *dst, VFrame *src, int row)
-{
- switch(src->get_color_model())
- {
- case BC_RGB888:
- case BC_YUV888:
- GET_ROW(unsigned char, 3);
- break;
- case BC_RGB_FLOAT:
- GET_ROW(float, 3);
- break;
- case BC_RGBA8888:
- case BC_YUVA8888:
- GET_ROW(unsigned char, 4);
- break;
- case BC_RGBA_FLOAT:
- GET_ROW(float, 4);
- break;
- case BC_YUV161616:
- GET_ROW(uint16_t, 3);
- break;
- case BC_YUVA16161616:
- GET_ROW(uint16_t, 4);
- break;
- }
-}
-
-static void get_column(float *dst, VFrame *src, int column)
-{
- int components = BC_CModels::components(src->get_color_model());
- for(int i = 0; i < src->get_h(); i++)
- {
- float *input_pixel = (float*)src->get_rows()[i] + column * components;
- memcpy(dst, input_pixel, sizeof(float) * components);
- dst += components;
- }
-}
-
-static void put_column(float *src, VFrame *dst, int column)
-{
- int components = BC_CModels::components(dst->get_color_model());
- for(int i = 0; i < dst->get_h(); i++)
- {
- float *output_pixel = (float*)dst->get_rows()[i] + column * components;
- memcpy(output_pixel, src, sizeof(float) * components);
- src += components;
- }
-}
-
-void UnsharpUnit::process_package(LoadPackage *package)
-{
- UnsharpPackage *pkg = (UnsharpPackage*)package;
-// int w = server->src->get_w();
-// int h = server->src->get_h();
- int color_model = server->src->get_color_model();
- int components = BC_CModels::components(color_model);
- double *cmatrix = 0;
- int cmatrix_length = 0;
- int padded_y1 = pkg->y1;
- int padded_y2 = pkg->y2;
-
- cmatrix_length = calculate_convolution_matrix(
- plugin->config.radius,
- &cmatrix);
-
-
- if(padded_y2 < server->src->get_h())
- {
- padded_y2 += cmatrix_length / 2;
- padded_y2 = MIN(server->src->get_h(), padded_y2);
- }
- if(padded_y1 > 0)
- {
- padded_y1 -= cmatrix_length / 2;
- padded_y1 = MAX(0, padded_y1);
- }
-
- int padded_rows = padded_y2 - padded_y1;
-
- if(!temp || temp->get_h() != padded_rows)
- {
- delete temp;
- temp = 0;
- }
-
- if(!temp)
- {
- temp = new VFrame;
- temp->set_use_shm(0);
- temp->reallocate(0,
- -1,
- 0,
- 0,
- 0,
- server->src->get_w(),
- padded_rows,
- components == 3 ? BC_RGB_FLOAT : BC_RGBA_FLOAT,
- -1);
- }
-
- float *temp_in = new float[MAX(temp->get_w(), padded_rows) * components];
- float *temp_out = new float[MAX(temp->get_w(), padded_rows) * components];
-
-// Blur rows
- for(int i = padded_y1; i < padded_y2; i++)
- {
- get_row(temp_in, server->src, i);
- blur_pixels(cmatrix,
- cmatrix_length,
- temp_in,
- temp_out,
- temp->get_w(),
- components);
-// printf("UnsharpUnit::process_package %d %p %p %p %d %d\n",
-// __LINE__,
-// temp,
-// temp->get_rows()[0],
-// temp_out,
-// i - padded_y1,
-// temp->get_bytes_per_line());
- memcpy(temp->get_rows()[i - padded_y1],
- temp_out,
- temp->get_bytes_per_line());
- }
-
-//Now we're 100% floating point. Blur the columns
- for(int i = 0; i < temp->get_w(); i++)
- {
- get_column(temp_in, temp, i);
- blur_pixels(cmatrix,
- cmatrix_length,
- temp_in,
- temp_out,
- padded_rows,
- components);
- put_column(temp_out, temp, i);
- }
-
-
-//printf("%f %f %d\n", plugin->config.radius,plugin->config.amount, plugin->config.threshold);
-
-
-#define UNSHARPEN(type, components, max) \
-{ \
- float threshold = (float)plugin->config.threshold * max / 0xff; \
- float amount = plugin->config.amount; \
- \
- for(int i = pkg->y1; i < pkg->y2; i++) \
- { \
- float *blurry_row = (float*)temp->get_rows()[i - padded_y1]; \
- type *orig_row = (type*)server->src->get_rows()[i]; \
- for(int j = 0; j < server->src->get_w(); j++) \
- { \
- for(int k = 0; k < components; k++) \
- { \
- float diff = *orig_row - *blurry_row; \
- if(fabsf(2 * diff) < threshold) \
- diff = 0; \
- float value = *orig_row + amount * diff; \
- if(sizeof(type) == 4) \
- *orig_row = (type)value; \
- else \
- *orig_row = (type)CLIP(value, 0, max); \
- blurry_row++; \
- orig_row++; \
- } \
- } \
- } \
-}
-
-// Apply unsharpening
- switch(color_model)
- {
- case BC_RGB888:
- case BC_YUV888:
- UNSHARPEN(unsigned char, 3, 0xff);
- break;
- case BC_RGBA8888:
- case BC_YUVA8888:
- UNSHARPEN(unsigned char, 4, 0xff);
- break;
- case BC_RGB_FLOAT:
- UNSHARPEN(float, 3, 1.0);
- break;
- case BC_RGBA_FLOAT:
- UNSHARPEN(float, 4, 1.0);
- break;
- case BC_YUV161616:
- UNSHARPEN(uint16_t, 3, 0xffff);
- break;
- case BC_YUVA16161616:
- UNSHARPEN(uint16_t, 4, 0xffff);
- break;
- }
-
- delete [] temp_in;
- delete [] temp_out;
- delete [] cmatrix;
-}
-
-
-
-
-
-
-
-
-
-UnsharpEngine::UnsharpEngine(UnsharpMain *plugin,
- int total_clients,
- int total_packages)
- : LoadServer(
-//1, 1
-total_clients, total_packages
-)
-{
- this->plugin = plugin;
-}
-
-UnsharpEngine::~UnsharpEngine()
-{
-}
-
-
-void UnsharpEngine::init_packages()
-{
- for(int i = 0; i < get_total_packages(); i++)
- {
- UnsharpPackage *pkg = (UnsharpPackage*)get_package(i);
- pkg->y1 = src->get_h() * i / get_total_packages();
- pkg->y2 = src->get_h() * (i + 1) / get_total_packages();
- }
-}
-
-LoadClient* UnsharpEngine::new_client()
-{
- return new UnsharpUnit(this, plugin);
-}
-
-LoadPackage* UnsharpEngine::new_package()
-{
- return new UnsharpPackage;
-}
-
-
-void UnsharpEngine::do_unsharp(VFrame *src)
-{
- this->src = src;
-
- process_packages();
-}
-
-
-
-
-
-
-
-
-