0407750bf261ea4324a55e8677f8e7500c27b8e7
[oota-llvm.git] / lib / Support / Triple.cpp
1 //===--- Triple.cpp - Target triple helper class --------------------------===//
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 "llvm/ADT/Triple.h"
11
12 #include "llvm/ADT/Twine.h"
13 #include <cassert>
14 #include <cstring>
15 using namespace llvm;
16
17 //
18
19 const char *Triple::getArchTypeName(ArchType Kind) {
20   switch (Kind) {
21   case InvalidArch: return "<invalid>";
22   case UnknownArch: return "unknown";
23     
24   case alpha:   return "alpha";
25   case arm:     return "arm";
26   case bfin:    return "bfin";
27   case cellspu: return "cellspu";
28   case mips:    return "mips";
29   case mipsel:  return "mipsel";
30   case msp430:  return "msp430";
31   case pic16:   return "pic16";
32   case ppc64:   return "powerpc64";
33   case ppc:     return "powerpc";
34   case sparc:   return "sparc";
35   case systemz: return "s390x";
36   case thumb:   return "thumb";
37   case x86:     return "i386";
38   case x86_64:  return "x86_64";
39   case xcore:   return "xcore";
40   }
41
42   return "<invalid>";
43 }
44
45 const char *Triple::getVendorTypeName(VendorType Kind) {
46   switch (Kind) {
47   case UnknownVendor: return "unknown";
48
49   case Apple: return "apple";
50   case PC: return "pc";
51   }
52
53   return "<invalid>";
54 }
55
56 const char *Triple::getOSTypeName(OSType Kind) {
57   switch (Kind) {
58   case UnknownOS: return "unknown";
59
60   case AuroraUX: return "auroraux";
61   case Cygwin: return "cygwin";
62   case Darwin: return "darwin";
63   case DragonFly: return "dragonfly";
64   case FreeBSD: return "freebsd";
65   case Linux: return "linux";
66   case MinGW32: return "mingw32";
67   case MinGW64: return "mingw64";
68   case NetBSD: return "netbsd";
69   case OpenBSD: return "openbsd";
70   case Solaris: return "solaris";
71   case Win32: return "win32";
72   }
73
74   return "<invalid>";
75 }
76
77 Triple::ArchType Triple::getArchTypeForLLVMName(const StringRef &Name) {
78   if (Name == "alpha")
79     return alpha;
80   if (Name == "arm")
81     return arm;
82   if (Name == "bfin")
83     return bfin;
84   if (Name == "cellspu")
85     return cellspu;
86   if (Name == "mips")
87     return mips;
88   if (Name == "mipsel")
89     return mipsel;
90   if (Name == "msp430")
91     return msp430;
92   if (Name == "pic16")
93     return pic16;
94   if (Name == "ppc64")
95     return ppc64;
96   if (Name == "ppc")
97     return ppc;
98   if (Name == "sparc")
99     return sparc;
100   if (Name == "systemz")
101     return systemz;
102   if (Name == "thumb")
103     return thumb;
104   if (Name == "x86")
105     return x86;
106   if (Name == "x86-64")
107     return x86_64;
108   if (Name == "xcore")
109     return xcore;
110
111   return UnknownArch;
112 }
113
114 //
115
116 void Triple::Parse() const {
117   assert(!isInitialized() && "Invalid parse call.");
118
119   StringRef ArchName = getArchName();
120   if (ArchName.size() == 4 && ArchName[0] == 'i' && 
121       ArchName[2] == '8' && ArchName[3] == '6' && 
122       ArchName[1] - '3' < 6) // i[3-9]86
123     Arch = x86;
124   else if (ArchName == "amd64" || ArchName == "x86_64")
125     Arch = x86_64;
126   else if (ArchName == "pic16")
127     Arch = pic16;
128   else if (ArchName == "powerpc")
129     Arch = ppc;
130   else if (ArchName == "powerpc64")
131     Arch = ppc64;
132   else if (ArchName == "arm" ||
133            ArchName.startswith("armv"))
134     Arch = arm;
135   else if (ArchName == "thumb" ||
136            ArchName.startswith("thumbv"))
137     Arch = thumb;
138   else if (ArchName.startswith("alpha"))
139     Arch = alpha;
140   else if (ArchName == "spu" || ArchName == "cellspu")
141     Arch = cellspu;
142   else if (ArchName == "msp430")
143     Arch = msp430;
144   else if (ArchName == "mips" || ArchName == "mipsallegrex")
145     Arch = mips;
146   else if (ArchName == "mipsel" || ArchName == "mipsallegrexel" ||
147            ArchName == "psp")
148     Arch = mipsel;
149   else if (ArchName == "sparc")
150     Arch = sparc;
151   else if (ArchName == "s390x")
152     Arch = systemz;
153   else
154     Arch = UnknownArch;
155
156   StringRef VendorName = getVendorName();
157   if (VendorName == "apple")
158     Vendor = Apple;
159   else if (VendorName == "pc")
160     Vendor = PC;
161   else
162     Vendor = UnknownVendor;
163
164   StringRef OSName = getOSName();
165   if (OSName.startswith("auroraux"))
166     OS = AuroraUX;
167   else if (OSName.startswith("cygwin"))
168     OS = Cygwin;
169   else if (OSName.startswith("darwin"))
170     OS = Darwin;
171   else if (OSName.startswith("dragonfly"))
172     OS = DragonFly;
173   else if (OSName.startswith("freebsd"))
174     OS = FreeBSD;
175   else if (OSName.startswith("linux"))
176     OS = Linux;
177   else if (OSName.startswith("mingw32"))
178     OS = MinGW32;
179   else if (OSName.startswith("mingw64"))
180     OS = MinGW64;
181   else if (OSName.startswith("netbsd"))
182     OS = NetBSD;
183   else if (OSName.startswith("openbsd"))
184     OS = OpenBSD;
185   else if (OSName.startswith("solaris"))
186     OS = Solaris;
187   else if (OSName.startswith("win32"))
188     OS = Win32;
189   else
190     OS = UnknownOS;
191
192   assert(isInitialized() && "Failed to initialize!");
193 }
194
195 StringRef Triple::getArchName() const {
196   return StringRef(Data).split('-').first;           // Isolate first component
197 }
198
199 StringRef Triple::getVendorName() const {
200   StringRef Tmp = StringRef(Data).split('-').second; // Strip first component
201   return Tmp.split('-').first;                       // Isolate second component
202 }
203
204 StringRef Triple::getOSName() const {
205   StringRef Tmp = StringRef(Data).split('-').second; // Strip first component
206   Tmp = Tmp.split('-').second;                       // Strip second component
207   return Tmp.split('-').first;                       // Isolate third component
208 }
209
210 StringRef Triple::getEnvironmentName() const {
211   StringRef Tmp = StringRef(Data).split('-').second; // Strip first component
212   Tmp = Tmp.split('-').second;                       // Strip second component
213   return Tmp.split('-').second;                      // Strip third component
214 }
215
216 StringRef Triple::getOSAndEnvironmentName() const {
217   StringRef Tmp = StringRef(Data).split('-').second; // Strip first component
218   return Tmp.split('-').second;                      // Strip second component
219 }
220
221 static unsigned EatNumber(StringRef &Str) {
222   assert(!Str.empty() && Str[0] >= '0' && Str[0] <= '9' && "Not a number");
223   unsigned Result = Str[0]-'0';
224   
225   // Eat the digit.
226   Str = Str.substr(1);
227   
228   // Handle "darwin11".
229   if (Result == 1 && !Str.empty() && Str[0] >= '0' && Str[0] <= '9') {
230     Result = Result*10 + (Str[0] - '0');
231     // Eat the digit.
232     Str = Str.substr(1);
233   }
234   
235   return Result;
236 }
237
238 /// getDarwinNumber - Parse the 'darwin number' out of the specific target
239 /// triple.  For example, if we have darwin8.5 return 8,5,0.  If any entry is
240 /// not defined, return 0's.  This requires that the triple have an OSType of
241 /// darwin before it is called.
242 void Triple::getDarwinNumber(unsigned &Maj, unsigned &Min,
243                              unsigned &Revision) const {
244   assert(getOS() == Darwin && "Not a darwin target triple!");
245   StringRef OSName = getOSName();
246   assert(OSName.startswith("darwin") && "Unknown darwin target triple!");
247   
248   // Strip off "darwin".
249   OSName = OSName.substr(6);
250   
251   Maj = Min = Revision = 0;
252
253   if (OSName.empty() || OSName[0] < '0' || OSName[0] > '9')
254     return;
255
256   // The major version is the first digit.
257   Maj = EatNumber(OSName);
258   if (OSName.empty()) return;
259   
260   // Handle minor version: 10.4.9 -> darwin8.9.
261   if (OSName[0] != '.')
262     return;
263   
264   // Eat the '.'.
265   OSName = OSName.substr(1);
266
267   if (OSName.empty() || OSName[0] < '0' || OSName[0] > '9')
268     return;
269   
270   Min = EatNumber(OSName);
271   if (OSName.empty()) return;
272
273   // Handle revision darwin8.9.1
274   if (OSName[0] != '.')
275     return;
276   
277   // Eat the '.'.
278   OSName = OSName.substr(1);
279   
280   if (OSName.empty() || OSName[0] < '0' || OSName[0] > '9')
281     return;
282
283   Revision = EatNumber(OSName);
284 }
285
286 void Triple::setTriple(const Twine &Str) {
287   Data = Str.str();
288   Arch = InvalidArch;
289 }
290
291 void Triple::setArch(ArchType Kind) {
292   setArchName(getArchTypeName(Kind));
293 }
294
295 void Triple::setVendor(VendorType Kind) {
296   setVendorName(getVendorTypeName(Kind));
297 }
298
299 void Triple::setOS(OSType Kind) {
300   setOSName(getOSTypeName(Kind));
301 }
302
303 void Triple::setArchName(const StringRef &Str) {
304   setTriple(Str + "-" + getVendorName() + "-" + getOSAndEnvironmentName());
305 }
306
307 void Triple::setVendorName(const StringRef &Str) {
308   setTriple(getArchName() + "-" + Str + "-" + getOSAndEnvironmentName());
309 }
310
311 void Triple::setOSName(const StringRef &Str) {
312   if (hasEnvironment())
313     setTriple(getArchName() + "-" + getVendorName() + "-" + Str +
314               "-" + getEnvironmentName());
315   else
316     setTriple(getArchName() + "-" + getVendorName() + "-" + Str);
317 }
318
319 void Triple::setEnvironmentName(const StringRef &Str) {
320   setTriple(getArchName() + "-" + getVendorName() + "-" + getOSName() + 
321             "-" + Str);
322 }
323
324 void Triple::setOSAndEnvironmentName(const StringRef &Str) {
325   setTriple(getArchName() + "-" + getVendorName() + "-" + Str);
326 }