Type safety for Constants.cpp! Some of this is temporary, as I'm planning to push...
[oota-llvm.git] / lib / VMCore / ValueTypes.cpp
index 9fb76554f7f31064d72f91c1eb1baaf05f25b370..fe4af053a2a60cbf5f3750f3994d1ff53b4ce4eb 100644 (file)
@@ -1,9 +1,9 @@
-//===-- ValueTypes.cpp - Implementation of MVT::ValueType methods ---------===//
+//===----------- ValueTypes.cpp - Implementation of MVT methods -----------===//
 //
 //                     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) {
-  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::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::v4f32: return "v4f32";
-  case MVT::v2f64: return "v2f64";
+MVT MVT::getExtendedIntegerVT(unsigned BitWidth) {
+  MVT VT;
+  VT.LLVMTy = IntegerType::get(BitWidth);
+  assert(VT.isExtended() && "Type is not extended!");
+  return VT;
+}
+
+MVT MVT::getExtendedVectorVT(MVT VT, unsigned NumElements) {
+  MVT ResultVT;
+  ResultVT.LLVMTy = VectorType::get(VT.getTypeForMVT(), NumElements);
+  assert(ResultVT.isExtended() && "Type is not extended!");
+  return ResultVT;
+}
+
+bool MVT::isExtendedFloatingPoint() const {
+  assert(isExtended() && "Type is not extended!");
+  return LLVMTy->isFPOrFPVector();
+}
+
+bool MVT::isExtendedInteger() const {
+  assert(isExtended() && "Type is not extended!");
+  return LLVMTy->isIntOrIntVector();
+}
+
+bool MVT::isExtendedVector() const {
+  assert(isExtended() && "Type is not extended!");
+  return isa<VectorType>(LLVMTy);
+}
+
+bool MVT::isExtended64BitVector() const {
+  return isExtendedVector() && getSizeInBits() == 64;
+}
+
+bool MVT::isExtended128BitVector() const {
+  return isExtendedVector() && getSizeInBits() == 128;
+}
+
+MVT MVT::getExtendedVectorElementType() const {
+  assert(isExtended() && "Type is not extended!");
+  return MVT::getMVT(cast<VectorType>(LLVMTy)->getElementType());
+}
+
+unsigned MVT::getExtendedVectorNumElements() const {
+  assert(isExtended() && "Type is not extended!");
+  return cast<VectorType>(LLVMTy)->getNumElements();
+}
+
+unsigned MVT::getExtendedSizeInBits() const {
+  assert(isExtended() && "Type is not extended!");
+  if (const IntegerType *ITy = dyn_cast<IntegerType>(LLVMTy))
+    return ITy->getBitWidth();
+  if (const VectorType *VTy = dyn_cast<VectorType>(LLVMTy))
+    return VTy->getBitWidth();
+  assert(false && "Unrecognized extended type!");
+  return 0; // Suppress warnings.
+}
+
+/// getMVTString - This function returns value type as a string, e.g. "i32".
+std::string MVT::getMVTString() const {
+  switch (V) {
+  default:
+    if (isVector())
+      return "v" + utostr(getVectorNumElements()) +
+             getVectorElementType().getMVTString();
+    if (isInteger())
+      return "i" + utostr(getSizeInBits());
+    assert(0 && "Invalid MVT!");
+    return "?";
+  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::v2i8:    return "v2i8";
+  case MVT::v4i8:    return "v4i8";
+  case MVT::v2i16:   return "v2i16";
+  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::getVectorType - Returns the ValueType that represents a vector
-/// NumElements in length, where each element is of type VT.  If there is no
-/// ValueType that represents this vector, a ValueType of Other is returned.
-///
-MVT::ValueType MVT::getVectorType(ValueType VT, unsigned NumElements) {
-  switch (VT) {
-  default: 
-    break;
-  case MVT::i8:
-    if (NumElements == 8)  return MVT::v8i8;
-    if (NumElements == 16) return MVT::v16i8;
-    break;
-  case MVT::i16:
-    if (NumElements == 4)  return MVT::v4i16;
-    if (NumElements == 8)  return MVT::v8i16;
-    break;
-  case MVT::i32:
-    if (NumElements == 2)  return MVT::v2i32;
-    if (NumElements == 4)  return MVT::v4i32;
-    break;
-  case MVT::i64:
-    if (NumElements == 1)  return MVT::v1i64;
-    if (NumElements == 2)  return MVT::v2i64;
-    break;
-  case MVT::f32:
-    if (NumElements == 2)  return MVT::v2f32;
-    if (NumElements == 4)  return MVT::v4f32;
-    break;
-  case MVT::f64:
-    if (NumElements == 2)  return MVT::v2f64;
-    break;
+/// getTypeForMVT - This method returns an LLVM type corresponding to the
+/// specified MVT.  For integer types, this returns an unsigned type.  Note
+/// that this will abort for types that cannot be represented.
+const Type *MVT::getTypeForMVT() const {
+  switch (V) {
+  default:
+    assert(isExtended() && "Type is not extended!");
+    return LLVMTy;
+  case MVT::isVoid:  return Type::VoidTy;
+  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::v2i8:    return VectorType::get(Type::Int8Ty, 2);
+  case MVT::v4i8:    return VectorType::get(Type::Int8Ty, 4);
+  case MVT::v2i16:    return VectorType::get(Type::Int16Ty, 2);
+  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);
   }
-  return MVT::Other;
 }
 
-/// MVT::getTypeForValueType - This method returns an LLVM type corresponding
-/// 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!");
-  case MVT::isVoid:return Type::VoidTy;
-  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::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::v4f32: return VectorType::get(Type::FloatTy, 4);
-  case MVT::v2f64: return VectorType::get(Type::DoubleTy, 2);
+/// getMVT - 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 MVT::getMVT(const Type *Ty, bool HandleUnknown){
+  switch (Ty->getTypeID()) {
+  default:
+    if (HandleUnknown) return MVT::Other;
+    assert(0 && "Unknown type!");
+    return MVT::isVoid;
+  case Type::VoidTyID:
+    return MVT::isVoid;
+  case Type::IntegerTyID:
+    return getIntegerVT(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 getVectorVT(getMVT(VTy->getElementType(), false),
+                       VTy->getNumElements());
+  }
   }
 }