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