snes9x/win32/AVIOutput.cpp

523 lines
14 KiB
C++
Raw Normal View History

2010-09-25 17:46:12 +02:00
/***********************************************************************************
Snes9x - Portable Super Nintendo Entertainment System (TM) emulator.
(c) Copyright 1996 - 2002 Gary Henderson (gary.henderson@ntlworld.com),
Jerremy Koot (jkoot@snes9x.com)
(c) Copyright 2002 - 2004 Matthew Kendora
(c) Copyright 2002 - 2005 Peter Bortas (peter@bortas.org)
(c) Copyright 2004 - 2005 Joel Yliluoma (http://iki.fi/bisqwit/)
(c) Copyright 2001 - 2006 John Weidman (jweidman@slip.net)
(c) Copyright 2002 - 2006 funkyass (funkyass@spam.shaw.ca),
Kris Bleakley (codeviolation@hotmail.com)
(c) Copyright 2002 - 2010 Brad Jorsch (anomie@users.sourceforge.net),
Nach (n-a-c-h@users.sourceforge.net),
zones (kasumitokoduck@yahoo.com)
(c) Copyright 2006 - 2007 nitsuja
(c) Copyright 2009 - 2010 BearOso,
OV2
BS-X C emulator code
(c) Copyright 2005 - 2006 Dreamer Nom,
zones
C4 x86 assembler and some C emulation code
(c) Copyright 2000 - 2003 _Demo_ (_demo_@zsnes.com),
Nach,
zsKnight (zsknight@zsnes.com)
C4 C++ code
(c) Copyright 2003 - 2006 Brad Jorsch,
Nach
DSP-1 emulator code
(c) Copyright 1998 - 2006 _Demo_,
Andreas Naive (andreasnaive@gmail.com),
Gary Henderson,
Ivar (ivar@snes9x.com),
John Weidman,
Kris Bleakley,
Matthew Kendora,
Nach,
neviksti (neviksti@hotmail.com)
DSP-2 emulator code
(c) Copyright 2003 John Weidman,
Kris Bleakley,
Lord Nightmare (lord_nightmare@users.sourceforge.net),
Matthew Kendora,
neviksti
DSP-3 emulator code
(c) Copyright 2003 - 2006 John Weidman,
Kris Bleakley,
Lancer,
z80 gaiden
DSP-4 emulator code
(c) Copyright 2004 - 2006 Dreamer Nom,
John Weidman,
Kris Bleakley,
Nach,
z80 gaiden
OBC1 emulator code
(c) Copyright 2001 - 2004 zsKnight,
pagefault (pagefault@zsnes.com),
Kris Bleakley
Ported from x86 assembler to C by sanmaiwashi
SPC7110 and RTC C++ emulator code used in 1.39-1.51
(c) Copyright 2002 Matthew Kendora with research by
zsKnight,
John Weidman,
Dark Force
SPC7110 and RTC C++ emulator code used in 1.52+
(c) Copyright 2009 byuu,
neviksti
S-DD1 C emulator code
(c) Copyright 2003 Brad Jorsch with research by
Andreas Naive,
John Weidman
S-RTC C emulator code
(c) Copyright 2001 - 2006 byuu,
John Weidman
ST010 C++ emulator code
(c) Copyright 2003 Feather,
John Weidman,
Kris Bleakley,
Matthew Kendora
Super FX x86 assembler emulator code
(c) Copyright 1998 - 2003 _Demo_,
pagefault,
zsKnight
Super FX C emulator code
(c) Copyright 1997 - 1999 Ivar,
Gary Henderson,
John Weidman
Sound emulator code used in 1.5-1.51
(c) Copyright 1998 - 2003 Brad Martin
(c) Copyright 1998 - 2006 Charles Bilyue'
Sound emulator code used in 1.52+
(c) Copyright 2004 - 2007 Shay Green (gblargg@gmail.com)
SH assembler code partly based on x86 assembler code
(c) Copyright 2002 - 2004 Marcus Comstedt (marcus@mc.pp.se)
2xSaI filter
(c) Copyright 1999 - 2001 Derek Liauw Kie Fa
HQ2x, HQ3x, HQ4x filters
(c) Copyright 2003 Maxim Stepin (maxim@hiend3d.com)
NTSC filter
(c) Copyright 2006 - 2007 Shay Green
GTK+ GUI code
(c) Copyright 2004 - 2010 BearOso
Win32 GUI code
(c) Copyright 2003 - 2006 blip,
funkyass,
Matthew Kendora,
Nach,
nitsuja
(c) Copyright 2009 - 2010 OV2
Mac OS GUI code
(c) Copyright 1998 - 2001 John Stiles
(c) Copyright 2001 - 2010 zones
Specific ports contains the works of other authors. See headers in
individual files.
Snes9x homepage: http://www.snes9x.com/
Permission to use, copy, modify and/or distribute Snes9x in both binary
and source form, for non-commercial purposes, is hereby granted without
fee, providing that this license information and copyright notice appear
with all copies and any derived work.
This software is provided 'as-is', without any express or implied
warranty. In no event shall the authors be held liable for any damages
arising from the use of this software or it's derivatives.
Snes9x is freeware for PERSONAL USE only. Commercial users should
seek permission of the copyright holders first. Commercial use includes,
but is not limited to, charging money for Snes9x or software derived from
Snes9x, including Snes9x or derivatives in commercial game bundles, and/or
using Snes9x as a promotion for your commercial product.
The copyright holders request that bug fixes and improvements to the code
should be forwarded to them so everyone can benefit from the modifications
in future versions.
Super NES and Super Nintendo Entertainment System are trademarks of
Nintendo Co., Limited and its subsidiary companies.
***********************************************************************************/
/*
Win32 AVI Output module
(c) 2004 blip
*/
#include <windows.h>
#include <vfw.h>
#include "wsnes9x.h" // for GUI.hWnd
#include "AVIOutput.h"
#define VIDEO_STREAM 0
#define AUDIO_STREAM 1
struct AVIFile
{
bool valid;
int fps;
int frameskip;
bool video_added;
BITMAPINFOHEADER bitmap_format;
bool sound_added;
WAVEFORMATEX wave_format;
AVISTREAMINFO avi_video_header;
AVISTREAMINFO avi_sound_header;
PAVIFILE avi_file;
PAVISTREAM streams[2];
PAVISTREAM compressed_streams[2];
AVICOMPRESSOPTIONS compress_options[2];
AVICOMPRESSOPTIONS* compress_options_ptr[2];
int video_frames;
int sound_samples;
long tBytes, ByteBuffer;
};
static char saved_cur_avi_fnameandext[MAX_PATH];
static char saved_avi_fname[MAX_PATH];
static char saved_avi_ext[MAX_PATH];
static int avi_segnum=0;
static struct AVIFile saved_avi_info;
static int use_prev_options=0;
static bool truncate_existing(const char* filename)
{
// this is only here because AVIFileOpen doesn't seem to do it for us
FILE* fd = fopen(filename, "wb");
if(fd)
{
fclose(fd);
return true;
}
return false;
}
static void clean_up(AVIFile* _avi)
{
AVIFile& avi = *_avi;
if(avi.sound_added)
{
if(avi.compressed_streams[AUDIO_STREAM])
{
AVIStreamClose(avi.compressed_streams[AUDIO_STREAM]);
avi.compressed_streams[AUDIO_STREAM] = NULL;
avi.streams[AUDIO_STREAM] = NULL; // compressed_streams[AUDIO_STREAM] is just a copy of streams[AUDIO_STREAM]
}
}
if(avi.video_added)
{
if(avi.compressed_streams[VIDEO_STREAM])
{
AVIStreamClose(avi.compressed_streams[VIDEO_STREAM]);
avi.compressed_streams[VIDEO_STREAM] = NULL;
}
if(avi.streams[VIDEO_STREAM])
{
AVIStreamClose(avi.streams[VIDEO_STREAM]);
avi.streams[VIDEO_STREAM] = NULL;
}
}
if(avi.avi_file)
{
AVIFileClose(avi.avi_file);
avi.avi_file = NULL;
}
}
#ifdef __cplusplus
extern "C" {
#endif
void AVICreate(struct AVIFile** avi_out)
{
*avi_out = new AVIFile;
memset(*avi_out, 0, sizeof(AVIFile));
AVIFileInit();
}
void AVIClose(struct AVIFile** avi_out)
{
if(*avi_out)
{
clean_up(*avi_out);
delete *avi_out;
}
*avi_out = NULL;
AVIFileExit();
}
void AVISetFramerate(const int fps, const int frameskip, struct AVIFile* avi_out)
{
avi_out->fps = fps;
avi_out->frameskip = frameskip;
}
void AVISetVideoFormat(const BITMAPINFOHEADER* bitmap_format, struct AVIFile* avi_out)
{
memcpy(&(avi_out->bitmap_format), bitmap_format, sizeof(BITMAPINFOHEADER));
avi_out->video_added = true;
}
void AVISetSoundFormat(const WAVEFORMATEX* wave_format, struct AVIFile* avi_out)
{
memcpy(&(avi_out->wave_format), wave_format, sizeof(WAVEFORMATEX));
avi_out->sound_added = true;
}
int AVIBegin(const char* filename, struct AVIFile* _avi_out)
{
AVIFile& avi = *_avi_out;
int result = 0;
do
{
if(!avi.video_added)
break;
if(!truncate_existing(filename))
break;
// open the file
if(FAILED(AVIFileOpen(&avi.avi_file, filename, OF_CREATE | OF_WRITE, NULL)))
break;
// create the video stream
memset(&avi.avi_video_header, 0, sizeof(AVISTREAMINFO));
avi.avi_video_header.fccType = streamtypeVIDEO;
avi.avi_video_header.dwScale = avi.frameskip;
avi.avi_video_header.dwRate = avi.fps;
avi.avi_video_header.dwSuggestedBufferSize = avi.bitmap_format.biSizeImage;
if(FAILED(AVIFileCreateStream(avi.avi_file, &avi.streams[VIDEO_STREAM], &avi.avi_video_header)))
break;
if(use_prev_options)
{
avi.compress_options[VIDEO_STREAM] = saved_avi_info.compress_options[VIDEO_STREAM];
avi.compress_options_ptr[VIDEO_STREAM] = &avi.compress_options[0];
}
else
{
// get compression options
memset(&avi.compress_options[VIDEO_STREAM], 0, sizeof(AVICOMPRESSOPTIONS));
avi.compress_options_ptr[VIDEO_STREAM] = &avi.compress_options[0];
if(!AVISaveOptions(GUI.hWnd, 0, 1, &avi.streams[VIDEO_STREAM], &avi.compress_options_ptr[VIDEO_STREAM]))
break;
}
// create compressed stream
if(FAILED(AVIMakeCompressedStream(&avi.compressed_streams[VIDEO_STREAM], avi.streams[VIDEO_STREAM], &avi.compress_options[VIDEO_STREAM], NULL)))
break;
// set the stream format
if(FAILED(AVIStreamSetFormat(avi.compressed_streams[VIDEO_STREAM], 0, (void*)&avi.bitmap_format, avi.bitmap_format.biSize)))
break;
// add sound (if requested)
if(avi.sound_added)
{
// create the audio stream
memset(&avi.avi_sound_header, 0, sizeof(AVISTREAMINFO));
avi.avi_sound_header.fccType = streamtypeAUDIO;
avi.avi_sound_header.dwQuality = (DWORD)-1;
avi.avi_sound_header.dwScale = avi.wave_format.nBlockAlign;
avi.avi_sound_header.dwRate = avi.wave_format.nAvgBytesPerSec;
avi.avi_sound_header.dwSampleSize = avi.wave_format.nBlockAlign;
avi.avi_sound_header.dwInitialFrames = 1;
if(FAILED(AVIFileCreateStream(avi.avi_file, &avi.streams[AUDIO_STREAM], &avi.avi_sound_header)))
break;
// AVISaveOptions doesn't seem to work for audio streams
// so here we just copy the pointer for the compressed stream
avi.compressed_streams[AUDIO_STREAM] = avi.streams[AUDIO_STREAM];
// set the stream format
if(FAILED(AVIStreamSetFormat(avi.compressed_streams[AUDIO_STREAM], 0, (void*)&avi.wave_format, sizeof(WAVEFORMATEX))))
break;
}
// initialize counters
avi.video_frames = 0;
avi.sound_samples = 0;
avi.tBytes = 0;
avi.ByteBuffer = 0;
strncpy(saved_cur_avi_fnameandext,filename,MAX_PATH);
strncpy(saved_avi_fname,filename,MAX_PATH);
char* dot = strrchr(saved_avi_fname, '.');
if(dot && dot > strrchr(saved_avi_fname, '/') && dot > strrchr(saved_avi_fname, '\\'))
{
strcpy(saved_avi_ext,dot);
dot[0]='\0';
}
// success
result = 1;
avi.valid = true;
} while(false);
if(!result)
{
clean_up(&avi);
avi.valid = false;
}
return result;
}
int AVIGetVideoFormat(const struct AVIFile* avi_out, const BITMAPINFOHEADER** ppFormat)
{
if(!avi_out->valid ||
!avi_out->video_added)
{
if(ppFormat)
{
*ppFormat = NULL;
}
return 0;
}
if(ppFormat)
{
*ppFormat = &avi_out->bitmap_format;
}
return 1;
}
int AVIGetSoundFormat(const struct AVIFile* avi_out, const WAVEFORMATEX** ppFormat)
{
if(!avi_out->valid ||
!avi_out->sound_added)
{
if(ppFormat)
{
*ppFormat = NULL;
}
return 0;
}
if(ppFormat)
{
*ppFormat = &avi_out->wave_format;
}
return 1;
}
static int AVINextSegment(struct AVIFile* avi_out)
{
char avi_fname[MAX_PATH];
strcpy(avi_fname,saved_avi_fname);
char avi_fname_temp[MAX_PATH];
sprintf(avi_fname_temp, "%s_part%d%s", avi_fname, avi_segnum+2, saved_avi_ext);
saved_avi_info=*avi_out;
use_prev_options=1;
avi_segnum++;
clean_up(avi_out);
int ret = AVIBegin(avi_fname_temp, avi_out);
use_prev_options=0;
strcpy(saved_avi_fname,avi_fname);
return ret;
}
void AVIAddVideoFrame(void* bitmap_data, struct AVIFile* avi_out)
{
if(!avi_out->valid ||
!avi_out->video_added)
{
return;
}
if(FAILED(AVIStreamWrite(avi_out->compressed_streams[VIDEO_STREAM],
avi_out->video_frames,
1,
bitmap_data,
avi_out->bitmap_format.biSizeImage,
AVIIF_KEYFRAME, NULL, &avi_out->ByteBuffer)))
{
avi_out->valid = false;
return;
}
avi_out->video_frames++;
avi_out->tBytes += avi_out->ByteBuffer;
// segment / split AVI when it's almost 2 GB (2000MB, to be precise)
if(!(avi_out->video_frames % 60) && avi_out->tBytes > 2097152000)
AVINextSegment(avi_out);
}
void AVIAddSoundSamples(void* sound_data, const int num_samples, struct AVIFile* avi_out)
{
if(!avi_out->valid ||
!avi_out->sound_added)
{
return;
}
int data_length = num_samples * avi_out->wave_format.nBlockAlign;
if(FAILED(AVIStreamWrite(avi_out->compressed_streams[AUDIO_STREAM],
avi_out->sound_samples,
num_samples,
sound_data,
data_length,
0, NULL, &avi_out->ByteBuffer)))
{
avi_out->valid = false;
return;
}
avi_out->sound_samples += num_samples;
avi_out->tBytes += avi_out->ByteBuffer;
}
#ifdef __cplusplus
}
#endif