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