gstgoom.c 18.3 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>
Wim Taymans's avatar
Wim Taymans committed
4
 *           (C) <2011> Wim Taymans <wim.taymans at gmail dot com>
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
 *
 * 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
22 23 24 25
/**
 * SECTION:element-goom
 * @see_also: synaesthesia
 *
26
 * Goom is an audio visualisation element. It creates warping structures
Jan Schmidt's avatar
Jan Schmidt committed
27
 * based on the incoming audio signal.
28 29
 *
 * <refsect2>
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
30
 * <title>Example launch line</title>
31
 * |[
Wim Taymans's avatar
Wim Taymans committed
32
 * gst-launch -v audiotestsrc ! goom ! videoconvert ! xvimagesink
33
 * ]|
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
34 35 36
 * </refsect2>
 */

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

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

48 49 50 51
#if HAVE_ORC
#include <orc/orc.h>
#endif

52
GST_DEBUG_CATEGORY (goom_debug);
53 54
#define GST_CAT_DEFAULT goom_debug

55 56 57 58 59
#define DEFAULT_WIDTH  320
#define DEFAULT_HEIGHT 240
#define DEFAULT_FPS_N  25
#define DEFAULT_FPS_D  1

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

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

Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
73 74 75
static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("src",
    GST_PAD_SRC,
    GST_PAD_ALWAYS,
Wim Taymans's avatar
Wim Taymans committed
76 77 78 79 80
#if G_BYTE_ORDER == G_BIG_ENDIAN
    GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE ("xRGB"))
#else
    GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE ("BGRx"))
#endif
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
81
    );
82

83 84 85
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 */
Wim Taymans's avatar
Wim Taymans committed
86 87
    GST_STATIC_CAPS ("audio/x-raw, "
        "format = (string) " GST_AUDIO_NE (S16) ", "
Wim Taymans's avatar
Wim Taymans committed
88
        "rate = (int) [ 8000, 96000 ], " "channels = (int) { 1, 2 }")
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
89 90 91
    );


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

Wim Taymans's avatar
Wim Taymans committed
103 104
#define gst_goom_parent_class parent_class
G_DEFINE_TYPE (GstGoom, gst_goom, GST_TYPE_ELEMENT);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
105

106
static void
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
107
gst_goom_class_init (GstGoomClass * klass)
108 109 110 111
{
  GObjectClass *gobject_class;
  GstElementClass *gstelement_class;

Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
112 113
  gobject_class = (GObjectClass *) klass;
  gstelement_class = (GstElementClass *) klass;
114

Wim Taymans's avatar
Wim Taymans committed
115
  gobject_class->finalize = gst_goom_finalize;
116

Wim Taymans's avatar
Wim Taymans committed
117 118 119 120 121 122 123 124 125
  gst_element_class_set_details_simple (gstelement_class, "GOOM: what a GOOM!",
      "Visualization",
      "Takes frames of data and outputs video frames using the GOOM filter",
      "Wim Taymans <wim@fluendo.com>");
  gst_element_class_add_pad_template (gstelement_class,
      gst_static_pad_template_get (&sink_template));
  gst_element_class_add_pad_template (gstelement_class,
      gst_static_pad_template_get (&src_template));

126
  gstelement_class->change_state = GST_DEBUG_FUNCPTR (gst_goom_change_state);
127 128 129
}

static void
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
130
gst_goom_init (GstGoom * goom)
131 132
{
  /* create the sink and src pads */
133
  goom->sinkpad = gst_pad_new_from_static_template (&sink_template, "sink");
134 135 136 137
  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));
Wim Taymans's avatar
Wim Taymans committed
138 139 140
  gst_element_add_pad (GST_ELEMENT (goom), goom->sinkpad);

  goom->srcpad = gst_pad_new_from_static_template (&src_template, "src");
141 142
  gst_pad_set_event_function (goom->srcpad,
      GST_DEBUG_FUNCPTR (gst_goom_src_event));
143 144
  gst_pad_set_query_function (goom->srcpad,
      GST_DEBUG_FUNCPTR (gst_goom_src_query));
Wim Taymans's avatar
Wim Taymans committed
145
  gst_element_add_pad (GST_ELEMENT (goom), goom->srcpad);
Wim Taymans's avatar
Wim Taymans committed
146

147 148
  goom->adapter = gst_adapter_new ();

149 150 151 152
  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
153
  goom->channels = 0;
Wim Taymans's avatar
Wim Taymans committed
154 155
  goom->rate = 0;
  goom->duration = 0;
156

157
  goom->plugin = goom_init (goom->width, goom->height);
Wim Taymans's avatar
Wim Taymans committed
158 159 160
}

static void
Wim Taymans's avatar
Wim Taymans committed
161
gst_goom_finalize (GObject * object)
Wim Taymans's avatar
Wim Taymans committed
162
{
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
163
  GstGoom *goom = GST_GOOM (object);
164

165
  goom_close (goom->plugin);
166
  goom->plugin = NULL;
167

Wim Taymans's avatar
Wim Taymans committed
168
  g_object_unref (goom->adapter);
Wim Taymans's avatar
Wim Taymans committed
169 170
  if (goom->pool)
    gst_object_unref (goom->pool);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
171

Wim Taymans's avatar
Wim Taymans committed
172 173 174 175 176 177 178 179 180 181 182 183 184
  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);
185 186
}

Wim Taymans's avatar
Wim Taymans committed
187
static gboolean
Wim Taymans's avatar
Wim Taymans committed
188
gst_goom_sink_setcaps (GstGoom * goom, GstCaps * caps)
189
{
David Schleef's avatar
David Schleef committed
190 191 192
  GstStructure *structure;

  structure = gst_caps_get_structure (caps, 0);
193

194 195
  gst_structure_get_int (structure, "channels", &goom->channels);
  gst_structure_get_int (structure, "rate", &goom->rate);
Wim Taymans's avatar
Wim Taymans committed
196

Wim Taymans's avatar
Wim Taymans committed
197 198
  goom->bps = goom->channels * sizeof (gint16);

Wim Taymans's avatar
Wim Taymans committed
199
  return TRUE;
Wim Taymans's avatar
Wim Taymans committed
200 201
}

Wim Taymans's avatar
Wim Taymans committed
202
static gboolean
Wim Taymans's avatar
Wim Taymans committed
203
gst_goom_src_setcaps (GstGoom * goom, GstCaps * caps)
Wim Taymans's avatar
Wim Taymans committed
204
{
David Schleef's avatar
David Schleef committed
205
  GstStructure *structure;
Wim Taymans's avatar
Wim Taymans committed
206
  gboolean res;
Wim Taymans's avatar
Wim Taymans committed
207

David Schleef's avatar
David Schleef committed
208
  structure = gst_caps_get_structure (caps, 0);
209
  if (!gst_structure_get_int (structure, "width", &goom->width) ||
Jan Schmidt's avatar
Jan Schmidt committed
210 211 212
      !gst_structure_get_int (structure, "height", &goom->height) ||
      !gst_structure_get_fraction (structure, "framerate", &goom->fps_n,
          &goom->fps_d))
Wim Taymans's avatar
Wim Taymans committed
213
    goto error;
214

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

Wim Taymans's avatar
Wim Taymans committed
217 218 219 220 221
  /* 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);
222
  goom->bpf = goom->spf * goom->bps;
Wim Taymans's avatar
Wim Taymans committed
223 224 225 226

  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
227 228 229 230 231 232 233 234 235 236
  res = gst_pad_push_event (goom->srcpad, gst_event_new_caps (caps));

  return res;

  /* ERRORS */
error:
  {
    GST_DEBUG_OBJECT (goom, "error parsing caps");
    return FALSE;
  }
237 238
}

Wim Taymans's avatar
Wim Taymans committed
239
static gboolean
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
240
gst_goom_src_negotiate (GstGoom * goom)
241
{
242
  GstCaps *othercaps, *target;
243
  GstStructure *structure;
Wim Taymans's avatar
Wim Taymans committed
244 245 246 247
  GstCaps *templ;
  GstQuery *query;
  GstBufferPool *pool = NULL;
  guint size, min, max, prefix, alignment;
248

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

251 252
  GST_DEBUG_OBJECT (goom, "performing negotiation");

Wim Taymans's avatar
Wim Taymans committed
253
  /* see what the peer can do */
Wim Taymans's avatar
Wim Taymans committed
254
  othercaps = gst_pad_peer_get_caps (goom->srcpad, NULL);
Wim Taymans's avatar
Wim Taymans committed
255
  if (othercaps) {
256
    target = gst_caps_intersect (othercaps, templ);
Wim Taymans's avatar
Wim Taymans committed
257
    gst_caps_unref (othercaps);
Wim Taymans's avatar
Wim Taymans committed
258
    gst_caps_unref (templ);
259

260
    if (gst_caps_is_empty (target))
Wim Taymans's avatar
Wim Taymans committed
261 262
      goto no_format;

263
    gst_caps_truncate (target);
Wim Taymans's avatar
Wim Taymans committed
264
  } else {
Wim Taymans's avatar
Wim Taymans committed
265
    target = templ;
266 267
  }

Wim Taymans's avatar
Wim Taymans committed
268
  structure = gst_caps_get_structure (target, 0);
269 270 271 272
  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
273

274
  gst_goom_src_setcaps (goom, target);
Wim Taymans's avatar
Wim Taymans committed
275 276 277 278 279 280 281 282 283 284 285 286 287

  /* try to get a bufferpool now */
  /* find a pool for the negotiated caps now */
  query = gst_query_new_allocation (target, TRUE);

  if (gst_pad_peer_query (goom->srcpad, query)) {
    /* we got configuration from our peer, parse them */
    gst_query_parse_allocation_params (query, &size, &min, &max, &prefix,
        &alignment, &pool);
  } else {
    size = goom->outsize;
    min = max = 0;
    prefix = 0;
288
    alignment = 0;
Wim Taymans's avatar
Wim Taymans committed
289 290 291 292 293 294 295 296 297
  }

  if (pool == NULL) {
    GstStructure *config;

    /* we did not get a pool, make one ourselves then */
    pool = gst_buffer_pool_new ();

    config = gst_buffer_pool_get_config (pool);
Wim Taymans's avatar
Wim Taymans committed
298
    gst_buffer_pool_config_set (config, target, size, min, max, prefix,
Wim Taymans's avatar
Wim Taymans committed
299 300 301 302 303 304 305 306 307 308 309
        alignment);
    gst_buffer_pool_set_config (pool, config);
  }

  if (goom->pool)
    gst_object_unref (goom->pool);
  goom->pool = pool;

  /* and activate */
  gst_buffer_pool_set_active (pool, TRUE);

Wim Taymans's avatar
Wim Taymans committed
310 311 312 313 314 315
  gst_caps_unref (target);

  return TRUE;

no_format:
  {
316
    gst_caps_unref (target);
Wim Taymans's avatar
Wim Taymans committed
317 318
    return FALSE;
  }
319 320
}

Wim Taymans's avatar
Wim Taymans committed
321
static gboolean
Wim Taymans's avatar
Wim Taymans committed
322
gst_goom_src_event (GstPad * pad, GstEvent * event)
323
{
Wim Taymans's avatar
Wim Taymans committed
324
  gboolean res;
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
325
  GstGoom *goom;
326

Wim Taymans's avatar
Wim Taymans committed
327
  goom = GST_GOOM (gst_pad_get_parent (pad));
Wim Taymans's avatar
Wim Taymans committed
328

Wim Taymans's avatar
Wim Taymans committed
329
  switch (GST_EVENT_TYPE (event)) {
Wim Taymans's avatar
Wim Taymans committed
330 331 332 333 334 335
    case GST_EVENT_QOS:
    {
      gdouble proportion;
      GstClockTimeDiff diff;
      GstClockTime timestamp;

Wim Taymans's avatar
Wim Taymans committed
336
      gst_event_parse_qos (event, NULL, &proportion, &diff, &timestamp);
Wim Taymans's avatar
Wim Taymans committed
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

      /* 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)) {
Wim Taymans's avatar
Wim Taymans committed
370 371 372 373 374 375 376 377
    case GST_EVENT_CAPS:
    {
      GstCaps *caps;

      gst_event_parse_caps (event, &caps);
      res = gst_goom_sink_setcaps (goom, caps);
      break;
    }
Wim Taymans's avatar
Wim Taymans committed
378 379 380 381 382 383 384
    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;
Wim Taymans's avatar
Wim Taymans committed
385
    case GST_EVENT_SEGMENT:
Wim Taymans's avatar
Wim Taymans committed
386
    {
Wim Taymans's avatar
Wim Taymans committed
387 388 389
      /* the newsegment values are used to clip the input samples
       * and to convert the incomming timestamps to running time so
       * we can do QoS */
Wim Taymans's avatar
Wim Taymans committed
390
      gst_event_copy_segment (event, &goom->segment);
Wim Taymans's avatar
Wim Taymans committed
391 392 393

      res = gst_pad_push_event (goom->srcpad, event);
      break;
Wim Taymans's avatar
Wim Taymans committed
394
    }
Wim Taymans's avatar
Wim Taymans committed
395
    default:
Wim Taymans's avatar
Wim Taymans committed
396
      res = gst_pad_push_event (goom->srcpad, event);
Wim Taymans's avatar
Wim Taymans committed
397
      break;
Wim Taymans's avatar
Wim Taymans committed
398
  }
Wim Taymans's avatar
Wim Taymans committed
399 400
  gst_object_unref (goom);

Wim Taymans's avatar
Wim Taymans committed
401 402
  return res;
}
Wim Taymans's avatar
Wim Taymans committed
403

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

422 423 424
      if (goom->rate == 0)
        break;

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

        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
464
/* make sure we are negotiated */
Wim Taymans's avatar
Wim Taymans committed
465
static GstFlowReturn
Wim Taymans's avatar
Wim Taymans committed
466
ensure_negotiated (GstGoom * goom)
Wim Taymans's avatar
Wim Taymans committed
467
{
Wim Taymans's avatar
Wim Taymans committed
468 469 470 471 472 473
  gboolean reconfigure;

  GST_OBJECT_LOCK (goom->srcpad);
  reconfigure = GST_PAD_NEEDS_RECONFIGURE (goom->srcpad);
  GST_OBJECT_FLAG_UNSET (goom->srcpad, GST_PAD_NEED_RECONFIGURE);
  GST_OBJECT_UNLOCK (goom->srcpad);
Wim Taymans's avatar
Wim Taymans committed
474

Wim Taymans's avatar
Wim Taymans committed
475 476
  /* we don't know an output format yet, pick one */
  if (reconfigure || !gst_pad_has_current_caps (goom->srcpad)) {
Wim Taymans's avatar
Wim Taymans committed
477 478 479 480 481
    if (!gst_goom_src_negotiate (goom))
      return GST_FLOW_NOT_NEGOTIATED;
  }
  return GST_FLOW_OK;
}
482

Wim Taymans's avatar
Wim Taymans committed
483

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

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

Wim Taymans's avatar
Wim Taymans committed
497 498 499 500 501
  /* Make sure have an output format */
  ret = ensure_negotiated (goom);
  if (ret != GST_FLOW_OK) {
    gst_buffer_unref (buffer);
    goto beach;
Wim Taymans's avatar
Wim Taymans committed
502 503 504 505 506
  }

  /* 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
507 508
  }

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

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

  ret = GST_FLOW_OK;

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

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

Wim Taymans's avatar
Wim Taymans committed
537 538
    GST_DEBUG_OBJECT (goom, "processing buffer");

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

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

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

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

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

Wim Taymans's avatar
Wim Taymans committed
584 585 586
    /* 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) {
Wim Taymans's avatar
Wim Taymans committed
587 588
      GST_DEBUG_OBJECT (goom, "allocating output buffer");
      ret = gst_buffer_pool_acquire_buffer (goom->pool, &outbuf, NULL);
Wim Taymans's avatar
Wim Taymans committed
589
      if (ret != GST_FLOW_OK) {
Wim Taymans's avatar
Wim Taymans committed
590
        gst_adapter_unmap (goom->adapter, 0);
Wim Taymans's avatar
Wim Taymans committed
591 592 593
        goto beach;
      }
    }
Wim Taymans's avatar
Wim Taymans committed
594

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

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

    gst_adapter_unmap (goom->adapter, 0);
Wim Taymans's avatar
Wim Taymans committed
602 603

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

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

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

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

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

beach:
  gst_object_unref (goom);

  return ret;
630 631
}

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

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

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

  switch (transition) {
651
    case GST_STATE_CHANGE_PAUSED_TO_READY:
Wim Taymans's avatar
Wim Taymans committed
652 653 654 655 656
      if (goom->pool) {
        gst_buffer_pool_set_active (goom->pool, FALSE);
        gst_object_unref (goom->pool);
        goom->pool = NULL;
      }
Wim Taymans's avatar
Wim Taymans committed
657
      break;
658
    case GST_STATE_CHANGE_READY_TO_NULL:
Wim Taymans's avatar
Wim Taymans committed
659 660 661 662
      break;
    default:
      break;
  }
663

Wim Taymans's avatar
Wim Taymans committed
664
  return ret;
665 666
}

667
static gboolean
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
668
plugin_init (GstPlugin * plugin)
669
{
670 671
  GST_DEBUG_CATEGORY_INIT (goom_debug, "goom", 0, "goom visualisation element");

672 673 674 675
#if HAVE_ORC
  orc_init ();
#endif

Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
676
  return gst_element_register (plugin, "goom", GST_RANK_NONE, GST_TYPE_GOOM);
677 678
}

Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
679 680 681 682
GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
    GST_VERSION_MINOR,
    "goom",
    "GOOM visualization filter",
683
    plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)