01652d889c
This commit adds a new PDLL specific LSP command, pdll.viewOutput, that allows for viewing the intermediate outputs of a given PDLL file. The available intermediate forms currently mirror those in mlir-pdll, namely: AST, MLIR, CPP. This is extremely useful for a developer of PDLL, as it simplifies various testing, and is also quite useful for users as they can easily view what is actually being generated for their PDLL files. This new command is added to the vscode client, and is available in the right client context menu of PDLL files, or via the vscode command palette. Differential Revision: https://reviews.llvm.org/D124783
119 lines
3.9 KiB
C++
119 lines
3.9 KiB
C++
//===- CompilationDatabase.cpp - LSP Compilation Database -----------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "CompilationDatabase.h"
|
|
#include "../lsp-server-support/Logging.h"
|
|
#include "Protocol.h"
|
|
#include "mlir/Support/FileUtilities.h"
|
|
#include "llvm/ADT/SetVector.h"
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/Support/YAMLTraits.h"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::lsp;
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// YamlFileInfo
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
namespace {
|
|
struct YamlFileInfo {
|
|
/// The absolute path to the file.
|
|
std::string filename;
|
|
/// The include directories available for the file.
|
|
std::vector<std::string> includeDirs;
|
|
};
|
|
} // namespace
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// CompilationDatabase
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
LLVM_YAML_IS_DOCUMENT_LIST_VECTOR(YamlFileInfo)
|
|
|
|
namespace llvm {
|
|
namespace yaml {
|
|
template <>
|
|
struct MappingTraits<YamlFileInfo> {
|
|
static void mapping(IO &io, YamlFileInfo &info) {
|
|
// Parse the filename and normalize it to the form we will expect from
|
|
// incoming URIs.
|
|
io.mapRequired("filepath", info.filename);
|
|
|
|
// Normalize the filename to avoid incompatability with incoming URIs.
|
|
if (Expected<lsp::URIForFile> uri =
|
|
lsp::URIForFile::fromFile(info.filename))
|
|
info.filename = uri->file().str();
|
|
|
|
// Parse the includes from the yaml stream. These are in the form of a
|
|
// semi-colon delimited list.
|
|
std::string combinedIncludes;
|
|
io.mapRequired("includes", combinedIncludes);
|
|
for (StringRef include : llvm::split(combinedIncludes, ";")) {
|
|
if (!include.empty())
|
|
info.includeDirs.push_back(include.str());
|
|
}
|
|
}
|
|
};
|
|
} // end namespace yaml
|
|
} // end namespace llvm
|
|
|
|
CompilationDatabase::CompilationDatabase(ArrayRef<std::string> databases) {
|
|
for (StringRef filename : databases)
|
|
loadDatabase(filename);
|
|
}
|
|
|
|
const CompilationDatabase::FileInfo &
|
|
CompilationDatabase::getFileInfo(StringRef filename) const {
|
|
auto it = files.find(filename);
|
|
return it == files.end() ? defaultFileInfo : it->second;
|
|
}
|
|
|
|
void CompilationDatabase::loadDatabase(StringRef filename) {
|
|
if (filename.empty())
|
|
return;
|
|
|
|
// Set up the input file.
|
|
std::string errorMessage;
|
|
std::unique_ptr<llvm::MemoryBuffer> inputFile =
|
|
openInputFile(filename, &errorMessage);
|
|
if (!inputFile) {
|
|
Logger::error("Failed to open compilation database: {0}", errorMessage);
|
|
return;
|
|
}
|
|
llvm::yaml::Input yaml(inputFile->getBuffer());
|
|
|
|
// Parse the yaml description and add any new files to the database.
|
|
std::vector<YamlFileInfo> parsedFiles;
|
|
yaml >> parsedFiles;
|
|
|
|
SetVector<StringRef> knownIncludes;
|
|
for (auto &file : parsedFiles) {
|
|
auto it = files.try_emplace(file.filename, std::move(file.includeDirs));
|
|
|
|
// If we encounter a duplicate file, log a warning and ignore it.
|
|
if (!it.second) {
|
|
Logger::info("Duplicate file in compilation database: {0}",
|
|
file.filename);
|
|
continue;
|
|
}
|
|
|
|
// Track the includes for the file.
|
|
for (StringRef include : it.first->second.includeDirs)
|
|
knownIncludes.insert(include);
|
|
}
|
|
|
|
// Add all of the known includes to the default file info. We don't know any
|
|
// information about how to treat these files, but these may be project files
|
|
// that we just don't yet have information for. In these cases, providing some
|
|
// heuristic information provides a better user experience, and generally
|
|
// shouldn't lead to any negative side effects.
|
|
for (StringRef include : knownIncludes)
|
|
defaultFileInfo.includeDirs.push_back(include.str());
|
|
}
|