Skip to content

F
ffmpeg
Commit c16ddcc9 authored by Diego Biurrun's avatar Diego Biurrun
Browse files
Options

avisynth: K&R formatting cosmetics

parent 97b052e5
Hide whitespace changes
Inline Side-by-side
Showing with 150 additions and 145 deletions
libavformat/avisynth.c
View file @ c16ddcc9
...@@ -19,201 +19,206 @@ ...@@ -19,201 +19,206 @@
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/ */
#include <windows.h>
#include <vfw.h>
#include "libavutil/internal.h" #include "libavutil/internal.h"
#include "avformat.h" #include "avformat.h"
#include "internal.h" #include "internal.h"
#include "riff.h" #include "riff.h"
#include <windows.h>
#include <vfw.h>
typedef struct { typedef struct {
PAVISTREAM handle; PAVISTREAM handle;
AVISTREAMINFO info; AVISTREAMINFO info;
DWORD read; DWORD read;
LONG chunck_size; LONG chunck_size;
LONG chunck_samples; LONG chunck_samples;
} AviSynthStream; } AviSynthStream;
typedef struct { typedef struct {
PAVIFILE file; PAVIFILE file;
AviSynthStream *streams; AviSynthStream *streams;
int nb_streams; int nb_streams;
int next_stream; int next_stream;
} AviSynthContext; } AviSynthContext;
static int avisynth_read_header(AVFormatContext *s) static int avisynth_read_header(AVFormatContext *s)
{ {
AviSynthContext *avs = s->priv_data; AviSynthContext *avs = s->priv_data;
HRESULT res; HRESULT res;
AVIFILEINFO info; AVIFILEINFO info;
DWORD id; DWORD id;
AVStream *st; AVStream *st;
AviSynthStream *stream; AviSynthStream *stream;
wchar_t filename_wchar[1024] = { 0 }; wchar_t filename_wchar[1024] = { 0 };
char filename_char[1024] = { 0 }; char filename_char[1024] = { 0 };
AVIFileInit(); AVIFileInit();
/* AviSynth cannot accept UTF-8 file names. */ /* AviSynth cannot accept UTF-8 file names. */
MultiByteToWideChar(CP_UTF8, 0, s->filename, -1, filename_wchar, 1024); MultiByteToWideChar(CP_UTF8, 0, s->filename, -1, filename_wchar, 1024);
WideCharToMultiByte(CP_THREAD_ACP, 0, filename_wchar, -1, filename_char, 1024, NULL, NULL); WideCharToMultiByte(CP_THREAD_ACP, 0, filename_wchar, -1, filename_char,
res = AVIFileOpen(&avs->file, filename_char, OF_READ|OF_SHARE_DENY_WRITE, NULL); 1024, NULL, NULL);
if (res != S_OK) res = AVIFileOpen(&avs->file, filename_char,
{ OF_READ | OF_SHARE_DENY_WRITE, NULL);
av_log(s, AV_LOG_ERROR, "AVIFileOpen failed with error %ld", res); if (res != S_OK) {
AVIFileExit(); av_log(s, AV_LOG_ERROR, "AVIFileOpen failed with error %ld", res);
return -1; AVIFileExit();
return -1;
} }
res = AVIFileInfo(avs->file, &info, sizeof(info)); res = AVIFileInfo(avs->file, &info, sizeof(info));
if (res != S_OK) if (res != S_OK) {
{ av_log(s, AV_LOG_ERROR, "AVIFileInfo failed with error %ld", res);
av_log(s, AV_LOG_ERROR, "AVIFileInfo failed with error %ld", res); AVIFileExit();
AVIFileExit(); return -1;
return -1;
} }
avs->streams = av_mallocz(info.dwStreams * sizeof(AviSynthStream)); avs->streams = av_mallocz(info.dwStreams * sizeof(AviSynthStream));
for (id=0; id<info.dwStreams; id++) for (id = 0; id < info.dwStreams; id++) {
{ stream = &avs->streams[id];
stream = &avs->streams[id]; stream->read = 0;
stream->read = 0; if (AVIFileGetStream(avs->file, &stream->handle, 0, id) == S_OK &&
if (AVIFileGetStream(avs->file, &stream->handle, 0, id) == S_OK) AVIStreamInfo(stream->handle, &stream->info,
{ sizeof(stream->info)) == S_OK) {
if (AVIStreamInfo(stream->handle, &stream->info, sizeof(stream->info)) == S_OK) if (stream->info.fccType == streamtypeAUDIO) {
{ WAVEFORMATEX wvfmt;
if (stream->info.fccType == streamtypeAUDIO) LONG struct_size = sizeof(WAVEFORMATEX);
{ if (AVIStreamReadFormat(stream->handle, 0,
WAVEFORMATEX wvfmt; &wvfmt, &struct_size) != S_OK)
LONG struct_size = sizeof(WAVEFORMATEX);
if (AVIStreamReadFormat(stream->handle, 0, &wvfmt, &struct_size) != S_OK)
continue; continue;
st = avformat_new_stream(s, NULL); st = avformat_new_stream(s, NULL);
st->id = id; st->id = id;
st->codec->codec_type = AVMEDIA_TYPE_AUDIO; st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
st->codec->block_align = wvfmt.nBlockAlign; st->codec->block_align = wvfmt.nBlockAlign;
st->codec->channels = wvfmt.nChannels; st->codec->channels = wvfmt.nChannels;
st->codec->sample_rate = wvfmt.nSamplesPerSec; st->codec->sample_rate = wvfmt.nSamplesPerSec;
st->codec->bit_rate = wvfmt.nAvgBytesPerSec * 8; st->codec->bit_rate = wvfmt.nAvgBytesPerSec * 8;
st->codec->bits_per_coded_sample = wvfmt.wBitsPerSample; st->codec->bits_per_coded_sample = wvfmt.wBitsPerSample;
stream->chunck_samples = wvfmt.nSamplesPerSec * (uint64_t)info.dwScale / (uint64_t)info.dwRate; stream->chunck_samples = wvfmt.nSamplesPerSec *
stream->chunck_size = stream->chunck_samples * wvfmt.nChannels * wvfmt.wBitsPerSample / 8; (uint64_t)info.dwScale /
(uint64_t)info.dwRate;
st->codec->codec_tag = wvfmt.wFormatTag; stream->chunck_size = stream->chunck_samples *
st->codec->codec_id = ff_wav_codec_get_id(wvfmt.wFormatTag, st->codec->bits_per_coded_sample); wvfmt.nChannels *
} wvfmt.wBitsPerSample / 8;
else if (stream->info.fccType == streamtypeVIDEO)
{ st->codec->codec_tag = wvfmt.wFormatTag;
BITMAPINFO imgfmt; st->codec->codec_id =
LONG struct_size = sizeof(BITMAPINFO); ff_wav_codec_get_id(wvfmt.wFormatTag,
st->codec->bits_per_coded_sample);
stream->chunck_size = stream->info.dwSampleSize; } else if (stream->info.fccType == streamtypeVIDEO) {
stream->chunck_samples = 1; BITMAPINFO imgfmt;
LONG struct_size = sizeof(BITMAPINFO);
if (AVIStreamReadFormat(stream->handle, 0, &imgfmt, &struct_size) != S_OK)
stream->chunck_size = stream->info.dwSampleSize;
stream->chunck_samples = 1;
if (AVIStreamReadFormat(stream->handle, 0, &imgfmt,
&struct_size) != S_OK)
continue; continue;
st = avformat_new_stream(s, NULL); st = avformat_new_stream(s, NULL);
st->id = id; st->id = id;
st->codec->codec_type = AVMEDIA_TYPE_VIDEO; st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
st->avg_frame_rate.num = stream->info.dwRate; st->avg_frame_rate.num = stream->info.dwRate;
st->avg_frame_rate.den = stream->info.dwScale; st->avg_frame_rate.den = stream->info.dwScale;
st->codec->width = imgfmt.bmiHeader.biWidth; st->codec->width = imgfmt.bmiHeader.biWidth;
st->codec->height = imgfmt.bmiHeader.biHeight; st->codec->height = imgfmt.bmiHeader.biHeight;
st->codec->bits_per_coded_sample = imgfmt.bmiHeader.biBitCount; st->codec->bits_per_coded_sample = imgfmt.bmiHeader.biBitCount;
st->codec->bit_rate = (uint64_t)stream->info.dwSampleSize * (uint64_t)stream->info.dwRate * 8 / (uint64_t)stream->info.dwScale; st->codec->bit_rate = (uint64_t)stream->info.dwSampleSize *
st->codec->codec_tag = imgfmt.bmiHeader.biCompression; (uint64_t)stream->info.dwRate * 8 /
st->codec->codec_id = ff_codec_get_id(ff_codec_bmp_tags, imgfmt.bmiHeader.biCompression); (uint64_t)stream->info.dwScale;
st->codec->codec_tag = imgfmt.bmiHeader.biCompression;
st->duration = stream->info.dwLength; st->codec->codec_id =
} ff_codec_get_id(ff_codec_bmp_tags,
else imgfmt.bmiHeader.biCompression);
{
AVIStreamRelease(stream->handle); st->duration = stream->info.dwLength;
continue; } else {
} AVIStreamRelease(stream->handle);
continue;
}
avs->nb_streams++; avs->nb_streams++;
st->codec->stream_codec_tag = stream->info.fccHandler; st->codec->stream_codec_tag = stream->info.fccHandler;
avpriv_set_pts_info(st, 64, info.dwScale, info.dwRate); avpriv_set_pts_info(st, 64, info.dwScale, info.dwRate);
st->start_time = stream->info.dwStart; st->start_time = stream->info.dwStart;
}
} }
} }
return 0; return 0;
} }
static int avisynth_read_packet(AVFormatContext *s, AVPacket *pkt) static int avisynth_read_packet(AVFormatContext *s, AVPacket *pkt)
{ {
AviSynthContext *avs = s->priv_data; AviSynthContext *avs = s->priv_data;
HRESULT res; HRESULT res;
AviSynthStream *stream; AviSynthStream *stream;
int stream_id = avs->next_stream; int stream_id = avs->next_stream;
LONG read_size; LONG read_size;
// handle interleaving manually... // handle interleaving manually...
stream = &avs->streams[stream_id]; stream = &avs->streams[stream_id];
if (stream->read >= stream->info.dwLength) if (stream->read >= stream->info.dwLength)
return AVERROR(EIO); return AVERROR(EIO);
if (av_new_packet(pkt, stream->chunck_size)) if (av_new_packet(pkt, stream->chunck_size))
return AVERROR(EIO); return AVERROR(EIO);
pkt->stream_index = stream_id; pkt->stream_index = stream_id;
pkt->pts = avs->streams[stream_id].read / avs->streams[stream_id].chunck_samples; pkt->pts = avs->streams[stream_id].read /
avs->streams[stream_id].chunck_samples;
res = AVIStreamRead(stream->handle, stream->read, stream->chunck_samples, pkt->data, stream->chunck_size, &read_size, NULL); res = AVIStreamRead(stream->handle, stream->read, stream->chunck_samples,
pkt->data, stream->chunck_size, &read_size, NULL);
pkt->pts = stream->read; pkt->pts = stream->read;
pkt->size = read_size; pkt->size = read_size;
stream->read += stream->chunck_samples; stream->read += stream->chunck_samples;
// prepare for the next stream to read // prepare for the next stream to read
do { do
avs->next_stream = (avs->next_stream+1) % avs->nb_streams; avs->next_stream = (avs->next_stream + 1) % avs->nb_streams;
} while (avs->next_stream != stream_id && s->streams[avs->next_stream]->discard >= AVDISCARD_ALL); while (avs->next_stream != stream_id &&
s->streams[avs->next_stream]->discard >= AVDISCARD_ALL);
return (res == S_OK) ? pkt->size : -1; return (res == S_OK) ? pkt->size : -1;
} }
static int avisynth_read_close(AVFormatContext *s) static int avisynth_read_close(AVFormatContext *s)
{ {
AviSynthContext *avs = s->priv_data; AviSynthContext *avs = s->priv_data;
int i; int i;
for (i=0;i<avs->nb_streams;i++) for (i = 0; i < avs->nb_streams; i++)
{ AVIStreamRelease(avs->streams[i].handle);
AVIStreamRelease(avs->streams[i].handle);
}
av_free(avs->streams); av_free(avs->streams);
AVIFileRelease(avs->file); AVIFileRelease(avs->file);
AVIFileExit(); AVIFileExit();
return 0; return 0;
} }
static int avisynth_read_seek(AVFormatContext *s, int stream_index, int64_t pts, int flags) static int avisynth_read_seek(AVFormatContext *s, int stream_index,
int64_t pts, int flags)
{ {
AviSynthContext *avs = s->priv_data; AviSynthContext *avs = s->priv_data;
int stream_id; int stream_id;
for (stream_id = 0; stream_id < avs->nb_streams; stream_id++) for (stream_id = 0; stream_id < avs->nb_streams; stream_id++)
{ avs->streams[stream_id].read =
avs->streams[stream_id].read = pts * avs->streams[stream_id].chunck_samples; pts * avs->streams[stream_id].chunck_samples;
}
return 0; return 0;
} }
AVInputFormat ff_avisynth_demuxer = { AVInputFormat ff_avisynth_demuxer = {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment