tests/examples/equalizer/demo.c

   1 #include <stdlib.h>
   2 #include <string.h>
   3 #include <math.h>
   4 #include <gst/gst.h>
   5 #include <gtk/gtk.h>
   6
   7 #define NBANDS 10
   8
   9 static GtkWidget *drawingarea = NULL;
  10 static guint spect_height = 128;
  11 static guint spect_bands = 256;
  12 static gfloat height_scale = 2.0;
  13
  14 static void
  15 on_window_destroy (GObject * object, gpointer user_data)
  16 {
  17   drawingarea = NULL;
  18   gtk_main_quit ();
  19 }
  20
  21 static gboolean
  22 on_configure_event (GtkWidget * widget, GdkEventConfigure * event,
  23     gpointer user_data)
  24 {
  25   GstElement *spectrum = GST_ELEMENT (user_data);
  26
  27   /*GST_INFO ("%d x %d", event->width, event->height); */
  28   spect_height = event->height;
  29   height_scale = event->height / 64.0;
  30   spect_bands = event->width;
  31
  32   g_object_set (G_OBJECT (spectrum), "bands", spect_bands, NULL);
  33   return FALSE;
  34 }
  35
  36 /* control gains */
  37 static void
  38 on_gain_changed (GtkRange * range, gpointer user_data)
  39 {
  40   GstObject *band = GST_OBJECT (user_data);
  41   gdouble value = gtk_range_get_value (range);
  42
  43   g_object_set (band, "gain", value, NULL);
  44 }
  45
  46 /* control bandwidths */
  47 static void
  48 on_bandwidth_changed (GtkRange * range, gpointer user_data)
  49 {
  50   GstObject *band = GST_OBJECT (user_data);
  51   gdouble value = gtk_range_get_value (range);
  52
  53   g_object_set (band, "bandwidth", value, NULL);
  54 }
  55
  56 /* control frequency */
  57 static void
  58 on_freq_changed (GtkRange * range, gpointer user_data)
  59 {
  60   GstObject *band = GST_OBJECT (user_data);
  61   gdouble value = gtk_range_get_value (range);
  62
  63   /* hbox */
  64   GtkWidget *parent = gtk_widget_get_parent (GTK_WIDGET (range));
  65
  66   /* frame */
  67   GtkWidget *parent_parent = gtk_widget_get_parent (parent);
  68   gchar *label = g_strdup_printf ("%d Hz", (int) (value + 0.5));
  69
  70   gtk_frame_set_label (GTK_FRAME (parent_parent), label);
  71   g_free (label);
  72
  73   g_object_set (band, "freq", value, NULL);
  74 }
  75
  76 /* draw frequency spectrum as a bunch of bars */
  77 static void
  78 draw_spectrum (gfloat * data)
  79 {
  80   gint i;
  81   GdkRectangle rect = { 0, 0, spect_bands, spect_height };
  82   cairo_t *cr;
  83
  84   if (!drawingarea)
  85     return;
  86
  87   gdk_window_begin_paint_rect (gtk_widget_get_window (drawingarea), &rect);
  88   cr = gdk_cairo_create (gtk_widget_get_window (drawingarea));
  89
  90   cairo_set_source_rgb (cr, 0, 0, 0);
  91   cairo_rectangle (cr, 0, 0, spect_bands, spect_height);
  92   cairo_fill (cr);
  93   for (i = 0; i < spect_bands; i++) {
  94     cairo_set_source_rgb (cr, 1, 1, 1);
  95     cairo_rectangle (cr, i, -data[i], 1, spect_height + data[i]);
  96     cairo_fill (cr);
  97   }
  98   cairo_destroy (cr);
  99
 100   gdk_window_end_paint (gtk_widget_get_window (drawingarea));
 101 }
 102
 103 /* receive spectral data from element message */
 104 static gboolean
 105 message_handler (GstBus * bus, GstMessage * message, gpointer data)
 106 {
 107   if (message->type == GST_MESSAGE_ELEMENT) {
 108     const GstStructure *s = gst_message_get_structure (message);
 109     const gchar *name = gst_structure_get_name (s);
 110
 111     if (strcmp (name, "spectrum") == 0) {
 112       gfloat *spect = g_new (gfloat, spect_bands);
 113       const GValue *list;
 114       const GValue *value;
 115       guint i;
 116
 117       list = gst_structure_get_value (s, "magnitude");
 118       for (i = 0; i < spect_bands; ++i) {
 119         value = gst_value_list_get_value (list, i);
 120         spect[i] = height_scale * g_value_get_float (value);
 121       }
 122       draw_spectrum (spect);
 123       g_free (spect);
 124     }
 125   }
 126   return TRUE;
 127 }
 128
 129 int
 130 main (int argc, char *argv[])
 131 {
 132   GstElement *bin;
 133   GstElement *src, *capsfilter, *equalizer, *spectrum, *audioconvert, *sink;
 134   GstCaps *caps;
 135   GstBus *bus;
 136   GtkWidget *appwindow, *vbox, *hbox, *widget;
 137   int i;
 138
 139   gst_init (&argc, &argv);
 140   gtk_init (&argc, &argv);
 141
 142   bin = gst_pipeline_new ("bin");
 143
 144   /* White noise */
 145   src = gst_element_factory_make ("audiotestsrc", "src");
 146   g_object_set (G_OBJECT (src), "wave", 5, "volume", 0.8, NULL);
 147
 148   /* Force float32 samples */
 149   capsfilter = gst_element_factory_make ("capsfilter", "capsfilter");
 150   caps =
 151       gst_caps_new_simple ("audio/x-raw-float", "width", G_TYPE_INT, 32, NULL);
 152   g_object_set (capsfilter, "caps", caps, NULL);
 153
 154   equalizer = gst_element_factory_make ("equalizer-nbands", "equalizer");
 155   g_object_set (G_OBJECT (equalizer), "num-bands", NBANDS, NULL);
 156
 157   spectrum = gst_element_factory_make ("spectrum", "spectrum");
 158   g_object_set (G_OBJECT (spectrum), "bands", spect_bands, "threshold", -80,
 159       "message", TRUE, "interval", 500 * GST_MSECOND, NULL);
 160
 161   audioconvert = gst_element_factory_make ("audioconvert", "audioconvert");
 162
 163   sink = gst_element_factory_make ("autoaudiosink", "sink");
 164
 165   gst_bin_add_many (GST_BIN (bin), src, capsfilter, equalizer, spectrum,
 166       audioconvert, sink, NULL);
 167   if (!gst_element_link_many (src, capsfilter, equalizer, spectrum,
 168           audioconvert, sink, NULL)) {
 169     fprintf (stderr, "can't link elements\n");
 170     exit (1);
 171   }
 172
 173   bus = gst_element_get_bus (bin);
 174   gst_bus_add_watch (bus, message_handler, NULL);
 175   gst_object_unref (bus);
 176
 177   appwindow = gtk_window_new (GTK_WINDOW_TOPLEVEL);
 178   g_signal_connect (G_OBJECT (appwindow), "destroy",
 179       G_CALLBACK (on_window_destroy), NULL);
 180   vbox = gtk_vbox_new (FALSE, 6);
 181
 182   drawingarea = gtk_drawing_area_new ();
 183   gtk_widget_set_size_request (drawingarea, spect_bands, spect_height);
 184   g_signal_connect (G_OBJECT (drawingarea), "configure-event",
 185       G_CALLBACK (on_configure_event), (gpointer) spectrum);
 186   gtk_box_pack_start (GTK_BOX (vbox), drawingarea, TRUE, TRUE, 0);
 187
 188   hbox = gtk_hbox_new (FALSE, 20);
 189
 190   for (i = 0; i < NBANDS; i++) {
 191     GstObject *band;
 192     gdouble freq;
 193     gdouble bw;
 194     gdouble gain;
 195     gchar *label;
 196     GtkWidget *frame, *scales_hbox;
 197
 198     band = gst_child_proxy_get_child_by_index (GST_CHILD_PROXY (equalizer), i);
 199     g_assert (band != NULL);
 200     g_object_get (G_OBJECT (band), "freq", &freq, NULL);
 201     g_object_get (G_OBJECT (band), "bandwidth", &bw, NULL);
 202     g_object_get (G_OBJECT (band), "gain", &gain, NULL);
 203
 204     label = g_strdup_printf ("%d Hz", (int) (freq + 0.5));
 205     frame = gtk_frame_new (label);
 206     g_free (label);
 207
 208     scales_hbox = gtk_hbox_new (FALSE, 6);
 209
 210     widget = gtk_vscale_new_with_range (-24.0, 12.0, 0.5);
 211     gtk_scale_set_draw_value (GTK_SCALE (widget), TRUE);
 212     gtk_scale_set_value_pos (GTK_SCALE (widget), GTK_POS_TOP);
 213     gtk_range_set_value (GTK_RANGE (widget), gain);
 214     gtk_widget_set_size_request (widget, 25, 150);
 215     g_signal_connect (G_OBJECT (widget), "value-changed",
 216         G_CALLBACK (on_gain_changed), (gpointer) band);
 217     gtk_box_pack_start (GTK_BOX (scales_hbox), widget, FALSE, FALSE, 0);
 218
 219     widget = gtk_vscale_new_with_range (0.0, 20000.0, 5.0);
 220     gtk_scale_set_draw_value (GTK_SCALE (widget), TRUE);
 221     gtk_scale_set_value_pos (GTK_SCALE (widget), GTK_POS_TOP);
 222     gtk_range_set_value (GTK_RANGE (widget), bw);
 223     gtk_widget_set_size_request (widget, 25, 150);
 224     g_signal_connect (G_OBJECT (widget), "value-changed",
 225         G_CALLBACK (on_bandwidth_changed), (gpointer) band);
 226     gtk_box_pack_start (GTK_BOX (scales_hbox), widget, TRUE, TRUE, 0);
 227
 228     widget = gtk_vscale_new_with_range (20.0, 20000.0, 5.0);
 229     gtk_scale_set_draw_value (GTK_SCALE (widget), TRUE);
 230     gtk_scale_set_value_pos (GTK_SCALE (widget), GTK_POS_TOP);
 231     gtk_range_set_value (GTK_RANGE (widget), freq);
 232     gtk_widget_set_size_request (widget, 25, 150);
 233     g_signal_connect (G_OBJECT (widget), "value-changed",
 234         G_CALLBACK (on_freq_changed), (gpointer) band);
 235     gtk_box_pack_start (GTK_BOX (scales_hbox), widget, TRUE, TRUE, 0);
 236
 237     gtk_container_add (GTK_CONTAINER (frame), scales_hbox);
 238
 239     gtk_box_pack_start (GTK_BOX (hbox), frame, TRUE, TRUE, 0);
 240   }
 241
 242   gtk_box_pack_start (GTK_BOX (vbox), hbox, TRUE, TRUE, 0);
 243
 244   gtk_container_add (GTK_CONTAINER (appwindow), vbox);
 245   gtk_widget_show_all (appwindow);
 246
 247   gst_element_set_state (bin, GST_STATE_PLAYING);
 248   gtk_main ();
 249   gst_element_set_state (bin, GST_STATE_NULL);
 250
 251   gst_object_unref (bin);
 252
 253   return 0;
 254 }