1 files changed, 85 insertions, 25 deletions

diff --git a/gst/spectrum/demo-audiotest.c b/gst/spectrum/demo-audiotest.c
index 11ebf9df..e69aef34 100644
--- a/gst/spectrum/demo-audiotest.c
+++ b/gst/spectrum/demo-audiotest.c
@@ -1,15 +1,45 @@
+/* GStreamer
+ * Copyright (C) 2006 Stefan Kost <ensonic@users.sf.net>
+ *
+ * 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.
+ */
+
 #ifdef HAVE_CONFIG_H
 #include "config.h"
 #endif
+
+#include <string.h>
+#include <math.h>
 #include <gst/gst.h>
 #include <gtk/gtk.h>
 
+#define DEFAULT_AUDIOSINK "alsasink"
 #define SPECT_BANDS 256
 
 static GtkWidget *drawingarea = NULL;
-static gint ct = 0;
 
 static void
+on_window_destroy (GtkObject * object, gpointer user_data)
+{
+  drawingarea = NULL;
+  gtk_main_quit ();
+}
+
+/* control audiotestsrc frequency */
+static void
 on_frequency_changed (GtkRange * range, gpointer user_data)
 {
   GstElement *machine = GST_ELEMENT (user_data);
@@ -18,26 +48,51 @@ on_frequency_changed (GtkRange * range, gpointer user_data)
   g_object_set (machine, "freq", value, NULL);
 }
 
+/* draw frequency spectrum as a bunch of bars */
 static void
-spectrum_chain (GstElement * sink, GstBuffer * buf, GstPad * pad,
-    gpointer unused)
+draw_spectrum (guchar * data)
 {
-  ct = (ct + 1) & 0x15;
-  if (!ct) {
-    gint i;
-    guchar *data = buf->data;
-    gint width = GST_BUFFER_SIZE (buf);
-    GdkRectangle rect = { 0, 0, width, 50 };
-
-    gdk_window_begin_paint_rect (drawingarea->window, &rect);
-    gdk_draw_rectangle (drawingarea->window, drawingarea->style->black_gc,
-        TRUE, 0, 0, width, 50);
-    for (i = 0; i < width; i++) {
-      gdk_draw_rectangle (drawingarea->window, drawingarea->style->white_gc,
-          TRUE, i, 64 - data[i], 1, data[i]);
+  gint i;
+  GdkRectangle rect = { 0, 0, SPECT_BANDS, 50 };
+
+  if (!drawingarea)
+    return;
+
+  gdk_window_begin_paint_rect (drawingarea->window, &rect);
+  gdk_draw_rectangle (drawingarea->window, drawingarea->style->black_gc,
+      TRUE, 0, 0, SPECT_BANDS, 50);
+  for (i = 0; i < SPECT_BANDS; i++) {
+    gdk_draw_rectangle (drawingarea->window, drawingarea->style->white_gc,
+        TRUE, i, 64 - data[i], 1, data[i]);
+  }
+  gdk_window_end_paint (drawingarea->window);
+}
+
+/* receive spectral data from element message */
+gboolean
+message_handler (GstBus * bus, GstMessage * message, gpointer data)
+{
+  if (message->type == GST_MESSAGE_ELEMENT) {
+    const GstStructure *s = gst_message_get_structure (message);
+    const gchar *name = gst_structure_get_name (s);
+
+    if (strcmp (name, "spectrum") == 0) {
+      guchar spect[SPECT_BANDS];
+      const GValue *list;
+      const GValue *value;
+      guint i;
+
+      list = gst_structure_get_value (s, "spectrum");
+      for (i = 0; i < SPECT_BANDS; ++i) {
+        value = gst_value_list_get_value (list, i);
+        spect[i] = g_value_get_uchar (value);
+      }
+      draw_spectrum (spect);
     }
-    gdk_window_end_paint (drawingarea->window);
   }
+  /* we handled the message we want, and ignored the ones we didn't want.
+   * so the core can unref the message for us */
+  return TRUE;
 }
 
 int
@@ -45,7 +100,7 @@ main (int argc, char *argv[])
 {
   GstElement *bin;
   GstElement *src, *spectrum, *sink;
-
+  GstBus *bus;
   GtkWidget *appwindow, *vbox, *widget;
 
   gst_init (&argc, &argv);
@@ -54,16 +109,15 @@ main (int argc, char *argv[])
   bin = gst_pipeline_new ("bin");
 
   src = gst_element_factory_make ("audiotestsrc", "src");
-  g_object_set (G_OBJECT (src), "blocksize", (gulong) 1024 * 2, NULL);
+  /*
+     g_object_set (G_OBJECT (src), "wave", 0, NULL);
+   */
 
   spectrum = gst_element_factory_make ("spectrum", "spectrum");
-  g_object_set (G_OBJECT (spectrum), "width", SPECT_BANDS, "threshold", -80,
-      NULL);
-
-  sink = gst_element_factory_make ("fakesink", "sink");
-  g_object_set (G_OBJECT (sink), "signal-handoffs", TRUE, NULL);
+  g_object_set (G_OBJECT (spectrum), "bands", SPECT_BANDS, "threshold", -80,
+      "message", TRUE, NULL);
 
-  g_signal_connect (sink, "handoff", G_CALLBACK (spectrum_chain), NULL);
+  sink = gst_element_factory_make (DEFAULT_AUDIOSINK, "sink");
 
   gst_bin_add_many (GST_BIN (bin), src, spectrum, sink, NULL);
   if (!gst_element_link_many (src, spectrum, sink, NULL)) {
@@ -71,7 +125,13 @@ main (int argc, char *argv[])
     exit (1);
   }
 
+  bus = gst_element_get_bus (bin);
+  gst_bus_add_watch (bus, message_handler, NULL);
+  gst_object_unref (bus);
+
   appwindow = gtk_window_new (GTK_WINDOW_TOPLEVEL);
+  g_signal_connect (G_OBJECT (appwindow), "destroy",
+      G_CALLBACK (on_window_destroy), NULL);
   vbox = gtk_vbox_new (FALSE, 6);
 
   widget = gtk_hscale_new_with_range (50.0, 20000.0, 10);