cinelerra-5.1/plugins/denoiseseltempavg/seltempavg.h

   1
   2 /*
   3  * CINELERRA
   4  * Copyright (C) 2008 Adam Williams <broadcast at earthling dot net>
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  *
  20  */
  21
  22 #ifndef SELTEMPAVG_H
  23 #define SELTEMPAVG_H
  24
  25 class SelTempAvgMain;
  26
  27 #include "bchash.inc"
  28 #include "pluginvclient.h"
  29 #include "seltempavgwindow.h"
  30 #include "vframe.inc"
  31
  32 class SelTempAvgConfig
  33 {
  34 public:
  35         SelTempAvgConfig();
  36         void copy_from(SelTempAvgConfig *src);
  37         int equivalent(SelTempAvgConfig *src);
  38
  39         int frames;
  40
  41         float avg_threshold_RY, avg_threshold_GU, avg_threshold_BV;
  42         float std_threshold_RY, std_threshold_GU, std_threshold_BV;
  43         int mask_RY, mask_GU, mask_BV;
  44
  45         int method;
  46         enum
  47         {
  48                 METHOD_NONE,
  49                 METHOD_SELTEMPAVG,
  50                 METHOD_AVERAGE,
  51                 METHOD_STDDEV
  52         };
  53
  54         int offsetmode;
  55         enum
  56         {
  57                 OFFSETMODE_FIXED,
  58                 OFFSETMODE_RESTARTMARKERSYS
  59         };
  60
  61
  62         int paranoid;
  63         int nosubtract;
  64         int offset_restartmarker_keyframe;
  65         int offset_fixed_value;
  66         float gain;
  67
  68 };
  69
  70
  71 class SelTempAvgMain : public PluginVClient
  72 {
  73 public:
  74         SelTempAvgMain(PluginServer *server);
  75         ~SelTempAvgMain();
  76
  77 // required for all realtime plugins
  78         PLUGIN_CLASS_MEMBERS(SelTempAvgConfig)
  79         void update_gui();
  80         int process_buffer(VFrame *frame, int64_t start_position, double frame_rate);
  81         int is_realtime();
  82         void save_data(KeyFrame *keyframe);
  83         void read_data(KeyFrame *keyframe);
  84         int nextkeyframeisoffsetrestart(KeyFrame *keyframe);
  85         void clear_accum(int w, int h, int color_model);
  86         void subtract_accum(VFrame *frame);
  87         void add_accum(VFrame *frame);
  88         void transfer_accum(VFrame *frame);
  89
  90         int64_t restartoffset;
  91         int onakeyframe;
  92
  93         char string[BCTEXTLEN];
  94
  95         VFrame **history;
  96
  97 // Frame of history in requested framerate
  98         int64_t *history_frame;
  99         int *history_valid;
 100         unsigned char *accumulation;
 101         unsigned char *accumulation_sq;
 102         unsigned char *accumulation_grey;
 103
 104         int history_size;
 105 // Starting frame of history in requested framerate
 106         int64_t history_start;
 107 // When subtraction is disabled, this detects no change for paranoid mode.
 108         int64_t prev_frame;
 109 };
 110
 111
 112 #endif