yading@11: /* yading@11: * Windows Television (WTV) muxer yading@11: * Copyright (c) 2011 Zhentan Feng yading@11: * Copyright (c) 2011 Peter Ross yading@11: * This file is part of FFmpeg. yading@11: * yading@11: * FFmpeg is free software; you can redistribute it and/or yading@11: * modify it under the terms of the GNU Lesser General Public yading@11: * License as published by the Free Software Foundation; either yading@11: * version 2.1 of the License, or (at your option) any later version. yading@11: * yading@11: * FFmpeg is distributed in the hope that it will be useful, yading@11: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@11: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@11: * Lesser General Public License for more details. yading@11: * yading@11: * You should have received a copy of the GNU Lesser General Public yading@11: * License along with FFmpeg; if not, write to the Free Software yading@11: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@11: */ yading@11: yading@11: /** yading@11: * @file yading@11: * Windows Television (WTV) demuxer yading@11: * @author Zhentan Feng yading@11: */ yading@11: yading@11: #include "libavutil/intreadwrite.h" yading@11: #include "libavutil/avassert.h" yading@11: #include "avformat.h" yading@11: #include "internal.h" yading@11: #include "wtv.h" yading@11: #include "asf.h" yading@11: yading@11: #define WTV_BIGSECTOR_SIZE (1 << WTV_BIGSECTOR_BITS) yading@11: #define INDEX_BASE 0x2 yading@11: #define MAX_NB_INDEX 10 yading@11: yading@11: /* declare utf16le strings */ yading@11: #define _ , 0, yading@11: static const uint8_t timeline_table_0_header_events[] = yading@11: {'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}; yading@11: static const uint8_t table_0_header_legacy_attrib[] = yading@11: {'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}; yading@11: static const uint8_t table_0_redirector_legacy_attrib[] = yading@11: {'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}; yading@11: static const uint8_t table_0_header_time[] = yading@11: {'t'_'a'_'b'_'l'_'e'_'.'_'0'_'.'_'h'_'e'_'a'_'d'_'e'_'r'_'.'_'t'_'i'_'m'_'e', 0}; yading@11: static const uint8_t legacy_attrib[] = yading@11: {'l'_'e'_'g'_'a'_'c'_'y'_'_'_'a'_'t'_'t'_'r'_'i'_'b', 0}; yading@11: #undef _ yading@11: yading@11: static const ff_asf_guid sub_wtv_guid = yading@11: {0x8C,0xC3,0xD2,0xC2,0x7E,0x9A,0xDA,0x11,0x8B,0xF7,0x00,0x07,0xE9,0x5E,0xAD,0x8D}; yading@11: yading@11: enum WtvFileIndex { yading@11: WTV_TIMELINE_TABLE_0_HEADER_EVENTS = 0, yading@11: WTV_TIMELINE_TABLE_0_ENTRIES_EVENTS, yading@11: WTV_TIMELINE, yading@11: WTV_TABLE_0_HEADER_LEGACY_ATTRIB, yading@11: WTV_TABLE_0_ENTRIES_LEGACY_ATTRIB, yading@11: WTV_TABLE_0_REDIRECTOR_LEGACY_ATTRIB, yading@11: WTV_TABLE_0_HEADER_TIME, yading@11: WTV_TABLE_0_ENTRIES_TIME, yading@11: WTV_FILES yading@11: }; yading@11: yading@11: typedef struct { yading@11: int64_t length; yading@11: const void *header; yading@11: int depth; yading@11: int first_sector; yading@11: } WtvFile; yading@11: yading@11: typedef struct { yading@11: int64_t pos; yading@11: int64_t serial; yading@11: const ff_asf_guid * guid; yading@11: int stream_id; yading@11: } WtvChunkEntry; yading@11: yading@11: typedef struct { yading@11: int64_t serial; yading@11: int64_t value; yading@11: } WtvSyncEntry; yading@11: yading@11: typedef struct { yading@11: int64_t timeline_start_pos; yading@11: WtvFile file[WTV_FILES]; yading@11: int64_t serial; /** chunk serial number */ yading@11: int64_t last_chunk_pos; /** last chunk position */ yading@11: int64_t last_timestamp_pos; /** last timestamp chunk position */ yading@11: int64_t first_index_pos; /** first index_chunk position */ yading@11: yading@11: WtvChunkEntry index[MAX_NB_INDEX]; yading@11: int nb_index; yading@11: int first_video_flag; yading@11: yading@11: WtvSyncEntry *st_pairs; /* (serial, timestamp) pairs */ yading@11: int nb_st_pairs; yading@11: WtvSyncEntry *sp_pairs; /* (serial, position) pairs */ yading@11: int nb_sp_pairs; yading@11: yading@11: int64_t last_pts; yading@11: int64_t last_serial; yading@11: yading@11: AVPacket thumbnail; yading@11: } WtvContext; yading@11: yading@11: yading@11: static void add_serial_pair(WtvSyncEntry ** list, int * count, int64_t serial, int64_t value) yading@11: { yading@11: int new_count = *count + 1; yading@11: WtvSyncEntry *new_list = av_realloc(*list, new_count * sizeof(WtvSyncEntry)); yading@11: if (!new_list) yading@11: return; yading@11: new_list[*count] = (WtvSyncEntry){serial, value}; yading@11: *list = new_list; yading@11: *count = new_count; yading@11: } yading@11: yading@11: typedef int WTVHeaderWriteFunc(AVIOContext *pb); yading@11: yading@11: typedef struct { yading@11: const uint8_t *header; yading@11: int header_size; yading@11: WTVHeaderWriteFunc *write_header; yading@11: } WTVRootEntryTable; yading@11: yading@11: static int write_pad(AVIOContext *pb, int size) yading@11: { yading@11: for (; size > 0; size--) yading@11: avio_w8(pb, 0); yading@11: return 0; yading@11: } yading@11: yading@11: static const ff_asf_guid *get_codec_guid(enum AVCodecID id, const AVCodecGuid *av_guid) yading@11: { yading@11: int i; yading@11: for (i = 0; av_guid[i].id != AV_CODEC_ID_NONE; i++) { yading@11: if (id == av_guid[i].id) yading@11: return &(av_guid[i].guid); yading@11: } yading@11: return NULL; yading@11: } yading@11: yading@11: /** yading@11: * Write chunk header. If header chunk (0x80000000 set) then add to list of header chunks yading@11: */ yading@11: static void write_chunk_header(AVFormatContext *s, const ff_asf_guid *guid, int length, int stream_id) yading@11: { yading@11: WtvContext *wctx = s->priv_data; yading@11: AVIOContext *pb = s->pb; yading@11: yading@11: wctx->last_chunk_pos = avio_tell(pb) - wctx->timeline_start_pos; yading@11: ff_put_guid(pb, guid); yading@11: avio_wl32(pb, 32 + length); yading@11: avio_wl32(pb, stream_id); yading@11: avio_wl64(pb, wctx->serial); yading@11: yading@11: if ((stream_id & 0x80000000) && guid != &ff_index_guid) { yading@11: WtvChunkEntry *t = wctx->index + wctx->nb_index; yading@11: av_assert0(wctx->nb_index < MAX_NB_INDEX); yading@11: t->pos = wctx->last_chunk_pos; yading@11: t->serial = wctx->serial; yading@11: t->guid = guid; yading@11: t->stream_id = stream_id & 0x3FFFFFFF; yading@11: wctx->nb_index++; yading@11: } yading@11: } yading@11: yading@11: static void write_chunk_header2(AVFormatContext *s, const ff_asf_guid *guid, int stream_id) yading@11: { yading@11: WtvContext *wctx = s->priv_data; yading@11: AVIOContext *pb = s->pb; yading@11: yading@11: int64_t last_chunk_pos = wctx->last_chunk_pos; yading@11: write_chunk_header(s, guid, 0, stream_id); // length updated later yading@11: avio_wl64(pb, last_chunk_pos); yading@11: } yading@11: yading@11: static void finish_chunk_noindex(AVFormatContext *s) yading@11: { yading@11: WtvContext *wctx = s->priv_data; yading@11: AVIOContext *pb = s->pb; yading@11: yading@11: // update the chunk_len field and pad. yading@11: int64_t chunk_len = avio_tell(pb) - (wctx->last_chunk_pos + wctx->timeline_start_pos); yading@11: avio_seek(pb, -(chunk_len - 16), SEEK_CUR); yading@11: avio_wl32(pb, chunk_len); yading@11: avio_seek(pb, chunk_len - (16 + 4), SEEK_CUR); yading@11: yading@11: write_pad(pb, WTV_PAD8(chunk_len) - chunk_len); yading@11: wctx->serial++; yading@11: } yading@11: yading@11: static void write_index(AVFormatContext *s) yading@11: { yading@11: AVIOContext *pb = s->pb; yading@11: WtvContext *wctx = s->priv_data; yading@11: int i; yading@11: yading@11: write_chunk_header2(s, &ff_index_guid, 0x80000000); yading@11: avio_wl32(pb, 0); yading@11: avio_wl32(pb, 0); yading@11: yading@11: for (i = 0; i < wctx->nb_index; i++) { yading@11: WtvChunkEntry *t = wctx->index + i; yading@11: ff_put_guid(pb, t->guid); yading@11: avio_wl64(pb, t->pos); yading@11: avio_wl32(pb, t->stream_id); yading@11: avio_wl32(pb, 0); // checksum? yading@11: avio_wl64(pb, t->serial); yading@11: } yading@11: wctx->nb_index = 0; // reset index yading@11: finish_chunk_noindex(s); yading@11: yading@11: if (!wctx->first_index_pos) yading@11: wctx->first_index_pos = wctx->last_chunk_pos; yading@11: } yading@11: yading@11: static void finish_chunk(AVFormatContext *s) yading@11: { yading@11: WtvContext *wctx = s->priv_data; yading@11: finish_chunk_noindex(s); yading@11: if (wctx->nb_index == MAX_NB_INDEX) yading@11: write_index(s); yading@11: } yading@11: yading@11: static int write_stream_codec_info(AVFormatContext *s, AVStream *st) yading@11: { yading@11: WtvContext *wctx = s->priv_data; yading@11: const ff_asf_guid *g, *media_type, *format_type; yading@11: AVIOContext *pb = s->pb; yading@11: int64_t hdr_pos_start; yading@11: int hdr_size = 0; yading@11: yading@11: if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) { yading@11: g = get_codec_guid(st->codec->codec_id, ff_video_guids); yading@11: media_type = &ff_mediatype_video; yading@11: format_type = &ff_format_mpeg2_video; yading@11: } else if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO) { yading@11: g = get_codec_guid(st->codec->codec_id, ff_codec_wav_guids); yading@11: media_type = &ff_mediatype_audio; yading@11: format_type = &ff_format_waveformatex; yading@11: } else { yading@11: av_log(s, AV_LOG_ERROR, "unknown codec_type (0x%x)\n", st->codec->codec_type); yading@11: return -1; yading@11: } yading@11: yading@11: if (g == NULL) { yading@11: av_log(s, AV_LOG_ERROR, "can't get video codec_id (0x%x) guid.\n", st->codec->codec_id); yading@11: return -1; yading@11: } yading@11: yading@11: ff_put_guid(pb, media_type); // mediatype yading@11: ff_put_guid(pb, &ff_mediasubtype_cpfilters_processed); // subtype yading@11: write_pad(pb, 12); yading@11: ff_put_guid(pb,&ff_format_cpfilters_processed); // format type yading@11: avio_wl32(pb, 0); // size yading@11: yading@11: hdr_pos_start = avio_tell(pb); yading@11: if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) { yading@11: if (wctx->first_video_flag) { yading@11: write_pad(pb, 216); //The size is sensitive. yading@11: wctx->first_video_flag = 0; yading@11: } else { yading@11: write_pad(pb, 72); // aspect ratio yading@11: ff_put_bmp_header(pb, st->codec, ff_codec_bmp_tags, 0); yading@11: } yading@11: } else { yading@11: ff_put_wav_header(pb, st->codec); yading@11: } yading@11: hdr_size = avio_tell(pb) - hdr_pos_start; yading@11: yading@11: // seek back write hdr_size yading@11: avio_seek(pb, -(hdr_size + 4), SEEK_CUR); yading@11: avio_wl32(pb, hdr_size + 32); yading@11: avio_seek(pb, hdr_size, SEEK_CUR); yading@11: ff_put_guid(pb, g); // actual_subtype yading@11: ff_put_guid(pb, format_type); // actual_formattype yading@11: yading@11: return 0; yading@11: } yading@11: yading@11: static int write_stream_codec(AVFormatContext *s, AVStream * st) yading@11: { yading@11: AVIOContext *pb = s->pb; yading@11: int ret; yading@11: write_chunk_header2(s, &ff_stream1_guid, 0x80000000 | 0x01); yading@11: yading@11: avio_wl32(pb, 0x01); yading@11: write_pad(pb, 4); yading@11: write_pad(pb, 4); yading@11: yading@11: ret = write_stream_codec_info(s, st); yading@11: if (ret < 0) { yading@11: av_log(s, AV_LOG_ERROR, "write stream codec info failed codec_type(0x%x)\n", st->codec->codec_type); yading@11: return -1; yading@11: } yading@11: yading@11: finish_chunk(s); yading@11: return 0; yading@11: } yading@11: yading@11: static void write_sync(AVFormatContext *s) yading@11: { yading@11: AVIOContext *pb = s->pb; yading@11: WtvContext *wctx = s->priv_data; yading@11: int64_t last_chunk_pos = wctx->last_chunk_pos; yading@11: yading@11: write_chunk_header(s, &ff_sync_guid, 0x18, 0); yading@11: avio_wl64(pb, wctx->first_index_pos); yading@11: avio_wl64(pb, wctx->last_timestamp_pos); yading@11: avio_wl64(pb, 0); yading@11: yading@11: finish_chunk(s); yading@11: add_serial_pair(&wctx->sp_pairs, &wctx->nb_sp_pairs, wctx->serial, wctx->last_chunk_pos); yading@11: yading@11: wctx->last_chunk_pos = last_chunk_pos; yading@11: } yading@11: yading@11: static int write_stream_data(AVFormatContext *s, AVStream *st) yading@11: { yading@11: AVIOContext *pb = s->pb; yading@11: int ret; yading@11: yading@11: write_chunk_header2(s, &ff_SBE2_STREAM_DESC_EVENT, 0x80000000 | (st->index + INDEX_BASE)); yading@11: avio_wl32(pb, 0x00000001); yading@11: avio_wl32(pb, st->index + INDEX_BASE); //stream_id yading@11: avio_wl32(pb, 0x00000001); yading@11: write_pad(pb, 8); yading@11: yading@11: ret = write_stream_codec_info(s, st); yading@11: if (ret < 0) { yading@11: av_log(s, AV_LOG_ERROR, "write stream codec info failed codec_type(0x%x)\n", st->codec->codec_type); yading@11: return -1; yading@11: } yading@11: finish_chunk(s); yading@11: yading@11: avpriv_set_pts_info(st, 64, 1, 10000000); yading@11: yading@11: return 0; yading@11: } yading@11: yading@11: static int write_header(AVFormatContext *s) yading@11: { yading@11: AVIOContext *pb = s->pb; yading@11: WtvContext *wctx = s->priv_data; yading@11: int i, pad, ret; yading@11: AVStream *st; yading@11: yading@11: wctx->last_chunk_pos = -1; yading@11: wctx->last_timestamp_pos = -1; yading@11: yading@11: ff_put_guid(pb, &ff_wtv_guid); yading@11: ff_put_guid(pb, &sub_wtv_guid); yading@11: yading@11: avio_wl32(pb, 0x01); yading@11: avio_wl32(pb, 0x02); yading@11: avio_wl32(pb, 1 << WTV_SECTOR_BITS); yading@11: avio_wl32(pb, 1 << WTV_BIGSECTOR_BITS); yading@11: yading@11: //write initial root fields yading@11: avio_wl32(pb, 0); // root_size, update later yading@11: write_pad(pb, 4); yading@11: avio_wl32(pb, 0); // root_sector, update it later. yading@11: yading@11: write_pad(pb, 32); yading@11: avio_wl32(pb, 0); // file ends pointer, update it later. yading@11: yading@11: pad = (1 << WTV_SECTOR_BITS) - avio_tell(pb); yading@11: write_pad(pb, pad); yading@11: yading@11: wctx->timeline_start_pos = avio_tell(pb); yading@11: yading@11: wctx->serial = 1; yading@11: wctx->last_chunk_pos = -1; yading@11: wctx->first_video_flag = 1; yading@11: yading@11: for (i = 0; i < s->nb_streams; i++) { yading@11: st = s->streams[i]; yading@11: if (st->codec->codec_id == AV_CODEC_ID_MJPEG) yading@11: continue; yading@11: ret = write_stream_codec(s, st); yading@11: if (ret < 0) { yading@11: av_log(s, AV_LOG_ERROR, "write stream codec failed codec_type(0x%x)\n", st->codec->codec_type); yading@11: return -1; yading@11: } yading@11: if (!i) yading@11: write_sync(s); yading@11: } yading@11: yading@11: for (i = 0; i < s->nb_streams; i++) { yading@11: st = s->streams[i]; yading@11: if (st->codec->codec_id == AV_CODEC_ID_MJPEG) yading@11: continue; yading@11: ret = write_stream_data(s, st); yading@11: if (ret < 0) { yading@11: av_log(s, AV_LOG_ERROR, "write stream data failed codec_type(0x%x)\n", st->codec->codec_type); yading@11: return -1; yading@11: } yading@11: } yading@11: yading@11: if (wctx->nb_index) yading@11: write_index(s); yading@11: yading@11: return 0; yading@11: } yading@11: yading@11: static void write_timestamp(AVFormatContext *s, AVPacket *pkt) yading@11: { yading@11: AVIOContext *pb = s->pb; yading@11: WtvContext *wctx = s->priv_data; yading@11: AVCodecContext *enc = s->streams[pkt->stream_index]->codec; yading@11: yading@11: write_chunk_header(s, &ff_timestamp_guid, 56, 0x40000000 | (INDEX_BASE + pkt->stream_index)); yading@11: write_pad(pb, 8); yading@11: avio_wl64(pb, pkt->pts == AV_NOPTS_VALUE ? -1 : pkt->pts); yading@11: avio_wl64(pb, pkt->pts == AV_NOPTS_VALUE ? -1 : pkt->pts); yading@11: avio_wl64(pb, pkt->pts == AV_NOPTS_VALUE ? -1 : pkt->pts); yading@11: avio_wl64(pb, 0); yading@11: avio_wl64(pb, enc->codec_type == AVMEDIA_TYPE_VIDEO && (pkt->flags & AV_PKT_FLAG_KEY) ? 1 : 0); yading@11: avio_wl64(pb, 0); yading@11: yading@11: wctx->last_timestamp_pos = wctx->last_chunk_pos; yading@11: } yading@11: yading@11: static int write_packet(AVFormatContext *s, AVPacket *pkt) yading@11: { yading@11: AVIOContext *pb = s->pb; yading@11: WtvContext *wctx = s->priv_data; yading@11: yading@11: if (s->streams[pkt->stream_index]->codec->codec_id == AV_CODEC_ID_MJPEG && !wctx->thumbnail.size) { yading@11: av_copy_packet(&wctx->thumbnail, pkt); yading@11: return 0; yading@11: } yading@11: yading@11: /* emit sync chunk and 'timeline.table.0.entries.Event' record every 50 frames */ yading@11: if (wctx->serial - (wctx->nb_sp_pairs ? wctx->sp_pairs[wctx->nb_sp_pairs - 1].serial : 0) >= 50) yading@11: write_sync(s); yading@11: yading@11: /* emit 'table.0.entries.time' record every 500ms */ yading@11: if (pkt->pts != AV_NOPTS_VALUE && pkt->pts - (wctx->nb_st_pairs ? wctx->st_pairs[wctx->nb_st_pairs - 1].value : 0) >= 5000000) yading@11: add_serial_pair(&wctx->st_pairs, &wctx->nb_st_pairs, wctx->serial, pkt->pts); yading@11: yading@11: if (pkt->pts != AV_NOPTS_VALUE && pkt->pts > wctx->last_pts) { yading@11: wctx->last_pts = pkt->pts; yading@11: wctx->last_serial = wctx->serial; yading@11: } yading@11: yading@11: // write timestamp chunk yading@11: write_timestamp(s, pkt); yading@11: yading@11: write_chunk_header(s, &ff_data_guid, pkt->size, INDEX_BASE + pkt->stream_index); yading@11: avio_write(pb, pkt->data, pkt->size); yading@11: write_pad(pb, WTV_PAD8(pkt->size) - pkt->size); yading@11: yading@11: wctx->serial++; yading@11: return 0; yading@11: } yading@11: yading@11: static int write_table0_header_events(AVIOContext *pb) yading@11: { yading@11: avio_wl32(pb, 0x10); yading@11: write_pad(pb, 84); yading@11: avio_wl64(pb, 0x32); yading@11: return 96; yading@11: } yading@11: yading@11: static int write_table0_header_legacy_attrib(AVIOContext *pb) yading@11: { yading@11: int pad = 0; yading@11: avio_wl32(pb, 0xFFFFFFFF); yading@11: write_pad(pb, 12); yading@11: avio_write(pb, legacy_attrib, sizeof(legacy_attrib)); yading@11: pad = WTV_PAD8(sizeof(legacy_attrib)) - sizeof(legacy_attrib); yading@11: write_pad(pb, pad); yading@11: write_pad(pb, 32); yading@11: return 48 + WTV_PAD8(sizeof(legacy_attrib)); yading@11: } yading@11: yading@11: static int write_table0_header_time(AVIOContext *pb) yading@11: { yading@11: avio_wl32(pb, 0x10); yading@11: write_pad(pb, 76); yading@11: avio_wl64(pb, 0x40); yading@11: return 88; yading@11: } yading@11: yading@11: static const WTVRootEntryTable wtv_root_entry_table[] = { yading@11: { timeline_table_0_header_events, sizeof(timeline_table_0_header_events), write_table0_header_events}, yading@11: { ff_timeline_table_0_entries_Events_le16, sizeof(ff_timeline_table_0_entries_Events_le16), NULL}, yading@11: { ff_timeline_le16, sizeof(ff_timeline_le16), NULL}, yading@11: { table_0_header_legacy_attrib, sizeof(table_0_header_legacy_attrib), write_table0_header_legacy_attrib}, yading@11: { ff_table_0_entries_legacy_attrib_le16, sizeof(ff_table_0_entries_legacy_attrib_le16), NULL}, yading@11: { table_0_redirector_legacy_attrib, sizeof(table_0_redirector_legacy_attrib), NULL}, yading@11: { table_0_header_time, sizeof(table_0_header_time), write_table0_header_time}, yading@11: { ff_table_0_entries_time_le16, sizeof(ff_table_0_entries_time_le16), NULL}, yading@11: }; yading@11: yading@11: static int write_root_table(AVFormatContext *s, int64_t sector_pos) yading@11: { yading@11: AVIOContext *pb = s->pb; yading@11: WtvContext *wctx = s->priv_data; yading@11: int size, pad; yading@11: int i; yading@11: yading@11: const WTVRootEntryTable *h = wtv_root_entry_table; yading@11: for (i = 0; i < sizeof(wtv_root_entry_table)/sizeof(WTVRootEntryTable); i++, h++) { yading@11: WtvFile *w = &wctx->file[i]; yading@11: int filename_padding = WTV_PAD8(h->header_size) - h->header_size; yading@11: WTVHeaderWriteFunc *write = h->write_header; yading@11: int len = 0; yading@11: int64_t len_pos; yading@11: yading@11: ff_put_guid(pb, &ff_dir_entry_guid); yading@11: len_pos = avio_tell(pb); yading@11: avio_wl16(pb, 40 + h->header_size + filename_padding + 8); // maybe updated later yading@11: write_pad(pb, 6); yading@11: avio_wl64(pb, write ? 0 : w->length);// maybe update later yading@11: avio_wl32(pb, (h->header_size + filename_padding) >> 1); yading@11: write_pad(pb, 4); yading@11: yading@11: avio_write(pb, h->header, h->header_size); yading@11: write_pad(pb, filename_padding); yading@11: yading@11: if (write) { yading@11: len = write(pb); yading@11: // update length field yading@11: avio_seek(pb, len_pos, SEEK_SET); yading@11: avio_wl64(pb, 40 + h->header_size + filename_padding + len); yading@11: avio_wl64(pb, len |(1ULL<<62) | (1ULL<<60)); yading@11: avio_seek(pb, 8 + h->header_size + filename_padding + len, SEEK_CUR); yading@11: } else { yading@11: avio_wl32(pb, w->first_sector); yading@11: avio_wl32(pb, w->depth); yading@11: } yading@11: } yading@11: yading@11: // caculate root table size yading@11: size = avio_tell(pb) - sector_pos; yading@11: pad = WTV_SECTOR_SIZE- size; yading@11: write_pad(pb, pad); yading@11: yading@11: return size; yading@11: } yading@11: yading@11: static void write_fat(AVIOContext *pb, int start_sector, int nb_sectors, int shift) yading@11: { yading@11: int i; yading@11: for (i = 0; i < nb_sectors; i++) { yading@11: avio_wl32(pb, start_sector + (i << shift)); yading@11: } yading@11: // pad left sector pointer size yading@11: write_pad(pb, WTV_SECTOR_SIZE - ((nb_sectors << 2) % WTV_SECTOR_SIZE)); yading@11: } yading@11: yading@11: static int64_t write_fat_sector(AVFormatContext *s, int64_t start_pos, int nb_sectors, int sector_bits, int depth) yading@11: { yading@11: int64_t start_sector = start_pos >> WTV_SECTOR_BITS; yading@11: int shift = sector_bits - WTV_SECTOR_BITS; yading@11: yading@11: int64_t fat = avio_tell(s->pb); yading@11: write_fat(s->pb, start_sector, nb_sectors, shift); yading@11: yading@11: if (depth == 2) { yading@11: int64_t start_sector1 = fat >> WTV_SECTOR_BITS; yading@11: int nb_sectors1 = ((nb_sectors << 2) + WTV_SECTOR_SIZE - 1) / WTV_SECTOR_SIZE; yading@11: int64_t fat1 = avio_tell(s->pb); yading@11: yading@11: write_fat(s->pb, start_sector1, nb_sectors1, 0); yading@11: return fat1; yading@11: } yading@11: yading@11: return fat; yading@11: } yading@11: yading@11: static void write_table_entries_events(AVFormatContext *s) yading@11: { yading@11: AVIOContext *pb = s->pb; yading@11: WtvContext *wctx = s->priv_data; yading@11: int i; yading@11: for (i = 0; i < wctx->nb_sp_pairs; i++) { yading@11: avio_wl64(pb, wctx->sp_pairs[i].serial); yading@11: avio_wl64(pb, wctx->sp_pairs[i].value); yading@11: } yading@11: } yading@11: yading@11: static void write_table_entries_time(AVFormatContext *s) yading@11: { yading@11: AVIOContext *pb = s->pb; yading@11: WtvContext *wctx = s->priv_data; yading@11: int i; yading@11: for (i = 0; i < wctx->nb_st_pairs; i++) { yading@11: avio_wl64(pb, wctx->st_pairs[i].value); yading@11: avio_wl64(pb, wctx->st_pairs[i].serial); yading@11: } yading@11: avio_wl64(pb, wctx->last_pts); yading@11: avio_wl64(pb, wctx->last_serial); yading@11: } yading@11: yading@11: static void write_metadata_header(AVIOContext *pb, int type, const char *key, int value_size) yading@11: { yading@11: ff_put_guid(pb, &ff_metadata_guid); yading@11: avio_wl32(pb, type); yading@11: avio_wl32(pb, value_size); yading@11: avio_put_str16le(pb, key); yading@11: } yading@11: yading@11: static int metadata_header_size(const char *key) yading@11: { yading@11: return 16 + 4 + 4 + strlen(key)*2 + 2; yading@11: } yading@11: yading@11: static void write_tag_int32(AVIOContext *pb, const char *key, int value) yading@11: { yading@11: write_metadata_header(pb, 0, key, 4); yading@11: avio_wl32(pb, value); yading@11: } yading@11: yading@11: static void write_tag(AVIOContext *pb, const char *key, const char *value) yading@11: { yading@11: write_metadata_header(pb, 1, key, strlen(value)*2 + 2); yading@11: avio_put_str16le(pb, value); yading@11: } yading@11: yading@11: static int attachment_value_size(const AVPacket *pkt, const AVDictionaryEntry *e) yading@11: { yading@11: return strlen("image/jpeg")*2 + 2 + 1 + (e ? strlen(e->value)*2 : 0) + 2 + 4 + pkt->size; yading@11: } yading@11: yading@11: static void write_table_entries_attrib(AVFormatContext *s) yading@11: { yading@11: WtvContext *wctx = s->priv_data; yading@11: AVIOContext *pb = s->pb; yading@11: AVDictionaryEntry *tag = 0; yading@11: yading@11: //FIXME: translate special tags (e.g. WM/Bitrate) to binary representation yading@11: ff_metadata_conv(&s->metadata, ff_asf_metadata_conv, NULL); yading@11: while ((tag = av_dict_get(s->metadata, "", tag, AV_DICT_IGNORE_SUFFIX))) yading@11: write_tag(pb, tag->key, tag->value); yading@11: yading@11: if (wctx->thumbnail.size) { yading@11: AVStream *st = s->streams[wctx->thumbnail.stream_index]; yading@11: tag = av_dict_get(st->metadata, "title", NULL, 0); yading@11: write_metadata_header(pb, 2, "WM/Picture", attachment_value_size(&wctx->thumbnail, tag)); yading@11: yading@11: avio_put_str16le(pb, "image/jpeg"); yading@11: avio_w8(pb, 0x10); yading@11: avio_put_str16le(pb, tag ? tag->value : ""); yading@11: yading@11: avio_wl32(pb, wctx->thumbnail.size); yading@11: avio_write(pb, wctx->thumbnail.data, wctx->thumbnail.size); yading@11: yading@11: write_tag_int32(pb, "WM/MediaThumbType", 2); yading@11: } yading@11: } yading@11: yading@11: static void write_table_redirector_legacy_attrib(AVFormatContext *s) yading@11: { yading@11: WtvContext *wctx = s->priv_data; yading@11: AVIOContext *pb = s->pb; yading@11: AVDictionaryEntry *tag = 0; yading@11: int64_t pos = 0; yading@11: yading@11: //FIXME: translate special tags to binary representation yading@11: while ((tag = av_dict_get(s->metadata, "", tag, AV_DICT_IGNORE_SUFFIX))) { yading@11: avio_wl64(pb, pos); yading@11: pos += metadata_header_size(tag->key) + strlen(tag->value)*2 + 2; yading@11: } yading@11: yading@11: if (wctx->thumbnail.size) { yading@11: AVStream *st = s->streams[wctx->thumbnail.stream_index]; yading@11: avio_wl64(pb, pos); yading@11: pos += metadata_header_size("WM/Picture") + attachment_value_size(&wctx->thumbnail, av_dict_get(st->metadata, "title", NULL, 0)); yading@11: yading@11: avio_wl64(pb, pos); yading@11: pos += metadata_header_size("WM/MediaThumbType") + 4; yading@11: } yading@11: } yading@11: yading@11: /** yading@11: * Pad the remainder of a file yading@11: * Write out fat table yading@11: * @return <0 on error yading@11: */ yading@11: static int finish_file(AVFormatContext *s, enum WtvFileIndex index, int64_t start_pos) yading@11: { yading@11: WtvContext *wctx = s->priv_data; yading@11: AVIOContext *pb = s->pb; yading@11: WtvFile *w = &wctx->file[index]; yading@11: int64_t end_pos = avio_tell(pb); yading@11: int sector_bits, nb_sectors, pad; yading@11: yading@11: av_assert0(index < WTV_FILES); yading@11: yading@11: w->length = (end_pos - start_pos); yading@11: yading@11: // determine optimal fat table depth, sector_bits, nb_sectors yading@11: if (w->length <= WTV_SECTOR_SIZE) { yading@11: w->depth = 0; yading@11: sector_bits = WTV_SECTOR_BITS; yading@11: } else if (w->length <= (WTV_SECTOR_SIZE / 4) * WTV_SECTOR_SIZE) { yading@11: w->depth = 1; yading@11: sector_bits = WTV_SECTOR_BITS; yading@11: } else if (w->length <= (WTV_SECTOR_SIZE / 4) * WTV_BIGSECTOR_SIZE) { yading@11: w->depth = 1; yading@11: sector_bits = WTV_BIGSECTOR_BITS; yading@11: } else if (w->length <= (int64_t)(WTV_SECTOR_SIZE / 4) * (WTV_SECTOR_SIZE / 4) * WTV_SECTOR_SIZE) { yading@11: w->depth = 2; yading@11: sector_bits = WTV_SECTOR_BITS; yading@11: } else if (w->length <= (int64_t)(WTV_SECTOR_SIZE / 4) * (WTV_SECTOR_SIZE / 4) * WTV_BIGSECTOR_SIZE) { yading@11: w->depth = 2; yading@11: sector_bits = WTV_BIGSECTOR_BITS; yading@11: } else { yading@11: av_log(s, AV_LOG_ERROR, "unsupported file allocation table depth (%"PRIi64" bytes)\n", w->length); yading@11: return -1; yading@11: } yading@11: yading@11: // determine the nb_sectors yading@11: nb_sectors = (int)(w->length >> sector_bits); yading@11: yading@11: // pad sector of timeline yading@11: pad = (1 << sector_bits) - (w->length % (1 << sector_bits)); yading@11: if (pad) { yading@11: nb_sectors++; yading@11: write_pad(pb, pad); yading@11: } yading@11: yading@11: //write fat table yading@11: if (w->depth > 0) { yading@11: w->first_sector = write_fat_sector(s, start_pos, nb_sectors, sector_bits, w->depth) >> WTV_SECTOR_BITS; yading@11: } else { yading@11: w->first_sector = start_pos >> WTV_SECTOR_BITS; yading@11: } yading@11: yading@11: w->length |= 1ULL<<60; yading@11: if (sector_bits == WTV_SECTOR_BITS) yading@11: w->length |= 1ULL<<63; yading@11: yading@11: return 0; yading@11: } yading@11: yading@11: static int write_trailer(AVFormatContext *s) yading@11: { yading@11: WtvContext *wctx = s->priv_data; yading@11: AVIOContext *pb = s->pb; yading@11: int root_size; yading@11: int64_t sector_pos; yading@11: int64_t start_pos, file_end_pos; yading@11: yading@11: if (finish_file(s, WTV_TIMELINE, wctx->timeline_start_pos) < 0) yading@11: return -1; yading@11: yading@11: start_pos = avio_tell(pb); yading@11: write_table_entries_events(s); yading@11: if (finish_file(s, WTV_TIMELINE_TABLE_0_ENTRIES_EVENTS, start_pos) < 0) yading@11: return -1; yading@11: yading@11: start_pos = avio_tell(pb); yading@11: write_table_entries_attrib(s); yading@11: if (finish_file(s, WTV_TABLE_0_ENTRIES_LEGACY_ATTRIB, start_pos) < 0) yading@11: return -1; yading@11: yading@11: start_pos = avio_tell(pb); yading@11: write_table_redirector_legacy_attrib(s); yading@11: if (finish_file(s, WTV_TABLE_0_REDIRECTOR_LEGACY_ATTRIB, start_pos) < 0) yading@11: return -1; yading@11: yading@11: start_pos = avio_tell(pb); yading@11: write_table_entries_time(s); yading@11: if (finish_file(s, WTV_TABLE_0_ENTRIES_TIME, start_pos) < 0) yading@11: return -1; yading@11: yading@11: // write root table yading@11: sector_pos = avio_tell(pb); yading@11: root_size = write_root_table(s, sector_pos); yading@11: yading@11: file_end_pos = avio_tell(pb); yading@11: // update root value yading@11: avio_seek(pb, 0x30, SEEK_SET); yading@11: avio_wl32(pb, root_size); yading@11: avio_seek(pb, 4, SEEK_CUR); yading@11: avio_wl32(pb, sector_pos >> WTV_SECTOR_BITS); yading@11: avio_seek(pb, 0x5c, SEEK_SET); yading@11: avio_wl32(pb, file_end_pos >> WTV_SECTOR_BITS); yading@11: yading@11: avio_flush(pb); yading@11: yading@11: av_free(wctx->sp_pairs); yading@11: av_free(wctx->st_pairs); yading@11: av_free_packet(&wctx->thumbnail); yading@11: return 0; yading@11: } yading@11: yading@11: AVOutputFormat ff_wtv_muxer = { yading@11: .name = "wtv", yading@11: .long_name = NULL_IF_CONFIG_SMALL("Windows Television (WTV)"), yading@11: .extensions = "wtv", yading@11: .priv_data_size = sizeof(WtvContext), yading@11: .audio_codec = AV_CODEC_ID_AC3, yading@11: .video_codec = AV_CODEC_ID_MPEG2VIDEO, yading@11: .write_header = write_header, yading@11: .write_packet = write_packet, yading@11: .write_trailer = write_trailer, yading@11: .codec_tag = (const AVCodecTag* const []){ ff_codec_bmp_tags, yading@11: ff_codec_wav_tags, 0 }, yading@11: };