mirror of
				https://github.com/nyanmisaka/ffmpeg-rockchip.git
				synced 2025-10-31 04:26:37 +08:00 
			
		
		
		
	
		
			
				
	
	
		
			659 lines
		
	
	
		
			18 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			659 lines
		
	
	
		
			18 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /*
 | |
|  * MPEG1 mux/demux
 | |
|  * Copyright (c) 2000, 2001, 2002 Fabrice Bellard.
 | |
|  *
 | |
|  * 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 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
 | |
|  * Lesser General Public License for more details.
 | |
|  *
 | |
|  * 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 "avformat.h"
 | |
| #include "tick.h"
 | |
| 
 | |
| #define MAX_PAYLOAD_SIZE 4096
 | |
| #define NB_STREAMS 2
 | |
| 
 | |
| typedef struct {
 | |
|     UINT8 buffer[MAX_PAYLOAD_SIZE];
 | |
|     int buffer_ptr;
 | |
|     UINT8 id;
 | |
|     int max_buffer_size; /* in bytes */
 | |
|     int packet_number;
 | |
|     INT64 pts;
 | |
|     Ticker pts_ticker;
 | |
|     INT64 start_pts;
 | |
| } StreamInfo;
 | |
| 
 | |
| typedef struct {
 | |
|     int packet_size; /* required packet size */
 | |
|     int packet_data_max_size; /* maximum data size inside a packet */
 | |
|     int packet_number;
 | |
|     int pack_header_freq;     /* frequency (in packets^-1) at which we send pack headers */
 | |
|     int system_header_freq;
 | |
|     int mux_rate; /* bitrate in units of 50 bytes/s */
 | |
|     /* stream info */
 | |
|     int audio_bound;
 | |
|     int video_bound;
 | |
| } MpegMuxContext;
 | |
| 
 | |
| #define PACK_START_CODE             ((unsigned int)0x000001ba)
 | |
| #define SYSTEM_HEADER_START_CODE    ((unsigned int)0x000001bb)
 | |
| #define SEQUENCE_END_CODE           ((unsigned int)0x000001b7)
 | |
| #define PACKET_START_CODE_MASK      ((unsigned int)0xffffff00)
 | |
| #define PACKET_START_CODE_PREFIX    ((unsigned int)0x00000100)
 | |
| #define ISO_11172_END_CODE          ((unsigned int)0x000001b9)
 | |
|   
 | |
| /* mpeg2 */
 | |
| #define PROGRAM_STREAM_MAP 0x1bc
 | |
| #define PRIVATE_STREAM_1   0x1bd
 | |
| #define PADDING_STREAM     0x1be
 | |
| #define PRIVATE_STREAM_2   0x1bf
 | |
| 
 | |
| 
 | |
| #define AUDIO_ID 0xc0
 | |
| #define VIDEO_ID 0xe0
 | |
| 
 | |
| static int put_pack_header(AVFormatContext *ctx, 
 | |
|                            UINT8 *buf, INT64 timestamp)
 | |
| {
 | |
|     MpegMuxContext *s = ctx->priv_data;
 | |
|     PutBitContext pb;
 | |
|     
 | |
|     init_put_bits(&pb, buf, 128, NULL, NULL);
 | |
| 
 | |
|     put_bits(&pb, 32, PACK_START_CODE);
 | |
|     put_bits(&pb, 4, 0x2);
 | |
|     put_bits(&pb, 3, (UINT32)((timestamp >> 30) & 0x07));
 | |
|     put_bits(&pb, 1, 1);
 | |
|     put_bits(&pb, 15, (UINT32)((timestamp >> 15) & 0x7fff));
 | |
|     put_bits(&pb, 1, 1);
 | |
|     put_bits(&pb, 15, (UINT32)((timestamp) & 0x7fff));
 | |
|     put_bits(&pb, 1, 1);
 | |
|     put_bits(&pb, 1, 1);
 | |
|     put_bits(&pb, 22, s->mux_rate);
 | |
|     put_bits(&pb, 1, 1);
 | |
| 
 | |
|     flush_put_bits(&pb);
 | |
|     return pbBufPtr(&pb) - pb.buf;
 | |
| }
 | |
| 
 | |
| static int put_system_header(AVFormatContext *ctx, UINT8 *buf)
 | |
| {
 | |
|     MpegMuxContext *s = ctx->priv_data;
 | |
|     int size, rate_bound, i, private_stream_coded, id;
 | |
|     PutBitContext pb;
 | |
| 
 | |
|     init_put_bits(&pb, buf, 128, NULL, NULL);
 | |
| 
 | |
|     put_bits(&pb, 32, SYSTEM_HEADER_START_CODE);
 | |
|     put_bits(&pb, 16, 0);
 | |
|     put_bits(&pb, 1, 1);
 | |
|     
 | |
|     rate_bound = s->mux_rate; /* maximum bit rate of the multiplexed stream */
 | |
|     put_bits(&pb, 22, rate_bound);
 | |
|     put_bits(&pb, 1, 1); /* marker */
 | |
|     put_bits(&pb, 6, s->audio_bound);
 | |
| 
 | |
|     put_bits(&pb, 1, 1); /* variable bitrate */
 | |
|     put_bits(&pb, 1, 1); /* non constrainted bit stream */
 | |
|     
 | |
|     put_bits(&pb, 1, 0); /* audio locked */
 | |
|     put_bits(&pb, 1, 0); /* video locked */
 | |
|     put_bits(&pb, 1, 1); /* marker */
 | |
| 
 | |
|     put_bits(&pb, 5, s->video_bound);
 | |
|     put_bits(&pb, 8, 0xff); /* reserved byte */
 | |
|     
 | |
|     /* audio stream info */
 | |
|     private_stream_coded = 0;
 | |
|     for(i=0;i<ctx->nb_streams;i++) {
 | |
|         StreamInfo *stream = ctx->streams[i]->priv_data;
 | |
|         id = stream->id;
 | |
|         if (id < 0xc0) {
 | |
|             /* special case for private streams (AC3 use that) */
 | |
|             if (private_stream_coded)
 | |
|                 continue;
 | |
|             private_stream_coded = 1;
 | |
|             id = 0xbd;
 | |
|         }
 | |
|         put_bits(&pb, 8, id); /* stream ID */
 | |
|         put_bits(&pb, 2, 3);
 | |
|         if (id < 0xe0) {
 | |
|             /* audio */
 | |
|             put_bits(&pb, 1, 0);
 | |
|             put_bits(&pb, 13, stream->max_buffer_size / 128);
 | |
|         } else {
 | |
|             /* video */
 | |
|             put_bits(&pb, 1, 1);
 | |
|             put_bits(&pb, 13, stream->max_buffer_size / 1024);
 | |
|         }
 | |
|     }
 | |
|     flush_put_bits(&pb);
 | |
|     size = pbBufPtr(&pb) - pb.buf;
 | |
|     /* patch packet size */
 | |
|     buf[4] = (size - 6) >> 8;
 | |
|     buf[5] = (size - 6) & 0xff;
 | |
| 
 | |
|     return size;
 | |
| }
 | |
| 
 | |
| static int mpeg_mux_init(AVFormatContext *ctx)
 | |
| {
 | |
|     MpegMuxContext *s = ctx->priv_data;
 | |
|     int bitrate, i, mpa_id, mpv_id, ac3_id;
 | |
|     AVStream *st;
 | |
|     StreamInfo *stream;
 | |
| 
 | |
|     s->packet_number = 0;
 | |
| 
 | |
|     /* XXX: hardcoded */
 | |
|     if (ctx->flags & AVF_FLAG_VCD)
 | |
|         s->packet_size = 2324; /* VCD packet size */
 | |
|     else
 | |
|         s->packet_size = 2048;
 | |
|         
 | |
|     /* startcode(4) + length(2) + flags(1) */
 | |
|     s->packet_data_max_size = s->packet_size - 7;
 | |
|     s->audio_bound = 0;
 | |
|     s->video_bound = 0;
 | |
|     mpa_id = AUDIO_ID;
 | |
|     ac3_id = 0x80;
 | |
|     mpv_id = VIDEO_ID;
 | |
|     for(i=0;i<ctx->nb_streams;i++) {
 | |
|         st = ctx->streams[i];
 | |
|         stream = av_mallocz(sizeof(StreamInfo));
 | |
|         if (!stream)
 | |
|             goto fail;
 | |
|         st->priv_data = stream;
 | |
| 
 | |
|         switch(st->codec.codec_type) {
 | |
|         case CODEC_TYPE_AUDIO:
 | |
|             if (st->codec.codec_id == CODEC_ID_AC3)
 | |
|                 stream->id = ac3_id++;
 | |
|             else
 | |
|                 stream->id = mpa_id++;
 | |
|             stream->max_buffer_size = 4 * 1024; 
 | |
|             s->audio_bound++;
 | |
|             break;
 | |
|         case CODEC_TYPE_VIDEO:
 | |
|             stream->id = mpv_id++;
 | |
|             stream->max_buffer_size = 46 * 1024; 
 | |
|             s->video_bound++;
 | |
|             break;
 | |
|         default:
 | |
|             abort();
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     /* we increase slightly the bitrate to take into account the
 | |
|        headers. XXX: compute it exactly */
 | |
|     bitrate = 2000;
 | |
|     for(i=0;i<ctx->nb_streams;i++) {
 | |
|         st = ctx->streams[i];
 | |
|         bitrate += st->codec.bit_rate;
 | |
|     }
 | |
|     s->mux_rate = (bitrate + (8 * 50) - 1) / (8 * 50);
 | |
|     
 | |
|     if (ctx->flags & AVF_FLAG_VCD)
 | |
|         /* every packet */
 | |
|         s->pack_header_freq = 1;
 | |
|     else
 | |
|         /* every 2 seconds */
 | |
|         s->pack_header_freq = 2 * bitrate / s->packet_size / 8;
 | |
|     
 | |
|     if (ctx->flags & AVF_FLAG_VCD)
 | |
|         /* every 40 packets, this is my invention */
 | |
|         s->system_header_freq = s->pack_header_freq * 40;
 | |
|     else
 | |
|         /* every 10 seconds */
 | |
|         s->system_header_freq = s->pack_header_freq * 5;
 | |
|     
 | |
|     
 | |
|     for(i=0;i<ctx->nb_streams;i++) {
 | |
|         stream = ctx->streams[i]->priv_data;
 | |
|         stream->buffer_ptr = 0;
 | |
|         stream->packet_number = 0;
 | |
|         stream->pts = 0;
 | |
|         stream->start_pts = -1;
 | |
| 
 | |
|         st = ctx->streams[i];
 | |
|         switch (st->codec.codec_type) {
 | |
|         case CODEC_TYPE_AUDIO:
 | |
|             ticker_init(&stream->pts_ticker,
 | |
|                         st->codec.sample_rate,
 | |
|                         90000 * st->codec.frame_size);
 | |
|             break;
 | |
|         case CODEC_TYPE_VIDEO:
 | |
|             ticker_init(&stream->pts_ticker,
 | |
|                         st->codec.frame_rate,
 | |
|                         90000 * FRAME_RATE_BASE);
 | |
|             break;
 | |
|         default:
 | |
|             abort();
 | |
|         }
 | |
|     }
 | |
|     return 0;
 | |
|  fail:
 | |
|     for(i=0;i<ctx->nb_streams;i++) {
 | |
|         av_free(ctx->streams[i]->priv_data);
 | |
|     }
 | |
|     return -ENOMEM;
 | |
| }
 | |
| 
 | |
| /* flush the packet on stream stream_index */
 | |
| static void flush_packet(AVFormatContext *ctx, int stream_index, int last_pkt)
 | |
| {
 | |
|     MpegMuxContext *s = ctx->priv_data;
 | |
|     StreamInfo *stream = ctx->streams[stream_index]->priv_data;
 | |
|     UINT8 *buf_ptr;
 | |
|     int size, payload_size, startcode, id, len, stuffing_size, i;
 | |
|     INT64 timestamp;
 | |
|     UINT8 buffer[128];
 | |
|     int last = last_pkt ? 4 : 0;
 | |
|     
 | |
|     id = stream->id;
 | |
|     timestamp = stream->start_pts;
 | |
| 
 | |
| #if 0
 | |
|     printf("packet ID=%2x PTS=%0.3f\n", 
 | |
|            id, timestamp / 90000.0);
 | |
| #endif
 | |
| 
 | |
|     buf_ptr = buffer;
 | |
|     if (((s->packet_number % s->pack_header_freq) == 0)) {
 | |
|         /* output pack and systems header if needed */
 | |
|         size = put_pack_header(ctx, buf_ptr, timestamp);
 | |
|         buf_ptr += size;
 | |
|         if ((s->packet_number % s->system_header_freq) == 0) {
 | |
|             size = put_system_header(ctx, buf_ptr);
 | |
|             buf_ptr += size;
 | |
|         }
 | |
|     }
 | |
|     size = buf_ptr - buffer;
 | |
|     put_buffer(&ctx->pb, buffer, size);
 | |
| 
 | |
|     /* packet header */
 | |
|     payload_size = s->packet_size - (size + 6 + 5 + last);
 | |
|     if (id < 0xc0) {
 | |
|         startcode = PRIVATE_STREAM_1;
 | |
|         payload_size -= 4;
 | |
|     } else {
 | |
|         startcode = 0x100 + id;
 | |
|     }
 | |
|     stuffing_size = payload_size - stream->buffer_ptr;
 | |
|     if (stuffing_size < 0)
 | |
|         stuffing_size = 0;
 | |
| 
 | |
|     put_be32(&ctx->pb, startcode);
 | |
| 
 | |
|     put_be16(&ctx->pb, payload_size + 5);
 | |
|     /* stuffing */
 | |
|     for(i=0;i<stuffing_size;i++)
 | |
|         put_byte(&ctx->pb, 0xff);
 | |
|     
 | |
|     /* presentation time stamp */
 | |
|     put_byte(&ctx->pb, 
 | |
|              (0x02 << 4) | 
 | |
|              (((timestamp >> 30) & 0x07) << 1) | 
 | |
|              1);
 | |
|     put_be16(&ctx->pb, (UINT16)((((timestamp >> 15) & 0x7fff) << 1) | 1));
 | |
|     put_be16(&ctx->pb, (UINT16)((((timestamp) & 0x7fff) << 1) | 1));
 | |
| 
 | |
|     if (startcode == PRIVATE_STREAM_1) {
 | |
|         put_byte(&ctx->pb, id);
 | |
|         if (id >= 0x80 && id <= 0xbf) {
 | |
|             /* XXX: need to check AC3 spec */
 | |
|             put_byte(&ctx->pb, 1);
 | |
|             put_byte(&ctx->pb, 0);
 | |
|             put_byte(&ctx->pb, 2);
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     if (last_pkt) {
 | |
|         put_be32(&ctx->pb, ISO_11172_END_CODE);
 | |
|     }
 | |
|     /* output data */
 | |
|     put_buffer(&ctx->pb, stream->buffer, payload_size - stuffing_size);
 | |
|     put_flush_packet(&ctx->pb);
 | |
|     
 | |
|     /* preserve remaining data */
 | |
|     len = stream->buffer_ptr - payload_size;
 | |
|     if (len < 0) 
 | |
|         len = 0;
 | |
|     memmove(stream->buffer, stream->buffer + stream->buffer_ptr - len, len);
 | |
|     stream->buffer_ptr = len;
 | |
| 
 | |
|     s->packet_number++;
 | |
|     stream->packet_number++;
 | |
|     stream->start_pts = -1;
 | |
| }
 | |
| 
 | |
| static int mpeg_mux_write_packet(AVFormatContext *ctx, int stream_index,
 | |
|                                  UINT8 *buf, int size, int force_pts)
 | |
| {
 | |
|     MpegMuxContext *s = ctx->priv_data;
 | |
|     AVStream *st = ctx->streams[stream_index];
 | |
|     StreamInfo *stream = st->priv_data;
 | |
|     int len;
 | |
|     
 | |
|     while (size > 0) {
 | |
|         /* set pts */
 | |
|         if (stream->start_pts == -1) {
 | |
|             if (force_pts)
 | |
|                 stream->pts = force_pts;
 | |
|             stream->start_pts = stream->pts;
 | |
|         }
 | |
|         len = s->packet_data_max_size - stream->buffer_ptr;
 | |
|         if (len > size)
 | |
|             len = size;
 | |
|         memcpy(stream->buffer + stream->buffer_ptr, buf, len);
 | |
|         stream->buffer_ptr += len;
 | |
|         buf += len;
 | |
|         size -= len;
 | |
|         while (stream->buffer_ptr >= s->packet_data_max_size) {
 | |
|             /* output the packet */
 | |
|             if (stream->start_pts == -1)
 | |
|                 stream->start_pts = stream->pts;
 | |
|             flush_packet(ctx, stream_index, 0);
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     stream->pts += ticker_tick(&stream->pts_ticker, 1);
 | |
|     //if (st->codec.codec_type == CODEC_TYPE_VIDEO)
 | |
|     //    fprintf(stderr,"\nVideo PTS: %6lld", stream->pts);
 | |
|     //else
 | |
|     //    fprintf(stderr,"\nAudio PTS: %6lld", stream->pts);
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| static int mpeg_mux_end(AVFormatContext *ctx)
 | |
| {
 | |
|     StreamInfo *stream;
 | |
|     int i;
 | |
| 
 | |
|     /* flush each packet */
 | |
|     for(i=0;i<ctx->nb_streams;i++) {
 | |
|         stream = ctx->streams[i]->priv_data;
 | |
|         if (stream->buffer_ptr > 0) {
 | |
|             if (i == (ctx->nb_streams - 1)) 
 | |
|                 flush_packet(ctx, i, 1);
 | |
|             else
 | |
|                 flush_packet(ctx, i, 0);
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     /* write the end header */
 | |
|     //put_be32(&ctx->pb, ISO_11172_END_CODE);
 | |
|     //put_flush_packet(&ctx->pb);
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| /*********************************************/
 | |
| /* demux code */
 | |
| 
 | |
| #define MAX_SYNC_SIZE 100000
 | |
| 
 | |
| static int mpegps_probe(AVProbeData *p)
 | |
| {
 | |
|     int code, c, i;
 | |
|     code = 0xff;
 | |
| 
 | |
|     /* we search the first start code. If it is a packet start code,
 | |
|        then we decide it is mpeg ps. We do not send highest value to
 | |
|        give a chance to mpegts */
 | |
|     for(i=0;i<p->buf_size;i++) {
 | |
|         c = p->buf[i];
 | |
|         code = (code << 8) | c;
 | |
|         if ((code & 0xffffff00) == 0x100) {
 | |
|             if (code == PACK_START_CODE ||
 | |
|                 code == SYSTEM_HEADER_START_CODE ||
 | |
|                 (code >= 0x1e0 && code <= 0x1ef) ||
 | |
|                 (code >= 0x1c0 && code <= 0x1df) ||
 | |
|                 code == PRIVATE_STREAM_2 ||
 | |
|                 code == PROGRAM_STREAM_MAP ||
 | |
|                 code == PRIVATE_STREAM_1 ||
 | |
|                 code == PADDING_STREAM)
 | |
|                 return AVPROBE_SCORE_MAX - 1;
 | |
|             else
 | |
|                 return 0;
 | |
|         }
 | |
|     }
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| 
 | |
| typedef struct MpegDemuxContext {
 | |
|     int header_state;
 | |
| } MpegDemuxContext;
 | |
| 
 | |
| static int find_start_code(ByteIOContext *pb, int *size_ptr, 
 | |
|                            UINT32 *header_state)
 | |
| {
 | |
|     unsigned int state, v;
 | |
|     int val, n;
 | |
| 
 | |
|     state = *header_state;
 | |
|     n = *size_ptr;
 | |
|     while (n > 0) {
 | |
|         if (url_feof(pb))
 | |
|             break;
 | |
|         v = get_byte(pb);
 | |
|         n--;
 | |
|         if (state == 0x000001) {
 | |
|             state = ((state << 8) | v) & 0xffffff;
 | |
|             val = state;
 | |
|             goto found;
 | |
|         }
 | |
|         state = ((state << 8) | v) & 0xffffff;
 | |
|     }
 | |
|     val = -1;
 | |
|  found:
 | |
|     *header_state = state;
 | |
|     *size_ptr = n;
 | |
|     return val;
 | |
| }
 | |
| 
 | |
| static int mpegps_read_header(AVFormatContext *s,
 | |
|                                   AVFormatParameters *ap)
 | |
| {
 | |
|     MpegDemuxContext *m = s->priv_data;
 | |
|     m->header_state = 0xff;
 | |
|     /* no need to do more */
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| static INT64 get_pts(ByteIOContext *pb, int c)
 | |
| {
 | |
|     INT64 pts;
 | |
|     int val;
 | |
| 
 | |
|     if (c < 0)
 | |
|         c = get_byte(pb);
 | |
|     pts = (INT64)((c >> 1) & 0x07) << 30;
 | |
|     val = get_be16(pb);
 | |
|     pts |= (INT64)(val >> 1) << 15;
 | |
|     val = get_be16(pb);
 | |
|     pts |= (INT64)(val >> 1);
 | |
|     return pts;
 | |
| }
 | |
| 
 | |
| static int mpegps_read_packet(AVFormatContext *s,
 | |
|                                   AVPacket *pkt)
 | |
| {
 | |
|     MpegDemuxContext *m = s->priv_data;
 | |
|     AVStream *st;
 | |
|     int len, size, startcode, i, c, flags, header_len, type, codec_id;
 | |
|     INT64 pts, dts;
 | |
| 
 | |
|     /* next start code (should be immediately after */
 | |
|  redo:
 | |
|     m->header_state = 0xff;
 | |
|     size = MAX_SYNC_SIZE;
 | |
|     startcode = find_start_code(&s->pb, &size, &m->header_state);
 | |
|     //printf("startcode=%x pos=0x%Lx\n", startcode, url_ftell(&s->pb));
 | |
|     if (startcode < 0)
 | |
|         return -EIO;
 | |
|     if (startcode == PACK_START_CODE)
 | |
|         goto redo;
 | |
|     if (startcode == SYSTEM_HEADER_START_CODE)
 | |
|         goto redo;
 | |
|     if (startcode == PADDING_STREAM ||
 | |
|         startcode == PRIVATE_STREAM_2) {
 | |
|         /* skip them */
 | |
|         len = get_be16(&s->pb);
 | |
|         url_fskip(&s->pb, len);
 | |
|         goto redo;
 | |
|     }
 | |
|     /* find matching stream */
 | |
|     if (!((startcode >= 0x1c0 && startcode <= 0x1df) ||
 | |
|           (startcode >= 0x1e0 && startcode <= 0x1ef) ||
 | |
|           (startcode == 0x1bd)))
 | |
|         goto redo;
 | |
| 
 | |
|     len = get_be16(&s->pb);
 | |
|     pts = 0;
 | |
|     dts = 0;
 | |
|     /* stuffing */
 | |
|     for(;;) {
 | |
|         c = get_byte(&s->pb);
 | |
|         len--;
 | |
|         /* XXX: for mpeg1, should test only bit 7 */
 | |
|         if (c != 0xff) 
 | |
|             break;
 | |
|     }
 | |
|     if ((c & 0xc0) == 0x40) {
 | |
|         /* buffer scale & size */
 | |
|         get_byte(&s->pb);
 | |
|         c = get_byte(&s->pb);
 | |
|         len -= 2;
 | |
|     }
 | |
|     if ((c & 0xf0) == 0x20) {
 | |
|         pts = get_pts(&s->pb, c);
 | |
|         len -= 4;
 | |
|         dts = pts;
 | |
|     } else if ((c & 0xf0) == 0x30) {
 | |
|         pts = get_pts(&s->pb, c);
 | |
|         dts = get_pts(&s->pb, -1);
 | |
|         len -= 9;
 | |
|     } else if ((c & 0xc0) == 0x80) {
 | |
|         /* mpeg 2 PES */
 | |
|         if ((c & 0x30) != 0) {
 | |
|             fprintf(stderr, "Encrypted multiplex not handled\n");
 | |
|             return -EIO;
 | |
|         }
 | |
|         flags = get_byte(&s->pb);
 | |
|         header_len = get_byte(&s->pb);
 | |
|         len -= 2;
 | |
|         if (header_len > len)
 | |
|             goto redo;
 | |
|         if ((flags & 0xc0) == 0x40) {
 | |
|             pts = get_pts(&s->pb, -1);
 | |
|             dts = pts;
 | |
|             header_len -= 5;
 | |
|             len -= 5;
 | |
|         } if ((flags & 0xc0) == 0xc0) {
 | |
|             pts = get_pts(&s->pb, -1);
 | |
|             dts = get_pts(&s->pb, -1);
 | |
|             header_len -= 10;
 | |
|             len -= 10;
 | |
|         }
 | |
|         len -= header_len;
 | |
|         while (header_len > 0) {
 | |
|             get_byte(&s->pb);
 | |
|             header_len--;
 | |
|         }
 | |
|     }
 | |
|     if (startcode == 0x1bd) {
 | |
|         startcode = get_byte(&s->pb);
 | |
|         len--;
 | |
|         if (startcode >= 0x80 && startcode <= 0xbf) {
 | |
|             /* audio: skip header */
 | |
|             get_byte(&s->pb);
 | |
|             get_byte(&s->pb);
 | |
|             get_byte(&s->pb);
 | |
|             len -= 3;
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     /* now find stream */
 | |
|     for(i=0;i<s->nb_streams;i++) {
 | |
|         st = s->streams[i];
 | |
|         if (st->id == startcode)
 | |
|             goto found;
 | |
|     }
 | |
|     /* no stream found: add a new stream */
 | |
|     st = av_new_stream(s, startcode);
 | |
|     if (!st) 
 | |
|         goto skip;
 | |
|     if (startcode >= 0x1e0 && startcode <= 0x1ef) {
 | |
|         type = CODEC_TYPE_VIDEO;
 | |
|         codec_id = CODEC_ID_MPEG1VIDEO;
 | |
|     } else if (startcode >= 0x1c0 && startcode <= 0x1df) {
 | |
|         type = CODEC_TYPE_AUDIO;
 | |
|         codec_id = CODEC_ID_MP2;
 | |
|     } else if (startcode >= 0x80 && startcode <= 0x9f) {
 | |
|         type = CODEC_TYPE_AUDIO;
 | |
|         codec_id = CODEC_ID_AC3;
 | |
|     } else {
 | |
|     skip:
 | |
|         /* skip packet */
 | |
|         url_fskip(&s->pb, len);
 | |
|         goto redo;
 | |
|     }
 | |
|     st->codec.codec_type = type;
 | |
|     st->codec.codec_id = codec_id;
 | |
|  found:
 | |
|     av_new_packet(pkt, len);
 | |
|     //printf("\nRead Packet ID: %x PTS: %f Size: %d", startcode,
 | |
|     //       (float)pts/90000, len);
 | |
|     get_buffer(&s->pb, pkt->data, pkt->size);
 | |
|     pkt->pts = pts;
 | |
|     pkt->stream_index = st->index;
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| static int mpegps_read_close(AVFormatContext *s)
 | |
| {
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| static AVOutputFormat mpegps_mux = {
 | |
|     "mpeg",
 | |
|     "MPEG PS format",
 | |
|     "video/x-mpeg",
 | |
|     "mpg,mpeg,vob",
 | |
|     sizeof(MpegMuxContext),
 | |
|     CODEC_ID_MP2,
 | |
|     CODEC_ID_MPEG1VIDEO,
 | |
|     mpeg_mux_init,
 | |
|     mpeg_mux_write_packet,
 | |
|     mpeg_mux_end,
 | |
| };
 | |
| 
 | |
| static AVInputFormat mpegps_demux = {
 | |
|     "mpeg",
 | |
|     "MPEG PS format",
 | |
|     sizeof(MpegDemuxContext),
 | |
|     mpegps_probe,
 | |
|     mpegps_read_header,
 | |
|     mpegps_read_packet,
 | |
|     mpegps_read_close,
 | |
|     flags: AVFMT_NOHEADER,
 | |
| };
 | |
| 
 | |
| int mpegps_init(void)
 | |
| {
 | |
|     av_register_output_format(&mpegps_mux);
 | |
|     av_register_input_format(&mpegps_demux);
 | |
|     return 0;
 | |
| }
 | 
