diff options
Diffstat (limited to 'ext/ofa/gstofa.c')
| -rw-r--r-- | ext/ofa/gstofa.c | 250 |
1 files changed, 250 insertions, 0 deletions
diff --git a/ext/ofa/gstofa.c b/ext/ofa/gstofa.c new file mode 100644 index 000000000..cf86048df --- /dev/null +++ b/ext/ofa/gstofa.c @@ -0,0 +1,250 @@ +/* GStreamer + * + * gstofa.c + * + * Copyright (C) 2006 M. Derezynski + * Copyright (C) 2008 Eric Buehl + * Copyright (C) 2008 Sebastian Dröge <slomo@circular-chaos.org> + * + * 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., 51 Franklin St, Fifth Floor, + * Boston, MA 02110-1301 USA. + */ + +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif + +#include <gst/gst.h> +#include <ofa1/ofa.h> +#include "gstofa.h" + +#define PAD_CAPS \ + "audio/x-raw-int, " \ + "rate = (int) [ 1, MAX ], " \ + "channels = (int) [ 1, 2 ], " \ + "endianness = (int) { LITTLE_ENDIAN, BIG_ENDIAN }, " \ + "width = (int) { 16 }, " \ + "depth = (int) { 16 }, " \ + "signed = (boolean) true" + +GST_DEBUG_CATEGORY_STATIC (gst_ofa_debug); +#define GST_CAT_DEFAULT gst_ofa_debug + +enum +{ + PROP_0, + PROP_FINGERPRINT, +}; + + +#define _do_init(bla) \ + GST_DEBUG_CATEGORY_INIT (gst_ofa_debug, "ofa", 0, "ofa element"); + +GST_BOILERPLATE_FULL (GstOFA, gst_ofa, GstAudioFilter, + GST_TYPE_AUDIO_FILTER, _do_init); + +static void gst_ofa_finalize (GObject * object); +static void gst_ofa_get_property (GObject * object, guint prop_id, + GValue * value, GParamSpec * pspec); +static GstFlowReturn gst_ofa_transform_ip (GstBaseTransform * trans, + GstBuffer * buf); +static gboolean gst_ofa_event (GstBaseTransform * trans, GstEvent * event); + +static void +gst_ofa_base_init (gpointer g_class) +{ + GstElementClass *gstelement_class = GST_ELEMENT_CLASS (g_class); + GstAudioFilterClass *audio_filter_class = (GstAudioFilterClass *) g_class; + GstCaps *caps; + + gst_element_class_set_details_simple (gstelement_class, "OFA", + "MusicIP Fingerprinting element", + "Find a music fingerprint using MusicIP's libofa", + "Milosz Derezynski <internalerror@gmail.com>, Eric Buehl <eric.buehl@gmail.com>"); + + caps = gst_caps_from_string (PAD_CAPS); + gst_audio_filter_class_add_pad_templates (audio_filter_class, caps); + gst_caps_unref (caps); +} + +static void +gst_ofa_finalize (GObject * object) +{ + GstOFA *ofa = GST_OFA (object); + + if (ofa->adapter) { + g_object_unref (ofa->adapter); + ofa->adapter = NULL; + } + + g_free (ofa->fingerprint); + ofa->fingerprint = NULL; + + G_OBJECT_CLASS (parent_class)->finalize (object); +} + +static void +gst_ofa_class_init (GstOFAClass * klass) +{ + GObjectClass *gobject_class = G_OBJECT_CLASS (klass); + GstBaseTransformClass *gstbasetrans_class = GST_BASE_TRANSFORM_CLASS (klass); + + gobject_class->get_property = GST_DEBUG_FUNCPTR (gst_ofa_get_property); + + g_object_class_install_property (gobject_class, PROP_FINGERPRINT, + g_param_spec_string ("fingerprint", "Resulting fingerprint", + "Resulting fingerprint", NULL, G_PARAM_READABLE)); + + gobject_class->finalize = GST_DEBUG_FUNCPTR (gst_ofa_finalize); + + gstbasetrans_class->transform_ip = GST_DEBUG_FUNCPTR (gst_ofa_transform_ip); + gstbasetrans_class->event = GST_DEBUG_FUNCPTR (gst_ofa_event); + gstbasetrans_class->passthrough_on_same_caps = TRUE; +} + +static void +create_fingerprint (GstOFA * ofa) +{ + GstBuffer *buf; + gint rate = GST_AUDIO_FILTER (ofa)->format.rate; + gint channels = GST_AUDIO_FILTER (ofa)->format.channels; + gint width = GST_AUDIO_FILTER (ofa)->format.width / 8; + gint endianness = + (GST_AUDIO_FILTER (ofa)->format. + bigend) ? OFA_BIG_ENDIAN : OFA_LITTLE_ENDIAN; + GstTagList *tags; + + buf = + gst_adapter_take_buffer (ofa->adapter, + gst_adapter_available (ofa->adapter)); + + ofa->fingerprint = g_strdup (ofa_create_print (GST_BUFFER_DATA (buf), + endianness, + GST_BUFFER_SIZE (buf) / width, rate, (channels == 2) ? 1 : 0)); + + gst_buffer_unref (buf); + + tags = gst_tag_list_new (); + gst_tag_list_add (tags, GST_TAG_MERGE_REPLACE, + GST_TAG_OFA_FINGERPRINT, ofa->fingerprint, NULL); + gst_element_found_tags (GST_ELEMENT (ofa), tags); + + ofa->record = FALSE; +} + +static gboolean +gst_ofa_event (GstBaseTransform * trans, GstEvent * event) +{ + GstOFA *ofa = GST_OFA (trans); + + switch (GST_EVENT_TYPE (event)) { + case GST_EVENT_FLUSH_STOP: + case GST_EVENT_NEWSEGMENT: + gst_adapter_clear (ofa->adapter); + ofa->record = TRUE; + g_free (ofa->fingerprint); + ofa->fingerprint = NULL; + break; + case GST_EVENT_EOS: + /* we got to the end of the stream but never generated a fingerprint + * (probably under 135 seconds) + */ + if (!ofa->fingerprint) + create_fingerprint (ofa); + break; + default: + break; + } + + return TRUE; +} + +static void +gst_ofa_init (GstOFA * ofa, GstOFAClass * g_class) +{ + gst_base_transform_set_passthrough (GST_BASE_TRANSFORM (ofa), TRUE); + + ofa->fingerprint = NULL; + ofa->record = TRUE; + + ofa->adapter = gst_adapter_new (); +} + +static GstFlowReturn +gst_ofa_transform_ip (GstBaseTransform * trans, GstBuffer * buf) +{ + GstOFA *ofa = GST_OFA (trans); + guint64 nframes; + GstClockTime duration; + gint rate = GST_AUDIO_FILTER (ofa)->format.rate; + gint channels = GST_AUDIO_FILTER (ofa)->format.channels; + gint width = GST_AUDIO_FILTER (ofa)->format.width / 8; + + g_return_val_if_fail (rate > 0 && channels > 0 + && width > 0, GST_FLOW_NOT_NEGOTIATED); + + if (ofa->record) + gst_adapter_push (ofa->adapter, gst_buffer_copy (buf)); + + nframes = + gst_util_uint64_scale (gst_adapter_available (ofa->adapter), 1, + channels * width); + duration = + GST_FRAMES_TO_CLOCK_TIME (gst_adapter_available (ofa->adapter), rate); + + if (duration >= 135 * GST_SECOND && ofa->fingerprint == NULL) + create_fingerprint (ofa); + + return GST_FLOW_OK; +} + +static void +gst_ofa_get_property (GObject * object, guint prop_id, GValue * value, + GParamSpec * pspec) +{ + GstOFA *ofa = GST_OFA (object); + + switch (prop_id) { + case PROP_FINGERPRINT: + g_value_set_string (value, ofa->fingerprint); + break; + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + + +static gboolean +plugin_init (GstPlugin * plugin) +{ + gboolean ret; + + ret = gst_element_register (plugin, "ofa", GST_RANK_NONE, GST_TYPE_OFA); + + if (ret) { + /* TODO: get this into core */ + gst_tag_register (GST_TAG_OFA_FINGERPRINT, GST_TAG_FLAG_META, + G_TYPE_STRING, "ofa fingerprint", "OFA fingerprint", NULL); + } + + return ret; +} + +GST_PLUGIN_DEFINE (GST_VERSION_MAJOR, + GST_VERSION_MINOR, + "ofa", + "Calculate MusicIP fingerprint from audio files", + plugin_init, VERSION, "LGPL", GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN) |