1*9880d681SAndroid Build Coastguard Worker //===-- llvm-split: command line tool for testing module splitter ---------===//
2*9880d681SAndroid Build Coastguard Worker //
3*9880d681SAndroid Build Coastguard Worker // The LLVM Compiler Infrastructure
4*9880d681SAndroid Build Coastguard Worker //
5*9880d681SAndroid Build Coastguard Worker // This file is distributed under the University of Illinois Open Source
6*9880d681SAndroid Build Coastguard Worker // License. See LICENSE.TXT for details.
7*9880d681SAndroid Build Coastguard Worker //
8*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
9*9880d681SAndroid Build Coastguard Worker //
10*9880d681SAndroid Build Coastguard Worker // This program can be used to test the llvm::SplitModule function.
11*9880d681SAndroid Build Coastguard Worker //
12*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
13*9880d681SAndroid Build Coastguard Worker
14*9880d681SAndroid Build Coastguard Worker #include "llvm/ADT/StringExtras.h"
15*9880d681SAndroid Build Coastguard Worker #include "llvm/Bitcode/ReaderWriter.h"
16*9880d681SAndroid Build Coastguard Worker #include "llvm/IR/LLVMContext.h"
17*9880d681SAndroid Build Coastguard Worker #include "llvm/IR/Verifier.h"
18*9880d681SAndroid Build Coastguard Worker #include "llvm/IRReader/IRReader.h"
19*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/CommandLine.h"
20*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/FileSystem.h"
21*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/SourceMgr.h"
22*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/ToolOutputFile.h"
23*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/raw_ostream.h"
24*9880d681SAndroid Build Coastguard Worker #include "llvm/Transforms/Utils/SplitModule.h"
25*9880d681SAndroid Build Coastguard Worker
26*9880d681SAndroid Build Coastguard Worker using namespace llvm;
27*9880d681SAndroid Build Coastguard Worker
28*9880d681SAndroid Build Coastguard Worker static cl::opt<std::string>
29*9880d681SAndroid Build Coastguard Worker InputFilename(cl::Positional, cl::desc("<input bitcode file>"),
30*9880d681SAndroid Build Coastguard Worker cl::init("-"), cl::value_desc("filename"));
31*9880d681SAndroid Build Coastguard Worker
32*9880d681SAndroid Build Coastguard Worker static cl::opt<std::string>
33*9880d681SAndroid Build Coastguard Worker OutputFilename("o", cl::desc("Override output filename"),
34*9880d681SAndroid Build Coastguard Worker cl::value_desc("filename"));
35*9880d681SAndroid Build Coastguard Worker
36*9880d681SAndroid Build Coastguard Worker static cl::opt<unsigned> NumOutputs("j", cl::Prefix, cl::init(2),
37*9880d681SAndroid Build Coastguard Worker cl::desc("Number of output files"));
38*9880d681SAndroid Build Coastguard Worker
39*9880d681SAndroid Build Coastguard Worker static cl::opt<bool>
40*9880d681SAndroid Build Coastguard Worker PreserveLocals("preserve-locals", cl::Prefix, cl::init(false),
41*9880d681SAndroid Build Coastguard Worker cl::desc("Split without externalizing locals"));
42*9880d681SAndroid Build Coastguard Worker
main(int argc,char ** argv)43*9880d681SAndroid Build Coastguard Worker int main(int argc, char **argv) {
44*9880d681SAndroid Build Coastguard Worker LLVMContext Context;
45*9880d681SAndroid Build Coastguard Worker SMDiagnostic Err;
46*9880d681SAndroid Build Coastguard Worker cl::ParseCommandLineOptions(argc, argv, "LLVM module splitter\n");
47*9880d681SAndroid Build Coastguard Worker
48*9880d681SAndroid Build Coastguard Worker std::unique_ptr<Module> M = parseIRFile(InputFilename, Err, Context);
49*9880d681SAndroid Build Coastguard Worker
50*9880d681SAndroid Build Coastguard Worker if (!M) {
51*9880d681SAndroid Build Coastguard Worker Err.print(argv[0], errs());
52*9880d681SAndroid Build Coastguard Worker return 1;
53*9880d681SAndroid Build Coastguard Worker }
54*9880d681SAndroid Build Coastguard Worker
55*9880d681SAndroid Build Coastguard Worker unsigned I = 0;
56*9880d681SAndroid Build Coastguard Worker SplitModule(std::move(M), NumOutputs, [&](std::unique_ptr<Module> MPart) {
57*9880d681SAndroid Build Coastguard Worker std::error_code EC;
58*9880d681SAndroid Build Coastguard Worker std::unique_ptr<tool_output_file> Out(new tool_output_file(
59*9880d681SAndroid Build Coastguard Worker OutputFilename + utostr(I++), EC, sys::fs::F_None));
60*9880d681SAndroid Build Coastguard Worker if (EC) {
61*9880d681SAndroid Build Coastguard Worker errs() << EC.message() << '\n';
62*9880d681SAndroid Build Coastguard Worker exit(1);
63*9880d681SAndroid Build Coastguard Worker }
64*9880d681SAndroid Build Coastguard Worker
65*9880d681SAndroid Build Coastguard Worker verifyModule(*MPart);
66*9880d681SAndroid Build Coastguard Worker WriteBitcodeToFile(MPart.get(), Out->os());
67*9880d681SAndroid Build Coastguard Worker
68*9880d681SAndroid Build Coastguard Worker // Declare success.
69*9880d681SAndroid Build Coastguard Worker Out->keep();
70*9880d681SAndroid Build Coastguard Worker }, PreserveLocals);
71*9880d681SAndroid Build Coastguard Worker
72*9880d681SAndroid Build Coastguard Worker return 0;
73*9880d681SAndroid Build Coastguard Worker }
74