bit: loose seekback & wrong bitrate code.
[ffmpeg.git] / libavformat / bit.c
1 #include "avformat.h"
2 #include "libavcodec/get_bits.h"
3 #include "libavcodec/put_bits.h"
4
5 #define MAX_FRAME_SIZE 10
6
7 #define SYNC_WORD  0x6b21
8 #define BIT_0      0x7f
9 #define BIT_1      0x81
10
11 static int probe(AVProbeData *p)
12 {
13     int i, j;
14
15     for(i=0; i+3<p->buf_size && i< 10*0x50; ){
16         if(AV_RL16(&p->buf[0]) != SYNC_WORD)
17             return 0;
18         j=AV_RL16(&p->buf[2]);
19         if(j!=0x40 && j!=0x50)
20             return 0;
21         i+=j;
22     }
23     return AVPROBE_SCORE_MAX/2;
24 }
25
26 static int read_header(AVFormatContext *s, AVFormatParameters *ap)
27 {
28     AVStream* st;
29     ByteIOContext *pb = s->pb;
30
31     st=av_new_stream(s, 0);
32     if (!st)
33         return AVERROR(ENOMEM);
34
35     st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
36     st->codec->codec_id=CODEC_ID_G729;
37     st->codec->sample_rate=8000;
38     st->codec->block_align = 16;
39     st->codec->channels=1;
40
41     av_set_pts_info(st, 64, 1, 100);
42     return 0;
43 }
44
45 static int read_packet(AVFormatContext *s,
46                           AVPacket *pkt)
47 {
48     ByteIOContext *pb = s->pb;
49     PutBitContext pbo;
50     uint16_t buf[8 * MAX_FRAME_SIZE + 2];
51     int packet_size;
52     int sync;
53     uint16_t* src=buf;
54     int i, j, ret;
55
56     if(url_feof(pb))
57         return AVERROR_EOF;
58
59     sync = get_le16(pb); // sync word
60     packet_size = get_le16(pb) / 8;
61     if(packet_size > MAX_FRAME_SIZE)
62         return AVERROR(EIO);
63
64     ret = get_buffer(pb, (uint8_t*)buf, (8 * packet_size) * sizeof(uint16_t));
65     if(ret<0)
66         return ret;
67     if(ret != 8 * packet_size * sizeof(uint16_t))
68         return AVERROR(EIO);
69
70     av_new_packet(pkt, packet_size);
71
72     init_put_bits(&pbo, pkt->data, packet_size);
73     for(j=0; j < packet_size; j++)
74         for(i=0; i<8;i++)
75             put_bits(&pbo,1, AV_RL16(src++) == BIT_1 ? 1 : 0);
76
77     flush_put_bits(&pbo);
78
79     pkt->duration=1;
80     return 0;
81 }
82
83 AVInputFormat ff_bit_demuxer = {
84     "bit",
85     "G.729 BIT file format",
86     0,
87     probe,
88     read_header,
89     read_packet,
90     .extensions = "bit"
91 };
92
93 #ifdef CONFIG_MUXERS
94 static int write_header(AVFormatContext *s)
95 {
96     AVCodecContext *enc = s->streams[0]->codec;
97
98     enc->codec_id = CODEC_ID_G729;
99     enc->channels = 1;
100     enc->bits_per_coded_sample = 16;
101     enc->block_align = (enc->bits_per_coded_sample * enc->channels) >> 3;
102
103     return 0;
104 }
105
106 static int write_packet(AVFormatContext *s, AVPacket *pkt)
107 {
108     ByteIOContext *pb = s->pb;
109     GetBitContext gb;
110     int i;
111     uint16_t b;
112
113     put_le16(pb, SYNC_WORD);
114     put_le16(pb, 8 * 10);
115
116     init_get_bits(&gb, pkt->data, 8*10);
117     for(i=0; i< 8 * 10; i++)
118         put_le16(pb, get_bits1(&gb) ? BIT_1 : BIT_0);
119     put_flush_packet(pb);
120
121     return 0;
122 }
123
124 AVOutputFormat ff_bit_muxer = {
125     "bit",
126     "G.729 BIT file format",
127     "audio/bit",
128     "bit",
129     0,
130     CODEC_ID_G729,
131     CODEC_ID_NONE,
132     write_header,
133     write_packet,
134     .extensions = "bit"
135 };
136 #endif