avisynth: K&R formatting cosmetics
[ffmpeg.git] / libavformat / avisynth.c
1 /*
2  * AviSynth support
3  * Copyright (c) 2006 DivX, Inc.
4  *
5  * This file is part of Libav.
6  *
7  * Libav 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  * Libav 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 Libav; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 #include <windows.h>
23 #include <vfw.h>
24
25 #include "libavutil/internal.h"
26 #include "avformat.h"
27 #include "internal.h"
28 #include "riff.h"
29
30 typedef struct {
31     PAVISTREAM handle;
32     AVISTREAMINFO info;
33     DWORD read;
34     LONG chunck_size;
35     LONG chunck_samples;
36 } AviSynthStream;
37
38 typedef struct {
39     PAVIFILE file;
40     AviSynthStream *streams;
41     int nb_streams;
42     int next_stream;
43 } AviSynthContext;
44
45 static int avisynth_read_header(AVFormatContext *s)
46 {
47     AviSynthContext *avs = s->priv_data;
48     HRESULT res;
49     AVIFILEINFO info;
50     DWORD id;
51     AVStream *st;
52     AviSynthStream *stream;
53     wchar_t filename_wchar[1024] = { 0 };
54     char filename_char[1024]     = { 0 };
55
56     AVIFileInit();
57
58     /* AviSynth cannot accept UTF-8 file names. */
59     MultiByteToWideChar(CP_UTF8, 0, s->filename, -1, filename_wchar, 1024);
60     WideCharToMultiByte(CP_THREAD_ACP, 0, filename_wchar, -1, filename_char,
61                         1024, NULL, NULL);
62     res = AVIFileOpen(&avs->file, filename_char,
63                       OF_READ | OF_SHARE_DENY_WRITE, NULL);
64     if (res != S_OK) {
65         av_log(s, AV_LOG_ERROR, "AVIFileOpen failed with error %ld", res);
66         AVIFileExit();
67         return -1;
68     }
69
70     res = AVIFileInfo(avs->file, &info, sizeof(info));
71     if (res != S_OK) {
72         av_log(s, AV_LOG_ERROR, "AVIFileInfo failed with error %ld", res);
73         AVIFileExit();
74         return -1;
75     }
76
77     avs->streams = av_mallocz(info.dwStreams * sizeof(AviSynthStream));
78
79     for (id = 0; id < info.dwStreams; id++) {
80         stream       = &avs->streams[id];
81         stream->read = 0;
82         if (AVIFileGetStream(avs->file, &stream->handle, 0, id) == S_OK &&
83             AVIStreamInfo(stream->handle, &stream->info,
84                           sizeof(stream->info)) == S_OK) {
85             if (stream->info.fccType == streamtypeAUDIO) {
86                 WAVEFORMATEX wvfmt;
87                 LONG struct_size = sizeof(WAVEFORMATEX);
88                 if (AVIStreamReadFormat(stream->handle, 0,
89                                         &wvfmt, &struct_size) != S_OK)
90                     continue;
91
92                 st                    = avformat_new_stream(s, NULL);
93                 st->id                = id;
94                 st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
95
96                 st->codec->block_align           = wvfmt.nBlockAlign;
97                 st->codec->channels              = wvfmt.nChannels;
98                 st->codec->sample_rate           = wvfmt.nSamplesPerSec;
99                 st->codec->bit_rate              = wvfmt.nAvgBytesPerSec * 8;
100                 st->codec->bits_per_coded_sample = wvfmt.wBitsPerSample;
101
102                 stream->chunck_samples = wvfmt.nSamplesPerSec *
103                                          (uint64_t)info.dwScale /
104                                          (uint64_t)info.dwRate;
105                 stream->chunck_size    = stream->chunck_samples *
106                                          wvfmt.nChannels *
107                                          wvfmt.wBitsPerSample / 8;
108
109                 st->codec->codec_tag = wvfmt.wFormatTag;
110                 st->codec->codec_id  =
111                     ff_wav_codec_get_id(wvfmt.wFormatTag,
112                                         st->codec->bits_per_coded_sample);
113             } else if (stream->info.fccType == streamtypeVIDEO) {
114                 BITMAPINFO imgfmt;
115                 LONG struct_size = sizeof(BITMAPINFO);
116
117                 stream->chunck_size    = stream->info.dwSampleSize;
118                 stream->chunck_samples = 1;
119
120                 if (AVIStreamReadFormat(stream->handle, 0, &imgfmt,
121                                         &struct_size) != S_OK)
122                     continue;
123
124                 st                     = avformat_new_stream(s, NULL);
125                 st->id                 = id;
126                 st->codec->codec_type  = AVMEDIA_TYPE_VIDEO;
127                 st->avg_frame_rate.num = stream->info.dwRate;
128                 st->avg_frame_rate.den = stream->info.dwScale;
129
130                 st->codec->width  = imgfmt.bmiHeader.biWidth;
131                 st->codec->height = imgfmt.bmiHeader.biHeight;
132
133                 st->codec->bits_per_coded_sample = imgfmt.bmiHeader.biBitCount;
134                 st->codec->bit_rate              = (uint64_t)stream->info.dwSampleSize *
135                                                    (uint64_t)stream->info.dwRate * 8 /
136                                                    (uint64_t)stream->info.dwScale;
137                 st->codec->codec_tag             = imgfmt.bmiHeader.biCompression;
138                 st->codec->codec_id              =
139                     ff_codec_get_id(ff_codec_bmp_tags,
140                                     imgfmt.bmiHeader.biCompression);
141
142                 st->duration = stream->info.dwLength;
143             } else {
144                 AVIStreamRelease(stream->handle);
145                 continue;
146             }
147
148             avs->nb_streams++;
149
150             st->codec->stream_codec_tag = stream->info.fccHandler;
151
152             avpriv_set_pts_info(st, 64, info.dwScale, info.dwRate);
153             st->start_time = stream->info.dwStart;
154         }
155     }
156
157     return 0;
158 }
159
160 static int avisynth_read_packet(AVFormatContext *s, AVPacket *pkt)
161 {
162     AviSynthContext *avs = s->priv_data;
163     HRESULT res;
164     AviSynthStream *stream;
165     int stream_id = avs->next_stream;
166     LONG read_size;
167
168     // handle interleaving manually...
169     stream = &avs->streams[stream_id];
170
171     if (stream->read >= stream->info.dwLength)
172         return AVERROR(EIO);
173
174     if (av_new_packet(pkt, stream->chunck_size))
175         return AVERROR(EIO);
176     pkt->stream_index = stream_id;
177     pkt->pts          = avs->streams[stream_id].read /
178                         avs->streams[stream_id].chunck_samples;
179
180     res = AVIStreamRead(stream->handle, stream->read, stream->chunck_samples,
181                         pkt->data, stream->chunck_size, &read_size, NULL);
182
183     pkt->pts  = stream->read;
184     pkt->size = read_size;
185
186     stream->read += stream->chunck_samples;
187
188     // prepare for the next stream to read
189     do
190         avs->next_stream = (avs->next_stream + 1) % avs->nb_streams;
191     while (avs->next_stream != stream_id &&
192            s->streams[avs->next_stream]->discard >= AVDISCARD_ALL);
193
194     return (res == S_OK) ? pkt->size : -1;
195 }
196
197 static int avisynth_read_close(AVFormatContext *s)
198 {
199     AviSynthContext *avs = s->priv_data;
200     int i;
201
202     for (i = 0; i < avs->nb_streams; i++)
203         AVIStreamRelease(avs->streams[i].handle);
204
205     av_free(avs->streams);
206     AVIFileRelease(avs->file);
207     AVIFileExit();
208     return 0;
209 }
210
211 static int avisynth_read_seek(AVFormatContext *s, int stream_index,
212                               int64_t pts, int flags)
213 {
214     AviSynthContext *avs = s->priv_data;
215     int stream_id;
216
217     for (stream_id = 0; stream_id < avs->nb_streams; stream_id++)
218         avs->streams[stream_id].read =
219             pts * avs->streams[stream_id].chunck_samples;
220
221     return 0;
222 }
223
224 AVInputFormat ff_avisynth_demuxer = {
225     .name           = "avisynth",
226     .long_name      = NULL_IF_CONFIG_SMALL("AviSynth"),
227     .priv_data_size = sizeof(AviSynthContext),
228     .read_header    = avisynth_read_header,
229     .read_packet    = avisynth_read_packet,
230     .read_close     = avisynth_read_close,
231     .read_seek      = avisynth_read_seek,
232     .extensions     = "avs",
233 };