1/*
2 * VC-1 test bitstreams format muxer.
3 * Copyright (c) 2008 Konstantin Shishkov
4 *
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#include "avformat.h"
22#include "internal.h"
23
24typedef struct RCVContext {
25    int frames;
26} RCVContext;
27
28static int vc1test_write_header(AVFormatContext *s)
29{
30    AVCodecParameters *par = s->streams[0]->codecpar;
31    AVIOContext *pb = s->pb;
32
33    if (par->codec_id != AV_CODEC_ID_WMV3) {
34        av_log(s, AV_LOG_ERROR, "Only WMV3 is accepted!\n");
35        return -1;
36    }
37    avio_wl24(pb, 0); //frames count will be here
38    avio_w8(pb, 0xC5);
39    avio_wl32(pb, 4);
40    avio_write(pb, par->extradata, 4);
41    avio_wl32(pb, par->height);
42    avio_wl32(pb, par->width);
43    avio_wl32(pb, 0xC);
44    avio_wl24(pb, 0); // hrd_buffer
45    avio_w8(pb, 0x80); // level|cbr|res1
46    avio_wl32(pb, 0); // hrd_rate
47    if (s->streams[0]->avg_frame_rate.den && s->streams[0]->avg_frame_rate.num == 1)
48        avio_wl32(pb, s->streams[0]->avg_frame_rate.den);
49    else
50        avio_wl32(pb, 0xFFFFFFFF); //variable framerate
51    avpriv_set_pts_info(s->streams[0], 32, 1, 1000);
52
53    return 0;
54}
55
56static int vc1test_write_packet(AVFormatContext *s, AVPacket *pkt)
57{
58    RCVContext *ctx = s->priv_data;
59    AVIOContext *pb = s->pb;
60
61    if (!pkt->size)
62        return 0;
63    avio_wl32(pb, pkt->size | ((pkt->flags & AV_PKT_FLAG_KEY) ? 0x80000000 : 0));
64    avio_wl32(pb, pkt->pts);
65    avio_write(pb, pkt->data, pkt->size);
66    ctx->frames++;
67
68    return 0;
69}
70
71static int vc1test_write_trailer(AVFormatContext *s)
72{
73    RCVContext *ctx = s->priv_data;
74    AVIOContext *pb = s->pb;
75
76    if (s->pb->seekable & AVIO_SEEKABLE_NORMAL) {
77        avio_seek(pb, 0, SEEK_SET);
78        avio_wl24(pb, ctx->frames);
79    }
80    return 0;
81}
82
83const AVOutputFormat ff_vc1t_muxer = {
84    .name              = "vc1test",
85    .long_name         = NULL_IF_CONFIG_SMALL("VC-1 test bitstream"),
86    .extensions        = "rcv",
87    .priv_data_size    = sizeof(RCVContext),
88    .audio_codec       = AV_CODEC_ID_NONE,
89    .video_codec       = AV_CODEC_ID_WMV3,
90    .write_header      = vc1test_write_header,
91    .write_packet      = vc1test_write_packet,
92    .write_trailer     = vc1test_write_trailer,
93};
94