gstreamer/gst/gl/gstgleffects.c

/*
 * GStreamer
 * Copyright (C) 2008 Filippo Argiolas <filippo.argiolas@gmail.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.
 */

#include <gstgleffects.h>

#define GST_TYPE_GL_EFFECTS            (gst_gl_effects_get_type())
#define GST_GL_EFFECTS(obj)            (G_TYPE_CHECK_INSTANCE_CAST((obj), GST_TYPE_GL_EFFECTS,GstGLEffects))
#define GST_IS_GL_EFFECTS(obj)         (G_TYPE_CHECK_INSTANCE_TYPE((obj), GST_TYPE_GL_EFFECTS))
#define GST_GL_EFFECTS_CLASS(klass)    (G_TYPE_CHECK_CLASS_CAST((klass) , GST_TYPE_GL_EFFECTS,GstGLEffectsClass))
#define GST_IS_GL_EFFECTS_CLASS(klass) (G_TYPE_CHECK_CLASS_TYPE((klass) , GST_TYPE_GL_EFFECTS))
#define GST_GL_EFFECTS_GET_CLASS(obj)  (G_TYPE_INSTANCE_GET_CLASS((obj) , GST_TYPE_GL_EFFECTS,GstGLEffectsClass))

#define GST_CAT_DEFAULT gst_gl_effects_debug
GST_DEBUG_CATEGORY_STATIC (GST_CAT_DEFAULT);

#define DEBUG_INIT(bla)							\
  GST_DEBUG_CATEGORY_INIT (gst_gl_effects_debug, "gleffects", 0, "gleffects element");

GST_BOILERPLATE_FULL (GstGLEffects, gst_gl_effects, GstGLFilter,
		      GST_TYPE_GL_FILTER, DEBUG_INIT);

static void gst_gl_effects_set_property (GObject * object, guint prop_id,
					 const GValue * value, GParamSpec * pspec);
static void gst_gl_effects_get_property (GObject * object, guint prop_id,
					 GValue * value, GParamSpec * pspec);

static void gst_gl_effects_init_resources (GstGLFilter* filter);
static void gst_gl_effects_reset_resources (GstGLFilter* filter);

static gboolean gst_gl_effects_filter (GstGLFilter * filter,
				       GstGLBuffer * inbuf, GstGLBuffer * outbuf);

static const GstElementDetails element_details = GST_ELEMENT_DETAILS (
  "Gstreamer OpenGL Effects",
  "Filter/Effect",
  "GL Shading Language effects",
  "Filippo Argiolas <filippo.argiolas@gmail.com>");

/* dont' forget to edit the following when a new effect is added */
typedef enum {
  GST_GL_EFFECT_IDENTITY,
  GST_GL_EFFECT_MIRROR,
  GST_GL_EFFECT_SQUEEZE,
  GST_GL_EFFECT_STRETCH,
  GST_GL_EFFECT_HEAT,
  GST_GL_EFFECT_SEPIA,
  GST_GL_EFFECT_GLOW,
  GST_GL_N_EFFECTS
} GstGLEffectsEffect;

#define GST_TYPE_GL_EFFECTS_EFFECT (gst_gl_effects_effect_get_type ())
static GType
gst_gl_effects_effect_get_type (void)
{
  static GType gl_effects_effect_type = 0;
  static const GEnumValue effect_types [] = {
    { GST_GL_EFFECT_IDENTITY, "Do nothing Effect", "identity" },
    { GST_GL_EFFECT_MIRROR, "Mirror Effect", "mirror" },
    { GST_GL_EFFECT_SQUEEZE, "Squeeze Effect", "squeeze" },
    { GST_GL_EFFECT_STRETCH, "Stretch Effect", "stretch" },
    { GST_GL_EFFECT_HEAT, "Heat Signature Effect", "heat" },
    { GST_GL_EFFECT_SEPIA, "Sepia Toning Effect", "sepia" },
    { GST_GL_EFFECT_GLOW, "Glow Lighting Effect", "glow" },
    { 0, NULL, NULL }
  };

  if (!gl_effects_effect_type) {
    gl_effects_effect_type =
      g_enum_register_static ("GstGLEffectsEffect", effect_types);
  }
  return gl_effects_effect_type;
}

static void
gst_gl_effects_set_effect (GstGLEffects *effects, gint effect_type) {

  switch (effect_type) {
  case GST_GL_EFFECT_IDENTITY:
    effects->effect = (GstGLEffectProcessFunc) gst_gl_effects_identity;
    break;
  case GST_GL_EFFECT_MIRROR:
    effects->effect = (GstGLEffectProcessFunc) gst_gl_effects_mirror;
    break;
  case GST_GL_EFFECT_SQUEEZE:
    effects->effect = (GstGLEffectProcessFunc) gst_gl_effects_squeeze;
    break;
  case GST_GL_EFFECT_STRETCH:
    effects->effect = (GstGLEffectProcessFunc) gst_gl_effects_stretch;
    break;
  case GST_GL_EFFECT_HEAT:
    effects->effect = (GstGLEffectProcessFunc) gst_gl_effects_heat;
    break;
  case GST_GL_EFFECT_SEPIA:
    effects->effect = (GstGLEffectProcessFunc) gst_gl_effects_sepia;
    break;
  case GST_GL_EFFECT_GLOW:
    effects->effect = (GstGLEffectProcessFunc) gst_gl_effects_glow;
    break;
  default:
    g_assert_not_reached ();
  }
  effects->current_effect = effect_type;
}

/* init resources that need a gl context */
static void
gst_gl_effects_init_gl_resources (GstGLFilter *filter)
{
  GstGLEffects *effects = GST_GL_EFFECTS (filter);
  gint i;

  for (i=0; i<NEEDED_TEXTURES; i++) {
    glGenTextures (1, &effects->midtexture[i]);
    glBindTexture(GL_TEXTURE_RECTANGLE_ARB, effects->midtexture[i]);
    glTexImage2D(GL_TEXTURE_RECTANGLE_ARB, 0, GL_RGBA8,
		 filter->width, filter->height, 0, GL_RGBA, GL_UNSIGNED_BYTE, NULL);
    glTexParameteri (GL_TEXTURE_RECTANGLE_ARB, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
    glTexParameteri (GL_TEXTURE_RECTANGLE_ARB, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
    glTexParameteri (GL_TEXTURE_RECTANGLE_ARB, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
    glTexParameteri (GL_TEXTURE_RECTANGLE_ARB, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
  }
}

/* free resources that need a gl context */
static void
gst_gl_effects_reset_gl_resources (GstGLFilter *filter)
{
  GstGLEffects *effects = GST_GL_EFFECTS (filter);
  gint i;

  for (i=0; i<10; i++) {
    glDeleteTextures (1, &effects->midtexture[i]);
    effects->midtexture[i] = 0;
  }
  for (i=0; i<GST_GL_EFFECTS_N_CURVES; i++) {
    glDeleteTextures (1, &effects->curve[i]);
    effects->curve[i] = 0;
  }
}

static void
gst_gl_effects_base_init (gpointer klass)
{
  GstElementClass *element_class = GST_ELEMENT_CLASS (klass);

  gst_element_class_set_details (element_class, &element_details);
}

static void
gst_gl_effects_class_init (GstGLEffectsClass * klass)
{
  GObjectClass *gobject_class;

  gobject_class = (GObjectClass *) klass;
  gobject_class->set_property = gst_gl_effects_set_property;
  gobject_class->get_property = gst_gl_effects_get_property;

  GST_GL_FILTER_CLASS (klass)->filter = gst_gl_effects_filter;
  GST_GL_FILTER_CLASS (klass)->display_init_cb = gst_gl_effects_init_gl_resources;
  GST_GL_FILTER_CLASS (klass)->display_reset_cb = gst_gl_effects_reset_gl_resources;
  GST_GL_FILTER_CLASS (klass)->onInitFBO = gst_gl_effects_init_resources;
  GST_GL_FILTER_CLASS (klass)->onReset = gst_gl_effects_reset_resources;

  g_object_class_install_property (
    gobject_class,
    PROP_EFFECT,
    g_param_spec_enum ("effect",
		       "Effect",
		       "Select which effect apply to GL video texture",
		       GST_TYPE_GL_EFFECTS_EFFECT,
		       GST_GL_EFFECT_IDENTITY,
		       G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
}

void
gst_gl_effects_draw_texture (GstGLEffects * effects, GLuint tex)
{
  GstGLFilter *filter = GST_GL_FILTER (effects);

  glActiveTexture (GL_TEXTURE0);
  glEnable (GL_TEXTURE_RECTANGLE_ARB);
  glBindTexture (GL_TEXTURE_RECTANGLE_ARB, tex);

  glBegin (GL_QUADS);

  glTexCoord2f (0.0, 0.0);
  glVertex2f (-1.0, -1.0);
  glTexCoord2f (filter->width, 0.0);
  glVertex2f (1.0, -1.0);
  glTexCoord2f (filter->width, filter->height);
  glVertex2f (1.0, 1.0);
  glTexCoord2f (0.0, filter->height);
  glVertex2f (-1.0, 1.0);

  glEnd ();
}

/* static void */
/* set_orizonthal_switch (GstGLDisplay *display, gpointer data) */
/* { */
/* //  GstGLEffects *effects = GST_GL_EFFECTS (data); */

/*   const double mirrormatrix[16] = { */
/*     -1.0, 0.0, 0.0, 0.0, */
/*     0.0, 1.0, 0.0, 0.0, */
/*     0.0, 0.0, 1.0, 0.0, */
/*     0.0, 0.0, 0.0, 1.0 */
/*   }; */

/*   glMatrixMode (GL_MODELVIEW); */
/*   glLoadMatrixd (mirrormatrix); */
/* } */

static void
gst_gl_effects_init (GstGLEffects * effects, GstGLEffectsClass * klass)
{
  gint i;
  effects->shaderstable = g_hash_table_new_full (g_str_hash,
						 g_str_equal,
						 NULL,
						 g_object_unref);
  for (i=0; i<GST_GL_EFFECTS_N_CURVES; i++) {
    effects->curve[i] = 0;
  }

  effects->effect = gst_gl_effects_identity;
}

static void
gst_gl_effects_reset_resources (GstGLFilter* filter)
{
  GstGLEffects* effects = GST_GL_EFFECTS(filter);

  g_hash_table_unref (effects->shaderstable);
  effects->shaderstable = NULL;
}

static void
gst_gl_effects_set_property (GObject * object, guint prop_id,
    const GValue * value, GParamSpec * pspec)
{
  GstGLEffects *effects = GST_GL_EFFECTS (object); 

  switch (prop_id) {
  case PROP_EFFECT:
    gst_gl_effects_set_effect (effects, g_value_get_enum (value));
    break;
  default:
    G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
    break;
  }
}

static void
gst_gl_effects_get_property (GObject * object, guint prop_id,
    GValue * value, GParamSpec * pspec)
{
  GstGLEffects *effects = GST_GL_EFFECTS (object);

  switch (prop_id) {
  case PROP_EFFECT:
    g_value_set_enum (value, effects->current_effect);
    break;
  default:
    G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
    break;
  }
}

static void
gst_gl_effects_init_resources (GstGLFilter* filter)
{
//  GstGLEffects* blur_filter = GST_GL_EFFECTS (filter);
}

static gboolean
gst_gl_effects_filter (GstGLFilter* filter, GstGLBuffer* inbuf,
				GstGLBuffer* outbuf)
{
  GstGLEffects* effects = GST_GL_EFFECTS(filter);

  effects->intexture = inbuf->texture;
  effects->outtexture = outbuf->texture;

  effects->effect (effects);

  return TRUE;
}