gst-gpac/src/gpacmux.c
branchtrunk
changeset 939 1418e08cee7a
parent 917 be87d28e8371
     1.1 --- a/gst-gpac/src/gpacmux.c	Thu Feb 21 17:44:16 2008 +0000
     1.2 +++ b/gst-gpac/src/gpacmux.c	Mon Mar 03 17:38:32 2008 +0000
     1.3 @@ -35,9 +35,14 @@
     1.4    gint track_number;
     1.5    guint32 di; /* outDescriptionIndex */
     1.6  
     1.7 +  GstBuffer *buffer;
     1.8 +  GstBuffer *next_buffer;
     1.9 +
    1.10    guint32 frame_count;
    1.11    gboolean is_video;
    1.12  
    1.13 +  gboolean eos;
    1.14 +
    1.15  } GstGpacPad;
    1.16  
    1.17  struct _GstGpacMux
    1.18 @@ -46,6 +51,7 @@
    1.19  
    1.20    GstPad *srcpad;
    1.21    GstCollectPads *collect;
    1.22 +  gint active_pads;
    1.23  
    1.24    GF_ISOFile *file;
    1.25  
    1.26 @@ -109,6 +115,13 @@
    1.27  static GstStateChangeReturn gst_gpac_mux_change_state (GstElement * element,
    1.28      GstStateChange transition);
    1.29  
    1.30 +static GstGpacPad* gst_gpac_mux_queue_pads (GstGpacMux * gpac_mux);
    1.31 +static GstFlowReturn gst_gpac_mux_process_pad (GstGpacMux * gpac_mux, GstGpacPad *pad);
    1.32 +static gboolean gst_gpac_mux_all_pads_eos (GstCollectPads * pads);
    1.33 +static gint gst_gpac_mux_compare_pads (GstGpacMux * ogg_mux, GstGpacPad *first,
    1.34 +    GstGpacPad *second);
    1.35 +
    1.36 +
    1.37  static GstElementClass *parent_class = NULL;
    1.38  
    1.39  GType
    1.40 @@ -216,13 +229,6 @@
    1.41    GstGpacPad *gpacpad = (GstGpacPad *) data;
    1.42    GstBuffer *buf;
    1.43  
    1.44 -/*  if (gpacpad->pagebuffers) {
    1.45 -    while ((buf = g_queue_pop_head (gpacpad->pagebuffers)) != NULL) {
    1.46 -      gst_buffer_unref (buf);
    1.47 -    }
    1.48 -    g_queue_free (gpacpad->pagebuffers);
    1.49 -    gpacpad->pagebuffers = NULL;
    1.50 -  }*/
    1.51  }
    1.52  
    1.53  static GstPadLinkReturn
    1.54 @@ -282,7 +288,7 @@
    1.55      
    1.56      /* create new pad with the name */
    1.57      GST_DEBUG_OBJECT (gpac_mux, "Creating new pad for serial %d", serial);
    1.58 -    printf ("XXXX new pad from template\n");
    1.59 +
    1.60      newpad = gst_pad_new_from_template (templ, padname);
    1.61      g_free (padname);
    1.62  
    1.63 @@ -294,6 +300,7 @@
    1.64        gpacpad = (GstGpacPad *)
    1.65            gst_collect_pads_add_pad_full (gpac_mux->collect, newpad,
    1.66            sizeof (GstGpacPad), gst_gpac_mux_gpac_pad_destroy_notify);
    1.67 +      gpac_mux->active_pads++;
    1.68  
    1.69        /* gpac new track */
    1.70        gpacpad->is_video = is_video;
    1.71 @@ -411,92 +418,128 @@
    1.72    return gst_pad_push (mux->srcpad, buffer);
    1.73  }
    1.74  
    1.75 +static GstFlowReturn
    1.76 +gst_gpac_mux_collected (GstCollectPads * pads, GstGpacMux * gpac_mux)
    1.77 +{
    1.78 +  GstFlowReturn ret = GST_FLOW_OK;
    1.79 +  GstGpacPad *pad;
    1.80 +  gint active_before;
    1.81 +
    1.82 +  GST_LOG_OBJECT (gpac_mux, "collected");
    1.83 +
    1.84 +  active_before = gpac_mux->active_pads;
    1.85 +
    1.86 +  pad = gst_gpac_mux_queue_pads (gpac_mux);
    1.87 +  if (!pad) {
    1.88 +    return GST_FLOW_WRONG_STATE;
    1.89 +  }
    1.90 +
    1.91 +  if (pad->buffer) {
    1.92 +    ret = gst_gpac_mux_process_pad (gpac_mux, pad);
    1.93 +  }
    1.94 +
    1.95 +  if (gpac_mux->active_pads < active_before) {
    1.96 +    /* If the active pad count went down, this mean at least one pad has gone
    1.97 +     * EOS. Since CollectPads only calls _collected() once when all pads are
    1.98 +     * EOS, and our code doesn't _pop() from all pads we need to check that by
    1.99 +     * peeking on all pads, else we won't be called again and the muxing will
   1.100 +     * not terminate (push out EOS). */
   1.101 +          printf ("XXXX um pad foi desativado %" GST_PTR_FORMAT "\n", pad);
   1.102 +
   1.103 +    /* if all the pads have been removed, flush all pending data */
   1.104 +    if ((ret == GST_FLOW_OK) && gst_gpac_mux_all_pads_eos (pads)) {
   1.105 +      GST_LOG_OBJECT (gpac_mux, "no pads remaining, flushing data");
   1.106 +
   1.107 +      do {
   1.108 +        pad = gst_gpac_mux_queue_pads (gpac_mux);
   1.109 +        if (pad)
   1.110 +          ret = gst_gpac_mux_process_pad (gpac_mux, pad);
   1.111 +      } while ((ret == GST_FLOW_OK) && (pad != NULL));
   1.112 +
   1.113 +      /* gpac file close (eos) */
   1.114 +      // Fixme: this should flush all data to src pad
   1.115 +      // Fixme: where to release gpac_mux->file?
   1.116 +      printf ("CCCCCCCCCCCCCCCCcclosing the file\n");
   1.117 +      gf_isom_close (gpac_mux->file);
   1.118 +
   1.119 +
   1.120 +      GST_DEBUG_OBJECT (gpac_mux, "Pushing EOS");
   1.121 +      gst_pad_push_event (gpac_mux->srcpad, gst_event_new_eos ());
   1.122 +    }
   1.123 +  }
   1.124 +
   1.125 +  return ret;
   1.126 +
   1.127 +}
   1.128 +
   1.129  static gboolean
   1.130 -all_pads_eos (GstCollectPads * pads)
   1.131 +gst_gpac_mux_all_pads_eos (GstCollectPads * pads)
   1.132  {
   1.133 -  GSList *iter;
   1.134 +  GSList *walk;
   1.135    gboolean alleos = TRUE;
   1.136  
   1.137 -  iter = pads->data;
   1.138 -  while (iter) {
   1.139 +  walk = pads->data;
   1.140 +  while (walk) {
   1.141      GstBuffer *buf;
   1.142 -    GstCollectData *data = (GstCollectData *) iter->data;
   1.143 +    GstCollectData *data = (GstCollectData *) walk->data;
   1.144  
   1.145      buf = gst_collect_pads_peek (pads, data);
   1.146      if (buf) {
   1.147        alleos = FALSE;
   1.148        gst_buffer_unref (buf);
   1.149 -      goto beach;
   1.150 +      break;
   1.151      }
   1.152 -    iter = iter->next;
   1.153 +    walk = walk->next;
   1.154    }
   1.155 -beach:
   1.156 +
   1.157    return alleos;
   1.158  }
   1.159 - 
   1.160 +
   1.161 +
   1.162  static GstFlowReturn
   1.163 -gst_gpac_mux_collected (GstCollectPads * pads, GstGpacMux * gpac_mux)
   1.164 +gst_gpac_mux_process_pad (GstGpacMux *gpac_mux, GstGpacPad *pad)
   1.165  {
   1.166    GstFlowReturn ret = GST_FLOW_OK;
   1.167 -  GSList *iter;
   1.168 -  GstBuffer *buf;
   1.169 -  gint i = 0;
   1.170 +  GF_ISOSample *sample;
   1.171  
   1.172 -  GST_LOG_OBJECT (gpac_mux, "collected");
   1.173 -
   1.174 -  iter = gpac_mux->collect->data;
   1.175 -  while (iter) {
   1.176 -    GstCollectData *data = (GstCollectData *) iter->data;
   1.177 -    GstGpacPad *pad = (GstGpacPad *) data;
   1.178 -
   1.179 -    buf = gst_collect_pads_pop (gpac_mux->collect, data);
   1.180 -  if (buf == NULL) {
   1.181 -    iter = g_slist_next (iter);
   1.182 -      continue;
   1.183 -  }
   1.184 -    GST_LOG_OBJECT (data->pad, "popped buffer %" GST_PTR_FORMAT, buf);
   1.185 -
   1.186 -    /* gpac output */
   1.187 -    printf ("xxxxxx buffer size: %d\n", GST_BUFFER_SIZE(buf)); fflush (stdout);
   1.188 -    if (pad->frame_count < 300) {
   1.189 -      GF_ISOSample *sample = gf_isom_sample_new();
   1.190 -
   1.191 -       sample->dataLength = GST_BUFFER_SIZE(buf);
   1.192 -       sample->data = GST_BUFFER_DATA(buf);
   1.193 -       sample->CTS_Offset = 0;
   1.194 -
   1.195 -      if (pad->is_video) {
   1.196 -        sample->IsRAP = 0;
   1.197 -        sample->DTS += 1000*pad->frame_count;
   1.198 -      } else {
   1.199 -        sample->IsRAP = 0;
   1.200 -        sample->DTS = 2048*pad->frame_count;
   1.201 -      }
   1.202 -
   1.203 -      gf_isom_add_sample(gpac_mux->file, pad->track_number, 
   1.204 -                      pad->di, sample);
   1.205 -      sample->data = NULL;
   1.206 -
   1.207 -      gf_isom_sample_del(&sample);
   1.208 -
   1.209 -      printf ("xxxx frames %d\n", pad->frame_count);
   1.210 -
   1.211 -    } else if (pad->frame_count == 300) {
   1.212 -      printf ("XXX closing gpac output file\n"); fflush (stdout);
   1.213 -      gf_isom_close (gpac_mux->file);
   1.214 -    }
   1.215 -
   1.216 -    /* gstreamer output (push) */
   1.217 -    if (gst_gpac_mux_push_buffer (gpac_mux, buf) != GST_FLOW_OK) {
   1.218 -      printf ("EEEEEEEE push failed\n");
   1.219 -    }
   1.220 -
   1.221 -    iter = g_slist_next (iter);
   1.222 -    pad->frame_count++;
   1.223 -    i++;
   1.224 +  if (pad->buffer == NULL) {
   1.225 +          printf ("Buffer is null, wrong state\n");
   1.226 +      return GST_FLOW_WRONG_STATE;
   1.227    }
   1.228  
   1.229 -  /* fixme */
   1.230 +  /* gpac output */
   1.231 +  printf ("xxxxxx buffer size: %d\n", GST_BUFFER_SIZE(pad->buffer)); fflush (stdout);
   1.232 +  printf ("xxxx frames %d\n", pad->frame_count);
   1.233 +
   1.234 +  sample = gf_isom_sample_new();
   1.235 +  sample->dataLength = GST_BUFFER_SIZE(pad->buffer);
   1.236 +  sample->data = GST_BUFFER_DATA(pad->buffer);
   1.237 +  sample->CTS_Offset = 0;
   1.238 +
   1.239 +  if (pad->is_video) {
   1.240 +    sample->IsRAP = 0;
   1.241 +    sample->DTS += 1000*pad->frame_count;
   1.242 +  } else {
   1.243 +    sample->IsRAP = 0;
   1.244 +    sample->DTS = 2048*pad->frame_count;
   1.245 +  }
   1.246 +
   1.247 +  gf_isom_add_sample(gpac_mux->file, pad->track_number, 
   1.248 +                     pad->di, sample);
   1.249 +  sample->data = NULL;
   1.250 +
   1.251 +  gf_isom_sample_del(&sample);
   1.252 +
   1.253 +
   1.254 +  /* gstreamer output (push) */
   1.255 +  ret = gst_gpac_mux_push_buffer (gpac_mux, pad->buffer);
   1.256 +  pad->buffer = NULL;
   1.257 +
   1.258 +  if (ret != GST_FLOW_OK) {
   1.259 +  }
   1.260 +
   1.261 +  pad->frame_count++;
   1.262 + 
   1.263    return ret;
   1.264  }
   1.265  
   1.266 @@ -504,13 +547,13 @@
   1.267  static void
   1.268  gst_gpac_mux_init_collectpads (GstCollectPads * collect)
   1.269  {
   1.270 -  GSList *walk;
   1.271 +  GSList *iter;
   1.272  
   1.273 -  walk = collect->data;
   1.274 -  while (walk) {
   1.275 -    GstGpacPad *gpacpad = (GstGpacPad *) walk->data;
   1.276 +  iter = collect->data;
   1.277 +  while (iter) {
   1.278 +    GstGpacPad *gpacpad = (GstGpacPad *) iter->data;
   1.279  
   1.280 -    //ogg_stream_init (&gpacpad->stream, gpacpad->serial);
   1.281 +    //gpac_stream_init (&gpacpad->stream, gpacpad->serial);
   1.282      //gpacpad->packetno = 0;
   1.283      //gpacpad->pageno = 0;
   1.284      //gpacpad->eos = FALSE;
   1.285 @@ -521,7 +564,7 @@
   1.286      //gpacpad->prev_delta = FALSE;
   1.287      //gpacpad->pagebuffers = g_queue_new ();
   1.288  
   1.289 -    walk = g_slist_next (walk);
   1.290 +    iter = g_slist_next (iter);
   1.291    }
   1.292  }
   1.293  
   1.294 @@ -529,10 +572,10 @@
   1.295  static void
   1.296  gst_gpac_mux_clear_collectpads (GstCollectPads * collect)
   1.297  {
   1.298 -  GSList *walk;
   1.299 +  GSList *iter;
   1.300  
   1.301 -  for (walk = collect->data; walk; walk = g_slist_next (walk)) {
   1.302 -    GstGpacPad *gpacpad = (GstGpacPad *) walk->data;
   1.303 +  for (iter = collect->data; iter; iter = g_slist_next (iter)) {
   1.304 +    GstGpacPad *gpacpad = (GstGpacPad *) iter->data;
   1.305      GstBuffer *buf;
   1.306  
   1.307      //gpac_stream_clear (&gpacpad->stream);
   1.308 @@ -588,6 +631,97 @@
   1.309  
   1.310  }
   1.311  
   1.312 +static GstGpacPad *
   1.313 +gst_gpac_mux_queue_pads (GstGpacMux * gpac_mux)
   1.314 +{
   1.315 +  GstGpacPad *bestpad = NULL;//, *still_hungry = NULL;
   1.316 +  GSList *iter;
   1.317 +
   1.318 +  /* try to make sure we have a buffer from each usable pad first */
   1.319 +  iter = gpac_mux->collect->data;
   1.320 +  while (iter) {
   1.321 +    GstGpacPad *pad;
   1.322 +    GstCollectData *data;
   1.323 +
   1.324 +    data = (GstCollectData *) iter->data;
   1.325 +    pad = (GstGpacPad *) data;
   1.326 +
   1.327 +    iter = g_slist_next (iter);
   1.328 +
   1.329 +    GST_LOG_OBJECT (data->pad, "looking at pad for buffer");
   1.330 +
   1.331 +    /* try to get a new buffer for this pad if needed and possible */
   1.332 +    if (pad->buffer == NULL) {
   1.333 +      GstBuffer *buf;
   1.334 +
   1.335 +      buf = gst_collect_pads_pop (gpac_mux->collect, data);
   1.336 +      GST_LOG_OBJECT (data->pad, "popped buffer %" GST_PTR_FORMAT, buf);
   1.337 +
   1.338 +      /* On EOS we get a NULL buffer */
   1.339 +      if (buf == NULL) {
   1.340 +              printf ("EENENENENENEND OF STREAM EOS\n");
   1.341 +        GST_DEBUG_OBJECT (data->pad, "EOS on pad");
   1.342 +        if (!pad->eos) {
   1.343 +          /* it's no longer active */
   1.344 +          gpac_mux->active_pads--;
   1.345 +          pad->eos = TRUE;
   1.346 +
   1.347 +        }
   1.348 +      }
   1.349 +
   1.350 +      pad->buffer = buf;
   1.351 +    } 
   1.352 +
   1.353 +    /* we should have a buffer now, see if it is the best pad to
   1.354 +     * pull on */
   1.355 +      if (gst_gpac_mux_compare_pads (gpac_mux, bestpad, pad) > 0) {
   1.356 +        GST_LOG_OBJECT (data->pad,
   1.357 +            "new best pad, with buffers %" GST_PTR_FORMAT, pad->buffer);
   1.358 +        bestpad = pad;
   1.359 +      }
   1.360 +  }
   1.361 + 
   1.362 +  return bestpad;
   1.363 +}
   1.364 +
   1.365 +static gint
   1.366 +gst_gpac_mux_compare_pads (GstGpacMux * ogg_mux, GstGpacPad *first,
   1.367 +    GstGpacPad *second)
   1.368 +{
   1.369 +  guint64 firsttime, secondtime;
   1.370 +
   1.371 +  /* if the first pad doesn't contain anything or is even NULL, return
   1.372 +   * the second pad as best candidate and vice versa */
   1.373 +  if (first == NULL || (first->buffer == NULL))
   1.374 +    return 1;
   1.375 +  if (second == NULL || (second->buffer == NULL))
   1.376 +    return -1;
   1.377 +
   1.378 +  /* no timestamp on first buffer, it must go first */
   1.379 +  if (first->buffer)
   1.380 +    firsttime = GST_BUFFER_TIMESTAMP (first->buffer);
   1.381 +  if (firsttime == GST_CLOCK_TIME_NONE)
   1.382 +    return -1;
   1.383 +
   1.384 +  /* no timestamp on second buffer, it must go first */
   1.385 +  if (second->buffer)
   1.386 +    secondtime = GST_BUFFER_TIMESTAMP (second->buffer);
   1.387 +  if (secondtime == GST_CLOCK_TIME_NONE)
   1.388 +    return 1;
   1.389 +
   1.390 +  /* first buffer has higher timestamp, second one should go first */
   1.391 +  if (secondtime < firsttime)
   1.392 +    return 1;
   1.393 +  /* second buffer has higher timestamp, first one should go first */
   1.394 +  else if (secondtime > firsttime)
   1.395 +    return -1;
   1.396 +
   1.397 +  /* same priority if all of the above failed */
   1.398 +  return 0;
   1.399 +}
   1.400 +
   1.401 +
   1.402 +
   1.403  static gboolean
   1.404  gst_gpac_mux_plugin_init (GstPlugin * plugin)
   1.405  {