Changes For Bug 352
[oota-llvm.git] / lib / Target / CBackend / Writer.cpp
1 //===-- Writer.cpp - Library for converting LLVM code to C ----------------===//
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 //
10 // This library converts LLVM code to C code, compilable by GCC and other C
11 // compilers.
12 //
13 //===----------------------------------------------------------------------===//
14
15 #include "CTargetMachine.h"
16 #include "llvm/Constants.h"
17 #include "llvm/DerivedTypes.h"
18 #include "llvm/Module.h"
19 #include "llvm/Instructions.h"
20 #include "llvm/Pass.h"
21 #include "llvm/PassManager.h"
22 #include "llvm/SymbolTable.h"
23 #include "llvm/Intrinsics.h"
24 #include "llvm/Analysis/ConstantsScanner.h"
25 #include "llvm/Analysis/FindUsedTypes.h"
26 #include "llvm/Analysis/LoopInfo.h"
27 #include "llvm/CodeGen/IntrinsicLowering.h"
28 #include "llvm/Transforms/Scalar.h"
29 #include "llvm/Target/TargetMachineRegistry.h"
30 #include "llvm/Support/CallSite.h"
31 #include "llvm/Support/CFG.h"
32 #include "llvm/Support/GetElementPtrTypeIterator.h"
33 #include "llvm/Support/InstVisitor.h"
34 #include "llvm/Support/Mangler.h"
35 #include "llvm/ADT/StringExtras.h"
36 #include "llvm/Support/MathExtras.h"
37 #include "llvm/Config/config.h"
38 #include <algorithm>
39 #include <iostream>
40 #include <sstream>
41 using namespace llvm;
42
43 namespace {
44   // Register the target.
45   RegisterTarget<CTargetMachine> X("c", "  C backend");
46
47   /// NameAllUsedStructs - This pass inserts names for any unnamed structure
48   /// types that are used by the program.
49   ///
50   class CBackendNameAllUsedStructs : public Pass {
51     void getAnalysisUsage(AnalysisUsage &AU) const {
52       AU.addRequired<FindUsedTypes>();
53     }
54
55     virtual const char *getPassName() const {
56       return "C backend type canonicalizer";
57     }
58
59     virtual bool run(Module &M);
60   };
61   
62   /// CWriter - This class is the main chunk of code that converts an LLVM
63   /// module to a C translation unit.
64   class CWriter : public FunctionPass, public InstVisitor<CWriter> {
65     std::ostream &Out; 
66     IntrinsicLowering &IL;
67     Mangler *Mang;
68     LoopInfo *LI;
69     const Module *TheModule;
70     std::map<const Type *, std::string> TypeNames;
71
72     std::map<const ConstantFP *, unsigned> FPConstantMap;
73   public:
74     CWriter(std::ostream &o, IntrinsicLowering &il) : Out(o), IL(il) {}
75
76     virtual const char *getPassName() const { return "C backend"; }
77
78     void getAnalysisUsage(AnalysisUsage &AU) const {
79       AU.addRequired<LoopInfo>();
80       AU.setPreservesAll();
81     }
82
83     virtual bool doInitialization(Module &M);
84
85     bool runOnFunction(Function &F) {
86       LI = &getAnalysis<LoopInfo>();
87
88       // Output all floating point constants that cannot be printed accurately.
89       printFloatingPointConstants(F);
90   
91       lowerIntrinsics(F);
92       printFunction(F);
93       FPConstantMap.clear();
94       return false;
95     }
96
97     virtual bool doFinalization(Module &M) {
98       // Free memory...
99       delete Mang;
100       TypeNames.clear();
101       return false;
102     }
103
104     std::ostream &printType(std::ostream &Out, const Type *Ty,
105                             const std::string &VariableName = "",
106                             bool IgnoreName = false);
107
108     void writeOperand(Value *Operand);
109     void writeOperandInternal(Value *Operand);
110
111   private :
112     void lowerIntrinsics(Function &F);
113
114     bool nameAllUsedStructureTypes(Module &M);
115     void printModule(Module *M);
116     void printModuleTypes(const SymbolTable &ST);
117     void printContainedStructs(const Type *Ty, std::set<const StructType *> &);
118     void printFloatingPointConstants(Function &F);
119     void printFunctionSignature(const Function *F, bool Prototype);
120
121     void printFunction(Function &);
122     void printBasicBlock(BasicBlock *BB);
123     void printLoop(Loop *L);
124
125     void printConstant(Constant *CPV);
126     void printConstantArray(ConstantArray *CPA);
127
128     // isInlinableInst - Attempt to inline instructions into their uses to build
129     // trees as much as possible.  To do this, we have to consistently decide
130     // what is acceptable to inline, so that variable declarations don't get
131     // printed and an extra copy of the expr is not emitted.
132     //
133     static bool isInlinableInst(const Instruction &I) {
134       // Always inline setcc instructions, even if they are shared by multiple
135       // expressions.  GCC generates horrible code if we don't.
136       if (isa<SetCondInst>(I)) return true;
137
138       // Must be an expression, must be used exactly once.  If it is dead, we
139       // emit it inline where it would go.
140       if (I.getType() == Type::VoidTy || !I.hasOneUse() ||
141           isa<TerminatorInst>(I) || isa<CallInst>(I) || isa<PHINode>(I) || 
142           isa<LoadInst>(I) || isa<VAArgInst>(I) || isa<VANextInst>(I))
143         // Don't inline a load across a store or other bad things!
144         return false;
145
146       // Only inline instruction it it's use is in the same BB as the inst.
147       return I.getParent() == cast<Instruction>(I.use_back())->getParent();
148     }
149
150     // isDirectAlloca - Define fixed sized allocas in the entry block as direct
151     // variables which are accessed with the & operator.  This causes GCC to
152     // generate significantly better code than to emit alloca calls directly.
153     //
154     static const AllocaInst *isDirectAlloca(const Value *V) {
155       const AllocaInst *AI = dyn_cast<AllocaInst>(V);
156       if (!AI) return false;
157       if (AI->isArrayAllocation())
158         return 0;   // FIXME: we can also inline fixed size array allocas!
159       if (AI->getParent() != &AI->getParent()->getParent()->getEntryBlock())
160         return 0;
161       return AI;
162     }
163
164     // Instruction visitation functions
165     friend class InstVisitor<CWriter>;
166
167     void visitReturnInst(ReturnInst &I);
168     void visitBranchInst(BranchInst &I);
169     void visitSwitchInst(SwitchInst &I);
170     void visitInvokeInst(InvokeInst &I) {
171       assert(0 && "Lowerinvoke pass didn't work!");
172     }
173
174     void visitUnwindInst(UnwindInst &I) {
175       assert(0 && "Lowerinvoke pass didn't work!");
176     }
177
178     void visitPHINode(PHINode &I);
179     void visitBinaryOperator(Instruction &I);
180
181     void visitCastInst (CastInst &I);
182     void visitSelectInst(SelectInst &I);
183     void visitCallInst (CallInst &I);
184     void visitCallSite (CallSite CS);
185     void visitShiftInst(ShiftInst &I) { visitBinaryOperator(I); }
186
187     void visitMallocInst(MallocInst &I);
188     void visitAllocaInst(AllocaInst &I);
189     void visitFreeInst  (FreeInst   &I);
190     void visitLoadInst  (LoadInst   &I);
191     void visitStoreInst (StoreInst  &I);
192     void visitGetElementPtrInst(GetElementPtrInst &I);
193     void visitVANextInst(VANextInst &I);
194     void visitVAArgInst (VAArgInst &I);
195
196     void visitInstruction(Instruction &I) {
197       std::cerr << "C Writer does not know about " << I;
198       abort();
199     }
200
201     void outputLValue(Instruction *I) {
202       Out << "  " << Mang->getValueName(I) << " = ";
203     }
204
205     bool isGotoCodeNecessary(BasicBlock *From, BasicBlock *To);
206     void printPHICopiesForSuccessors(BasicBlock *CurBlock, 
207                                      unsigned Indent);
208     void printBranchToBlock(BasicBlock *CurBlock, BasicBlock *SuccBlock,
209                             unsigned Indent);
210     void printIndexingExpression(Value *Ptr, gep_type_iterator I,
211                                  gep_type_iterator E);
212   };
213 }
214
215 /// This method inserts names for any unnamed structure types that are used by
216 /// the program, and removes names from structure types that are not used by the
217 /// program.
218 ///
219 bool CBackendNameAllUsedStructs::run(Module &M) {
220   // Get a set of types that are used by the program...
221   std::set<const Type *> UT = getAnalysis<FindUsedTypes>().getTypes();
222   
223   // Loop over the module symbol table, removing types from UT that are
224   // already named, and removing names for structure types that are not used.
225   //
226   SymbolTable &MST = M.getSymbolTable();
227   for (SymbolTable::type_iterator TI = MST.type_begin(), TE = MST.type_end();
228        TI != TE; ) {
229     SymbolTable::type_iterator I = TI++;
230     if (const StructType *STy = dyn_cast<StructType>(I->second)) {
231       // If this is not used, remove it from the symbol table.
232       std::set<const Type *>::iterator UTI = UT.find(STy);
233       if (UTI == UT.end())
234         MST.remove(I);
235       else
236         UT.erase(UTI);
237     }
238   }
239
240   // UT now contains types that are not named.  Loop over it, naming
241   // structure types.
242   //
243   bool Changed = false;
244   unsigned RenameCounter = 0;
245   for (std::set<const Type *>::const_iterator I = UT.begin(), E = UT.end();
246        I != E; ++I)
247     if (const StructType *ST = dyn_cast<StructType>(*I)) {
248       while (M.addTypeName("unnamed"+utostr(RenameCounter), ST))
249         ++RenameCounter;
250       Changed = true;
251     }
252   return Changed;
253 }
254
255
256 // Pass the Type* and the variable name and this prints out the variable
257 // declaration.
258 //
259 std::ostream &CWriter::printType(std::ostream &Out, const Type *Ty,
260                                  const std::string &NameSoFar,
261                                  bool IgnoreName) {
262   if (Ty->isPrimitiveType())
263     switch (Ty->getTypeID()) {
264     case Type::VoidTyID:   return Out << "void "               << NameSoFar;
265     case Type::BoolTyID:   return Out << "bool "               << NameSoFar;
266     case Type::UByteTyID:  return Out << "unsigned char "      << NameSoFar;
267     case Type::SByteTyID:  return Out << "signed char "        << NameSoFar;
268     case Type::UShortTyID: return Out << "unsigned short "     << NameSoFar;
269     case Type::ShortTyID:  return Out << "short "              << NameSoFar;
270     case Type::UIntTyID:   return Out << "unsigned "           << NameSoFar;
271     case Type::IntTyID:    return Out << "int "                << NameSoFar;
272     case Type::ULongTyID:  return Out << "unsigned long long " << NameSoFar;
273     case Type::LongTyID:   return Out << "signed long long "   << NameSoFar;
274     case Type::FloatTyID:  return Out << "float "              << NameSoFar;
275     case Type::DoubleTyID: return Out << "double "             << NameSoFar;
276     default :
277       std::cerr << "Unknown primitive type: " << *Ty << "\n";
278       abort();
279     }
280   
281   // Check to see if the type is named.
282   if (!IgnoreName || isa<OpaqueType>(Ty)) {
283     std::map<const Type *, std::string>::iterator I = TypeNames.find(Ty);
284     if (I != TypeNames.end()) return Out << I->second << " " << NameSoFar;
285   }
286
287   switch (Ty->getTypeID()) {
288   case Type::FunctionTyID: {
289     const FunctionType *MTy = cast<FunctionType>(Ty);
290     std::stringstream FunctionInnards; 
291     FunctionInnards << " (" << NameSoFar << ") (";
292     for (FunctionType::param_iterator I = MTy->param_begin(),
293            E = MTy->param_end(); I != E; ++I) {
294       if (I != MTy->param_begin())
295         FunctionInnards << ", ";
296       printType(FunctionInnards, *I, "");
297     }
298     if (MTy->isVarArg()) {
299       if (MTy->getNumParams()) 
300         FunctionInnards << ", ...";
301     } else if (!MTy->getNumParams()) {
302       FunctionInnards << "void";
303     }
304     FunctionInnards << ")";
305     std::string tstr = FunctionInnards.str();
306     printType(Out, MTy->getReturnType(), tstr);
307     return Out;
308   }
309   case Type::StructTyID: {
310     const StructType *STy = cast<StructType>(Ty);
311     Out << NameSoFar + " {\n";
312     unsigned Idx = 0;
313     for (StructType::element_iterator I = STy->element_begin(),
314            E = STy->element_end(); I != E; ++I) {
315       Out << "  ";
316       printType(Out, *I, "field" + utostr(Idx++));
317       Out << ";\n";
318     }
319     return Out << "}";
320   }  
321
322   case Type::PointerTyID: {
323     const PointerType *PTy = cast<PointerType>(Ty);
324     std::string ptrName = "*" + NameSoFar;
325
326     if (isa<ArrayType>(PTy->getElementType()))
327       ptrName = "(" + ptrName + ")";
328
329     return printType(Out, PTy->getElementType(), ptrName);
330   }
331
332   case Type::ArrayTyID: {
333     const ArrayType *ATy = cast<ArrayType>(Ty);
334     unsigned NumElements = ATy->getNumElements();
335     return printType(Out, ATy->getElementType(),
336                      NameSoFar + "[" + utostr(NumElements) + "]");
337   }
338
339   case Type::OpaqueTyID: {
340     static int Count = 0;
341     std::string TyName = "struct opaque_" + itostr(Count++);
342     assert(TypeNames.find(Ty) == TypeNames.end());
343     TypeNames[Ty] = TyName;
344     return Out << TyName << " " << NameSoFar;
345   }
346   default:
347     assert(0 && "Unhandled case in getTypeProps!");
348     abort();
349   }
350
351   return Out;
352 }
353
354 void CWriter::printConstantArray(ConstantArray *CPA) {
355
356   // As a special case, print the array as a string if it is an array of
357   // ubytes or an array of sbytes with positive values.
358   // 
359   const Type *ETy = CPA->getType()->getElementType();
360   bool isString = (ETy == Type::SByteTy || ETy == Type::UByteTy);
361
362   // Make sure the last character is a null char, as automatically added by C
363   if (isString && (CPA->getNumOperands() == 0 ||
364                    !cast<Constant>(*(CPA->op_end()-1))->isNullValue()))
365     isString = false;
366   
367   if (isString) {
368     Out << "\"";
369     // Keep track of whether the last number was a hexadecimal escape
370     bool LastWasHex = false;
371
372     // Do not include the last character, which we know is null
373     for (unsigned i = 0, e = CPA->getNumOperands()-1; i != e; ++i) {
374       unsigned char C = cast<ConstantInt>(CPA->getOperand(i))->getRawValue();
375       
376       // Print it out literally if it is a printable character.  The only thing
377       // to be careful about is when the last letter output was a hex escape
378       // code, in which case we have to be careful not to print out hex digits
379       // explicitly (the C compiler thinks it is a continuation of the previous
380       // character, sheesh...)
381       //
382       if (isprint(C) && (!LastWasHex || !isxdigit(C))) {
383         LastWasHex = false;
384         if (C == '"' || C == '\\')
385           Out << "\\" << C;
386         else
387           Out << C;
388       } else {
389         LastWasHex = false;
390         switch (C) {
391         case '\n': Out << "\\n"; break;
392         case '\t': Out << "\\t"; break;
393         case '\r': Out << "\\r"; break;
394         case '\v': Out << "\\v"; break;
395         case '\a': Out << "\\a"; break;
396         case '\"': Out << "\\\""; break;
397         case '\'': Out << "\\\'"; break;           
398         default:
399           Out << "\\x";
400           Out << (char)(( C/16  < 10) ? ( C/16 +'0') : ( C/16 -10+'A'));
401           Out << (char)(((C&15) < 10) ? ((C&15)+'0') : ((C&15)-10+'A'));
402           LastWasHex = true;
403           break;
404         }
405       }
406     }
407     Out << "\"";
408   } else {
409     Out << "{";
410     if (CPA->getNumOperands()) {
411       Out << " ";
412       printConstant(cast<Constant>(CPA->getOperand(0)));
413       for (unsigned i = 1, e = CPA->getNumOperands(); i != e; ++i) {
414         Out << ", ";
415         printConstant(cast<Constant>(CPA->getOperand(i)));
416       }
417     }
418     Out << " }";
419   }
420 }
421
422 // isFPCSafeToPrint - Returns true if we may assume that CFP may be written out
423 // textually as a double (rather than as a reference to a stack-allocated
424 // variable). We decide this by converting CFP to a string and back into a
425 // double, and then checking whether the conversion results in a bit-equal
426 // double to the original value of CFP. This depends on us and the target C
427 // compiler agreeing on the conversion process (which is pretty likely since we
428 // only deal in IEEE FP).
429 //
430 static bool isFPCSafeToPrint(const ConstantFP *CFP) {
431 #if HAVE_PRINTF_A
432   char Buffer[100];
433   sprintf(Buffer, "%a", CFP->getValue());
434
435   if (!strncmp(Buffer, "0x", 2) ||
436       !strncmp(Buffer, "-0x", 3) ||
437       !strncmp(Buffer, "+0x", 3))
438     return atof(Buffer) == CFP->getValue();
439   return false;
440 #else
441   std::string StrVal = ftostr(CFP->getValue());
442
443   while (StrVal[0] == ' ')
444     StrVal.erase(StrVal.begin());
445
446   // Check to make sure that the stringized number is not some string like "Inf"
447   // or NaN.  Check that the string matches the "[-+]?[0-9]" regex.
448   if ((StrVal[0] >= '0' && StrVal[0] <= '9') ||
449       ((StrVal[0] == '-' || StrVal[0] == '+') &&
450        (StrVal[1] >= '0' && StrVal[1] <= '9')))
451     // Reparse stringized version!
452     return atof(StrVal.c_str()) == CFP->getValue();
453   return false;
454 #endif
455 }
456
457 // printConstant - The LLVM Constant to C Constant converter.
458 void CWriter::printConstant(Constant *CPV) {
459   if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CPV)) {
460     switch (CE->getOpcode()) {
461     case Instruction::Cast:
462       Out << "((";
463       printType(Out, CPV->getType());
464       Out << ")";
465       printConstant(CE->getOperand(0));
466       Out << ")";
467       return;
468
469     case Instruction::GetElementPtr:
470       Out << "(&(";
471       printIndexingExpression(CE->getOperand(0), gep_type_begin(CPV),
472                               gep_type_end(CPV));
473       Out << "))";
474       return;
475     case Instruction::Select:
476       Out << "(";
477       printConstant(CE->getOperand(0));
478       Out << "?";
479       printConstant(CE->getOperand(1));
480       Out << ":";
481       printConstant(CE->getOperand(2));
482       Out << ")";
483       return;
484     case Instruction::Add:
485     case Instruction::Sub:
486     case Instruction::Mul:
487     case Instruction::Div:
488     case Instruction::Rem:
489     case Instruction::SetEQ:
490     case Instruction::SetNE:
491     case Instruction::SetLT:
492     case Instruction::SetLE:
493     case Instruction::SetGT:
494     case Instruction::SetGE:
495     case Instruction::Shl:
496     case Instruction::Shr:
497       Out << "(";
498       printConstant(CE->getOperand(0));
499       switch (CE->getOpcode()) {
500       case Instruction::Add: Out << " + "; break;
501       case Instruction::Sub: Out << " - "; break;
502       case Instruction::Mul: Out << " * "; break;
503       case Instruction::Div: Out << " / "; break;
504       case Instruction::Rem: Out << " % "; break;
505       case Instruction::SetEQ: Out << " == "; break;
506       case Instruction::SetNE: Out << " != "; break;
507       case Instruction::SetLT: Out << " < "; break;
508       case Instruction::SetLE: Out << " <= "; break;
509       case Instruction::SetGT: Out << " > "; break;
510       case Instruction::SetGE: Out << " >= "; break;
511       case Instruction::Shl: Out << " << "; break;
512       case Instruction::Shr: Out << " >> "; break;
513       default: assert(0 && "Illegal opcode here!");
514       }
515       printConstant(CE->getOperand(1));
516       Out << ")";
517       return;
518
519     default:
520       std::cerr << "CWriter Error: Unhandled constant expression: "
521                 << *CE << "\n";
522       abort();
523     }
524   }
525
526   switch (CPV->getType()->getTypeID()) {
527   case Type::BoolTyID:
528     Out << (CPV == ConstantBool::False ? "0" : "1"); break;
529   case Type::SByteTyID:
530   case Type::ShortTyID:
531     Out << cast<ConstantSInt>(CPV)->getValue(); break;
532   case Type::IntTyID:
533     if ((int)cast<ConstantSInt>(CPV)->getValue() == (int)0x80000000)
534       Out << "((int)0x80000000)";   // Handle MININT specially to avoid warning
535     else
536       Out << cast<ConstantSInt>(CPV)->getValue();
537     break;
538
539   case Type::LongTyID:
540     Out << cast<ConstantSInt>(CPV)->getValue() << "ll"; break;
541
542   case Type::UByteTyID:
543   case Type::UShortTyID:
544     Out << cast<ConstantUInt>(CPV)->getValue(); break;
545   case Type::UIntTyID:
546     Out << cast<ConstantUInt>(CPV)->getValue() << "u"; break;
547   case Type::ULongTyID:
548     Out << cast<ConstantUInt>(CPV)->getValue() << "ull"; break;
549
550   case Type::FloatTyID:
551   case Type::DoubleTyID: {
552     ConstantFP *FPC = cast<ConstantFP>(CPV);
553     std::map<const ConstantFP*, unsigned>::iterator I = FPConstantMap.find(FPC);
554     if (I != FPConstantMap.end()) {
555       // Because of FP precision problems we must load from a stack allocated
556       // value that holds the value in hex.
557       Out << "(*(" << (FPC->getType() == Type::FloatTy ? "float" : "double")
558           << "*)&FPConstant" << I->second << ")";
559     } else {
560       if (IsNAN(FPC->getValue())) {
561         // The value is NaN
562  
563         // The prefix for a quiet NaN is 0x7FF8. For a signalling NaN,
564         // it's 0x7ff4.
565         const unsigned long QuietNaN = 0x7ff8UL;
566         const unsigned long SignalNaN = 0x7ff4UL;
567
568         // We need to grab the first part of the FP #
569         union {
570           double   d;
571           uint64_t ll;
572         } DHex;
573         char Buffer[100];
574
575         DHex.d = FPC->getValue();
576         sprintf(Buffer, "0x%llx", DHex.ll);
577
578         std::string Num(&Buffer[0], &Buffer[6]);
579         unsigned long Val = strtoul(Num.c_str(), 0, 16);
580
581         if (FPC->getType() == Type::FloatTy)
582           Out << "LLVM_NAN" << (Val == QuietNaN ? "" : "S") << "F(\""
583               << Buffer << "\") /*nan*/ ";
584         else
585           Out << "LLVM_NAN" << (Val == QuietNaN ? "" : "S") << "(\""
586               << Buffer << "\") /*nan*/ ";
587       } else if (IsInf(FPC->getValue())) {
588         // The value is Inf
589         if (FPC->getValue() < 0) Out << "-";
590         Out << "LLVM_INF" << (FPC->getType() == Type::FloatTy ? "F" : "")
591             << " /*inf*/ ";
592       } else {
593         std::string Num;
594 #if HAVE_PRINTF_A
595         // Print out the constant as a floating point number.
596         char Buffer[100];
597         sprintf(Buffer, "%a", FPC->getValue());
598         Num = Buffer;
599 #else
600         Num = ftostr(FPC->getValue());
601 #endif
602         Out << Num;
603       }
604     }
605     break;
606   }
607
608   case Type::ArrayTyID:
609     if (isa<ConstantAggregateZero>(CPV)) {
610       const ArrayType *AT = cast<ArrayType>(CPV->getType());
611       Out << "{";
612       if (AT->getNumElements()) {
613         Out << " ";
614         Constant *CZ = Constant::getNullValue(AT->getElementType());
615         printConstant(CZ);
616         for (unsigned i = 1, e = AT->getNumElements(); i != e; ++i) {
617           Out << ", ";
618           printConstant(CZ);
619         }
620       }
621       Out << " }";
622     } else {
623       printConstantArray(cast<ConstantArray>(CPV));
624     }
625     break;
626
627   case Type::StructTyID:
628     if (isa<ConstantAggregateZero>(CPV)) {
629       const StructType *ST = cast<StructType>(CPV->getType());
630       Out << "{";
631       if (ST->getNumElements()) {
632         Out << " ";
633         printConstant(Constant::getNullValue(ST->getElementType(0)));
634         for (unsigned i = 1, e = ST->getNumElements(); i != e; ++i) {
635           Out << ", ";
636           printConstant(Constant::getNullValue(ST->getElementType(i)));
637         }
638       }
639       Out << " }";
640     } else {
641       Out << "{";
642       if (CPV->getNumOperands()) {
643         Out << " ";
644         printConstant(cast<Constant>(CPV->getOperand(0)));
645         for (unsigned i = 1, e = CPV->getNumOperands(); i != e; ++i) {
646           Out << ", ";
647           printConstant(cast<Constant>(CPV->getOperand(i)));
648         }
649       }
650       Out << " }";
651     }
652     break;
653
654   case Type::PointerTyID:
655     if (isa<ConstantPointerNull>(CPV)) {
656       Out << "((";
657       printType(Out, CPV->getType());
658       Out << ")/*NULL*/0)";
659       break;
660     } else if (GlobalValue *GV = dyn_cast<GlobalValue>(CPV)) {
661       writeOperand(GV);
662       break;
663     }
664     // FALL THROUGH
665   default:
666     std::cerr << "Unknown constant type: " << *CPV << "\n";
667     abort();
668   }
669 }
670
671 void CWriter::writeOperandInternal(Value *Operand) {
672   if (Instruction *I = dyn_cast<Instruction>(Operand))
673     if (isInlinableInst(*I) && !isDirectAlloca(I)) {
674       // Should we inline this instruction to build a tree?
675       Out << "(";
676       visit(*I);
677       Out << ")";    
678       return;
679     }
680   
681   Constant* CPV = dyn_cast<Constant>(Operand);
682   if (CPV && !isa<GlobalValue>(CPV)) {
683     printConstant(CPV); 
684   } else {
685     Out << Mang->getValueName(Operand);
686   }
687 }
688
689 void CWriter::writeOperand(Value *Operand) {
690   if (isa<GlobalVariable>(Operand) || isDirectAlloca(Operand))
691     Out << "(&";  // Global variables are references as their addresses by llvm
692
693   writeOperandInternal(Operand);
694
695   if (isa<GlobalVariable>(Operand) || isDirectAlloca(Operand))
696     Out << ")";
697 }
698
699 // generateCompilerSpecificCode - This is where we add conditional compilation
700 // directives to cater to specific compilers as need be.
701 //
702 static void generateCompilerSpecificCode(std::ostream& Out) {
703   // Alloca is hard to get, and we don't want to include stdlib.h here...
704   Out << "/* get a declaration for alloca */\n"
705       << "#if defined(sun) || defined(__CYGWIN__) || defined(__APPLE__)\n"
706       << "extern void *__builtin_alloca(unsigned long);\n"
707       << "#define alloca(x) __builtin_alloca(x)\n"
708       << "#else\n"
709       << "#ifndef __FreeBSD__\n"
710       << "#include <alloca.h>\n"
711       << "#endif\n"
712       << "#endif\n\n";
713
714   // We output GCC specific attributes to preserve 'linkonce'ness on globals.
715   // If we aren't being compiled with GCC, just drop these attributes.
716   Out << "#ifndef __GNUC__  /* Can only support \"linkonce\" vars with GCC */\n"
717       << "#define __attribute__(X)\n"
718       << "#endif\n\n";
719
720 #if 0
721   // At some point, we should support "external weak" vs. "weak" linkages.
722   // On Mac OS X, "external weak" is spelled "__attribute__((weak_import))".
723   Out << "#if defined(__GNUC__) && defined(__APPLE_CC__)\n"
724       << "#define __EXTERNAL_WEAK__ __attribute__((weak_import))\n"
725       << "#elif defined(__GNUC__)\n"
726       << "#define __EXTERNAL_WEAK__ __attribute__((weak))\n"
727       << "#else\n"
728       << "#define __EXTERNAL_WEAK__\n"
729       << "#endif\n\n";
730 #endif
731
732   // For now, turn off the weak linkage attribute on Mac OS X. (See above.)
733   Out << "#if defined(__GNUC__) && defined(__APPLE_CC__)\n"
734       << "#define __ATTRIBUTE_WEAK__\n"
735       << "#elif defined(__GNUC__)\n"
736       << "#define __ATTRIBUTE_WEAK__ __attribute__((weak))\n"
737       << "#else\n"
738       << "#define __ATTRIBUTE_WEAK__\n"
739       << "#endif\n\n";
740
741   // Define NaN and Inf as GCC builtins if using GCC, as 0 otherwise
742   // From the GCC documentation:
743   // 
744   //   double __builtin_nan (const char *str)
745   //
746   // This is an implementation of the ISO C99 function nan.
747   //
748   // Since ISO C99 defines this function in terms of strtod, which we do
749   // not implement, a description of the parsing is in order. The string is
750   // parsed as by strtol; that is, the base is recognized by leading 0 or
751   // 0x prefixes. The number parsed is placed in the significand such that
752   // the least significant bit of the number is at the least significant
753   // bit of the significand. The number is truncated to fit the significand
754   // field provided. The significand is forced to be a quiet NaN.
755   //
756   // This function, if given a string literal, is evaluated early enough
757   // that it is considered a compile-time constant.
758   //
759   //   float __builtin_nanf (const char *str)
760   //
761   // Similar to __builtin_nan, except the return type is float.
762   //
763   //   double __builtin_inf (void)
764   //
765   // Similar to __builtin_huge_val, except a warning is generated if the
766   // target floating-point format does not support infinities. This
767   // function is suitable for implementing the ISO C99 macro INFINITY.
768   //
769   //   float __builtin_inff (void)
770   //
771   // Similar to __builtin_inf, except the return type is float.
772   Out << "#ifdef __GNUC__\n"
773       << "#define LLVM_NAN(NanStr)   __builtin_nan(NanStr)   /* Double */\n"
774       << "#define LLVM_NANF(NanStr)  __builtin_nanf(NanStr)  /* Float */\n"
775       << "#define LLVM_NANS(NanStr)  __builtin_nans(NanStr)  /* Double */\n"
776       << "#define LLVM_NANSF(NanStr) __builtin_nansf(NanStr) /* Float */\n"
777       << "#define LLVM_INF           __builtin_inf()         /* Double */\n"
778       << "#define LLVM_INFF          __builtin_inff()        /* Float */\n"
779       << "#else\n"
780       << "#define LLVM_NAN(NanStr)   ((double)0.0)           /* Double */\n"
781       << "#define LLVM_NANF(NanStr)  0.0F                    /* Float */\n"
782       << "#define LLVM_NANS(NanStr)  ((double)0.0)           /* Double */\n"
783       << "#define LLVM_NANSF(NanStr) 0.0F                    /* Float */\n"
784       << "#define LLVM_INF           ((double)0.0)           /* Double */\n"
785       << "#define LLVM_INFF          0.0F                    /* Float */\n"
786       << "#endif\n";
787 }
788
789 bool CWriter::doInitialization(Module &M) {
790   // Initialize
791   TheModule = &M;
792
793   IL.AddPrototypes(M);
794   
795   // Ensure that all structure types have names...
796   Mang = new Mangler(M);
797
798   // get declaration for alloca
799   Out << "/* Provide Declarations */\n";
800   Out << "#include <stdarg.h>\n";      // Varargs support
801   Out << "#include <setjmp.h>\n";      // Unwind support
802   generateCompilerSpecificCode(Out);
803
804   // Provide a definition for `bool' if not compiling with a C++ compiler.
805   Out << "\n"
806       << "#ifndef __cplusplus\ntypedef unsigned char bool;\n#endif\n"
807     
808       << "\n\n/* Support for floating point constants */\n"
809       << "typedef unsigned long long ConstantDoubleTy;\n"
810       << "typedef unsigned int        ConstantFloatTy;\n"
811     
812       << "\n\n/* Global Declarations */\n";
813
814   // First output all the declarations for the program, because C requires
815   // Functions & globals to be declared before they are used.
816   //
817
818   // Loop over the symbol table, emitting all named constants...
819   printModuleTypes(M.getSymbolTable());
820
821   // Global variable declarations...
822   if (!M.gempty()) {
823     Out << "\n/* External Global Variable Declarations */\n";
824     for (Module::giterator I = M.gbegin(), E = M.gend(); I != E; ++I) {
825       if (I->hasExternalLinkage()) {
826         Out << "extern ";
827         printType(Out, I->getType()->getElementType(), Mang->getValueName(I));
828         Out << ";\n";
829       }
830     }
831   }
832
833   // Function declarations
834   if (!M.empty()) {
835     Out << "\n/* Function Declarations */\n";
836     for (Module::iterator I = M.begin(), E = M.end(); I != E; ++I) {
837       // Don't print declarations for intrinsic functions.
838       if (!I->getIntrinsicID() && 
839           I->getName() != "setjmp" && I->getName() != "longjmp") {
840         printFunctionSignature(I, true);
841         if (I->hasWeakLinkage()) Out << " __ATTRIBUTE_WEAK__";
842         if (I->hasLinkOnceLinkage()) Out << " __ATTRIBUTE_WEAK__";
843         Out << ";\n";
844       }
845     }
846   }
847
848   // Output the global variable declarations
849   if (!M.gempty()) {
850     Out << "\n\n/* Global Variable Declarations */\n";
851     for (Module::giterator I = M.gbegin(), E = M.gend(); I != E; ++I)
852       if (!I->isExternal()) {
853         Out << "extern ";
854         printType(Out, I->getType()->getElementType(), Mang->getValueName(I));
855
856         if (I->hasLinkOnceLinkage())
857           Out << " __attribute__((common))";
858         else if (I->hasWeakLinkage())
859           Out << " __ATTRIBUTE_WEAK__";
860         Out << ";\n";
861       }
862   }
863
864   // Output the global variable definitions and contents...
865   if (!M.gempty()) {
866     Out << "\n\n/* Global Variable Definitions and Initialization */\n";
867     for (Module::giterator I = M.gbegin(), E = M.gend(); I != E; ++I)
868       if (!I->isExternal()) {
869         if (I->hasInternalLinkage())
870           Out << "static ";
871         printType(Out, I->getType()->getElementType(), Mang->getValueName(I));
872         if (I->hasLinkOnceLinkage())
873           Out << " __attribute__((common))";
874         else if (I->hasWeakLinkage())
875           Out << " __ATTRIBUTE_WEAK__";
876
877         // If the initializer is not null, emit the initializer.  If it is null,
878         // we try to avoid emitting large amounts of zeros.  The problem with
879         // this, however, occurs when the variable has weak linkage.  In this
880         // case, the assembler will complain about the variable being both weak
881         // and common, so we disable this optimization.
882         if (!I->getInitializer()->isNullValue()) {
883           Out << " = " ;
884           writeOperand(I->getInitializer());
885         } else if (I->hasWeakLinkage()) {
886           // We have to specify an initializer, but it doesn't have to be
887           // complete.  If the value is an aggregate, print out { 0 }, and let
888           // the compiler figure out the rest of the zeros.
889           Out << " = " ;
890           if (isa<StructType>(I->getInitializer()->getType()) ||
891               isa<ArrayType>(I->getInitializer()->getType())) {
892             Out << "{ 0 }";
893           } else {
894             // Just print it out normally.
895             writeOperand(I->getInitializer());
896           }
897         }
898         Out << ";\n";
899       }
900   }
901
902   if (!M.empty())
903     Out << "\n\n/* Function Bodies */\n";
904   return false;
905 }
906
907
908 /// Output all floating point constants that cannot be printed accurately...
909 void CWriter::printFloatingPointConstants(Function &F) {
910   union {
911     double D;
912     uint64_t U;
913   } DBLUnion;
914
915   union {
916     float F;
917     unsigned U;
918   } FLTUnion;
919
920   // Scan the module for floating point constants.  If any FP constant is used
921   // in the function, we want to redirect it here so that we do not depend on
922   // the precision of the printed form, unless the printed form preserves
923   // precision.
924   //
925   static unsigned FPCounter = 0;
926   for (constant_iterator I = constant_begin(&F), E = constant_end(&F);
927        I != E; ++I)
928     if (const ConstantFP *FPC = dyn_cast<ConstantFP>(*I))
929       if (!isFPCSafeToPrint(FPC) && // Do not put in FPConstantMap if safe.
930           !FPConstantMap.count(FPC)) {
931         double Val = FPC->getValue();
932         
933         FPConstantMap[FPC] = FPCounter;  // Number the FP constants
934         
935         if (FPC->getType() == Type::DoubleTy) {
936           DBLUnion.D = Val;
937           Out << "static const ConstantDoubleTy FPConstant" << FPCounter++
938               << " = 0x" << std::hex << DBLUnion.U << std::dec
939               << "ULL;    /* " << Val << " */\n";
940         } else if (FPC->getType() == Type::FloatTy) {
941           FLTUnion.F = Val;
942           Out << "static const ConstantFloatTy FPConstant" << FPCounter++
943               << " = 0x" << std::hex << FLTUnion.U << std::dec
944               << "U;    /* " << Val << " */\n";
945         } else
946           assert(0 && "Unknown float type!");
947       }
948   
949   Out << "\n";
950 }
951
952
953 /// printSymbolTable - Run through symbol table looking for type names.  If a
954 /// type name is found, emit it's declaration...
955 ///
956 void CWriter::printModuleTypes(const SymbolTable &ST) {
957   // If there are no type names, exit early.
958   if ( ! ST.hasTypes() )
959     return;
960
961   // We are only interested in the type plane of the symbol table...
962   SymbolTable::type_const_iterator I   = ST.type_begin();
963   SymbolTable::type_const_iterator End = ST.type_end();
964   
965   // Print out forward declarations for structure types before anything else!
966   Out << "/* Structure forward decls */\n";
967   for (; I != End; ++I)
968     if (const Type *STy = dyn_cast<StructType>(I->second)) {
969       std::string Name = "struct l_" + Mangler::makeNameProper(I->first);
970       Out << Name << ";\n";
971       TypeNames.insert(std::make_pair(STy, Name));
972     }
973
974   Out << "\n";
975
976   // Now we can print out typedefs...
977   Out << "/* Typedefs */\n";
978   for (I = ST.type_begin(); I != End; ++I) {
979     const Type *Ty = cast<Type>(I->second);
980     std::string Name = "l_" + Mangler::makeNameProper(I->first);
981     Out << "typedef ";
982     printType(Out, Ty, Name);
983     Out << ";\n";
984   }
985   
986   Out << "\n";
987
988   // Keep track of which structures have been printed so far...
989   std::set<const StructType *> StructPrinted;
990
991   // Loop over all structures then push them into the stack so they are
992   // printed in the correct order.
993   //
994   Out << "/* Structure contents */\n";
995   for (I = ST.type_begin(); I != End; ++I)
996     if (const StructType *STy = dyn_cast<StructType>(I->second))
997       // Only print out used types!
998       printContainedStructs(STy, StructPrinted);
999 }
1000
1001 // Push the struct onto the stack and recursively push all structs
1002 // this one depends on.
1003 void CWriter::printContainedStructs(const Type *Ty,
1004                                     std::set<const StructType*> &StructPrinted){
1005   if (const StructType *STy = dyn_cast<StructType>(Ty)) {
1006     //Check to see if we have already printed this struct
1007     if (StructPrinted.count(STy) == 0) {
1008       // Print all contained types first...
1009       for (StructType::element_iterator I = STy->element_begin(),
1010              E = STy->element_end(); I != E; ++I) {
1011         const Type *Ty1 = I->get();
1012         if (isa<StructType>(Ty1) || isa<ArrayType>(Ty1))
1013           printContainedStructs(*I, StructPrinted);
1014       }
1015       
1016       //Print structure type out..
1017       StructPrinted.insert(STy);
1018       std::string Name = TypeNames[STy];  
1019       printType(Out, STy, Name, true);
1020       Out << ";\n\n";
1021     }
1022
1023     // If it is an array, check contained types and continue
1024   } else if (const ArrayType *ATy = dyn_cast<ArrayType>(Ty)){
1025     const Type *Ty1 = ATy->getElementType();
1026     if (isa<StructType>(Ty1) || isa<ArrayType>(Ty1))
1027       printContainedStructs(Ty1, StructPrinted);
1028   }
1029 }
1030
1031
1032 void CWriter::printFunctionSignature(const Function *F, bool Prototype) {
1033   if (F->hasInternalLinkage()) Out << "static ";
1034   
1035   // Loop over the arguments, printing them...
1036   const FunctionType *FT = cast<FunctionType>(F->getFunctionType());
1037   
1038   std::stringstream FunctionInnards; 
1039     
1040   // Print out the name...
1041   FunctionInnards << Mang->getValueName(F) << "(";
1042     
1043   if (!F->isExternal()) {
1044     if (!F->aempty()) {
1045       std::string ArgName;
1046       if (F->abegin()->hasName() || !Prototype)
1047         ArgName = Mang->getValueName(F->abegin());
1048       printType(FunctionInnards, F->afront().getType(), ArgName);
1049       for (Function::const_aiterator I = ++F->abegin(), E = F->aend();
1050            I != E; ++I) {
1051         FunctionInnards << ", ";
1052         if (I->hasName() || !Prototype)
1053           ArgName = Mang->getValueName(I);
1054         else 
1055           ArgName = "";
1056         printType(FunctionInnards, I->getType(), ArgName);
1057       }
1058     }
1059   } else {
1060     // Loop over the arguments, printing them...
1061     for (FunctionType::param_iterator I = FT->param_begin(),
1062            E = FT->param_end(); I != E; ++I) {
1063       if (I != FT->param_begin()) FunctionInnards << ", ";
1064       printType(FunctionInnards, *I);
1065     }
1066   }
1067
1068   // Finish printing arguments... if this is a vararg function, print the ...,
1069   // unless there are no known types, in which case, we just emit ().
1070   //
1071   if (FT->isVarArg() && FT->getNumParams()) {
1072     if (FT->getNumParams()) FunctionInnards << ", ";
1073     FunctionInnards << "...";  // Output varargs portion of signature!
1074   } else if (!FT->isVarArg() && FT->getNumParams() == 0) {
1075     FunctionInnards << "void"; // ret() -> ret(void) in C.
1076   }
1077   FunctionInnards << ")";
1078   // Print out the return type and the entire signature for that matter
1079   printType(Out, F->getReturnType(), FunctionInnards.str());
1080 }
1081
1082 void CWriter::printFunction(Function &F) {
1083   printFunctionSignature(&F, false);
1084   Out << " {\n";
1085
1086   // print local variable information for the function
1087   for (inst_iterator I = inst_begin(&F), E = inst_end(&F); I != E; ++I)
1088     if (const AllocaInst *AI = isDirectAlloca(&*I)) {
1089       Out << "  ";
1090       printType(Out, AI->getAllocatedType(), Mang->getValueName(AI));
1091       Out << ";    /* Address exposed local */\n";
1092     } else if (I->getType() != Type::VoidTy && !isInlinableInst(*I)) {
1093       Out << "  ";
1094       printType(Out, I->getType(), Mang->getValueName(&*I));
1095       Out << ";\n";
1096       
1097       if (isa<PHINode>(*I)) {  // Print out PHI node temporaries as well...
1098         Out << "  ";
1099         printType(Out, I->getType(),
1100                   Mang->getValueName(&*I)+"__PHI_TEMPORARY");
1101         Out << ";\n";
1102       }
1103     }
1104
1105   Out << "\n";
1106
1107   // print the basic blocks
1108   for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB) {
1109     if (Loop *L = LI->getLoopFor(BB)) {
1110       if (L->getHeader() == BB && L->getParentLoop() == 0)
1111         printLoop(L);
1112     } else {
1113       printBasicBlock(BB);
1114     }
1115   }
1116   
1117   Out << "}\n\n";
1118 }
1119
1120 void CWriter::printLoop(Loop *L) {
1121   Out << "  do {     /* Syntactic loop '" << L->getHeader()->getName()
1122       << "' to make GCC happy */\n";
1123   for (unsigned i = 0, e = L->getBlocks().size(); i != e; ++i) {
1124     BasicBlock *BB = L->getBlocks()[i];
1125     Loop *BBLoop = LI->getLoopFor(BB);
1126     if (BBLoop == L)
1127       printBasicBlock(BB);
1128     else if (BB == BBLoop->getHeader() && BBLoop->getParentLoop() == L)
1129       printLoop(BBLoop);      
1130   }
1131   Out << "  } while (1); /* end of syntactic loop '"
1132       << L->getHeader()->getName() << "' */\n";
1133 }
1134
1135 void CWriter::printBasicBlock(BasicBlock *BB) {
1136
1137   // Don't print the label for the basic block if there are no uses, or if
1138   // the only terminator use is the predecessor basic block's terminator.
1139   // We have to scan the use list because PHI nodes use basic blocks too but
1140   // do not require a label to be generated.
1141   //
1142   bool NeedsLabel = false;
1143   for (pred_iterator PI = pred_begin(BB), E = pred_end(BB); PI != E; ++PI)
1144     if (isGotoCodeNecessary(*PI, BB)) {
1145       NeedsLabel = true;
1146       break;
1147     }
1148       
1149   if (NeedsLabel) Out << Mang->getValueName(BB) << ":\n";
1150       
1151   // Output all of the instructions in the basic block...
1152   for (BasicBlock::iterator II = BB->begin(), E = --BB->end(); II != E;
1153        ++II) {
1154     if (!isInlinableInst(*II) && !isDirectAlloca(II)) {
1155       if (II->getType() != Type::VoidTy)
1156         outputLValue(II);
1157       else
1158         Out << "  ";
1159       visit(*II);
1160       Out << ";\n";
1161     }
1162   }
1163       
1164   // Don't emit prefix or suffix for the terminator...
1165   visit(*BB->getTerminator());
1166 }
1167
1168
1169 // Specific Instruction type classes... note that all of the casts are
1170 // necessary because we use the instruction classes as opaque types...
1171 //
1172 void CWriter::visitReturnInst(ReturnInst &I) {
1173   // Don't output a void return if this is the last basic block in the function
1174   if (I.getNumOperands() == 0 && 
1175       &*--I.getParent()->getParent()->end() == I.getParent() &&
1176       !I.getParent()->size() == 1) {
1177     return;
1178   }
1179
1180   Out << "  return";
1181   if (I.getNumOperands()) {
1182     Out << " ";
1183     writeOperand(I.getOperand(0));
1184   }
1185   Out << ";\n";
1186 }
1187
1188 void CWriter::visitSwitchInst(SwitchInst &SI) {
1189   printPHICopiesForSuccessors(SI.getParent(), 0);
1190
1191   Out << "  switch (";
1192   writeOperand(SI.getOperand(0));
1193   Out << ") {\n  default:\n";
1194   printBranchToBlock(SI.getParent(), SI.getDefaultDest(), 2);
1195   Out << ";\n";
1196   for (unsigned i = 2, e = SI.getNumOperands(); i != e; i += 2) {
1197     Out << "  case ";
1198     writeOperand(SI.getOperand(i));
1199     Out << ":\n";
1200     BasicBlock *Succ = cast<BasicBlock>(SI.getOperand(i+1));
1201     printBranchToBlock(SI.getParent(), Succ, 2);
1202     if (Succ == SI.getParent()->getNext())
1203       Out << "    break;\n";
1204   }
1205   Out << "  }\n";
1206 }
1207
1208 bool CWriter::isGotoCodeNecessary(BasicBlock *From, BasicBlock *To) {
1209   /// FIXME: This should be reenabled, but loop reordering safe!!
1210   return true;
1211
1212   if (From->getNext() != To) // Not the direct successor, we need a goto
1213     return true; 
1214
1215   //isa<SwitchInst>(From->getTerminator())
1216
1217
1218   if (LI->getLoopFor(From) != LI->getLoopFor(To))
1219     return true;
1220   return false;
1221 }
1222
1223 void CWriter::printPHICopiesForSuccessors(BasicBlock *CurBlock, 
1224                                           unsigned Indent) {
1225   for (succ_iterator SI = succ_begin(CurBlock), E = succ_end(CurBlock);
1226        SI != E; ++SI)
1227     for (BasicBlock::iterator I = SI->begin();
1228          PHINode *PN = dyn_cast<PHINode>(I); ++I) {
1229       //  now we have to do the printing
1230       Out << std::string(Indent, ' ');
1231       Out << "  " << Mang->getValueName(I) << "__PHI_TEMPORARY = ";
1232       writeOperand(PN->getIncomingValue(PN->getBasicBlockIndex(CurBlock)));
1233       Out << ";   /* for PHI node */\n";
1234     }
1235 }
1236
1237
1238 void CWriter::printBranchToBlock(BasicBlock *CurBB, BasicBlock *Succ,
1239                                  unsigned Indent) {
1240   if (isGotoCodeNecessary(CurBB, Succ)) {
1241     Out << std::string(Indent, ' ') << "  goto ";
1242     writeOperand(Succ);
1243     Out << ";\n";
1244   }
1245 }
1246
1247 // Branch instruction printing - Avoid printing out a branch to a basic block
1248 // that immediately succeeds the current one.
1249 //
1250 void CWriter::visitBranchInst(BranchInst &I) {
1251   printPHICopiesForSuccessors(I.getParent(), 0);
1252
1253   if (I.isConditional()) {
1254     if (isGotoCodeNecessary(I.getParent(), I.getSuccessor(0))) {
1255       Out << "  if (";
1256       writeOperand(I.getCondition());
1257       Out << ") {\n";
1258       
1259       printBranchToBlock(I.getParent(), I.getSuccessor(0), 2);
1260       
1261       if (isGotoCodeNecessary(I.getParent(), I.getSuccessor(1))) {
1262         Out << "  } else {\n";
1263         printBranchToBlock(I.getParent(), I.getSuccessor(1), 2);
1264       }
1265     } else {
1266       // First goto not necessary, assume second one is...
1267       Out << "  if (!";
1268       writeOperand(I.getCondition());
1269       Out << ") {\n";
1270
1271       printBranchToBlock(I.getParent(), I.getSuccessor(1), 2);
1272     }
1273
1274     Out << "  }\n";
1275   } else {
1276     printBranchToBlock(I.getParent(), I.getSuccessor(0), 0);
1277   }
1278   Out << "\n";
1279 }
1280
1281 // PHI nodes get copied into temporary values at the end of predecessor basic
1282 // blocks.  We now need to copy these temporary values into the REAL value for
1283 // the PHI.
1284 void CWriter::visitPHINode(PHINode &I) {
1285   writeOperand(&I);
1286   Out << "__PHI_TEMPORARY";
1287 }
1288
1289
1290 void CWriter::visitBinaryOperator(Instruction &I) {
1291   // binary instructions, shift instructions, setCond instructions.
1292   assert(!isa<PointerType>(I.getType()));
1293
1294   // We must cast the results of binary operations which might be promoted.
1295   bool needsCast = false;
1296   if ((I.getType() == Type::UByteTy) || (I.getType() == Type::SByteTy)
1297       || (I.getType() == Type::UShortTy) || (I.getType() == Type::ShortTy)
1298       || (I.getType() == Type::FloatTy)) {
1299     needsCast = true;
1300     Out << "((";
1301     printType(Out, I.getType());
1302     Out << ")(";
1303   }
1304       
1305   writeOperand(I.getOperand(0));
1306
1307   switch (I.getOpcode()) {
1308   case Instruction::Add: Out << " + "; break;
1309   case Instruction::Sub: Out << " - "; break;
1310   case Instruction::Mul: Out << "*"; break;
1311   case Instruction::Div: Out << "/"; break;
1312   case Instruction::Rem: Out << "%"; break;
1313   case Instruction::And: Out << " & "; break;
1314   case Instruction::Or: Out << " | "; break;
1315   case Instruction::Xor: Out << " ^ "; break;
1316   case Instruction::SetEQ: Out << " == "; break;
1317   case Instruction::SetNE: Out << " != "; break;
1318   case Instruction::SetLE: Out << " <= "; break;
1319   case Instruction::SetGE: Out << " >= "; break;
1320   case Instruction::SetLT: Out << " < "; break;
1321   case Instruction::SetGT: Out << " > "; break;
1322   case Instruction::Shl : Out << " << "; break;
1323   case Instruction::Shr : Out << " >> "; break;
1324   default: std::cerr << "Invalid operator type!" << I; abort();
1325   }
1326
1327   writeOperand(I.getOperand(1));
1328
1329   if (needsCast) {
1330     Out << "))";
1331   }
1332 }
1333
1334 void CWriter::visitCastInst(CastInst &I) {
1335   if (I.getType() == Type::BoolTy) {
1336     Out << "(";
1337     writeOperand(I.getOperand(0));
1338     Out << " != 0)";
1339     return;
1340   }
1341   Out << "(";
1342   printType(Out, I.getType());
1343   Out << ")";
1344   if (isa<PointerType>(I.getType())&&I.getOperand(0)->getType()->isIntegral() ||
1345       isa<PointerType>(I.getOperand(0)->getType())&&I.getType()->isIntegral()) {
1346     // Avoid "cast to pointer from integer of different size" warnings
1347     Out << "(long)";  
1348   }
1349   
1350   writeOperand(I.getOperand(0));
1351 }
1352
1353 void CWriter::visitSelectInst(SelectInst &I) {
1354   Out << "((";
1355   writeOperand(I.getCondition());
1356   Out << ") ? (";
1357   writeOperand(I.getTrueValue());
1358   Out << ") : (";
1359   writeOperand(I.getFalseValue());
1360   Out << "))";    
1361 }
1362
1363
1364 void CWriter::lowerIntrinsics(Function &F) {
1365   for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB)
1366     for (BasicBlock::iterator I = BB->begin(), E = BB->end(); I != E; )
1367       if (CallInst *CI = dyn_cast<CallInst>(I++))
1368         if (Function *F = CI->getCalledFunction())
1369           switch (F->getIntrinsicID()) {
1370           case Intrinsic::not_intrinsic:
1371           case Intrinsic::vastart:
1372           case Intrinsic::vacopy:
1373           case Intrinsic::vaend:
1374           case Intrinsic::returnaddress:
1375           case Intrinsic::frameaddress:
1376           case Intrinsic::setjmp:
1377           case Intrinsic::longjmp:
1378             // We directly implement these intrinsics
1379             break;
1380           default:
1381             // All other intrinsic calls we must lower.
1382             Instruction *Before = CI->getPrev();
1383             IL.LowerIntrinsicCall(CI);
1384             if (Before) {        // Move iterator to instruction after call
1385               I = Before; ++I;
1386             } else {
1387               I = BB->begin();
1388             }
1389           }
1390 }
1391
1392
1393
1394 void CWriter::visitCallInst(CallInst &I) {
1395   // Handle intrinsic function calls first...
1396   if (Function *F = I.getCalledFunction())
1397     if (Intrinsic::ID ID = (Intrinsic::ID)F->getIntrinsicID()) {
1398       switch (ID) {
1399       default: assert(0 && "Unknown LLVM intrinsic!");
1400       case Intrinsic::vastart: 
1401         Out << "0; ";
1402         
1403         Out << "va_start(*(va_list*)&" << Mang->getValueName(&I) << ", ";
1404         // Output the last argument to the enclosing function...
1405         if (I.getParent()->getParent()->aempty()) {
1406           std::cerr << "The C backend does not currently support zero "
1407                     << "argument varargs functions, such as '"
1408                     << I.getParent()->getParent()->getName() << "'!\n";
1409           abort();
1410         }
1411         writeOperand(&I.getParent()->getParent()->aback());
1412         Out << ")";
1413         return;
1414       case Intrinsic::vaend:
1415         if (!isa<ConstantPointerNull>(I.getOperand(1))) {
1416           Out << "va_end(*(va_list*)&";
1417           writeOperand(I.getOperand(1));
1418           Out << ")";
1419         } else {
1420           Out << "va_end(*(va_list*)0)";
1421         }
1422         return;
1423       case Intrinsic::vacopy:
1424         Out << "0;";
1425         Out << "va_copy(*(va_list*)&" << Mang->getValueName(&I) << ", ";
1426         Out << "*(va_list*)&";
1427         writeOperand(I.getOperand(1));
1428         Out << ")";
1429         return;
1430       case Intrinsic::returnaddress:
1431         Out << "__builtin_return_address(";
1432         writeOperand(I.getOperand(1));
1433         Out << ")";
1434         return;
1435       case Intrinsic::frameaddress:
1436         Out << "__builtin_frame_address(";
1437         writeOperand(I.getOperand(1));
1438         Out << ")";
1439         return;
1440       case Intrinsic::setjmp:
1441         Out << "setjmp(*(jmp_buf*)";
1442         writeOperand(I.getOperand(1));
1443         Out << ")";
1444         return;
1445       case Intrinsic::longjmp:
1446         Out << "longjmp(*(jmp_buf*)";
1447         writeOperand(I.getOperand(1));
1448         Out << ", ";
1449         writeOperand(I.getOperand(2));
1450         Out << ")";
1451         return;
1452       }
1453     }
1454   visitCallSite(&I);
1455 }
1456
1457 void CWriter::visitCallSite(CallSite CS) {
1458   const PointerType  *PTy   = cast<PointerType>(CS.getCalledValue()->getType());
1459   const FunctionType *FTy   = cast<FunctionType>(PTy->getElementType());
1460   const Type         *RetTy = FTy->getReturnType();
1461   
1462   writeOperand(CS.getCalledValue());
1463   Out << "(";
1464
1465   if (CS.arg_begin() != CS.arg_end()) {
1466     CallSite::arg_iterator AI = CS.arg_begin(), AE = CS.arg_end();
1467     writeOperand(*AI);
1468
1469     for (++AI; AI != AE; ++AI) {
1470       Out << ", ";
1471       writeOperand(*AI);
1472     }
1473   }
1474   Out << ")";
1475 }  
1476
1477 void CWriter::visitMallocInst(MallocInst &I) {
1478   assert(0 && "lowerallocations pass didn't work!");
1479 }
1480
1481 void CWriter::visitAllocaInst(AllocaInst &I) {
1482   Out << "(";
1483   printType(Out, I.getType());
1484   Out << ") alloca(sizeof(";
1485   printType(Out, I.getType()->getElementType());
1486   Out << ")";
1487   if (I.isArrayAllocation()) {
1488     Out << " * " ;
1489     writeOperand(I.getOperand(0));
1490   }
1491   Out << ")";
1492 }
1493
1494 void CWriter::visitFreeInst(FreeInst &I) {
1495   assert(0 && "lowerallocations pass didn't work!");
1496 }
1497
1498 void CWriter::printIndexingExpression(Value *Ptr, gep_type_iterator I,
1499                                       gep_type_iterator E) {
1500   bool HasImplicitAddress = false;
1501   // If accessing a global value with no indexing, avoid *(&GV) syndrome
1502   if (GlobalValue *V = dyn_cast<GlobalValue>(Ptr)) {
1503     HasImplicitAddress = true;
1504   } else if (isDirectAlloca(Ptr)) {
1505     HasImplicitAddress = true;
1506   }
1507
1508   if (I == E) {
1509     if (!HasImplicitAddress)
1510       Out << "*";  // Implicit zero first argument: '*x' is equivalent to 'x[0]'
1511
1512     writeOperandInternal(Ptr);
1513     return;
1514   }
1515
1516   const Constant *CI = dyn_cast<Constant>(I.getOperand());
1517   if (HasImplicitAddress && (!CI || !CI->isNullValue()))
1518     Out << "(&";
1519
1520   writeOperandInternal(Ptr);
1521
1522   if (HasImplicitAddress && (!CI || !CI->isNullValue())) {
1523     Out << ")";
1524     HasImplicitAddress = false;  // HIA is only true if we haven't addressed yet
1525   }
1526
1527   assert(!HasImplicitAddress || (CI && CI->isNullValue()) &&
1528          "Can only have implicit address with direct accessing");
1529
1530   if (HasImplicitAddress) {
1531     ++I;
1532   } else if (CI && CI->isNullValue()) {
1533     gep_type_iterator TmpI = I; ++TmpI;
1534
1535     // Print out the -> operator if possible...
1536     if (TmpI != E && isa<StructType>(*TmpI)) {
1537       Out << (HasImplicitAddress ? "." : "->");
1538       Out << "field" << cast<ConstantUInt>(TmpI.getOperand())->getValue();
1539       I = ++TmpI;
1540     }
1541   }
1542
1543   for (; I != E; ++I)
1544     if (isa<StructType>(*I)) {
1545       Out << ".field" << cast<ConstantUInt>(I.getOperand())->getValue();
1546     } else {
1547       Out << "[";
1548       writeOperand(I.getOperand());
1549       Out << "]";
1550     }
1551 }
1552
1553 void CWriter::visitLoadInst(LoadInst &I) {
1554   Out << "*";
1555   writeOperand(I.getOperand(0));
1556 }
1557
1558 void CWriter::visitStoreInst(StoreInst &I) {
1559   Out << "*";
1560   writeOperand(I.getPointerOperand());
1561   Out << " = ";
1562   writeOperand(I.getOperand(0));
1563 }
1564
1565 void CWriter::visitGetElementPtrInst(GetElementPtrInst &I) {
1566   Out << "&";
1567   printIndexingExpression(I.getPointerOperand(), gep_type_begin(I),
1568                           gep_type_end(I));
1569 }
1570
1571 void CWriter::visitVANextInst(VANextInst &I) {
1572   Out << Mang->getValueName(I.getOperand(0));
1573   Out << ";  va_arg(*(va_list*)&" << Mang->getValueName(&I) << ", ";
1574   printType(Out, I.getArgType());
1575   Out << ")";  
1576 }
1577
1578 void CWriter::visitVAArgInst(VAArgInst &I) {
1579   Out << "0;\n";
1580   Out << "{ va_list Tmp; va_copy(Tmp, *(va_list*)&";
1581   writeOperand(I.getOperand(0));
1582   Out << ");\n  " << Mang->getValueName(&I) << " = va_arg(Tmp, ";
1583   printType(Out, I.getType());
1584   Out << ");\n  va_end(Tmp); }";
1585 }
1586
1587 //===----------------------------------------------------------------------===//
1588 //                       External Interface declaration
1589 //===----------------------------------------------------------------------===//
1590
1591 bool CTargetMachine::addPassesToEmitAssembly(PassManager &PM, std::ostream &o) {
1592   PM.add(createLowerGCPass());
1593   PM.add(createLowerAllocationsPass());
1594   PM.add(createLowerInvokePass());
1595   PM.add(new CBackendNameAllUsedStructs());
1596   PM.add(new CWriter(o, getIntrinsicLowering()));
1597   return false;
1598 }
1599
1600 // vim: sw=2