In getID(), don't call getValidSymbolName to mangle external names!
[oota-llvm.git] / lib / Target / SparcV9 / SparcV9AsmPrinter.cpp
1 //===-- EmitAssembly.cpp - Emit Sparc Specific .s File ---------------------==//
2 //
3 // This file implements all of the stuff neccesary to output a .s file from
4 // LLVM.  The code in this file assumes that the specified module has already
5 // been compiled into the internal data structures of the Module.
6 //
7 // This code largely consists of two LLVM Pass's: a FunctionPass and a Pass.
8 // The FunctionPass is pipelined together with all of the rest of the code
9 // generation stages, and the Pass runs at the end to emit code for global
10 // variables and such.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "SparcInternals.h"
15 #include "llvm/CodeGen/MachineInstr.h"
16 #include "llvm/CodeGen/MachineFunction.h"
17 #include "llvm/Constants.h"
18 #include "llvm/DerivedTypes.h"
19 #include "llvm/Module.h"
20 #include "llvm/SlotCalculator.h"
21 #include "llvm/Pass.h"
22 #include "llvm/Assembly/Writer.h"
23 #include "Support/StringExtras.h"
24 using std::string;
25
26 namespace {
27
28 class GlobalIdTable: public Annotation {
29   static AnnotationID AnnotId;
30   friend class AsmPrinter;              // give access to AnnotId
31   
32   typedef hash_map<const Value*, int> ValIdMap;
33   typedef ValIdMap::const_iterator ValIdMapConstIterator;
34   typedef ValIdMap::      iterator ValIdMapIterator;
35 public:
36   SlotCalculator Table;    // map anonymous values to unique integer IDs
37   ValIdMap valToIdMap;     // used for values not handled by SlotCalculator 
38   
39   GlobalIdTable(Module* M) : Annotation(AnnotId), Table(M, true) {}
40 };
41
42 AnnotationID GlobalIdTable::AnnotId =
43   AnnotationManager::getID("ASM PRINTER GLOBAL TABLE ANNOT");
44   
45 //===---------------------------------------------------------------------===//
46 //   Code Shared By the two printer passes, as a mixin
47 //===---------------------------------------------------------------------===//
48
49 class AsmPrinter {
50   GlobalIdTable* idTable;
51 public:
52   std::ostream &toAsm;
53   const TargetMachine &Target;
54   
55   enum Sections {
56     Unknown,
57     Text,
58     ReadOnlyData,
59     InitRWData,
60     ZeroInitRWData,
61   } CurSection;
62
63   AsmPrinter(std::ostream &os, const TargetMachine &T)
64     : idTable(0), toAsm(os), Target(T), CurSection(Unknown) {}
65   
66   // (start|end)(Module|Function) - Callback methods to be invoked by subclasses
67   void startModule(Module &M) {
68     // Create the global id table if it does not already exist
69     idTable = (GlobalIdTable*)M.getAnnotation(GlobalIdTable::AnnotId);
70     if (idTable == NULL) {
71       idTable = new GlobalIdTable(&M);
72       M.addAnnotation(idTable);
73     }
74   }
75   void startFunction(Function &F) {
76     // Make sure the slot table has information about this function...
77     idTable->Table.incorporateFunction(&F);
78   }
79   void endFunction(Function &) {
80     idTable->Table.purgeFunction();  // Forget all about F
81   }
82   void endModule() {
83   }
84
85   // Check if a value is external or accessible from external code.
86   bool isExternal(const Value* V) {
87     const GlobalValue *GV = dyn_cast<GlobalValue>(V);
88     return GV && GV->hasExternalLinkage();
89   }
90   
91   // enterSection - Use this method to enter a different section of the output
92   // executable.  This is used to only output neccesary section transitions.
93   //
94   void enterSection(enum Sections S) {
95     if (S == CurSection) return;        // Only switch section if neccesary
96     CurSection = S;
97
98     toAsm << "\n\t.section ";
99     switch (S)
100       {
101       default: assert(0 && "Bad section name!");
102       case Text:         toAsm << "\".text\""; break;
103       case ReadOnlyData: toAsm << "\".rodata\",#alloc"; break;
104       case InitRWData:   toAsm << "\".data\",#alloc,#write"; break;
105       case ZeroInitRWData: toAsm << "\".bss\",#alloc,#write"; break;
106       }
107     toAsm << "\n";
108   }
109
110   static string getValidSymbolName(const string &S) {
111     string Result;
112     
113     // Symbol names in Sparc assembly language have these rules:
114     // (a) Must match { letter | _ | . | $ } { letter | _ | . | $ | digit }*
115     // (b) A name beginning in "." is treated as a local name.
116     // 
117     if (isdigit(S[0]))
118       Result = "ll";
119     
120     for (unsigned i = 0; i < S.size(); ++i)
121       {
122         char C = S[i];
123         if (C == '_' || C == '.' || C == '$' || isalpha(C) || isdigit(C))
124           Result += C;
125         else
126           {
127             Result += '_';
128             Result += char('0' + ((unsigned char)C >> 4));
129             Result += char('0' + (C & 0xF));
130           }
131       }
132     return Result;
133   }
134
135   // getID - Return a valid identifier for the specified value.  Base it on
136   // the name of the identifier if possible (qualified by the type), and
137   // use a numbered value based on prefix otherwise.
138   // FPrefix is always prepended to the output identifier.
139   //
140   string getID(const Value *V, const char *Prefix, const char *FPrefix = 0) {
141     string Result = FPrefix ? FPrefix : "";  // "Forced prefix"
142
143     Result +=  V->hasName() ? V->getName() : string(Prefix);
144
145     // Qualify all internal names with a unique id.
146     if (!isExternal(V)) {
147       int valId = idTable->Table.getValSlot(V);
148       if (valId == -1) {
149         GlobalIdTable::ValIdMapConstIterator I = idTable->valToIdMap.find(V);
150         if (I == idTable->valToIdMap.end())
151           valId = idTable->valToIdMap[V] = idTable->valToIdMap.size();
152         else
153           valId = I->second;
154       }
155       Result = Result + "_" + itostr(valId);
156
157       // Replace or prefix problem characters in the name
158       Result = getValidSymbolName(Result);
159     }
160
161     return Result;
162   }
163   
164   // getID Wrappers - Ensure consistent usage...
165   string getID(const Function *F) {
166     return getID(F, "LLVMFunction_");
167   }
168   string getID(const BasicBlock *BB) {
169     return getID(BB, "LL", (".L_"+getID(BB->getParent())+"_").c_str());
170   }
171   string getID(const GlobalVariable *GV) {
172     return getID(GV, "LLVMGlobal_");
173   }
174   string getID(const Constant *CV) {
175     return getID(CV, "LLVMConst_", ".C_");
176   }
177   string getID(const GlobalValue *GV) {
178     if (const GlobalVariable *V = dyn_cast<GlobalVariable>(GV))
179       return getID(V);
180     else if (const Function *F = dyn_cast<Function>(GV))
181       return getID(F);
182     assert(0 && "Unexpected type of GlobalValue!");
183     return "";
184   }
185
186   // ConstantExprToString() - Convert a ConstantExpr to an asm expression
187   // and return this as a string.
188   string ConstantExprToString(const ConstantExpr* CE,
189                               const TargetMachine& target) {
190     string S;
191     switch(CE->getOpcode()) {
192     case Instruction::GetElementPtr:
193       { // generate a symbolic expression for the byte address
194         const Value* ptrVal = CE->getOperand(0);
195         std::vector<Value*> idxVec(CE->op_begin()+1, CE->op_end());
196         S += "(" + valToExprString(ptrVal, target) + ") + ("
197           + utostr(target.DataLayout.getIndexedOffset(ptrVal->getType(),idxVec))
198           + ")";
199         break;
200       }
201
202     case Instruction::Cast:
203       // Support only non-converting casts for now, i.e., a no-op.
204       // This assertion is not a complete check.
205       assert(target.DataLayout.getTypeSize(CE->getType()) ==
206              target.DataLayout.getTypeSize(CE->getOperand(0)->getType()));
207       S += "(" + valToExprString(CE->getOperand(0), target) + ")";
208       break;
209
210     case Instruction::Add:
211       S += "(" + valToExprString(CE->getOperand(0), target) + ") + ("
212                + valToExprString(CE->getOperand(1), target) + ")";
213       break;
214
215     default:
216       assert(0 && "Unsupported operator in ConstantExprToString()");
217       break;
218     }
219
220     return S;
221   }
222
223   // valToExprString - Helper function for ConstantExprToString().
224   // Appends result to argument string S.
225   // 
226   string valToExprString(const Value* V, const TargetMachine& target) {
227     string S;
228     bool failed = false;
229     if (const Constant* CV = dyn_cast<Constant>(V)) { // symbolic or known
230
231       if (const ConstantBool *CB = dyn_cast<ConstantBool>(CV))
232         S += string(CB == ConstantBool::True ? "1" : "0");
233       else if (const ConstantSInt *CI = dyn_cast<ConstantSInt>(CV))
234         S += itostr(CI->getValue());
235       else if (const ConstantUInt *CI = dyn_cast<ConstantUInt>(CV))
236         S += utostr(CI->getValue());
237       else if (const ConstantFP *CFP = dyn_cast<ConstantFP>(CV))
238         S += ftostr(CFP->getValue());
239       else if (isa<ConstantPointerNull>(CV))
240         S += "0";
241       else if (const ConstantPointerRef *CPR = dyn_cast<ConstantPointerRef>(CV))
242         S += valToExprString(CPR->getValue(), target);
243       else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CV))
244         S += ConstantExprToString(CE, target);
245       else
246         failed = true;
247
248     } else if (const GlobalValue* GV = dyn_cast<GlobalValue>(V)) {
249       S += getID(GV);
250     }
251     else
252       failed = true;
253
254     if (failed) {
255       assert(0 && "Cannot convert value to string");
256       S += "<illegal-value>";
257     }
258     return S;
259   }
260
261 };
262
263
264
265 //===----------------------------------------------------------------------===//
266 //   SparcFunctionAsmPrinter Code
267 //===----------------------------------------------------------------------===//
268
269 struct SparcFunctionAsmPrinter : public FunctionPass, public AsmPrinter {
270   inline SparcFunctionAsmPrinter(std::ostream &os, const TargetMachine &t)
271     : AsmPrinter(os, t) {}
272
273   const char *getPassName() const {
274     return "Output Sparc Assembly for Functions";
275   }
276
277   virtual bool doInitialization(Module &M) {
278     startModule(M);
279     return false;
280   }
281
282   virtual bool runOnFunction(Function &F) {
283     startFunction(F);
284     emitFunction(F);
285     endFunction(F);
286     return false;
287   }
288
289   virtual bool doFinalization(Module &M) {
290     endModule();
291     return false;
292   }
293
294   virtual void getAnalysisUsage(AnalysisUsage &AU) const {
295     AU.setPreservesAll();
296   }
297
298   void emitFunction(const Function &F);
299 private :
300   void emitBasicBlock(const MachineBasicBlock &MBB);
301   void emitMachineInst(const MachineInstr *MI);
302   
303   unsigned int printOperands(const MachineInstr *MI, unsigned int opNum);
304   void printOneOperand(const MachineOperand &Op);
305
306   bool OpIsBranchTargetLabel(const MachineInstr *MI, unsigned int opNum);
307   bool OpIsMemoryAddressBase(const MachineInstr *MI, unsigned int opNum);
308   
309   unsigned getOperandMask(unsigned Opcode) {
310     switch (Opcode) {
311     case SUBcc:   return 1 << 3;  // Remove CC argument
312   //case BA:      return 1 << 0;  // Remove Arg #0, which is always null or xcc
313     default:      return 0;       // By default, don't hack operands...
314     }
315   }
316 };
317
318 inline bool
319 SparcFunctionAsmPrinter::OpIsBranchTargetLabel(const MachineInstr *MI,
320                                                unsigned int opNum) {
321   switch (MI->getOpCode()) {
322   case JMPLCALL:
323   case JMPLRET: return (opNum == 0);
324   default:      return false;
325   }
326 }
327
328
329 inline bool
330 SparcFunctionAsmPrinter::OpIsMemoryAddressBase(const MachineInstr *MI,
331                                                unsigned int opNum) {
332   if (Target.getInstrInfo().isLoad(MI->getOpCode()))
333     return (opNum == 0);
334   else if (Target.getInstrInfo().isStore(MI->getOpCode()))
335     return (opNum == 1);
336   else
337     return false;
338 }
339
340
341 #define PrintOp1PlusOp2(mop1, mop2) \
342   printOneOperand(mop1); \
343   toAsm << "+"; \
344   printOneOperand(mop2);
345
346 unsigned int
347 SparcFunctionAsmPrinter::printOperands(const MachineInstr *MI,
348                                unsigned int opNum)
349 {
350   const MachineOperand& mop = MI->getOperand(opNum);
351   
352   if (OpIsBranchTargetLabel(MI, opNum))
353     {
354       PrintOp1PlusOp2(mop, MI->getOperand(opNum+1));
355       return 2;
356     }
357   else if (OpIsMemoryAddressBase(MI, opNum))
358     {
359       toAsm << "[";
360       PrintOp1PlusOp2(mop, MI->getOperand(opNum+1));
361       toAsm << "]";
362       return 2;
363     }
364   else
365     {
366       printOneOperand(mop);
367       return 1;
368     }
369 }
370
371
372 void
373 SparcFunctionAsmPrinter::printOneOperand(const MachineOperand &mop)
374 {
375   bool needBitsFlag = true;
376   
377   if (mop.opHiBits32())
378     toAsm << "%lm(";
379   else if (mop.opLoBits32())
380     toAsm << "%lo(";
381   else if (mop.opHiBits64())
382     toAsm << "%hh(";
383   else if (mop.opLoBits64())
384     toAsm << "%hm(";
385   else
386     needBitsFlag = false;
387   
388   switch (mop.getType())
389     {
390     case MachineOperand::MO_VirtualRegister:
391     case MachineOperand::MO_CCRegister:
392     case MachineOperand::MO_MachineRegister:
393       {
394         int RegNum = (int)mop.getAllocatedRegNum();
395         
396         // better to print code with NULL registers than to die
397         if (RegNum == Target.getRegInfo().getInvalidRegNum()) {
398           toAsm << "<NULL VALUE>";
399         } else {
400           toAsm << "%" << Target.getRegInfo().getUnifiedRegName(RegNum);
401         }
402         break;
403       }
404     
405     case MachineOperand::MO_PCRelativeDisp:
406       {
407         const Value *Val = mop.getVRegValue();
408         assert(Val && "\tNULL Value in SparcFunctionAsmPrinter");
409         
410         if (const BasicBlock *BB = dyn_cast<const BasicBlock>(Val))
411           toAsm << getID(BB);
412         else if (const Function *M = dyn_cast<Function>(Val))
413           toAsm << getID(M);
414         else if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(Val))
415           toAsm << getID(GV);
416         else if (const Constant *CV = dyn_cast<Constant>(Val))
417           toAsm << getID(CV);
418         else
419           assert(0 && "Unrecognized value in SparcFunctionAsmPrinter");
420         break;
421       }
422     
423     case MachineOperand::MO_SignExtendedImmed:
424       toAsm << mop.getImmedValue();
425       break;
426
427     case MachineOperand::MO_UnextendedImmed:
428       toAsm << (uint64_t) mop.getImmedValue();
429       break;
430     
431     default:
432       toAsm << mop;      // use dump field
433       break;
434     }
435   
436   if (needBitsFlag)
437     toAsm << ")";
438 }
439
440
441 void
442 SparcFunctionAsmPrinter::emitMachineInst(const MachineInstr *MI)
443 {
444   unsigned Opcode = MI->getOpCode();
445
446   if (Target.getInstrInfo().isDummyPhiInstr(Opcode));
447     return;  // IGNORE PHI NODES
448
449   toAsm << "\t" << Target.getInstrInfo().getName(Opcode) << "\t";
450
451   unsigned Mask = getOperandMask(Opcode);
452   
453   bool NeedComma = false;
454   unsigned N = 1;
455   for (unsigned OpNum = 0; OpNum < MI->getNumOperands(); OpNum += N)
456     if (! ((1 << OpNum) & Mask)) {        // Ignore this operand?
457       if (NeedComma) toAsm << ", ";         // Handle comma outputing
458       NeedComma = true;
459       N = printOperands(MI, OpNum);
460     }
461   else
462     N = 1;
463   
464   toAsm << "\n";
465 }
466
467 void
468 SparcFunctionAsmPrinter::emitBasicBlock(const MachineBasicBlock &MBB)
469 {
470   // Emit a label for the basic block
471   toAsm << getID(MBB.getBasicBlock()) << ":\n";
472
473   // Loop over all of the instructions in the basic block...
474   for (MachineBasicBlock::const_iterator MII = MBB.begin(), MIE = MBB.end();
475        MII != MIE; ++MII)
476     emitMachineInst(*MII);
477   toAsm << "\n";  // Seperate BB's with newlines
478 }
479
480 void
481 SparcFunctionAsmPrinter::emitFunction(const Function &F)
482 {
483   string methName = getID(&F);
484   toAsm << "!****** Outputing Function: " << methName << " ******\n";
485   enterSection(AsmPrinter::Text);
486   toAsm << "\t.align\t4\n\t.global\t" << methName << "\n";
487   //toAsm << "\t.type\t" << methName << ",#function\n";
488   toAsm << "\t.type\t" << methName << ", 2\n";
489   toAsm << methName << ":\n";
490
491   // Output code for all of the basic blocks in the function...
492   MachineFunction &MF = MachineFunction::get(&F);
493   for (MachineFunction::const_iterator I = MF.begin(), E = MF.end(); I != E; ++I)
494     emitBasicBlock(*I);
495
496   // Output a .size directive so the debugger knows the extents of the function
497   toAsm << ".EndOf_" << methName << ":\n\t.size "
498            << methName << ", .EndOf_"
499            << methName << "-" << methName << "\n";
500
501   // Put some spaces between the functions
502   toAsm << "\n\n";
503 }
504
505 }  // End anonymous namespace
506
507 Pass *UltraSparc::getFunctionAsmPrinterPass(std::ostream &Out) {
508   return new SparcFunctionAsmPrinter(Out, *this);
509 }
510
511
512
513
514
515 //===----------------------------------------------------------------------===//
516 //   SparcFunctionAsmPrinter Code
517 //===----------------------------------------------------------------------===//
518
519 namespace {
520
521 class SparcModuleAsmPrinter : public Pass, public AsmPrinter {
522 public:
523   SparcModuleAsmPrinter(std::ostream &os, TargetMachine &t)
524     : AsmPrinter(os, t) {}
525
526   const char *getPassName() const { return "Output Sparc Assembly for Module"; }
527
528   virtual bool run(Module &M) {
529     startModule(M);
530     emitGlobalsAndConstants(M);
531     endModule();
532     return false;
533   }
534
535   virtual void getAnalysisUsage(AnalysisUsage &AU) const {
536     AU.setPreservesAll();
537   }
538
539 private:
540   void emitGlobalsAndConstants  (const Module &M);
541
542   void printGlobalVariable      (const GlobalVariable *GV);
543   void PrintZeroBytesToPad      (int numBytes);
544   void printSingleConstantValue (const Constant* CV);
545   void printConstantValueOnly   (const Constant* CV, int numPadBytes = 0);
546   void printConstant            (const Constant* CV, string valID = "");
547
548   static void FoldConstants     (const Module &M,
549                                  hash_set<const Constant*> &moduleConstants);
550 };
551
552
553 // Can we treat the specified array as a string?  Only if it is an array of
554 // ubytes or non-negative sbytes.
555 //
556 static bool isStringCompatible(const ConstantArray *CVA) {
557   const Type *ETy = cast<ArrayType>(CVA->getType())->getElementType();
558   if (ETy == Type::UByteTy) return true;
559   if (ETy != Type::SByteTy) return false;
560
561   for (unsigned i = 0; i < CVA->getNumOperands(); ++i)
562     if (cast<ConstantSInt>(CVA->getOperand(i))->getValue() < 0)
563       return false;
564
565   return true;
566 }
567
568 // toOctal - Convert the low order bits of X into an octal letter
569 static inline char toOctal(int X) {
570   return (X&7)+'0';
571 }
572
573 // getAsCString - Return the specified array as a C compatible string, only if
574 // the predicate isStringCompatible is true.
575 //
576 static string getAsCString(const ConstantArray *CVA) {
577   assert(isStringCompatible(CVA) && "Array is not string compatible!");
578
579   string Result;
580   const Type *ETy = cast<ArrayType>(CVA->getType())->getElementType();
581   Result = "\"";
582   for (unsigned i = 0; i < CVA->getNumOperands(); ++i) {
583     unsigned char C = (ETy == Type::SByteTy) ?
584       (unsigned char)cast<ConstantSInt>(CVA->getOperand(i))->getValue() :
585       (unsigned char)cast<ConstantUInt>(CVA->getOperand(i))->getValue();
586
587     if (C == '"') {
588       Result += "\\\"";
589     } else if (C == '\\') {
590       Result += "\\\\";
591     } else if (isprint(C)) {
592       Result += C;
593     } else {
594       switch(C) {
595       case '\a': Result += "\\a"; break;
596       case '\b': Result += "\\b"; break;
597       case '\f': Result += "\\f"; break;
598       case '\n': Result += "\\n"; break;
599       case '\r': Result += "\\r"; break;
600       case '\t': Result += "\\t"; break;
601       case '\v': Result += "\\v"; break;
602       default:
603         Result += '\\';
604         Result += toOctal(C >> 6);
605         Result += toOctal(C >> 3);
606         Result += toOctal(C >> 0);
607         break;
608       }
609     }
610   }
611   Result += "\"";
612
613   return Result;
614 }
615
616 inline bool
617 ArrayTypeIsString(const ArrayType* arrayType)
618 {
619   return (arrayType->getElementType() == Type::UByteTy ||
620           arrayType->getElementType() == Type::SByteTy);
621 }
622
623
624 inline const string
625 TypeToDataDirective(const Type* type)
626 {
627   switch(type->getPrimitiveID())
628     {
629     case Type::BoolTyID: case Type::UByteTyID: case Type::SByteTyID:
630       return ".byte";
631     case Type::UShortTyID: case Type::ShortTyID:
632       return ".half";
633     case Type::UIntTyID: case Type::IntTyID:
634       return ".word";
635     case Type::ULongTyID: case Type::LongTyID: case Type::PointerTyID:
636       return ".xword";
637     case Type::FloatTyID:
638       return ".word";
639     case Type::DoubleTyID:
640       return ".xword";
641     case Type::ArrayTyID:
642       if (ArrayTypeIsString((ArrayType*) type))
643         return ".ascii";
644       else
645         return "<InvaliDataTypeForPrinting>";
646     default:
647       return "<InvaliDataTypeForPrinting>";
648     }
649 }
650
651 // Get the size of the type
652 // 
653 inline unsigned int
654 TypeToSize(const Type* type, const TargetMachine& target)
655 {
656   return target.findOptimalStorageSize(type);
657 }
658
659 // Get the size of the constant for the given target.
660 // If this is an unsized array, return 0.
661 // 
662 inline unsigned int
663 ConstantToSize(const Constant* CV, const TargetMachine& target)
664 {
665   if (const ConstantArray* CVA = dyn_cast<ConstantArray>(CV))
666     {
667       const ArrayType *aty = cast<ArrayType>(CVA->getType());
668       if (ArrayTypeIsString(aty))
669         return 1 + CVA->getNumOperands();
670     }
671   
672   return TypeToSize(CV->getType(), target);
673 }
674
675 // Align data larger than one L1 cache line on L1 cache line boundaries.
676 // Align all smaller data on the next higher 2^x boundary (4, 8, ...).
677 // 
678 inline unsigned int
679 SizeToAlignment(unsigned int size, const TargetMachine& target)
680 {
681   unsigned short cacheLineSize = target.getCacheInfo().getCacheLineSize(1); 
682   if (size > (unsigned) cacheLineSize / 2)
683     return cacheLineSize;
684   else
685     for (unsigned sz=1; /*no condition*/; sz *= 2)
686       if (sz >= size)
687         return sz;
688 }
689
690 // Get the size of the type and then use SizeToAlignment.
691 // 
692 inline unsigned int
693 TypeToAlignment(const Type* type, const TargetMachine& target)
694 {
695   return SizeToAlignment(TypeToSize(type, target), target);
696 }
697
698 // Get the size of the constant and then use SizeToAlignment.
699 // Handles strings as a special case;
700 inline unsigned int
701 ConstantToAlignment(const Constant* CV, const TargetMachine& target)
702 {
703   if (const ConstantArray* CVA = dyn_cast<ConstantArray>(CV))
704     if (ArrayTypeIsString(cast<ArrayType>(CVA->getType())))
705       return SizeToAlignment(1 + CVA->getNumOperands(), target);
706   
707   return TypeToAlignment(CV->getType(), target);
708 }
709
710
711 // Print a single constant value.
712 void
713 SparcModuleAsmPrinter::printSingleConstantValue(const Constant* CV)
714 {
715   assert(CV->getType() != Type::VoidTy &&
716          CV->getType() != Type::TypeTy &&
717          CV->getType() != Type::LabelTy &&
718          "Unexpected type for Constant");
719   
720   assert((!isa<ConstantArray>(CV) && ! isa<ConstantStruct>(CV))
721          && "Aggregate types should be handled outside this function");
722   
723   toAsm << "\t" << TypeToDataDirective(CV->getType()) << "\t";
724   
725   if (CV->getType()->isPrimitiveType())
726     {
727       if (CV->getType()->isFloatingPoint()) {
728         // FP Constants are printed as integer constants to avoid losing
729         // precision...
730         double Val = cast<ConstantFP>(CV)->getValue();
731         if (CV->getType() == Type::FloatTy) {
732           float FVal = (float)Val;
733           char *ProxyPtr = (char*)&FVal;        // Abide by C TBAA rules
734           toAsm << *(unsigned int*)ProxyPtr;            
735         } else if (CV->getType() == Type::DoubleTy) {
736           char *ProxyPtr = (char*)&Val;         // Abide by C TBAA rules
737           toAsm << *(uint64_t*)ProxyPtr;            
738         } else {
739           assert(0 && "Unknown floating point type!");
740         }
741         
742         toAsm << "\t! " << CV->getType()->getDescription()
743               << " value: " << Val << "\n";
744       } else {
745         WriteAsOperand(toAsm, CV, false, false) << "\n";
746       }
747     }
748   else if (const ConstantPointerRef* CPR = dyn_cast<ConstantPointerRef>(CV))
749     { // This is a constant address for a global variable or method.
750       // Use the name of the variable or method as the address value.
751       toAsm << getID(CPR->getValue()) << "\n";
752     }
753   else if (isa<ConstantPointerNull>(CV))
754     { // Null pointer value
755       toAsm << "0\n";
756     }
757   else if (const ConstantExpr* CE = dyn_cast<ConstantExpr>(CV))
758     { // Constant expression built from operators, constants, and symbolic addrs
759       toAsm << ConstantExprToString(CE, Target) << "\n";
760     }
761   else
762     {
763       assert(0 && "Unknown elementary type for constant");
764     }
765 }
766
767 void
768 SparcModuleAsmPrinter::PrintZeroBytesToPad(int numBytes)
769 {
770   for ( ; numBytes >= 8; numBytes -= 8)
771     printSingleConstantValue(Constant::getNullValue(Type::ULongTy));
772
773   if (numBytes >= 4)
774     {
775       printSingleConstantValue(Constant::getNullValue(Type::UIntTy));
776       numBytes -= 4;
777     }
778
779   while (numBytes--)
780     printSingleConstantValue(Constant::getNullValue(Type::UByteTy));
781 }
782
783 // Print a constant value or values (it may be an aggregate).
784 // Uses printSingleConstantValue() to print each individual value.
785 void
786 SparcModuleAsmPrinter::printConstantValueOnly(const Constant* CV,
787                                               int numPadBytes /* = 0*/)
788 {
789   const ConstantArray *CVA = dyn_cast<ConstantArray>(CV);
790
791   if (numPadBytes)
792     PrintZeroBytesToPad(numPadBytes);
793
794   if (CVA && isStringCompatible(CVA))
795     { // print the string alone and return
796       toAsm << "\t" << ".ascii" << "\t" << getAsCString(CVA) << "\n";
797     }
798   else if (CVA)
799     { // Not a string.  Print the values in successive locations
800       const std::vector<Use> &constValues = CVA->getValues();
801       for (unsigned i=0; i < constValues.size(); i++)
802         printConstantValueOnly(cast<Constant>(constValues[i].get()));
803     }
804   else if (const ConstantStruct *CVS = dyn_cast<ConstantStruct>(CV))
805     { // Print the fields in successive locations. Pad to align if needed!
806       const StructLayout *cvsLayout =
807         Target.DataLayout.getStructLayout(CVS->getType());
808       const std::vector<Use>& constValues = CVS->getValues();
809       unsigned sizeSoFar = 0;
810       for (unsigned i=0, N = constValues.size(); i < N; i++)
811         {
812           const Constant* field = cast<Constant>(constValues[i].get());
813
814           // Check if padding is needed and insert one or more 0s.
815           unsigned fieldSize = Target.DataLayout.getTypeSize(field->getType());
816           int padSize = ((i == N-1? cvsLayout->StructSize
817                                   : cvsLayout->MemberOffsets[i+1])
818                          - cvsLayout->MemberOffsets[i]) - fieldSize;
819           sizeSoFar += (fieldSize + padSize);
820
821           // Now print the actual field value
822           printConstantValueOnly(field, padSize);
823         }
824       assert(sizeSoFar == cvsLayout->StructSize &&
825              "Layout of constant struct may be incorrect!");
826     }
827   else
828     printSingleConstantValue(CV);
829 }
830
831 // Print a constant (which may be an aggregate) prefixed by all the
832 // appropriate directives.  Uses printConstantValueOnly() to print the
833 // value or values.
834 void
835 SparcModuleAsmPrinter::printConstant(const Constant* CV, string valID)
836 {
837   if (valID.length() == 0)
838     valID = getID(CV);
839   
840   toAsm << "\t.align\t" << ConstantToAlignment(CV, Target) << "\n";
841   
842   // Print .size and .type only if it is not a string.
843   const ConstantArray *CVA = dyn_cast<ConstantArray>(CV);
844   if (CVA && isStringCompatible(CVA))
845     { // print it as a string and return
846       toAsm << valID << ":\n";
847       toAsm << "\t" << ".ascii" << "\t" << getAsCString(CVA) << "\n";
848       return;
849     }
850   
851   toAsm << "\t.type" << "\t" << valID << ",#object\n";
852
853   unsigned int constSize = ConstantToSize(CV, Target);
854   if (constSize)
855     toAsm << "\t.size" << "\t" << valID << "," << constSize << "\n";
856   
857   toAsm << valID << ":\n";
858   
859   printConstantValueOnly(CV);
860 }
861
862
863 void SparcModuleAsmPrinter::FoldConstants(const Module &M,
864                                           hash_set<const Constant*> &MC) {
865   for (Module::const_iterator I = M.begin(), E = M.end(); I != E; ++I)
866     if (!I->isExternal()) {
867       const hash_set<const Constant*> &pool =
868         MachineFunction::get(I).getConstantPoolValues();
869       MC.insert(pool.begin(), pool.end());
870     }
871 }
872
873 void SparcModuleAsmPrinter::printGlobalVariable(const GlobalVariable* GV)
874 {
875   if (GV->hasExternalLinkage())
876     toAsm << "\t.global\t" << getID(GV) << "\n";
877   
878   if (GV->hasInitializer() && ! GV->getInitializer()->isNullValue())
879     printConstant(GV->getInitializer(), getID(GV));
880   else {
881     toAsm << "\t.align\t" << TypeToAlignment(GV->getType()->getElementType(),
882                                                 Target) << "\n";
883     toAsm << "\t.type\t" << getID(GV) << ",#object\n";
884     toAsm << "\t.reserve\t" << getID(GV) << ","
885           << TypeToSize(GV->getType()->getElementType(), Target)
886           << "\n";
887   }
888 }
889
890
891 void SparcModuleAsmPrinter::emitGlobalsAndConstants(const Module &M) {
892   // First, get the constants there were marked by the code generator for
893   // inclusion in the assembly code data area and fold them all into a
894   // single constant pool since there may be lots of duplicates.  Also,
895   // lets force these constants into the slot table so that we can get
896   // unique names for unnamed constants also.
897   // 
898   hash_set<const Constant*> moduleConstants;
899   FoldConstants(M, moduleConstants);
900     
901   // Output constants spilled to memory
902   enterSection(AsmPrinter::ReadOnlyData);
903   for (hash_set<const Constant*>::const_iterator I = moduleConstants.begin(),
904          E = moduleConstants.end();  I != E; ++I)
905     printConstant(*I);
906
907   // Output global variables...
908   for (Module::const_giterator GI = M.gbegin(), GE = M.gend(); GI != GE; ++GI)
909     if (! GI->isExternal()) {
910       assert(GI->hasInitializer());
911       if (GI->isConstant())
912         enterSection(AsmPrinter::ReadOnlyData);   // read-only, initialized data
913       else if (GI->getInitializer()->isNullValue())
914         enterSection(AsmPrinter::ZeroInitRWData); // read-write zero data
915       else
916         enterSection(AsmPrinter::InitRWData);     // read-write non-zero data
917
918       printGlobalVariable(GI);
919     }
920
921   toAsm << "\n";
922 }
923
924 }  // End anonymous namespace
925
926 Pass *UltraSparc::getModuleAsmPrinterPass(std::ostream &Out) {
927   return new SparcModuleAsmPrinter(Out, *this);
928 }