gst/imagefreeze/gstimagefreeze.c

   1 /* GStreamer
   2  * Copyright (c) 2005 Edward Hervey <bilboed@bilboed.com>
   3  * Copyright (C) 2010 Sebastian Dröge <sebastian.droege@collabora.co.uk>
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Library General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Library General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Library General Public
  16  * License along with this library; if not, write to the
  17  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18  * Boston, MA 02111-1307, USA.
  19  */
  20
  21 /**
  22  * SECTION:element-imagefreeze
  23  *
  24  * The imagefreeze element generates a still frame video stream from
  25  * the input. It duplicates the first frame with the framerate requested
  26  * by downstream, allows seeking and answers queries.
  27  *
  28  * <refsect2>
  29  * <title>Example launch line</title>
  30  * |[
  31  * gst-launch -v filesrc location=some.png ! decodebin2 ! imagefreeze ! autovideosink
  32  * ]| This pipeline shows a still frame stream of a PNG file.
  33  * </refsect2>
  34  */
  35
  36 /* This is based on the imagefreeze element from PiTiVi:
  37  * http://git.gnome.org/browse/pitivi/tree/pitivi/elements/imagefreeze.py
  38  */
  39
  40 #ifdef HAVE_CONFIG_H
  41 #include "config.h"
  42 #endif
  43
  44 #include <gst/glib-compat-private.h>
  45
  46 #include "gstimagefreeze.h"
  47
  48 static void gst_image_freeze_finalize (GObject * object);
  49
  50 static void gst_image_freeze_reset (GstImageFreeze * self);
  51
  52 static GstStateChangeReturn gst_image_freeze_change_state (GstElement * element,
  53     GstStateChange transition);
  54
  55 static GstFlowReturn gst_image_freeze_sink_chain (GstPad * pad,
  56     GstBuffer * buffer);
  57 static gboolean gst_image_freeze_sink_event (GstPad * pad, GstEvent * event);
  58 static gboolean gst_image_freeze_sink_setcaps (GstPad * pad, GstCaps * caps);
  59 static GstCaps *gst_image_freeze_sink_getcaps (GstPad * pad);
  60 static gboolean gst_image_freeze_sink_query (GstPad * pad, GstQuery * query);
  61 static GstFlowReturn gst_image_freeze_sink_bufferalloc (GstPad * pad,
  62     guint64 offset, guint size, GstCaps * caps, GstBuffer ** buf);
  63 static void gst_image_freeze_src_loop (GstPad * pad);
  64 static gboolean gst_image_freeze_src_event (GstPad * pad, GstEvent * event);
  65 static gboolean gst_image_freeze_src_query (GstPad * pad, GstQuery * query);
  66 static const GstQueryType *gst_image_freeze_src_query_type (GstPad * pad);
  67
  68 static GstStaticPadTemplate sink_pad_template = GST_STATIC_PAD_TEMPLATE ("sink",
  69     GST_PAD_SINK,
  70     GST_PAD_ALWAYS,
  71     GST_STATIC_CAPS ("video/x-raw-yuv; video/x-raw-rgb; video/x-raw-gray"));
  72
  73 static GstStaticPadTemplate src_pad_template =
  74     GST_STATIC_PAD_TEMPLATE ("src", GST_PAD_SRC, GST_PAD_ALWAYS,
  75     GST_STATIC_CAPS ("video/x-raw-yuv; video/x-raw-rgb; video/x-raw-gray"));
  76
  77 GST_DEBUG_CATEGORY_STATIC (gst_image_freeze_debug);
  78 #define GST_CAT_DEFAULT gst_image_freeze_debug
  79
  80 GST_BOILERPLATE (GstImageFreeze, gst_image_freeze, GstElement,
  81     GST_TYPE_ELEMENT);
  82
  83 static void
  84 gst_image_freeze_base_init (gpointer g_class)
  85 {
  86   GstElementClass *gstelement_class = GST_ELEMENT_CLASS (g_class);
  87
  88   gst_element_class_set_details_simple (gstelement_class,
  89       "Still frame stream generator",
  90       "Filter/Video",
  91       "Generates a still frame stream from an image",
  92       "Sebastian Dröge <sebastian.droege@collabora.co.uk>");
  93
  94   gst_element_class_add_pad_template (gstelement_class,
  95       gst_static_pad_template_get (&sink_pad_template));
  96   gst_element_class_add_pad_template (gstelement_class,
  97       gst_static_pad_template_get (&src_pad_template));
  98 }
  99
 100 static void
 101 gst_image_freeze_class_init (GstImageFreezeClass * klass)
 102 {
 103   GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
 104   GstElementClass *gstelement_class = GST_ELEMENT_CLASS (klass);
 105
 106   gobject_class->finalize = gst_image_freeze_finalize;
 107
 108   gstelement_class->change_state =
 109       GST_DEBUG_FUNCPTR (gst_image_freeze_change_state);
 110 }
 111
 112 static void
 113 gst_image_freeze_init (GstImageFreeze * self, GstImageFreezeClass * g_class)
 114 {
 115   self->sinkpad = gst_pad_new_from_static_template (&sink_pad_template, "sink");
 116   gst_pad_set_chain_function (self->sinkpad,
 117       GST_DEBUG_FUNCPTR (gst_image_freeze_sink_chain));
 118   gst_pad_set_event_function (self->sinkpad,
 119       GST_DEBUG_FUNCPTR (gst_image_freeze_sink_event));
 120   gst_pad_set_query_function (self->sinkpad,
 121       GST_DEBUG_FUNCPTR (gst_image_freeze_sink_query));
 122   gst_pad_set_setcaps_function (self->sinkpad,
 123       GST_DEBUG_FUNCPTR (gst_image_freeze_sink_setcaps));
 124   gst_pad_set_getcaps_function (self->sinkpad,
 125       GST_DEBUG_FUNCPTR (gst_image_freeze_sink_getcaps));
 126   gst_pad_set_bufferalloc_function (self->sinkpad,
 127       GST_DEBUG_FUNCPTR (gst_image_freeze_sink_bufferalloc));
 128   gst_element_add_pad (GST_ELEMENT (self), self->sinkpad);
 129
 130   self->srcpad = gst_pad_new_from_static_template (&src_pad_template, "src");
 131   gst_pad_set_event_function (self->srcpad,
 132       GST_DEBUG_FUNCPTR (gst_image_freeze_src_event));
 133   gst_pad_set_query_function (self->srcpad,
 134       GST_DEBUG_FUNCPTR (gst_image_freeze_src_query));
 135   gst_pad_set_query_type_function (self->srcpad,
 136       GST_DEBUG_FUNCPTR (gst_image_freeze_src_query_type));
 137   gst_pad_use_fixed_caps (self->srcpad);
 138   gst_element_add_pad (GST_ELEMENT (self), self->srcpad);
 139
 140   self->lock = g_mutex_new ();
 141
 142   gst_image_freeze_reset (self);
 143 }
 144
 145 static void
 146 gst_image_freeze_finalize (GObject * object)
 147 {
 148   GstImageFreeze *self = GST_IMAGE_FREEZE (object);
 149
 150   gst_image_freeze_reset (self);
 151
 152   if (self->lock)
 153     g_mutex_free (self->lock);
 154   self->lock = NULL;
 155
 156   G_OBJECT_CLASS (parent_class)->finalize (object);
 157 }
 158
 159 static void
 160 gst_image_freeze_reset (GstImageFreeze * self)
 161 {
 162   GST_DEBUG_OBJECT (self, "Resetting internal state");
 163
 164   g_mutex_lock (self->lock);
 165   gst_buffer_replace (&self->buffer, NULL);
 166
 167   gst_segment_init (&self->segment, GST_FORMAT_TIME);
 168   self->need_segment = TRUE;
 169   gst_event_replace (&self->close_segment, NULL);
 170
 171   self->fps_n = self->fps_d = 0;
 172   self->offset = 0;
 173   g_mutex_unlock (self->lock);
 174
 175   g_atomic_int_set (&self->seeking, 0);
 176 }
 177
 178 static gboolean
 179 gst_image_freeze_sink_setcaps (GstPad * pad, GstCaps * caps)
 180 {
 181   GstImageFreeze *self = GST_IMAGE_FREEZE (gst_pad_get_parent (pad));
 182   gboolean ret = FALSE;
 183   GstStructure *s;
 184   gint fps_n, fps_d;
 185   GstCaps *othercaps, *intersection;
 186   guint i, n;
 187
 188   caps = gst_caps_make_writable (gst_caps_ref (caps));
 189
 190   GST_DEBUG_OBJECT (pad, "Setting caps: %" GST_PTR_FORMAT, caps);
 191
 192   s = gst_caps_get_structure (caps, 0);
 193
 194   /* 1. Remove framerate */
 195   gst_structure_remove_field (s, "framerate");
 196   gst_structure_set (s, "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, G_MAXINT, 1,
 197       NULL);
 198
 199   /* 2. Intersect with template caps */
 200   othercaps = (GstCaps *) gst_pad_get_pad_template_caps (pad);
 201   intersection = gst_caps_intersect (caps, othercaps);
 202   GST_DEBUG_OBJECT (pad, "Intersecting: %" GST_PTR_FORMAT, caps);
 203   GST_DEBUG_OBJECT (pad, "with: %" GST_PTR_FORMAT, othercaps);
 204   GST_DEBUG_OBJECT (pad, "gave: %" GST_PTR_FORMAT, intersection);
 205   gst_caps_unref (caps);
 206   caps = intersection;
 207   intersection = othercaps = NULL;
 208
 209   /* 3. Intersect with downstream peer caps */
 210   othercaps = gst_pad_peer_get_caps (self->srcpad);
 211   if (othercaps) {
 212     intersection = gst_caps_intersect (caps, othercaps);
 213     GST_DEBUG_OBJECT (pad, "Intersecting: %" GST_PTR_FORMAT, caps);
 214     GST_DEBUG_OBJECT (pad, "with: %" GST_PTR_FORMAT, othercaps);
 215     GST_DEBUG_OBJECT (pad, "gave: %" GST_PTR_FORMAT, intersection);
 216     gst_caps_unref (othercaps);
 217     gst_caps_unref (caps);
 218     caps = intersection;
 219     intersection = othercaps = NULL;
 220   }
 221
 222   /* 4. For every candidate check if it's accepted downstream
 223    * and fixate framerate to nearest 25/1 */
 224   n = gst_caps_get_size (caps);
 225   for (i = 0; i < n; i++) {
 226     GstCaps *candidate = gst_caps_new_empty ();
 227     GstStructure *s = gst_structure_copy (gst_caps_get_structure (caps, i));
 228
 229     gst_caps_append_structure (candidate, s);
 230     if (gst_pad_peer_accept_caps (self->srcpad, candidate)) {
 231       if (gst_structure_has_field_typed (s, "framerate", GST_TYPE_FRACTION) ||
 232           gst_structure_fixate_field_nearest_fraction (s, "framerate", 25, 1)) {
 233         gst_structure_get_fraction (s, "framerate", &fps_n, &fps_d);
 234         if (fps_d != 0) {
 235           g_mutex_lock (self->lock);
 236           self->fps_n = fps_n;
 237           self->fps_d = fps_d;
 238           g_mutex_unlock (self->lock);
 239           GST_DEBUG_OBJECT (pad, "Setting caps %" GST_PTR_FORMAT, candidate);
 240           gst_pad_set_caps (self->srcpad, candidate);
 241           gst_caps_unref (candidate);
 242           ret = TRUE;
 243           goto done;
 244         } else {
 245           GST_WARNING_OBJECT (pad, "Invalid caps with framerate %d/%d", fps_n,
 246               fps_d);
 247         }
 248       }
 249     }
 250     gst_caps_unref (candidate);
 251   }
 252
 253 done:
 254   if (!ret)
 255     GST_ERROR_OBJECT (pad, "No usable caps found");
 256
 257   gst_caps_unref (caps);
 258   gst_object_unref (self);
 259
 260   return ret;
 261 }
 262
 263 static GstCaps *
 264 gst_image_freeze_sink_getcaps (GstPad * pad)
 265 {
 266   GstImageFreeze *self = GST_IMAGE_FREEZE (gst_pad_get_parent (pad));
 267   GstCaps *ret, *tmp;
 268   guint i, n;
 269
 270   if (GST_PAD_CAPS (pad)) {
 271     ret = gst_caps_copy (GST_PAD_CAPS (pad));
 272     goto done;
 273   }
 274
 275   tmp = gst_pad_peer_get_caps (self->srcpad);
 276   if (tmp) {
 277     ret = gst_caps_intersect (tmp, gst_pad_get_pad_template_caps (pad));
 278     gst_caps_unref (tmp);
 279   } else {
 280     ret = gst_caps_copy (gst_pad_get_pad_template_caps (pad));
 281   }
 282
 283   n = gst_caps_get_size (ret);
 284   for (i = 0; i < n; i++) {
 285     GstStructure *s = gst_caps_get_structure (ret, i);
 286
 287     gst_structure_remove_field (s, "framerate");
 288     gst_structure_set (s, "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, G_MAXINT,
 289         1, NULL);
 290   }
 291
 292 done:
 293   gst_object_unref (self);
 294
 295   GST_LOG_OBJECT (pad, "Returning caps: %" GST_PTR_FORMAT, ret);
 296
 297   return ret;
 298 }
 299
 300 static gboolean
 301 gst_image_freeze_sink_query (GstPad * pad, GstQuery * query)
 302 {
 303   GstImageFreeze *self = GST_IMAGE_FREEZE (gst_pad_get_parent (pad));
 304   gboolean ret;
 305   GstPad *peer = gst_pad_get_peer (self->srcpad);
 306
 307   GST_LOG_OBJECT (pad, "Handling query of type '%s'",
 308       gst_query_type_get_name (GST_QUERY_TYPE (query)));
 309
 310   if (!peer) {
 311     GST_INFO_OBJECT (pad, "No peer yet, dropping query");
 312     ret = FALSE;
 313   } else {
 314     ret = gst_pad_query (peer, query);
 315     gst_object_unref (peer);
 316   }
 317
 318   gst_object_unref (self);
 319   return ret;
 320 }
 321
 322 static GstFlowReturn
 323 gst_image_freeze_sink_bufferalloc (GstPad * pad, guint64 offset, guint size,
 324     GstCaps * caps, GstBuffer ** buf)
 325 {
 326   GstImageFreeze *self = GST_IMAGE_FREEZE (gst_pad_get_parent (pad));
 327   GstFlowReturn ret;
 328   gboolean do_alloc;
 329
 330   GST_LOG_OBJECT (pad, "Allocating buffer with offset 0x%" G_GINT64_MODIFIER
 331       "x and size %u with caps: %" GST_PTR_FORMAT, offset, size, caps);
 332
 333   *buf = NULL;
 334
 335   g_mutex_lock (self->lock);
 336   do_alloc = self->buffer == NULL;
 337   g_mutex_unlock (self->lock);
 338
 339   if (do_alloc) {
 340     gboolean seeking = FALSE;
 341
 342     do {
 343       GST_PAD_STREAM_LOCK (self->srcpad);
 344       ret = gst_pad_alloc_buffer (self->srcpad, offset, size, caps, buf);
 345
 346       seeking = ret == GST_FLOW_WRONG_STATE
 347           && g_atomic_int_get (&self->seeking);
 348       GST_PAD_STREAM_UNLOCK (self->srcpad);
 349     } while (seeking);
 350
 351     if (G_UNLIKELY (ret != GST_FLOW_OK))
 352       GST_ERROR_OBJECT (pad, "Allocating buffer failed: %s",
 353           gst_flow_get_name (ret));
 354   } else {
 355     /* Let upstream go EOS if we already have a buffer */
 356     ret = GST_FLOW_EOS;
 357   }
 358
 359   gst_object_unref (self);
 360
 361   return ret;
 362 }
 363
 364 static gboolean
 365 gst_image_freeze_convert (GstImageFreeze * self,
 366     GstFormat src_format, gint64 src_value,
 367     GstFormat * dest_format, gint64 * dest_value)
 368 {
 369   gboolean ret = FALSE;
 370
 371   if (src_format == *dest_format) {
 372     *dest_value = src_value;
 373     return TRUE;
 374   }
 375
 376   if (src_value == -1) {
 377     *dest_value = -1;
 378     return TRUE;
 379   }
 380
 381   switch (src_format) {
 382     case GST_FORMAT_DEFAULT:{
 383       switch (*dest_format) {
 384         case GST_FORMAT_TIME:
 385           g_mutex_lock (self->lock);
 386           if (self->fps_n == 0)
 387             *dest_value = -1;
 388           else
 389             *dest_value =
 390                 gst_util_uint64_scale (src_value, GST_SECOND * self->fps_d,
 391                 self->fps_n);
 392           g_mutex_unlock (self->lock);
 393           ret = TRUE;
 394           break;
 395         default:
 396           break;
 397       }
 398       break;
 399     }
 400     case GST_FORMAT_TIME:{
 401       switch (*dest_format) {
 402         case GST_FORMAT_DEFAULT:
 403           g_mutex_lock (self->lock);
 404           *dest_value =
 405               gst_util_uint64_scale (src_value, self->fps_n,
 406               self->fps_d * GST_SECOND);
 407           g_mutex_unlock (self->lock);
 408           ret = TRUE;
 409           break;
 410         default:
 411           break;
 412       }
 413       break;
 414     }
 415     default:
 416       break;
 417   }
 418
 419   return ret;
 420 }
 421
 422 static const GstQueryType *
 423 gst_image_freeze_src_query_type (GstPad * pad)
 424 {
 425   static const GstQueryType types[] = {
 426     GST_QUERY_POSITION,
 427     GST_QUERY_DURATION,
 428     GST_QUERY_SEEKING,
 429     GST_QUERY_CONVERT,
 430     0
 431   };
 432
 433   return types;
 434 }
 435
 436 static gboolean
 437 gst_image_freeze_src_query (GstPad * pad, GstQuery * query)
 438 {
 439   GstImageFreeze *self = GST_IMAGE_FREEZE (gst_pad_get_parent (pad));
 440   gboolean ret = FALSE;
 441
 442   GST_LOG_OBJECT (pad, "Handling query of type '%s'",
 443       gst_query_type_get_name (GST_QUERY_TYPE (query)));
 444
 445   switch (GST_QUERY_TYPE (query)) {
 446     case GST_QUERY_CONVERT:{
 447       GstFormat src_format, dest_format;
 448       gint64 src_value, dest_value;
 449
 450       gst_query_parse_convert (query, &src_format, &src_value, &dest_format,
 451           &dest_value);
 452       ret =
 453           gst_image_freeze_convert (self, src_format, src_value, &dest_format,
 454           &dest_value);
 455       if (ret)
 456         gst_query_set_convert (query, src_format, src_value, dest_format,
 457             dest_value);
 458       break;
 459     }
 460     case GST_QUERY_POSITION:{
 461       GstFormat format;
 462       gint64 position;
 463
 464       gst_query_parse_position (query, &format, NULL);
 465       switch (format) {
 466         case GST_FORMAT_DEFAULT:{
 467           g_mutex_lock (self->lock);
 468           position = self->offset;
 469           g_mutex_unlock (self->lock);
 470           ret = TRUE;
 471           break;
 472         }
 473         case GST_FORMAT_TIME:{
 474           g_mutex_lock (self->lock);
 475           position = self->segment.last_stop;
 476           g_mutex_unlock (self->lock);
 477           ret = TRUE;
 478           break;
 479         }
 480         default:
 481           break;
 482       }
 483
 484       if (ret) {
 485         gst_query_set_position (query, format, position);
 486         GST_DEBUG_OBJECT (pad,
 487             "Returning position %" G_GINT64_FORMAT " in format %s", position,
 488             gst_format_get_name (format));
 489       } else {
 490         GST_DEBUG_OBJECT (pad, "Position query failed");
 491       }
 492       break;
 493     }
 494     case GST_QUERY_DURATION:{
 495       GstFormat format;
 496       gint64 duration;
 497
 498       gst_query_parse_duration (query, &format, NULL);
 499       switch (format) {
 500         case GST_FORMAT_TIME:{
 501           g_mutex_lock (self->lock);
 502           duration = self->segment.stop;
 503           g_mutex_unlock (self->lock);
 504           ret = TRUE;
 505           break;
 506         }
 507         case GST_FORMAT_DEFAULT:{
 508           g_mutex_lock (self->lock);
 509           duration = self->segment.stop;
 510           if (duration != -1)
 511             duration =
 512                 gst_util_uint64_scale (duration, self->fps_n,
 513                 GST_SECOND * self->fps_d);
 514           g_mutex_unlock (self->lock);
 515           ret = TRUE;
 516           break;
 517         }
 518         default:
 519           break;
 520       }
 521
 522       if (ret) {
 523         gst_query_set_duration (query, format, duration);
 524         GST_DEBUG_OBJECT (pad,
 525             "Returning duration %" G_GINT64_FORMAT " in format %s", duration,
 526             gst_format_get_name (format));
 527       } else {
 528         GST_DEBUG_OBJECT (pad, "Duration query failed");
 529       }
 530       break;
 531     }
 532     case GST_QUERY_SEEKING:{
 533       GstFormat format;
 534       gboolean seekable;
 535
 536       gst_query_parse_seeking (query, &format, NULL, NULL, NULL);
 537       seekable = (format == GST_FORMAT_TIME || format == GST_FORMAT_DEFAULT);
 538
 539       gst_query_set_seeking (query, format, seekable, (seekable ? 0 : -1), -1);
 540       ret = TRUE;
 541       break;
 542     }
 543     default:
 544       ret = FALSE;
 545       break;
 546   }
 547
 548   gst_object_unref (self);
 549   return ret;
 550 }
 551
 552
 553 static gboolean
 554 gst_image_freeze_sink_event (GstPad * pad, GstEvent * event)
 555 {
 556   GstImageFreeze *self = GST_IMAGE_FREEZE (gst_pad_get_parent (pad));
 557   gboolean ret;
 558
 559   GST_LOG_OBJECT (pad, "Got %s event", GST_EVENT_TYPE_NAME (event));
 560
 561   switch (GST_EVENT_TYPE (event)) {
 562     case GST_EVENT_EOS:
 563       if (!self->buffer) {
 564         /* if we receive EOS before a buffer arrives, then let it pass */
 565         GST_DEBUG_OBJECT (self, "EOS without input buffer, passing on");
 566         ret = gst_pad_push_event (self->srcpad, event);
 567         break;
 568       }
 569       /* fall-through */
 570     case GST_EVENT_NEWSEGMENT:
 571       GST_DEBUG_OBJECT (pad, "Dropping event");
 572       gst_event_unref (event);
 573       ret = TRUE;
 574       break;
 575     case GST_EVENT_FLUSH_START:
 576       gst_image_freeze_reset (self);
 577       /* fall through */
 578     default:
 579       ret = gst_pad_push_event (self->srcpad, event);
 580       break;
 581   }
 582
 583   gst_object_unref (self);
 584   return ret;
 585 }
 586
 587 static gboolean
 588 gst_image_freeze_src_event (GstPad * pad, GstEvent * event)
 589 {
 590   GstImageFreeze *self = GST_IMAGE_FREEZE (gst_pad_get_parent (pad));
 591   gboolean ret;
 592
 593   GST_LOG_OBJECT (pad, "Got %s event", GST_EVENT_TYPE_NAME (event));
 594
 595   switch (GST_EVENT_TYPE (event)) {
 596     case GST_EVENT_NAVIGATION:
 597     case GST_EVENT_QOS:
 598     case GST_EVENT_LATENCY:
 599     case GST_EVENT_STEP:
 600       GST_DEBUG_OBJECT (pad, "Dropping event");
 601       gst_event_unref (event);
 602       ret = TRUE;
 603       break;
 604     case GST_EVENT_SEEK:{
 605       gdouble rate;
 606       GstFormat format;
 607       GstSeekFlags flags;
 608       GstSeekType start_type, stop_type;
 609       gint64 start, stop;
 610       gint64 last_stop;
 611       gboolean start_task;
 612       gboolean flush;
 613
 614       gst_event_parse_seek (event, &rate, &format, &flags, &start_type, &start,
 615           &stop_type, &stop);
 616       gst_event_unref (event);
 617
 618       flush = ! !(flags & GST_SEEK_FLAG_FLUSH);
 619
 620       if (format != GST_FORMAT_TIME && format != GST_FORMAT_DEFAULT) {
 621         GST_ERROR_OBJECT (pad, "Seek in invalid format: %s",
 622             gst_format_get_name (format));
 623         ret = FALSE;
 624         break;
 625       }
 626
 627       if (format == GST_FORMAT_DEFAULT) {
 628         format = GST_FORMAT_TIME;
 629         if (!gst_image_freeze_convert (self, GST_FORMAT_DEFAULT, start, &format,
 630                 &start)
 631             || !gst_image_freeze_convert (self, GST_FORMAT_DEFAULT, stop,
 632                 &format, &stop)
 633             || start == -1 || stop == -1) {
 634           GST_ERROR_OBJECT (pad,
 635               "Failed to convert seek from DEFAULT format into TIME format");
 636           ret = FALSE;
 637           break;
 638         }
 639       }
 640
 641       if (flush) {
 642         GstEvent *e;
 643
 644         g_atomic_int_set (&self->seeking, 1);
 645         e = gst_event_new_flush_start ();
 646         gst_pad_push_event (self->srcpad, e);
 647       } else {
 648         gst_pad_pause_task (self->srcpad);
 649       }
 650
 651       GST_PAD_STREAM_LOCK (self->srcpad);
 652
 653       g_mutex_lock (self->lock);
 654       gst_event_replace (&self->close_segment, NULL);
 655       if (!flush) {
 656         if (!self->need_segment && self->segment.rate >= 0) {
 657           self->close_segment =
 658               gst_event_new_new_segment_full (TRUE, self->segment.rate,
 659               self->segment.applied_rate, self->segment.format,
 660               self->segment.start, self->segment.last_stop, self->segment.time);
 661         } else if (!self->need_segment) {
 662           gint64 stop;
 663
 664           if ((stop = self->segment.stop) == -1)
 665             stop = self->segment.duration;
 666
 667           self->close_segment =
 668               gst_event_new_new_segment_full (TRUE, self->segment.rate,
 669               self->segment.applied_rate, self->segment.format,
 670               self->segment.last_stop, stop, self->segment.last_stop);
 671         }
 672       }
 673
 674       gst_segment_set_seek (&self->segment, rate, format, flags, start_type,
 675           start, stop_type, stop, NULL);
 676       self->need_segment = TRUE;
 677       last_stop = self->segment.last_stop;
 678
 679       start_task = self->buffer != NULL;
 680       g_mutex_unlock (self->lock);
 681
 682       if (flush) {
 683         GstEvent *e;
 684
 685         e = gst_event_new_flush_stop ();
 686         gst_pad_push_event (self->srcpad, e);
 687         g_atomic_int_set (&self->seeking, 0);
 688       }
 689
 690       if (flags & GST_SEEK_FLAG_SEGMENT) {
 691         GstMessage *m;
 692
 693         m = gst_message_new_segment_start (GST_OBJECT (self),
 694             format, last_stop);
 695         gst_element_post_message (GST_ELEMENT (self), m);
 696       }
 697
 698       GST_PAD_STREAM_UNLOCK (self->srcpad);
 699
 700       GST_DEBUG_OBJECT (pad, "Seek successful");
 701
 702       if (start_task) {
 703         g_mutex_lock (self->lock);
 704
 705         if (self->buffer != NULL)
 706           gst_pad_start_task (self->srcpad,
 707               (GstTaskFunction) gst_image_freeze_src_loop, self->srcpad);
 708
 709         g_mutex_unlock (self->lock);
 710       }
 711
 712       ret = TRUE;
 713       break;
 714     }
 715     case GST_EVENT_FLUSH_START:
 716       gst_image_freeze_reset (self);
 717       /* fall through */
 718     default:
 719       ret = gst_pad_push_event (self->sinkpad, event);
 720       break;
 721   }
 722
 723   gst_object_unref (self);
 724   return ret;
 725 }
 726
 727 static GstFlowReturn
 728 gst_image_freeze_sink_chain (GstPad * pad, GstBuffer * buffer)
 729 {
 730   GstImageFreeze *self = GST_IMAGE_FREEZE (GST_PAD_PARENT (pad));
 731
 732   g_mutex_lock (self->lock);
 733   if (self->buffer) {
 734     GST_DEBUG_OBJECT (pad, "Already have a buffer, dropping");
 735     gst_buffer_unref (buffer);
 736     g_mutex_unlock (self->lock);
 737     return GST_FLOW_EOS;
 738   }
 739
 740   self->buffer = buffer;
 741
 742   gst_pad_start_task (self->srcpad, (GstTaskFunction) gst_image_freeze_src_loop,
 743       self->srcpad);
 744   g_mutex_unlock (self->lock);
 745   return GST_FLOW_OK;
 746 }
 747
 748 static void
 749 gst_image_freeze_src_loop (GstPad * pad)
 750 {
 751   GstImageFreeze *self = GST_IMAGE_FREEZE (GST_PAD_PARENT (pad));
 752   GstBuffer *buffer;
 753   guint64 offset;
 754   GstClockTime timestamp, timestamp_end;
 755   gint64 cstart, cstop;
 756   gboolean in_seg, eos;
 757
 758   g_mutex_lock (self->lock);
 759   if (!self->buffer) {
 760     GST_ERROR_OBJECT (pad, "Have no buffer yet");
 761     g_mutex_unlock (self->lock);
 762     gst_pad_pause_task (self->srcpad);
 763     return;
 764   }
 765   buffer = gst_buffer_ref (self->buffer);
 766   buffer = gst_buffer_make_metadata_writable (buffer);
 767   g_mutex_unlock (self->lock);
 768
 769   if (self->close_segment) {
 770     GST_DEBUG_OBJECT (pad, "Closing previous segment");
 771     gst_pad_push_event (self->srcpad, self->close_segment);
 772     self->close_segment = NULL;
 773   }
 774
 775   if (self->need_segment) {
 776     GstEvent *e;
 777
 778     GST_DEBUG_OBJECT (pad, "Pushing NEWSEGMENT event: %" GST_SEGMENT_FORMAT,
 779         &self->segment);
 780     e = gst_event_new_new_segment_full (FALSE, self->segment.rate,
 781         self->segment.applied_rate, self->segment.format, self->segment.start,
 782         self->segment.stop, self->segment.start);
 783
 784     g_mutex_lock (self->lock);
 785     if (self->segment.rate >= 0) {
 786       self->offset =
 787           gst_util_uint64_scale (self->segment.start, self->fps_n,
 788           self->fps_d * GST_SECOND);
 789     } else {
 790       self->offset =
 791           gst_util_uint64_scale (self->segment.stop, self->fps_n,
 792           self->fps_d * GST_SECOND);
 793     }
 794     g_mutex_unlock (self->lock);
 795
 796     self->need_segment = FALSE;
 797
 798     gst_pad_push_event (self->srcpad, e);
 799   }
 800
 801   g_mutex_lock (self->lock);
 802   offset = self->offset;
 803
 804   if (self->fps_n != 0) {
 805     timestamp =
 806         gst_util_uint64_scale (offset, self->fps_d * GST_SECOND, self->fps_n);
 807     timestamp_end =
 808         gst_util_uint64_scale (offset + 1, self->fps_d * GST_SECOND,
 809         self->fps_n);
 810   } else {
 811     timestamp = self->segment.start;
 812     timestamp_end = GST_CLOCK_TIME_NONE;
 813   }
 814
 815   eos = (self->fps_n == 0 && offset > 0) ||
 816       (self->segment.rate >= 0 && self->segment.stop != -1
 817       && timestamp > self->segment.stop) || (self->segment.rate < 0
 818       && offset == 0) || (self->segment.rate < 0
 819       && self->segment.start != -1 && timestamp_end < self->segment.start);
 820
 821   if (self->fps_n == 0 && offset > 0)
 822     in_seg = FALSE;
 823   else
 824     in_seg =
 825         gst_segment_clip (&self->segment, GST_FORMAT_TIME, timestamp,
 826         timestamp_end, &cstart, &cstop);
 827
 828   if (in_seg)
 829     gst_segment_set_last_stop (&self->segment, GST_FORMAT_TIME, cstart);
 830
 831   if (self->segment.rate >= 0)
 832     self->offset++;
 833   else
 834     self->offset--;
 835   g_mutex_unlock (self->lock);
 836
 837   GST_DEBUG_OBJECT (pad, "Handling buffer with timestamp %" GST_TIME_FORMAT,
 838       GST_TIME_ARGS (timestamp));
 839
 840   if (in_seg) {
 841     GstFlowReturn ret;
 842
 843     GST_BUFFER_TIMESTAMP (buffer) = cstart;
 844     GST_BUFFER_DURATION (buffer) = cstop - cstart;
 845     GST_BUFFER_OFFSET (buffer) = offset;
 846     GST_BUFFER_OFFSET_END (buffer) = offset + 1;
 847     gst_buffer_set_caps (buffer, GST_PAD_CAPS (self->srcpad));
 848     ret = gst_pad_push (self->srcpad, buffer);
 849     GST_DEBUG_OBJECT (pad, "Pushing buffer resulted in %s",
 850         gst_flow_get_name (ret));
 851     if (ret != GST_FLOW_OK)
 852       gst_pad_pause_task (self->srcpad);
 853   } else {
 854     gst_buffer_unref (buffer);
 855   }
 856
 857   if (eos) {
 858     if ((self->segment.flags & GST_SEEK_FLAG_SEGMENT)) {
 859       GstMessage *m;
 860
 861       GST_DEBUG_OBJECT (pad, "Sending segment done at end of segment");
 862       if (self->segment.rate >= 0)
 863         m = gst_message_new_segment_done (GST_OBJECT_CAST (self),
 864             GST_FORMAT_TIME, self->segment.stop);
 865       else
 866         m = gst_message_new_segment_done (GST_OBJECT_CAST (self),
 867             GST_FORMAT_TIME, self->segment.start);
 868       gst_element_post_message (GST_ELEMENT_CAST (self), m);
 869     } else {
 870       GST_DEBUG_OBJECT (pad, "Sending EOS at end of segment");
 871       gst_pad_push_event (self->srcpad, gst_event_new_eos ());
 872     }
 873     gst_pad_pause_task (self->srcpad);
 874   }
 875 }
 876
 877 static GstStateChangeReturn
 878 gst_image_freeze_change_state (GstElement * element, GstStateChange transition)
 879 {
 880   GstImageFreeze *self = GST_IMAGE_FREEZE (element);
 881   GstStateChangeReturn ret = GST_STATE_CHANGE_SUCCESS;
 882
 883   switch (transition) {
 884     case GST_STATE_CHANGE_READY_TO_PAUSED:
 885       gst_image_freeze_reset (self);
 886       break;
 887     case GST_STATE_CHANGE_PAUSED_TO_READY:
 888       gst_pad_stop_task (self->srcpad);
 889       gst_image_freeze_reset (self);
 890       break;
 891     default:
 892       break;
 893   }
 894
 895   if (GST_ELEMENT_CLASS (parent_class)->change_state)
 896     ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
 897
 898   switch (transition) {
 899     default:
 900       break;
 901   }
 902
 903   return ret;
 904 }
 905
 906 static gboolean
 907 plugin_init (GstPlugin * plugin)
 908 {
 909   GST_DEBUG_CATEGORY_INIT (gst_image_freeze_debug, "imagefreeze", 0,
 910       "imagefreeze element");
 911
 912   if (!gst_element_register (plugin, "imagefreeze", GST_RANK_NONE,
 913           GST_TYPE_IMAGE_FREEZE))
 914     return FALSE;
 915
 916   return TRUE;
 917 }
 918
 919 GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
 920     GST_VERSION_MINOR,
 921     "imagefreeze",
 922     "Still frame stream generator",
 923     plugin_init, VERSION, "LGPL", GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)