blob: f199365dc8153de3b9558ac577effc49d633eb53 [file] [log] [blame]
//===--- AllTUsExecution.h - Execute actions on all TUs. -*- C++ --------*-===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
// This file defines a tool executor that runs given actions on all TUs in the
// compilation database. Tool results are deuplicated by the result key.
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_CLANG_TOOLING_ALLTUSEXECUTION_H
#define LLVM_CLANG_TOOLING_ALLTUSEXECUTION_H
#include "clang/Tooling/ArgumentsAdjusters.h"
#include "clang/Tooling/Execution.h"
namespace clang {
namespace tooling {
/// Executes given frontend actions on all files/TUs in the compilation
/// database.
class AllTUsToolExecutor : public ToolExecutor {
public:
static const char *ExecutorName;
/// Init with \p CompilationDatabase.
/// This uses \p ThreadCount threads to exececute the actions on all files in
/// parallel. If \p ThreadCount is 0, this uses `llvm::hardware_concurrency`.
AllTUsToolExecutor(const CompilationDatabase &Compilations,
unsigned ThreadCount,
std::shared_ptr<PCHContainerOperations> PCHContainerOps =
std::make_shared<PCHContainerOperations>());
/// Init with \p CommonOptionsParser. This is expected to be used by
/// `createExecutorFromCommandLineArgs` based on commandline options.
///
/// The executor takes ownership of \p Options.
AllTUsToolExecutor(CommonOptionsParser Options, unsigned ThreadCount,
std::shared_ptr<PCHContainerOperations> PCHContainerOps =
std::make_shared<PCHContainerOperations>());
StringRef getExecutorName() const override { return ExecutorName; }
using ToolExecutor::execute;
llvm::Error
execute(llvm::ArrayRef<
std::pair<std::unique_ptr<FrontendActionFactory>, ArgumentsAdjuster>>
Actions) override;
ExecutionContext *getExecutionContext() override { return &Context; };
ToolResults *getToolResults() override { return Results.get(); }
void mapVirtualFile(StringRef FilePath, StringRef Content) override {
OverlayFiles[FilePath] = Content;
}
private:
// Used to store the parser when the executor is initialized with parser.
llvm::Optional<CommonOptionsParser> OptionsParser;
const CompilationDatabase &Compilations;
std::unique_ptr<ToolResults> Results;
ExecutionContext Context;
llvm::StringMap<std::string> OverlayFiles;
unsigned ThreadCount;
};
} // end namespace tooling
} // end namespace clang
#endif // LLVM_CLANG_TOOLING_ALLTUSEXECUTION_H