]> git.sesse.net Git - ffmpeg/blob - libavformat/wtvenc.c
hevc: C code update for new motion compensation
[ffmpeg] / libavformat / wtvenc.c
1 /*
2  * Windows Television (WTV) muxer
3  * Copyright (c) 2011 Zhentan Feng <spyfeng at gmail dot com>
4  * Copyright (c) 2011 Peter Ross <pross@xvid.org>
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
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
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 /**
23  * @file
24  * Windows Television (WTV) demuxer
25  * @author Zhentan Feng <spyfeng at gmail dot com>
26  */
27
28 #include "libavutil/intreadwrite.h"
29 #include "libavutil/avassert.h"
30 #include "avformat.h"
31 #include "avio_internal.h"
32 #include "internal.h"
33 #include "mpegts.h"
34 #include "wtv.h"
35
36 #define WTV_BIGSECTOR_SIZE (1 << WTV_BIGSECTOR_BITS)
37 #define INDEX_BASE 0x2
38 #define MAX_NB_INDEX 10
39
40 /* declare utf16le strings */
41 #define _ , 0,
42 static const uint8_t timeline_table_0_header_events[] =
43     {'t'_'i'_'m'_'e'_'l'_'i'_'n'_'e'_'.'_'t'_'a'_'b'_'l'_'e'_'.'_'0'_'.'_'h'_'e'_'a'_'d'_'e'_'r'_'.'_'E'_'v'_'e'_'n'_'t'_'s', 0};
44 static const uint8_t table_0_header_legacy_attrib[] =
45     {'t'_'a'_'b'_'l'_'e'_'.'_'0'_'.'_'h'_'e'_'a'_'d'_'e'_'r'_'.'_'l'_'e'_'g'_'a'_'c'_'y'_'_'_'a'_'t'_'t'_'r'_'i'_'b', 0};
46 static const uint8_t table_0_redirector_legacy_attrib[] =
47     {'t'_'a'_'b'_'l'_'e'_'.'_'0'_'.'_'r'_'e'_'d'_'i'_'r'_'e'_'c'_'t'_'o'_'r'_'.'_'l'_'e'_'g'_'a'_'c'_'y'_'_'_'a'_'t'_'t'_'r'_'i'_'b', 0};
48 static const uint8_t table_0_header_time[] =
49     {'t'_'a'_'b'_'l'_'e'_'.'_'0'_'.'_'h'_'e'_'a'_'d'_'e'_'r'_'.'_'t'_'i'_'m'_'e', 0};
50 static const uint8_t legacy_attrib[] =
51     {'l'_'e'_'g'_'a'_'c'_'y'_'_'_'a'_'t'_'t'_'r'_'i'_'b', 0};
52 #undef _
53
54 static const ff_asf_guid sub_wtv_guid =
55     {0x8C,0xC3,0xD2,0xC2,0x7E,0x9A,0xDA,0x11,0x8B,0xF7,0x00,0x07,0xE9,0x5E,0xAD,0x8D};
56
57 enum WtvFileIndex {
58     WTV_TIMELINE_TABLE_0_HEADER_EVENTS = 0,
59     WTV_TIMELINE_TABLE_0_ENTRIES_EVENTS,
60     WTV_TIMELINE,
61     WTV_TABLE_0_HEADER_LEGACY_ATTRIB,
62     WTV_TABLE_0_ENTRIES_LEGACY_ATTRIB,
63     WTV_TABLE_0_REDIRECTOR_LEGACY_ATTRIB,
64     WTV_TABLE_0_HEADER_TIME,
65     WTV_TABLE_0_ENTRIES_TIME,
66     WTV_FILES
67 };
68
69 typedef struct {
70     int64_t length;
71     const void *header;
72     int depth;
73     int first_sector;
74 } WtvFile;
75
76 typedef struct {
77     int64_t             pos;
78     int64_t             serial;
79     const ff_asf_guid * guid;
80     int                 stream_id;
81 } WtvChunkEntry;
82
83 typedef struct {
84     int64_t serial;
85     int64_t value;
86 } WtvSyncEntry;
87
88 typedef struct {
89     int64_t timeline_start_pos;
90     WtvFile file[WTV_FILES];
91     int64_t serial;         /**< chunk serial number */
92     int64_t last_chunk_pos; /**< last chunk position */
93     int64_t last_timestamp_pos; /**< last timestamp chunk position */
94     int64_t first_index_pos;    /**< first index_chunk position */
95
96     WtvChunkEntry index[MAX_NB_INDEX];
97     int nb_index;
98     int first_video_flag;
99
100     WtvSyncEntry *st_pairs; /* (serial, timestamp) pairs */
101     int nb_st_pairs;
102     WtvSyncEntry *sp_pairs; /* (serial, position) pairs */
103     int nb_sp_pairs;
104
105     int64_t last_pts;
106     int64_t last_serial;
107
108     AVPacket thumbnail;
109 } WtvContext;
110
111
112 static void add_serial_pair(WtvSyncEntry ** list, int * count, int64_t serial, int64_t value)
113 {
114     int new_count = *count + 1;
115     WtvSyncEntry *new_list = av_realloc(*list, new_count * sizeof(WtvSyncEntry));
116     if (!new_list)
117         return;
118     new_list[*count] = (WtvSyncEntry){serial, value};
119     *list  = new_list;
120     *count = new_count;
121 }
122
123 typedef int WTVHeaderWriteFunc(AVIOContext *pb);
124
125 typedef struct {
126     const uint8_t *header;
127     int header_size;
128     WTVHeaderWriteFunc *write_header;
129 } WTVRootEntryTable;
130
131 #define write_pad(pb, size) ffio_fill(pb, 0, size)
132
133 /**
134  * Write chunk header. If header chunk (0x80000000 set) then add to list of header chunks
135  */
136 static void write_chunk_header(AVFormatContext *s, const ff_asf_guid *guid, int length, int stream_id)
137 {
138     WtvContext *wctx = s->priv_data;
139     AVIOContext *pb = s->pb;
140
141     wctx->last_chunk_pos = avio_tell(pb) - wctx->timeline_start_pos;
142     ff_put_guid(pb, guid);
143     avio_wl32(pb, 32 + length);
144     avio_wl32(pb, stream_id);
145     avio_wl64(pb, wctx->serial);
146
147     if ((stream_id & 0x80000000) && guid != &ff_index_guid) {
148         WtvChunkEntry *t = wctx->index + wctx->nb_index;
149         av_assert0(wctx->nb_index < MAX_NB_INDEX);
150         t->pos       = wctx->last_chunk_pos;
151         t->serial    = wctx->serial;
152         t->guid      = guid;
153         t->stream_id = stream_id & 0x3FFFFFFF;
154         wctx->nb_index++;
155     }
156 }
157
158 static void write_chunk_header2(AVFormatContext *s, const ff_asf_guid *guid, int stream_id)
159 {
160     WtvContext *wctx = s->priv_data;
161     AVIOContext *pb = s->pb;
162
163     int64_t last_chunk_pos = wctx->last_chunk_pos;
164     write_chunk_header(s, guid, 0, stream_id); // length updated later
165     avio_wl64(pb, last_chunk_pos);
166 }
167
168 static void finish_chunk_noindex(AVFormatContext *s)
169 {
170     WtvContext *wctx = s->priv_data;
171     AVIOContext *pb = s->pb;
172
173     // update the chunk_len field and pad.
174     int64_t chunk_len = avio_tell(pb) - (wctx->last_chunk_pos + wctx->timeline_start_pos);
175     avio_seek(pb, -(chunk_len - 16), SEEK_CUR);
176     avio_wl32(pb, chunk_len);
177     avio_seek(pb, chunk_len - (16 + 4), SEEK_CUR);
178
179     write_pad(pb, WTV_PAD8(chunk_len) - chunk_len);
180     wctx->serial++;
181 }
182
183 static void write_index(AVFormatContext *s)
184 {
185     AVIOContext *pb = s->pb;
186     WtvContext *wctx = s->priv_data;
187     int i;
188
189     write_chunk_header2(s, &ff_index_guid, 0x80000000);
190     avio_wl32(pb, 0);
191     avio_wl32(pb, 0);
192
193     for (i = 0; i < wctx->nb_index; i++) {
194         WtvChunkEntry *t = wctx->index + i;
195         ff_put_guid(pb,  t->guid);
196         avio_wl64(pb, t->pos);
197         avio_wl32(pb, t->stream_id);
198         avio_wl32(pb, 0); // checksum?
199         avio_wl64(pb, t->serial);
200     }
201     wctx->nb_index = 0;   // reset index
202     finish_chunk_noindex(s);
203
204     if (!wctx->first_index_pos)
205         wctx->first_index_pos = wctx->last_chunk_pos;
206 }
207
208 static void finish_chunk(AVFormatContext *s)
209 {
210     WtvContext *wctx = s->priv_data;
211     finish_chunk_noindex(s);
212     if (wctx->nb_index == MAX_NB_INDEX)
213         write_index(s);
214 }
215
216 static void put_videoinfoheader2(AVIOContext *pb, AVStream *st)
217 {
218     AVRational dar = av_mul_q(st->sample_aspect_ratio, (AVRational){st->codec->width, st->codec->height});
219     unsigned int num, den;
220     av_reduce(&num, &den, dar.num, dar.den, 0xFFFFFFFF);
221
222     /* VIDEOINFOHEADER2 */
223     avio_wl32(pb, 0);
224     avio_wl32(pb, 0);
225     avio_wl32(pb, st->codec->width);
226     avio_wl32(pb, st->codec->height);
227
228     avio_wl32(pb, 0);
229     avio_wl32(pb, 0);
230     avio_wl32(pb, 0);
231     avio_wl32(pb, 0);
232
233     avio_wl32(pb, st->codec->bit_rate);
234     avio_wl32(pb, 0);
235     avio_wl64(pb, st->avg_frame_rate.num && st->avg_frame_rate.den ? INT64_C(10000000) / av_q2d(st->avg_frame_rate) : 0);
236     avio_wl32(pb, 0);
237     avio_wl32(pb, 0);
238
239     avio_wl32(pb, num);
240     avio_wl32(pb, den);
241     avio_wl32(pb, 0);
242     avio_wl32(pb, 0);
243
244     ff_put_bmp_header(pb, st->codec, ff_codec_bmp_tags, 0, 1);
245
246     if (st->codec->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
247         int padding = (st->codec->extradata_size & 3) ? 4 - (st->codec->extradata_size & 3) : 0;
248         /* MPEG2VIDEOINFO */
249         avio_wl32(pb, 0);
250         avio_wl32(pb, st->codec->extradata_size + padding);
251         avio_wl32(pb, -1);
252         avio_wl32(pb, -1);
253         avio_wl32(pb, 0);
254         avio_write(pb, st->codec->extradata, st->codec->extradata_size);
255         ffio_fill(pb, 0, padding);
256     }
257 }
258
259 static int write_stream_codec_info(AVFormatContext *s, AVStream *st)
260 {
261     const ff_asf_guid *g, *media_type, *format_type;
262     const AVCodecTag *tags;
263     AVIOContext *pb = s->pb;
264     int64_t  hdr_pos_start;
265     int hdr_size = 0;
266
267     if (st->codec->codec_type  == AVMEDIA_TYPE_VIDEO) {
268         g = get_codec_guid(st->codec->codec_id, ff_video_guids);
269         media_type = &ff_mediatype_video;
270         format_type = st->codec->codec_id == AV_CODEC_ID_MPEG2VIDEO ? &ff_format_mpeg2_video : &ff_format_videoinfo2;
271         tags = ff_codec_bmp_tags;
272     } else if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
273         g = get_codec_guid(st->codec->codec_id, ff_codec_wav_guids);
274         media_type = &ff_mediatype_audio;
275         format_type = &ff_format_waveformatex;
276         tags = ff_codec_wav_tags;
277     } else {
278         av_log(s, AV_LOG_ERROR, "unknown codec_type (0x%x)\n", st->codec->codec_type);
279         return -1;
280     }
281
282     ff_put_guid(pb, media_type); // mediatype
283     ff_put_guid(pb, &ff_mediasubtype_cpfilters_processed); // subtype
284     write_pad(pb, 12);
285     ff_put_guid(pb,&ff_format_cpfilters_processed); // format type
286     avio_wl32(pb, 0); // size
287
288     hdr_pos_start = avio_tell(pb);
289     if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
290         put_videoinfoheader2(pb, st);
291     } else {
292         if (ff_put_wav_header(pb, st->codec, 0) < 0)
293             format_type = &ff_format_none;
294     }
295     hdr_size = avio_tell(pb) - hdr_pos_start;
296
297     // seek back write hdr_size
298     avio_seek(pb, -(hdr_size + 4), SEEK_CUR);
299     avio_wl32(pb, hdr_size + 32);
300     avio_seek(pb, hdr_size, SEEK_CUR);
301     if (g) {
302         ff_put_guid(pb, g);           // actual_subtype
303     } else {
304         int tag = ff_codec_get_tag(tags, st->codec->codec_id);
305         if (!tag) {
306             av_log(s, AV_LOG_ERROR, "unsupported codec_id (0x%x)\n", st->codec->codec_id);
307             return -1;
308         }
309         avio_wl32(pb, tag);
310         avio_write(pb, (const uint8_t[]){FF_MEDIASUBTYPE_BASE_GUID}, 12);
311     }
312     ff_put_guid(pb, format_type); // actual_formattype
313
314     return 0;
315 }
316
317 static int write_stream_codec(AVFormatContext *s, AVStream * st)
318 {
319     AVIOContext *pb = s->pb;
320     int ret;
321     write_chunk_header2(s, &ff_stream1_guid, 0x80000000 | 0x01);
322
323     avio_wl32(pb,  0x01);
324     write_pad(pb, 4);
325     write_pad(pb, 4);
326
327     ret = write_stream_codec_info(s, st);
328     if (ret < 0) {
329         av_log(s, AV_LOG_ERROR, "write stream codec info failed codec_type(0x%x)\n", st->codec->codec_type);
330         return -1;
331     }
332
333     finish_chunk(s);
334     return 0;
335 }
336
337 static void write_sync(AVFormatContext *s)
338 {
339     AVIOContext *pb = s->pb;
340     WtvContext *wctx = s->priv_data;
341     int64_t last_chunk_pos = wctx->last_chunk_pos;
342
343     write_chunk_header(s, &ff_sync_guid, 0x18, 0);
344     avio_wl64(pb, wctx->first_index_pos);
345     avio_wl64(pb, wctx->last_timestamp_pos);
346     avio_wl64(pb, 0);
347
348     finish_chunk(s);
349     add_serial_pair(&wctx->sp_pairs, &wctx->nb_sp_pairs, wctx->serial, wctx->last_chunk_pos);
350
351     wctx->last_chunk_pos = last_chunk_pos;
352 }
353
354 static int write_stream_data(AVFormatContext *s, AVStream *st)
355 {
356     AVIOContext *pb = s->pb;
357     int ret;
358
359     write_chunk_header2(s, &ff_SBE2_STREAM_DESC_EVENT, 0x80000000 | (st->index + INDEX_BASE));
360     avio_wl32(pb, 0x00000001);
361     avio_wl32(pb, st->index + INDEX_BASE); //stream_id
362     avio_wl32(pb, 0x00000001);
363     write_pad(pb, 8);
364
365     ret = write_stream_codec_info(s, st);
366     if (ret < 0) {
367         av_log(s, AV_LOG_ERROR, "write stream codec info failed codec_type(0x%x)\n", st->codec->codec_type);
368         return -1;
369     }
370     finish_chunk(s);
371
372     avpriv_set_pts_info(st, 64, 1, 10000000);
373
374     return 0;
375 }
376
377 static int write_header(AVFormatContext *s)
378 {
379     AVIOContext *pb = s->pb;
380     WtvContext *wctx = s->priv_data;
381     int i, pad, ret;
382     AVStream *st;
383
384     wctx->last_chunk_pos     = -1;
385     wctx->last_timestamp_pos = -1;
386
387     ff_put_guid(pb, &ff_wtv_guid);
388     ff_put_guid(pb, &sub_wtv_guid);
389
390     avio_wl32(pb, 0x01);
391     avio_wl32(pb, 0x02);
392     avio_wl32(pb, 1 << WTV_SECTOR_BITS);
393     avio_wl32(pb, 1 << WTV_BIGSECTOR_BITS);
394
395     //write initial root fields
396     avio_wl32(pb, 0); // root_size, update later
397     write_pad(pb, 4);
398     avio_wl32(pb, 0); // root_sector, update it later.
399
400     write_pad(pb, 32);
401     avio_wl32(pb, 0); // file ends pointer, update it later.
402
403     pad = (1 << WTV_SECTOR_BITS) - avio_tell(pb);
404     write_pad(pb, pad);
405
406     wctx->timeline_start_pos = avio_tell(pb);
407
408     wctx->serial = 1;
409     wctx->last_chunk_pos = -1;
410     wctx->first_video_flag = 1;
411
412     for (i = 0; i < s->nb_streams; i++) {
413         st = s->streams[i];
414         if (st->codec->codec_id == AV_CODEC_ID_MJPEG)
415             continue;
416         ret = write_stream_codec(s, st);
417         if (ret < 0) {
418             av_log(s, AV_LOG_ERROR, "write stream codec failed codec_type(0x%x)\n", st->codec->codec_type);
419             return -1;
420         }
421         if (!i)
422             write_sync(s);
423     }
424
425     for (i = 0; i < s->nb_streams; i++) {
426         st = s->streams[i];
427         if (st->codec->codec_id == AV_CODEC_ID_MJPEG)
428             continue;
429         ret  = write_stream_data(s, st);
430         if (ret < 0) {
431             av_log(s, AV_LOG_ERROR, "write stream data failed codec_type(0x%x)\n", st->codec->codec_type);
432             return -1;
433         }
434     }
435
436     if (wctx->nb_index)
437         write_index(s);
438
439     return 0;
440 }
441
442 static void write_timestamp(AVFormatContext *s, AVPacket *pkt)
443 {
444     AVIOContext *pb = s->pb;
445     WtvContext  *wctx = s->priv_data;
446     AVCodecContext *enc = s->streams[pkt->stream_index]->codec;
447
448     write_chunk_header(s, &ff_timestamp_guid, 56, 0x40000000 | (INDEX_BASE + pkt->stream_index));
449     write_pad(pb, 8);
450     avio_wl64(pb, pkt->pts == AV_NOPTS_VALUE ? -1 : pkt->pts);
451     avio_wl64(pb, pkt->pts == AV_NOPTS_VALUE ? -1 : pkt->pts);
452     avio_wl64(pb, pkt->pts == AV_NOPTS_VALUE ? -1 : pkt->pts);
453     avio_wl64(pb, 0);
454     avio_wl64(pb, enc->codec_type == AVMEDIA_TYPE_VIDEO && (pkt->flags & AV_PKT_FLAG_KEY) ? 1 : 0);
455     avio_wl64(pb, 0);
456
457     wctx->last_timestamp_pos = wctx->last_chunk_pos;
458 }
459
460 static int write_packet(AVFormatContext *s, AVPacket *pkt)
461 {
462     AVIOContext *pb = s->pb;
463     WtvContext  *wctx = s->priv_data;
464     AVStream    *st   = s->streams[pkt->stream_index];
465
466     if (st->codec->codec_id == AV_CODEC_ID_MJPEG && !wctx->thumbnail.size) {
467         av_copy_packet(&wctx->thumbnail, pkt);
468         return 0;
469     } else if (st->codec->codec_id == AV_CODEC_ID_H264) {
470         int ret = ff_check_h264_startcode(s, st, pkt);
471         if (ret < 0)
472             return ret;
473     }
474
475     /* emit sync chunk and 'timeline.table.0.entries.Event' record every 50 frames */
476     if (wctx->serial - (wctx->nb_sp_pairs ? wctx->sp_pairs[wctx->nb_sp_pairs - 1].serial : 0) >= 50)
477         write_sync(s);
478
479     /* emit 'table.0.entries.time' record every 500ms */
480     if (pkt->pts != AV_NOPTS_VALUE && pkt->pts - (wctx->nb_st_pairs ? wctx->st_pairs[wctx->nb_st_pairs - 1].value : 0) >= 5000000)
481         add_serial_pair(&wctx->st_pairs, &wctx->nb_st_pairs, wctx->serial, pkt->pts);
482
483     if (pkt->pts != AV_NOPTS_VALUE && pkt->pts > wctx->last_pts) {
484         wctx->last_pts = pkt->pts;
485         wctx->last_serial = wctx->serial;
486     }
487
488     // write timestamp chunk
489     write_timestamp(s, pkt);
490
491     write_chunk_header(s, &ff_data_guid, pkt->size, INDEX_BASE + pkt->stream_index);
492     avio_write(pb, pkt->data, pkt->size);
493     write_pad(pb, WTV_PAD8(pkt->size) - pkt->size);
494
495     wctx->serial++;
496     return 0;
497 }
498
499 static int write_table0_header_events(AVIOContext *pb)
500 {
501     avio_wl32(pb, 0x10);
502     write_pad(pb, 84);
503     avio_wl64(pb, 0x32);
504     return 96;
505 }
506
507 static int write_table0_header_legacy_attrib(AVIOContext *pb)
508 {
509     int pad = 0;
510     avio_wl32(pb, 0xFFFFFFFF);
511     write_pad(pb, 12);
512     avio_write(pb, legacy_attrib, sizeof(legacy_attrib));
513     pad = WTV_PAD8(sizeof(legacy_attrib)) - sizeof(legacy_attrib);
514     write_pad(pb, pad);
515     write_pad(pb, 32);
516     return 48 + WTV_PAD8(sizeof(legacy_attrib));
517 }
518
519 static int write_table0_header_time(AVIOContext *pb)
520 {
521     avio_wl32(pb, 0x10);
522     write_pad(pb, 76);
523     avio_wl64(pb, 0x40);
524     return 88;
525 }
526
527 static const WTVRootEntryTable wtv_root_entry_table[] = {
528     { timeline_table_0_header_events,          sizeof(timeline_table_0_header_events),          write_table0_header_events},
529     { ff_timeline_table_0_entries_Events_le16, sizeof(ff_timeline_table_0_entries_Events_le16), NULL},
530     { ff_timeline_le16,                        sizeof(ff_timeline_le16),                        NULL},
531     { table_0_header_legacy_attrib,            sizeof(table_0_header_legacy_attrib),            write_table0_header_legacy_attrib},
532     { ff_table_0_entries_legacy_attrib_le16,   sizeof(ff_table_0_entries_legacy_attrib_le16),   NULL},
533     { table_0_redirector_legacy_attrib,        sizeof(table_0_redirector_legacy_attrib),        NULL},
534     { table_0_header_time,                     sizeof(table_0_header_time),                     write_table0_header_time},
535     { ff_table_0_entries_time_le16,            sizeof(ff_table_0_entries_time_le16),            NULL},
536 };
537
538 static int write_root_table(AVFormatContext *s, int64_t sector_pos)
539 {
540     AVIOContext *pb = s->pb;
541     WtvContext  *wctx = s->priv_data;
542     int size, pad;
543     int i;
544
545     const WTVRootEntryTable *h = wtv_root_entry_table;
546     for (i = 0; i < sizeof(wtv_root_entry_table)/sizeof(WTVRootEntryTable); i++, h++) {
547         WtvFile *w = &wctx->file[i];
548         int filename_padding = WTV_PAD8(h->header_size) - h->header_size;
549         WTVHeaderWriteFunc *write = h->write_header;
550         int len = 0;
551         int64_t len_pos;
552
553         ff_put_guid(pb, &ff_dir_entry_guid);
554         len_pos = avio_tell(pb);
555         avio_wl16(pb, 40 + h->header_size + filename_padding + 8); // maybe updated later
556         write_pad(pb, 6);
557         avio_wl64(pb, write ? 0 : w->length);// maybe update later
558         avio_wl32(pb, (h->header_size + filename_padding) >> 1);
559         write_pad(pb, 4);
560
561         avio_write(pb, h->header, h->header_size);
562         write_pad(pb, filename_padding);
563
564         if (write) {
565             len = write(pb);
566             // update length field
567             avio_seek(pb, len_pos, SEEK_SET);
568             avio_wl64(pb, 40 + h->header_size + filename_padding + len);
569             avio_wl64(pb, len |(1ULL<<62) | (1ULL<<60));
570             avio_seek(pb, 8 + h->header_size + filename_padding + len, SEEK_CUR);
571         } else {
572             avio_wl32(pb, w->first_sector);
573             avio_wl32(pb, w->depth);
574         }
575     }
576
577     // caculate root table size
578     size = avio_tell(pb) - sector_pos;
579     pad = WTV_SECTOR_SIZE- size;
580     write_pad(pb, pad);
581
582     return size;
583 }
584
585 static void write_fat(AVIOContext *pb, int start_sector, int nb_sectors, int shift)
586 {
587     int i;
588     for (i = 0; i < nb_sectors; i++) {
589         avio_wl32(pb, start_sector + (i << shift));
590     }
591     // pad left sector pointer size
592     write_pad(pb, WTV_SECTOR_SIZE - ((nb_sectors << 2) % WTV_SECTOR_SIZE));
593 }
594
595 static int64_t write_fat_sector(AVFormatContext *s, int64_t start_pos, int nb_sectors, int sector_bits, int depth)
596 {
597     int64_t start_sector = start_pos >> WTV_SECTOR_BITS;
598     int shift = sector_bits - WTV_SECTOR_BITS;
599
600     int64_t fat = avio_tell(s->pb);
601     write_fat(s->pb, start_sector, nb_sectors, shift);
602
603     if (depth == 2) {
604         int64_t start_sector1 = fat >> WTV_SECTOR_BITS;
605         int nb_sectors1 = ((nb_sectors << 2) + WTV_SECTOR_SIZE - 1) / WTV_SECTOR_SIZE;
606         int64_t fat1 = avio_tell(s->pb);
607
608        write_fat(s->pb, start_sector1, nb_sectors1, 0);
609        return fat1;
610     }
611
612     return fat;
613 }
614
615 static void write_table_entries_events(AVFormatContext *s)
616 {
617     AVIOContext *pb = s->pb;
618     WtvContext *wctx = s->priv_data;
619     int i;
620     for (i = 0; i < wctx->nb_sp_pairs; i++) {
621         avio_wl64(pb, wctx->sp_pairs[i].serial);
622         avio_wl64(pb, wctx->sp_pairs[i].value);
623     }
624 }
625
626 static void write_table_entries_time(AVFormatContext *s)
627 {
628     AVIOContext *pb = s->pb;
629     WtvContext *wctx = s->priv_data;
630     int i;
631     for (i = 0; i < wctx->nb_st_pairs; i++) {
632         avio_wl64(pb, wctx->st_pairs[i].value);
633         avio_wl64(pb, wctx->st_pairs[i].serial);
634     }
635     avio_wl64(pb, wctx->last_pts);
636     avio_wl64(pb, wctx->last_serial);
637 }
638
639 static void write_metadata_header(AVIOContext *pb, int type, const char *key, int value_size)
640 {
641     ff_put_guid(pb, &ff_metadata_guid);
642     avio_wl32(pb, type);
643     avio_wl32(pb, value_size);
644     avio_put_str16le(pb, key);
645 }
646
647 static int metadata_header_size(const char *key)
648 {
649     return 16 + 4 + 4 + strlen(key)*2 + 2;
650 }
651
652 static void write_tag_int32(AVIOContext *pb, const char *key, int value)
653 {
654     write_metadata_header(pb, 0, key, 4);
655     avio_wl32(pb, value);
656 }
657
658 static void write_tag(AVIOContext *pb, const char *key, const char *value)
659 {
660     write_metadata_header(pb, 1, key, strlen(value)*2 + 2);
661     avio_put_str16le(pb, value);
662 }
663
664 static int attachment_value_size(const AVPacket *pkt, const AVDictionaryEntry *e)
665 {
666     return strlen("image/jpeg")*2 + 2 + 1 + (e ? strlen(e->value)*2 : 0) + 2 + 4 + pkt->size;
667 }
668
669 static void write_table_entries_attrib(AVFormatContext *s)
670 {
671     WtvContext *wctx = s->priv_data;
672     AVIOContext *pb = s->pb;
673     AVDictionaryEntry *tag = 0;
674
675     //FIXME: translate special tags (e.g. WM/Bitrate) to binary representation
676     ff_metadata_conv(&s->metadata, ff_asf_metadata_conv, NULL);
677     while ((tag = av_dict_get(s->metadata, "", tag, AV_DICT_IGNORE_SUFFIX)))
678         write_tag(pb, tag->key, tag->value);
679
680     if (wctx->thumbnail.size) {
681         AVStream *st = s->streams[wctx->thumbnail.stream_index];
682         tag = av_dict_get(st->metadata, "title", NULL, 0);
683         write_metadata_header(pb, 2, "WM/Picture", attachment_value_size(&wctx->thumbnail, tag));
684
685         avio_put_str16le(pb, "image/jpeg");
686         avio_w8(pb, 0x10);
687         avio_put_str16le(pb, tag ? tag->value : "");
688
689         avio_wl32(pb, wctx->thumbnail.size);
690         avio_write(pb, wctx->thumbnail.data, wctx->thumbnail.size);
691
692         write_tag_int32(pb, "WM/MediaThumbType", 2);
693     }
694 }
695
696 static void write_table_redirector_legacy_attrib(AVFormatContext *s)
697 {
698     WtvContext *wctx = s->priv_data;
699     AVIOContext *pb = s->pb;
700     AVDictionaryEntry *tag = 0;
701     int64_t pos = 0;
702
703     //FIXME: translate special tags to binary representation
704     while ((tag = av_dict_get(s->metadata, "", tag, AV_DICT_IGNORE_SUFFIX))) {
705         avio_wl64(pb, pos);
706         pos += metadata_header_size(tag->key) + strlen(tag->value)*2 + 2;
707     }
708
709     if (wctx->thumbnail.size) {
710         AVStream *st = s->streams[wctx->thumbnail.stream_index];
711         avio_wl64(pb, pos);
712         pos += metadata_header_size("WM/Picture") + attachment_value_size(&wctx->thumbnail, av_dict_get(st->metadata, "title", NULL, 0));
713
714         avio_wl64(pb, pos);
715         pos += metadata_header_size("WM/MediaThumbType") + 4;
716     }
717 }
718
719 /**
720  * Pad the remainder of a file
721  * Write out fat table
722  * @return <0 on error
723  */
724 static int finish_file(AVFormatContext *s, enum WtvFileIndex index, int64_t start_pos)
725 {
726     WtvContext *wctx = s->priv_data;
727     AVIOContext *pb = s->pb;
728     WtvFile *w = &wctx->file[index];
729     int64_t end_pos = avio_tell(pb);
730     int sector_bits, nb_sectors, pad;
731
732     av_assert0(index < WTV_FILES);
733
734     w->length = (end_pos - start_pos);
735
736     // determine optimal fat table depth, sector_bits, nb_sectors
737     if (w->length <= WTV_SECTOR_SIZE) {
738         w->depth = 0;
739         sector_bits = WTV_SECTOR_BITS;
740     } else if (w->length <= (WTV_SECTOR_SIZE / 4) * WTV_SECTOR_SIZE) {
741         w->depth = 1;
742         sector_bits = WTV_SECTOR_BITS;
743     } else if (w->length <= (WTV_SECTOR_SIZE / 4) * WTV_BIGSECTOR_SIZE) {
744         w->depth = 1;
745         sector_bits = WTV_BIGSECTOR_BITS;
746     } else if (w->length <= (int64_t)(WTV_SECTOR_SIZE / 4) * (WTV_SECTOR_SIZE / 4) * WTV_SECTOR_SIZE) {
747         w->depth = 2;
748         sector_bits = WTV_SECTOR_BITS;
749     } else if (w->length <= (int64_t)(WTV_SECTOR_SIZE / 4) * (WTV_SECTOR_SIZE / 4) * WTV_BIGSECTOR_SIZE) {
750         w->depth = 2;
751         sector_bits = WTV_BIGSECTOR_BITS;
752     } else {
753         av_log(s, AV_LOG_ERROR, "unsupported file allocation table depth (%"PRIi64" bytes)\n", w->length);
754         return -1;
755     }
756
757     // determine the nb_sectors
758     nb_sectors = (int)(w->length >> sector_bits);
759
760     // pad sector of timeline
761     pad = (1 << sector_bits) - (w->length % (1 << sector_bits));
762     if (pad) {
763         nb_sectors++;
764         write_pad(pb, pad);
765     }
766
767     //write fat table
768     if (w->depth > 0) {
769         w->first_sector = write_fat_sector(s, start_pos, nb_sectors, sector_bits, w->depth) >> WTV_SECTOR_BITS;
770     } else {
771         w->first_sector = start_pos >> WTV_SECTOR_BITS;
772     }
773
774     w->length |= 1ULL<<60;
775     if (sector_bits == WTV_SECTOR_BITS)
776         w->length |= 1ULL<<63;
777
778     return 0;
779 }
780
781 static int write_trailer(AVFormatContext *s)
782 {
783     WtvContext *wctx = s->priv_data;
784     AVIOContext *pb = s->pb;
785     int root_size;
786     int64_t sector_pos;
787     int64_t start_pos, file_end_pos;
788
789     if (finish_file(s, WTV_TIMELINE, wctx->timeline_start_pos) < 0)
790         return -1;
791
792     start_pos = avio_tell(pb);
793     write_table_entries_events(s);
794     if (finish_file(s, WTV_TIMELINE_TABLE_0_ENTRIES_EVENTS, start_pos) < 0)
795         return -1;
796
797     start_pos = avio_tell(pb);
798     write_table_entries_attrib(s);
799     if (finish_file(s, WTV_TABLE_0_ENTRIES_LEGACY_ATTRIB, start_pos) < 0)
800         return -1;
801
802     start_pos = avio_tell(pb);
803     write_table_redirector_legacy_attrib(s);
804     if (finish_file(s, WTV_TABLE_0_REDIRECTOR_LEGACY_ATTRIB, start_pos) < 0)
805         return -1;
806
807     start_pos = avio_tell(pb);
808     write_table_entries_time(s);
809     if (finish_file(s, WTV_TABLE_0_ENTRIES_TIME, start_pos) < 0)
810         return -1;
811
812     // write root table
813     sector_pos = avio_tell(pb);
814     root_size = write_root_table(s, sector_pos);
815
816     file_end_pos = avio_tell(pb);
817     // update root value
818     avio_seek(pb, 0x30, SEEK_SET);
819     avio_wl32(pb, root_size);
820     avio_seek(pb, 4, SEEK_CUR);
821     avio_wl32(pb, sector_pos >> WTV_SECTOR_BITS);
822     avio_seek(pb, 0x5c, SEEK_SET);
823     avio_wl32(pb, file_end_pos >> WTV_SECTOR_BITS);
824
825     avio_flush(pb);
826
827     av_free(wctx->sp_pairs);
828     av_free(wctx->st_pairs);
829     av_free_packet(&wctx->thumbnail);
830     return 0;
831 }
832
833 AVOutputFormat ff_wtv_muxer = {
834     .name           = "wtv",
835     .long_name      = NULL_IF_CONFIG_SMALL("Windows Television (WTV)"),
836     .extensions     = "wtv",
837     .priv_data_size = sizeof(WtvContext),
838     .audio_codec    = AV_CODEC_ID_AC3,
839     .video_codec    = AV_CODEC_ID_MPEG2VIDEO,
840     .write_header   = write_header,
841     .write_packet   = write_packet,
842     .write_trailer  = write_trailer,
843     .codec_tag      = (const AVCodecTag* const []){ ff_codec_bmp_tags,
844                                                     ff_codec_wav_tags, 0 },
845 };