gst/imagefreeze/gstimagefreeze.c

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