|
150
|
1 //===- GuessTargetAndModeCompilationDatabase.cpp --------------------------===//
|
|
|
2 //
|
|
|
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
|
4 // See https://llvm.org/LICENSE.txt for license information.
|
|
|
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
|
6 //
|
|
|
7 //===----------------------------------------------------------------------===//
|
|
|
8
|
|
|
9 #include "clang/Tooling/CompilationDatabase.h"
|
|
|
10 #include "clang/Tooling/Tooling.h"
|
|
|
11 #include <memory>
|
|
|
12
|
|
|
13 namespace clang {
|
|
|
14 namespace tooling {
|
|
|
15
|
|
|
16 namespace {
|
|
|
17 class TargetAndModeAdderDatabase : public CompilationDatabase {
|
|
|
18 public:
|
|
|
19 TargetAndModeAdderDatabase(std::unique_ptr<CompilationDatabase> Base)
|
|
|
20 : Base(std::move(Base)) {
|
|
|
21 assert(this->Base != nullptr);
|
|
|
22 }
|
|
|
23
|
|
|
24 std::vector<std::string> getAllFiles() const override {
|
|
|
25 return Base->getAllFiles();
|
|
|
26 }
|
|
|
27
|
|
|
28 std::vector<CompileCommand> getAllCompileCommands() const override {
|
|
|
29 return addTargetAndMode(Base->getAllCompileCommands());
|
|
|
30 }
|
|
|
31
|
|
|
32 std::vector<CompileCommand>
|
|
|
33 getCompileCommands(StringRef FilePath) const override {
|
|
|
34 return addTargetAndMode(Base->getCompileCommands(FilePath));
|
|
|
35 }
|
|
|
36
|
|
|
37 private:
|
|
|
38 std::vector<CompileCommand>
|
|
|
39 addTargetAndMode(std::vector<CompileCommand> Cmds) const {
|
|
|
40 for (auto &Cmd : Cmds) {
|
|
|
41 if (Cmd.CommandLine.empty())
|
|
|
42 continue;
|
|
|
43 addTargetAndModeForProgramName(Cmd.CommandLine, Cmd.CommandLine.front());
|
|
|
44 }
|
|
|
45 return Cmds;
|
|
|
46 }
|
|
|
47 std::unique_ptr<CompilationDatabase> Base;
|
|
|
48 };
|
|
|
49 } // namespace
|
|
|
50
|
|
|
51 std::unique_ptr<CompilationDatabase>
|
|
|
52 inferTargetAndDriverMode(std::unique_ptr<CompilationDatabase> Base) {
|
|
|
53 return std::make_unique<TargetAndModeAdderDatabase>(std::move(Base));
|
|
|
54 }
|
|
|
55
|
|
|
56 } // namespace tooling
|
|
|
57 } // namespace clang
|
