From: Marcelo Juchem <marcelo@fb.com>
Date: Mon, 1 Apr 2013 23:00:01 +0000 (-0700)
Subject: as_stl_allocator "template alias"
X-Git-Tag: v0.22.0~1017
X-Git-Url: http://demsky.eecs.uci.edu/git/?a=commitdiff_plain;h=c246dd22a0dc296fc3bab7837e790b0fbf4deed5;p=folly.git

as_stl_allocator "template alias"

Summary: Implementing as_stl_allocator as a companion to the existing make_stl_allocator

Test Plan: unit tests added

Reviewed By: jon.coens@fb.com

FB internal diff: D755207
---

diff --git a/folly/Memory.h b/folly/Memory.h
index 09f10f36..00534177 100644
--- a/folly/Memory.h
+++ b/folly/Memory.h
@@ -210,6 +210,18 @@ public:
     && !has_destroy<allocator, void(void*)>::value;
 };
 
+template <typename T, typename Allocator>
+struct as_stl_allocator {
+  typedef typename std::conditional<
+    is_simple_allocator<T, Allocator>::value,
+    folly::StlAllocator<
+      typename std::remove_reference<Allocator>::type,
+      typename std::remove_reference<T>::type
+    >,
+    typename std::remove_reference<Allocator>::type
+  >::type type;
+};
+
 template <typename T, typename Allocator>
 typename std::enable_if<
   is_simple_allocator<T, Allocator>::value,
@@ -232,6 +244,13 @@ typename std::enable_if<
   return std::move(allocator);
 }
 
+/**
+ * AllocatorUniquePtr: a unique_ptr that supports both STL-style
+ * allocators and SimpleAllocator
+ *
+ * @author: Marcelo Juchem <marcelo@fb.com>
+ */
+
 template <typename T, typename Allocator>
 struct AllocatorUniquePtr {
   typedef std::unique_ptr<T,
@@ -245,6 +264,13 @@ struct AllocatorUniquePtr {
   > type;
 };
 
+/**
+ * Functions to allocate a unique_ptr / shared_ptr, supporting both
+ * STL-style allocators and SimpleAllocator, analog to std::allocate_shared
+ *
+ * @author: Marcelo Juchem <marcelo@fb.com>
+ */
+
 template <typename T, typename Allocator, typename ...Args>
 typename AllocatorUniquePtr<T, Allocator>::type allocate_unique(
   Allocator&& allocator, Args&&... args
diff --git a/folly/test/MemoryTest.cpp b/folly/test/MemoryTest.cpp
new file mode 100644
index 00000000..82f1a50b
--- /dev/null
+++ b/folly/test/MemoryTest.cpp
@@ -0,0 +1,52 @@
+/*
+ * Copyright 2013 Facebook, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "folly/Memory.h"
+#include "folly/Arena.h"
+#include "folly/String.h"
+
+#include <glog/logging.h>
+#include <gtest/gtest.h>
+
+#include <type_traits>
+
+using namespace folly;
+
+template <std::size_t> struct T {};
+template <std::size_t> struct S {};
+template <std::size_t> struct P {};
+
+TEST(as_stl_allocator, sanity_check) {
+  typedef StlAllocator<SysArena, int> stl_arena_alloc;
+
+  EXPECT_TRUE((std::is_same<
+    as_stl_allocator<int, SysArena>::type,
+    stl_arena_alloc
+  >::value));
+
+  EXPECT_TRUE((std::is_same<
+    as_stl_allocator<int, stl_arena_alloc>::type,
+    stl_arena_alloc
+  >::value));
+}
+
+int main(int argc, char **argv) {
+  FLAGS_logtostderr = true;
+  google::InitGoogleLogging(argv[0]);
+  testing::InitGoogleTest(&argc, argv);
+
+  return RUN_ALL_TESTS();
+}