blob: 7b926c42ff73e234a3608593a0d72bff78790fe6 [file] [log] [blame]
Chris Lattner16c7bb22002-05-09 02:28:59 +00001//===-- Writer.cpp - Library for writing C files --------------------------===//
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +00002//
3// This library implements the functionality defined in llvm/Assembly/CWriter.h
4// and CLocalVars.h
5//
6// TODO : Recursive types.
Chris Lattner16c7bb22002-05-09 02:28:59 +00007//
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +00008//===-----------------------------------------------------------------------==//
Chris Lattner16c7bb22002-05-09 02:28:59 +00009
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +000010#include "llvm/Assembly/CWriter.h"
11#include "CLocalVars.h"
12#include "llvm/SlotCalculator.h"
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +000013#include "llvm/Constants.h"
Chris Lattner2f5f51a2002-05-09 03:28:37 +000014#include "llvm/DerivedTypes.h"
15#include "llvm/Module.h"
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +000016#include "llvm/GlobalVariable.h"
Chris Lattner2f5f51a2002-05-09 03:28:37 +000017#include "llvm/Function.h"
18#include "llvm/Argument.h"
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +000019#include "llvm/BasicBlock.h"
20#include "llvm/iMemory.h"
21#include "llvm/iTerminators.h"
22#include "llvm/iPHINode.h"
23#include "llvm/iOther.h"
Chris Lattner2f5f51a2002-05-09 03:28:37 +000024#include "llvm/iOperators.h"
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +000025#include "llvm/SymbolTable.h"
26#include "llvm/Support/InstVisitor.h"
27#include "Support/StringExtras.h"
28#include "Support/STLExtras.h"
29
30#include <algorithm>
31#include <strstream>
32using std::string;
33using std::map;
34using std::vector;
35using std::ostream;
36
Chris Lattner16c7bb22002-05-09 02:28:59 +000037//===-----------------------------------------------------------------------==//
38//
39// Implementation of the CLocalVars methods
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +000040
41// Appends a variable to the LocalVars map if it does not already exist
42// Also check that the type exists on the map.
43void CLocalVars::addLocalVar(const Type *t, const string & var) {
44 if (!LocalVars.count(t) ||
45 find(LocalVars[t].begin(), LocalVars[t].end(), var)
46 == LocalVars[t].end()) {
47 LocalVars[t].push_back(var);
48 }
49}
50
Chris Lattner16c7bb22002-05-09 02:28:59 +000051static string calcTypeNameVar(const Type *Ty,
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +000052 map<const Type *, string> &TypeNames,
53 string VariableName, string NameSoFar);
54
55static std::string getConstStrValue(const Constant* CPV);
56
57
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +000058static std::string getConstArrayStrValue(const Constant* CPV) {
59 std::string Result;
60
61 // As a special case, print the array as a string if it is an array of
62 // ubytes or an array of sbytes with positive values.
63 //
64 const Type *ETy = cast<ArrayType>(CPV->getType())->getElementType();
65 bool isString = (ETy == Type::SByteTy || ETy == Type::UByteTy);
66
67 if (ETy == Type::SByteTy) {
68 for (unsigned i = 0; i < CPV->getNumOperands(); ++i)
69 if (ETy == Type::SByteTy &&
70 cast<ConstantSInt>(CPV->getOperand(i))->getValue() < 0) {
71 isString = false;
72 break;
73 }
74 }
Chris Lattner2f5eb4e2002-05-09 03:56:52 +000075 if (isString) {
76 // Make sure the last character is a null char, as automatically added by C
77 if (CPV->getNumOperands() == 0 ||
78 !cast<Constant>(*(CPV->op_end()-1))->isNullValue())
79 isString = false;
80 }
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +000081
82 if (isString) {
83 Result = "\"";
Chris Lattner2f5eb4e2002-05-09 03:56:52 +000084 // Do not include the last character, which we know is null
85 for (unsigned i = 0, e = CPV->getNumOperands()-1; i != e; ++i) {
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +000086 unsigned char C = (ETy == Type::SByteTy) ?
87 (unsigned char)cast<ConstantSInt>(CPV->getOperand(i))->getValue() :
88 (unsigned char)cast<ConstantUInt>(CPV->getOperand(i))->getValue();
89
90 if (isprint(C)) {
91 Result += C;
92 } else {
Chris Lattner2f5eb4e2002-05-09 03:56:52 +000093 switch (C) {
94 case '\n': Result += "\\n"; break;
95 case '\t': Result += "\\t"; break;
96 case '\r': Result += "\\r"; break;
97 case '\v': Result += "\\v"; break;
98 case '\a': Result += "\\a"; break;
99 default:
100 Result += "\\x";
101 Result += ( C/16 < 10) ? ( C/16 +'0') : ( C/16 -10+'A');
102 Result += ((C&15) < 10) ? ((C&15)+'0') : ((C&15)-10+'A');
103 break;
104 }
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000105 }
106 }
107 Result += "\"";
108
109 } else {
110 Result = "{";
111 if (CPV->getNumOperands()) {
112 Result += " " + getConstStrValue(cast<Constant>(CPV->getOperand(0)));
113 for (unsigned i = 1; i < CPV->getNumOperands(); i++)
114 Result += ", " + getConstStrValue(cast<Constant>(CPV->getOperand(i)));
115 }
116 Result += " }";
117 }
118
119 return Result;
120}
121
122static std::string getConstStructStrValue(const Constant* CPV) {
123 std::string Result = "{";
124 if (CPV->getNumOperands()) {
125 Result += " " + getConstStrValue(cast<Constant>(CPV->getOperand(0)));
126 for (unsigned i = 1; i < CPV->getNumOperands(); i++)
127 Result += ", " + getConstStrValue(cast<Constant>(CPV->getOperand(i)));
128 }
129
130 return Result + " }";
131}
132
133// our own getStrValue function for constant initializers
134static std::string getConstStrValue(const Constant* CPV) {
135 // Does not handle null pointers, that needs to be checked explicitly
136 string tempstr;
137 if (CPV == ConstantBool::False)
138 return "0";
139 else if (CPV == ConstantBool::True)
140 return "1";
141
142 else if (isa<ConstantArray>(CPV)) {
143 tempstr = getConstArrayStrValue(CPV);
144 }
145 else if (isa<ConstantStruct>(CPV)) {
146 tempstr = getConstStructStrValue(CPV);
147 }
148 else if (ConstantUInt *CUI = dyn_cast<ConstantUInt>(CPV)) {
Chris Lattner16c7bb22002-05-09 02:28:59 +0000149 tempstr = utostr(CUI->getValue());
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000150 }
151 else if (ConstantSInt *CSI = dyn_cast<ConstantSInt>(CPV)) {
152 tempstr = itostr(CSI->getValue());
153 }
154 else if (ConstantFP *CFP = dyn_cast<ConstantFP>(CPV)) {
155 tempstr = ftostr(CFP->getValue());
156 }
157
158 if (CPV->getType() == Type::ULongTy)
159 tempstr += "ull";
160 else if (CPV->getType() == Type::LongTy)
161 tempstr += "ll";
162 else if (CPV->getType() == Type::UIntTy ||
163 CPV->getType() == Type::UShortTy)
164 tempstr += "u";
165
166 return tempstr;
167
168}
169
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000170// Internal function
171// Essentially pass the Type* variable, an empty typestack and this prints
172// out the C type
Chris Lattner16c7bb22002-05-09 02:28:59 +0000173static string calcTypeName(const Type *Ty, map<const Type *, string> &TypeNames,
174 string &FunctionInfo) {
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000175
176 // Takin' care of the fact that boolean would be int in C
177 // and that ushort would be unsigned short etc.
178
179 // Base Case
180 if (Ty->isPrimitiveType())
181 switch (Ty->getPrimitiveID()) {
Chris Lattner16c7bb22002-05-09 02:28:59 +0000182 case Type::VoidTyID: return "void";
183 case Type::BoolTyID: return "bool";
184 case Type::UByteTyID: return "unsigned char";
185 case Type::SByteTyID: return "signed char";
186 case Type::UShortTyID: return "unsigned short";
187 case Type::ShortTyID: return "short";
188 case Type::UIntTyID: return "unsigned";
189 case Type::IntTyID: return "int";
190 case Type::ULongTyID: return "unsigned long long";
191 case Type::LongTyID: return "signed long long";
192 case Type::FloatTyID: return "float";
193 case Type::DoubleTyID: return "double";
194 default : assert(0 && "Unknown primitive type!");
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000195 }
196
197 // Check to see if the type is named.
198 map<const Type *, string>::iterator I = TypeNames.find(Ty);
199 if (I != TypeNames.end())
200 return I->second;
201
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000202 string Result;
203 string MInfo = "";
204 switch (Ty->getPrimitiveID()) {
205 case Type::FunctionTyID: {
206 const FunctionType *MTy = cast<const FunctionType>(Ty);
Chris Lattner16c7bb22002-05-09 02:28:59 +0000207 Result = calcTypeName(MTy->getReturnType(), TypeNames, MInfo);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000208 if (MInfo != "")
209 Result += ") " + MInfo;
210 Result += "(";
Chris Lattner16c7bb22002-05-09 02:28:59 +0000211 FunctionInfo += " (";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000212 for (FunctionType::ParamTypes::const_iterator
213 I = MTy->getParamTypes().begin(),
214 E = MTy->getParamTypes().end(); I != E; ++I) {
215 if (I != MTy->getParamTypes().begin())
Chris Lattner16c7bb22002-05-09 02:28:59 +0000216 FunctionInfo += ", ";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000217 MInfo = "";
Chris Lattner16c7bb22002-05-09 02:28:59 +0000218 FunctionInfo += calcTypeName(*I, TypeNames, MInfo);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000219 if (MInfo != "")
220 Result += ") " + MInfo;
221 }
222 if (MTy->isVarArg()) {
223 if (!MTy->getParamTypes().empty())
Chris Lattner16c7bb22002-05-09 02:28:59 +0000224 FunctionInfo += ", ";
225 FunctionInfo += "...";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000226 }
Chris Lattner16c7bb22002-05-09 02:28:59 +0000227 FunctionInfo += ")";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000228 break;
229 }
230 case Type::StructTyID: {
231 string tempstr = "";
232 const StructType *STy = cast<const StructType>(Ty);
233 Result = " struct {\n ";
234 int indx = 0;
235 for (StructType::ElementTypes::const_iterator
236 I = STy->getElementTypes().begin(),
237 E = STy->getElementTypes().end(); I != E; ++I) {
Chris Lattner16c7bb22002-05-09 02:28:59 +0000238 Result += calcTypeNameVar(*I, TypeNames,
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000239 "field" + itostr(indx++), tempstr);
240 Result += ";\n ";
241 }
242 Result += " } ";
243 break;
244 }
245 case Type::PointerTyID:
246 Result = calcTypeName(cast<const PointerType>(Ty)->getElementType(),
Chris Lattner16c7bb22002-05-09 02:28:59 +0000247 TypeNames, MInfo);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000248 Result += "*";
249 break;
250 case Type::ArrayTyID: {
251 const ArrayType *ATy = cast<const ArrayType>(Ty);
252 int NumElements = ATy->getNumElements();
Chris Lattner16c7bb22002-05-09 02:28:59 +0000253 Result = calcTypeName(ATy->getElementType(), TypeNames, MInfo);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000254 Result += "*";
255 break;
256 }
257 default:
258 assert(0 && "Unhandled case in getTypeProps!");
259 Result = "<error>";
260 }
261
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000262 return Result;
263}
264
265// Internal function
266// Pass the Type* variable and and the variable name and this prints out the
267// variable declaration.
268// This is different from calcTypeName because if you need to declare an array
269// the size of the array would appear after the variable name itself
270// For eg. int a[10];
Chris Lattner16c7bb22002-05-09 02:28:59 +0000271static string calcTypeNameVar(const Type *Ty,
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000272 map<const Type *, string> &TypeNames,
273 string VariableName, string NameSoFar) {
274 if (Ty->isPrimitiveType())
275 switch (Ty->getPrimitiveID()) {
276 case Type::BoolTyID:
Chris Lattner16c7bb22002-05-09 02:28:59 +0000277 return "bool " + NameSoFar + VariableName;
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000278 case Type::UByteTyID:
279 return "unsigned char " + NameSoFar + VariableName;
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000280 case Type::SByteTyID:
281 return "signed char " + NameSoFar + VariableName;
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000282 case Type::UShortTyID:
283 return "unsigned long long " + NameSoFar + VariableName;
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000284 case Type::ULongTyID:
285 return "unsigned long long " + NameSoFar + VariableName;
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000286 case Type::LongTyID:
287 return "signed long long " + NameSoFar + VariableName;
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000288 case Type::UIntTyID:
Chris Lattner16c7bb22002-05-09 02:28:59 +0000289 return "unsigned " + NameSoFar + VariableName;
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000290 default :
291 return Ty->getDescription() + " " + NameSoFar + VariableName;
292 }
293
294 // Check to see if the type is named.
295 map<const Type *, string>::iterator I = TypeNames.find(Ty);
296 if (I != TypeNames.end())
297 return I->second + " " + NameSoFar + VariableName;
298
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000299 string Result;
300 string tempstr = "";
301
302 switch (Ty->getPrimitiveID()) {
303 case Type::FunctionTyID: {
304 string MInfo = "";
305 const FunctionType *MTy = cast<const FunctionType>(Ty);
Chris Lattner16c7bb22002-05-09 02:28:59 +0000306 Result += calcTypeName(MTy->getReturnType(), TypeNames, MInfo);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000307 if (MInfo != "")
308 Result += ") " + MInfo;
309 Result += " " + NameSoFar + VariableName;
310 Result += " (";
311 for (FunctionType::ParamTypes::const_iterator
312 I = MTy->getParamTypes().begin(),
313 E = MTy->getParamTypes().end(); I != E; ++I) {
314 if (I != MTy->getParamTypes().begin())
315 Result += ", ";
316 MInfo = "";
Chris Lattner16c7bb22002-05-09 02:28:59 +0000317 Result += calcTypeName(*I, TypeNames, MInfo);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000318 if (MInfo != "")
319 Result += ") " + MInfo;
320 }
321 if (MTy->isVarArg()) {
322 if (!MTy->getParamTypes().empty())
323 Result += ", ";
324 Result += "...";
325 }
326 Result += ")";
327 break;
328 }
329 case Type::StructTyID: {
330 const StructType *STy = cast<const StructType>(Ty);
331 Result = " struct {\n ";
332 int indx = 0;
333 for (StructType::ElementTypes::const_iterator
334 I = STy->getElementTypes().begin(),
335 E = STy->getElementTypes().end(); I != E; ++I) {
Chris Lattner16c7bb22002-05-09 02:28:59 +0000336 Result += calcTypeNameVar(*I, TypeNames,
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000337 "field" + itostr(indx++), "");
338 Result += ";\n ";
339 }
340 Result += " }";
341 Result += " " + NameSoFar + VariableName;
342 break;
343 }
344
345 case Type::PointerTyID: {
346 Result = calcTypeNameVar(cast<const PointerType>(Ty)->getElementType(),
Chris Lattner16c7bb22002-05-09 02:28:59 +0000347 TypeNames, tempstr,
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000348 "(*" + NameSoFar + VariableName + ")");
349 break;
350 }
351
352 case Type::ArrayTyID: {
353 const ArrayType *ATy = cast<const ArrayType>(Ty);
354 int NumElements = ATy->getNumElements();
Chris Lattner16c7bb22002-05-09 02:28:59 +0000355 Result = calcTypeNameVar(ATy->getElementType(), TypeNames,
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000356 tempstr, NameSoFar + VariableName + "[" +
357 itostr(NumElements) + "]");
358 break;
359 }
360 default:
361 assert(0 && "Unhandled case in getTypeProps!");
362 Result = "<error>";
363 }
364
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000365 return Result;
366}
367
368// printTypeVarInt - The internal guts of printing out a type that has a
369// potentially named portion and the variable associated with the type.
370static ostream &printTypeVarInt(ostream &Out, const Type *Ty,
371 map<const Type *, string> &TypeNames,
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000372 const string &VariableName) {
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000373 // Primitive types always print out their description, regardless of whether
374 // they have been named or not.
375
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000376 if (Ty->isPrimitiveType())
377 switch (Ty->getPrimitiveID()) {
378 case Type::BoolTyID:
Chris Lattner16c7bb22002-05-09 02:28:59 +0000379 return Out << "bool " << VariableName;
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000380 case Type::UByteTyID:
381 return Out << "unsigned char " << VariableName;
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000382 case Type::SByteTyID:
383 return Out << "signed char " << VariableName;
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000384 case Type::UShortTyID:
385 return Out << "unsigned long long " << VariableName;
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000386 case Type::ULongTyID:
387 return Out << "unsigned long long " << VariableName;
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000388 case Type::LongTyID:
389 return Out << "signed long long " << VariableName;
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000390 case Type::UIntTyID:
Chris Lattner16c7bb22002-05-09 02:28:59 +0000391 return Out << "unsigned " << VariableName;
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000392 default :
393 return Out << Ty->getDescription() << " " << VariableName;
394 }
395
396 // Check to see if the type is named.
397 map<const Type *, string>::iterator I = TypeNames.find(Ty);
398 if (I != TypeNames.end()) return Out << I->second << " " << VariableName;
399
400 // Otherwise we have a type that has not been named but is a derived type.
401 // Carefully recurse the type hierarchy to print out any contained symbolic
402 // names.
403 //
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000404 string TypeNameVar, tempstr = "";
Chris Lattner16c7bb22002-05-09 02:28:59 +0000405 TypeNameVar = calcTypeNameVar(Ty, TypeNames, VariableName, tempstr);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000406 return Out << TypeNameVar;
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000407}
408
409// Internal guts of printing a type name
410static ostream &printTypeInt(ostream &Out, const Type *Ty,
411 map<const Type *, string> &TypeNames) {
412 // Primitive types always print out their description, regardless of whether
413 // they have been named or not.
414
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000415 if (Ty->isPrimitiveType())
416 switch (Ty->getPrimitiveID()) {
417 case Type::BoolTyID:
Chris Lattner16c7bb22002-05-09 02:28:59 +0000418 return Out << "bool";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000419 case Type::UByteTyID:
420 return Out << "unsigned char";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000421 case Type::SByteTyID:
422 return Out << "signed char";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000423 case Type::UShortTyID:
Chris Lattner16c7bb22002-05-09 02:28:59 +0000424 return Out << "unsigned short";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000425 case Type::ULongTyID:
426 return Out << "unsigned long long";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000427 case Type::LongTyID:
428 return Out << "signed long long";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000429 case Type::UIntTyID:
Chris Lattner16c7bb22002-05-09 02:28:59 +0000430 return Out << "unsigned";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000431 default :
432 return Out << Ty->getDescription();
433 }
434
435 // Check to see if the type is named.
436 map<const Type *, string>::iterator I = TypeNames.find(Ty);
437 if (I != TypeNames.end()) return Out << I->second;
438
439 // Otherwise we have a type that has not been named but is a derived type.
440 // Carefully recurse the type hierarchy to print out any contained symbolic
441 // names.
442 //
Chris Lattner16c7bb22002-05-09 02:28:59 +0000443 string MInfo;
444 string TypeName = calcTypeName(Ty, TypeNames, MInfo);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000445 // TypeNames.insert(std::make_pair(Ty, TypeName));
446 //Cache type name for later use
447 if (MInfo != "")
448 return Out << TypeName << ")" << MInfo;
449 else
450 return Out << TypeName;
451}
452
453namespace {
454
455 //Internal CWriter class mimics AssemblyWriter.
456 class CWriter {
457 ostream& Out;
458 SlotCalculator &Table;
459 const Module *TheModule;
460 map<const Type *, string> TypeNames;
461 public:
462 inline CWriter(ostream &o, SlotCalculator &Tab, const Module *M)
463 : Out(o), Table(Tab), TheModule(M) {
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000464 }
465
Chris Lattner16c7bb22002-05-09 02:28:59 +0000466 inline void write(const Module *M) { printModule(M); }
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000467
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000468 ostream& printTypeVar(const Type *Ty, const string &VariableName) {
469 return printTypeVarInt(Out, Ty, TypeNames, VariableName);
470 }
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000471
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000472 ostream& printType(const Type *Ty) {
473 return printTypeInt(Out, Ty, TypeNames);
474 }
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000475
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000476 void writeOperand(const Value *Operand, bool PrintName = true);
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000477
478 string getValueName(const Value *V);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000479 private :
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000480
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000481 void printModule(const Module *M);
482 void printSymbolTable(const SymbolTable &ST);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000483 void printGlobal(const GlobalVariable *GV);
Chris Lattner16c7bb22002-05-09 02:28:59 +0000484 void printFunctionSignature(const Function *F);
485 void printFunctionDecl(const Function *F); // Print just the forward decl
486 void printFunctionArgument(const Argument *FA);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000487
488 void printFunction(const Function *);
489
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000490 void outputBasicBlock(const BasicBlock *);
491 };
492 /* END class CWriter */
493
494
495 /* CLASS InstLocalVarsVisitor */
496 class InstLocalVarsVisitor : public InstVisitor<InstLocalVarsVisitor> {
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000497 CWriter& CW;
Chris Lattner16c7bb22002-05-09 02:28:59 +0000498 void handleTerminator(TerminatorInst *tI, int indx);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000499 public:
500 CLocalVars CLV;
501
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000502 InstLocalVarsVisitor(CWriter &cw) : CW(cw) {}
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000503
504 void visitInstruction(Instruction *I) {
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000505 if (I->getType() != Type::VoidTy) {
506 string tempostr = CW.getValueName(I);
507 CLV.addLocalVar(I->getType(), tempostr);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000508 }
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000509 }
510
511 void visitBranchInst(BranchInst *I) {
Chris Lattner16c7bb22002-05-09 02:28:59 +0000512 handleTerminator(I, 0);
513 if (I->isConditional())
514 handleTerminator(I, 1);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000515 }
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000516 };
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000517}
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000518
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000519void InstLocalVarsVisitor::handleTerminator(TerminatorInst *tI,int indx) {
520 BasicBlock *bb = tI->getSuccessor(indx);
521
522 BasicBlock::const_iterator insIt = bb->begin();
523 while (insIt != bb->end()) {
524 if (const PHINode *pI = dyn_cast<PHINode>(*insIt)) {
525 // Its a phinode!
526 // Calculate the incoming index for this
527 assert(pI->getBasicBlockIndex(tI->getParent()) != -1);
528
529 CLV.addLocalVar(pI->getType(), CW.getValueName(pI));
530 } else
531 break;
532 insIt++;
533 }
534}
535
536namespace {
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000537 /* CLASS CInstPrintVisitor */
538
539 class CInstPrintVisitor: public InstVisitor<CInstPrintVisitor> {
540 CWriter& CW;
541 SlotCalculator& Table;
542 ostream &Out;
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000543
544 void outputLValue(Instruction *);
545 void printPhiFromNextBlock(TerminatorInst *tI, int indx);
Chris Lattner44408262002-05-09 03:50:42 +0000546 void printIndexingExpr(MemAccessInst *MAI);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000547
548 public:
549 CInstPrintVisitor (CWriter &cw, SlotCalculator& table, ostream& o)
Chris Lattner44408262002-05-09 03:50:42 +0000550 : CW(cw), Table(table), Out(o) {}
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000551
552 void visitCastInst(CastInst *I);
553 void visitCallInst(CallInst *I);
Chris Lattner2f5f51a2002-05-09 03:28:37 +0000554 void visitShiftInst(ShiftInst *I) { visitBinaryOperator(I); }
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000555 void visitReturnInst(ReturnInst *I);
556 void visitBranchInst(BranchInst *I);
557 void visitSwitchInst(SwitchInst *I);
558 void visitInvokeInst(InvokeInst *I) ;
559 void visitMallocInst(MallocInst *I);
560 void visitAllocaInst(AllocaInst *I);
561 void visitFreeInst(FreeInst *I);
562 void visitLoadInst(LoadInst *I);
563 void visitStoreInst(StoreInst *I);
564 void visitGetElementPtrInst(GetElementPtrInst *I);
Chris Lattner2f5f51a2002-05-09 03:28:37 +0000565 void visitPHINode(PHINode *I) {}
566
567 void visitNot(GenericUnaryInst *I);
568 void visitBinaryOperator(Instruction *I);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000569 };
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000570}
571
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000572void CInstPrintVisitor::outputLValue(Instruction *I) {
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000573 Out << " " << CW.getValueName(I) << " = ";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000574}
575
576void CInstPrintVisitor::printPhiFromNextBlock(TerminatorInst *tI, int indx) {
577 BasicBlock *bb = tI->getSuccessor(indx);
578 BasicBlock::const_iterator insIt = bb->begin();
579 while (insIt != bb->end()) {
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000580 if (PHINode *pI = dyn_cast<PHINode>(*insIt)) {
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000581 //Its a phinode!
582 //Calculate the incoming index for this
583 int incindex = pI->getBasicBlockIndex(tI->getParent());
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000584 if (incindex != -1) {
585 //now we have to do the printing
586 outputLValue(pI);
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000587 CW.writeOperand(pI->getIncomingValue(incindex));
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000588 Out << ";\n";
589 }
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000590 }
591 else break;
592 insIt++;
593 }
594}
595
596// Implement all "other" instructions, except for PHINode
597void CInstPrintVisitor::visitCastInst(CastInst *I) {
598 outputLValue(I);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000599 Out << "(";
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000600 CW.printType(I->getType());
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000601 Out << ")";
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000602 CW.writeOperand(I->getOperand(0));
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000603 Out << ";\n";
604}
605
606void CInstPrintVisitor::visitCallInst(CallInst *I) {
Chris Lattnerf34ee812002-05-09 03:12:34 +0000607 if (I->getType() != Type::VoidTy)
608 outputLValue(I);
Chris Lattner2f5f51a2002-05-09 03:28:37 +0000609 else
610 Out << " ";
Chris Lattnerf34ee812002-05-09 03:12:34 +0000611
Chris Lattner2f499022002-05-09 04:21:21 +0000612 const PointerType *PTy = cast<PointerType>(I->getCalledValue()->getType());
613 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
614 const Type *RetTy = FTy->getReturnType();
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000615
Chris Lattner2f499022002-05-09 04:21:21 +0000616 Out << CW.getValueName(I->getOperand(0)) << "(";
617
618 if (I->getNumOperands() != 0) {
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000619 CW.writeOperand(I->getOperand(1));
Chris Lattner2f499022002-05-09 04:21:21 +0000620
621 for (unsigned op = 2, Eop = I->getNumOperands(); op != Eop; ++op) {
622 Out << ", ";
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000623 CW.writeOperand(I->getOperand(op));
Chris Lattner2f499022002-05-09 04:21:21 +0000624 }
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000625 }
Chris Lattner2f499022002-05-09 04:21:21 +0000626 Out << ");\n";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000627}
628
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000629// Specific Instruction type classes... note that all of the casts are
630// neccesary because we use the instruction classes as opaque types...
631//
632void CInstPrintVisitor::visitReturnInst(ReturnInst *I) {
Chris Lattner2f5f51a2002-05-09 03:28:37 +0000633 Out << " return ";
Chris Lattner16c7bb22002-05-09 02:28:59 +0000634 if (I->getNumOperands())
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000635 CW.writeOperand(I->getOperand(0));
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000636 Out << ";\n";
637}
638
639void CInstPrintVisitor::visitBranchInst(BranchInst *I) {
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000640 TerminatorInst *tI = cast<TerminatorInst>(I);
Chris Lattner16c7bb22002-05-09 02:28:59 +0000641 if (I->isConditional()) {
642 Out << " if (";
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000643 CW.writeOperand(I->getCondition());
Chris Lattner2f5f51a2002-05-09 03:28:37 +0000644 Out << ") {\n";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000645 printPhiFromNextBlock(tI,0);
Chris Lattner16c7bb22002-05-09 02:28:59 +0000646 Out << " goto ";
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000647 CW.writeOperand(I->getOperand(0));
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000648 Out << ";\n";
Chris Lattner2f5f51a2002-05-09 03:28:37 +0000649 Out << " } else {\n";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000650 printPhiFromNextBlock(tI,1);
Chris Lattner16c7bb22002-05-09 02:28:59 +0000651 Out << " goto ";
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000652 CW.writeOperand(I->getOperand(1));
Chris Lattner2f5f51a2002-05-09 03:28:37 +0000653 Out << ";\n }\n";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000654 } else {
655 printPhiFromNextBlock(tI,0);
656 Out << " goto ";
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000657 CW.writeOperand(I->getOperand(0));
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000658 Out << ";\n";
659 }
660 Out << "\n";
661}
662
663void CInstPrintVisitor::visitSwitchInst(SwitchInst *I) {
Chris Lattnerf34ee812002-05-09 03:12:34 +0000664 assert(0 && "Switch not implemented!");
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000665}
666
667void CInstPrintVisitor::visitInvokeInst(InvokeInst *I) {
Chris Lattnerf34ee812002-05-09 03:12:34 +0000668 assert(0 && "Invoke not implemented!");
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000669}
670
671void CInstPrintVisitor::visitMallocInst(MallocInst *I) {
672 outputLValue(I);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000673 Out << "(";
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000674 CW.printType(I->getType()->getElementType());
Chris Lattner2f5f51a2002-05-09 03:28:37 +0000675 Out << "*)malloc(sizeof(";
676 CW.printTypeVar(I->getType()->getElementType(), "");
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000677 Out << ")";
Chris Lattner2f5f51a2002-05-09 03:28:37 +0000678
679 if (I->isArrayAllocation()) {
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000680 Out << " * " ;
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000681 CW.writeOperand(I->getOperand(0));
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000682 }
683 Out << ");";
684}
685
686void CInstPrintVisitor::visitAllocaInst(AllocaInst *I) {
687 outputLValue(I);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000688 string tempstr = "";
689 Out << "(";
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000690 CW.printTypeVar(I->getType(), tempstr);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000691 Out << ") alloca(sizeof(";
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000692 CW.printTypeVar(cast<PointerType>(I->getType())->getElementType(),
693 tempstr);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000694 Out << ")";
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000695 if (I->isArrayAllocation()) {
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000696 Out << " * " ;
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000697 CW.writeOperand(I->getOperand(0));
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000698 }
699 Out << ");\n";
700}
701
702void CInstPrintVisitor::visitFreeInst(FreeInst *I) {
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000703 Out << "free(";
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000704 CW.writeOperand(I->getOperand(0));
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000705 Out << ");\n";
706}
707
Chris Lattner44408262002-05-09 03:50:42 +0000708void CInstPrintVisitor::printIndexingExpr(MemAccessInst *MAI) {
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000709 CW.writeOperand(MAI->getPointerOperand());
Chris Lattner44408262002-05-09 03:50:42 +0000710
711 for (MemAccessInst::op_iterator I = MAI->idx_begin(), E = MAI->idx_end();
712 I != E; ++I)
713 if ((*I)->getType() == Type::UIntTy) {
714 Out << "[";
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000715 CW.writeOperand(*I);
Chris Lattner44408262002-05-09 03:50:42 +0000716 Out << "]";
717 } else {
718 Out << ".field" << cast<ConstantUInt>(*I)->getValue();
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000719 }
Chris Lattner44408262002-05-09 03:50:42 +0000720}
721
722void CInstPrintVisitor::visitLoadInst(LoadInst *I) {
723 outputLValue(I);
724 printIndexingExpr(I);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000725 Out << ";\n";
726}
727
Chris Lattner44408262002-05-09 03:50:42 +0000728void CInstPrintVisitor::visitStoreInst(StoreInst *I) {
729 Out << " ";
730 printIndexingExpr(I);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000731 Out << " = ";
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000732 CW.writeOperand(I->getOperand(0));
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000733 Out << ";\n";
734}
735
736void CInstPrintVisitor::visitGetElementPtrInst(GetElementPtrInst *I) {
737 outputLValue(I);
Chris Lattner44408262002-05-09 03:50:42 +0000738 Out << "&";
739 printIndexingExpr(I);
740 Out << ";\n";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000741}
742
Chris Lattner2f5f51a2002-05-09 03:28:37 +0000743void CInstPrintVisitor::visitNot(GenericUnaryInst *I) {
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000744 outputLValue(I);
Chris Lattner2f5f51a2002-05-09 03:28:37 +0000745 Out << "~";
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000746 CW.writeOperand(I->getOperand(0));
Chris Lattner2f5f51a2002-05-09 03:28:37 +0000747 Out << ";\n";
748}
749
750void CInstPrintVisitor::visitBinaryOperator(Instruction *I) {
751 // binary instructions, shift instructions, setCond instructions.
752 outputLValue(I);
753 if (isa<PointerType>(I->getType())) {
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000754 Out << "(";
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000755 CW.printType(I->getType());
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000756 Out << ")";
757 }
Chris Lattner2f5f51a2002-05-09 03:28:37 +0000758
759 if (isa<PointerType>(I->getType())) Out << "(long long)";
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000760 CW.writeOperand(I->getOperand(0));
Chris Lattner2f5f51a2002-05-09 03:28:37 +0000761
762 switch (I->getOpcode()) {
Chris Lattner2d05a1a2002-05-09 05:16:40 +0000763 case Instruction::Add: Out << " + "; break;
764 case Instruction::Sub: Out << " - "; break;
Chris Lattner2f5f51a2002-05-09 03:28:37 +0000765 case Instruction::Mul: Out << "*"; break;
766 case Instruction::Div: Out << "/"; break;
767 case Instruction::Rem: Out << "%"; break;
Chris Lattner2d05a1a2002-05-09 05:16:40 +0000768 case Instruction::And: Out << " & "; break;
769 case Instruction::Or: Out << " | "; break;
770 case Instruction::Xor: Out << " ^ "; break;
771 case Instruction::SetEQ: Out << " == "; break;
772 case Instruction::SetNE: Out << " != "; break;
773 case Instruction::SetLE: Out << " <= "; break;
774 case Instruction::SetGE: Out << " >= "; break;
775 case Instruction::SetLT: Out << " < "; break;
776 case Instruction::SetGT: Out << " > "; break;
777 case Instruction::Shl : Out << " << "; break;
778 case Instruction::Shr : Out << " >> "; break;
Chris Lattner2f5f51a2002-05-09 03:28:37 +0000779 default: cerr << "Invalid operator type!" << I; abort();
780 }
781
782 if (isa<PointerType>(I->getType())) Out << "(long long)";
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000783 CW.writeOperand(I->getOperand(1));
Chris Lattner2f5f51a2002-05-09 03:28:37 +0000784 Out << ";\n";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000785}
786
787/* END : CInstPrintVisitor implementation */
788
Chris Lattner2f499022002-05-09 04:21:21 +0000789// We dont want identifier names with ., space, - in them.
790// So we replace them with _
791static string makeNameProper(string x) {
792 string tmp;
793 for (string::iterator sI = x.begin(), sEnd = x.end(); sI != sEnd; sI++)
794 switch (*sI) {
795 case '.': tmp += "_d"; break;
796 case ' ': tmp += "_s"; break;
797 case '-': tmp += "_D"; break;
798 case '_': tmp += "__"; break;
799 default: tmp += *sI;
800 }
801
802 return tmp;
803}
804
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000805string CWriter::getValueName(const Value *V) {
Chris Lattner2f499022002-05-09 04:21:21 +0000806 if (V->hasName()) { // Print out the label if it exists...
807 if (isa<GlobalValue>(V)) // Do not mangle globals...
808 return makeNameProper(V->getName());
809
Chris Lattner2d05a1a2002-05-09 05:16:40 +0000810 return "l" + utostr(V->getType()->getUniqueID()) + "_" +
811 makeNameProper(V->getName());
Chris Lattner2f499022002-05-09 04:21:21 +0000812 }
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000813
814 int Slot = Table.getValSlot(V);
815 assert(Slot >= 0 && "Invalid value!");
Chris Lattner2d05a1a2002-05-09 05:16:40 +0000816 return "ltmp_" + itostr(Slot) + "_" + utostr(V->getType()->getUniqueID());
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000817}
818
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000819void CWriter::printModule(const Module *M) {
820 // printing stdlib inclusion
821 // Out << "#include <stdlib.h>\n";
822
Chris Lattner16c7bb22002-05-09 02:28:59 +0000823 // get declaration for alloca
824 Out << "/* Provide Declarations */\n"
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000825 << "#include <alloca.h>\n\n"
Chris Lattner16c7bb22002-05-09 02:28:59 +0000826
827 // Provide a definition for null if one does not already exist.
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000828 << "#ifndef NULL\n#define NULL 0\n#endif\n\n"
Chris Lattner16c7bb22002-05-09 02:28:59 +0000829 << "typedef unsigned char bool;\n"
830
831 << "\n\n/* Global Symbols */\n";
832
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000833 // Loop over the symbol table, emitting all named constants...
834 if (M->hasSymbolTable())
835 printSymbolTable(*M->getSymbolTable());
836
Chris Lattner16c7bb22002-05-09 02:28:59 +0000837 Out << "\n\n/* Global Data */\n";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000838 for_each(M->gbegin(), M->gend(),
839 bind_obj(this, &CWriter::printGlobal));
840
Chris Lattner16c7bb22002-05-09 02:28:59 +0000841 // First output all the declarations of the functions as C requires Functions
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000842 // be declared before they are used.
Chris Lattner16c7bb22002-05-09 02:28:59 +0000843 //
844 Out << "\n\n/* Function Declarations */\n";
845 for_each(M->begin(), M->end(), bind_obj(this, &CWriter::printFunctionDecl));
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000846
Chris Lattner16c7bb22002-05-09 02:28:59 +0000847 // Output all of the functions...
848 Out << "\n\n/* Function Bodies */\n";
849 for_each(M->begin(), M->end(), bind_obj(this, &CWriter::printFunction));
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000850}
851
852// prints the global constants
853void CWriter::printGlobal(const GlobalVariable *GV) {
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000854 string tempostr = getValueName(GV);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000855 if (GV->hasInternalLinkage()) Out << "static ";
856
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000857 printTypeVar(GV->getType()->getElementType(), tempostr);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000858
859 if (GV->hasInitializer()) {
860 Out << " = " ;
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000861 writeOperand(GV->getInitializer(), false);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000862 }
863
864 Out << ";\n";
865}
866
867// printSymbolTable - Run through symbol table looking for named constants
868// if a named constant is found, emit it's declaration...
869// Assuming that symbol table has only types and constants.
870void CWriter::printSymbolTable(const SymbolTable &ST) {
871 // GraphT G;
872 for (SymbolTable::const_iterator TI = ST.begin(); TI != ST.end(); ++TI) {
873 SymbolTable::type_const_iterator I = ST.type_begin(TI->first);
874 SymbolTable::type_const_iterator End = ST.type_end(TI->first);
875
876 // TODO
877 // Need to run through all the used types in the program
878 // FindUsedTypes &FUT = new FindUsedTypes();
879 // const std::set<const Type *> &UsedTypes = FUT.getTypes();
880 // Filter out the structures printing forward definitions for each of them
881 // and creating the dependency graph.
882 // Print forward definitions to all of them
883 // print the typedefs topologically sorted
884
885 // But for now we have
886 for (; I != End; ++I) {
887 const Value *V = I->second;
Chris Lattner2d05a1a2002-05-09 05:16:40 +0000888 if (const Type *Ty = dyn_cast<const Type>(V)) {
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000889 string tempostr;
890 string tempstr = "";
891 Out << "typedef ";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000892 tempostr = "llvm__" + I->first;
Chris Lattner16c7bb22002-05-09 02:28:59 +0000893 string TypeNameVar = calcTypeNameVar(Ty, TypeNames,
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000894 tempostr, tempstr);
895 Out << TypeNameVar << ";\n";
896 if (!isa<PointerType>(Ty) ||
897 !cast<PointerType>(Ty)->getElementType()->isPrimitiveType())
898 TypeNames.insert(std::make_pair(Ty, "llvm__"+I->first));
899 }
900 }
901 }
902}
903
Chris Lattner16c7bb22002-05-09 02:28:59 +0000904// printFunctionDecl - Print function declaration
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000905//
Chris Lattner16c7bb22002-05-09 02:28:59 +0000906void CWriter::printFunctionDecl(const Function *F) {
907 printFunctionSignature(F);
908 Out << ";\n";
909}
910
911void CWriter::printFunctionSignature(const Function *F) {
912 if (F->hasInternalLinkage()) Out << "static ";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000913
914 // Loop over the arguments, printing them...
Chris Lattner16c7bb22002-05-09 02:28:59 +0000915 const FunctionType *FT = cast<FunctionType>(F->getFunctionType());
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000916
Chris Lattner16c7bb22002-05-09 02:28:59 +0000917 // Print out the return type and name...
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000918 printType(F->getReturnType());
Chris Lattner2f499022002-05-09 04:21:21 +0000919 Out << " " << getValueName(F) << "(";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000920
Chris Lattner16c7bb22002-05-09 02:28:59 +0000921 if (!F->isExternal()) {
922 for_each(F->getArgumentList().begin(), F->getArgumentList().end(),
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000923 bind_obj(this, &CWriter::printFunctionArgument));
924 } else {
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000925 // Loop over the arguments, printing them...
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000926 for (FunctionType::ParamTypes::const_iterator I =
Chris Lattner16c7bb22002-05-09 02:28:59 +0000927 FT->getParamTypes().begin(),
928 E = FT->getParamTypes().end(); I != E; ++I) {
929 if (I != FT->getParamTypes().begin()) Out << ", ";
Chris Lattner2a7ab2e2002-05-09 04:39:00 +0000930 printType(*I);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000931 }
932 }
933
934 // Finish printing arguments...
Chris Lattner16c7bb22002-05-09 02:28:59 +0000935 if (FT->isVarArg()) {
936 if (FT->getParamTypes().size()) Out << ", ";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000937 Out << "..."; // Output varargs portion of signature!
938 }
Chris Lattner16c7bb22002-05-09 02:28:59 +0000939 Out << ")";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000940}
941
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000942
943// printFunctionArgument - This member is called for every argument that
944// is passed into the method. Simply print it out
945//
946void CWriter::printFunctionArgument(const Argument *Arg) {
947 // Insert commas as we go... the first arg doesn't get a comma
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000948 if (Arg != Arg->getParent()->getArgumentList().front()) Out << ", ";
949
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000950 // Output type...
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000951 printTypeVar(Arg->getType(), getValueName(Arg));
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000952}
953
Chris Lattner16c7bb22002-05-09 02:28:59 +0000954void CWriter::printFunction(const Function *F) {
955 if (F->isExternal()) return;
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000956
Chris Lattnerf34ee812002-05-09 03:12:34 +0000957 Table.incorporateFunction(F);
958
Chris Lattner16c7bb22002-05-09 02:28:59 +0000959 // Process each of the basic blocks, gather information and call the
960 // output methods on the CLocalVars and Function* objects.
961
962 // gather local variable information for each basic block
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000963 InstLocalVarsVisitor ILV(*this);
Chris Lattner16c7bb22002-05-09 02:28:59 +0000964 ILV.visit((Function *)F);
965
966 printFunctionSignature(F);
967 Out << " {\n";
968
969 // Loop over the symbol table, emitting all named constants...
970 if (F->hasSymbolTable())
971 printSymbolTable(*F->getSymbolTable());
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000972
Chris Lattner16c7bb22002-05-09 02:28:59 +0000973 // print the local variables
974 // we assume that every local variable is alloca'ed in the C code.
975 std::map<const Type*, VarListType> &locals = ILV.CLV.LocalVars;
976
977 map<const Type*, VarListType>::iterator iter;
978 for (iter = locals.begin(); iter != locals.end(); ++iter) {
979 VarListType::iterator listiter;
980 for (listiter = iter->second.begin(); listiter != iter->second.end();
981 ++listiter) {
982 Out << " ";
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000983 printTypeVar(iter->first, *listiter);
Chris Lattner16c7bb22002-05-09 02:28:59 +0000984 Out << ";\n";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000985 }
986 }
Chris Lattner16c7bb22002-05-09 02:28:59 +0000987
988 // print the basic blocks
989 for_each(F->begin(), F->end(), bind_obj(this, &CWriter::outputBasicBlock));
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000990
Chris Lattner16c7bb22002-05-09 02:28:59 +0000991 Out << "}\n";
Chris Lattnerf34ee812002-05-09 03:12:34 +0000992 Table.purgeFunction();
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000993}
994
995void CWriter::outputBasicBlock(const BasicBlock* BB) {
Chris Lattnerb5af06a2002-05-09 03:06:06 +0000996 Out << getValueName(BB) << ":\n";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +0000997
998 // Output all of the instructions in the basic block...
999 // print the basic blocks
1000 CInstPrintVisitor CIPV(*this, Table, Out);
1001 CIPV.visit((BasicBlock *) BB);
1002}
1003
Chris Lattner2a7ab2e2002-05-09 04:39:00 +00001004void CWriter::writeOperand(const Value *Operand, bool PrintName = true) {
Chris Lattner2d05a1a2002-05-09 05:16:40 +00001005 if (isa<GlobalVariable>(Operand))
1006 Out << "(&"; // Global variables are references as their addresses by llvm
Chris Lattner44408262002-05-09 03:50:42 +00001007
Chris Lattner16c7bb22002-05-09 02:28:59 +00001008 if (PrintName && Operand->hasName()) {
Chris Lattner2f499022002-05-09 04:21:21 +00001009 Out << getValueName(Operand);
Chris Lattner44408262002-05-09 03:50:42 +00001010 } else if (const Constant *CPV = dyn_cast<const Constant>(Operand)) {
Chris Lattner16c7bb22002-05-09 02:28:59 +00001011 if (isa<ConstantPointerNull>(CPV))
1012 Out << "NULL";
1013 else
1014 Out << getConstStrValue(CPV);
Chris Lattner44408262002-05-09 03:50:42 +00001015 } else {
Chris Lattner16c7bb22002-05-09 02:28:59 +00001016 int Slot = Table.getValSlot(Operand);
Chris Lattner2d05a1a2002-05-09 05:16:40 +00001017 assert(Slot >= 0 && "Malformed LLVM!");
1018 Out << "ltmp_" << Slot << "_" << Operand->getType()->getUniqueID();
Chris Lattner16c7bb22002-05-09 02:28:59 +00001019 }
Chris Lattner44408262002-05-09 03:50:42 +00001020
Chris Lattner2d05a1a2002-05-09 05:16:40 +00001021 if (isa<GlobalVariable>(Operand))
Chris Lattner44408262002-05-09 03:50:42 +00001022 Out << ")";
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +00001023}
1024
1025
1026//===----------------------------------------------------------------------===//
1027// External Interface declaration
1028//===----------------------------------------------------------------------===//
1029
Chris Lattnerf34ee812002-05-09 03:12:34 +00001030void WriteToC(const Module *M, ostream &Out) {
1031 assert(M && "You can't write a null module!!");
1032 SlotCalculator SlotTable(M, false);
1033 CWriter W(Out, SlotTable, M);
1034 W.write(M);
Sumant Kowshik9ddc86c2002-05-08 18:09:58 +00001035 Out.flush();
1036}