summaryrefslogtreecommitdiff
path: root/include
diff options
context:
space:
mode:
authorNick Kledzik <kledzik@apple.com>2014-01-28 19:21:27 +0000
committerNick Kledzik <kledzik@apple.com>2014-01-28 19:21:27 +0000
commit1c71a20f32b6c6af85fb26d56b26effe71c710b9 (patch)
tree90da4f83c000aff69d4a6bb33e2b1fced970a661 /include
parent3dcb2a2d92cca5d5ebac11fafab923520341b265 (diff)
downloadllvm-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.h36
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;
};