llvm-6502/tools/opt/opt.cpp
Chris Lattner 84608e43b2 Added a stupid testcase for iterators.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@17 91177308-0d34-0410-b5e6-96231b3b80d8
2001-06-08 00:35:25 +00:00

119 lines
3.8 KiB
C++

//===------------------------------------------------------------------------===
// LLVM 'OPT' UTILITY
//
// This utility may be invoked in the following manner:
// opt --help - Output information about command line switches
// opt [options] -dce - Run a dead code elimination pass on input
// bytecodes
// opt [options] -constprop - Run a constant propogation pass on input
// bytecodes
// opt [options] -inline - Run a method inlining pass on input bytecodes
// opt [options] -strip - Strip symbol tables out of methods
// opt [options] -mstrip - Strip module & method symbol tables
//
// Optimizations may be specified an arbitrary number of times on the command
// line, they are run in the order specified.
//
// TODO: Add a -all option to keep applying all optimizations until the program
// stops permuting.
// TODO: Add a -h command line arg that prints all available optimizations
// TODO: Add a -q command line arg that quiets "XXX pass made modifications"
//
//===------------------------------------------------------------------------===
#include <iostream.h>
#include <fstream.h>
#include "llvm/Module.h"
#include "llvm/Bytecode/Reader.h"
#include "llvm/Bytecode/Writer.h"
#include "llvm/Tools/CommandLine.h"
#include "llvm/Opt/AllOpts.h"
#if 1 // Testcase, TODO: REMOVE
#include "llvm/CFG.h"
#include "llvm/Assembly/Writer.h"
#include "llvm/Method.h"
static bool DoPrintM(Method *M) {
df_iterator I = df_begin(M->getBasicBlocks().front(), false);
df_iterator E = df_end(M->getBasicBlocks().front());
unsigned i = 0;
for (; I != E; ++I, ++i) {
cerr << "Basic Block Visited #" << i << *I;
}
return false;
}
static bool DoPrint(Module *C) {
return ApplyOptToAllMethods(C, DoPrintM);
}
#endif
struct {
const string ArgName, Name;
bool (*OptPtr)(Module *C);
} OptTable[] = {
{ "-dce", "Dead Code Elimination", DoDeadCodeElimination },
{ "-constprop","Constant Propogation", DoConstantPropogation },
{ "-inline" ,"Method Inlining", DoMethodInlining },
{ "-strip" ,"Strip Symbols", DoSymbolStripping },
{ "-mstrip" ,"Strip Module Symbols", DoFullSymbolStripping },
{ "-print" ,"Test printing stuff", DoPrint },
};
int main(int argc, char **argv) {
ToolCommandLine Opts(argc, argv, false);
bool Quiet = false;
for (int i = 1; i < argc; i++) {
if (string(argv[i]) == string("--help")) {
cerr << argv[0] << " usage:\n"
<< " " << argv[0] << " --help - Print this usage information\n";
return 1;
} else if (string(argv[i]) == string("-q")) {
Quiet = true; argv[i] = 0;
}
}
ostream *Out = &cout; // Default to printing to stdout...
Module *C = ParseBytecodeFile(Opts.getInputFilename());
if (C == 0) {
cerr << "bytecode didn't read correctly.\n";
return 1;
}
for (int i = 1; i < argc; i++) {
if (argv[i] == 0) continue;
unsigned j;
for (j = 0; j < sizeof(OptTable)/sizeof(OptTable[0]); j++) {
if (string(argv[i]) == OptTable[j].ArgName) {
if (OptTable[j].OptPtr(C) && !Quiet)
cerr << OptTable[j].Name << " pass made modifications!\n";
break;
}
}
if (j == sizeof(OptTable)/sizeof(OptTable[0]))
cerr << "'" << argv[i] << "' argument unrecognized: ignored\n";
}
if (Opts.getOutputFilename() != "-") {
Out = new ofstream(Opts.getOutputFilename().c_str(),
(Opts.getForce() ? 0 : ios::noreplace)|ios::out);
if (!Out->good()) {
cerr << "Error opening " << Opts.getOutputFilename()
<< "!\n";
delete C;
return 1;
}
}
// Okay, we're done now... write out result...
WriteBytecodeToFile(C, *Out);
delete C;
if (Out != &cout) delete Out;
return 0;
}