Repository navigation
Expand file tree
/
Copy pathcpp2rust_lib.cpp
More file actions
87 lines (77 loc) · 3.25 KB
/
Copy pathcpp2rust_lib.cpp
File metadata and controls
87 lines (77 loc) · 3.25 KB
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
// Copyright (c) 2022-present INESC-ID.
// Distributed under the MIT license that can be found in the LICENSE file.
#include "cpp2rust_lib.h"
#include <clang/Tooling/ArgumentsAdjusters.h>
#include <clang/Tooling/CompilationDatabase.h>
#include <clang/Tooling/Tooling.h>
#include <filesystem>
#include "compat/platform_flags.h"
#include "converter/converter.h"
#include "converter/models/converter_refcount.h"
#include "frontend_action.h"
namespace cpp2rust {
std::string TranspileSrc(std::string_view cc_code, Model model,
const std::vector<std::string_view> &cxx_flags,
const std::string &rules_dir,
std::string_view filename) {
auto tool_args = getPlatformClangBeginFlags();
tool_args.push_back("-fparse-all-comments");
tool_args.insert(tool_args.end(), cxx_flags.begin(), cxx_flags.end());
auto end_flags = getPlatformClangEndFlags();
tool_args.insert(tool_args.end(), end_flags.begin(), end_flags.end());
std::string rs_code;
clang::tooling::runToolOnCodeWithArgs(
std::make_unique<FrontendAction>(rs_code, model, /*first=*/true,
rules_dir),
cc_code, tool_args, std::filesystem::path(filename).filename().string(),
filename.ends_with(".c") ? CLANG_C_COMPILER : CLANG_CXX_COMPILER);
Converter::EmitOpaqueRecords(rs_code);
Converter::EmitVirtualMethods(rs_code);
if (model == Model::kRefCount) {
ConverterRefCount::EmitMethodsOnPtr(rs_code);
}
Converter::EmitGlobalInits(model, rs_code);
return rs_code;
}
std::string TranspileDir(std::string_view build_dir, Model model,
const std::string &rules_dir) {
std::string error_message;
auto compile_dbase = clang::tooling::CompilationDatabase::loadFromDirectory(
build_dir, error_message);
if (!compile_dbase) {
return {};
}
std::vector<std::string> files;
for (const auto &compile_command : compile_dbase->getAllCompileCommands()) {
files.emplace_back(compile_command.Filename);
}
clang::tooling::ClangTool Tool(*compile_dbase, files);
Tool.appendArgumentsAdjuster(clang::tooling::getInsertArgumentAdjuster(
getPlatformClangBeginFlags(),
clang::tooling::ArgumentInsertPosition::BEGIN));
Tool.appendArgumentsAdjuster(clang::tooling::getInsertArgumentAdjuster(
getPlatformClangEndFlags(), clang::tooling::ArgumentInsertPosition::END));
// Redefine __FILE__ to use just the basename, so the generated code
// doesn't contain system-specific absolute paths.
Tool.appendArgumentsAdjuster(
[](const clang::tooling::CommandLineArguments &args,
llvm::StringRef filename) {
auto result = args;
auto basename =
std::filesystem::path(filename.str()).filename().string();
result.push_back("-Wno-builtin-macro-redefined");
result.push_back("-D__FILE__=\"" + basename + "\"");
return result;
});
std::string rs_code;
FrontendActionFactory factory(rs_code, model, rules_dir);
Tool.run(&factory);
Converter::EmitOpaqueRecords(rs_code);
Converter::EmitVirtualMethods(rs_code);
if (model == Model::kRefCount) {
ConverterRefCount::EmitMethodsOnPtr(rs_code);
}
Converter::EmitGlobalInits(model, rs_code);
return rs_code;
}
} // namespace cpp2rust