tests/examples/spectrum/demo-audiotest.c

   1 /* GStreamer
   2  * Copyright (C) 2006 Stefan Kost <ensonic@users.sf.net>
   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., 59 Temple Place - Suite 330,
  17  * Boston, MA 02111-1307, USA.
  18  */
  19 /* TODO: add wave selection */
  20
  21 #ifdef HAVE_CONFIG_H
  22 #include "config.h"
  23 #endif
  24
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27 #include <string.h>
  28 #include <math.h>
  29 #include <gst/gst.h>
  30 #include <gtk/gtk.h>
  31
  32 #ifndef DEFAULT_AUDIOSINK
  33 #define DEFAULT_AUDIOSINK "autoaudiosink"
  34 #endif
  35
  36 static guint spect_height = 64;
  37 static guint spect_bands = 256;
  38 static gfloat height_scale = 1.0;
  39
  40 static GtkWidget *drawingarea = NULL;
  41 static GstClock *sync_clock = NULL;
  42
  43 static void
  44 on_window_destroy (GtkObject * object, gpointer user_data)
  45 {
  46   drawingarea = NULL;
  47   gtk_main_quit ();
  48 }
  49
  50 /* control audiotestsrc frequency */
  51 static void
  52 on_frequency_changed (GtkRange * range, gpointer user_data)
  53 {
  54   GstElement *machine = GST_ELEMENT (user_data);
  55   gdouble value = gtk_range_get_value (range);
  56
  57   g_object_set (machine, "freq", value, NULL);
  58 }
  59
  60 static gboolean
  61 on_configure_event (GtkWidget * widget, GdkEventConfigure * event,
  62     gpointer user_data)
  63 {
  64   GstElement *spectrum = GST_ELEMENT (user_data);
  65
  66   /*GST_INFO ("%d x %d", event->width, event->height); */
  67   spect_height = event->height;
  68   height_scale = event->height / 64.0;
  69   spect_bands = event->width;
  70
  71   g_object_set (G_OBJECT (spectrum), "bands", spect_bands, NULL);
  72   return FALSE;
  73 }
  74
  75 /* draw frequency spectrum as a bunch of bars */
  76 static void
  77 draw_spectrum (gfloat * data)
  78 {
  79   gint i;
  80   GdkRectangle rect = { 0, 0, spect_bands, spect_height };
  81   cairo_t *cr;
  82
  83   if (!drawingarea)
  84     return;
  85
  86   gdk_window_begin_paint_rect (gtk_widget_get_window (drawingarea), &rect);
  87   cr = gdk_cairo_create (gtk_widget_get_window (drawingarea));
  88
  89   cairo_set_source_rgb (cr, 0, 0, 0);
  90   cairo_rectangle (cr, 0, 0, spect_bands, spect_height);
  91   cairo_fill (cr);
  92   cairo_set_source_rgb (cr, 1, 1, 1);
  93   for (i = 0; i < spect_bands; i++) {
  94     cairo_rectangle (cr, i, -data[i], 1, spect_height + data[i]);
  95     cairo_fill (cr);
  96   }
  97   cairo_destroy (cr);
  98
  99   gdk_window_end_paint (gtk_widget_get_window (drawingarea));
 100 }
 101
 102 /* process delayed message */
 103 static gboolean
 104 delayed_idle_spectrum_update (gpointer user_data)
 105 {
 106   draw_spectrum ((gfloat *) user_data);
 107   g_free (user_data);
 108   return (FALSE);
 109 }
 110
 111 static gboolean
 112 delayed_spectrum_update (GstClock * sync_clock, GstClockTime time,
 113     GstClockID id, gpointer user_data)
 114 {
 115   if (GST_CLOCK_TIME_IS_VALID (time))
 116     g_idle_add (delayed_idle_spectrum_update, user_data);
 117   else
 118     g_free (user_data);
 119   return (TRUE);
 120 }
 121
 122 /* receive spectral data from element message */
 123 static gboolean
 124 message_handler (GstBus * bus, GstMessage * message, gpointer data)
 125 {
 126   if (message->type == GST_MESSAGE_ELEMENT) {
 127     const GstStructure *s = gst_message_get_structure (message);
 128     const gchar *name = gst_structure_get_name (s);
 129
 130     if (strcmp (name, "spectrum") == 0) {
 131       GstElement *spectrum = GST_ELEMENT (GST_MESSAGE_SRC (message));
 132       GstClockTime timestamp, duration;
 133       GstClockTime waittime = GST_CLOCK_TIME_NONE;
 134
 135       if (gst_structure_get_clock_time (s, "running-time", &timestamp) &&
 136           gst_structure_get_clock_time (s, "duration", &duration)) {
 137         /* wait for middle of buffer */
 138         waittime = timestamp + duration / 2;
 139       } else if (gst_structure_get_clock_time (s, "endtime", &timestamp)) {
 140         waittime = timestamp;
 141       }
 142       if (GST_CLOCK_TIME_IS_VALID (waittime)) {
 143         GstClockID clock_id;
 144         GstClockTime basetime = gst_element_get_base_time (spectrum);
 145         gfloat *spect = g_new (gfloat, spect_bands);
 146         const GValue *list;
 147         const GValue *value;
 148         guint i;
 149
 150         list = gst_structure_get_value (s, "magnitude");
 151         for (i = 0; i < spect_bands; ++i) {
 152           value = gst_value_list_get_value (list, i);
 153           spect[i] = height_scale * g_value_get_float (value);
 154         }
 155
 156         clock_id =
 157             gst_clock_new_single_shot_id (sync_clock, waittime + basetime);
 158         gst_clock_id_wait_async (clock_id, delayed_spectrum_update,
 159             (gpointer) spect);
 160         gst_clock_id_unref (clock_id);
 161       }
 162     }
 163   }
 164   return TRUE;
 165 }
 166
 167 int
 168 main (int argc, char *argv[])
 169 {
 170   GstElement *bin;
 171   GstElement *src, *spectrum, *audioconvert, *sink;
 172   GstBus *bus;
 173   GtkWidget *appwindow, *vbox, *widget;
 174
 175   gst_init (&argc, &argv);
 176   gtk_init (&argc, &argv);
 177
 178   bin = gst_pipeline_new ("bin");
 179
 180   src = gst_element_factory_make ("audiotestsrc", "src");
 181   g_object_set (G_OBJECT (src), "wave", 0, NULL);
 182
 183   spectrum = gst_element_factory_make ("spectrum", "spectrum");
 184   g_object_set (G_OBJECT (spectrum), "bands", spect_bands, "threshold", -80,
 185       "message", TRUE, NULL);
 186
 187   audioconvert = gst_element_factory_make ("audioconvert", "audioconvert");
 188
 189   sink = gst_element_factory_make (DEFAULT_AUDIOSINK, "sink");
 190
 191   gst_bin_add_many (GST_BIN (bin), src, spectrum, audioconvert, sink, NULL);
 192   if (!gst_element_link_many (src, spectrum, audioconvert, sink, NULL)) {
 193     fprintf (stderr, "can't link elements\n");
 194     exit (1);
 195   }
 196
 197   bus = gst_element_get_bus (bin);
 198   gst_bus_add_watch (bus, message_handler, NULL);
 199   gst_object_unref (bus);
 200
 201   sync_clock = gst_pipeline_get_clock (GST_PIPELINE (bin));
 202
 203   appwindow = gtk_window_new (GTK_WINDOW_TOPLEVEL);
 204   g_signal_connect (G_OBJECT (appwindow), "destroy",
 205       G_CALLBACK (on_window_destroy), NULL);
 206   vbox = gtk_vbox_new (FALSE, 6);
 207
 208   widget = gtk_hscale_new_with_range (50.0, 20000.0, 10);
 209   gtk_scale_set_draw_value (GTK_SCALE (widget), TRUE);
 210   gtk_scale_set_value_pos (GTK_SCALE (widget), GTK_POS_TOP);
 211   gtk_range_set_value (GTK_RANGE (widget), 440.0);
 212   g_signal_connect (G_OBJECT (widget), "value-changed",
 213       G_CALLBACK (on_frequency_changed), (gpointer) src);
 214   gtk_box_pack_start (GTK_BOX (vbox), widget, FALSE, FALSE, 0);
 215
 216   drawingarea = gtk_drawing_area_new ();
 217   gtk_widget_set_size_request (drawingarea, spect_bands, spect_height);
 218   g_signal_connect (G_OBJECT (drawingarea), "configure-event",
 219       G_CALLBACK (on_configure_event), (gpointer) spectrum);
 220   gtk_box_pack_start (GTK_BOX (vbox), drawingarea, TRUE, TRUE, 0);
 221
 222   gtk_container_add (GTK_CONTAINER (appwindow), vbox);
 223   gtk_widget_show_all (appwindow);
 224
 225   gst_element_set_state (bin, GST_STATE_PLAYING);
 226   gtk_main ();
 227   gst_element_set_state (bin, GST_STATE_NULL);
 228
 229   gst_object_unref (sync_clock);
 230   gst_object_unref (bin);
 231
 232   return 0;
 233 }