X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=projects%2FStacker%2Ftools%2Fstkrc%2Fstkrc.cpp;h=727b044c20fdc4a68458f9d0b2249c00265f7e15;hb=40bb451c94a00502358546214d6d00a7c523c31d;hp=6006bb476e2ab6eb0c5bcff033926aaa0c02679d;hpb=551ccae044b0ff658fe629dd67edd5ffe75d10e8;p=oota-llvm.git diff --git a/projects/Stacker/tools/stkrc/stkrc.cpp b/projects/Stacker/tools/stkrc/stkrc.cpp index 6006bb476e2..727b044c20f 100644 --- a/projects/Stacker/tools/stkrc/stkrc.cpp +++ b/projects/Stacker/tools/stkrc/stkrc.cpp @@ -1,11 +1,11 @@ //===--- stkrc.cpp --- The Stacker Compiler -------------------------------===// -// +// // The LLVM Compiler Infrastructure // -// This file was developed by Reid Spencer and donated to the LLVM research -// group and is distributed under the University of Illinois Open Source +// This file was developed by Reid Spencer and donated to the LLVM research +// group and is distributed under the University of Illinois Open Source // License. See LICENSE.TXT for details. -// +// //===----------------------------------------------------------------------===// // // This is the "main" program for the Stacker Compiler. It is simply a utility @@ -14,7 +14,7 @@ // To get help using this utility, you can invoke it with: // stkrc --help - Output information about command line switches // -// +// //===------------------------------------------------------------------------=== #include "../../lib/compiler/StackerCompiler.h" @@ -29,24 +29,24 @@ using namespace llvm; -static cl::opt +static cl::opt InputFilename(cl::Positional, cl::desc(""), cl::init("-")); static cl::opt -OutputFilename("o", cl::desc("Override output filename"), - cl::value_desc("filename")); +OutputFilename("o", cl::desc("Override output filename"), + cl::value_desc("filename")); static cl::opt Force("f", cl::desc("Overwrite output files")); static cl::opt StackSize("s", cl::desc("Specify program maximum stack size"), - cl::init(1024), cl::value_desc("stack size")); + cl::init(1024), cl::value_desc("stack size")); static cl::opt DumpAsm("d", cl::desc("Print LLVM Assembly as parsed"), cl::Hidden); -#ifdef PARSE_DEBUG +#ifdef PARSE_DEBUG static cl::opt ParseDebug("g", cl::desc("Turn on Bison Debugging"), cl::Hidden); #endif @@ -59,94 +59,118 @@ FlexDebug("x", cl::desc("Turn on Flex Debugging"), cl::Hidden); static cl::opt EchoSource("e", cl::desc("Print Stacker Source as parsed"), cl::Hidden); -int main(int argc, char **argv) +enum OptLev { + None = 0, + One = 1, + Two = 2, + Three = 3, + Four = 4, + Five = 5 +}; + +static cl::opt OptLevel( + cl::desc("Choose optimization level to apply:"), + cl::init(One), + cl::values( + clEnumValN(None,"O0","An alias for the -O1 option"), + clEnumValN(One,"O1","Optimize for compilation speed"), + clEnumValN(Two,"O2","Perform simple optimizations to reduce code size"), + clEnumValN(Three,"O3","More aggressive optimizations"), + clEnumValN(Four,"O4","High level of optimization"), + clEnumValN(Five,"O5","An alias for the -O4 option"), + clEnumValEnd + )); + +int main(int argc, char **argv) { cl::ParseCommandLineOptions(argc, argv, " stacker .st -> .bc compiler\n"); std::ostream *Out = 0; - StackerCompiler compiler; - try - { -#ifdef PARSE_DEBUG + try { + StackerCompiler compiler; + try { - extern int Stackerdebug; - Stackerdebug = ParseDebug; - } +#ifdef PARSE_DEBUG + { + extern int Stackerdebug; + Stackerdebug = ParseDebug; + } #endif #ifdef FLEX_DEBUG - { - extern int Stacker_flex_debug; - Stacker_flex_debug = FlexDebug; - } + { + extern int Stacker_flex_debug; + Stacker_flex_debug = FlexDebug; + } #endif - // Parse the file now... - - std::auto_ptr M ( - compiler.compile(InputFilename,EchoSource, StackSize) ); - if (M.get() == 0) { - std::cerr << argv[0] << ": assembly didn't read correctly.\n"; - return 1; - } + // Parse the file now... - if (verifyModule(*M.get())) { - std::cerr << argv[0] - << ": assembly parsed, but does not verify as correct!\n"; - return 1; - } - - if (DumpAsm) std::cerr << "Here's the assembly:\n" << M.get(); - - if (OutputFilename != "") { // Specified an output filename? - if (OutputFilename != "-") { // Not stdout? - if (!Force && std::ifstream(OutputFilename.c_str())) { - // If force is not specified, make sure not to overwrite a file! - std::cerr << argv[0] << ": error opening '" << OutputFilename - << "': file exists!\n" - << "Use -f command line argument to force output\n"; - return 1; - } - Out = new std::ofstream(OutputFilename.c_str()); - } else { // Specified stdout - Out = &std::cout; + std::auto_ptr M ( + compiler.compile(InputFilename,EchoSource,OptLevel,StackSize)); + if (M.get() == 0) { + throw std::string("program didn't parse correctly."); + } + + if (verifyModule(*M.get())) { + throw std::string("program parsed, but does not verify as correct!"); } - } else { - if (InputFilename == "-") { - OutputFilename = "-"; - Out = &std::cout; + + if (DumpAsm) + std::cerr << "Here's the assembly:" << M.get(); + + if (OutputFilename != "") { // Specified an output filename? + if (OutputFilename != "-") { // Not stdout? + if (!Force && std::ifstream(OutputFilename.c_str())) { + // If force is not specified, make sure not to overwrite a file! + throw std::string("error opening '") + OutputFilename + + "': file exists!\n" + + "Use -f command line argument to force output"; + return 1; + } + Out = new std::ofstream(OutputFilename.c_str()); + } else { // Specified stdout + Out = &std::cout; + } } else { - std::string IFN = InputFilename; - int Len = IFN.length(); - if (IFN[Len-3] == '.' && IFN[Len-2] == 's' && IFN[Len-1] == 't') { - // Source ends in .ll - OutputFilename = std::string(IFN.begin(), IFN.end()-3); + if (InputFilename == "-") { + OutputFilename = "-"; + Out = &std::cout; } else { - OutputFilename = IFN; // Append a .bc to it - } - OutputFilename += ".bc"; - - if (!Force && std::ifstream(OutputFilename.c_str())) { - // If force is not specified, make sure not to overwrite a file! - std::cerr << argv[0] << ": error opening '" << OutputFilename - << "': file exists!\n" - << "Use -f command line argument to force output\n"; - return 1; + std::string IFN = InputFilename; + int Len = IFN.length(); + if (IFN[Len-3] == '.' && IFN[Len-2] == 's' && IFN[Len-1] == 't') { + // Source ends in .ll + OutputFilename = std::string(IFN.begin(), IFN.end()-3); + } else { + OutputFilename = IFN; // Append a .bc to it + } + OutputFilename += ".bc"; + + if (!Force && std::ifstream(OutputFilename.c_str())) { + // If force is not specified, make sure not to overwrite a file! + throw std::string("error opening '") + OutputFilename + + "': file exists!\n" + + "Use -f command line argument to force output\n"; + } + + Out = new std::ofstream(OutputFilename.c_str()); + // Make sure that the Out file gets unlinked from the disk if we get a + // SIGINT + sys::RemoveFileOnSignal(sys::Path(OutputFilename)); } + } - Out = new std::ofstream(OutputFilename.c_str()); - // Make sure that the Out file gets unlinked from the disk if we get a - // SIGINT - sys::RemoveFileOnSignal(OutputFilename); + if (!Out->good()) { + throw std::string("error opening ") + OutputFilename + "!"; } - } - - if (!Out->good()) { - std::cerr << argv[0] << ": error opening " << OutputFilename << "!\n"; + + WriteBytecodeToFile(M.get(), *Out); + } catch (const ParseError &E) { + std::cerr << argv[0] << ": " << E.getMessage() << "\n"; return 1; } - - WriteBytecodeToFile(M.get(), *Out); - } catch (const ParseException &E) { - std::cerr << argv[0] << ": " << E.getMessage() << "\n"; + } + catch (const std::string& msg ) { + std::cerr << argv[0] << ": " << msg << "\n"; return 1; }