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