Mercurial > libavformat.hg
annotate ffmenc.c @ 4563:90e0047f90b6 libavformat
consistent naming of Chinese AVS raw demuxer
author | stefang |
---|---|
date | Sun, 22 Feb 2009 14:27:50 +0000 |
parents | 77e0c7511d41 |
children | d426504e401b |
rev | line source |
---|---|
0 | 1 /* |
3348 | 2 * FFM (ffserver live feed) muxer |
4251
77e0c7511d41
cosmetics: Remove pointless period after copyright statement non-sentences.
diego
parents:
4201
diff
changeset
|
3 * Copyright (c) 2001 Fabrice Bellard |
0 | 4 * |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1169
diff
changeset
|
5 * This file is part of FFmpeg. |
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1169
diff
changeset
|
6 * |
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1169
diff
changeset
|
7 * FFmpeg is free software; you can redistribute it and/or |
0 | 8 * modify it under the terms of the GNU Lesser General Public |
9 * License as published by the Free Software Foundation; either | |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1169
diff
changeset
|
10 * version 2.1 of the License, or (at your option) any later version. |
0 | 11 * |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1169
diff
changeset
|
12 * FFmpeg is distributed in the hope that it will be useful, |
0 | 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
15 * Lesser General Public License for more details. | |
16 * | |
17 * You should have received a copy of the GNU Lesser General Public | |
1358
0899bfe4105c
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
1169
diff
changeset
|
18 * License along with FFmpeg; if not, write to the Free Software |
896
edbe5c3717f9
Update licensing information: The FSF changed postal address.
diego
parents:
887
diff
changeset
|
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
0 | 20 */ |
3348 | 21 |
4201
7d2f3f1b68d8
Fix build: Add intreadwrite.h and bswap.h #includes where necessary.
diego
parents:
3731
diff
changeset
|
22 #include "libavutil/intreadwrite.h" |
0 | 23 #include "avformat.h" |
3348 | 24 #include "ffm.h" |
0 | 25 |
26 static void flush_packet(AVFormatContext *s) | |
27 { | |
28 FFMContext *ffm = s->priv_data; | |
29 int fill_size, h; | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2222
diff
changeset
|
30 ByteIOContext *pb = s->pb; |
0 | 31 |
32 fill_size = ffm->packet_end - ffm->packet_ptr; | |
33 memset(ffm->packet_ptr, 0, fill_size); | |
34 | |
885 | 35 if (url_ftell(pb) % ffm->packet_size) |
318
54e915169d48
Add more resilience in reading ffm files. In particular, don't assume
philipjsg
parents:
277
diff
changeset
|
36 av_abort(); |
54e915169d48
Add more resilience in reading ffm files. In particular, don't assume
philipjsg
parents:
277
diff
changeset
|
37 |
0 | 38 /* put header */ |
39 put_be16(pb, PACKET_ID); | |
40 put_be16(pb, fill_size); | |
3492 | 41 put_be64(pb, ffm->dts); |
0 | 42 h = ffm->frame_offset; |
43 if (ffm->first_packet) | |
44 h |= 0x8000; | |
45 put_be16(pb, h); | |
46 put_buffer(pb, ffm->packet, ffm->packet_end - ffm->packet); | |
751
dcb459ca11eb
Flush the ffm packet to the wire (or file) whenever we flush the ffm packet.
philipjsg
parents:
744
diff
changeset
|
47 put_flush_packet(pb); |
0 | 48 |
49 /* prepare next packet */ | |
50 ffm->frame_offset = 0; /* no key frame */ | |
51 ffm->packet_ptr = ffm->packet; | |
52 ffm->first_packet = 0; | |
53 } | |
54 | |
55 /* 'first' is true if first data of a frame */ | |
56 static void ffm_write_data(AVFormatContext *s, | |
241 | 57 const uint8_t *buf, int size, |
3492 | 58 int64_t dts, int header) |
0 | 59 { |
60 FFMContext *ffm = s->priv_data; | |
61 int len; | |
62 | |
3488
222810cefa11
do not reset ffm pts, this avoids setting last pts to 0 and screwing up seeking
bcoudurier
parents:
3443
diff
changeset
|
63 if (header && ffm->frame_offset == 0) { |
0 | 64 ffm->frame_offset = ffm->packet_ptr - ffm->packet + FFM_HEADER_SIZE; |
3492 | 65 ffm->dts = dts; |
3488
222810cefa11
do not reset ffm pts, this avoids setting last pts to 0 and screwing up seeking
bcoudurier
parents:
3443
diff
changeset
|
66 } |
0 | 67 |
68 /* write as many packets as needed */ | |
69 while (size > 0) { | |
70 len = ffm->packet_end - ffm->packet_ptr; | |
71 if (len > size) | |
72 len = size; | |
73 memcpy(ffm->packet_ptr, buf, len); | |
74 | |
75 ffm->packet_ptr += len; | |
76 buf += len; | |
77 size -= len; | |
3490 | 78 if (ffm->packet_ptr >= ffm->packet_end) |
0 | 79 flush_packet(s); |
80 } | |
81 } | |
82 | |
83 static int ffm_write_header(AVFormatContext *s) | |
84 { | |
85 FFMContext *ffm = s->priv_data; | |
86 AVStream *st; | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2222
diff
changeset
|
87 ByteIOContext *pb = s->pb; |
0 | 88 AVCodecContext *codec; |
89 int bit_rate, i; | |
90 | |
91 ffm->packet_size = FFM_PACKET_SIZE; | |
92 | |
93 /* header */ | |
862
aa0abab5e320
fix feed read_header, avoid using put_tag in write_header, to be consistent with read_header, also some minor cosmetics
alex
parents:
858
diff
changeset
|
94 put_le32(pb, MKTAG('F', 'F', 'M', '1')); |
0 | 95 put_be32(pb, ffm->packet_size); |
96 /* XXX: store write position in other file ? */ | |
97 put_be64(pb, ffm->packet_size); /* current write position */ | |
98 | |
99 put_be32(pb, s->nb_streams); | |
100 bit_rate = 0; | |
101 for(i=0;i<s->nb_streams;i++) { | |
102 st = s->streams[i]; | |
820
feca73904e67
changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents:
775
diff
changeset
|
103 bit_rate += st->codec->bit_rate; |
0 | 104 } |
105 put_be32(pb, bit_rate); | |
106 | |
107 /* list of streams */ | |
108 for(i=0;i<s->nb_streams;i++) { | |
109 st = s->streams[i]; | |
502
813b0119a98e
ffserver fixes by (Koos Vriezen <koos.vriezen at xs4all dot nl>)
michael
parents:
468
diff
changeset
|
110 av_set_pts_info(st, 64, 1, 1000000); |
0 | 111 |
820
feca73904e67
changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents:
775
diff
changeset
|
112 codec = st->codec; |
0 | 113 /* generic info */ |
114 put_be32(pb, codec->codec_id); | |
115 put_byte(pb, codec->codec_type); | |
116 put_be32(pb, codec->bit_rate); | |
887 | 117 put_be32(pb, st->quality); |
0 | 118 put_be32(pb, codec->flags); |
744
da5b3b9e898e
Add in many fields that have been added to the Codec structure. This means
philipjsg
parents:
743
diff
changeset
|
119 put_be32(pb, codec->flags2); |
da5b3b9e898e
Add in many fields that have been added to the Codec structure. This means
philipjsg
parents:
743
diff
changeset
|
120 put_be32(pb, codec->debug); |
0 | 121 /* specific info */ |
122 switch(codec->codec_type) { | |
123 case CODEC_TYPE_VIDEO: | |
743 | 124 put_be32(pb, codec->time_base.num); |
125 put_be32(pb, codec->time_base.den); | |
0 | 126 put_be16(pb, codec->width); |
127 put_be16(pb, codec->height); | |
128 put_be16(pb, codec->gop_size); | |
744
da5b3b9e898e
Add in many fields that have been added to the Codec structure. This means
philipjsg
parents:
743
diff
changeset
|
129 put_be32(pb, codec->pix_fmt); |
0 | 130 put_byte(pb, codec->qmin); |
131 put_byte(pb, codec->qmax); | |
132 put_byte(pb, codec->max_qdiff); | |
133 put_be16(pb, (int) (codec->qcompress * 10000.0)); | |
134 put_be16(pb, (int) (codec->qblur * 10000.0)); | |
135 put_be32(pb, codec->bit_rate_tolerance); | |
3731 | 136 put_strz(pb, codec->rc_eq ? codec->rc_eq : "tex^qComp"); |
0 | 137 put_be32(pb, codec->rc_max_rate); |
138 put_be32(pb, codec->rc_min_rate); | |
139 put_be32(pb, codec->rc_buffer_size); | |
823 | 140 put_be64(pb, av_dbl2int(codec->i_quant_factor)); |
141 put_be64(pb, av_dbl2int(codec->b_quant_factor)); | |
142 put_be64(pb, av_dbl2int(codec->i_quant_offset)); | |
143 put_be64(pb, av_dbl2int(codec->b_quant_offset)); | |
0 | 144 put_be32(pb, codec->dct_algo); |
744
da5b3b9e898e
Add in many fields that have been added to the Codec structure. This means
philipjsg
parents:
743
diff
changeset
|
145 put_be32(pb, codec->strict_std_compliance); |
da5b3b9e898e
Add in many fields that have been added to the Codec structure. This means
philipjsg
parents:
743
diff
changeset
|
146 put_be32(pb, codec->max_b_frames); |
da5b3b9e898e
Add in many fields that have been added to the Codec structure. This means
philipjsg
parents:
743
diff
changeset
|
147 put_be32(pb, codec->luma_elim_threshold); |
da5b3b9e898e
Add in many fields that have been added to the Codec structure. This means
philipjsg
parents:
743
diff
changeset
|
148 put_be32(pb, codec->chroma_elim_threshold); |
da5b3b9e898e
Add in many fields that have been added to the Codec structure. This means
philipjsg
parents:
743
diff
changeset
|
149 put_be32(pb, codec->mpeg_quant); |
da5b3b9e898e
Add in many fields that have been added to the Codec structure. This means
philipjsg
parents:
743
diff
changeset
|
150 put_be32(pb, codec->intra_dc_precision); |
da5b3b9e898e
Add in many fields that have been added to the Codec structure. This means
philipjsg
parents:
743
diff
changeset
|
151 put_be32(pb, codec->me_method); |
da5b3b9e898e
Add in many fields that have been added to the Codec structure. This means
philipjsg
parents:
743
diff
changeset
|
152 put_be32(pb, codec->mb_decision); |
da5b3b9e898e
Add in many fields that have been added to the Codec structure. This means
philipjsg
parents:
743
diff
changeset
|
153 put_be32(pb, codec->nsse_weight); |
da5b3b9e898e
Add in many fields that have been added to the Codec structure. This means
philipjsg
parents:
743
diff
changeset
|
154 put_be32(pb, codec->frame_skip_cmp); |
823 | 155 put_be64(pb, av_dbl2int(codec->rc_buffer_aggressivity)); |
1809
491581a2b9a7
codec_tag settable via VideoTag, and transmit codec_tag in ffm
alex
parents:
1787
diff
changeset
|
156 put_be32(pb, codec->codec_tag); |
3489 | 157 put_byte(pb, codec->thread_count); |
0 | 158 break; |
159 case CODEC_TYPE_AUDIO: | |
160 put_be32(pb, codec->sample_rate); | |
161 put_le16(pb, codec->channels); | |
162 put_le16(pb, codec->frame_size); | |
163 break; | |
164 default: | |
537 | 165 return -1; |
0 | 166 } |
3441 | 167 if (codec->flags & CODEC_FLAG_GLOBAL_HEADER) { |
168 put_be32(pb, codec->extradata_size); | |
169 put_buffer(pb, codec->extradata, codec->extradata_size); | |
170 } | |
0 | 171 } |
172 | |
173 /* flush until end of block reached */ | |
174 while ((url_ftell(pb) % ffm->packet_size) != 0) | |
175 put_byte(pb, 0); | |
176 | |
177 put_flush_packet(pb); | |
178 | |
179 /* init packet mux */ | |
180 ffm->packet_ptr = ffm->packet; | |
181 ffm->packet_end = ffm->packet + ffm->packet_size - FFM_HEADER_SIZE; | |
537 | 182 assert(ffm->packet_end >= ffm->packet); |
0 | 183 ffm->frame_offset = 0; |
3492 | 184 ffm->dts = 0; |
0 | 185 ffm->first_packet = 1; |
186 | |
187 return 0; | |
188 } | |
189 | |
468 | 190 static int ffm_write_packet(AVFormatContext *s, AVPacket *pkt) |
0 | 191 { |
3491 | 192 int64_t dts; |
3527 | 193 uint8_t header[FRAME_HEADER_SIZE+4]; |
3442
a6f4d53b738d
pass dts and pts through ffm, should fix streaming b frames
bcoudurier
parents:
3441
diff
changeset
|
194 int header_size = FRAME_HEADER_SIZE; |
0 | 195 |
3491 | 196 dts = s->timestamp + pkt->dts; |
0 | 197 /* packet size & key_frame */ |
468 | 198 header[0] = pkt->stream_index; |
0 | 199 header[1] = 0; |
468 | 200 if (pkt->flags & PKT_FLAG_KEY) |
0 | 201 header[1] |= FLAG_KEY_FRAME; |
3312 | 202 AV_WB24(header+2, pkt->size); |
3310 | 203 AV_WB24(header+5, pkt->duration); |
3491 | 204 AV_WB64(header+8, s->timestamp + pkt->pts); |
3442
a6f4d53b738d
pass dts and pts through ffm, should fix streaming b frames
bcoudurier
parents:
3441
diff
changeset
|
205 if (pkt->pts != pkt->dts) { |
a6f4d53b738d
pass dts and pts through ffm, should fix streaming b frames
bcoudurier
parents:
3441
diff
changeset
|
206 header[1] |= FLAG_DTS; |
a6f4d53b738d
pass dts and pts through ffm, should fix streaming b frames
bcoudurier
parents:
3441
diff
changeset
|
207 AV_WB32(header+16, pkt->pts - pkt->dts); |
a6f4d53b738d
pass dts and pts through ffm, should fix streaming b frames
bcoudurier
parents:
3441
diff
changeset
|
208 header_size += 4; |
a6f4d53b738d
pass dts and pts through ffm, should fix streaming b frames
bcoudurier
parents:
3441
diff
changeset
|
209 } |
3491 | 210 ffm_write_data(s, header, header_size, dts, 1); |
211 ffm_write_data(s, pkt->data, pkt->size, dts, 0); | |
0 | 212 |
213 return 0; | |
214 } | |
215 | |
216 static int ffm_write_trailer(AVFormatContext *s) | |
217 { | |
2771
d52c718e83f9
Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents:
2222
diff
changeset
|
218 ByteIOContext *pb = s->pb; |
0 | 219 FFMContext *ffm = s->priv_data; |
220 | |
221 /* flush packets */ | |
222 if (ffm->packet_ptr > ffm->packet) | |
223 flush_packet(s); | |
224 | |
225 put_flush_packet(pb); | |
226 | |
227 if (!url_is_streamed(pb)) { | |
65 | 228 int64_t size; |
0 | 229 /* update the write offset */ |
230 size = url_ftell(pb); | |
231 url_fseek(pb, 8, SEEK_SET); | |
232 put_be64(pb, size); | |
233 put_flush_packet(pb); | |
234 } | |
235 | |
236 return 0; | |
237 } | |
238 | |
1169 | 239 AVOutputFormat ffm_muxer = { |
0 | 240 "ffm", |
3603 | 241 NULL_IF_CONFIG_SMALL("FFM (FFserver live feed) format"), |
0 | 242 "", |
243 "ffm", | |
244 sizeof(FFMContext), | |
245 /* not really used */ | |
246 CODEC_ID_MP2, | |
247 CODEC_ID_MPEG1VIDEO, | |
248 ffm_write_header, | |
249 ffm_write_packet, | |
250 ffm_write_trailer, | |
251 }; |