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