gstgoom.c 18 KB
Newer Older
1 2
/* gstgoom.c: implementation of goom drawing element
 * Copyright (C) <2001> Richard Boulton <richard@tartarus.org>
Wim Taymans's avatar
Wim Taymans committed
3
 *           (C) <2006> Wim Taymans <wim at fluendo dot com>
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

Wim Taymans's avatar
Wim Taymans committed
101 102
static gboolean gst_goom_src_query (GstPad * pad, GstObject * parent,
    GstQuery * query);
103

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

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

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

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

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

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

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

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

148 149
  goom->adapter = gst_adapter_new ();

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

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

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

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

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

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

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

  structure = gst_caps_get_structure (caps, 0);
194

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  return TRUE;

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

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

Wim Taymans's avatar
Wim Taymans committed
328
  goom = GST_GOOM (GST_PAD_PARENT (pad));
Wim Taymans's avatar
Wim Taymans committed
329

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

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

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

  return res;
}

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

Wim Taymans's avatar
Wim Taymans committed
367
  goom = GST_GOOM (GST_PAD_PARENT (pad));
Wim Taymans's avatar
Wim Taymans committed
368 369

  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

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

403
static gboolean
Wim Taymans's avatar
Wim Taymans committed
404
gst_goom_src_query (GstPad * pad, GstObject * parent, GstQuery * query)
405
{
406
  gboolean res = FALSE;
407 408
  GstGoom *goom;

Wim Taymans's avatar
Wim Taymans committed
409
  goom = GST_GOOM (parent);
410 411 412 413 414 415 416 417 418 419 420

  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;

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

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

        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:
Wim Taymans's avatar
Wim Taymans committed
454
      res = gst_pad_query_default (pad, parent, query);
455 456 457
      break;
  }

458 459 460
  return res;
}

Wim Taymans's avatar
Wim Taymans committed
461
/* make sure we are negotiated */
Wim Taymans's avatar
Wim Taymans committed
462
static GstFlowReturn
Wim Taymans's avatar
Wim Taymans committed
463
ensure_negotiated (GstGoom * goom)
Wim Taymans's avatar
Wim Taymans committed
464
{
Wim Taymans's avatar
Wim Taymans committed
465 466
  gboolean reconfigure;

Wim Taymans's avatar
Wim Taymans committed
467
  reconfigure = gst_pad_check_reconfigure (goom->srcpad);
Wim Taymans's avatar
Wim Taymans committed
468

Wim Taymans's avatar
Wim Taymans committed
469 470
  /* 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
471 472 473 474 475
    if (!gst_goom_src_negotiate (goom))
      return GST_FLOW_NOT_NEGOTIATED;
  }
  return GST_FLOW_OK;
}
476

Wim Taymans's avatar
Wim Taymans committed
477

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

Wim Taymans's avatar
Wim Taymans committed
485
  goom = GST_GOOM (GST_PAD_PARENT (pad));
486 487 488 489
  if (goom->bps == 0) {
    ret = GST_FLOW_NOT_NEGOTIATED;
    goto beach;
  }
Wim Taymans's avatar
Wim Taymans committed
490

Wim Taymans's avatar
Wim Taymans committed
491 492 493 494 495
  /* 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
496 497 498 499 500
  }

  /* 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
501 502
  }

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

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

  ret = GST_FLOW_OK;

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

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

Wim Taymans's avatar
Wim Taymans committed
530 531
    GST_DEBUG_OBJECT (goom, "processing buffer");

532 533 534 535 536 537 538 539
    /* 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);
    }

Stefan Sauer's avatar
Stefan Sauer committed
540
    if (GST_CLOCK_TIME_IS_VALID (timestamp)) {
Wim Taymans's avatar
Wim Taymans committed
541
      gint64 qostime;
Stefan Sauer's avatar
Stefan Sauer committed
542
      gboolean need_skip;
Wim Taymans's avatar
Wim Taymans committed
543 544

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

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

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

565 566 567 568 569 570 571 572 573 574 575
    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++;
      }
    }
576

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

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

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

Wim Taymans's avatar
Wim Taymans committed
594
    gst_adapter_unmap (goom->adapter);
Wim Taymans's avatar
Wim Taymans committed
595 596

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

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

Wim Taymans's avatar
Wim Taymans committed
603
  skip:
604 605 606 607 608 609 610
    /* 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
611 612 613

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

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

beach:

  return ret;
622 623
}

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

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

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

  switch (transition) {
643
    case GST_STATE_CHANGE_PAUSED_TO_READY:
Wim Taymans's avatar
Wim Taymans committed
644 645
      if (goom->pool) {
        gst_buffer_pool_set_active (goom->pool, FALSE);
Stefan Sauer's avatar
Stefan Sauer committed
646
        gst_object_replace ((GstObject **) & goom->pool, NULL);
Wim Taymans's avatar
Wim Taymans committed
647
      }
Wim Taymans's avatar
Wim Taymans committed
648
      break;
649
    case GST_STATE_CHANGE_READY_TO_NULL:
Wim Taymans's avatar
Wim Taymans committed
650 651 652 653
      break;
    default:
      break;
  }
654

Wim Taymans's avatar
Wim Taymans committed
655
  return ret;
656 657
}

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

663 664 665 666
#if HAVE_ORC
  orc_init ();
#endif

Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
667
  return gst_element_register (plugin, "goom", GST_RANK_NONE, GST_TYPE_GOOM);
668 669
}

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