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
UTF8ToFilePath(const std::string & s)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
FilePathToUTF8(const base::FilePath & path)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
RepoPathToPathName(const std::vector<int> & repo_path)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
TargetIndexToLetter(int target_index)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
RepoPathToTargetName(const std::vector<int> & repo_path,int target_index)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
RepoPathToFullTargetName(const std::vector<int> & repo_path,int target_index)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
WriteLevel(const std::vector<int> & repo_path,int spread,int max_depth,int targets_per_level,int files_per_target)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 base::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
main()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