diff --git a/llvm/include/llvm/ADT/SmallPtrSet.h b/llvm/include/llvm/ADT/SmallPtrSet.h index 6bd4d0927fe4..40e9bb4139c8 100644 --- a/llvm/include/llvm/ADT/SmallPtrSet.h +++ b/llvm/include/llvm/ADT/SmallPtrSet.h @@ -53,6 +53,9 @@ protected: unsigned NumElements; unsigned NumTombstones; void *SmallArray[1]; // Must be last ivar. + + // Helper to copy construct a SmallPtrSet. + SmallPtrSetImpl(const SmallPtrSetImpl& that); public: SmallPtrSetImpl(unsigned SmallSize) { assert(SmallSize && (SmallSize & (SmallSize-1)) == 0 && @@ -214,6 +217,7 @@ class SmallPtrSet : public SmallPtrSetImpl { void *SmallArray[SmallSizePowTwo]; public: SmallPtrSet() : SmallPtrSetImpl(NextPowerOfTwo::Val) {} + SmallPtrSet(const SmallPtrSet &that) : SmallPtrSetImpl(that) {} template SmallPtrSet(It I, It E) diff --git a/llvm/lib/Support/SmallPtrSet.cpp b/llvm/lib/Support/SmallPtrSet.cpp index 98d5bc99adc1..61fad5ea5063 100644 --- a/llvm/lib/Support/SmallPtrSet.cpp +++ b/llvm/lib/Support/SmallPtrSet.cpp @@ -141,5 +141,33 @@ void SmallPtrSetImpl::Grow() { } delete [] OldBuckets; + NumTombstones = 0; + } +} + +SmallPtrSetImpl::SmallPtrSetImpl(const SmallPtrSetImpl& that) { + NumElements = that.NumElements; + NumTombstones = 0; + if (that.isSmall()) { + CurArraySize = that.CurArraySize; + CurArray = &SmallArray[0]; + memcpy(CurArray, that.CurArray, sizeof(void*)*CurArraySize); + } else { + CurArraySize = that.NumElements < 64 ? 128 : that.NumElements*2; + CurArray = new void*[CurArraySize+1]; + memset(CurArray, -1, CurArraySize*sizeof(void*)); + + // The end pointer, always valid, is set to a valid element to help the + // iterator. + CurArray[CurArraySize] = 0; + + // Copy over all valid entries. + for (void **BucketPtr = that.CurArray, **E = that.CurArray+CurArraySize; + BucketPtr != E; ++BucketPtr) { + // Copy over the element if it is valid. + void *Elt = *BucketPtr; + if (Elt != getTombstoneMarker() && Elt != getEmptyMarker()) + *const_cast(FindBucketFor(Elt)) = Elt; + } } }