+++ /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 "clip.h"
-#include "bccmodels.h"
-#include "filexml.h"
-#include "ivtc.h"
-#include "ivtcwindow.h"
-#include "language.h"
-
-#include <stdio.h>
-#include <string.h>
-
-
-#if 0
-static const char *pattern_text[] =
-{
- N_("A B BC CD D"),
- N_("AB BC CD DE EF"),
- N_("Automatic")
-};
-#endif
-
-REGISTER_PLUGIN(IVTCMain)
-
-IVTCConfig::IVTCConfig()
-{
- frame_offset = 0;
- first_field = 0;
- automatic = 1;
- auto_threshold = 2;
- pattern = IVTCConfig::PULLDOWN32;
-}
-
-IVTCMain::IVTCMain(PluginServer *server)
- : PluginVClient(server)
-{
-
- engine = 0;
- previous_min = 0x4000000000000000LL;
- previous_strategy = 0;
-}
-
-IVTCMain::~IVTCMain()
-{
-
-
- if(engine)
- {
- if(temp_frame[0]) delete temp_frame[0];
- if(temp_frame[1]) delete temp_frame[1];
- temp_frame[0] = 0;
- temp_frame[1] = 0;
- delete engine;
- }
-}
-
-const char* IVTCMain::plugin_title() { return _("Inverse Telecine"); }
-int IVTCMain::is_realtime() { return 1; }
-
-
-
-NEW_WINDOW_MACRO(IVTCMain, IVTCWindow)
-
-
-
-int IVTCMain::load_configuration()
-{
- KeyFrame *prev_keyframe;
-
- prev_keyframe = get_prev_keyframe(get_source_position());
-// Must also switch between interpolation between keyframes and using first keyframe
- read_data(prev_keyframe);
-
- return 0;
-}
-
-void IVTCMain::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("IVTC");
- output.tag.set_property("FRAME_OFFSET", config.frame_offset);
- output.tag.set_property("FIRST_FIELD", config.first_field);
- output.tag.set_property("AUTOMATIC", config.automatic);
- output.tag.set_property("AUTO_THRESHOLD", config.auto_threshold);
- output.tag.set_property("PATTERN", config.pattern);
- output.append_tag();
- output.tag.set_title("/IVTC");
- output.append_tag();
- output.append_newline();
- output.terminate_string();
-}
-
-void IVTCMain::read_data(KeyFrame *keyframe)
-{
- FileXML input;
-
- input.set_shared_input(keyframe->get_data(), strlen(keyframe->get_data()));
-
- int result = 0;
- //float new_threshold;
-
- while(!result)
- {
- result = input.read_tag();
-
- if(!result)
- {
- if(input.tag.title_is("IVTC"))
- {
- config.frame_offset = input.tag.get_property("FRAME_OFFSET", config.frame_offset);
- config.first_field = input.tag.get_property("FIRST_FIELD", config.first_field);
- config.automatic = input.tag.get_property("AUTOMATIC", config.automatic);
- //new_threshold = input.tag.get_property("AUTO_THRESHOLD", config.auto_threshold);
- config.pattern = input.tag.get_property("PATTERN", config.pattern);
- }
- }
- }
-}
-
-
-
-void IVTCMain::render_stop()
-{
- previous_min = 0x4000000000000000LL;
-}
-
-
-
-// Pattern A B BC CD D
-int IVTCMain::process_realtime(VFrame *input_ptr, VFrame *output_ptr)
-{
- load_configuration();
-
- if(!engine)
- {
- temp_frame[0] = 0;
- temp_frame[1] = 0;
-
- engine = new IVTCEngine(this, smp + 1);
- }
-
-// Determine position in pattern
- int pattern_position = (PluginClient::source_position + config.frame_offset) % 5;
-
-//printf("IVTCMain::process_realtime %d %d\n", pattern_position, config.first_field);
- if(!temp_frame[0]) temp_frame[0] = new VFrame(0,
- -1,
- input_ptr->get_w(),
- input_ptr->get_h(),
- input_ptr->get_color_model(),
- -1);
- if(!temp_frame[1]) temp_frame[1] = new VFrame(0,
- -1,
- input_ptr->get_w(),
- input_ptr->get_h(),
- input_ptr->get_color_model(),
- -1);
-
- int row_size = VFrame::calculate_bytes_per_pixel(input_ptr->get_color_model()) * input_ptr->get_w();
- this->input = input_ptr;
- this->output = output_ptr;
-
-// Determine pattern
- if(config.pattern == IVTCConfig::PULLDOWN32)
- {
- switch(pattern_position)
- {
-// Direct copy
- case 0:
- case 4:
- if(input_ptr->get_rows()[0] != output_ptr->get_rows()[0])
- output_ptr->copy_from(input_ptr);
- break;
-
- case 1:
- temp_frame[0]->copy_from(input_ptr);
- if(input_ptr->get_rows()[0] != output_ptr->get_rows()[0])
- output_ptr->copy_from(input_ptr);
- break;
-
- case 2:
-// Save one field for next frame. Reuse previous frame.
- temp_frame[1]->copy_from(input_ptr);
- output_ptr->copy_from(temp_frame[0]);
- break;
-
- case 3:
-// Combine previous field with current field.
- for(int i = 0; i < input_ptr->get_h(); i++)
- {
- unsigned char *dst = output_ptr->get_rows()[i];
- unsigned char *src = ((i + config.first_field) & 1) ?
- input_ptr->get_rows()[i] :
- temp_frame[1]->get_rows()[i];
- if( src != dst )
- memcpy(dst, src, row_size);
- }
- break;
- }
- }
- else
- if(config.pattern == IVTCConfig::SHIFTFIELD)
- {
- temp_frame[1]->copy_from(input_ptr);
-
-// Recycle previous bottom or top
- for(int i = 0; i < input_ptr->get_h(); i++)
- {
- unsigned char *dst = output_ptr->get_rows()[i];
- unsigned char *src = ((i + config.first_field) & 1) ?
- input_ptr->get_rows()[i] :
- temp_frame[0]->get_rows()[i];
- if( src != dst )
- memcpy(dst, src, row_size);
- }
-
-// Swap temp frames
- VFrame *temp = temp_frame[0];
- temp_frame[0] = temp_frame[1];
- temp_frame[1] = temp;
- }
- else
- if(config.pattern == IVTCConfig::AUTOMATIC)
- {
-// Compare averaged rows with original rows and
-// with previous rows.
-// Take rows which are most similar to the averaged rows.
-// Process frame.
- engine->process_packages();
-// Copy current for future use
- temp_frame[1]->copy_from(input_ptr);
-
-// Add results
- even_vs_current = 0;
- even_vs_prev = 0;
- odd_vs_current = 0;
- odd_vs_prev = 0;
-
- for(int i = 0; i < engine->get_total_clients(); i++)
- {
- IVTCUnit *unit = (IVTCUnit*)engine->get_client(i);
- even_vs_current += unit->even_vs_current;
- even_vs_prev += unit->even_vs_prev;
- odd_vs_current += unit->odd_vs_current;
- odd_vs_prev += unit->odd_vs_prev;
- }
-
-
- int64_t min;
- int strategy;
-
-
-// First strategy.
-// Even lines from previous frame are more similar to
-// averaged even lines in current frame.
-// Take even lines from previous frame
- min = even_vs_prev;
- strategy = 0;
-
- if(even_vs_current < min)
- {
-// Even lines from current frame are more similar to averaged
-// even lines in current frame than previous combinations.
-// Take all lines from current frame
- min = even_vs_current;
- strategy = 2;
- }
-
- if(min > odd_vs_prev)
- {
-// Odd lines from previous frame are more similar to averaged
-// odd lines in current frame than previous combinations.
-// Take odd lines from previous frame
- min = odd_vs_prev;
- strategy = 1;
- }
-
- if(min > odd_vs_current)
- {
-// Odd lines from current frame are more similar to averaged
-// odd lines in current frame than previous combinations.
-// Take odd lines from current frame.
- min = odd_vs_current;
- strategy = 2;
- }
-
-// int confident = 1;
-// Do something if not confident.
-// Sometimes we never get the other field.
-// Currently nothing is done because it doesn't fix the timing.
-// if(min > previous_min * 4 && previous_strategy == 2)
-// {
-// confident = 0;
-// strategy = 3;
-// }
-// printf("IVTCMain::process_realtime 1: previous_min=%lld min=%lld strategy=%d confident=%d\n",
-// previous_min,
-// min,
-// strategy,
-// confident);
-
-
-
-// printf("IVTCMain::process_realtime:\n even_vs_current=%lld\n even_vs_prev=%lld\n odd_vs_current=%lld\n odd_vs_prev=%lld\n strategy=%d confident=%d\n",
-// even_vs_current,
-// even_vs_prev,
-// odd_vs_current,
-// odd_vs_prev,
-// strategy,
-// strategy == 2 && !use_direct_copy);
-
- switch(strategy)
- {
- case 0:
- for(int i = 0; i < input_ptr->get_h(); i++)
- {
- unsigned char *dst = output_ptr->get_rows()[i];
- unsigned char *src = (!(i & 1)) ?
- temp_frame[0]->get_rows()[i] :
- input_ptr->get_rows()[i];
- if( src != dst )
- memcpy(dst, src, row_size);
- }
- break;
- case 1:
- for(int i = 0; i < input_ptr->get_h(); i++)
- {
- unsigned char *dst = output_ptr->get_rows()[i];
- unsigned char *src = (i & 1) ?
- temp_frame[0]->get_rows()[i] :
- input_ptr->get_rows()[i];
- if( src != dst )
- memcpy(dst, src, row_size);
- }
- break;
- case 2:
- output_ptr->copy_from(input_ptr);
- break;
- case 3:
-// output_ptr->copy_from(temp_frame[0]);
-// Deinterlace
- for(int i = 0; i < input_ptr->get_h(); i++)
- {
- unsigned char *dst = output_ptr->get_rows()[i];
- unsigned char *src = (i & 1) ?
- input_ptr->get_rows()[i - 1] :
- input_ptr->get_rows()[i];
- if( src != dst )
- memcpy(dst, src, row_size);
- }
- break;
- }
-
- previous_min = min;
- previous_strategy = strategy;
- VFrame *temp = temp_frame[1];
- temp_frame[1] = temp_frame[0];
- temp_frame[0] = temp;
- }
- return 0;
-}
-
-
-
-void IVTCMain::update_gui()
-{
- if(thread)
- {
- load_configuration();
- ((IVTCWindow*)thread->window)->lock_window();
- if(config.pattern == IVTCConfig::AUTOMATIC)
- {
- ((IVTCWindow*)thread->window)->frame_offset->disable();
- ((IVTCWindow*)thread->window)->first_field->disable();
- }
- else
- {
- ((IVTCWindow*)thread->window)->frame_offset->enable();
- ((IVTCWindow*)thread->window)->first_field->enable();
- }
- ((IVTCWindow*)thread->window)->frame_offset->update((int64_t)config.frame_offset);
- ((IVTCWindow*)thread->window)->first_field->update(config.first_field);
-// ((IVTCWindow*)thread->window)->automatic->update(config.automatic);
- for(int i = 0; i < TOTAL_PATTERNS; i++)
- {
- ((IVTCWindow*)thread->window)->pattern[i]->update(config.pattern == i);
- }
- ((IVTCWindow*)thread->window)->unlock_window();
- }
-}
-
-
-
-// labs returns different values on x86_64 causing our accumulators to explode
-#define ABS local_abs
-
-
-#ifdef __x86_64__
-
-static int local_abs(int value)
-{
- return (value < 0 ? -value : value);
-}
-
-static float local_abs(float value)
-{
- return (value < 0 ? -value : value);
-}
-
-#else
-
-static int local_abs(int value)
-{
- return abs(value);
-}
-
-static float local_abs(float value)
-{
- return fabsf(value);
-}
-
-
-#endif
-
-
-
-
-IVTCPackage::IVTCPackage()
- : LoadPackage()
-{
-}
-
-
-
-
-IVTCUnit::IVTCUnit(IVTCEngine *server, IVTCMain *plugin)
- : LoadClient(server)
-{
- this->server = server;
- this->plugin = plugin;
-}
-
-#define IVTC_MACRO(type, temp_type, components, is_yuv) \
-{ \
- type **curr_rows = (type**)plugin->input->get_rows(); \
- type **prev_rows = (type**)plugin->temp_frame[0]->get_rows(); \
- /* int skip = components - 1; // Components to skip for YUV */\
- for(int i = ptr->y1; i < ptr->y2; i++) \
- { /* Rows to average in the input frame */ \
- int input_row1_number = i - 1; \
- int input_row2_number = i + 1; \
- input_row1_number = MAX(0, input_row1_number); \
- input_row2_number = MIN(h - 1, input_row2_number); \
- type *input_row1 = curr_rows[input_row1_number]; \
- type *input_row2 = curr_rows[input_row2_number]; \
-/* Rows to compare the averaged rows to */ \
- type *current_row = curr_rows[i]; \
- type *prev_row = prev_rows[i]; \
- temp_type current_difference = 0; \
- temp_type prev_difference = 0; \
- for(int j = 0; j < w; j++) \
- { \
-/* This only compares luminance */ \
-/* Get average of current rows */ \
- temp_type average = ((temp_type)*input_row1 + *input_row2) / 2; \
-/* Difference between averaged current rows and original inbetween row */ \
- current_difference += ABS(average - *current_row); \
-/* Difference between averaged current rows and previous inbetween row */ \
- prev_difference += ABS(average - *prev_row); \
- \
-/* Do RGB channels */ \
- if(!is_yuv) \
- { \
- average = ((temp_type)input_row1[1] + input_row2[1]) / 2; \
- current_difference += ABS(average - current_row[1]); \
- prev_difference += ABS(average - prev_row[1]); \
- average = ((temp_type)input_row1[2] + input_row2[2]) / 2; \
- current_difference += ABS(average - current_row[2]); \
- prev_difference += ABS(average - prev_row[2]); \
- } \
- \
-/* Add to row accumulators */ \
- current_row += components; \
- prev_row += components; \
- input_row1 += components; \
- input_row2 += components; \
- } \
- \
-/* Store row differences in even or odd variables */ \
- if(sizeof(type) == 4) \
- { \
- if(i % 2) \
- { \
- odd_vs_current += (int64_t)(current_difference * 0xffff); \
- odd_vs_prev += (int64_t)(prev_difference); \
- } \
- else \
- { \
- even_vs_current += (int64_t)(current_difference); \
- even_vs_prev += (int64_t)(prev_difference); \
- } \
- } \
- else \
- { \
- if(i % 2) \
- { \
- odd_vs_current += (int64_t)current_difference; \
- odd_vs_prev += (int64_t)prev_difference; \
- } \
- else \
- { \
- even_vs_current += (int64_t)current_difference; \
- even_vs_prev += (int64_t)prev_difference; \
- } \
- } \
- } \
-}
-
-void IVTCUnit::clear_totals()
-{
- even_vs_current = 0;
- even_vs_prev = 0;
- odd_vs_current = 0;
- odd_vs_prev = 0;
-}
-
-void IVTCUnit::process_package(LoadPackage *package)
-{
- IVTCPackage *ptr = (IVTCPackage*)package;
- int w = plugin->input->get_w();
- int h = plugin->input->get_h();
-
- switch(plugin->input->get_color_model())
- {
- case BC_RGB_FLOAT:
- IVTC_MACRO(float, float, 3, 0);
- break;
- case BC_RGB888:
- IVTC_MACRO(unsigned char, int, 3, 0);
- break;
- case BC_YUV888:
- IVTC_MACRO(unsigned char, int, 3, 1);
- break;
- case BC_RGBA_FLOAT:
- IVTC_MACRO(float, float, 4, 0);
- break;
- case BC_RGBA8888:
- IVTC_MACRO(unsigned char, int, 4, 0);
- break;
- case BC_YUVA8888:
- IVTC_MACRO(unsigned char, int, 4, 1);
- break;
- case BC_RGB161616:
- IVTC_MACRO(uint16_t, int, 3, 0);
- break;
- case BC_YUV161616:
- IVTC_MACRO(uint16_t, int, 3, 1);
- break;
- case BC_RGBA16161616:
- IVTC_MACRO(uint16_t, int, 4, 0);
- break;
- case BC_YUVA16161616:
- IVTC_MACRO(uint16_t, int, 4, 1);
- break;
- }
-
-}
-
-
-
-
-
-IVTCEngine::IVTCEngine(IVTCMain *plugin, int cpus)
- : LoadServer(cpus, cpus)
-{
- this->plugin = plugin;
-}
-
-IVTCEngine::~IVTCEngine()
-{
-}
-
-void IVTCEngine::init_packages()
-{
- int increment = plugin->input->get_h() / get_total_packages();
- increment /= 2;
- increment *= 2;
- if(!increment) increment = 2;
- int y1 = 0;
- for(int i = 0; i < get_total_packages(); i++)
- {
- IVTCPackage *package = (IVTCPackage*)get_package(i);
- package->y1 = y1;
- y1 += increment;
- if(y1 > plugin->input->get_h()) y1 = plugin->input->get_h();
- package->y2 = y1;
- }
- for(int i = 0; i < get_total_clients(); i++)
- {
- IVTCUnit *unit = (IVTCUnit*)get_client(i);
- unit->clear_totals();
- }
-}
-
-LoadClient* IVTCEngine::new_client()
-{
- return new IVTCUnit(this, plugin);
-}
-
-LoadPackage* IVTCEngine::new_package()
-{
- return new IVTCPackage;
-}
-
-
-
-