www.pudn.com > XvidQP.rar > CReportFile.cpp

// CReportFile.cpp: implementation of the CReportFile class. 
// 
////////////////////////////////////////////////////////////////////// 
 
#include "stdafx.h" 
#include "XvidQuantsParser.h" 
#include "CReportFile.h" 
 
#ifdef _DEBUG 
#undef THIS_FILE 
static char THIS_FILE[]=__FILE__; 
#define new DEBUG_NEW 
#endif 
 
////////////////////////////////////////////////////////////////////// 
// Construction/Destruction 
////////////////////////////////////////////////////////////////////// 
 
CReportFile::CReportFile() 
{ 
	//data = NULL; 
} 
 
CReportFile::~CReportFile() 
{ 
	//if (data != NULL) delete[] data; 
} 
 
//----------------------------------------------------------------------------------------------- 
bool CReportFile::Save(CString filepath, CXvid &xvid, int start, int stop) 
{ 
	CFile rfile; 
 
	//File header 
	fileHdr.rtyp	= 0x08; 
	fileHdr.rmajor	= 1; 
	fileHdr.rminor	= 1; 
 
	//Info header 
	infoHdr.fps		= xvid.GetRate(); 
	infoHdr.width	= xvid.GetWidth(); 
	infoHdr.height	= xvid.GetHeight(); 
	infoHdr.nframes	= stop-start; 
	infoHdr.nmax	= xvid.GetMaxQuant(); 
	infoHdr.nmin	= xvid.GetMinQuant(); 
	infoHdr.navg	= xvid.GetAvgQuants(); 
	infoHdr.icount	= xvid.GetIFrameCount(); 
	infoHdr.iavg	= xvid.GetIAvgQuants(); 
	infoHdr.pcount	= xvid.GetPFrameCount(); 
	infoHdr.pavg	= xvid.GetPAvgQuants(); 
	infoHdr.nfsmax	= xvid.GetMaxFrameSize(); 
	infoHdr.nfsmin	= xvid.GetMinFrameSize(); 
	infoHdr.ifsmax	= xvid.GetIMaxFrameSize(); 
	infoHdr.ifsmin	= xvid.GetIMinFrameSize(); 
	infoHdr.pfsmax	= xvid.GetPMaxFrameSize(); 
	infoHdr.pfsmin	= xvid.GetPMinFrameSize(); 
 
	//save file... 
	if (!rfile.Open(filepath, CFile::modeCreate|CFile::modeWrite)) 
	{ 
		return false; 
	} 
 
	rfile.Write(&fileHdr, sizeof(REPORTFILEHEADER)); 
	rfile.Write(&infoHdr, sizeof(REPORTINFOHEADER)); 
	rfile.Write(xvid.GetQuantsArray(), sizeof(CXvid::QUANTS)*infoHdr.nframes); 
 
	//cleanup 
	rfile.Close(); 
	return true; 
} 
 
//----------------------------------------------------------------------------------------------- 
bool CReportFile::Load(CString filepath, CXvid &xvid) 
{ 
	CFile rfile; 
	CXvid::QUANTS* databuf=NULL; 
 
	//load file... 
	if (!rfile.Open(filepath, CFile::modeRead)) 
	{ 
		return false; 
	} 
 
	//Check file header 
	rfile.Read(&fileHdr, sizeof(REPORTFILEHEADER)); 
	if (fileHdr.rtyp != 0x08) 
	{ 
		return false; 
	} 
	if (fileHdr.rminor != 1) 
	{ 
		AfxMessageBox("Wrong report file version!"); 
		return false; 
	} 
 
	//load info header 
	rfile.Read(&infoHdr, sizeof(REPORTINFOHEADER)); 
	xvid.SetRate(infoHdr.fps); 
	xvid.SetWidth(infoHdr.width); 
	xvid.SetHeight(infoHdr.height); 
	xvid.SetFrameCount(infoHdr.nframes); 
	xvid.SetMaxQuant(infoHdr.nmax); 
	xvid.SetMinQuant(infoHdr.nmin); 
	xvid.SetAvgQuants(infoHdr.navg); 
	xvid.SetIFrameCount(infoHdr.icount); 
	xvid.SetIAvgQuants(infoHdr.iavg); 
	xvid.SetPFrameCount(infoHdr.pcount); 
	xvid.SetPAvgQuants(infoHdr.pavg); 
	xvid.SetMaxFrameSize(infoHdr.nfsmax); 
	xvid.SetMinFrameSize(infoHdr.nfsmin); 
	xvid.SetIMaxFrameSize(infoHdr.ifsmax); 
	xvid.SetIMinFrameSize(infoHdr.ifsmin); 
	xvid.SetPMaxFrameSize(infoHdr.pfsmax); 
	xvid.SetPMinFrameSize(infoHdr.pfsmin); 
 
	//load quants data 
	databuf = new CXvid::QUANTS[infoHdr.nframes]; 
	rfile.Read(databuf, sizeof(CXvid::QUANTS)*infoHdr.nframes); 
	xvid.SetQuantsArray(databuf, infoHdr.nframes); 
 
	//cleanup 
	rfile.Close(); 
	xvid.SetOpened(true); 
	xvid.SetAnalysed(true); 
	return true; 
}