gsttrack.h

/* GStreamer
 * Copyright (C) 2014 Henry Kroll <nospam@thenerdshow.com>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library 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
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */

#ifndef _GST_TRACK_H_
#define _GST_TRACK_H_

#include <gst/videofilters/gstvideofilter2.h>
#include <gst/video/video.h>

enum
{
  PROP_0,
  PROP_MESSAGE,
  PROP_MARK_METHOD,
  PROP_SIZE,
  PROP_BGCOLOR,
  PROP_FGCOLOR0,
  PROP_FGCOLOR1,
  PROP_MCOLOR,
  PROP_THRESHOLD,
  PROP_MAX_OBJECTS,
};

typedef enum {
  GST_TRACK_MARK_METHOD_NOTHING,
  GST_TRACK_MARK_METHOD_CROSSHAIRS,
  GST_TRACK_MARK_METHOD_BOX,
  GST_TRACK_MARK_METHOD_BOTH,
  GST_TRACK_MARK_METHOD_CLOAK,
  GST_TRACK_MARK_METHOD_BLUR,
  GST_TRACK_MARK_METHOD_BLUR8,
  GST_TRACK_MARK_METHOD_DECIMATE,
  GST_TRACK_MARK_METHOD_EDGE,
  GST_TRACK_MARK_METHOD_OUTLINE,
  GST_TRACK_MARK_METHOD_COLORIZE,
} GstTrackMarkMethod;

#define RED   0xff0000
#define GREEN 0x00ff00
#define BLUE  0x0000ff
#define WHITE 0x0ffffff
#define DEFAULT_MESSAGE TRUE
#define DEFAULT_THRESHOLD 88
#define DEFAULT_SIZE 20
#define DEFAULT_MAX_OBJECTS 1
#define DEFAULT_COLOR 0xFF0000
#define MAX_OBJECTS 1024
#define DEFAULT_MARK_METHOD GST_TRACK_MARK_METHOD_BOTH

G_BEGIN_DECLS

#define GST_TYPE_TRACK   (gst_track_get_type())
#define GST_TRACK(obj)   (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_TRACK,GstTrack))
#define GST_TRACK_CLASS(klass)   (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_TRACK,GstTrackClass))
#define GST_IS_TRACK(obj)   (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_TRACK))
#define GST_IS_TRACK_CLASS(obj)   (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_TRACK))

typedef struct _GstTrack
{
  GstVideoFilter2 video_filter2;

  /* properties */
  gboolean message;             /* whether to post messages */
  guint size;                   /* minimum detection size */
  guint color0;                /* object color to track */
  guint color1;               /* object highlight or text */
  guint color2;               /* object spot or outline */
  guint mcolor;                 /* marker color */
  guint threshold;              /* color tracking threshold */
  guint max_objects;            /* number of objects to track */
  guint mark_method;            /* mark method */

  /* state */
  guint *rect;                  /* bounding box of tracked object */
  guint8 bgyuv[3];              /* background color YUV */
  guint8 fgyuv0[3];
  guint8 fgyuv1[3];
  guint8 mcyuv[3];
  guint obj_found[MAX_OBJECTS][6]; /* array of found objects */
  guint obj_count;
} GstTrack;

typedef struct _GstTrackClass
{
  GstVideoFilter2Class video_filter2_class;
} GstTrackClass;

GType gst_track_get_type (void);

G_END_DECLS

#endif