f66432899acfa22d480d0e5846c79336d16470f9
[oota-llvm.git] / tools / gccas / gccas.cpp
1 //===----------------------------------------------------------------------===//
2 // LLVM 'GCCAS' UTILITY 
3 //
4 //  This utility is designed to be used by the GCC frontend for creating
5 // bytecode files from it's intermediate llvm assembly.  The requirements for
6 // this utility are thus slightly different than that of the standard as util.
7 //
8 //===----------------------------------------------------------------------===//
9
10 #include "llvm/Module.h"
11 #include "llvm/PassManager.h"
12 #include "llvm/Assembly/Parser.h"
13 #include "llvm/Transforms/RaisePointerReferences.h"
14 #include "llvm/Transforms/IPO.h"
15 #include "llvm/Transforms/Scalar.h"
16 #include "llvm/Analysis/LoadValueNumbering.h"
17 #include "llvm/Analysis/Verifier.h"
18 #include "llvm/Bytecode/WriteBytecodePass.h"
19 #include "llvm/Target/TargetData.h"
20 #include "Support/CommandLine.h"
21 #include "Support/Signals.h"
22 #include <memory>
23 #include <fstream>
24 using std::cerr;
25
26 // FIXME: This should eventually be parameterized...
27 static TargetData TD("opt target");
28
29 static cl::opt<std::string>
30 InputFilename(cl::Positional, cl::desc("<input llvm assembly>"), cl::Required);
31
32 static cl::opt<std::string> 
33 OutputFilename("o", cl::desc("Override output filename"),
34                cl::value_desc("filename"));
35
36 static cl::opt<int>
37 RunNPasses("stopAfterNPasses",
38            cl::desc("Only run the first N passes of gccas"), cl::Hidden,
39            cl::value_desc("# passes"));
40
41 static cl::opt<bool> 
42 StopAtLevelRaise("stopraise", cl::desc("Stop optimization before level raise"),
43                  cl::Hidden);
44
45 static cl::opt<bool>   
46 Verify("verify", cl::desc("Verify each pass result"));
47
48
49 static inline void addPass(PassManager &PM, Pass *P) {
50   static int NumPassesCreated = 0;
51   
52   // If we haven't already created the number of passes that was requested...
53   if (RunNPasses == 0 || RunNPasses > NumPassesCreated) {
54     // Add the pass to the pass manager...
55     PM.add(P);
56
57     // If we are verifying all of the intermediate steps, add the verifier...
58     if (Verify) PM.add(createVerifierPass());
59
60     // Keep track of how many passes we made for -stopAfterNPasses
61     ++NumPassesCreated;
62   } else {
63     delete P;             // We don't want this pass to run, just delete it now
64   }
65 }
66
67
68 void AddConfiguredTransformationPasses(PassManager &PM) {
69   if (Verify) PM.add(createVerifierPass());
70
71   addPass(PM, createFunctionResolvingPass());    // Resolve (...) functions
72   addPass(PM, createGlobalDCEPass());            // Kill unused uinit g-vars
73   addPass(PM, createDeadTypeEliminationPass());  // Eliminate dead types
74   addPass(PM, createConstantMergePass());        // Merge dup global constants
75   addPass(PM, createVerifierPass());             // Verify that input is correct
76   addPass(PM, createDeadInstEliminationPass());  // Remove Dead code/vars
77   addPass(PM, createRaiseAllocationsPass());     // call %malloc -> malloc inst
78   addPass(PM, createIndVarSimplifyPass());       // Simplify indvars
79
80   // Level raise is eternally buggy/in need of enhancements.  Allow
81   // transformation to stop right before it runs.
82   if (StopAtLevelRaise) return;
83
84   addPass(PM, createRaisePointerReferencesPass(TD));// Recover type information
85   addPass(PM, createPromoteMemoryToRegister());  // Promote alloca's to regs
86   // Disabling until this is fixed -- Vikram, 7/7/02.
87   // addPass(PM, createReassociatePass());          // Reassociate expressions
88   addPass(PM, createInstructionCombiningPass()); // Combine silly seq's
89   addPass(PM, createCorrelatedExpressionEliminationPass());
90   addPass(PM, createInstructionCombiningPass()); // Combine silly seq's
91   addPass(PM, createCFGSimplificationPass());    // Merge & remove BBs
92   addPass(PM, createLICMPass());                 // Hoist loop invariants
93   addPass(PM, createLoadValueNumberingPass());   // GVN for load instructions
94   addPass(PM, createGCSEPass());                 // Remove common subexprs
95   addPass(PM, createSCCPPass());                 // Constant prop with SCCP
96
97   // Run instcombine after redundancy elimination to exploit opportunities
98   // opened up by them.
99   addPass(PM, createInstructionCombiningPass());
100   addPass(PM, createAggressiveDCEPass());        // SSA based 'Agressive DCE'
101   addPass(PM, createCFGSimplificationPass());    // Merge & remove BBs
102 }
103
104
105 int main(int argc, char **argv) {
106   cl::ParseCommandLineOptions(argc, argv, " llvm .s -> .o assembler for GCC\n");
107
108   std::auto_ptr<Module> M;
109   try {
110     // Parse the file now...
111     M.reset(ParseAssemblyFile(InputFilename));
112   } catch (const ParseException &E) {
113     cerr << argv[0] << ": " << E.getMessage() << "\n";
114     return 1;
115   }
116
117   if (M.get() == 0) {
118     cerr << argv[0] << ": assembly didn't read correctly.\n";
119     return 1;
120   }
121   
122   if (OutputFilename == "") {   // Didn't specify an output filename?
123     std::string IFN = InputFilename;
124     int Len = IFN.length();
125     if (IFN[Len-2] == '.' && IFN[Len-1] == 's') {   // Source ends in .s?
126       OutputFilename = std::string(IFN.begin(), IFN.end()-2);
127     } else {
128       OutputFilename = IFN;   // Append a .o to it
129     }
130     OutputFilename += ".o";
131   }
132
133   std::ofstream Out(OutputFilename.c_str(), std::ios::out);
134   if (!Out.good()) {
135     cerr << argv[0] << ": error opening " << OutputFilename << "!\n";
136     return 1;
137   }
138
139   // Make sure that the Out file gets unlink'd from the disk if we get a SIGINT
140   RemoveFileOnSignal(OutputFilename);
141
142   // In addition to just parsing the input from GCC, we also want to spiff it up
143   // a little bit.  Do this now.
144   //
145   PassManager Passes;
146
147   // Add all of the transformation passes to the pass manager to do the cleanup
148   // and optimization of the GCC output.
149   //
150   AddConfiguredTransformationPasses(Passes);
151
152   // Write bytecode to file...
153   Passes.add(new WriteBytecodePass(&Out));
154
155   // Run our queue of passes all at once now, efficiently.
156   Passes.run(*M.get());
157   return 0;
158 }