diff options
author | Nick Kledzik <kledzik@apple.com> | 2014-01-28 19:21:27 +0000 |
---|---|---|
committer | Nick Kledzik <kledzik@apple.com> | 2014-01-28 19:21:27 +0000 |
commit | 1c71a20f32b6c6af85fb26d56b26effe71c710b9 (patch) | |
tree | 90da4f83c000aff69d4a6bb33e2b1fced970a661 /include | |
parent | 3dcb2a2d92cca5d5ebac11fafab923520341b265 (diff) | |
download | llvm-1c71a20f32b6c6af85fb26d56b26effe71c710b9.tar.gz llvm-1c71a20f32b6c6af85fb26d56b26effe71c710b9.tar.bz2 llvm-1c71a20f32b6c6af85fb26d56b26effe71c710b9.tar.xz |
Add BumpPtrAllocator::allocateCopy() utilities
Makes it easy to use BumpPtrAllocator to make a copy of StringRef strings.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@200331 91177308-0d34-0410-b5e6-96231b3b80d8
Diffstat (limited to 'include')
-rw-r--r-- | include/llvm/Support/Allocator.h | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/include/llvm/Support/Allocator.h b/include/llvm/Support/Allocator.h index 397f50fbe3..9f7611be02 100644 --- a/include/llvm/Support/Allocator.h +++ b/include/llvm/Support/Allocator.h @@ -14,6 +14,8 @@ #ifndef LLVM_SUPPORT_ALLOCATOR_H #define LLVM_SUPPORT_ALLOCATOR_H +#include "llvm/ADT/ArrayRef.h" +#include "llvm/ADT/StringRef.h" #include "llvm/Support/AlignOf.h" #include "llvm/Support/DataTypes.h" #include "llvm/Support/MathExtras.h" @@ -179,6 +181,40 @@ public: void PrintStats() const; + + /// Allocate space and copy content into it. + void *allocateCopy(const void *Src, size_t Size, size_t Alignment=1) { + void *P = Allocate(Size, Alignment); + memcpy(P, Src, Size); + return P; + } + + /// Allocate space for an array of type T, and use std::copy() + /// to copy the array contents. + template <typename T> + typename enable_if<isPodLike<T>, T*>::type + allocateCopy(const T Src[], size_t Num) { + T *P = Allocate<T>(Num); + std::copy(Src, &Src[Num], P); + return P; + } + + /// Copy a StringRef by allocating copy in BumpPtrAllocator. + StringRef allocateCopy(StringRef Str) { + size_t Length = Str.size(); + char *P = allocateCopy<char>(Str.data(), Length); + return StringRef(P, Length); + } + + /// Copy a ArrayRef<T> by allocating copy in BumpPtrAllocator. + template <typename T> + typename enable_if<isPodLike<T>, ArrayRef<T>>::type + allocateCopy(ArrayRef<T> Src) { + size_t Length = Src.size(); + T *P = allocateCopy(Src.data(), Length*sizeof(T)); + return makeArrayRef(P, Length); + } + /// Compute the total physical memory allocated by this allocator. size_t getTotalMemory() const; }; |