gstgoom.c 18 KB
Newer Older
1
2
/* gstgoom.c: implementation of goom drawing element
 * Copyright (C) <2001> Richard Boulton <richard@tartarus.org>
Wim Taymans's avatar
Wim Taymans committed
3
 *           (C) <2006> Wim Taymans <wim at fluendo dot com>
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
 *
 * 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.
 */

Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
21
22
23
24
/**
 * SECTION:element-goom
 * @see_also: synaesthesia
 *
25
 * Goom is an audio visualisation element. It creates warping structures
Jan Schmidt's avatar
Jan Schmidt committed
26
 * based on the incoming audio signal.
27
28
 *
 * <refsect2>
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
29
 * <title>Example launch line</title>
30
 * |[
31
 * gst-launch -v audiotestsrc ! goom ! ffmpegcolorspace ! xvimagesink
32
 * ]|
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
33
34
35
 * </refsect2>
 */

36
37
38
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
39

40
#include <string.h>
41
#include <gst/gst.h>
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
42
#include "gstgoom.h"
43
#include <gst/video/video.h>
44
#include "goom.h"
45

46
47
48
49
#if HAVE_ORC
#include <orc/orc.h>
#endif

50
GST_DEBUG_CATEGORY (goom_debug);
51
52
#define GST_CAT_DEFAULT goom_debug

53
54
55
56
57
#define DEFAULT_WIDTH  320
#define DEFAULT_HEIGHT 240
#define DEFAULT_FPS_N  25
#define DEFAULT_FPS_D  1

58
/* signals and args */
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
59
60
enum
{
61
62
63
64
  /* FILL ME */
  LAST_SIGNAL
};

Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
65
66
enum
{
67
68
  ARG_0
      /* FILL ME */
69
70
};

Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
71
72
73
74
75
static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("src",
    GST_PAD_SRC,
    GST_PAD_ALWAYS,
    GST_STATIC_CAPS (GST_VIDEO_CAPS_xRGB_HOST_ENDIAN)
    );
76

77
78
79
static GstStaticPadTemplate sink_template = GST_STATIC_PAD_TEMPLATE ("sink",    /* the name of the pads */
    GST_PAD_SINK,               /* type of the pad */
    GST_PAD_ALWAYS,             /* ALWAYS/SOMETIMES */
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
80
    GST_STATIC_CAPS ("audio/x-raw-int, "
81
82
83
84
        "endianness = (int) BYTE_ORDER, "
        "signed = (boolean) TRUE, "
        "width = (int) 16, "
        "depth = (int) 16, "
Wim Taymans's avatar
Wim Taymans committed
85
        "rate = (int) [ 8000, 96000 ], " "channels = (int) { 1, 2 }")
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
86
87
88
    );


Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
89
90
91
static void gst_goom_class_init (GstGoomClass * klass);
static void gst_goom_base_init (GstGoomClass * klass);
static void gst_goom_init (GstGoom * goom);
Wim Taymans's avatar
Wim Taymans committed
92
static void gst_goom_finalize (GObject * object);
93

94
95
static GstStateChangeReturn gst_goom_change_state (GstElement * element,
    GstStateChange transition);
96

Wim Taymans's avatar
Wim Taymans committed
97
static GstFlowReturn gst_goom_chain (GstPad * pad, GstBuffer * buffer);
Wim Taymans's avatar
Wim Taymans committed
98
99
static gboolean gst_goom_src_event (GstPad * pad, GstEvent * event);
static gboolean gst_goom_sink_event (GstPad * pad, GstEvent * event);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
100

101
102
static gboolean gst_goom_src_query (GstPad * pad, GstQuery * query);

103
104
static gboolean gst_goom_sink_setcaps (GstPad * pad, GstCaps * caps);
static gboolean gst_goom_src_setcaps (GstPad * pad, GstCaps * caps);
105
106
107
108
109
110
111
112
113
114

static GstElementClass *parent_class = NULL;

GType
gst_goom_get_type (void)
{
  static GType type = 0;

  if (!type) {
    static const GTypeInfo info = {
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
115
      sizeof (GstGoomClass),
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
116
117
      (GBaseInitFunc) gst_goom_base_init,
      NULL,
118
119
120
      (GClassInitFunc) gst_goom_class_init,
      NULL,
      NULL,
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
121
      sizeof (GstGoom),
122
123
124
      0,
      (GInstanceInitFunc) gst_goom_init,
    };
125

Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
126
    type = g_type_register_static (GST_TYPE_ELEMENT, "GstGoom", &info, 0);
127
128
129
130
  }
  return type;
}

Ronald S. Bultje's avatar
Ronald S. Bultje committed
131
static void
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
132
gst_goom_base_init (GstGoomClass * klass)
Ronald S. Bultje's avatar
Ronald S. Bultje committed
133
134
135
{
  GstElementClass *element_class = GST_ELEMENT_CLASS (klass);

136
137
138
139
  gst_element_class_set_details_simple (element_class, "GOOM: what a GOOM!",
      "Visualization",
      "Takes frames of data and outputs video frames using the GOOM filter",
      "Wim Taymans <wim@fluendo.com>");
Ronald S. Bultje's avatar
Ronald S. Bultje committed
140
  gst_element_class_add_pad_template (element_class,
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
141
      gst_static_pad_template_get (&sink_template));
Ronald S. Bultje's avatar
Ronald S. Bultje committed
142
  gst_element_class_add_pad_template (element_class,
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
143
      gst_static_pad_template_get (&src_template));
Ronald S. Bultje's avatar
Ronald S. Bultje committed
144
145
}

146
static void
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
147
gst_goom_class_init (GstGoomClass * klass)
148
149
150
151
{
  GObjectClass *gobject_class;
  GstElementClass *gstelement_class;

Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
152
153
  gobject_class = (GObjectClass *) klass;
  gstelement_class = (GstElementClass *) klass;
154

155
  parent_class = g_type_class_peek_parent (klass);
156

Wim Taymans's avatar
Wim Taymans committed
157
  gobject_class->finalize = gst_goom_finalize;
158

159
  gstelement_class->change_state = GST_DEBUG_FUNCPTR (gst_goom_change_state);
160
161
162
}

static void
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
163
gst_goom_init (GstGoom * goom)
164
165
{
  /* create the sink and src pads */
166
  goom->sinkpad = gst_pad_new_from_static_template (&sink_template, "sink");
167
168
169
170
171
172
  gst_pad_set_chain_function (goom->sinkpad,
      GST_DEBUG_FUNCPTR (gst_goom_chain));
  gst_pad_set_event_function (goom->sinkpad,
      GST_DEBUG_FUNCPTR (gst_goom_sink_event));
  gst_pad_set_setcaps_function (goom->sinkpad,
      GST_DEBUG_FUNCPTR (gst_goom_sink_setcaps));
Wim Taymans's avatar
Wim Taymans committed
173
174
175
  gst_element_add_pad (GST_ELEMENT (goom), goom->sinkpad);

  goom->srcpad = gst_pad_new_from_static_template (&src_template, "src");
176
177
178
179
  gst_pad_set_setcaps_function (goom->srcpad,
      GST_DEBUG_FUNCPTR (gst_goom_src_setcaps));
  gst_pad_set_event_function (goom->srcpad,
      GST_DEBUG_FUNCPTR (gst_goom_src_event));
180
181
  gst_pad_set_query_function (goom->srcpad,
      GST_DEBUG_FUNCPTR (gst_goom_src_query));
Wim Taymans's avatar
Wim Taymans committed
182
  gst_element_add_pad (GST_ELEMENT (goom), goom->srcpad);
Wim Taymans's avatar
Wim Taymans committed
183

184
185
  goom->adapter = gst_adapter_new ();

186
187
188
189
  goom->width = DEFAULT_WIDTH;
  goom->height = DEFAULT_HEIGHT;
  goom->fps_n = DEFAULT_FPS_N;  /* desired frame rate */
  goom->fps_d = DEFAULT_FPS_D;  /* desired frame rate */
Wim Taymans's avatar
Wim Taymans committed
190
  goom->channels = 0;
Wim Taymans's avatar
Wim Taymans committed
191
192
  goom->rate = 0;
  goom->duration = 0;
193

194
  goom->plugin = goom_init (goom->width, goom->height);
Wim Taymans's avatar
Wim Taymans committed
195
196
197
}

static void
Wim Taymans's avatar
Wim Taymans committed
198
gst_goom_finalize (GObject * object)
Wim Taymans's avatar
Wim Taymans committed
199
{
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
200
  GstGoom *goom = GST_GOOM (object);
201

202
  goom_close (goom->plugin);
203
  goom->plugin = NULL;
204

Wim Taymans's avatar
Wim Taymans committed
205
  g_object_unref (goom->adapter);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
206

Wim Taymans's avatar
Wim Taymans committed
207
208
209
210
211
212
213
214
215
216
217
218
219
  G_OBJECT_CLASS (parent_class)->finalize (object);
}

static void
gst_goom_reset (GstGoom * goom)
{
  gst_adapter_clear (goom->adapter);
  gst_segment_init (&goom->segment, GST_FORMAT_UNDEFINED);

  GST_OBJECT_LOCK (goom);
  goom->proportion = 1.0;
  goom->earliest_time = -1;
  GST_OBJECT_UNLOCK (goom);
220
221
}

Wim Taymans's avatar
Wim Taymans committed
222
223
static gboolean
gst_goom_sink_setcaps (GstPad * pad, GstCaps * caps)
224
{
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
225
  GstGoom *goom;
David Schleef's avatar
David Schleef committed
226
  GstStructure *structure;
Wim Taymans's avatar
Wim Taymans committed
227
  gboolean res;
David Schleef's avatar
David Schleef committed
228

Wim Taymans's avatar
Wim Taymans committed
229
  goom = GST_GOOM (GST_PAD_PARENT (pad));
230

David Schleef's avatar
David Schleef committed
231
  structure = gst_caps_get_structure (caps, 0);
232

Wim Taymans's avatar
Wim Taymans committed
233
234
  res = gst_structure_get_int (structure, "channels", &goom->channels);
  res &= gst_structure_get_int (structure, "rate", &goom->rate);
Wim Taymans's avatar
Wim Taymans committed
235

Wim Taymans's avatar
Wim Taymans committed
236
237
238
  goom->bps = goom->channels * sizeof (gint16);

  return res;
Wim Taymans's avatar
Wim Taymans committed
239
240
}

Wim Taymans's avatar
Wim Taymans committed
241
242
static gboolean
gst_goom_src_setcaps (GstPad * pad, GstCaps * caps)
Wim Taymans's avatar
Wim Taymans committed
243
{
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
244
  GstGoom *goom;
David Schleef's avatar
David Schleef committed
245
246
  GstStructure *structure;

Wim Taymans's avatar
Wim Taymans committed
247
  goom = GST_GOOM (GST_PAD_PARENT (pad));
Wim Taymans's avatar
Wim Taymans committed
248

David Schleef's avatar
David Schleef committed
249
  structure = gst_caps_get_structure (caps, 0);
Wim Taymans's avatar
Wim Taymans committed
250

251
  if (!gst_structure_get_int (structure, "width", &goom->width) ||
Jan Schmidt's avatar
Jan Schmidt committed
252
253
254
      !gst_structure_get_int (structure, "height", &goom->height) ||
      !gst_structure_get_fraction (structure, "framerate", &goom->fps_n,
          &goom->fps_d))
255
256
    return FALSE;

257
  goom_set_resolution (goom->plugin, goom->width, goom->height);
Wim Taymans's avatar
Wim Taymans committed
258

Wim Taymans's avatar
Wim Taymans committed
259
260
261
262
263
  /* size of the output buffer in bytes, depth is always 4 bytes */
  goom->outsize = goom->width * goom->height * 4;
  goom->duration =
      gst_util_uint64_scale_int (GST_SECOND, goom->fps_d, goom->fps_n);
  goom->spf = gst_util_uint64_scale_int (goom->rate, goom->fps_d, goom->fps_n);
264
  goom->bpf = goom->spf * goom->bps;
Wim Taymans's avatar
Wim Taymans committed
265
266
267
268

  GST_DEBUG_OBJECT (goom, "dimension %dx%d, framerate %d/%d, spf %d",
      goom->width, goom->height, goom->fps_n, goom->fps_d, goom->spf);

Wim Taymans's avatar
Wim Taymans committed
269
  return TRUE;
270
271
}

Wim Taymans's avatar
Wim Taymans committed
272
static gboolean
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
273
gst_goom_src_negotiate (GstGoom * goom)
274
{
275
  GstCaps *othercaps, *target;
276
  GstStructure *structure;
Wim Taymans's avatar
Wim Taymans committed
277
  const GstCaps *templ;
278

Wim Taymans's avatar
Wim Taymans committed
279
  templ = gst_pad_get_pad_template_caps (goom->srcpad);
280

281
282
  GST_DEBUG_OBJECT (goom, "performing negotiation");

Wim Taymans's avatar
Wim Taymans committed
283
284
285
  /* see what the peer can do */
  othercaps = gst_pad_peer_get_caps (goom->srcpad);
  if (othercaps) {
286
    target = gst_caps_intersect (othercaps, templ);
Wim Taymans's avatar
Wim Taymans committed
287
    gst_caps_unref (othercaps);
288

289
    if (gst_caps_is_empty (target))
Wim Taymans's avatar
Wim Taymans committed
290
291
      goto no_format;

292
    gst_caps_truncate (target);
Wim Taymans's avatar
Wim Taymans committed
293
294
  } else {
    target = gst_caps_ref ((GstCaps *) templ);
295
296
  }

Wim Taymans's avatar
Wim Taymans committed
297
  structure = gst_caps_get_structure (target, 0);
298
299
300
301
  gst_structure_fixate_field_nearest_int (structure, "width", DEFAULT_WIDTH);
  gst_structure_fixate_field_nearest_int (structure, "height", DEFAULT_HEIGHT);
  gst_structure_fixate_field_nearest_fraction (structure, "framerate",
      DEFAULT_FPS_N, DEFAULT_FPS_D);
Wim Taymans's avatar
Wim Taymans committed
302
303
304
305
306
307
308
309

  gst_pad_set_caps (goom->srcpad, target);
  gst_caps_unref (target);

  return TRUE;

no_format:
  {
310
    gst_caps_unref (target);
Wim Taymans's avatar
Wim Taymans committed
311
312
    return FALSE;
  }
313
314
}

Wim Taymans's avatar
Wim Taymans committed
315
static gboolean
Wim Taymans's avatar
Wim Taymans committed
316
gst_goom_src_event (GstPad * pad, GstEvent * event)
317
{
Wim Taymans's avatar
Wim Taymans committed
318
  gboolean res;
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
319
  GstGoom *goom;
320

Wim Taymans's avatar
Wim Taymans committed
321
  goom = GST_GOOM (gst_pad_get_parent (pad));
Wim Taymans's avatar
Wim Taymans committed
322

Wim Taymans's avatar
Wim Taymans committed
323
  switch (GST_EVENT_TYPE (event)) {
Wim Taymans's avatar
Wim Taymans committed
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
    case GST_EVENT_QOS:
    {
      gdouble proportion;
      GstClockTimeDiff diff;
      GstClockTime timestamp;

      gst_event_parse_qos (event, &proportion, &diff, &timestamp);

      /* save stuff for the _chain() function */
      GST_OBJECT_LOCK (goom);
      goom->proportion = proportion;
      if (diff >= 0)
        /* we're late, this is a good estimate for next displayable
         * frame (see part-qos.txt) */
        goom->earliest_time = timestamp + 2 * diff + goom->duration;
      else
        goom->earliest_time = timestamp + diff;
      GST_OBJECT_UNLOCK (goom);

      res = gst_pad_push_event (goom->sinkpad, event);
      break;
    }
    default:
      res = gst_pad_push_event (goom->sinkpad, event);
      break;
  }
  gst_object_unref (goom);

  return res;
}

static gboolean
gst_goom_sink_event (GstPad * pad, GstEvent * event)
{
  gboolean res;
  GstGoom *goom;

  goom = GST_GOOM (gst_pad_get_parent (pad));

  switch (GST_EVENT_TYPE (event)) {
    case GST_EVENT_FLUSH_START:
      res = gst_pad_push_event (goom->srcpad, event);
      break;
    case GST_EVENT_FLUSH_STOP:
      gst_goom_reset (goom);
      res = gst_pad_push_event (goom->srcpad, event);
      break;
371
    case GST_EVENT_NEWSEGMENT:
Wim Taymans's avatar
Wim Taymans committed
372
    {
373
      GstFormat format;
Wim Taymans's avatar
Wim Taymans committed
374
375
376
377
378
379
380
381
382
      gdouble rate, arate;
      gint64 start, stop, time;
      gboolean update;

      /* the newsegment values are used to clip the input samples
       * and to convert the incomming timestamps to running time so
       * we can do QoS */
      gst_event_parse_new_segment_full (event, &update, &rate, &arate, &format,
          &start, &stop, &time);
Wim Taymans's avatar
Wim Taymans committed
383

Wim Taymans's avatar
Wim Taymans committed
384
385
386
387
388
389
      /* now configure the values */
      gst_segment_set_newsegment_full (&goom->segment, update,
          rate, arate, format, start, stop, time);

      res = gst_pad_push_event (goom->srcpad, event);
      break;
Wim Taymans's avatar
Wim Taymans committed
390
    }
Wim Taymans's avatar
Wim Taymans committed
391
    default:
Wim Taymans's avatar
Wim Taymans committed
392
      res = gst_pad_push_event (goom->srcpad, event);
Wim Taymans's avatar
Wim Taymans committed
393
      break;
Wim Taymans's avatar
Wim Taymans committed
394
  }
Wim Taymans's avatar
Wim Taymans committed
395
396
  gst_object_unref (goom);

Wim Taymans's avatar
Wim Taymans committed
397
398
  return res;
}
Wim Taymans's avatar
Wim Taymans committed
399

400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
static gboolean
gst_goom_src_query (GstPad * pad, GstQuery * query)
{
  gboolean res;
  GstGoom *goom;

  goom = GST_GOOM (gst_pad_get_parent (pad));

  switch (GST_QUERY_TYPE (query)) {
    case GST_QUERY_LATENCY:
    {
      /* We need to send the query upstream and add the returned latency to our
       * own */
      GstClockTime min_latency, max_latency;
      gboolean us_live;
      GstClockTime our_latency;
      guint max_samples;

      if ((res = gst_pad_peer_query (goom->sinkpad, query))) {
        gst_query_parse_latency (query, &us_live, &min_latency, &max_latency);

        GST_DEBUG_OBJECT (goom, "Peer latency: min %"
            GST_TIME_FORMAT " max %" GST_TIME_FORMAT,
            GST_TIME_ARGS (min_latency), GST_TIME_ARGS (max_latency));

        /* the max samples we must buffer buffer */
        max_samples = MAX (GOOM_SAMPLES, goom->spf);
        our_latency =
            gst_util_uint64_scale_int (max_samples, GST_SECOND, goom->rate);

        GST_DEBUG_OBJECT (goom, "Our latency: %" GST_TIME_FORMAT,
            GST_TIME_ARGS (our_latency));

        /* we add some latency but only if we need to buffer more than what
         * upstream gives us */
435
        min_latency += our_latency;
436
        if (max_latency != -1)
437
          max_latency += our_latency;
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456

        GST_DEBUG_OBJECT (goom, "Calculated total latency : min %"
            GST_TIME_FORMAT " max %" GST_TIME_FORMAT,
            GST_TIME_ARGS (min_latency), GST_TIME_ARGS (max_latency));

        gst_query_set_latency (query, TRUE, min_latency, max_latency);
      }
      break;
    }
    default:
      res = gst_pad_peer_query (goom->sinkpad, query);
      break;
  }

  gst_object_unref (goom);

  return res;
}

Wim Taymans's avatar
Wim Taymans committed
457
static GstFlowReturn
Wim Taymans's avatar
Wim Taymans committed
458
get_buffer (GstGoom * goom, GstBuffer ** outbuf)
Wim Taymans's avatar
Wim Taymans committed
459
460
{
  GstFlowReturn ret;
Wim Taymans's avatar
Wim Taymans committed
461

Wim Taymans's avatar
Wim Taymans committed
462
463
464
465
  if (GST_PAD_CAPS (goom->srcpad) == NULL) {
    if (!gst_goom_src_negotiate (goom))
      return GST_FLOW_NOT_NEGOTIATED;
  }
Wim Taymans's avatar
Wim Taymans committed
466

Wim Taymans's avatar
Wim Taymans committed
467
468
  GST_DEBUG_OBJECT (goom, "allocating output buffer with caps %"
      GST_PTR_FORMAT, GST_PAD_CAPS (goom->srcpad));
469

Wim Taymans's avatar
Wim Taymans committed
470
471
472
473
474
475
  ret =
      gst_pad_alloc_buffer_and_set_caps (goom->srcpad,
      GST_BUFFER_OFFSET_NONE, goom->outsize,
      GST_PAD_CAPS (goom->srcpad), outbuf);
  if (ret != GST_FLOW_OK)
    return ret;
Wim Taymans's avatar
Wim Taymans committed
476

Wim Taymans's avatar
Wim Taymans committed
477
478
  return GST_FLOW_OK;
}
479

Wim Taymans's avatar
Wim Taymans committed
480
481
482
483
484
485
static GstFlowReturn
gst_goom_chain (GstPad * pad, GstBuffer * buffer)
{
  GstGoom *goom;
  GstFlowReturn ret;
  GstBuffer *outbuf = NULL;
486

Wim Taymans's avatar
Wim Taymans committed
487
  goom = GST_GOOM (gst_pad_get_parent (pad));
Wim Taymans's avatar
Wim Taymans committed
488

Wim Taymans's avatar
Wim Taymans committed
489
490
  /* If we don't have an output format yet, preallocate a buffer to try and
   * set one */
491
  if (GST_PAD_CAPS (goom->srcpad) == NULL) {
Wim Taymans's avatar
Wim Taymans committed
492
493
494
495
496
497
498
499
500
501
    ret = get_buffer (goom, &outbuf);
    if (ret != GST_FLOW_OK) {
      gst_buffer_unref (buffer);
      goto beach;
    }
  }

  /* don't try to combine samples from discont buffer */
  if (GST_BUFFER_FLAG_IS_SET (buffer, GST_BUFFER_FLAG_DISCONT)) {
    gst_adapter_clear (goom->adapter);
Wim Taymans's avatar
Wim Taymans committed
502
503
  }

Wim Taymans's avatar
Wim Taymans committed
504
505
506
507
  GST_DEBUG_OBJECT (goom,
      "Input buffer has %d samples, time=%" G_GUINT64_FORMAT,
      GST_BUFFER_SIZE (buffer) / goom->bps, GST_BUFFER_TIMESTAMP (buffer));

508
  /* Collect samples until we have enough for an output frame */
Wim Taymans's avatar
Wim Taymans committed
509
510
511
512
  gst_adapter_push (goom->adapter, buffer);

  ret = GST_FLOW_OK;

513
  while (TRUE) {
Wim Taymans's avatar
Wim Taymans committed
514
    const guint16 *data;
Wim Taymans's avatar
Wim Taymans committed
515
    gboolean need_skip;
516
517
    guchar *out_frame;
    gint i;
518
    guint avail, to_flush;
519
    guint64 dist, timestamp;
520
521
522
523
524
525
526
527
528
529
530

    avail = gst_adapter_available (goom->adapter);
    GST_DEBUG_OBJECT (goom, "avail now %u", avail);

    /* we need GOOM_SAMPLES to get a meaningful result from goom. */
    if (avail < (GOOM_SAMPLES * goom->bps))
      break;

    /* we also need enough samples to produce one frame at least */
    if (avail < goom->bpf)
      break;
531

Wim Taymans's avatar
Wim Taymans committed
532
533
    GST_DEBUG_OBJECT (goom, "processing buffer");

534
535
536
537
538
539
540
541
542
    /* get timestamp of the current adapter byte */
    timestamp = gst_adapter_prev_timestamp (goom->adapter, &dist);
    if (GST_CLOCK_TIME_IS_VALID (timestamp)) {
      /* convert bytes to time */
      dist /= goom->bps;
      timestamp += gst_util_uint64_scale_int (dist, GST_SECOND, goom->rate);
    }

    if (timestamp != -1) {
Wim Taymans's avatar
Wim Taymans committed
543
544
545
      gint64 qostime;

      qostime = gst_segment_to_running_time (&goom->segment, GST_FORMAT_TIME,
546
          timestamp);
Wim Taymans's avatar
Wim Taymans committed
547
548
549
550
551
552
553
554
555
556
557
558
559
560

      GST_OBJECT_LOCK (goom);
      /* check for QoS, don't compute buffers that are known to be late */
      need_skip = goom->earliest_time != -1 && qostime <= goom->earliest_time;
      GST_OBJECT_UNLOCK (goom);

      if (need_skip) {
        GST_WARNING_OBJECT (goom,
            "QoS: skip ts: %" GST_TIME_FORMAT ", earliest: %" GST_TIME_FORMAT,
            GST_TIME_ARGS (qostime), GST_TIME_ARGS (goom->earliest_time));
        goto skip;
      }
    }

561
    /* get next GOOM_SAMPLES, we have at least this amount of samples */
Wim Taymans's avatar
Wim Taymans committed
562
563
564
    data =
        (const guint16 *) gst_adapter_peek (goom->adapter,
        GOOM_SAMPLES * goom->bps);
Wim Taymans's avatar
Wim Taymans committed
565

566
567
568
569
570
571
572
573
574
575
576
    if (goom->channels == 2) {
      for (i = 0; i < GOOM_SAMPLES; i++) {
        goom->datain[0][i] = *data++;
        goom->datain[1][i] = *data++;
      }
    } else {
      for (i = 0; i < GOOM_SAMPLES; i++) {
        goom->datain[0][i] = *data;
        goom->datain[1][i] = *data++;
      }
    }
577

Wim Taymans's avatar
Wim Taymans committed
578
579
580
581
582
583
584
585
    /* alloc a buffer if we don't have one yet, this happens
     * when we pushed a buffer in this while loop before */
    if (outbuf == NULL) {
      ret = get_buffer (goom, &outbuf);
      if (ret != GST_FLOW_OK) {
        goto beach;
      }
    }
Wim Taymans's avatar
Wim Taymans committed
586

587
    GST_BUFFER_TIMESTAMP (outbuf) = timestamp;
Wim Taymans's avatar
Wim Taymans committed
588
589
    GST_BUFFER_DURATION (outbuf) = goom->duration;
    GST_BUFFER_SIZE (outbuf) = goom->outsize;
Wim Taymans's avatar
Wim Taymans committed
590

591
    out_frame = (guchar *) goom_update (goom->plugin, goom->datain, 0, 0);
Wim Taymans's avatar
Wim Taymans committed
592
593
594
    memcpy (GST_BUFFER_DATA (outbuf), out_frame, goom->outsize);

    GST_DEBUG ("Pushing frame with time=%" GST_TIME_FORMAT ", duration=%"
595
        GST_TIME_FORMAT, GST_TIME_ARGS (timestamp),
Wim Taymans's avatar
Wim Taymans committed
596
597
598
599
        GST_TIME_ARGS (goom->duration));

    ret = gst_pad_push (goom->srcpad, outbuf);
    outbuf = NULL;
600

Wim Taymans's avatar
Wim Taymans committed
601
  skip:
602
603
604
605
606
607
608
    /* Now flush the samples we needed for this frame, which might be more than
     * the samples we used (GOOM_SAMPLES). */
    to_flush = goom->bpf;

    GST_DEBUG_OBJECT (goom, "finished frame, flushing %u bytes from input",
        to_flush);
    gst_adapter_flush (goom->adapter, to_flush);
Wim Taymans's avatar
Wim Taymans committed
609
610
611

    if (ret != GST_FLOW_OK)
      break;
612
  }
Wim Taymans's avatar
Wim Taymans committed
613
614
615
616
617
618
619
620

  if (outbuf != NULL)
    gst_buffer_unref (outbuf);

beach:
  gst_object_unref (goom);

  return ret;
621
622
}

623
624
static GstStateChangeReturn
gst_goom_change_state (GstElement * element, GstStateChange transition)
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
625
{
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
626
  GstGoom *goom = GST_GOOM (element);
627
  GstStateChangeReturn ret;
628

Wim Taymans's avatar
Wim Taymans committed
629
  switch (transition) {
630
    case GST_STATE_CHANGE_NULL_TO_READY:
631
      break;
632
    case GST_STATE_CHANGE_READY_TO_PAUSED:
Wim Taymans's avatar
Wim Taymans committed
633
      gst_goom_reset (goom);
634
635
636
637
638
      break;
    default:
      break;
  }

639
  ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
Wim Taymans's avatar
Wim Taymans committed
640
641

  switch (transition) {
642
    case GST_STATE_CHANGE_PAUSED_TO_READY:
Wim Taymans's avatar
Wim Taymans committed
643
      break;
644
    case GST_STATE_CHANGE_READY_TO_NULL:
Wim Taymans's avatar
Wim Taymans committed
645
646
647
648
      break;
    default:
      break;
  }
649

Wim Taymans's avatar
Wim Taymans committed
650
  return ret;
651
652
}

653
static gboolean
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
654
plugin_init (GstPlugin * plugin)
655
{
656
657
  GST_DEBUG_CATEGORY_INIT (goom_debug, "goom", 0, "goom visualisation element");

658
659
660
661
#if HAVE_ORC
  orc_init ();
#endif

Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
662
  return gst_element_register (plugin, "goom", GST_RANK_NONE, GST_TYPE_GOOM);
663
664
}

Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
665
666
667
668
GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
    GST_VERSION_MINOR,
    "goom",
    "GOOM visualization filter",
669
    plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)