gstgoom.c 18.1 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
static gboolean
gst_goom_src_query (GstPad * pad, GstQuery * query)
{
403
  gboolean res = FALSE;
404 405 406 407 408 409 410 411 412 413 414 415 416 417
  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;

418 419 420
      if (goom->rate == 0)
        break;

421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437
      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 */
438
        min_latency += our_latency;
439
        if (max_latency != -1)
440
          max_latency += our_latency;
441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459

        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
460
static GstFlowReturn
Wim Taymans's avatar
Wim Taymans committed
461
get_buffer (GstGoom * goom, GstBuffer ** outbuf)
Wim Taymans's avatar
Wim Taymans committed
462 463
{
  GstFlowReturn ret;
Wim Taymans's avatar
Wim Taymans committed
464

Wim Taymans's avatar
Wim Taymans committed
465 466 467 468
  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
469

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

Wim Taymans's avatar
Wim Taymans committed
473 474 475 476 477 478
  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
479

Wim Taymans's avatar
Wim Taymans committed
480 481
  return GST_FLOW_OK;
}
482

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

Wim Taymans's avatar
Wim Taymans committed
490
  goom = GST_GOOM (gst_pad_get_parent (pad));
491 492 493 494
  if (goom->bps == 0) {
    ret = GST_FLOW_NOT_NEGOTIATED;
    goto beach;
  }
Wim Taymans's avatar
Wim Taymans committed
495

Wim Taymans's avatar
Wim Taymans committed
496 497
  /* If we don't have an output format yet, preallocate a buffer to try and
   * set one */
498
  if (GST_PAD_CAPS (goom->srcpad) == NULL) {
Wim Taymans's avatar
Wim Taymans committed
499 500 501 502 503 504 505 506 507 508
    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
509 510
  }

Wim Taymans's avatar
Wim Taymans committed
511 512 513 514
  GST_DEBUG_OBJECT (goom,
      "Input buffer has %d samples, time=%" G_GUINT64_FORMAT,
      GST_BUFFER_SIZE (buffer) / goom->bps, GST_BUFFER_TIMESTAMP (buffer));

515
  /* Collect samples until we have enough for an output frame */
Wim Taymans's avatar
Wim Taymans committed
516 517 518 519
  gst_adapter_push (goom->adapter, buffer);

  ret = GST_FLOW_OK;

520
  while (TRUE) {
Wim Taymans's avatar
Wim Taymans committed
521
    const guint16 *data;
Wim Taymans's avatar
Wim Taymans committed
522
    gboolean need_skip;
523 524
    guchar *out_frame;
    gint i;
525
    guint avail, to_flush;
526
    guint64 dist, timestamp;
527 528 529 530 531 532 533 534 535 536 537

    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;
538

Wim Taymans's avatar
Wim Taymans committed
539 540
    GST_DEBUG_OBJECT (goom, "processing buffer");

541 542 543 544 545 546 547 548 549
    /* 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
550 551 552
      gint64 qostime;

      qostime = gst_segment_to_running_time (&goom->segment, GST_FORMAT_TIME,
553
          timestamp);
554
      qostime += goom->duration;
Wim Taymans's avatar
Wim Taymans committed
555 556 557 558 559 560 561 562 563 564 565 566 567 568

      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;
      }
    }

569
    /* get next GOOM_SAMPLES, we have at least this amount of samples */
Wim Taymans's avatar
Wim Taymans committed
570 571 572
    data =
        (const guint16 *) gst_adapter_peek (goom->adapter,
        GOOM_SAMPLES * goom->bps);
Wim Taymans's avatar
Wim Taymans committed
573

574 575 576 577 578 579 580 581 582 583 584
    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++;
      }
    }
585

Wim Taymans's avatar
Wim Taymans committed
586 587 588 589 590 591 592 593
    /* 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
594

595
    GST_BUFFER_TIMESTAMP (outbuf) = timestamp;
Wim Taymans's avatar
Wim Taymans committed
596 597
    GST_BUFFER_DURATION (outbuf) = goom->duration;
    GST_BUFFER_SIZE (outbuf) = goom->outsize;
Wim Taymans's avatar
Wim Taymans committed
598

599
    out_frame = (guchar *) goom_update (goom->plugin, goom->datain, 0, 0);
Wim Taymans's avatar
Wim Taymans committed
600 601 602
    memcpy (GST_BUFFER_DATA (outbuf), out_frame, goom->outsize);

    GST_DEBUG ("Pushing frame with time=%" GST_TIME_FORMAT ", duration=%"
603
        GST_TIME_FORMAT, GST_TIME_ARGS (timestamp),
Wim Taymans's avatar
Wim Taymans committed
604 605 606 607
        GST_TIME_ARGS (goom->duration));

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

Wim Taymans's avatar
Wim Taymans committed
609
  skip:
610 611 612 613 614 615 616
    /* 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
617 618 619

    if (ret != GST_FLOW_OK)
      break;
620
  }
Wim Taymans's avatar
Wim Taymans committed
621 622 623 624 625 626 627 628

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

beach:
  gst_object_unref (goom);

  return ret;
629 630
}

631 632
static GstStateChangeReturn
gst_goom_change_state (GstElement * element, GstStateChange transition)
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
633
{
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
634
  GstGoom *goom = GST_GOOM (element);
635
  GstStateChangeReturn ret;
636

Wim Taymans's avatar
Wim Taymans committed
637
  switch (transition) {
638
    case GST_STATE_CHANGE_NULL_TO_READY:
639
      break;
640
    case GST_STATE_CHANGE_READY_TO_PAUSED:
Wim Taymans's avatar
Wim Taymans committed
641
      gst_goom_reset (goom);
642 643 644 645 646
      break;
    default:
      break;
  }

647
  ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
Wim Taymans's avatar
Wim Taymans committed
648 649

  switch (transition) {
650
    case GST_STATE_CHANGE_PAUSED_TO_READY:
Wim Taymans's avatar
Wim Taymans committed
651
      break;
652
    case GST_STATE_CHANGE_READY_TO_NULL:
Wim Taymans's avatar
Wim Taymans committed
653 654 655 656
      break;
    default:
      break;
  }
657

Wim Taymans's avatar
Wim Taymans committed
658
  return ret;
659 660
}

661
static gboolean
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
662
plugin_init (GstPlugin * plugin)
663
{
664 665
  GST_DEBUG_CATEGORY_INIT (goom_debug, "goom", 0, "goom visualisation element");

666 667 668 669
#if HAVE_ORC
  orc_init ();
#endif

Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
670
  return gst_element_register (plugin, "goom", GST_RANK_NONE, GST_TYPE_GOOM);
671 672
}

Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
673 674 675 676
GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
    GST_VERSION_MINOR,
    "goom",
    "GOOM visualization filter",
677
    plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)