Chris Lattner | 2148bcb | 2003-09-10 19:42:51 +0000 | [diff] [blame] | 1 | //===- extract.cpp - LLVM function extraction utility ---------------------===// |
John Criswell | 7c0e022 | 2003-10-20 17:47:21 +0000 | [diff] [blame^] | 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file was developed by the LLVM research group and is distributed under |
| 6 | // the University of Illinois Open Source License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
Chris Lattner | 579d914 | 2002-05-22 20:27:00 +0000 | [diff] [blame] | 10 | // |
| 11 | // This utility changes the input module to only contain a single function, |
| 12 | // which is primarily used for debugging transformations. |
| 13 | // |
| 14 | //===----------------------------------------------------------------------===// |
| 15 | |
| 16 | #include "llvm/Module.h" |
| 17 | #include "llvm/PassManager.h" |
| 18 | #include "llvm/Bytecode/Reader.h" |
| 19 | #include "llvm/Bytecode/WriteBytecodePass.h" |
Chris Lattner | 33974ca | 2002-07-23 22:04:40 +0000 | [diff] [blame] | 20 | #include "llvm/Transforms/IPO.h" |
Chris Lattner | 579d914 | 2002-05-22 20:27:00 +0000 | [diff] [blame] | 21 | #include "Support/CommandLine.h" |
| 22 | #include <memory> |
| 23 | |
Chris Lattner | 5ff62e9 | 2002-07-22 02:10:13 +0000 | [diff] [blame] | 24 | // InputFilename - The filename to read from. |
Chris Lattner | c7a0985 | 2002-07-25 16:31:09 +0000 | [diff] [blame] | 25 | static cl::opt<std::string> |
Chris Lattner | 5ff62e9 | 2002-07-22 02:10:13 +0000 | [diff] [blame] | 26 | InputFilename(cl::Positional, cl::desc("<input bytecode file>"), |
| 27 | cl::init("-"), cl::value_desc("filename")); |
| 28 | |
| 29 | |
| 30 | // ExtractFunc - The function to extract from the module... defaults to main. |
Chris Lattner | c7a0985 | 2002-07-25 16:31:09 +0000 | [diff] [blame] | 31 | static cl::opt<std::string> |
Chris Lattner | 5ff62e9 | 2002-07-22 02:10:13 +0000 | [diff] [blame] | 32 | ExtractFunc("func", cl::desc("Specify function to extract"), cl::init("main"), |
| 33 | cl::value_desc("function")); |
| 34 | |
Chris Lattner | 579d914 | 2002-05-22 20:27:00 +0000 | [diff] [blame] | 35 | int main(int argc, char **argv) { |
| 36 | cl::ParseCommandLineOptions(argc, argv, " llvm extractor\n"); |
| 37 | |
| 38 | std::auto_ptr<Module> M(ParseBytecodeFile(InputFilename)); |
| 39 | if (M.get() == 0) { |
Chris Lattner | 50e3a20 | 2002-07-30 21:43:22 +0000 | [diff] [blame] | 40 | std::cerr << argv[0] << ": bytecode didn't read correctly.\n"; |
Chris Lattner | 579d914 | 2002-05-22 20:27:00 +0000 | [diff] [blame] | 41 | return 1; |
| 42 | } |
| 43 | |
Chris Lattner | 5113eb0 | 2002-11-19 18:42:59 +0000 | [diff] [blame] | 44 | // Figure out which function we should extract |
| 45 | Function *F = M.get()->getNamedFunction(ExtractFunc); |
| 46 | if (F == 0) { |
| 47 | std::cerr << argv[0] << ": program doesn't contain function named '" |
| 48 | << ExtractFunc << "'!\n"; |
| 49 | return 1; |
| 50 | } |
| 51 | |
Chris Lattner | 579d914 | 2002-05-22 20:27:00 +0000 | [diff] [blame] | 52 | // In addition to just parsing the input from GCC, we also want to spiff it up |
| 53 | // a little bit. Do this now. |
| 54 | // |
| 55 | PassManager Passes; |
Chris Lattner | 5113eb0 | 2002-11-19 18:42:59 +0000 | [diff] [blame] | 56 | Passes.add(createFunctionExtractionPass(F)); // Extract the function |
Chris Lattner | 579d914 | 2002-05-22 20:27:00 +0000 | [diff] [blame] | 57 | Passes.add(createGlobalDCEPass()); // Delete unreachable globals |
Chris Lattner | 68ed318 | 2002-10-12 20:50:16 +0000 | [diff] [blame] | 58 | Passes.add(createFunctionResolvingPass()); // Delete prototypes |
Chris Lattner | 33974ca | 2002-07-23 22:04:40 +0000 | [diff] [blame] | 59 | Passes.add(createDeadTypeEliminationPass()); // Remove dead types... |
Chris Lattner | 579d914 | 2002-05-22 20:27:00 +0000 | [diff] [blame] | 60 | Passes.add(new WriteBytecodePass(&std::cout)); // Write bytecode to file... |
| 61 | |
Chris Lattner | 7e70829 | 2002-06-25 16:13:24 +0000 | [diff] [blame] | 62 | Passes.run(*M.get()); |
Chris Lattner | 579d914 | 2002-05-22 20:27:00 +0000 | [diff] [blame] | 63 | return 0; |
| 64 | } |