diff --git a/include/llvm/Support/GetElementPtrTypeIterator.h b/include/llvm/Support/GetElementPtrTypeIterator.h new file mode 100644 index 00000000000..b587a7e4e14 --- /dev/null +++ b/include/llvm/Support/GetElementPtrTypeIterator.h @@ -0,0 +1,92 @@ +//===- llvm/Support/GetElementPtrTypeIterator.h -----------------*- C++ -*-===// +// +// 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 implements an iterator for walking through the types indexed by +// getelementptr instructions. +// +//===----------------------------------------------------------------------===// + +#ifndef LLVM_SUPPORT_GETELEMENTPTRTYPE_H +#define LLVM_SUPPORT_GETELEMENTPTRTYPE_H + +#include "Support/iterator" +#include "llvm/iMemory.h" +#include "llvm/DerivedTypes.h" + +namespace llvm { + class GetElementPtrTypeIterator + : public forward_iterator { + typedef forward_iterator super; + + GetElementPtrInst *TheGEP; + const Type *CurTy; + unsigned Operand; + + GetElementPtrTypeIterator() {} + public: + + static GetElementPtrTypeIterator begin(GetElementPtrInst *gep) { + GetElementPtrTypeIterator I; + I.TheGEP = gep; + I.CurTy = gep->getOperand(0)->getType(); + I.Operand = 1; + return I; + } + static GetElementPtrTypeIterator end(GetElementPtrInst *gep) { + GetElementPtrTypeIterator I; + I.TheGEP = gep; + I.CurTy = 0; + I.Operand = gep->getNumOperands(); + return I; + } + + bool operator==(const GetElementPtrTypeIterator& x) const { + return Operand == x.Operand; + } + bool operator!=(const GetElementPtrTypeIterator& x) const { + return !operator==(x); + } + + const Type *operator*() const { + return CurTy; + } + + // This is a non-standard operator->. It allows you to call methods on the + // current type directly. + const Type *operator->() const { return operator*(); } + + unsigned getOperandNum() const { return Operand; } + + Value *getOperand() const { return TheGEP->getOperand(Operand); } + + GetElementPtrTypeIterator& operator++() { // Preincrement + if (const CompositeType *CT = dyn_cast(CurTy)) { + CurTy = CT->getTypeAtIndex(getOperand()); + } else { + CurTy = 0; + } + ++Operand; + return *this; + } + + GetElementPtrTypeIterator operator++(int) { // Postincrement + GetElementPtrTypeIterator tmp = *this; ++*this; return tmp; + } + }; + + inline GetElementPtrTypeIterator gep_type_begin(GetElementPtrInst *GEP) { + return GetElementPtrTypeIterator::begin(GEP); + } + + inline GetElementPtrTypeIterator gep_type_end(GetElementPtrInst *GEP) { + return GetElementPtrTypeIterator::end(GEP); + } +} // end namespace llvm + +#endif