|
|
#include <bits/stdc++.h>
|
|
|
#include <io.h>
|
|
|
|
|
|
using namespace std;
|
|
|
|
|
|
/**
|
|
|
* 功能：获取指定目录下所有的文件
|
|
|
* 作者：黄海
|
|
|
* 时间：2019-12-04
|
|
|
* @param cate_dir
|
|
|
* @return
|
|
|
*/
|
|
|
void getFiles(const string &path, vector<std::string> &files) {
|
|
|
//文件句柄
|
|
|
intptr_t hFile = 0;
|
|
|
//文件信息，_finddata_t需要io.h头文件
|
|
|
struct _finddata_t fileinfo;
|
|
|
std::string p;
|
|
|
if ((hFile = _findfirst(p.assign(path).append("\\*").c_str(), &fileinfo)) != -1) {
|
|
|
do {
|
|
|
//如果是目录,迭代之
|
|
|
//如果不是,加入列表
|
|
|
if ((fileinfo.attrib & _A_SUBDIR)) {
|
|
|
if (strcmp(fileinfo.name, ".") != 0 && strcmp(fileinfo.name, "..") != 0)
|
|
|
getFiles(p.assign(path).append("\\").append(fileinfo.name), files);
|
|
|
} else {
|
|
|
files.push_back(p.assign(path).append("\\").append(fileinfo.name));
|
|
|
}
|
|
|
} while (_findnext(hFile, &fileinfo) == 0);
|
|
|
_findclose(hFile);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
int startsWith(string s, string sub) {
|
|
|
return s.find(sub) == 0 ? 1 : 0;
|
|
|
}
|
|
|
|
|
|
int endsWith(string s, string sub) {
|
|
|
return s.rfind(sub) == (s.length() - sub.length()) ? 1 : 0;
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
* 功能：替换指定字符串
|
|
|
* 作者：黄海
|
|
|
* 时间：2019-11-21
|
|
|
* @param str
|
|
|
* @param old_value
|
|
|
* @param new_value
|
|
|
* @return
|
|
|
*/
|
|
|
string &replace_all(string &str, const string &old_value, const string &new_value) {
|
|
|
while (true) {
|
|
|
string::size_type pos(0);
|
|
|
if ((pos = str.find(old_value)) != string::npos)
|
|
|
str.replace(pos, old_value.length(), new_value);
|
|
|
else break;
|
|
|
}
|
|
|
return str;
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
* 功能：将字符串进行分隔的工具函数
|
|
|
* 作者：黄海
|
|
|
* 时间：2019-12-04
|
|
|
* @param s
|
|
|
* @param sv
|
|
|
* @param flag
|
|
|
*/
|
|
|
void split(const string &s, vector<string> &sv, const char flag = ' ') {
|
|
|
sv.clear();
|
|
|
istringstream iss(s);
|
|
|
string temp;
|
|
|
|
|
|
while (getline(iss, temp, flag)) {
|
|
|
sv.push_back(temp);
|
|
|
}
|
|
|
return;
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
* 功能：获取当前exe执行文件的上一级目录
|
|
|
* 作者：黄海
|
|
|
* 时间：2019-12-04
|
|
|
* @return
|
|
|
*/
|
|
|
string getCurrentParentPath() {
|
|
|
//当前目录
|
|
|
char currentPath[256];
|
|
|
_getcwd(currentPath, sizeof(currentPath));
|
|
|
vector<string> pathVector;
|
|
|
split(string(currentPath), pathVector, '\\');
|
|
|
string parentPath = "";
|
|
|
for (int i = 0; i < pathVector.size() - 1; i++) {
|
|
|
parentPath += pathVector[i] + "\\";
|
|
|
}
|
|
|
//去掉最后的分隔符
|
|
|
parentPath = parentPath.substr(0, parentPath.size() - 1);
|
|
|
return parentPath;
|
|
|
}
|
|
|
|
|
|
string inFile = "../../CMakeLists.config";
|
|
|
string configFile = "../CMakeLists.txt";
|
|
|
|
|
|
|
|
|
int main() {
|
|
|
//获取当前目录的上一级目录
|
|
|
string path = getCurrentParentPath();
|
|
|
|
|
|
//中文？
|
|
|
locale loc = locale::global(locale(locale(), "", LC_CTYPE));
|
|
|
//输入文件
|
|
|
ifstream in;
|
|
|
in.open(inFile);
|
|
|
|
|
|
//输出文件
|
|
|
ofstream out;
|
|
|
out.open(configFile);
|
|
|
|
|
|
//遍历
|
|
|
string line;
|
|
|
while (getline(in, line)) {
|
|
|
string str = line;
|
|
|
out << str << endl;
|
|
|
}
|
|
|
|
|
|
//开始处理目录下的所有CPP文件
|
|
|
vector<string> filenames;//用来存储文件名
|
|
|
getFiles(path, filenames);
|
|
|
|
|
|
int count = 0;
|
|
|
for (auto file : filenames) {
|
|
|
if (endsWith(file, ".cpp") && !endsWith(file, "CMakeCXXCompilerId.cpp")) {
|
|
|
string key = replace_all(file, path, "").substr(1);
|
|
|
key = replace_all(key, "\\", "/");
|
|
|
//获取程序名称
|
|
|
vector<string> sv;
|
|
|
split(key, sv, '/');
|
|
|
string fName;
|
|
|
if (sv.size() > 1)fName = sv[1]; else fName = sv[0];
|
|
|
fName = replace_all(fName, ".cpp", "");
|
|
|
out << "add_executable(" + fName + " " + key + ")" << endl;
|
|
|
count++;
|
|
|
}
|
|
|
}
|
|
|
//关闭文件
|
|
|
out.close();
|
|
|
in.close();
|
|
|
|
|
|
cout << "恭喜，配置文件修改成功完成，成功添加.cpp文件共" << count << "个！" << endl;
|
|
|
return 0;
|
|
|
} |
