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