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