由于申请软件著作权的时候,需要统计代码的行数,然后自己就写了一个,分享给大家。

// 获取文件的行数
//filepath:文件路径
//ruleData:一些配置信息,bCountNullLine:是否统计空行
int getFileLine(string filePath, const RuleData& ruleData){
	int nLines = 0;
	ifstream ReadFile;
	int n = 0;
	string tmp;
	ReadFile.open(filePath, ios::in);//ios::in 表示以只读的方式读取文件
	if (ReadFile.fail())//文件打开失败:返回0
		return 0;
	else//文件存在
	{
		while (getline(ReadFile, tmp))
		{
			if (!ruleData.bCountNullLine && tmp == "")
				continue;	//不统计空行

			n++;
		}
	}
	ReadFile.close();
	return n;
}

// 获取文件夹下的所有文件行数
//dirPath:总文件夹的路径
//secDirPath:次文件夹路径
//mapFileExt:文件后缀列表
//ruleData:一些配置信息
int countDirTotalLines(const string& dirPath, string secDirPath, const std::map<string, bool>& mapFileExt, const RuleData& ruleData){
	int nTotalLines = 0;

	long hFile = 0;
	struct _finddata_t fileInfo;
	std::string pathName, exdName;
	string folderPath = dirPath + secDirPath;
	if ((hFile = _findfirst(pathName.assign(folderPath).append("\\*").c_str(), &fileInfo)) == -1) {
		return nTotalLines;
	}
	do {
		const std::string filename = fileInfo.name;
		if (filename == "." || filename == "..")
			continue;

		string secDirPath2 = secDirPath + "/" + filename;
		if (fileInfo.attrib&_A_SUBDIR)	//文件夹-
		{
			int nLines = countDirTotalLines(dirPath, secDirPath2, mapFileExt, ruleData);
			nTotalLines += nLines;
			printf("%s总行数:%d\n", secDirPath2.c_str(), nLines);
		}
		else{
			const std::string& filepath = folderPath + "/" + filename;
			string fileExt = getFileExt(filename);
			if (mapFileExt.count(fileExt) > 0)
			{
				int nLines = getFileLine(filepath, ruleData);
				nTotalLines += nLines;
				printf("%s/%s行数:%d\n", secDirPath2.c_str(), filename.c_str(), nLines);
			}
		}

	} while (_findnext(hFile, &fileInfo) == 0);
	_findclose(hFile);
	return nTotalLines;
}