Code Complete

// main.cpp

#include "stdafx.h"

#include "Markup.h"

int _tmain(int argc, _TCHAR* argv[])

{

// 시스템 지역 설정을 따르게 셋팅한다.

setlocale(LC_ALL, "");

const bool IsCommandArgumentsError = argc != 5;

if (true == IsCommandArgumentsError) 

{

printf("명령어 오류\n");

return 1;

}

CMarkup xml;

const TCHAR* pVCProjectFileName = argv[1];

unityMaker::g_strFileConfigurationName = argv[2];

const TCHAR* pExcludedNameFilter = argv[3];

const int iSplitCount = _ttoi(argv[4]);

unityMaker::g_strFileConfigurationName += _T("|Win32");

// 파일 경로를 얻어낸다.

TCHAR czPath[MAX_PATH] = {0,};

memcpy(czPath, pVCProjectFileName, sizeof(TCHAR)*_tcslen(pVCProjectFileName));

PathRemoveFileSpec(czPath);

if (!xml.Load(pVCProjectFileName))

{

_tprintf(_T("xml 파일 오픈 실패%s \n"), pVCProjectFileName);

return 2;

}

// 소스 파일 목록이 있는 엘리먼트를 찾아 들어간다.

xml.FindChildElem(_T("Files"));

xml.IntoElem();

// 프로젝트 파일을 파싱하면서 소스를 빌드에서 제외시키고 해당 파일들의 목록을 얻어온다.

std::vector<MCD_STR> vecExcludedFromBuild;

unityMaker::ExcludedFromBuild(xml, MCD_STR(), pExcludedNameFilter, vecExcludedFromBuild);

const int iFileChunkSize = static_cast<int>(ceil((static_cast<float>(vecExcludedFromBuild.size()/iSplitCount))));

const bool IsAllInOne = iSplitCount <= 1;

if (true == IsAllInOne)

{

MCD_STR strUnityBuildFileName = unityMaker::GetUnityBuildFileNamePathCombine(czPath, pExcludedNameFilter, 0);

unityMaker::MakeUnityBuildFile(strUnityBuildFileName, vecExcludedFromBuild);

}

else

{

int i = 0;

for (i = 0; i < iSplitCount-1; ++i) 

{

std::vector<MCD_STR> vecChunk(vecExcludedFromBuild.begin() + (i*iFileChunkSize), vecExcludedFromBuild.begin() + (i+1)*iFileChunkSize);

MCD_STR strUnityBuildFileName = unityMaker::GetUnityBuildFileNamePathCombine(czPath, pExcludedNameFilter, i);

unityMaker::MakeUnityBuildFile(strUnityBuildFileName, vecChunk);

}

std::vector<MCD_STR> vecChunk(vecExcludedFromBuild.begin() + (i*iFileChunkSize), vecExcludedFromBuild.end());

MCD_STR strUnityBuildFileName = unityMaker::GetUnityBuildFileNamePathCombine(czPath, pExcludedNameFilter, i);

unityMaker::MakeUnityBuildFile(strUnityBuildFileName, vecChunk);

}

if (true == unityMaker::g_bIsSaveXML)

{

xml.Save(pVCProjectFileName);

}

return 0;

}

namespace unityMaker

{

MCD_STR g_strFileConfigurationName;

bool g_bIsSaveXML = false;

/**

@brief 재귀적으로 호출되면서 소스코드를 빌드에서 제외시키고 제외된 .cpp의 목록을 얻는다.

@param xml - xml 파서

@param

@param strExcludedNameFilter - 검색에서 제외할 필더 이름

@param output_vecExcludedFromBuild - 빌드에서 제외된 소스 파일 목록

*/

void ExcludedFromBuild(CMarkup& xml, const MCD_STR& nameFilterParent, const MCD_STR& strExcludedNameFilter, std::vector<MCD_STR>& output_vecExcludedFromBuild)

{

while (xml.FindChildElem(_T("Filter")))

{

xml.IntoElem();

MCD_STR nameFilter = xml.GetAttrib(_T("Name"));

// 대상에서 제외한다.

if (nameFilter == strExcludedNameFilter)

{

xml.OutOfElem();

continue;

}

ExcludedFromBuild(xml, nameFilter, strExcludedNameFilter, output_vecExcludedFromBuild);

xml.OutOfElem();

}

xml.ResetChildPos();

while (xml.FindChildElem(_T("File")))

{

xml.IntoElem();

MCD_STR namePath = xml.GetAttrib(_T("RelativePath"));

const bool IsTargetSourceCode = namePath.find(_T(".cpp")) != MCD_STR::npos 

&& namePath.find(_T("stdafx.cpp")) == MCD_STR::npos

&& namePath.find(_T("StdAfx.cpp")) == MCD_STR::npos;

if (true == IsTargetSourceCode)

{

bool IsExclude = false;

while (xml.FindChildElem(_T("FileConfiguration")))

{

xml.IntoElem();

if (xml.GetAttrib(_T("Name")) == g_strFileConfigurationName)

{

if (xml.GetAttrib(_T("ExcludedFromBuild")) != _T("true"))

{

xml.SetAttrib(_T("Name"), g_strFileConfigurationName);

xml.SetAttrib(_T("ExcludedFromBuild"), _T("true"));

unityMaker::g_bIsSaveXML = true;

}

else

{

IsExclude = true;

}

}

xml.OutOfElem();

}

if (false == IsExclude)

{

xml.AddChildElem(_T("FileConfiguration"));

xml.IntoElem();

xml.AddAttrib(_T("Name"), g_strFileConfigurationName);

xml.AddAttrib(_T("ExcludedFromBuild"), _T("true"));

xml.AddChildElem(_T("Tool"));

{

xml.IntoElem();

xml.AddAttrib(_T("Name"), _T("VCCLCompilerTool"));

xml.OutOfElem();

}

xml.OutOfElem();

unityMaker::g_bIsSaveXML = true;

}

output_vecExcludedFromBuild.push_back(namePath);

}

xml.OutOfElem();

}

}

/**

@brief 빌드에서 제외된 .cpp 파일을 포함하는 통합 빌드 파일을 만든다.

*/

void MakeUnityBuildFile(MCD_STR strMakeFileName, std::vector<MCD_STR>& vecExcludedFromBuild)

{

if (true == strMakeFileName.empty())

{

return;

}

FILE* pFileStream = _tfopen(strMakeFileName.c_str(), _T("wb"));

#if defined(UNICODE)

TCHAR tag = 0xFEFF;

fwrite(&tag, sizeof(TCHAR), 1, pFileStream);

#endif

MCD_STR strPrecompiledHeader = _T("#include \"stdafx.h\"\r\n\r\n");

fwrite(strPrecompiledHeader.c_str(), sizeof(TCHAR), strPrecompiledHeader.length(), pFileStream);

for (std::vector<MCD_STR>::iterator iter_Source = vecExcludedFromBuild.begin();

iter_Source != vecExcludedFromBuild.end();

++iter_Source)

{

MCD_STR strSourceCode = _T("#include \"");

strSourceCode += *iter_Source;

strSourceCode += _T("\"\r\n");

fwrite(strSourceCode.c_str(), sizeof(TCHAR), strSourceCode.length(), pFileStream);

}

fclose(pFileStream);

}

/**

@brief 통합 빌드 파일의 이름이 포함된 풀경로를 얻는다.

@warning 생성되는 파일의 이름은 빌드에서 제외되는 필터의 이름에 "_" 추가해서 만든다.

*/

MCD_STR GetUnityBuildFileNamePathCombine(const MCD_STR& strPath, const MCD_STR& strExcludedNameFilter, const int iFileIndex)

{

TCHAR cBuffer[4] = {0,};

MCD_STR strUnityBuildFileName = strExcludedNameFilter;

strUnityBuildFileName += _T("_");

strUnityBuildFileName += _itot(iFileIndex, cBuffer, 10);

strUnityBuildFileName += _T(".cpp");

if (false == strPath.empty())

{

strUnityBuildFileName = strPath + _T("\\") + strUnityBuildFileName;

}

return strUnityBuildFileName;

}

}