1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include <fstream> 6 #include <iostream> 7 8 #include "base/file_util.h" 9 #include "base/files/file_path.h" 10 #include "base/strings/string_number_conversions.h" 11 #include "base/strings/utf_string_conversions.h" 12 #include "build/build_config.h" 13 14 // Usage: just run in the directory where you want your test source root to be. 15 16 int files_written = 0; 17 int targets_written = 0; 18 19 base::FilePath UTF8ToFilePath(const std::string& s) { 20 #if defined(OS_WIN) 21 return base::FilePath(UTF8ToWide(s)); 22 #else 23 return base::FilePath(s); 24 #endif 25 } 26 27 std::string FilePathToUTF8(const base::FilePath& path) { 28 #if defined(OS_WIN) 29 return WideToUTF8(path.value()); 30 #else 31 return path.value(); 32 #endif 33 } 34 35 base::FilePath RepoPathToPathName(const std::vector<int>& repo_path) { 36 base::FilePath ret; 37 for (size_t i = 0; i < repo_path.size(); i++) { 38 ret = ret.Append(UTF8ToFilePath(base::IntToString(repo_path[i]))); 39 } 40 return ret; 41 } 42 43 std::string TargetIndexToLetter(int target_index) { 44 char ret[2]; 45 ret[0] = 'a' + target_index; 46 ret[1] = 0; 47 return ret; 48 } 49 50 std::string RepoPathToTargetName(const std::vector<int>& repo_path, 51 int target_index) { 52 std::string ret; 53 for (size_t i = 0; i < repo_path.size(); i++) { 54 if (i != 0) 55 ret.push_back('_'); 56 ret.append(base::IntToString(repo_path[i])); 57 } 58 ret += TargetIndexToLetter(target_index); 59 return ret; 60 } 61 62 std::string RepoPathToFullTargetName(const std::vector<int>& repo_path, 63 int target_index) { 64 std::string ret; 65 for (size_t i = 0; i < repo_path.size(); i++) { 66 ret.push_back('/'); 67 ret.append(base::IntToString(repo_path[i])); 68 } 69 70 ret += ":" + RepoPathToTargetName(repo_path, target_index); 71 return ret; 72 } 73 74 void WriteLevel(const std::vector<int>& repo_path, 75 int spread, 76 int max_depth, 77 int targets_per_level, 78 int files_per_target) { 79 base::FilePath dirname = RepoPathToPathName(repo_path); 80 base::FilePath filename = dirname.AppendASCII("BUILD.gn"); 81 std::cout << "Writing " << FilePathToUTF8(filename) << "\n"; 82 83 // Don't keep the file open while recursing. 84 { 85 file_util::CreateDirectory(dirname); 86 87 std::ofstream file; 88 file.open(FilePathToUTF8(filename).c_str(), 89 std::ios_base::out | std::ios_base::binary); 90 files_written++; 91 92 for (int i = 0; i < targets_per_level; i++) { 93 targets_written++; 94 file << "executable(\"" << RepoPathToTargetName(repo_path, i) 95 << "\") {\n"; 96 file << " sources = [\n"; 97 for (int f = 0; f < files_per_target; f++) 98 file << " \"" << base::IntToString(f) << ".cc\",\n"; 99 100 if (repo_path.size() < (size_t)max_depth) { 101 file << " ]\n"; 102 file << " deps = [\n"; 103 for (int d = 0; d < spread; d++) { 104 std::vector<int> cur = repo_path; 105 cur.push_back(d); 106 for (int t = 0; t < targets_per_level; t++) 107 file << " \"" << RepoPathToFullTargetName(cur, t) << "\",\n"; 108 } 109 } 110 file << " ]\n}\n\n"; 111 } 112 } 113 if (repo_path.size() < (size_t)max_depth) { 114 // Recursively generate subdirs. 115 for (int i = 0; i < spread; i++) { 116 std::vector<int> cur = repo_path; 117 cur.push_back(i); 118 WriteLevel(cur, spread, max_depth, targets_per_level, files_per_target); 119 } 120 } 121 } 122 123 int main() { 124 WriteLevel(std::vector<int>(), 5, 4, 3, 50); // 781 files, 2343 targets 125 //WriteLevel(std::vector<int>(), 6, 4, 2, 50); 126 std::cout << "Wrote " << files_written << " files and " 127 << targets_written << " targets.\n"; 128 return 0; 129 } 130
