mirror of
https://foundry.openuru.org/gitblit/r/CWE-ou-minkata.git
synced 2025-07-14 10:37:41 -04:00
Fix line endings and tabs
This commit is contained in:
@ -1,294 +1,294 @@
|
||||
/*==LICENSE==*
|
||||
|
||||
CyanWorlds.com Engine - MMOG client, server and tools
|
||||
Copyright (C) 2011 Cyan Worlds, Inc.
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
You can contact Cyan Worlds, Inc. by email legal@cyan.com
|
||||
or by snail mail at:
|
||||
Cyan Worlds, Inc.
|
||||
14617 N Newport Hwy
|
||||
Mead, WA 99021
|
||||
|
||||
*==LICENSE==*/
|
||||
#include "plAVIWriter.h"
|
||||
|
||||
#include "hsTypes.h"
|
||||
|
||||
#include "hsWindows.h"
|
||||
#include <vfw.h>
|
||||
|
||||
#include "hsTimer.h"
|
||||
#include "plMipmap.h"
|
||||
#include "plMessage/plRenderMsg.h"
|
||||
#include "plPipeline.h"
|
||||
#include "pnDispatch/plDispatch.h"
|
||||
#include "pnKeyedObject/plFixedKey.h"
|
||||
|
||||
bool plAVIWriter::fInitialized = false;
|
||||
|
||||
class plAVIWriterImp : public plAVIWriter
|
||||
{
|
||||
protected:
|
||||
PAVIFILE fFileHandle;
|
||||
PAVISTREAM fStreamHandle;
|
||||
PAVISTREAM fCompressedHandle;
|
||||
BITMAPINFOHEADER fBitmapInfo;
|
||||
|
||||
hsBool fOldRealTime;
|
||||
hsScalar fOldFrameTimeInc;
|
||||
|
||||
double fStartTime;
|
||||
|
||||
void IFillStreamInfo(AVISTREAMINFO* inf, plPipeline* pipeline);
|
||||
void IFillBitmapInfo(BITMAPINFOHEADER* inf, plPipeline* pipeline);
|
||||
|
||||
bool ICaptureFrame(plPipeline* pipeline);
|
||||
|
||||
public:
|
||||
plAVIWriterImp();
|
||||
virtual ~plAVIWriterImp();
|
||||
|
||||
virtual hsBool MsgReceive(plMessage* msg);
|
||||
|
||||
virtual void Shutdown();
|
||||
|
||||
virtual bool Open(const char* fileName, plPipeline* pipeline);
|
||||
virtual void Close();
|
||||
};
|
||||
|
||||
plAVIWriter::~plAVIWriter()
|
||||
{
|
||||
}
|
||||
|
||||
plAVIWriter& plAVIWriter::Instance()
|
||||
{
|
||||
static plAVIWriterImp theInstance;
|
||||
|
||||
if (!fInitialized)
|
||||
{
|
||||
theInstance.RegisterAs(kAVIWriter_KEY);
|
||||
fInitialized = true;
|
||||
}
|
||||
|
||||
return theInstance;
|
||||
}
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
plAVIWriterImp::plAVIWriterImp() :
|
||||
fStartTime(0),
|
||||
fOldRealTime(false),
|
||||
fStreamHandle(nil),
|
||||
fCompressedHandle(nil),
|
||||
fFileHandle(nil)
|
||||
{
|
||||
AVIFileInit();
|
||||
}
|
||||
|
||||
plAVIWriterImp::~plAVIWriterImp()
|
||||
{
|
||||
}
|
||||
|
||||
void plAVIWriterImp::Shutdown()
|
||||
{
|
||||
Close();
|
||||
UnRegisterAs(kAVIWriter_KEY);
|
||||
SetKey(nil);
|
||||
}
|
||||
|
||||
|
||||
|
||||
#include "plProfile.h"
|
||||
plProfile_CreateTimer("AviCapture", "RenderSetup", AviCapture);
|
||||
|
||||
hsBool plAVIWriterImp::MsgReceive(plMessage* msg)
|
||||
{
|
||||
|
||||
plRenderMsg* renderMsg = plRenderMsg::ConvertNoRef(msg);
|
||||
if (renderMsg)
|
||||
{
|
||||
plProfile_BeginTiming(AviCapture);
|
||||
|
||||
ICaptureFrame(renderMsg->Pipeline());
|
||||
plProfile_EndTiming(AviCapture);
|
||||
|
||||
}
|
||||
|
||||
return hsKeyedObject::MsgReceive(msg);
|
||||
}
|
||||
|
||||
static const int kFramesPerSec = 30;
|
||||
|
||||
bool plAVIWriterImp::Open(const char* fileName, plPipeline* pipeline)
|
||||
{
|
||||
// Already writing, fail
|
||||
if (fStreamHandle)
|
||||
return false;
|
||||
|
||||
fStartTime = hsTimer::GetSysSeconds();
|
||||
|
||||
// If we're running in real time, set to frame time
|
||||
fOldRealTime = hsTimer::IsRealTime();
|
||||
if (fOldRealTime)
|
||||
{
|
||||
hsTimer::SetRealTime(false);
|
||||
hsTimer::SetFrameTimeInc(1.f / kFramesPerSec);
|
||||
}
|
||||
|
||||
// Open AVI file
|
||||
HRESULT err;
|
||||
err = AVIFileOpen( &fFileHandle, // returned file pointer
|
||||
fileName, // file name
|
||||
OF_WRITE | OF_CREATE, // mode to open file with
|
||||
NULL); // use handler determined
|
||||
hsAssert(err == AVIERR_OK, "Error creating AVI file in plAVIWriter::Open");
|
||||
if (err != AVIERR_OK)
|
||||
{
|
||||
Close();
|
||||
return false;
|
||||
}
|
||||
|
||||
AVISTREAMINFO streamInfo;
|
||||
IFillStreamInfo(&streamInfo, pipeline);
|
||||
|
||||
// Create a video stream in the file
|
||||
err = AVIFileCreateStream( fFileHandle, // file pointer
|
||||
&fStreamHandle, // returned stream pointer
|
||||
&streamInfo ); // stream header
|
||||
hsAssert(err == AVIERR_OK, "Error creating video stream in plAVIWriter::Open");
|
||||
if (err != AVIERR_OK)
|
||||
{
|
||||
Close();
|
||||
return false;
|
||||
}
|
||||
|
||||
do
|
||||
{
|
||||
AVICOMPRESSOPTIONS opts;
|
||||
AVICOMPRESSOPTIONS FAR * aopts[1] = {&opts};
|
||||
memset(&opts, 0, sizeof(opts));
|
||||
|
||||
BOOL bErr = AVISaveOptions(NULL, ICMF_CHOOSE_DATARATE, 1, &fStreamHandle, (LPAVICOMPRESSOPTIONS FAR*)&aopts);
|
||||
hsAssert(bErr, "Error saving stream options in plAVIWriter::Open");
|
||||
if (!bErr)
|
||||
{
|
||||
Close();
|
||||
return false;
|
||||
}
|
||||
|
||||
err = AVIMakeCompressedStream(&fCompressedHandle, fStreamHandle, &opts, NULL);
|
||||
hsAssert(err == AVIERR_OK, "Error creating compressed stream in plAVIWriter::Open");
|
||||
if (err != AVIERR_OK)
|
||||
{
|
||||
Close();
|
||||
return false;
|
||||
}
|
||||
|
||||
IFillBitmapInfo(&fBitmapInfo, pipeline);
|
||||
err = AVIStreamSetFormat( fCompressedHandle, 0,
|
||||
&fBitmapInfo, // stream format
|
||||
fBitmapInfo.biSize);
|
||||
} while (err != AVIERR_OK &&
|
||||
hsMessageBox("Codec unavailable, try again?", "AVI Writer", hsMessageBoxYesNo) == hsMBoxYes);
|
||||
|
||||
if (err != AVIERR_OK)
|
||||
{
|
||||
Close();
|
||||
return false;
|
||||
}
|
||||
|
||||
plgDispatch::Dispatch()->RegisterForExactType(plRenderMsg::Index(), GetKey());
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void plAVIWriterImp::Close()
|
||||
{
|
||||
plgDispatch::Dispatch()->UnRegisterForExactType(plRenderMsg::Index(), GetKey());
|
||||
|
||||
hsTimer::SetRealTime(fOldRealTime);
|
||||
|
||||
if (fStreamHandle)
|
||||
{
|
||||
AVIStreamClose(fStreamHandle);
|
||||
fStreamHandle = nil;
|
||||
}
|
||||
|
||||
if (fCompressedHandle)
|
||||
{
|
||||
AVIStreamClose(fCompressedHandle);
|
||||
fCompressedHandle = nil;
|
||||
}
|
||||
|
||||
if (fFileHandle)
|
||||
{
|
||||
AVIFileClose(fFileHandle);
|
||||
fFileHandle = nil;
|
||||
}
|
||||
|
||||
AVIFileExit();
|
||||
}
|
||||
|
||||
void plAVIWriterImp::IFillStreamInfo(AVISTREAMINFO* inf, plPipeline* pipeline)
|
||||
{
|
||||
memset(inf, 0, sizeof(AVISTREAMINFO));
|
||||
inf->fccType = streamtypeVIDEO;
|
||||
inf->fccHandler = 0;
|
||||
inf->dwScale = 1;
|
||||
inf->dwRate = kFramesPerSec;
|
||||
|
||||
SetRect(&inf->rcFrame,
|
||||
0,0,
|
||||
pipeline->Width(),
|
||||
pipeline->Height());
|
||||
}
|
||||
|
||||
void plAVIWriterImp::IFillBitmapInfo(BITMAPINFOHEADER* inf, plPipeline* pipeline)
|
||||
{
|
||||
memset(inf,0,sizeof(BITMAPINFOHEADER));
|
||||
inf->biSize = sizeof(BITMAPINFOHEADER);
|
||||
inf->biPlanes = 1;
|
||||
inf->biBitCount = 32;
|
||||
inf->biCompression = BI_RGB;
|
||||
inf->biSizeImage = 0;
|
||||
inf->biXPelsPerMeter = 0;
|
||||
inf->biYPelsPerMeter = 0;
|
||||
inf->biClrUsed = 0;
|
||||
inf->biClrImportant = 0;
|
||||
inf->biWidth = pipeline->Width();
|
||||
inf->biHeight = pipeline->Height();
|
||||
}
|
||||
|
||||
bool plAVIWriterImp::ICaptureFrame(plPipeline* pipeline)
|
||||
{
|
||||
plMipmap frame;
|
||||
pipeline->CaptureScreen(&frame, true);
|
||||
|
||||
double time = hsTimer::GetSysSeconds() - fStartTime;
|
||||
time *= kFramesPerSec;
|
||||
|
||||
HRESULT err;
|
||||
err = AVIStreamWrite( fCompressedHandle,
|
||||
int(time),
|
||||
1,
|
||||
(LPBYTE)frame.GetAddr32(0,0),
|
||||
frame.GetTotalSize(),
|
||||
AVIIF_KEYFRAME,
|
||||
NULL,
|
||||
NULL);
|
||||
|
||||
return (err == AVIERR_OK);
|
||||
}
|
||||
/*==LICENSE==*
|
||||
|
||||
CyanWorlds.com Engine - MMOG client, server and tools
|
||||
Copyright (C) 2011 Cyan Worlds, Inc.
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
You can contact Cyan Worlds, Inc. by email legal@cyan.com
|
||||
or by snail mail at:
|
||||
Cyan Worlds, Inc.
|
||||
14617 N Newport Hwy
|
||||
Mead, WA 99021
|
||||
|
||||
*==LICENSE==*/
|
||||
#include "plAVIWriter.h"
|
||||
|
||||
#include "hsTypes.h"
|
||||
|
||||
#include "hsWindows.h"
|
||||
#include <vfw.h>
|
||||
|
||||
#include "hsTimer.h"
|
||||
#include "plMipmap.h"
|
||||
#include "plMessage/plRenderMsg.h"
|
||||
#include "plPipeline.h"
|
||||
#include "pnDispatch/plDispatch.h"
|
||||
#include "pnKeyedObject/plFixedKey.h"
|
||||
|
||||
bool plAVIWriter::fInitialized = false;
|
||||
|
||||
class plAVIWriterImp : public plAVIWriter
|
||||
{
|
||||
protected:
|
||||
PAVIFILE fFileHandle;
|
||||
PAVISTREAM fStreamHandle;
|
||||
PAVISTREAM fCompressedHandle;
|
||||
BITMAPINFOHEADER fBitmapInfo;
|
||||
|
||||
hsBool fOldRealTime;
|
||||
hsScalar fOldFrameTimeInc;
|
||||
|
||||
double fStartTime;
|
||||
|
||||
void IFillStreamInfo(AVISTREAMINFO* inf, plPipeline* pipeline);
|
||||
void IFillBitmapInfo(BITMAPINFOHEADER* inf, plPipeline* pipeline);
|
||||
|
||||
bool ICaptureFrame(plPipeline* pipeline);
|
||||
|
||||
public:
|
||||
plAVIWriterImp();
|
||||
virtual ~plAVIWriterImp();
|
||||
|
||||
virtual hsBool MsgReceive(plMessage* msg);
|
||||
|
||||
virtual void Shutdown();
|
||||
|
||||
virtual bool Open(const char* fileName, plPipeline* pipeline);
|
||||
virtual void Close();
|
||||
};
|
||||
|
||||
plAVIWriter::~plAVIWriter()
|
||||
{
|
||||
}
|
||||
|
||||
plAVIWriter& plAVIWriter::Instance()
|
||||
{
|
||||
static plAVIWriterImp theInstance;
|
||||
|
||||
if (!fInitialized)
|
||||
{
|
||||
theInstance.RegisterAs(kAVIWriter_KEY);
|
||||
fInitialized = true;
|
||||
}
|
||||
|
||||
return theInstance;
|
||||
}
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
plAVIWriterImp::plAVIWriterImp() :
|
||||
fStartTime(0),
|
||||
fOldRealTime(false),
|
||||
fStreamHandle(nil),
|
||||
fCompressedHandle(nil),
|
||||
fFileHandle(nil)
|
||||
{
|
||||
AVIFileInit();
|
||||
}
|
||||
|
||||
plAVIWriterImp::~plAVIWriterImp()
|
||||
{
|
||||
}
|
||||
|
||||
void plAVIWriterImp::Shutdown()
|
||||
{
|
||||
Close();
|
||||
UnRegisterAs(kAVIWriter_KEY);
|
||||
SetKey(nil);
|
||||
}
|
||||
|
||||
|
||||
|
||||
#include "plProfile.h"
|
||||
plProfile_CreateTimer("AviCapture", "RenderSetup", AviCapture);
|
||||
|
||||
hsBool plAVIWriterImp::MsgReceive(plMessage* msg)
|
||||
{
|
||||
|
||||
plRenderMsg* renderMsg = plRenderMsg::ConvertNoRef(msg);
|
||||
if (renderMsg)
|
||||
{
|
||||
plProfile_BeginTiming(AviCapture);
|
||||
|
||||
ICaptureFrame(renderMsg->Pipeline());
|
||||
plProfile_EndTiming(AviCapture);
|
||||
|
||||
}
|
||||
|
||||
return hsKeyedObject::MsgReceive(msg);
|
||||
}
|
||||
|
||||
static const int kFramesPerSec = 30;
|
||||
|
||||
bool plAVIWriterImp::Open(const char* fileName, plPipeline* pipeline)
|
||||
{
|
||||
// Already writing, fail
|
||||
if (fStreamHandle)
|
||||
return false;
|
||||
|
||||
fStartTime = hsTimer::GetSysSeconds();
|
||||
|
||||
// If we're running in real time, set to frame time
|
||||
fOldRealTime = hsTimer::IsRealTime();
|
||||
if (fOldRealTime)
|
||||
{
|
||||
hsTimer::SetRealTime(false);
|
||||
hsTimer::SetFrameTimeInc(1.f / kFramesPerSec);
|
||||
}
|
||||
|
||||
// Open AVI file
|
||||
HRESULT err;
|
||||
err = AVIFileOpen( &fFileHandle, // returned file pointer
|
||||
fileName, // file name
|
||||
OF_WRITE | OF_CREATE, // mode to open file with
|
||||
NULL); // use handler determined
|
||||
hsAssert(err == AVIERR_OK, "Error creating AVI file in plAVIWriter::Open");
|
||||
if (err != AVIERR_OK)
|
||||
{
|
||||
Close();
|
||||
return false;
|
||||
}
|
||||
|
||||
AVISTREAMINFO streamInfo;
|
||||
IFillStreamInfo(&streamInfo, pipeline);
|
||||
|
||||
// Create a video stream in the file
|
||||
err = AVIFileCreateStream( fFileHandle, // file pointer
|
||||
&fStreamHandle, // returned stream pointer
|
||||
&streamInfo ); // stream header
|
||||
hsAssert(err == AVIERR_OK, "Error creating video stream in plAVIWriter::Open");
|
||||
if (err != AVIERR_OK)
|
||||
{
|
||||
Close();
|
||||
return false;
|
||||
}
|
||||
|
||||
do
|
||||
{
|
||||
AVICOMPRESSOPTIONS opts;
|
||||
AVICOMPRESSOPTIONS FAR * aopts[1] = {&opts};
|
||||
memset(&opts, 0, sizeof(opts));
|
||||
|
||||
BOOL bErr = AVISaveOptions(NULL, ICMF_CHOOSE_DATARATE, 1, &fStreamHandle, (LPAVICOMPRESSOPTIONS FAR*)&aopts);
|
||||
hsAssert(bErr, "Error saving stream options in plAVIWriter::Open");
|
||||
if (!bErr)
|
||||
{
|
||||
Close();
|
||||
return false;
|
||||
}
|
||||
|
||||
err = AVIMakeCompressedStream(&fCompressedHandle, fStreamHandle, &opts, NULL);
|
||||
hsAssert(err == AVIERR_OK, "Error creating compressed stream in plAVIWriter::Open");
|
||||
if (err != AVIERR_OK)
|
||||
{
|
||||
Close();
|
||||
return false;
|
||||
}
|
||||
|
||||
IFillBitmapInfo(&fBitmapInfo, pipeline);
|
||||
err = AVIStreamSetFormat( fCompressedHandle, 0,
|
||||
&fBitmapInfo, // stream format
|
||||
fBitmapInfo.biSize);
|
||||
} while (err != AVIERR_OK &&
|
||||
hsMessageBox("Codec unavailable, try again?", "AVI Writer", hsMessageBoxYesNo) == hsMBoxYes);
|
||||
|
||||
if (err != AVIERR_OK)
|
||||
{
|
||||
Close();
|
||||
return false;
|
||||
}
|
||||
|
||||
plgDispatch::Dispatch()->RegisterForExactType(plRenderMsg::Index(), GetKey());
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void plAVIWriterImp::Close()
|
||||
{
|
||||
plgDispatch::Dispatch()->UnRegisterForExactType(plRenderMsg::Index(), GetKey());
|
||||
|
||||
hsTimer::SetRealTime(fOldRealTime);
|
||||
|
||||
if (fStreamHandle)
|
||||
{
|
||||
AVIStreamClose(fStreamHandle);
|
||||
fStreamHandle = nil;
|
||||
}
|
||||
|
||||
if (fCompressedHandle)
|
||||
{
|
||||
AVIStreamClose(fCompressedHandle);
|
||||
fCompressedHandle = nil;
|
||||
}
|
||||
|
||||
if (fFileHandle)
|
||||
{
|
||||
AVIFileClose(fFileHandle);
|
||||
fFileHandle = nil;
|
||||
}
|
||||
|
||||
AVIFileExit();
|
||||
}
|
||||
|
||||
void plAVIWriterImp::IFillStreamInfo(AVISTREAMINFO* inf, plPipeline* pipeline)
|
||||
{
|
||||
memset(inf, 0, sizeof(AVISTREAMINFO));
|
||||
inf->fccType = streamtypeVIDEO;
|
||||
inf->fccHandler = 0;
|
||||
inf->dwScale = 1;
|
||||
inf->dwRate = kFramesPerSec;
|
||||
|
||||
SetRect(&inf->rcFrame,
|
||||
0,0,
|
||||
pipeline->Width(),
|
||||
pipeline->Height());
|
||||
}
|
||||
|
||||
void plAVIWriterImp::IFillBitmapInfo(BITMAPINFOHEADER* inf, plPipeline* pipeline)
|
||||
{
|
||||
memset(inf,0,sizeof(BITMAPINFOHEADER));
|
||||
inf->biSize = sizeof(BITMAPINFOHEADER);
|
||||
inf->biPlanes = 1;
|
||||
inf->biBitCount = 32;
|
||||
inf->biCompression = BI_RGB;
|
||||
inf->biSizeImage = 0;
|
||||
inf->biXPelsPerMeter = 0;
|
||||
inf->biYPelsPerMeter = 0;
|
||||
inf->biClrUsed = 0;
|
||||
inf->biClrImportant = 0;
|
||||
inf->biWidth = pipeline->Width();
|
||||
inf->biHeight = pipeline->Height();
|
||||
}
|
||||
|
||||
bool plAVIWriterImp::ICaptureFrame(plPipeline* pipeline)
|
||||
{
|
||||
plMipmap frame;
|
||||
pipeline->CaptureScreen(&frame, true);
|
||||
|
||||
double time = hsTimer::GetSysSeconds() - fStartTime;
|
||||
time *= kFramesPerSec;
|
||||
|
||||
HRESULT err;
|
||||
err = AVIStreamWrite( fCompressedHandle,
|
||||
int(time),
|
||||
1,
|
||||
(LPBYTE)frame.GetAddr32(0,0),
|
||||
frame.GetTotalSize(),
|
||||
AVIIF_KEYFRAME,
|
||||
NULL,
|
||||
NULL);
|
||||
|
||||
return (err == AVIERR_OK);
|
||||
}
|
||||
|
Reference in New Issue
Block a user