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 
12 #include "llvm/ADT/StringSwitch.h"
13 #include <string.h>
14 #include <cassert>
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 = sizeof(TypeInfos) / sizeof(TypeInfos[0]);
33 
34 static const TypeInfo &getInfo(unsigned id) {
35   assert(id > 0 && id - 1 < numTypes && "Invalid Type ID.");
36   return TypeInfos[id - 1];
37 }
38 
39 const char *types::getTypeName(ID Id) {
40   return getInfo(Id).Name;
41 }
42 
43 types::ID types::getPreprocessedType(ID Id) {
44   return getInfo(Id).PreprocessedType;
45 }
46 
47 const char *types::getTypeTempSuffix(ID Id) {
48   return getInfo(Id).TempSuffix;
49 }
50 
51 bool types::onlyAssembleType(ID Id) {
52   return strchr(getInfo(Id).Flags, 'a');
53 }
54 
55 bool types::onlyPrecompileType(ID Id) {
56   return strchr(getInfo(Id).Flags, 'p');
57 }
58 
59 bool types::canTypeBeUserSpecified(ID Id) {
60   return strchr(getInfo(Id).Flags, 'u');
61 }
62 
63 bool types::appendSuffixForType(ID Id) {
64   return strchr(getInfo(Id).Flags, 'A');
65 }
66 
67 bool types::canLipoType(ID Id) {
68   return (Id == TY_Nothing ||
69           Id == TY_Image ||
70           Id == TY_Object);
71 }
72 
73 bool types::isAcceptedByClang(ID Id) {
74   switch (Id) {
75   default:
76     return false;
77 
78   case TY_Asm:
79   case TY_C: case TY_PP_C:
80   case TY_CL:
81   case TY_ObjC: case TY_PP_ObjC:
82   case TY_CXX: case TY_PP_CXX:
83   case TY_ObjCXX: case TY_PP_ObjCXX:
84   case TY_CHeader: case TY_PP_CHeader:
85   case TY_ObjCHeader: case TY_PP_ObjCHeader:
86   case TY_CXXHeader: case TY_PP_CXXHeader:
87   case TY_ObjCXXHeader: case TY_PP_ObjCXXHeader:
88   case TY_AST:
89   case TY_LLVM_IR: case TY_LLVM_BC:
90     return true;
91   }
92 }
93 
94 bool types::isOnlyAcceptedByClang(ID Id) {
95   switch (Id) {
96   default:
97     return false;
98 
99   case TY_AST:
100   case TY_LLVM_IR:
101   case TY_LLVM_BC:
102   case TY_RewrittenObjC:
103     return true;
104   }
105 }
106 
107 bool types::isObjC(ID Id) {
108   switch (Id) {
109   default:
110     return false;
111 
112   case TY_ObjC: case TY_PP_ObjC:
113   case TY_ObjCXX: case TY_PP_ObjCXX:
114   case TY_ObjCHeader: case TY_PP_ObjCHeader:
115   case TY_ObjCXXHeader: case TY_PP_ObjCXXHeader:
116     return true;
117   }
118 }
119 
120 bool types::isCXX(ID Id) {
121   switch (Id) {
122   default:
123     return false;
124 
125   case TY_CXX: case TY_PP_CXX:
126   case TY_ObjCXX: case TY_PP_ObjCXX:
127   case TY_CXXHeader: case TY_PP_CXXHeader:
128   case TY_ObjCXXHeader: case TY_PP_ObjCXXHeader:
129     return true;
130   }
131 }
132 
133 types::ID types::lookupTypeForExtension(const char *Ext) {
134   return llvm::StringSwitch<types::ID>(Ext)
135            .Case("c", TY_C)
136            .Case("i", TY_PP_C)
137            .Case("m", TY_ObjC)
138            .Case("M", TY_ObjCXX)
139            .Case("h", TY_CHeader)
140            .Case("C", TY_CXX)
141            .Case("H", TY_CXXHeader)
142            .Case("f", TY_PP_Fortran)
143            .Case("F", TY_Fortran)
144            .Case("s", TY_PP_Asm)
145            .Case("S", TY_Asm)
146            .Case("ii", TY_PP_CXX)
147            .Case("mi", TY_PP_ObjC)
148            .Case("mm", TY_ObjCXX)
149            .Case("bc", TY_LLVM_BC)
150            .Case("cc", TY_CXX)
151            .Case("CC", TY_CXX)
152            .Case("cl", TY_CL)
153            .Case("cp", TY_CXX)
154            .Case("hh", TY_CXXHeader)
155            .Case("ll", TY_LLVM_IR)
156            .Case("hpp", TY_CXXHeader)
157            .Case("ads", TY_Ada)
158            .Case("adb", TY_Ada)
159            .Case("ast", TY_AST)
160            .Case("cxx", TY_CXX)
161            .Case("cpp", TY_CXX)
162            .Case("CPP", TY_CXX)
163            .Case("CXX", TY_CXX)
164            .Case("for", TY_PP_Fortran)
165            .Case("FOR", TY_PP_Fortran)
166            .Case("fpp", TY_Fortran)
167            .Case("FPP", TY_Fortran)
168            .Case("f90", TY_PP_Fortran)
169            .Case("f95", TY_PP_Fortran)
170            .Case("F90", TY_Fortran)
171            .Case("F95", TY_Fortran)
172            .Case("mii", TY_PP_ObjCXX)
173            .Default(TY_INVALID);
174 }
175 
176 types::ID types::lookupTypeForTypeSpecifier(const char *Name) {
177   unsigned N = strlen(Name);
178 
179   for (unsigned i=0; i<numTypes; ++i) {
180     types::ID Id = (types::ID) (i + 1);
181     if (canTypeBeUserSpecified(Id) &&
182         memcmp(Name, getInfo(Id).Name, N + 1) == 0)
183       return Id;
184   }
185 
186   return TY_INVALID;
187 }
188 
189 // FIXME: Why don't we just put this list in the defs file, eh.
190 
191 unsigned types::getNumCompilationPhases(ID Id) {
192   if (Id == TY_Object)
193     return 1;
194 
195   unsigned N = 0;
196   if (getPreprocessedType(Id) != TY_INVALID)
197     N += 1;
198 
199   if (onlyAssembleType(Id))
200     return N + 2; // assemble, link
201   if (onlyPrecompileType(Id))
202     return N + 1; // precompile
203 
204   return N + 3; // compile, assemble, link
205 }
206 
207 phases::ID types::getCompilationPhase(ID Id, unsigned N) {
208   assert(N < getNumCompilationPhases(Id) && "Invalid index.");
209 
210   if (Id == TY_Object)
211     return phases::Link;
212 
213   if (getPreprocessedType(Id) != TY_INVALID) {
214     if (N == 0)
215       return phases::Preprocess;
216     --N;
217   }
218 
219   if (onlyAssembleType(Id))
220     return N == 0 ? phases::Assemble : phases::Link;
221 
222   if (onlyPrecompileType(Id))
223     return phases::Precompile;
224 
225   if (N == 0)
226     return phases::Compile;
227   if (N == 1)
228     return phases::Assemble;
229 
230   return phases::Link;
231 }
232 
233 ID types::lookupCXXTypeForCType(ID Id) {
234   switch (Id) {
235   default:
236     return Id;
237 
238   case types::TY_C:
239     return types::TY_CXX;
240   case types::TY_PP_C:
241     return types::TY_PP_CXX;
242   case types::TY_CHeader:
243     return types::TY_CXXHeader;
244   case types::TY_PP_CHeader:
245     return types::TY_PP_CXXHeader;
246   }
247 }
248