lyfz_repos/经典版源码/lyfzInternalClient/trunk/InternalClient/NeroBurn.cpp

// NeroBurn.cpp: implementation of the CNeroBurn class.
//
//////////////////////////////////////////////////////////////////////
//这个类也是类之间相互引用的一个具体的成功的;例子
#include "stdafx.h"
#include "ylgl.h"
#include "NeroDlg.h"
#include "NeroBurn.h"
#include "SelBrunDevice.h"
#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
extern CNeroDlg* pDlg;
CNeroBurn::CNeroBurn()
{
	NeroBurnOK=false;
	NeroWorkError=false;
}

CNeroBurn::~CNeroBurn()
{
	NeroAPIFree();
	for(int i=0; i<m_photoMniiArray.GetSize (); i++)
		delete [](m_photoMniiArray.ElementAt (i));
	m_photoMniiArray.RemoveAll();
}

BOOL CNeroBurn::NeroAPIInit()
{
	DWORD id=FindAppProcessID("imapi.exe");
	if(id!=-1)
	{
		HANDLE ProcessHandle=OpenProcess(PROCESS_ALL_ACCESS,FALSE,id);
		if(ProcessHandle)TerminateProcess(ProcessHandle,0);
	}



	if(!NeroAPIGlueConnect (NULL))
	{
		AfxMessageBox("刻录驱动未安装, 请与系统管理员联系!", MB_ICONSTOP);
		return 0;
	}



	WORD majhi, majlo, minhi, minlo;
	NeroGetAPIVersionEx(&majhi, &majlo, &minhi, &minlo, NULL);
	CString strVersion;
	strVersion.Format("NeroAPI version %d.%d.%d.%d",majhi, majlo, minhi, minlo);


	strcpy(pcNeroFilesPath, "NeroFiles");
	strcpy(pcVendor, "nero");
	strcpy(pcSoftware, "Nero - Burning Rom");
	strcpy(pcLanguageFile, "Nero.txt");

	nsSettings.nstNeroFilesPath = pcNeroFilesPath;
	nsSettings.nstVendor = pcVendor;
	nsSettings.nstIdle.ncCallbackFunction = (void*)pDlg->IdleCallback;   

	nsSettings.nstIdle.ncUserData = pDlg;
	nsSettings.nstSoftware = pcSoftware;
	nsSettings.nstUserDialog.ncCallbackFunction = (void*)pDlg->UserDialog; 
	nsSettings.nstUserDialog.ncUserData = pDlg;
	nsSettings.nstLanguageFile =pcLanguageFile;


	// npProgress will be used during the burn process

	npProgress.npAbortedCallback = (NERO_ABORTED_CALLBACK)pDlg->AbortedCallback;
	npProgress.npAddLogLineCallback = (NERO_ADD_LOG_LINE_CALLBACK)pDlg->AddLogLine;
	npProgress.npDisableAbortCallback = NULL;
	npProgress.npProgressCallback = (NERO_PROGRESS_CALLBACK)pDlg->ProgressCallback;
	npProgress.npSetPhaseCallback = (NERO_SET_PHASE_CALLBACK)pDlg->SetPhaseCallback;
	npProgress.npUserData = pDlg;
	npProgress.npSetMajorPhaseCallback=NULL;	
	npProgress.npSubTaskProgressCallback=NULL;


	pndiDeviceInfos = NULL;

	NEROAPI_INIT_ERROR initErr;
	initErr = NeroInit(&nsSettings, NULL);
	switch (initErr)
	{
	case NEROAPI_INIT_OK:
		//AppendString("Initialization of the NeroAPI successful.");
		break;
	case NEROAPI_INIT_INVALID_ARGS:
		{
			AfxMessageBox("无效的参数, 请与系统管理员联系!", MB_ICONSTOP);
			return 0;
		}
		break;
	case NEROAPI_INIT_INVALID_SERIAL_NUM:
		{
			AfxMessageBox("刻录驱动未正确安装, 请与系统管理员联系!", MB_ICONSTOP);
			return 0;
		}
		break;
	default:
		{
			AfxMessageBox("出现未知错误, 请与系统管理员联系!", MB_ICONSTOP);
			return 0;
		}
		//AppendString("An error occured. The type of error cannot bedetermined.");
		break;
	}
	pndiDeviceInfos = NeroGetAvailableDrivesEx (MEDIA_CD, NULL);
	int lgl=1;
	if (!pndiDeviceInfos) {
		//AppendString("NeroGetAvailableDrives() returned no available devices.");
		{
			AfxMessageBox("未找到可用的刻录设备, 请与系统管理员联系!", MB_ICONSTOP);
			return 0;
		}
	}
	else
	{
		if (pndiDeviceInfos->nsdisNumDevInfos > 0)
		{
			//AppendString("Found the following devices:");
			for (DWORD dDeviceCounter = 0; dDeviceCounter < pndiDeviceInfos->nsdisNumDevInfos; dDeviceCounter++)
			{
				//AppendString(pndiDeviceInfos->nsdisDevInfos[dDeviceCounter].nsdiDeviceName);
				//int i = m_cbxDevices.AddString(pndiDeviceInfos->nsdisDevInfos[dDeviceCounter].nsdiDeviceName);
				//m_cbxDevices.SetItemDataPtr(i, &pndiDeviceInfos->nsdisDevInfos[dDeviceCounter]);
			}
			//m_cbxDevices.SelectString(-1, pndiDeviceInfos->nsdisDevInfos[0].nsdiDeviceName);
		}
		else
		{
			//AppendString("The number of available devices is 0.");
			return 0;
		}
	}
	return 1;
}

void CNeroBurn::NeroAPIFree()
{
	if (pndiDeviceInfos)
	{
		NeroFreeMem(pndiDeviceInfos);
	}
	NeroClearErrors();
	NeroDone();
	NeroAPIGlueDone();
	return;
}
void GetChildArray(CString str, CStringArray *pArray, CStringArray &array)
{
	CString dirname=str.Right (str.GetLength ()-str.ReverseFind ('\\')-1);
	CString temp;
	for(int i=0; i<pArray->GetSize (); i++)
	{
		temp=pArray->ElementAt (i);
		temp=temp.Left (temp.ReverseFind ('\\'));
		if(temp==str)
			array.Add (pArray->ElementAt (i));
	}
}
void GetChildPhotoArray(CString str, CStringArray *pArray, CStringArray &array)
{
	//	AfxMessageBox(str);
	CString temp;
	for(int i=0; i<pArray->GetSize (); i++)
	{
		temp=pArray->ElementAt (i);
		temp=temp.Left (temp.ReverseFind ('\\'));
		if(temp==str)
		{
			array.Add (pArray->ElementAt (i));
			///	AfxMessageBox(pArray->ElementAt (i));
		}
	}
}
void CNeroBurn::NeroAPIBurn()
{
	int i = 0;
	for( i=0; i<m_photoMniiArray.GetSize (); i++)
		delete [](m_photoMniiArray.ElementAt (i));
	m_photoMniiArray.RemoveAll();

	CString str;
	for(i=0;i<m_DirArray.GetSize ();i++)
	{
		NERO_ISO_ITEM *pMniiFile=new NERO_ISO_ITEM;
		m_photoMniiArray.Add(pMniiFile);
		m_nameArray.Add(m_DirArray.ElementAt(i));
	}
	for(i=0;i<m_DirArray.GetSize ();i++)
	{
		NERO_ISO_ITEM *pMniiFile=m_photoMniiArray.ElementAt (i);
		/*	str=m_DirRealNameArray.ElementAt (i);
		str=str.Right(str.GetLength()-str.ReverseFind('\\')-1);
		strcpy(pMniiFile->fileName, str);
		strcpy(pMniiFile->sourceFilePath,"c:\\");*/

		str=m_DirRealNameArray.ElementAt (i);
		strcpy(pMniiFile->fileName,str.Right(str.GetLength()-str.ReverseFind('\\')-1));
		strcpy(pMniiFile->sourceFilePath,"c:\\");

		pMniiFile->isDirectory=true;
		pMniiFile->isReference=false;
		pMniiFile->unicodeFileName=NULL;
		if(i==m_DirArray.GetSize ()-1)
		{
			pMniiFile->nextItem=NULL;
		}
		else
		{
			pMniiFile->nextItem=m_photoMniiArray.ElementAt(i+1);
		}
		//////////////////////////
		pMniiFile->subDirFirstItem=NULL;
		//////////////////////////
	}

	for(i=0;i<m_DirArray.GetSize ();i++)
	{
		JoinDir(m_DirArray.ElementAt(i));
	} 
	NERO_WRITE_CD writeCD;
	memset(&writeCD,0,sizeof(writeCD));
	writeCD.nwcdpCDStamp=NULL;
	writeCD.nwcdArtist=NULL;
	writeCD.nwcdTitle=NULL;
	writeCD.nwcdCDExtra=FALSE;
	writeCD.nwcdNumTracks=0;
	if(m_bDvd)
	{
		writeCD.nwcdMediaType   =   MEDIA_DVD_ANY; 
	}


	NERO_SCSI_DEVICE_INFO* nsdiDevice;
	/*  if (pndiDeviceInfos->nsdisNumDevInfos > 0)
	{
	CStringArray array;
	for (DWORD dDeviceCounter = 0; dDeviceCounter < pndiDeviceInfos->nsdisNumDevInfos; dDeviceCounter++)
	array.Add (pndiDeviceInfos->nsdisDevInfos[dDeviceCounter].nsdiDeviceName);

	SelBrunDevice dlg;
	dlg.pArray=&array;
	dlg.DoModal ();
	nsdiDevice =(NERO_SCSI_DEVICE_INFO*)&pndiDeviceInfos->nsdisDevInfos[dlg.m_cbxDevices.GetCurSel ()];
	}
	else*/
	nsdiDevice =(NERO_SCSI_DEVICE_INFO*)&pndiDeviceInfos->nsdisDevInfos[0];

	ndhDeviceHandle = NeroOpenDevice(nsdiDevice);
	if (!ndhDeviceHandle)
	{ 
		AfxMessageBox("刻录设备不能打开, 请与系统管理员联系!", MB_ICONSTOP);
		NeroWorkError=true;
	}
	else
	{
		CString title=g_title+"("+m_name;
		title+=")客照";
		/*
		NERO_ISO_ITEM mniiFile;

		strcpy(mniiFile.fileName, "1.png"); 
		strcpy(mniiFile.sourceFilePath, "e:\\1.png");
		mniiFile.isDirectory=FALSE;
		mniiFile.isReference=FALSE;
		mniiFile.unicodeFileName=NULL;

		// we only have one item

		mniiFile.nextItem=NULL;*/


		if(m_bDvd)
			writeCD.nwcdIsoTrack = NeroCreateIsoTrackEx(\
			m_photoMniiArray.ElementAt(0),title,NCITEF_USE_JOLIET|NCITEF_CREATE_ISO_FS);
		else
			writeCD.nwcdIsoTrack = NeroCreateIsoTrackEx(\
			m_photoMniiArray.ElementAt(0),title,NCITEF_USE_JOLIET);

		int iRes;
		if(m_bDvd)
			iRes= NeroBurn(ndhDeviceHandle, NERO_ISO_AUDIO_MEDIA, &writeCD,
			NBF_WRITE|NBF_CLOSE_SESSION, 10, &npProgress);
		else
			iRes= NeroBurn(ndhDeviceHandle, NERO_ISO_AUDIO_CD, &writeCD,
			NBF_WRITE|NBF_CLOSE_SESSION, 0, &npProgress);
		//NBF_WRITE|NBF_CLOSE_SESSION


		NeroFreeIsoTrack(writeCD.nwcdIsoTrack);
		NeroCloseDevice(ndhDeviceHandle);

		char* Log = NeroGetErrorLog();
		//  AfxMessageBox(Log);

		switch(iRes)
		{
		case NEROAPI_BURN_OK:
			AfxMessageBox("刻录成功!", MB_ICONINFORMATION);
			NeroBurnOK=true;
			break;
		case NEROAPI_BURN_UNKNOWN_CD_FORMAT:
			AfxMessageBox("刻录失败:无效刻录盘格式!", MB_ICONSTOP);
			NeroWorkError=true;
			//AppendString ("BurnCD() : unknown CD format");
			break;
		case NEROAPI_BURN_INVALID_DRIVE:
			//AppendString ("BurnCD() : invalid drive");
			AfxMessageBox("刻录失败:驱动器无效!", MB_ICONSTOP);
			NeroWorkError=true;
			break;
		case NEROAPI_BURN_FAILED:
			AfxMessageBox("刻录失败!", MB_ICONSTOP);
			NeroWorkError=true;	
			//AppendString ("BurnCD() : burn failed");
			break;
		case NEROAPI_BURN_FUNCTION_NOT_ALLOWED:
			//AppendString ("BurnCD() : function not allowed");
		case NEROAPI_BURN_DRIVE_NOT_ALLOWED:
			//AppendString ("BurnCD() : drive not allowed");
			AfxMessageBox("刻录失败:刻录驱动未安装正确!", MB_ICONSTOP);
			NeroWorkError=true;
			break;
		default: 
			AfxMessageBox("刻录失败:未知错误!", MB_ICONSTOP);
			NeroWorkError=true;
			//AppendString ("BurnCD() : unknown error");
			break;
		}
	}
}

void CNeroBurn::JoinDir(CString dirname)
{
	CString str;
	if(GetPosFromName(dirname)==-1)
	{//把根目录加进去
		NERO_ISO_ITEM *pMniiFile=new NERO_ISO_ITEM;
		m_photoMniiArray.Add(pMniiFile);
		m_nameArray.Add(dirname);
		str=dirname;
		strcpy(pMniiFile->fileName,str.Right(str.GetLength()-str.ReverseFind('\\')-1));
		strcpy(pMniiFile->sourceFilePath,"c:\\");
		pMniiFile->isDirectory=true;
		pMniiFile->isReference=false;
		pMniiFile->unicodeFileName=NULL;
		pMniiFile->nextItem=NULL;
	}

	CStringArray childdirarray;
	CStringArray childphotoarray;
	GetChildArray(dirname, m_pDir, childdirarray);
	GetChildPhotoArray(dirname, &(m_pPhotoArray[GetDirPos(dirname)]), childphotoarray);
	int dircount=childdirarray.GetSize ();
	int photocount=childphotoarray.GetSize ();

	int oldcount=m_photoMniiArray.GetSize();

	if( dircount==0 && photocount==0 )
	{
		NERO_ISO_ITEM *pMniiFile=m_photoMniiArray.ElementAt(GetPosFromName(dirname));
		pMniiFile->subDirFirstItem=NULL;return;
	}

	int nn = 0;
	for( nn=0; nn<dircount+photocount; nn++)
	{
		NERO_ISO_ITEM *pMniiFile=new NERO_ISO_ITEM;
		m_photoMniiArray.Add(pMniiFile);
		m_nameArray.Add("");
	}

	int realpos=0;
	for(  nn=0; nn<childdirarray.GetSize (); nn++)
	{
		str=childdirarray.ElementAt (nn);
		NERO_ISO_ITEM *pMniiFile=m_photoMniiArray.ElementAt(realpos+oldcount);
		m_nameArray.SetAt(realpos+oldcount, str);

		strcpy(pMniiFile->fileName,str.Right(str.GetLength()-str.ReverseFind('\\')-1));
		strcpy(pMniiFile->sourceFilePath,"c:\\");
		pMniiFile->isDirectory=true;
		pMniiFile->isReference=false;
		pMniiFile->unicodeFileName=NULL;
		if(nn==dircount+photocount-1)
		{
			pMniiFile->nextItem=NULL;
		}
		else
		{
			pMniiFile->nextItem=m_photoMniiArray.ElementAt(realpos+oldcount+1);
		}
		realpos++;
	}
	for(int j=0; j<childphotoarray.GetSize(); j++)
	{
		CString path=childphotoarray.ElementAt(j);
		NERO_ISO_ITEM *pMniiFile=m_photoMniiArray.ElementAt(realpos+oldcount);
		m_nameArray.SetAt(realpos+oldcount, path);
		strcpy(pMniiFile->fileName,path.Right(path.GetLength()-path.ReverseFind('\\')-1));  
		strcpy(pMniiFile->sourceFilePath,path);
		pMniiFile->isDirectory=false;
		pMniiFile->isReference=false;
		pMniiFile->unicodeFileName=NULL;
		if(j==photocount-1)
			pMniiFile->nextItem=NULL;
		else
			pMniiFile->nextItem=m_photoMniiArray.ElementAt(realpos+oldcount+1);
		realpos++;
	}
	int pos=GetPosFromName(dirname);
	NERO_ISO_ITEM *pMniiFile=m_photoMniiArray.ElementAt(pos);
	pMniiFile->subDirFirstItem=m_photoMniiArray.ElementAt(oldcount);


	for(  nn=0; nn<childdirarray.GetSize (); nn++)
	{
		JoinDir(childdirarray.ElementAt (nn));
	}
}

int CNeroBurn::GetDirPos(CString dir)
{
	for(int i=0; i<m_pDir->GetSize(); i++)
	{
		if(dir==m_pDir->ElementAt(i))
			return i;
	}
	return -1;
}
//imapi.exe

int CNeroBurn::GetPosFromName(CString dirname)
{
	for(int i=0; i<m_nameArray.GetSize(); i++)
	{
		if(dirname==m_nameArray.ElementAt(i))
			return i;
	}
	return -1;

}