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 
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 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 
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 
72 bool types::onlyAssembleType(ID Id) {
73   return strchr(getInfo(Id).Flags, 'a');
74 }
75 
76 bool types::onlyPrecompileType(ID Id) {
77   return strchr(getInfo(Id).Flags, 'p');
78 }
79 
80 bool types::canTypeBeUserSpecified(ID Id) {
81   return strchr(getInfo(Id).Flags, 'u');
82 }
83 
84 bool types::appendSuffixForType(ID Id) {
85   return strchr(getInfo(Id).Flags, 'A');
86 }
87 
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 
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 
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 
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 
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 
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 
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 
191 bool types::isSrcFile(ID Id) {
192   return Id != TY_Object && getPreprocessedType(Id) != TY_INVALID;
193 }
194 
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 
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.
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 
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 
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