cinelerra-5.1/plugins/crikey/crikey.h

   1 /*
   2  * CINELERRA
   3  * Copyright (C) 1997-2014 Adam Williams <broadcast at earthling dot net>
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18  *
  19  */
  20
  21
  22
  23 #ifndef __CRIKEY_H__
  24 #define __CRIKEY_H__
  25
  26 #include "loadbalance.h"
  27 #include "pluginvclient.h"
  28
  29 class CriKeyEngine;
  30 class CriKey;
  31
  32 #define DRAW_ALPHA      0
  33 #define DRAW_EDGE       1
  34 #define DRAW_MASK       2
  35 #define DRAW_MODES      3
  36
  37 enum { PT_E, PT_X, PT_Y, PT_T, PT_TAG, PT_SZ }; // enable, x,y,threshold, tag
  38
  39 class CriKeyPoint
  40 {
  41 public:
  42         int tag, e;
  43         float x, y, t;
  44
  45         CriKeyPoint(int tag, int e, float x, float y, float t);
  46         ~CriKeyPoint();
  47         void update_parameter(CriKeyPoint *prev, CriKeyPoint *src);
  48 };
  49 class CriKeyPoints : public ArrayList<CriKeyPoint *>
  50 {
  51 public:
  52         CriKeyPoints() {}
  53         ~CriKeyPoints() { remove_all_objects(); }
  54 };
  55
  56 class CriKeyConfig
  57 {
  58 public:
  59         CriKeyConfig();
  60         ~CriKeyConfig();
  61
  62         void save_data(KeyFrame *keyframe);
  63         void read_data(KeyFrame *keyframe);
  64         int equivalent(CriKeyConfig &that);
  65         void copy_from(CriKeyConfig &that);
  66         void interpolate(CriKeyConfig &prev, CriKeyConfig &next,
  67                 long prev_frame, long next_frame, long current_frame);
  68         void limits();
  69
  70         CriKeyPoints points;
  71         int add_point(int tag, int e, float x, float y, float t);
  72         int add_point();
  73         void del_point(int i);
  74
  75         float threshold;
  76         int draw_mode;
  77         int drag, selected;
  78 };
  79
  80 class CriKeyPackage : public LoadPackage
  81 {
  82 public:
  83         CriKeyPackage() : LoadPackage() {}
  84         int y1, y2;
  85 };
  86
  87 class CriKeyEngine : public LoadServer
  88 {
  89 public:
  90         CriKeyEngine(CriKey *plugin, int total_clients, int total_packages)
  91          : LoadServer(total_clients, total_packages) { this->plugin = plugin; }
  92         ~CriKeyEngine() {}
  93
  94         void init_packages();
  95         LoadPackage* new_package();
  96         LoadClient* new_client();
  97
  98         CriKey *plugin;
  99         int set_color(int x, int y, float t);
 100         float color[3], threshold;
 101 };
 102
 103 class CriKeyUnit : public LoadClient
 104 {
 105 public:
 106         CriKeyUnit(CriKeyEngine *server)
 107          : LoadClient(server) { this->server = server; }
 108         ~CriKeyUnit() {}
 109
 110         float edge_detect(float *data, float max, int do_max);
 111         void process_package(LoadPackage *package);
 112         CriKeyEngine *server;
 113 };
 114
 115
 116 class CriKey : public PluginVClient
 117 {
 118 public:
 119         CriKey(PluginServer *server);
 120         ~CriKey();
 121 // required for all realtime plugins
 122         PLUGIN_CLASS_MEMBERS2(CriKeyConfig)
 123         int is_realtime();
 124         void update_gui();
 125         int new_point();
 126         int set_target(float *color, int x, int y);
 127         void save_data(KeyFrame *keyframe);
 128         void read_data(KeyFrame *keyframe);
 129         int process_buffer(VFrame *frame, int64_t start_position, double frame_rate);
 130         void draw_alpha(VFrame *msk);
 131         void draw_edge(VFrame *frm);
 132         void draw_mask(VFrame *frm);
 133         void draw_point(VFrame *msk, CriKeyPoint *pt);
 134         void span_keyframes(KeyFrame *src, int64_t start, int64_t end);
 135         void update_parameter(CriKeyConfig &prev_config, CriKeyConfig &src_config,
 136                 KeyFrame *keyframe);
 137
 138         CriKeyEngine *engine;
 139         VFrame *src, *edg, *msk;
 140         int w, h, color_model, bpp, comp;
 141         int is_yuv, is_float;
 142 };
 143
 144 #endif