* Add code to pace sending a stream when it is being sent from a file.
[ffmpeg.git] / libav / swf.c
index a93c4f7..14f8707 100644 (file)
@@ -1,30 +1,23 @@
 /*
  * Flash Compatible Streaming Format
- * Copyright (c) 2000 Gerard Lantau.
+ * Copyright (c) 2000 Fabrice Bellard.
  *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2 of the License, or (at your option) any later version.
  *
- * This program is distributed in the hope that it will be useful,
+ * This library is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
- * GNU General Public License for more details.
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
  *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  */
-#include <stdlib.h>
-#include <stdio.h>
-#include <string.h>
-#include <errno.h>
-
 #include "avformat.h"
 
-#include <assert.h>
-
 /* should have a generic way to indicate probable size */
 #define DUMMY_FILE_SIZE   (100 * 1024 * 1024)
 #define DUMMY_DURATION    600 /* in seconds */
@@ -51,8 +44,8 @@
 #define SHAPE_ID  1
 
 typedef struct {
-    long long duration_pos;
-    long long tag_pos;
+    offset_t duration_pos;
+    offset_t tag_pos;
     int tag;
 } SWFContext;
 
@@ -76,7 +69,7 @@ static void put_swf_end_tag(AVFormatContext *s)
 {
     SWFContext *swf = s->priv_data;
     ByteIOContext *pb = &s->pb;
-    long long pos;
+    offset_t pos;
     int tag_len, tag;
 
     pos = url_ftell(pb);
@@ -134,7 +127,7 @@ static void put_swf_rect(ByteIOContext *pb,
     put_bits(&p, nbits, ymax & mask);
     
     flush_put_bits(&p);
-    put_buffer(pb, buf, p.buf_ptr - p.buf);
+    put_buffer(pb, buf, pbBufPtr(&p) - p.buf);
 }
 
 static void put_swf_line_edge(PutBitContext *pb, int dx, int dy)
@@ -190,7 +183,7 @@ static void put_swf_matrix(ByteIOContext *pb,
     put_bits(&p, 20, ty);
 
     flush_put_bits(&p);
-    put_buffer(pb, buf, p.buf_ptr - p.buf);
+    put_buffer(pb, buf, pbBufPtr(&p) - p.buf);
 }
 
 /* XXX: handle audio only */
@@ -203,7 +196,7 @@ static int swf_write_header(AVFormatContext *s)
     UINT8 buf1[256];
     int i, width, height, rate;
 
-    swf = malloc(sizeof(SWFContext));
+    swf = av_malloc(sizeof(SWFContext));
     if (!swf)
         return -1;
     s->priv_data = swf;
@@ -237,7 +230,7 @@ static int swf_write_header(AVFormatContext *s)
     put_swf_rect(pb, 0, width, 0, height);
     put_le16(pb, (rate * 256) / FRAME_RATE_BASE); /* frame rate */
     swf->duration_pos = url_ftell(pb);
-    put_le16(pb, DUMMY_DURATION * (INT64)rate / FRAME_RATE_BASE); /* frame count */
+    put_le16(pb, (UINT16)(DUMMY_DURATION * (INT64)rate / FRAME_RATE_BASE)); /* frame count */
     
     /* define a shape with the jpeg inside */
 
@@ -278,7 +271,7 @@ static int swf_write_header(AVFormatContext *s)
     put_bits(&p, 5, 0);
 
     flush_put_bits(&p);
-    put_buffer(pb, buf1, p.buf_ptr - p.buf);
+    put_buffer(pb, buf1, pbBufPtr(&p) - p.buf);
 
     put_swf_end_tag(s);
 
@@ -301,7 +294,7 @@ static int swf_write_header(AVFormatContext *s)
             break;
         default:
             /* not supported */
-            free(swf);
+            av_free(swf);
             return -1;
         }
         if (audio_enc->channels == 2)
@@ -385,7 +378,7 @@ static int swf_write_audio(AVFormatContext *s, UINT8 *buf, int size)
 }
 
 static int swf_write_packet(AVFormatContext *s, int stream_index, 
-                           UINT8 *buf, int size)
+                           UINT8 *buf, int size, int force_pts)
 {
     AVCodecContext *codec = &s->streams[stream_index]->codec;
     if (codec->codec_type == CODEC_TYPE_AUDIO)
@@ -421,7 +414,6 @@ static int swf_write_trailer(AVFormatContext *s)
         url_fseek(pb, swf->duration_pos, SEEK_SET);
         put_le16(pb, video_enc->frame_number);
     }
-    free(swf);
     return 0;
 }
 
@@ -445,6 +437,19 @@ static int get_swf_tag(ByteIOContext *pb, int *len_ptr)
     return tag;
 }
 
+
+static int swf_probe(AVProbeData *p)
+{
+    /* check file header */
+    if (p->buf_size <= 16)
+        return 0;
+    if (p->buf[0] == 'F' && p->buf[1] == 'W' &&
+        p->buf[2] == 'S')
+        return AVPROBE_SCORE_MAX;
+    else
+        return 0;
+}
+
 static int swf_read_header(AVFormatContext *s, AVFormatParameters *ap)
 {
     ByteIOContext *pb = &s->pb;
@@ -495,7 +500,7 @@ static int swf_read_header(AVFormatContext *s, AVFormatParameters *ap)
                     st->codec.sample_rate = 44100;
                     break;
                 default:
-                    free(st);
+                    av_free(st);
                     return -EIO;
                 }
                 st->codec.codec_type = CODEC_TYPE_AUDIO;
@@ -535,18 +540,32 @@ static int swf_read_close(AVFormatContext *s)
      return 0;
 }
 
-AVFormat swf_format = {
+static AVInputFormat swf_iformat = {
+    "swf",
+    "Flash format",
+    0,
+    swf_probe,
+    swf_read_header,
+    swf_read_packet,
+    swf_read_close,
+};
+
+static AVOutputFormat swf_oformat = {
     "swf",
     "Flash format",
     "application/x-shockwave-flash",
     "swf",
+    sizeof(SWFContext),
     CODEC_ID_MP2,
     CODEC_ID_MJPEG,
     swf_write_header,
     swf_write_packet,
     swf_write_trailer,
-
-    swf_read_header,
-    swf_read_packet,
-    swf_read_close,
 };
+
+int swf_init(void)
+{
+    av_register_input_format(&swf_iformat);
+    av_register_output_format(&swf_oformat);
+    return 0;
+}