gstgoom.c 18.2 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
 * |[
32
 * gst-launch -v audiotestsrc ! goom ! ffmpegcolorspace ! 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>
45
#include "goom.h"
46

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

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

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

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

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

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

78 79 80
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
81
    GST_STATIC_CAPS ("audio/x-raw-int, "
82 83 84 85
        "endianness = (int) BYTE_ORDER, "
        "signed = (boolean) TRUE, "
        "width = (int) 16, "
        "depth = (int) 16, "
Wim Taymans's avatar
Wim Taymans committed
86
        "rate = (int) [ 8000, 96000 ], " "channels = (int) { 1, 2 }")
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
87 88 89
    );


Wim Taymans's avatar
Wim Taymans committed
90
static void gst_goom_finalize (GObject * object);
91

92 93
static GstStateChangeReturn gst_goom_change_state (GstElement * element,
    GstStateChange transition);
94

Wim Taymans's avatar
Wim Taymans committed
95
static GstFlowReturn gst_goom_chain (GstPad * pad, GstBuffer * buffer);
Wim Taymans's avatar
Wim Taymans committed
96 97
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
98

99 100
static gboolean gst_goom_src_query (GstPad * pad, GstQuery * query);

Wim Taymans's avatar
Wim Taymans committed
101 102
#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
103

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

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

Wim Taymans's avatar
Wim Taymans committed
113
  gobject_class->finalize = gst_goom_finalize;
114

Wim Taymans's avatar
Wim Taymans committed
115 116 117 118 119 120 121 122 123
  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));

124
  gstelement_class->change_state = GST_DEBUG_FUNCPTR (gst_goom_change_state);
125 126 127
}

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

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

145 146
  goom->adapter = gst_adapter_new ();

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

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

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

163
  goom_close (goom->plugin);
164
  goom->plugin = NULL;
165

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

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

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

  structure = gst_caps_get_structure (caps, 0);
191

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

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

Wim Taymans's avatar
Wim Taymans committed
197
  return TRUE;
Wim Taymans's avatar
Wim Taymans committed
198 199
}

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

David Schleef's avatar
David Schleef committed
206
  structure = gst_caps_get_structure (caps, 0);
207
  if (!gst_structure_get_int (structure, "width", &goom->width) ||
Jan Schmidt's avatar
Jan Schmidt committed
208 209 210
      !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
211
    goto error;
212

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

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

  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
225 226 227 228 229 230 231 232 233 234
  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;
  }
235 236
}

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

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

249 250
  GST_DEBUG_OBJECT (goom, "performing negotiation");

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

258
    if (gst_caps_is_empty (target))
Wim Taymans's avatar
Wim Taymans committed
259 260
      goto no_format;

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

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

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

  /* 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;
286
    alignment = 0;
Wim Taymans's avatar
Wim Taymans committed
287 288 289 290 291 292 293 294 295
  }

  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
296
    gst_buffer_pool_config_set (config, target, size, min, max, prefix,
Wim Taymans's avatar
Wim Taymans committed
297 298 299 300 301 302 303 304 305 306 307
        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
308 309 310 311 312 313
  gst_caps_unref (target);

  return TRUE;

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

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

Wim Taymans's avatar
Wim Taymans committed
325
  goom = GST_GOOM (gst_pad_get_parent (pad));
Wim Taymans's avatar
Wim Taymans committed
326

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

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

      /* 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
368 369 370 371 372 373 374 375
    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
376 377 378 379 380 381 382
    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
383
    case GST_EVENT_SEGMENT:
Wim Taymans's avatar
Wim Taymans committed
384
    {
Wim Taymans's avatar
Wim Taymans committed
385 386 387
      /* 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
388
      gst_event_copy_segment (event, &goom->segment);
Wim Taymans's avatar
Wim Taymans committed
389 390 391

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

Wim Taymans's avatar
Wim Taymans committed
399 400
  return res;
}
Wim Taymans's avatar
Wim Taymans committed
401

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

420 421 422
      if (goom->rate == 0)
        break;

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

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

Wim Taymans's avatar
Wim Taymans committed
473 474
  /* 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
475 476 477 478 479
    if (!gst_goom_src_negotiate (goom))
      return GST_FLOW_NOT_NEGOTIATED;
  }
  return GST_FLOW_OK;
}
480

Wim Taymans's avatar
Wim Taymans committed
481

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

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

Wim Taymans's avatar
Wim Taymans committed
495 496 497 498 499
  /* 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
500 501 502 503 504
  }

  /* 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
505 506
  }

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

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

  ret = GST_FLOW_OK;

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

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

Wim Taymans's avatar
Wim Taymans committed
535 536
    GST_DEBUG_OBJECT (goom, "processing buffer");

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

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

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

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

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

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

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

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

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

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

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

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

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

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

beach:
  gst_object_unref (goom);

  return ret;
628 629
}

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

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

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

  switch (transition) {
649
    case GST_STATE_CHANGE_PAUSED_TO_READY:
Wim Taymans's avatar
Wim Taymans committed
650 651 652 653 654
      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
655
      break;
656
    case GST_STATE_CHANGE_READY_TO_NULL:
Wim Taymans's avatar
Wim Taymans committed
657 658 659 660
      break;
    default:
      break;
  }
661

Wim Taymans's avatar
Wim Taymans committed
662
  return ret;
663 664
}

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

670 671 672 673
#if HAVE_ORC
  orc_init ();
#endif

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

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