365c083695339ffe069ed2e3b41e6fb6cb0174d5
[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/SmallString.h"
13 #include "llvm/ADT/Twine.h"
14 #include <cassert>
15 #include <cstring>
16 using namespace llvm;
17
18 //
19
20 const char *Triple::getArchTypeName(ArchType Kind) {
21   switch (Kind) {
22   case InvalidArch: return "<invalid>";
23   case UnknownArch: return "unknown";
24     
25   case alpha:   return "alpha";
26   case arm:     return "arm";
27   case bfin:    return "bfin";
28   case cellspu: return "cellspu";
29   case mips:    return "mips";
30   case mipsel:  return "mipsel";
31   case msp430:  return "msp430";
32   case pic16:   return "pic16";
33   case ppc64:   return "powerpc64";
34   case ppc:     return "powerpc";
35   case sparc:   return "sparc";
36   case sparcv9: return "sparcv9";
37   case systemz: return "s390x";
38   case tce:     return "tce";
39   case thumb:   return "thumb";
40   case x86:     return "i386";
41   case x86_64:  return "x86_64";
42   case xcore:   return "xcore";
43   case mblaze:  return "mblaze";
44   case ptx:     return "ptx";
45   }
46
47   return "<invalid>";
48 }
49
50 const char *Triple::getArchTypePrefix(ArchType Kind) {
51   switch (Kind) {
52   default:
53     return 0;
54
55   case alpha:   return "alpha";
56
57   case arm:
58   case thumb:   return "arm";
59
60   case bfin:    return "bfin";
61
62   case cellspu: return "spu";
63
64   case ppc64:
65   case ppc:     return "ppc";
66
67   case mblaze:  return "mblaze";
68
69   case sparcv9:
70   case sparc:   return "sparc";
71
72   case x86:
73   case x86_64:  return "x86";
74
75   case xcore:   return "xcore";
76
77   case ptx:     return "ptx";
78   }
79 }
80
81 const char *Triple::getVendorTypeName(VendorType Kind) {
82   switch (Kind) {
83   case UnknownVendor: return "unknown";
84
85   case Apple: return "apple";
86   case PC: return "pc";
87   }
88
89   return "<invalid>";
90 }
91
92 const char *Triple::getOSTypeName(OSType Kind) {
93   switch (Kind) {
94   case UnknownOS: return "unknown";
95
96   case AuroraUX: return "auroraux";
97   case Cygwin: return "cygwin";
98   case Darwin: return "darwin";
99   case DragonFly: return "dragonfly";
100   case FreeBSD: return "freebsd";
101   case Linux: return "linux";
102   case Lv2: return "lv2";
103   case MinGW32: return "mingw32";
104   case MinGW64: return "mingw64";
105   case NetBSD: return "netbsd";
106   case OpenBSD: return "openbsd";
107   case Psp: return "psp";
108   case Solaris: return "solaris";
109   case Win32: return "win32";
110   case Haiku: return "haiku";
111   case Minix: return "minix";
112   }
113
114   return "<invalid>";
115 }
116
117 Triple::ArchType Triple::getArchTypeForLLVMName(StringRef Name) {
118   if (Name == "alpha")
119     return alpha;
120   if (Name == "arm")
121     return arm;
122   if (Name == "bfin")
123     return bfin;
124   if (Name == "cellspu")
125     return cellspu;
126   if (Name == "mips")
127     return mips;
128   if (Name == "mipsel")
129     return mipsel;
130   if (Name == "msp430")
131     return msp430;
132   if (Name == "pic16")
133     return pic16;
134   if (Name == "ppc64")
135     return ppc64;
136   if (Name == "ppc")
137     return ppc;
138   if (Name == "mblaze")
139     return mblaze;
140   if (Name == "sparc")
141     return sparc;
142   if (Name == "sparcv9")
143     return sparcv9;
144   if (Name == "systemz")
145     return systemz;
146   if (Name == "tce")
147     return tce;
148   if (Name == "thumb")
149     return thumb;
150   if (Name == "x86")
151     return x86;
152   if (Name == "x86-64")
153     return x86_64;
154   if (Name == "xcore")
155     return xcore;
156   if (Name == "ptx")
157     return ptx;
158
159   return UnknownArch;
160 }
161
162 Triple::ArchType Triple::getArchTypeForDarwinArchName(StringRef Str) {
163   // See arch(3) and llvm-gcc's driver-driver.c. We don't implement support for
164   // archs which Darwin doesn't use.
165
166   // The matching this routine does is fairly pointless, since it is neither the
167   // complete architecture list, nor a reasonable subset. The problem is that
168   // historically the driver driver accepts this and also ties its -march=
169   // handling to the architecture name, so we need to be careful before removing
170   // support for it.
171
172   // This code must be kept in sync with Clang's Darwin specific argument
173   // translation.
174
175   if (Str == "ppc" || Str == "ppc601" || Str == "ppc603" || Str == "ppc604" ||
176       Str == "ppc604e" || Str == "ppc750" || Str == "ppc7400" ||
177       Str == "ppc7450" || Str == "ppc970")
178     return Triple::ppc;
179
180   if (Str == "ppc64")
181     return Triple::ppc64;
182
183   if (Str == "i386" || Str == "i486" || Str == "i486SX" || Str == "pentium" ||
184       Str == "i586" || Str == "pentpro" || Str == "i686" || Str == "pentIIm3" ||
185       Str == "pentIIm5" || Str == "pentium4")
186     return Triple::x86;
187
188   if (Str == "x86_64")
189     return Triple::x86_64;
190
191   // This is derived from the driver driver.
192   if (Str == "arm" || Str == "armv4t" || Str == "armv5" || Str == "xscale" ||
193       Str == "armv6" || Str == "armv7")
194     return Triple::arm;
195
196   if (Str == "ptx")
197     return Triple::ptx;
198
199   return Triple::UnknownArch;
200 }
201
202 // Returns architecture name that is understood by the target assembler.
203 const char *Triple::getArchNameForAssembler() {
204   if (getOS() != Triple::Darwin && getVendor() != Triple::Apple)
205     return NULL;
206
207   StringRef Str = getArchName();
208   if (Str == "i386")
209     return "i386";
210   if (Str == "x86_64")
211     return "x86_64";
212   if (Str == "powerpc")
213     return "ppc";
214   if (Str == "powerpc64")
215     return "ppc64";
216   if (Str == "mblaze" || Str == "microblaze")
217     return "mblaze";
218   if (Str == "arm")
219     return "arm";
220   if (Str == "armv4t" || Str == "thumbv4t")
221     return "armv4t";
222   if (Str == "armv5" || Str == "armv5e" || Str == "thumbv5" || Str == "thumbv5e")
223     return "armv5";
224   if (Str == "armv6" || Str == "thumbv6")
225     return "armv6";
226   if (Str == "armv7" || Str == "thumbv7")
227     return "armv7";
228   if (Str == "ptx")
229     return "ptx";
230   return NULL;
231 }
232
233 //
234
235 Triple::ArchType Triple::ParseArch(StringRef ArchName) {
236   if (ArchName.size() == 4 && ArchName[0] == 'i' && 
237       ArchName[2] == '8' && ArchName[3] == '6' && 
238       ArchName[1] - '3' < 6) // i[3-9]86
239     return x86;
240   else if (ArchName == "amd64" || ArchName == "x86_64")
241     return x86_64;
242   else if (ArchName == "bfin")
243     return bfin;
244   else if (ArchName == "pic16")
245     return pic16;
246   else if (ArchName == "powerpc")
247     return ppc;
248   else if ((ArchName == "powerpc64") || (ArchName == "ppu"))
249     return ppc64;
250   else if (ArchName == "mblaze")
251     return mblaze;
252   else if (ArchName == "arm" ||
253            ArchName.startswith("armv") ||
254            ArchName == "xscale")
255     return arm;
256   else if (ArchName == "thumb" ||
257            ArchName.startswith("thumbv"))
258     return thumb;
259   else if (ArchName.startswith("alpha"))
260     return alpha;
261   else if (ArchName == "spu" || ArchName == "cellspu")
262     return cellspu;
263   else if (ArchName == "msp430")
264     return msp430;
265   else if (ArchName == "mips" || ArchName == "mipsallegrex")
266     return mips;
267   else if (ArchName == "mipsel" || ArchName == "mipsallegrexel" ||
268            ArchName == "psp")
269     return mipsel;
270   else if (ArchName == "sparc")
271     return sparc;
272   else if (ArchName == "sparcv9")
273     return sparcv9;
274   else if (ArchName == "s390x")
275     return systemz;
276   else if (ArchName == "tce")
277     return tce;
278   else if (ArchName == "xcore")
279     return xcore;
280   else if (ArchName == "ptx")
281     return ptx;
282   else
283     return UnknownArch;
284 }
285
286 Triple::VendorType Triple::ParseVendor(StringRef VendorName) {
287   if (VendorName == "apple")
288     return Apple;
289   else if (VendorName == "pc")
290     return PC;
291   else
292     return UnknownVendor;
293 }
294
295 Triple::OSType Triple::ParseOS(StringRef OSName) {
296   if (OSName.startswith("auroraux"))
297     return AuroraUX;
298   else if (OSName.startswith("cygwin"))
299     return Cygwin;
300   else if (OSName.startswith("darwin"))
301     return Darwin;
302   else if (OSName.startswith("dragonfly"))
303     return DragonFly;
304   else if (OSName.startswith("freebsd"))
305     return FreeBSD;
306   else if (OSName.startswith("linux"))
307     return Linux;
308   else if (OSName.startswith("lv2"))
309     return Lv2;
310   else if (OSName.startswith("mingw32"))
311     return MinGW32;
312   else if (OSName.startswith("mingw64"))
313     return MinGW64;
314   else if (OSName.startswith("netbsd"))
315     return NetBSD;
316   else if (OSName.startswith("openbsd"))
317     return OpenBSD;
318   else if (OSName.startswith("psp"))
319     return Psp;
320   else if (OSName.startswith("solaris"))
321     return Solaris;
322   else if (OSName.startswith("win32"))
323     return Win32;
324   else if (OSName.startswith("haiku"))
325     return Haiku;
326   else if (OSName.startswith("minix"))
327     return Minix;
328   else
329     return UnknownOS;
330 }
331
332 void Triple::Parse() const {
333   assert(!isInitialized() && "Invalid parse call.");
334
335   Arch = ParseArch(getArchName());
336   Vendor = ParseVendor(getVendorName());
337   OS = ParseOS(getOSName());
338
339   assert(isInitialized() && "Failed to initialize!");
340 }
341
342 std::string Triple::normalize(StringRef Str) {
343   // Parse into components.
344   SmallVector<StringRef, 4> Components;
345   for (size_t First = 0, Last = 0; Last != StringRef::npos; First = Last + 1) {
346     Last = Str.find('-', First);
347     Components.push_back(Str.slice(First, Last));
348   }
349
350   // If the first component corresponds to a known architecture, preferentially
351   // use it for the architecture.  If the second component corresponds to a
352   // known vendor, preferentially use it for the vendor, etc.  This avoids silly
353   // component movement when a component parses as (eg) both a valid arch and a
354   // valid os.
355   ArchType Arch = UnknownArch;
356   if (Components.size() > 0)
357     Arch = ParseArch(Components[0]);
358   VendorType Vendor = UnknownVendor;
359   if (Components.size() > 1)
360     Vendor = ParseVendor(Components[1]);
361   OSType OS = UnknownOS;
362   if (Components.size() > 2)
363     OS = ParseOS(Components[2]);
364
365   // Note which components are already in their final position.  These will not
366   // be moved.
367   bool Found[3];
368   Found[0] = Arch != UnknownArch;
369   Found[1] = Vendor != UnknownVendor;
370   Found[2] = OS != UnknownOS;
371
372   // If they are not there already, permute the components into their canonical
373   // positions by seeing if they parse as a valid architecture, and if so moving
374   // the component to the architecture position etc.
375   for (unsigned Pos = 0; Pos != 3; ++Pos) {
376     if (Found[Pos])
377       continue; // Already in the canonical position.
378
379     for (unsigned Idx = 0; Idx != Components.size(); ++Idx) {
380       // Do not reparse any components that already matched.
381       if (Idx < 3 && Found[Idx])
382         continue;
383
384       // Does this component parse as valid for the target position?
385       bool Valid = false;
386       StringRef Comp = Components[Idx];
387       switch (Pos) {
388       default:
389         assert(false && "unexpected component type!");
390       case 0:
391         Arch = ParseArch(Comp);
392         Valid = Arch != UnknownArch;
393         break;
394       case 1:
395         Vendor = ParseVendor(Comp);
396         Valid = Vendor != UnknownVendor;
397         break;
398       case 2:
399         OS = ParseOS(Comp);
400         Valid = OS != UnknownOS;
401         break;
402       }
403       if (!Valid)
404         continue; // Nope, try the next component.
405
406       // Move the component to the target position, pushing any non-fixed
407       // components that are in the way to the right.  This tends to give
408       // good results in the common cases of a forgotten vendor component
409       // or a wrongly positioned environment.
410       if (Pos < Idx) {
411         // Insert left, pushing the existing components to the right.  For
412         // example, a-b-i386 -> i386-a-b when moving i386 to the front.
413         StringRef CurrentComponent(""); // The empty component.
414         // Replace the component we are moving with an empty component.
415         std::swap(CurrentComponent, Components[Idx]);
416         // Insert the component being moved at Pos, displacing any existing
417         // components to the right.
418         for (unsigned i = Pos; !CurrentComponent.empty(); ++i) {
419           // Skip over any fixed components.
420           while (i < 3 && Found[i]) ++i;
421           // Place the component at the new position, getting the component
422           // that was at this position - it will be moved right.
423           std::swap(CurrentComponent, Components[i]);
424         }
425       } else if (Pos > Idx) {
426         // Push right by inserting empty components until the component at Idx
427         // reaches the target position Pos.  For example, pc-a -> -pc-a when
428         // moving pc to the second position.
429         do {
430           // Insert one empty component at Idx.
431           StringRef CurrentComponent(""); // The empty component.
432           for (unsigned i = Idx; i < Components.size(); ++i) {
433             // Skip over any fixed components.
434             while (i < 3 && Found[i]) ++i;
435             // Place the component at the new position, getting the component
436             // that was at this position - it will be moved right.
437             std::swap(CurrentComponent, Components[i]);
438             // If it was placed on top of an empty component then we are done.
439             if (CurrentComponent.empty())
440               break;
441           }
442           // The last component was pushed off the end - append it.
443           if (!CurrentComponent.empty())
444             Components.push_back(CurrentComponent);
445
446           // Advance Idx to the component's new position.
447           while (++Idx < 3 && Found[Idx]) {}
448         } while (Idx < Pos); // Add more until the final position is reached.
449       }
450       assert(Pos < Components.size() && Components[Pos] == Comp &&
451              "Component moved wrong!");
452       Found[Pos] = true;
453       break;
454     }
455   }
456
457   // Special case logic goes here.  At this point Arch, Vendor and OS have the
458   // correct values for the computed components.
459
460   // Stick the corrected components back together to form the normalized string.
461   std::string Normalized;
462   for (unsigned i = 0, e = Components.size(); i != e; ++i) {
463     if (i) Normalized += '-';
464     Normalized += Components[i];
465   }
466   return Normalized;
467 }
468
469 StringRef Triple::getArchName() const {
470   return StringRef(Data).split('-').first;           // Isolate first component
471 }
472
473 StringRef Triple::getVendorName() const {
474   StringRef Tmp = StringRef(Data).split('-').second; // Strip first component
475   return Tmp.split('-').first;                       // Isolate second component
476 }
477
478 StringRef Triple::getOSName() const {
479   StringRef Tmp = StringRef(Data).split('-').second; // Strip first component
480   Tmp = Tmp.split('-').second;                       // Strip second component
481   return Tmp.split('-').first;                       // Isolate third component
482 }
483
484 StringRef Triple::getEnvironmentName() const {
485   StringRef Tmp = StringRef(Data).split('-').second; // Strip first component
486   Tmp = Tmp.split('-').second;                       // Strip second component
487   return Tmp.split('-').second;                      // Strip third component
488 }
489
490 StringRef Triple::getOSAndEnvironmentName() const {
491   StringRef Tmp = StringRef(Data).split('-').second; // Strip first component
492   return Tmp.split('-').second;                      // Strip second component
493 }
494
495 static unsigned EatNumber(StringRef &Str) {
496   assert(!Str.empty() && Str[0] >= '0' && Str[0] <= '9' && "Not a number");
497   unsigned Result = Str[0]-'0';
498   
499   // Eat the digit.
500   Str = Str.substr(1);
501   
502   // Handle "darwin11".
503   if (Result == 1 && !Str.empty() && Str[0] >= '0' && Str[0] <= '9') {
504     Result = Result*10 + (Str[0] - '0');
505     // Eat the digit.
506     Str = Str.substr(1);
507   }
508   
509   return Result;
510 }
511
512 /// getDarwinNumber - Parse the 'darwin number' out of the specific target
513 /// triple.  For example, if we have darwin8.5 return 8,5,0.  If any entry is
514 /// not defined, return 0's.  This requires that the triple have an OSType of
515 /// darwin before it is called.
516 void Triple::getDarwinNumber(unsigned &Maj, unsigned &Min,
517                              unsigned &Revision) const {
518   assert(getOS() == Darwin && "Not a darwin target triple!");
519   StringRef OSName = getOSName();
520   assert(OSName.startswith("darwin") && "Unknown darwin target triple!");
521   
522   // Strip off "darwin".
523   OSName = OSName.substr(6);
524   
525   Maj = Min = Revision = 0;
526
527   if (OSName.empty() || OSName[0] < '0' || OSName[0] > '9')
528     return;
529
530   // The major version is the first digit.
531   Maj = EatNumber(OSName);
532   if (OSName.empty()) return;
533   
534   // Handle minor version: 10.4.9 -> darwin8.9.
535   if (OSName[0] != '.')
536     return;
537   
538   // Eat the '.'.
539   OSName = OSName.substr(1);
540
541   if (OSName.empty() || OSName[0] < '0' || OSName[0] > '9')
542     return;
543   
544   Min = EatNumber(OSName);
545   if (OSName.empty()) return;
546
547   // Handle revision darwin8.9.1
548   if (OSName[0] != '.')
549     return;
550   
551   // Eat the '.'.
552   OSName = OSName.substr(1);
553   
554   if (OSName.empty() || OSName[0] < '0' || OSName[0] > '9')
555     return;
556
557   Revision = EatNumber(OSName);
558 }
559
560 void Triple::setTriple(const Twine &Str) {
561   Data = Str.str();
562   Arch = InvalidArch;
563 }
564
565 void Triple::setArch(ArchType Kind) {
566   setArchName(getArchTypeName(Kind));
567 }
568
569 void Triple::setVendor(VendorType Kind) {
570   setVendorName(getVendorTypeName(Kind));
571 }
572
573 void Triple::setOS(OSType Kind) {
574   setOSName(getOSTypeName(Kind));
575 }
576
577 void Triple::setArchName(StringRef Str) {
578   // Work around a miscompilation bug for Twines in gcc 4.0.3.
579   SmallString<64> Triple;
580   Triple += Str;
581   Triple += "-";
582   Triple += getVendorName();
583   Triple += "-";
584   Triple += getOSAndEnvironmentName();
585   setTriple(Triple.str());
586 }
587
588 void Triple::setVendorName(StringRef Str) {
589   setTriple(getArchName() + "-" + Str + "-" + getOSAndEnvironmentName());
590 }
591
592 void Triple::setOSName(StringRef Str) {
593   if (hasEnvironment())
594     setTriple(getArchName() + "-" + getVendorName() + "-" + Str +
595               "-" + getEnvironmentName());
596   else
597     setTriple(getArchName() + "-" + getVendorName() + "-" + Str);
598 }
599
600 void Triple::setEnvironmentName(StringRef Str) {
601   setTriple(getArchName() + "-" + getVendorName() + "-" + getOSName() +
602             "-" + Str);
603 }
604
605 void Triple::setOSAndEnvironmentName(StringRef Str) {
606   setTriple(getArchName() + "-" + getVendorName() + "-" + Str);
607 }