1 //===--- Types.cpp - Driver input & temporary type information ------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "clang/Driver/Types.h"
11 #include "llvm/ADT/STLExtras.h"
12 #include "llvm/ADT/StringSwitch.h"
13 #include <cassert>
14 #include <string.h>
15 
16 using namespace clang::driver;
17 using namespace clang::driver::types;
18 
19 struct TypeInfo {
20   const char *Name;
21   const char *Flags;
22   const char *TempSuffix;
23   ID PreprocessedType;
24 };
25 
26 static const TypeInfo TypeInfos[] = {
27 #define TYPE(NAME, ID, PP_TYPE, TEMP_SUFFIX, FLAGS) \
28   { NAME, FLAGS, TEMP_SUFFIX, TY_##PP_TYPE, },
29 #include "clang/Driver/Types.def"
30 #undef TYPE
31 };
32 static const unsigned numTypes = llvm::array_lengthof(TypeInfos);
33 
getInfo(unsigned id)34 static const TypeInfo &getInfo(unsigned id) {
35   assert(id > 0 && id - 1 < numTypes && "Invalid Type ID.");
36   return TypeInfos[id - 1];
37 }
38 
getTypeName(ID Id)39 const char *types::getTypeName(ID Id) {
40   return getInfo(Id).Name;
41 }
42 
getPreprocessedType(ID Id)43 types::ID types::getPreprocessedType(ID Id) {
44   return getInfo(Id).PreprocessedType;
45 }
46 
getTypeTempSuffix(ID Id,bool CLMode)47 const char *types::getTypeTempSuffix(ID Id, bool CLMode) {
48   if (Id == TY_Object && CLMode)
49     return "obj";
50   if (Id == TY_Image && CLMode)
51     return "exe";
52   if (Id == TY_PP_Asm && CLMode)
53     return "asm";
54   return getInfo(Id).TempSuffix;
55 }
56 
onlyAssembleType(ID Id)57 bool types::onlyAssembleType(ID Id) {
58   return strchr(getInfo(Id).Flags, 'a');
59 }
60 
onlyPrecompileType(ID Id)61 bool types::onlyPrecompileType(ID Id) {
62   return strchr(getInfo(Id).Flags, 'p');
63 }
64 
canTypeBeUserSpecified(ID Id)65 bool types::canTypeBeUserSpecified(ID Id) {
66   return strchr(getInfo(Id).Flags, 'u');
67 }
68 
appendSuffixForType(ID Id)69 bool types::appendSuffixForType(ID Id) {
70   return strchr(getInfo(Id).Flags, 'A');
71 }
72 
canLipoType(ID Id)73 bool types::canLipoType(ID Id) {
74   return (Id == TY_Nothing ||
75           Id == TY_Image ||
76           Id == TY_Object ||
77           Id == TY_LTO_BC);
78 }
79 
isAcceptedByClang(ID Id)80 bool types::isAcceptedByClang(ID Id) {
81   switch (Id) {
82   default:
83     return false;
84 
85   case TY_Asm:
86   case TY_C: case TY_PP_C:
87   case TY_CL:
88   case TY_CUDA:
89   case TY_ObjC: case TY_PP_ObjC: case TY_PP_ObjC_Alias:
90   case TY_CXX: case TY_PP_CXX:
91   case TY_ObjCXX: case TY_PP_ObjCXX: case TY_PP_ObjCXX_Alias:
92   case TY_CHeader: case TY_PP_CHeader:
93   case TY_CLHeader:
94   case TY_ObjCHeader: case TY_PP_ObjCHeader:
95   case TY_CXXHeader: case TY_PP_CXXHeader:
96   case TY_ObjCXXHeader: case TY_PP_ObjCXXHeader:
97   case TY_AST: case TY_ModuleFile:
98   case TY_LLVM_IR: case TY_LLVM_BC:
99     return true;
100   }
101 }
102 
isObjC(ID Id)103 bool types::isObjC(ID Id) {
104   switch (Id) {
105   default:
106     return false;
107 
108   case TY_ObjC: case TY_PP_ObjC: case TY_PP_ObjC_Alias:
109   case TY_ObjCXX: case TY_PP_ObjCXX:
110   case TY_ObjCHeader: case TY_PP_ObjCHeader:
111   case TY_ObjCXXHeader: case TY_PP_ObjCXXHeader: case TY_PP_ObjCXX_Alias:
112     return true;
113   }
114 }
115 
isCXX(ID Id)116 bool types::isCXX(ID Id) {
117   switch (Id) {
118   default:
119     return false;
120 
121   case TY_CXX: case TY_PP_CXX:
122   case TY_ObjCXX: case TY_PP_ObjCXX: case TY_PP_ObjCXX_Alias:
123   case TY_CXXHeader: case TY_PP_CXXHeader:
124   case TY_ObjCXXHeader: case TY_PP_ObjCXXHeader:
125   case TY_CUDA:
126     return true;
127   }
128 }
129 
lookupTypeForExtension(const char * Ext)130 types::ID types::lookupTypeForExtension(const char *Ext) {
131   return llvm::StringSwitch<types::ID>(Ext)
132            .Case("c", TY_C)
133            .Case("i", TY_PP_C)
134            .Case("m", TY_ObjC)
135            .Case("M", TY_ObjCXX)
136            .Case("h", TY_CHeader)
137            .Case("C", TY_CXX)
138            .Case("H", TY_CXXHeader)
139            .Case("f", TY_PP_Fortran)
140            .Case("F", TY_Fortran)
141            .Case("s", TY_PP_Asm)
142            .Case("asm", TY_PP_Asm)
143            .Case("S", TY_Asm)
144            .Case("o", TY_Object)
145            .Case("obj", TY_Object)
146            .Case("lib", TY_Object)
147            .Case("ii", TY_PP_CXX)
148            .Case("mi", TY_PP_ObjC)
149            .Case("mm", TY_ObjCXX)
150            .Case("bc", TY_LLVM_BC)
151            .Case("cc", TY_CXX)
152            .Case("CC", TY_CXX)
153            .Case("cl", TY_CL)
154            .Case("cp", TY_CXX)
155            .Case("cu", TY_CUDA)
156            .Case("hh", TY_CXXHeader)
157            .Case("ll", TY_LLVM_IR)
158            .Case("hpp", TY_CXXHeader)
159            .Case("ads", TY_Ada)
160            .Case("adb", TY_Ada)
161            .Case("ast", TY_AST)
162            .Case("c++", TY_CXX)
163            .Case("C++", TY_CXX)
164            .Case("cxx", TY_CXX)
165            .Case("cpp", TY_CXX)
166            .Case("CPP", TY_CXX)
167            .Case("CXX", TY_CXX)
168            .Case("for", TY_PP_Fortran)
169            .Case("FOR", TY_PP_Fortran)
170            .Case("fpp", TY_Fortran)
171            .Case("FPP", TY_Fortran)
172            .Case("f90", TY_PP_Fortran)
173            .Case("f95", TY_PP_Fortran)
174            .Case("F90", TY_Fortran)
175            .Case("F95", TY_Fortran)
176            .Case("mii", TY_PP_ObjCXX)
177            .Case("pcm", TY_ModuleFile)
178            .Case("pch", TY_PCH)
179            .Case("gch", TY_PCH)
180            .Default(TY_INVALID);
181 }
182 
lookupTypeForTypeSpecifier(const char * Name)183 types::ID types::lookupTypeForTypeSpecifier(const char *Name) {
184   for (unsigned i=0; i<numTypes; ++i) {
185     types::ID Id = (types::ID) (i + 1);
186     if (canTypeBeUserSpecified(Id) &&
187         strcmp(Name, getInfo(Id).Name) == 0)
188       return Id;
189   }
190 
191   return TY_INVALID;
192 }
193 
194 // FIXME: Why don't we just put this list in the defs file, eh.
getCompilationPhases(ID Id,llvm::SmallVectorImpl<phases::ID> & P)195 void types::getCompilationPhases(ID Id, llvm::SmallVectorImpl<phases::ID> &P) {
196   if (Id != TY_Object) {
197     if (getPreprocessedType(Id) != TY_INVALID) {
198       P.push_back(phases::Preprocess);
199     }
200 
201     if (onlyPrecompileType(Id)) {
202       P.push_back(phases::Precompile);
203     } else {
204       if (!onlyAssembleType(Id)) {
205         P.push_back(phases::Compile);
206         P.push_back(phases::Backend);
207       }
208       P.push_back(phases::Assemble);
209     }
210   }
211   if (!onlyPrecompileType(Id)) {
212     P.push_back(phases::Link);
213   }
214   assert(0 < P.size() && "Not enough phases in list");
215   assert(P.size() <= phases::MaxNumberOfPhases && "Too many phases in list");
216   return;
217 }
218 
lookupCXXTypeForCType(ID Id)219 ID types::lookupCXXTypeForCType(ID Id) {
220   switch (Id) {
221   default:
222     return Id;
223 
224   case types::TY_C:
225     return types::TY_CXX;
226   case types::TY_PP_C:
227     return types::TY_PP_CXX;
228   case types::TY_CHeader:
229     return types::TY_CXXHeader;
230   case types::TY_PP_CHeader:
231     return types::TY_PP_CXXHeader;
232   }
233 }
234