subprojects/gst-plugins-bad/gst/debugutils/gstchecksumsink.c

   1 /* GStreamer
   2  * Copyright (C) 2010 David Schleef <ds@schleef.org>
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Library General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Library General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Library General Public
  15  * License along with this library; if not, write to the
  16  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
  17  * Boston, MA 02110-1301, USA.
  18  */
  19
  20 #ifdef HAVE_CONFIG_H
  21 #include "config.h"
  22 #endif
  23
  24 #include <gst/gst.h>
  25 #include <gst/base/gstbasesink.h>
  26 #include "gstdebugutilsbadelements.h"
  27 #include "gstchecksumsink.h"
  28
  29 static void gst_checksum_sink_set_property (GObject * object, guint prop_id,
  30     const GValue * value, GParamSpec * pspec);
  31 static void gst_checksum_sink_get_property (GObject * object, guint prop_id,
  32     GValue * value, GParamSpec * pspec);
  33 static void gst_checksum_sink_dispose (GObject * object);
  34 static void gst_checksum_sink_finalize (GObject * object);
  35
  36 static gboolean gst_checksum_sink_start (GstBaseSink * sink);
  37 static gboolean gst_checksum_sink_stop (GstBaseSink * sink);
  38 static GstFlowReturn
  39 gst_checksum_sink_render (GstBaseSink * sink, GstBuffer * buffer);
  40
  41 enum
  42 {
  43   PROP_0,
  44   PROP_HASH,
  45 };
  46
  47 static GstStaticPadTemplate gst_checksum_sink_sink_template =
  48 GST_STATIC_PAD_TEMPLATE ("sink",
  49     GST_PAD_SINK,
  50     GST_PAD_ALWAYS,
  51     GST_STATIC_CAPS_ANY);
  52
  53 /* class initialization */
  54
  55 #define GST_TYPE_CHECKSUM_SINK_HASH (gst_checksum_sink_hash_get_type ())
  56 static GType
  57 gst_checksum_sink_hash_get_type (void)
  58 {
  59   static GType gtype = 0;
  60
  61   if (gtype == 0) {
  62     static const GEnumValue values[] = {
  63       {G_CHECKSUM_MD5, "MD5", "md5"},
  64       {G_CHECKSUM_SHA1, "SHA-1", "sha1"},
  65       {G_CHECKSUM_SHA256, "SHA-256", "sha256"},
  66       {G_CHECKSUM_SHA512, "SHA-512", "sha512"},
  67       {0, NULL, NULL},
  68     };
  69
  70     gtype = g_enum_register_static ("GstChecksumSinkHash", values);
  71   }
  72   return gtype;
  73 }
  74
  75 #define gst_checksum_sink_parent_class parent_class
  76 G_DEFINE_TYPE (GstChecksumSink, gst_checksum_sink, GST_TYPE_BASE_SINK);
  77 GST_ELEMENT_REGISTER_DEFINE (checksumsink, "checksumsink",
  78     GST_RANK_NONE, gst_checksum_sink_get_type ());
  79
  80 static void
  81 gst_checksum_sink_class_init (GstChecksumSinkClass * klass)
  82 {
  83   GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
  84   GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
  85   GstBaseSinkClass *base_sink_class = GST_BASE_SINK_CLASS (klass);
  86
  87   gobject_class->set_property = gst_checksum_sink_set_property;
  88   gobject_class->get_property = gst_checksum_sink_get_property;
  89   gobject_class->dispose = gst_checksum_sink_dispose;
  90   gobject_class->finalize = gst_checksum_sink_finalize;
  91   base_sink_class->start = GST_DEBUG_FUNCPTR (gst_checksum_sink_start);
  92   base_sink_class->stop = GST_DEBUG_FUNCPTR (gst_checksum_sink_stop);
  93   base_sink_class->render = GST_DEBUG_FUNCPTR (gst_checksum_sink_render);
  94
  95   gst_element_class_add_static_pad_template (element_class,
  96       &gst_checksum_sink_sink_template);
  97
  98   g_object_class_install_property (gobject_class, PROP_HASH,
  99       g_param_spec_enum ("hash", "Hash", "Checksum type",
 100           gst_checksum_sink_hash_get_type (), G_CHECKSUM_SHA1,
 101           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 102
 103   gst_element_class_set_static_metadata (element_class, "Checksum sink",
 104       "Debug/Sink", "Calculates a checksum for buffers",
 105       "David Schleef <ds@schleef.org>");
 106
 107   gst_type_mark_as_plugin_api (gst_checksum_sink_hash_get_type (), 0);
 108 }
 109
 110 static void
 111 gst_checksum_sink_init (GstChecksumSink * checksumsink)
 112 {
 113   gst_base_sink_set_sync (GST_BASE_SINK (checksumsink), FALSE);
 114   checksumsink->hash = G_CHECKSUM_SHA1;
 115 }
 116
 117 static void
 118 gst_checksum_sink_set_property (GObject * object, guint prop_id,
 119     const GValue * value, GParamSpec * pspec)
 120 {
 121   GstChecksumSink *checksumsink = GST_CHECKSUM_SINK (object);
 122
 123   switch (prop_id) {
 124     case PROP_HASH:
 125       checksumsink->hash = g_value_get_enum (value);
 126       break;
 127     default:
 128       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 129       break;
 130   }
 131 }
 132
 133 static void
 134 gst_checksum_sink_get_property (GObject * object, guint prop_id,
 135     GValue * value, GParamSpec * pspec)
 136 {
 137   GstChecksumSink *checksumsink = GST_CHECKSUM_SINK (object);
 138
 139   switch (prop_id) {
 140     case PROP_HASH:
 141       g_value_set_enum (value, checksumsink->hash);
 142       break;
 143     default:
 144       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 145       break;
 146   }
 147 }
 148
 149 static void
 150 gst_checksum_sink_dispose (GObject * object)
 151 {
 152   G_OBJECT_CLASS (parent_class)->dispose (object);
 153 }
 154
 155 static void
 156 gst_checksum_sink_finalize (GObject * object)
 157 {
 158   G_OBJECT_CLASS (parent_class)->finalize (object);
 159 }
 160
 161 static gboolean
 162 gst_checksum_sink_start (GstBaseSink * sink)
 163 {
 164   return TRUE;
 165 }
 166
 167 static gboolean
 168 gst_checksum_sink_stop (GstBaseSink * sink)
 169 {
 170   return TRUE;
 171 }
 172
 173 static GstFlowReturn
 174 gst_checksum_sink_render (GstBaseSink * sink, GstBuffer * buffer)
 175 {
 176   gchar *s;
 177   GstMapInfo map;
 178   GstChecksumSink *checksumsink;
 179
 180   checksumsink = GST_CHECKSUM_SINK (sink);
 181   gst_buffer_map (buffer, &map, GST_MAP_READ);
 182   s = g_compute_checksum_for_data (checksumsink->hash, map.data, map.size);
 183   gst_buffer_unmap (buffer, &map);
 184   g_print ("%" GST_TIME_FORMAT " %s\n",
 185       GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buffer)), s);
 186
 187   g_free (s);
 188
 189   return GST_FLOW_OK;
 190 }