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
 * |[
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
static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("src",
    GST_PAD_SRC,
    GST_PAD_ALWAYS,
Wim Taymans's avatar
Wim Taymans committed
75 76 77 78 79
#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
80
    );
81

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


Wim Taymans's avatar
Wim Taymans committed
94
static void gst_goom_finalize (GObject * object);
95

96 97
static GstStateChangeReturn gst_goom_change_state (GstElement * element,
    GstStateChange transition);
98

Wim Taymans's avatar
Wim Taymans committed
99
static GstFlowReturn gst_goom_chain (GstPad * pad, GstBuffer * buffer);
Wim Taymans's avatar
Wim Taymans committed
100 101
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
102

103 104
static gboolean gst_goom_src_query (GstPad * pad, GstQuery * query);

Wim Taymans's avatar
Wim Taymans committed
105 106
#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
107

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

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

Wim Taymans's avatar
Wim Taymans committed
117
  gobject_class->finalize = gst_goom_finalize;
118

Wim Taymans's avatar
Wim Taymans committed
119 120 121 122 123 124 125 126 127
  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));

128
  gstelement_class->change_state = GST_DEBUG_FUNCPTR (gst_goom_change_state);
129 130 131
}

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

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

149 150
  goom->adapter = gst_adapter_new ();

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

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

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

167
  goom_close (goom->plugin);
168
  goom->plugin = NULL;
169

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

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

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

  structure = gst_caps_get_structure (caps, 0);
195

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

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

Wim Taymans's avatar
Wim Taymans committed
201
  return TRUE;
Wim Taymans's avatar
Wim Taymans committed
202 203
}

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

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

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

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

  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
229 230 231 232 233 234 235 236 237 238
  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;
  }
239 240
}

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

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

253 254
  GST_DEBUG_OBJECT (goom, "performing negotiation");

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

262
    if (gst_caps_is_empty (target))
Wim Taymans's avatar
Wim Taymans committed
263 264
      goto no_format;

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

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

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

  /* 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;
290
    alignment = 0;
Wim Taymans's avatar
Wim Taymans committed
291 292 293 294 295 296 297 298 299
  }

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

  return TRUE;

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

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

Wim Taymans's avatar
Wim Taymans committed
329
  goom = GST_GOOM (gst_pad_get_parent (pad));
Wim Taymans's avatar
Wim Taymans committed
330

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

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

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

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

Wim Taymans's avatar
Wim Taymans committed
403 404
  return res;
}
Wim Taymans's avatar
Wim Taymans committed
405

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

424 425 426
      if (goom->rate == 0)
        break;

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

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

Wim Taymans's avatar
Wim Taymans committed
477 478
  /* 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
479 480 481 482 483
    if (!gst_goom_src_negotiate (goom))
      return GST_FLOW_NOT_NEGOTIATED;
  }
  return GST_FLOW_OK;
}
484

Wim Taymans's avatar
Wim Taymans committed
485

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

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

Wim Taymans's avatar
Wim Taymans committed
499 500 501 502 503
  /* 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
504 505 506 507 508
  }

  /* 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
  GST_DEBUG_OBJECT (goom,
      "Input buffer has %d samples, time=%" G_GUINT64_FORMAT,
Wim Taymans's avatar
Wim Taymans committed
513
      gst_buffer_get_size (buffer) / goom->bps, GST_BUFFER_TIMESTAMP (buffer));
Wim Taymans's avatar
Wim Taymans committed
514

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
    data =
Wim Taymans's avatar
Wim Taymans committed
571
        (const guint16 *) gst_adapter_map (goom->adapter,
Wim Taymans's avatar
Wim Taymans committed
572
        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
    /* 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
589 590
      GST_DEBUG_OBJECT (goom, "allocating output buffer");
      ret = gst_buffer_pool_acquire_buffer (goom->pool, &outbuf, NULL);
Wim Taymans's avatar
Wim Taymans committed
591
      if (ret != GST_FLOW_OK) {
Wim Taymans's avatar
Wim Taymans committed
592
        gst_adapter_unmap (goom->adapter, 0);
Wim Taymans's avatar
Wim Taymans committed
593 594 595
        goto beach;
      }
    }
Wim Taymans's avatar
Wim Taymans committed
596

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

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

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

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

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

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

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

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

beach:
  gst_object_unref (goom);

  return ret;
632 633
}

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

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

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

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

Wim Taymans's avatar
Wim Taymans committed
666
  return ret;
667 668
}

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

674 675 676 677
#if HAVE_ORC
  orc_init ();
#endif

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

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