+++ /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 "bccmodels.h"
-#include "filexml.h"
-#include "language.h"
-#include "sharpen.h"
-#include "sharpenwindow.h"
-
-#include <stdio.h>
-#include <string.h>
-
-
-PluginClient* new_plugin(PluginServer *server)
-{
- return new SharpenMain(server);
-}
-
-SharpenMain::SharpenMain(PluginServer *server)
- : PluginVClient(server)
-{
- sharpness = 0;
- thread = 0;
-}
-
-SharpenMain::~SharpenMain()
-{
- if(thread)
- {
-// Set result to 0 to indicate a server side close
- thread->window->set_done(0);
- thread->completion.lock();
- delete thread;
- }
-
-}
-
-char* SharpenMain::plugin_title() { return _("Quark"); }
-int SharpenMain::is_realtime() { return 1; }
-
-int SharpenMain::start_realtime()
-{
-// Initialize
- last_sharpness = sharpness;
-
- total_engines = smp > 1 ? 2 : 1;
- engine = new SharpenEngine*[total_engines];
- for(int i = 0; i < total_engines; i++)
- {
- engine[i] = new SharpenEngine(this);
- engine[i]->start();
- }
- return 0;
-}
-
-int SharpenMain::stop_realtime()
-{
- for(int i = 0; i < total_engines; i++)
- {
- delete engine[i];
- }
- delete engine;
- return 0;
-}
-
-int SharpenMain::process_realtime(VFrame *input_ptr, VFrame *output_ptr)
-{
- int i, j, k;
-
- load_configuration();
-
- get_luts(pos_lut, neg_lut, input_ptr->get_color_model());
-
- if(sharpness != 0)
- {
-// Arm first row
- row_step = (interlace || horizontal) ? 2 : 1;
-
- for(j = 0; j < row_step; j += total_engines)
- {
- for(k = 0; k < total_engines && k + j < row_step; k++)
- {
- engine[k]->start_process_frame(input_ptr, input_ptr, k + j);
- }
- for(k = 0; k < total_engines && k + j < row_step; k++)
- {
- engine[k]->wait_process_frame();
- }
- }
- }
- else
- if(input_ptr->get_rows()[0] != output_ptr->get_rows()[0])
- {
- output_ptr->copy_from(input_ptr);
- }
- return 0;
-}
-
-int SharpenMain::show_gui()
-{
- load_configuration();
- thread = new SharpenThread(this);
- thread->start();
- return 0;
-}
-
-int SharpenMain::set_string()
-{
- if(thread) thread->window->set_title(gui_string);
- return 0;
-}
-
-void SharpenMain::raise_window()
-{
- if(thread)
- {
- thread->window->raise_window();
- thread->window->flush();
- }
-}
-
-
-void SharpenMain::load_configuration()
-{
- KeyFrame *prev_keyframe, *next_keyframe;
-//printf("BlurMain::load_configuration 1\n");
-
- prev_keyframe = get_prev_keyframe(-1);
- next_keyframe = get_next_keyframe(-1);
-// Must also switch between interpolation between keyframes and using first keyframe
-//printf("BlurMain::load_configuration %s\n", prev_keyframe->get_data());
- read_data(prev_keyframe);
-}
-
-
-int SharpenMain::get_luts(int *pos_lut, int *neg_lut, int color_model)
-{
- int i, inv_sharpness, vmax;
-
- vmax = cmodel_calculate_max(color_model);
-
- inv_sharpness = (int)(100 - sharpness);
- if(horizontal) inv_sharpness /= 2;
- if(inv_sharpness < 1) inv_sharpness = 1;
-
- for(i = 0; i < vmax + 1; i++)
- {
- pos_lut[i] = 800 * i / inv_sharpness;
- neg_lut[i] = (4 + pos_lut[i] - (i << 3)) >> 3;
- }
-
- return 0;
-}
-
-void SharpenMain::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("SHARPNESS");
- output.tag.set_property("VALUE", sharpness);
- output.append_tag();
-
- if(interlace)
- {
- output.tag.set_title("INTERLACE");
- output.append_tag();
- }
-
- if(horizontal)
- {
- output.tag.set_title("HORIZONTAL");
- output.append_tag();
- }
-
- if(luminance)
- {
- output.tag.set_title("LUMINANCE");
- output.append_tag();
- }
- output.terminate_string();
-}
-
-void SharpenMain::read_data(KeyFrame *keyframe)
-{
- FileXML input;
-
- input.set_shared_input(keyframe->get_data(), strlen(keyframe->get_data()));
-
- int result = 0;
- int new_interlace = 0;
- int new_horizontal = 0;
- int new_luminance = 0;
-
- while(!result)
- {
- result = input.read_tag();
-
- if(!result)
- {
- if(input.tag.title_is("SHARPNESS"))
- {
- sharpness = input.tag.get_property("VALUE", sharpness);
- last_sharpness = sharpness;
- }
- else
- if(input.tag.title_is("INTERLACE"))
- {
- new_interlace = 1;
- }
- else
- if(input.tag.title_is("HORIZONTAL"))
- {
- new_horizontal = 1;
- }
- else
- if(input.tag.title_is("LUMINANCE"))
- {
- new_luminance = 1;
- }
- }
- }
-
- interlace = new_interlace;
- horizontal = new_horizontal;
- luminance = new_luminance;
-
- if(sharpness > MAXSHARPNESS)
- sharpness = MAXSHARPNESS;
- else
- if(sharpness < 0) sharpness = 0;
-
- if(thread)
- {
- thread->window->sharpen_slider->update((int)sharpness);
- thread->window->sharpen_interlace->update(interlace);
- thread->window->sharpen_horizontal->update(horizontal);
- thread->window->sharpen_luminance->update(luminance);
- }
-}
-
-
-
-
-SharpenEngine::SharpenEngine(SharpenMain *plugin)
- : Thread(1, 0, 0)
-{
- this->plugin = plugin;
- last_frame = 0;
- for(int i = 0; i < 4; i++)
- {
- neg_rows[i] = new int[plugin->project_frame_w * 4];
- }
- input_lock.lock();
- output_lock.lock();
- set_synchronous(1);
-}
-
-SharpenEngine::~SharpenEngine()
-{
- last_frame = 1;
- input_lock.unlock();
- Thread::join();
-
- for(int i = 0; i < 4; i++)
- {
- delete [] neg_rows[i];
- }
-}
-
-int SharpenEngine::start_process_frame(VFrame *output, VFrame *input, int field)
-{
- this->output = output;
- this->input = input;
- this->field = field;
- input_lock.unlock();
- return 0;
-}
-
-int SharpenEngine::wait_process_frame()
-{
- output_lock.lock();
- return 0;
-}
-
-#define FILTER(components, vmax, wordsize) \
-{ \
- int *pos_lut = plugin->pos_lut; \
- \
-/* Skip first pixel in row */ \
- memcpy(dst, src, components * wordsize); \
- dst += components; \
- src += components; \
- \
- w -= 2; \
- \
- while(w > 0) \
- { \
- long pixel; \
- pixel = pos_lut[src[0]] - \
- neg0[-components] - \
- neg0[0] - \
- neg0[components] - \
- neg1[-components] - \
- neg1[components] - \
- neg2[-components] - \
- neg2[0] - \
- neg2[components]; \
- pixel = (pixel + 4) >> 3; \
- if(pixel < 0) dst[0] = 0; \
- else \
- if(pixel > vmax) dst[0] = vmax; \
- else \
- dst[0] = pixel; \
- \
- pixel = pos_lut[src[1]] - \
- neg0[-components + 1] - \
- neg0[1] - \
- neg0[components + 1] - \
- neg1[-components + 1] - \
- neg1[components + 1] - \
- neg2[-components + 1] - \
- neg2[1] - \
- neg2[components + 1]; \
- pixel = (pixel + 4) >> 3; \
- if(pixel < 0) dst[1] = 0; \
- else \
- if(pixel > vmax) dst[1] = vmax; \
- else \
- dst[1] = pixel; \
- \
- pixel = pos_lut[src[2]] - \
- neg0[-components + 2] - \
- neg0[2] - \
- neg0[components + 2] - \
- neg1[-components + 2] - \
- neg1[components + 2] - \
- neg2[-components + 2] - \
- neg2[2] - \
- neg2[components + 2]; \
- pixel = (pixel + 4) >> 3; \
- if(pixel < 0) dst[2] = 0; \
- else \
- if(pixel > vmax) dst[2] = vmax; \
- else \
- dst[2] = pixel; \
- \
- if(components == 4) \
- dst[3] = src[3]; \
- \
- src += components; \
- dst += components; \
- \
- neg0 += components; \
- neg1 += components; \
- neg2 += components; \
- w--; \
- } \
- \
-/* Skip last pixel in row */ \
- memcpy(dst, src, components * wordsize); \
-}
-
-void SharpenEngine::filter(int components,
- int wordsize,
- int vmax,
- int w,
- u_int16_t *src,
- u_int16_t *dst,
- int *neg0,
- int *neg1,
- int *neg2)
-{
- FILTER(components, vmax, wordsize);
-}
-
-void SharpenEngine::filter(int components,
- int wordsize,
- int vmax,
- int w,
- unsigned char *src,
- unsigned char *dst,
- int *neg0,
- int *neg1,
- int *neg2)
-{
- FILTER(components, vmax, wordsize);
-}
-
-
-
-
-
-
-
-#define SHARPEN(components, wordsize, wordtype, vmax) \
-{ \
- int count, row; \
- unsigned char **input_rows, **output_rows; \
- \
- input_rows = input->get_rows(); \
- output_rows = output->get_rows(); \
- src_rows[0] = input_rows[field]; \
- src_rows[1] = input_rows[field]; \
- src_rows[2] = input_rows[field]; \
- src_rows[3] = input_rows[field]; \
- \
- for(int j = 0; j < plugin->project_frame_w; j++) \
- { \
- for(int k = 0; k < components; k++) \
- neg_rows[0][j * components + k] = plugin->neg_lut[((wordtype*)src_rows[0])[j * components + k]]; \
- } \
- \
- row = 1; \
- count = 1; \
- \
- for(int i = field; i < plugin->project_frame_h; i += plugin->row_step) \
- { \
- if((i + plugin->row_step) < plugin->project_frame_h) \
- { \
- if(count >= 3) count--; \
-/* Arm next row */ \
- src_rows[row] = input_rows[i + plugin->row_step]; \
- for(int k = 0; k < plugin->project_frame_w; k++) \
- { \
- for(int j = 0; j < components; j++) \
- neg_rows[row][k * components + j] = plugin->neg_lut[((wordtype*)src_rows[row])[k * components + j]]; \
- } \
- \
- count++; \
- row = (row + 1) & 3; \
- } \
- else \
- { \
- count--; \
- } \
- \
- dst_row = output_rows[i]; \
- if(count == 3) \
- { \
-/* Do the filter */ \
- if(plugin->horizontal) \
- filter(components, \
- wordsize, \
- vmax, \
- plugin->project_frame_w, \
- (wordtype*)src_rows[(row + 2) & 3], \
- (wordtype*)dst_row, \
- neg_rows[(row + 2) & 3] + components, \
- neg_rows[(row + 2) & 3] + components, \
- neg_rows[(row + 2) & 3] + components); \
- else \
- filter(components, \
- wordsize, \
- vmax, \
- plugin->project_frame_w, \
- (wordtype*)src_rows[(row + 2) & 3], \
- (wordtype*)dst_row, \
- neg_rows[(row + 1) & 3] + components, \
- neg_rows[(row + 2) & 3] + components, \
- neg_rows[(row + 3) & 3] + components); \
- } \
- else \
- if(count == 2) \
- { \
- if(i == 0) \
- memcpy(dst_row, src_rows[0], plugin->project_frame_w * components * wordsize); \
- else \
- memcpy(dst_row, src_rows[2], plugin->project_frame_w * components * wordsize); \
- } \
- } \
-}
-
-void SharpenEngine::sharpen_888()
-{
- SHARPEN(3, 1, unsigned char, 0xff);
-}
-
-void SharpenEngine::sharpen_8888()
-{
- SHARPEN(4, 1, unsigned char, 0xff);
-}
-
-void SharpenEngine::sharpen_161616()
-{
- SHARPEN(3, 2, u_int16_t, 0xffff);
-}
-
-void SharpenEngine::sharpen_16161616()
-{
- SHARPEN(4, 2, u_int16_t, 0xffff);
-}
-
-
-void SharpenEngine::run()
-{
- while(1)
- {
- input_lock.lock();
- if(last_frame)
- {
- output_lock.unlock();
- return;
- }
-
-
- switch(input->get_color_model())
- {
- case BC_RGB888:
- case BC_YUV888:
- sharpen_888();
- break;
-
- case BC_RGBA8888:
- case BC_YUVA8888:
- sharpen_8888();
- break;
-
- case BC_RGB161616:
- case BC_YUV161616:
- sharpen_161616();
- break;
-
- case BC_RGBA16161616:
- case BC_YUVA16161616:
- sharpen_16161616();
- break;
- }
-
- output_lock.unlock();
- }
-}
-