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
getPrecompiledType(ID Id)47 types::ID types::getPrecompiledType(ID Id) {
48 if (strchr(getInfo(Id).Flags, 'm'))
49 return TY_ModuleFile;
50 if (onlyPrecompileType(Id))
51 return TY_PCH;
52 return TY_INVALID;
53 }
54
getTypeTempSuffix(ID Id,bool CLMode)55 const char *types::getTypeTempSuffix(ID Id, bool CLMode) {
56 if (CLMode) {
57 switch (Id) {
58 case TY_Object:
59 case TY_LTO_BC:
60 return "obj";
61 case TY_Image:
62 return "exe";
63 case TY_PP_Asm:
64 return "asm";
65 default:
66 break;
67 }
68 }
69 return getInfo(Id).TempSuffix;
70 }
71
onlyAssembleType(ID Id)72 bool types::onlyAssembleType(ID Id) {
73 return strchr(getInfo(Id).Flags, 'a');
74 }
75
onlyPrecompileType(ID Id)76 bool types::onlyPrecompileType(ID Id) {
77 return strchr(getInfo(Id).Flags, 'p');
78 }
79
canTypeBeUserSpecified(ID Id)80 bool types::canTypeBeUserSpecified(ID Id) {
81 return strchr(getInfo(Id).Flags, 'u');
82 }
83
appendSuffixForType(ID Id)84 bool types::appendSuffixForType(ID Id) {
85 return strchr(getInfo(Id).Flags, 'A');
86 }
87
canLipoType(ID Id)88 bool types::canLipoType(ID Id) {
89 return (Id == TY_Nothing ||
90 Id == TY_Image ||
91 Id == TY_Object ||
92 Id == TY_LTO_BC);
93 }
94
isAcceptedByClang(ID Id)95 bool types::isAcceptedByClang(ID Id) {
96 switch (Id) {
97 default:
98 return false;
99
100 case TY_Asm:
101 case TY_C: case TY_PP_C:
102 case TY_CL:
103 case TY_CUDA: case TY_PP_CUDA:
104 case TY_CUDA_DEVICE:
105 case TY_HIP:
106 case TY_PP_HIP:
107 case TY_HIP_DEVICE:
108 case TY_ObjC: case TY_PP_ObjC: case TY_PP_ObjC_Alias:
109 case TY_CXX: case TY_PP_CXX:
110 case TY_ObjCXX: case TY_PP_ObjCXX: case TY_PP_ObjCXX_Alias:
111 case TY_CHeader: case TY_PP_CHeader:
112 case TY_CLHeader:
113 case TY_ObjCHeader: case TY_PP_ObjCHeader:
114 case TY_CXXHeader: case TY_PP_CXXHeader:
115 case TY_ObjCXXHeader: case TY_PP_ObjCXXHeader:
116 case TY_CXXModule: case TY_PP_CXXModule:
117 case TY_AST: case TY_ModuleFile:
118 case TY_LLVM_IR: case TY_LLVM_BC:
119 return true;
120 }
121 }
122
isObjC(ID Id)123 bool types::isObjC(ID Id) {
124 switch (Id) {
125 default:
126 return false;
127
128 case TY_ObjC: case TY_PP_ObjC: case TY_PP_ObjC_Alias:
129 case TY_ObjCXX: case TY_PP_ObjCXX:
130 case TY_ObjCHeader: case TY_PP_ObjCHeader:
131 case TY_ObjCXXHeader: case TY_PP_ObjCXXHeader: case TY_PP_ObjCXX_Alias:
132 return true;
133 }
134 }
135
isCXX(ID Id)136 bool types::isCXX(ID Id) {
137 switch (Id) {
138 default:
139 return false;
140
141 case TY_CXX: case TY_PP_CXX:
142 case TY_ObjCXX: case TY_PP_ObjCXX: case TY_PP_ObjCXX_Alias:
143 case TY_CXXHeader: case TY_PP_CXXHeader:
144 case TY_ObjCXXHeader: case TY_PP_ObjCXXHeader:
145 case TY_CXXModule: case TY_PP_CXXModule:
146 case TY_CUDA: case TY_PP_CUDA: case TY_CUDA_DEVICE:
147 case TY_HIP:
148 case TY_PP_HIP:
149 case TY_HIP_DEVICE:
150 return true;
151 }
152 }
153
isLLVMIR(ID Id)154 bool types::isLLVMIR(ID Id) {
155 switch (Id) {
156 default:
157 return false;
158
159 case TY_LLVM_IR:
160 case TY_LLVM_BC:
161 case TY_LTO_IR:
162 case TY_LTO_BC:
163 return true;
164 }
165 }
166
isCuda(ID Id)167 bool types::isCuda(ID Id) {
168 switch (Id) {
169 default:
170 return false;
171
172 case TY_CUDA:
173 case TY_PP_CUDA:
174 case TY_CUDA_DEVICE:
175 return true;
176 }
177 }
178
isHIP(ID Id)179 bool types::isHIP(ID Id) {
180 switch (Id) {
181 default:
182 return false;
183
184 case TY_HIP:
185 case TY_PP_HIP:
186 case TY_HIP_DEVICE:
187 return true;
188 }
189 }
190
isSrcFile(ID Id)191 bool types::isSrcFile(ID Id) {
192 return Id != TY_Object && getPreprocessedType(Id) != TY_INVALID;
193 }
194
lookupTypeForExtension(llvm::StringRef Ext)195 types::ID types::lookupTypeForExtension(llvm::StringRef Ext) {
196 return llvm::StringSwitch<types::ID>(Ext)
197 .Case("c", TY_C)
198 .Case("C", TY_CXX)
199 .Case("F", TY_Fortran)
200 .Case("f", TY_PP_Fortran)
201 .Case("h", TY_CHeader)
202 .Case("H", TY_CXXHeader)
203 .Case("i", TY_PP_C)
204 .Case("m", TY_ObjC)
205 .Case("M", TY_ObjCXX)
206 .Case("o", TY_Object)
207 .Case("S", TY_Asm)
208 .Case("s", TY_PP_Asm)
209 .Case("bc", TY_LLVM_BC)
210 .Case("cc", TY_CXX)
211 .Case("CC", TY_CXX)
212 .Case("cl", TY_CL)
213 .Case("cp", TY_CXX)
214 .Case("cu", TY_CUDA)
215 .Case("hh", TY_CXXHeader)
216 .Case("ii", TY_PP_CXX)
217 .Case("ll", TY_LLVM_IR)
218 .Case("mi", TY_PP_ObjC)
219 .Case("mm", TY_ObjCXX)
220 .Case("rs", TY_RenderScript)
221 .Case("adb", TY_Ada)
222 .Case("ads", TY_Ada)
223 .Case("asm", TY_PP_Asm)
224 .Case("ast", TY_AST)
225 .Case("ccm", TY_CXXModule)
226 .Case("cpp", TY_CXX)
227 .Case("CPP", TY_CXX)
228 .Case("c++", TY_CXX)
229 .Case("C++", TY_CXX)
230 .Case("cui", TY_PP_CUDA)
231 .Case("cxx", TY_CXX)
232 .Case("CXX", TY_CXX)
233 .Case("F90", TY_Fortran)
234 .Case("f90", TY_PP_Fortran)
235 .Case("F95", TY_Fortran)
236 .Case("f95", TY_PP_Fortran)
237 .Case("for", TY_PP_Fortran)
238 .Case("FOR", TY_PP_Fortran)
239 .Case("fpp", TY_Fortran)
240 .Case("FPP", TY_Fortran)
241 .Case("gch", TY_PCH)
242 .Case("hip", TY_HIP)
243 .Case("hpp", TY_CXXHeader)
244 .Case("iim", TY_PP_CXXModule)
245 .Case("lib", TY_Object)
246 .Case("mii", TY_PP_ObjCXX)
247 .Case("obj", TY_Object)
248 .Case("pch", TY_PCH)
249 .Case("pcm", TY_ModuleFile)
250 .Case("c++m", TY_CXXModule)
251 .Case("cppm", TY_CXXModule)
252 .Case("cxxm", TY_CXXModule)
253 .Default(TY_INVALID);
254 }
255
lookupTypeForTypeSpecifier(const char * Name)256 types::ID types::lookupTypeForTypeSpecifier(const char *Name) {
257 for (unsigned i=0; i<numTypes; ++i) {
258 types::ID Id = (types::ID) (i + 1);
259 if (canTypeBeUserSpecified(Id) &&
260 strcmp(Name, getInfo(Id).Name) == 0)
261 return Id;
262 }
263
264 return TY_INVALID;
265 }
266
267 // FIXME: Why don't we just put this list in the defs file, eh.
getCompilationPhases(ID Id,llvm::SmallVectorImpl<phases::ID> & P)268 void types::getCompilationPhases(ID Id, llvm::SmallVectorImpl<phases::ID> &P) {
269 if (Id != TY_Object) {
270 if (getPreprocessedType(Id) != TY_INVALID) {
271 P.push_back(phases::Preprocess);
272 }
273
274 if (getPrecompiledType(Id) != TY_INVALID) {
275 P.push_back(phases::Precompile);
276 }
277
278 if (!onlyPrecompileType(Id)) {
279 if (!onlyAssembleType(Id)) {
280 P.push_back(phases::Compile);
281 P.push_back(phases::Backend);
282 }
283 P.push_back(phases::Assemble);
284 }
285 }
286
287 if (!onlyPrecompileType(Id)) {
288 P.push_back(phases::Link);
289 }
290 assert(0 < P.size() && "Not enough phases in list");
291 assert(P.size() <= phases::MaxNumberOfPhases && "Too many phases in list");
292 }
293
lookupCXXTypeForCType(ID Id)294 ID types::lookupCXXTypeForCType(ID Id) {
295 switch (Id) {
296 default:
297 return Id;
298
299 case types::TY_C:
300 return types::TY_CXX;
301 case types::TY_PP_C:
302 return types::TY_PP_CXX;
303 case types::TY_CHeader:
304 return types::TY_CXXHeader;
305 case types::TY_PP_CHeader:
306 return types::TY_PP_CXXHeader;
307 }
308 }
309
lookupHeaderTypeForSourceType(ID Id)310 ID types::lookupHeaderTypeForSourceType(ID Id) {
311 switch (Id) {
312 default:
313 return Id;
314
315 // FIXME: Handle preprocessed input types.
316 case types::TY_C:
317 return types::TY_CHeader;
318 case types::TY_CXX:
319 case types::TY_CXXModule:
320 return types::TY_CXXHeader;
321 case types::TY_ObjC:
322 return types::TY_ObjCHeader;
323 case types::TY_ObjCXX:
324 return types::TY_ObjCXXHeader;
325 case types::TY_CL:
326 return types::TY_CLHeader;
327 }
328 }
329