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