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