Add comment.
[oota-llvm.git] / lib / VMCore / ValueTypes.cpp
index 81e19645b1fcc89f57718d66c732aff7e447d3fb..88add7c174c41cad04389b4dd76ed8d73b416461 100644 (file)
@@ -2,8 +2,8 @@
 //
 //                     The LLVM Compiler Infrastructure
 //
-// This file was developed by the LLVM research group and is distributed under
-// the University of Illinois Open Source License. See LICENSE.TXT for details.
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
 //
 //===----------------------------------------------------------------------===//
 //
 //
 //===----------------------------------------------------------------------===//
 
+#include "llvm/ADT/StringExtras.h"
 #include "llvm/CodeGen/ValueTypes.h"
 #include "llvm/Type.h"
+#include "llvm/DerivedTypes.h"
 using namespace llvm;
 
 /// MVT::getValueTypeString - This function returns value type as a string,
 /// e.g. "i32".
-const char *MVT::getValueTypeString(MVT::ValueType VT) {
+std::string MVT::getValueTypeString(MVT::ValueType VT) {
   switch (VT) {
-  default: assert(0 && "Invalid ValueType!");
-  case MVT::i1:    return "i1";
-  case MVT::i8:    return "i8";
-  case MVT::i16:   return "i16";
-  case MVT::i32:   return "i32";
-  case MVT::i64:   return "i64";
-  case MVT::i128:  return "i128";
-  case MVT::f32:   return "f32";
-  case MVT::f64:   return "f64";
-  case MVT::f80:   return "f80";
-  case MVT::f128:  return "f128";
-  case MVT::isVoid:return "isVoid";
-  case MVT::Other: return "ch";
-  case MVT::Flag:  return "flag";
-  case MVT::Vector:return "vec";
-  case MVT::v8i8:  return "v8i8";
-  case MVT::v4i16: return "v4i16";
-  case MVT::v2i32: return "v2i32";
-  case MVT::v16i8: return "v16i8";
-  case MVT::v8i16: return "v8i16";
-  case MVT::v4i32: return "v4i32";
-  case MVT::v2i64: return "v2i64";
-  case MVT::v4f32: return "v4f32";
-  case MVT::v2f64: return "v2f64";
+  default:
+    if (isVector(VT))
+      return "v" + utostr(getVectorNumElements(VT)) +
+             getValueTypeString(getVectorElementType(VT));
+    if (isInteger(VT))
+      return "i" + utostr(getSizeInBits(VT));
+    assert(0 && "Invalid ValueType!");
+  case MVT::i1:      return "i1";
+  case MVT::i8:      return "i8";
+  case MVT::i16:     return "i16";
+  case MVT::i32:     return "i32";
+  case MVT::i64:     return "i64";
+  case MVT::i128:    return "i128";
+  case MVT::f32:     return "f32";
+  case MVT::f64:     return "f64";
+  case MVT::f80:     return "f80";
+  case MVT::f128:    return "f128";
+  case MVT::ppcf128: return "ppcf128";
+  case MVT::isVoid:  return "isVoid";
+  case MVT::Other:   return "ch";
+  case MVT::Flag:    return "flag";
+  case MVT::v8i8:    return "v8i8";
+  case MVT::v4i16:   return "v4i16";
+  case MVT::v2i32:   return "v2i32";
+  case MVT::v1i64:   return "v1i64";
+  case MVT::v16i8:   return "v16i8";
+  case MVT::v8i16:   return "v8i16";
+  case MVT::v4i32:   return "v4i32";
+  case MVT::v2i64:   return "v2i64";
+  case MVT::v2f32:   return "v2f32";
+  case MVT::v4f32:   return "v4f32";
+  case MVT::v2f64:   return "v2f64";
+  case MVT::v3i32:   return "v3i32";
+  case MVT::v3f32:   return "v3f32";
   }
 }
 
 /// MVT::getTypeForValueType - This method returns an LLVM type corresponding
-/// to the specified ValueType.  For integer types, this returns an unsigned
-/// type.  Note that this will abort for types that cannot be represented.
+/// to the specified ValueType.  Note that this will abort for types that cannot
+/// be represented.
 const Type *MVT::getTypeForValueType(MVT::ValueType VT) {
   switch (VT) {
-  default: assert(0 && "ValueType does not correspond to LLVM type!");
+  default:
+    if (isVector(VT))
+      return VectorType::get(getTypeForValueType(getVectorElementType(VT)),
+                             getVectorNumElements(VT));
+    if (isInteger(VT))
+      return IntegerType::get(getSizeInBits(VT));
+    assert(0 && "ValueType does not correspond to LLVM type!");
   case MVT::isVoid:return Type::VoidTy;
-  case MVT::i1:    return Type::BoolTy;
-  case MVT::i8:    return Type::UByteTy;
-  case MVT::i16:   return Type::UShortTy;
-  case MVT::i32:   return Type::UIntTy;
-  case MVT::i64:   return Type::ULongTy;
-  case MVT::f32:   return Type::FloatTy;
-  case MVT::f64:   return Type::DoubleTy;
+  case MVT::i1:      return Type::Int1Ty;
+  case MVT::i8:      return Type::Int8Ty;
+  case MVT::i16:     return Type::Int16Ty;
+  case MVT::i32:     return Type::Int32Ty;
+  case MVT::i64:     return Type::Int64Ty;
+  case MVT::i128:    return IntegerType::get(128);
+  case MVT::f32:     return Type::FloatTy;
+  case MVT::f64:     return Type::DoubleTy;
+  case MVT::f80:     return Type::X86_FP80Ty;
+  case MVT::f128:    return Type::FP128Ty;
+  case MVT::ppcf128: return Type::PPC_FP128Ty;
+  case MVT::v8i8:    return VectorType::get(Type::Int8Ty, 8);
+  case MVT::v4i16:   return VectorType::get(Type::Int16Ty, 4);
+  case MVT::v2i32:   return VectorType::get(Type::Int32Ty, 2);
+  case MVT::v1i64:   return VectorType::get(Type::Int64Ty, 1);
+  case MVT::v16i8:   return VectorType::get(Type::Int8Ty, 16);
+  case MVT::v8i16:   return VectorType::get(Type::Int16Ty, 8);
+  case MVT::v4i32:   return VectorType::get(Type::Int32Ty, 4);
+  case MVT::v2i64:   return VectorType::get(Type::Int64Ty, 2);
+  case MVT::v2f32:   return VectorType::get(Type::FloatTy, 2);
+  case MVT::v4f32:   return VectorType::get(Type::FloatTy, 4);
+  case MVT::v2f64:   return VectorType::get(Type::DoubleTy, 2);
+  case MVT::v3i32:   return VectorType::get(Type::Int32Ty, 3);
+  case MVT::v3f32:   return VectorType::get(Type::FloatTy, 3);
+  }
+}
+
+/// MVT::getValueType - Return the value type corresponding to the specified
+/// type.  This returns all pointers as MVT::iPTR.  If HandleUnknown is true,
+/// unknown types are returned as Other, otherwise they are invalid.
+MVT::ValueType MVT::getValueType(const Type *Ty, bool HandleUnknown) {
+  switch (Ty->getTypeID()) {
+  default:
+    if (HandleUnknown) return MVT::Other;
+    assert(0 && "Unknown type!");
+  case Type::VoidTyID:
+    return MVT::isVoid;
+  case Type::IntegerTyID:
+    return getIntegerType(cast<IntegerType>(Ty)->getBitWidth());
+  case Type::FloatTyID:     return MVT::f32;
+  case Type::DoubleTyID:    return MVT::f64;
+  case Type::X86_FP80TyID:  return MVT::f80;
+  case Type::FP128TyID:     return MVT::f128;
+  case Type::PPC_FP128TyID: return MVT::ppcf128;
+  case Type::PointerTyID:   return MVT::iPTR;
+  case Type::VectorTyID: {
+    const VectorType *VTy = cast<VectorType>(Ty);
+    return getVectorType(getValueType(VTy->getElementType(), false),
+                         VTy->getNumElements());
+  }
   }
 }