hvirtual/plugins/ivtc/ivtc.h

   1 #ifndef IVTC_H
   2 #define IVTC_H
   3
   4 class IVTCMain;
   5 class IVTCEngine;
   6
   7 #define TOTAL_PATTERNS 3
   8
   9 static char *pattern_text[] =
  10 {
  11         "A  B  BC  CD  D",
  12         "AB  BC  CD  DE  EF",
  13         "Automatic"
  14 };
  15
  16 #include "defaults.h"
  17 #include "loadbalance.h"
  18 #include "mutex.h"
  19 #include "pluginvclient.h"
  20 #include "ivtcwindow.h"
  21 #include <sys/types.h>
  22
  23 class IVTCEngine;
  24
  25 class IVTCConfig
  26 {
  27 public:
  28         IVTCConfig();
  29         int frame_offset;
  30 // 0 - even     1 - odd
  31         int first_field;
  32         int automatic;
  33         float auto_threshold;
  34         int pattern;
  35         enum
  36         {
  37                 PULLDOWN32,
  38                 SHIFTFIELD,
  39                 AUTOMATIC
  40         };
  41 };
  42
  43 class IVTCMain : public PluginVClient
  44 {
  45 public:
  46         IVTCMain(PluginServer *server);
  47         ~IVTCMain();
  48
  49 // required for all realtime plugins
  50         int process_realtime(VFrame *input_ptr, VFrame *output_ptr);
  51         int is_realtime();
  52         void save_data(KeyFrame *keyframe);
  53         void read_data(KeyFrame *keyframe);
  54         PLUGIN_CLASS_MEMBERS(IVTCConfig, IVTCThread)
  55         void update_gui();
  56
  57         int load_defaults();
  58         int save_defaults();
  59
  60         void compare_fields(VFrame *frame1,
  61                 VFrame *frame2,
  62                 int64_t &field1,
  63                 int64_t &field2);
  64         int64_t compare(VFrame *current_avg,
  65                 VFrame *current_orig,
  66                 VFrame *previous,
  67                 int field);
  68
  69         void deinterlace_avg(VFrame *output, VFrame *input, int dominance);
  70
  71
  72         VFrame *temp_frame[2];
  73         VFrame *input, *output;
  74         int64_t even_vs_current;
  75         int64_t even_vs_prev;
  76         int64_t odd_vs_current;
  77         int64_t odd_vs_prev;
  78         IVTCEngine *engine;
  79 };
  80
  81
  82 class IVTCPackage : public LoadPackage
  83 {
  84 public:
  85         IVTCPackage();
  86         int y1, y2;
  87 };
  88
  89 class IVTCUnit : public LoadClient
  90 {
  91 public:
  92         IVTCUnit(IVTCEngine *server, IVTCMain *plugin);
  93         void process_package(LoadPackage *package);
  94         void clear_totals();
  95         IVTCEngine *server;
  96         IVTCMain *plugin;
  97         int64_t even_vs_current;
  98         int64_t even_vs_prev;
  99         int64_t odd_vs_current;
 100         int64_t odd_vs_prev;
 101 };
 102
 103
 104 class IVTCEngine : public LoadServer
 105 {
 106 public:
 107         IVTCEngine(IVTCMain *plugin, int cpus);
 108         ~IVTCEngine();
 109         void init_packages();
 110         LoadClient* new_client();
 111         LoadPackage* new_package();
 112         IVTCMain *plugin;
 113 };
 114
 115 #endif