Changeset 4765 for libpipi/trunk/pipi/sequence.c

Timestamp:

02/08/11 23:09:53 (2 years ago)

Author:

sam

Message:

Allow to choose between RGB and YUV in pipi_open_sequence().

File:

• libpipi/trunk/pipi/sequence.c (modified) (7 diffs)

libpipi/trunk/pipi/sequence.c

@@ -50 +50 @@
 
     struct SwsContext *sws_ctx;
-    int src_width, src_height;
+    int src_width, src_height, src_fmt;
 }
 ffmpeg_codec_t;
@@ -56 +56 @@
 
 pipi_sequence_t *pipi_open_sequence(char const *file,
-                                    int width, int height, int fps,
+                                    int width, int height, int rgb, int fps,
                                     int par_num, int par_den, int bitrate)
 {
@@ -183 +183 @@
     ff->buf = (uint8_t *)av_malloc(ff->buf_len);
 
+    ff->src_fmt = rgb ? PIX_FMT_RGB32 : PIX_FMT_YUV444P;
+
     av_write_header(ff->fmt_ctx);
 
@@ -220 +222 @@
     if (!ff->sws_ctx)
     {
-        ff->sws_ctx = sws_getContext(width, height, PIX_FMT_YUV444P,
+        ff->sws_ctx = sws_getContext(width, height, ff->src_fmt,
                                      ff->cod_ctx->width,
                                      ff->cod_ctx->height,
@@ -235 +237 @@
 
     /* Convert interleaved YUV to planar YUV */
-    if (!seq->convert_buf)
-        seq->convert_buf = malloc(width * height * 3);
-
-    for (n = 0; n < width * height; n++)
-    {
-        seq->convert_buf[n] = buffer[4 * n];
-        seq->convert_buf[n + width * height] = buffer[4 * n + 1];
-        seq->convert_buf[n + 2 * width * height] = buffer[4 * n + 2];
-    }
-
-    /* Feed the buffers to FFmpeg */
-    buflist[0] = seq->convert_buf;
-    buflist[1] = seq->convert_buf + 2 * width * height;
-    buflist[2] = seq->convert_buf + width * height;
-    pitchlist[0] = pitchlist[1] = pitchlist[2] = width;
+    if (ff->src_fmt == PIX_FMT_YUV444P)
+    {
+        if (!seq->convert_buf)
+            seq->convert_buf = malloc(width * height * 3);
+
+        for (n = 0; n < width * height; n++)
+        {
+            seq->convert_buf[n] = buffer[4 * n];
+            seq->convert_buf[n + width * height] = buffer[4 * n + 1];
+            seq->convert_buf[n + 2 * width * height] = buffer[4 * n + 2];
+        }
+
+        /* Feed the buffers to FFmpeg */
+        buflist[0] = seq->convert_buf;
+        buflist[1] = seq->convert_buf + 2 * width * height;
+        buflist[2] = seq->convert_buf + width * height;
+        pitchlist[0] = pitchlist[1] = pitchlist[2] = width;
+    }
+    else
+    {
+        buflist[0] = buffer;
+        pitchlist[0] = 4 * width;
+    }
+
     sws_scale(ff->sws_ctx, buflist, pitchlist, 0, height,
               ff->frame->data, ff->frame->linesize);
@@ -280 +291 @@
     ffmpeg_codec_t *ff = (ffmpeg_codec_t *)seq->codec_priv;
 
-    if (ff->fmt_ctx)
+    /* Finish the sequence */
+    if (ff->buf)
     {
         av_write_trailer(ff->fmt_ctx);
     }
 
+    /* Close everything */
     if (ff->buf)
     {
@@ -312 +325 @@
         ff->stream = NULL;
 
-        if (!(ff->fmt_ctx->flags & AVFMT_NOFILE))
+        if (!(ff->fmt_ctx->flags & AVFMT_NOFILE) && ff->fmt_ctx->pb)
             url_fclose(ff->fmt_ctx->pb);