summaryrefslogtreecommitdiffstats
path: root/tools/gn/generate_test_gn_data.cc
blob: cebf3a0491d462831909edbb404004cccc5325ea (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
// Copyright (c) 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <fstream>
#include <iostream>

#include "base/file_util.h"
#include "base/files/file_path.h"
#include "base/strings/string_number_conversions.h"
#include "base/strings/utf_string_conversions.h"
#include "build/build_config.h"

// Usage: just run in the directory where you want your test source root to be.

int files_written = 0;
int targets_written = 0;

base::FilePath UTF8ToFilePath(const std::string& s) {
#if defined(OS_WIN)
  return base::FilePath(base::UTF8ToWide(s));
#else
  return base::FilePath(s);
#endif
}

std::string FilePathToUTF8(const base::FilePath& path) {
#if defined(OS_WIN)
  return base::WideToUTF8(path.value());
#else
  return path.value();
#endif
}

base::FilePath RepoPathToPathName(const std::vector<int>& repo_path) {
  base::FilePath ret;
  for (size_t i = 0; i < repo_path.size(); i++) {
    ret = ret.Append(UTF8ToFilePath(base::IntToString(repo_path[i])));
  }
  return ret;
}

std::string TargetIndexToLetter(int target_index) {
  char ret[2];
  ret[0] = 'a' + target_index;
  ret[1] = 0;
  return ret;
}

std::string RepoPathToTargetName(const std::vector<int>& repo_path,
                                 int target_index) {
  std::string ret;
  for (size_t i = 0; i < repo_path.size(); i++) {
    if (i != 0)
      ret.push_back('_');
    ret.append(base::IntToString(repo_path[i]));
  }
  ret += TargetIndexToLetter(target_index);
  return ret;
}

std::string RepoPathToFullTargetName(const std::vector<int>& repo_path,
                                 int target_index) {
  std::string ret;
  for (size_t i = 0; i < repo_path.size(); i++) {
    ret.push_back('/');
    ret.append(base::IntToString(repo_path[i]));
  }

  ret += ":" + RepoPathToTargetName(repo_path, target_index);
  return ret;
}

void WriteLevel(const std::vector<int>& repo_path,
                int spread,
                int max_depth,
                int targets_per_level,
                int files_per_target) {
  base::FilePath dirname = RepoPathToPathName(repo_path);
  base::FilePath filename = dirname.AppendASCII("BUILD.gn");
  std::cout << "Writing " << FilePathToUTF8(filename) << "\n";

  // Don't keep the file open while recursing.
  {
    base::CreateDirectory(dirname);

    std::ofstream file;
    file.open(FilePathToUTF8(filename).c_str(),
              std::ios_base::out | std::ios_base::binary);
    files_written++;

    for (int i = 0; i < targets_per_level; i++) {
      targets_written++;
      file << "executable(\"" << RepoPathToTargetName(repo_path, i)
           << "\") {\n";
      file << "  sources = [\n";
      for (int f = 0; f < files_per_target; f++)
        file << "    \"" << base::IntToString(f) << ".cc\",\n";

      if (repo_path.size() < (size_t)max_depth) {
        file << "  ]\n";
        file << "  deps = [\n";
        for (int d = 0; d < spread; d++) {
          std::vector<int> cur = repo_path;
          cur.push_back(d);
          for (int t = 0; t < targets_per_level; t++)
            file << "    \"" << RepoPathToFullTargetName(cur, t) << "\",\n";
        }
      }
      file << "  ]\n}\n\n";
    }
  }
  if (repo_path.size() < (size_t)max_depth) {
    // Recursively generate subdirs.
    for (int i = 0; i < spread; i++) {
      std::vector<int> cur = repo_path;
      cur.push_back(i);
      WriteLevel(cur, spread, max_depth, targets_per_level, files_per_target);
    }
  }
}

int main() {
  WriteLevel(std::vector<int>(), 5, 4, 3, 50);  // 781 files, 2343 targets
  //WriteLevel(std::vector<int>(), 6, 4, 2, 50);
  std::cout << "Wrote " << files_written << " files and "
            << targets_written << " targets.\n";
  return 0;
}