setting path to prcontext.tcl script. Right now it searches for tclsh in your path...
[oota-llvm.git] / tools / llc / llc.cpp
index 71667c34a3ccd2fe336c2ed2ac4b3794af0fc671..4e1e0cce7bb03c873873a3aba09fb00ae3ffb318 100644 (file)
@@ -1,20 +1,33 @@
 //===-- llc.cpp - Implement the LLVM Native Code Generator ----------------===//
+// 
+//                     The LLVM Compiler Infrastructure
 //
-// This is the llc code generator.
+// This file was developed by the LLVM research group and is distributed under
+// the University of Illinois Open Source License. See LICENSE.TXT for details.
+// 
+//===----------------------------------------------------------------------===//
+//
+// This is the llc code generator driver. It provides a convenient
+// command-line interface for generating native assembly-language code 
+// or C code, given LLVM bytecode.
 //
 //===----------------------------------------------------------------------===//
 
 #include "llvm/Bytecode/Reader.h"
-#include "llvm/Target/TargetMachineImpls.h"
 #include "llvm/Target/TargetMachine.h"
+#include "llvm/Target/TargetMachineRegistry.h"
 #include "llvm/Transforms/Scalar.h"
 #include "llvm/Module.h"
 #include "llvm/PassManager.h"
 #include "llvm/Pass.h"
-#include "Support/CommandLine.h"
-#include "Support/Signals.h"
-#include <memory>
+#include "llvm/Support/CommandLine.h"
+#include "llvm/Support/PluginLoader.h"
+#include "llvm/System/Signals.h"
 #include <fstream>
+#include <iostream>
+#include <memory>
+
+using namespace llvm;
 
 // General options for llc.  Other pass-specific options are specified
 // within the corresponding llc passes, and target-specific options
@@ -28,30 +41,17 @@ OutputFilename("o", cl::desc("Output filename"), cl::value_desc("filename"));
 
 static cl::opt<bool> Force("f", cl::desc("Overwrite output files"));
 
-enum ArchName { noarch, x86, Sparc };
-
-static cl::opt<ArchName>
-Arch("march", cl::desc("Architecture to generate assembly for:"), cl::Prefix,
-     cl::values(clEnumVal(x86, "  IA-32 (Pentium and above)"),
-                clEnumValN(Sparc, "sparc", "  SPARC V9"),
-               0),
-#if defined(i386) || defined(__i386__) || defined(__x86__)
-     cl::init(x86)
-#elif defined(sparc) || defined(__sparc__) || defined(__sparcv9)
-     cl::init(Sparc)
-#else
-     cl::init(noarch)
-#endif
-     );
-
+static cl::opt<const TargetMachineRegistry::Entry*, false, TargetNameParser>
+MArch("march", cl::desc("Architecture to generate assembly for:"));
+               
 // GetFileNameRoot - Helper function to get the basename of a filename...
 static inline std::string
-GetFileNameRoot(const std::string &InputFilename)
-{
+GetFileNameRoot(const std::string &InputFilename) {
   std::string IFN = InputFilename;
   std::string outputFilename;
   int Len = IFN.length();
-  if (IFN[Len-3] == '.' && IFN[Len-2] == 'b' && IFN[Len-1] == 'c') {
+  if ((Len > 2) &&
+      IFN[Len-3] == '.' && IFN[Len-2] == 'b' && IFN[Len-1] == 'c') {
     outputFilename = std::string(IFN.begin(), IFN.end()-3); // s/.bc/.s/
   } else {
     outputFilename = IFN;
@@ -64,7 +64,8 @@ GetFileNameRoot(const std::string &InputFilename)
 //
 int main(int argc, char **argv) {
   cl::ParseCommandLineOptions(argc, argv, " llvm system compiler\n");
-  
+  sys::PrintStackTraceOnErrorSignal();
+
   // Load the module to be compiled...
   std::auto_ptr<Module> M(ParseBytecodeFile(InputFilename));
   if (M.get() == 0) {
@@ -75,38 +76,26 @@ int main(int argc, char **argv) {
 
   // Allocate target machine.  First, check whether the user has
   // explicitly specified an architecture to compile for.
-  unsigned Config = (mod.isLittleEndian()   ? TM::LittleEndian : TM::BigEndian)|
-                    (mod.has32BitPointers() ? TM::PtrSize32    : TM::PtrSize64);
-  TargetMachine* (*TargetMachineAllocator)(unsigned) = 0;
-  switch (Arch) {
-  case x86:
-    TargetMachineAllocator = allocateX86TargetMachine;
-    break;
-  case Sparc:
-    TargetMachineAllocator = allocateSparcTargetMachine;
-    break;
-  default:
-    // Decide what the default target machine should be, by looking at
-    // the module. This heuristic (ILP32, LE -> IA32; LP64, BE ->
-    // SPARCV9) is kind of gross, but it will work until we have more
-    // sophisticated target information to work from.
-    if (mod.isLittleEndian() && mod.has32BitPointers()) { 
-      TargetMachineAllocator = allocateX86TargetMachine;
-    } else if (mod.isBigEndian() && mod.has64BitPointers()) {
-      TargetMachineAllocator = allocateSparcTargetMachine;
-    } else {
-      assert(0 && "You must specify -march; I could not guess the default");
-    } 
-    break;
+  TargetMachine* (*TargetMachineAllocator)(const Module&,
+                                           IntrinsicLowering *) = 0;
+  if (MArch == 0) {
+    std::string Err;
+    MArch = TargetMachineRegistry::getClosestStaticTargetForModule(mod, Err);
+    if (MArch == 0) {
+      std::cerr << argv[0] << ": error auto-selecting target for module '"
+                << Err << "'.  Please use the -march option to explicitly "
+                << "pick a target.\n";
+      return 1;
+    }
   }
-  std::auto_ptr<TargetMachine> target((*TargetMachineAllocator)(Config));
+
+  std::auto_ptr<TargetMachine> target(MArch->CtorFn(mod, 0));
   assert(target.get() && "Could not allocate target machine!");
   TargetMachine &Target = *target.get();
   const TargetData &TD = Target.getTargetData();
 
   // Build up all of the passes that we want to do to the module...
   PassManager Passes;
-
   Passes.add(new TargetData("llc", TD.isLittleEndian(), TD.getPointerSize(),
                             TD.getPointerAlignment(), TD.getDoubleAlignment()));
 
@@ -116,17 +105,17 @@ int main(int argc, char **argv) {
     if (OutputFilename != "-") {
       // Specified an output filename?
       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;
+        // 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());
 
-      // Make sure that the Out file gets unlink'd from the disk if we get a
+      // Make sure that the Out file gets unlinked from the disk if we get a
       // SIGINT
-      RemoveFileOnSignal(OutputFilename);
+      sys::RemoveFileOnSignal(OutputFilename);
     } else {
       Out = &std::cout;
     }
@@ -136,7 +125,11 @@ int main(int argc, char **argv) {
       Out = &std::cout;
     } else {
       OutputFilename = GetFileNameRoot(InputFilename); 
-      OutputFilename += ".s";
+
+      if (MArch->Name[0] != 'c' || MArch->Name[1] != 0)  // not CBE
+        OutputFilename += ".s";
+      else
+        OutputFilename += ".cbe.c";
       
       if (!Force && std::ifstream(OutputFilename.c_str())) {
         // If force is not specified, make sure not to overwrite a file!
@@ -153,9 +146,9 @@ int main(int argc, char **argv) {
         return 1;
       }
       
-      // Make sure that the Out file gets unlink'd from the disk if we get a
+      // Make sure that the Out file gets unlinked from the disk if we get a
       // SIGINT
-      RemoveFileOnSignal(OutputFilename);
+      sys::RemoveFileOnSignal(OutputFilename);
     }
   }