From 7b0b426e9d1cb2176b2f67ed9af813d051696d8b Mon Sep 17 00:00:00 2001
From: =?utf8?q?=EB=B0=95=EC=A2=85=ED=98=84/=EB=8F=99=EC=9E=91=EC=A0=9C?=
 =?utf8?q?=EC=96=B4Lab=28SR=29/Staff=20Engineer/=EC=82=BC=EC=84=B1?=
 =?utf8?q?=EC=A0=84=EC=9E=90?= <jh1302.park@samsung.com>
Date: Thu, 19 Jul 2018 19:21:33 +0900
Subject: [PATCH] [coco] Add 'PtrManager' class (#724)

This commit adds 'PtrManager' class which automatically manages a set of
allocated objects (which is passed via take method).

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
---
 contrib/coco/core/include/coco/ADT/PtrManager.h | 34 +++++++++++
 contrib/coco/core/src/ADT/PtrManager.test.cpp   | 78 +++++++++++++++++++++++++
 2 files changed, 112 insertions(+)
 create mode 100644 contrib/coco/core/include/coco/ADT/PtrManager.h
 create mode 100644 contrib/coco/core/src/ADT/PtrManager.test.cpp

diff --git a/contrib/coco/core/include/coco/ADT/PtrManager.h b/contrib/coco/core/include/coco/ADT/PtrManager.h
new file mode 100644
index 0000000..fc94745
--- /dev/null
+++ b/contrib/coco/core/include/coco/ADT/PtrManager.h
@@ -0,0 +1,34 @@
+#ifndef __COCO_ADT_PTR_MANAGER_H__
+#define __COCO_ADT_PTR_MANAGER_H__
+
+#include <vector>
+
+#include <memory>
+
+namespace coco
+{
+
+template<typename T> class PtrManager
+{
+public:
+  /// @breif Return the number of managed objects
+  uint32_t size(void) const { return _ptrs.size(); }
+
+public:
+  T *at(uint32_t n) const { return _ptrs.at(n).get(); }
+
+protected:
+  template<typename U> U *take(std::unique_ptr<U> &&o)
+  {
+    auto res = o.get();
+    _ptrs.emplace_back(std::move(o));
+    return res;
+  }
+
+private:
+  std::vector<std::unique_ptr<T>> _ptrs;
+};
+
+} // namespace coco
+
+#endif // __COCO_ADT_PTR_MANAGER_H__
diff --git a/contrib/coco/core/src/ADT/PtrManager.test.cpp b/contrib/coco/core/src/ADT/PtrManager.test.cpp
new file mode 100644
index 0000000..422c3e7
--- /dev/null
+++ b/contrib/coco/core/src/ADT/PtrManager.test.cpp
@@ -0,0 +1,78 @@
+#include "coco/ADT/PtrManager.h"
+
+#include <memory>
+
+#include <gtest/gtest.h>
+
+namespace
+{
+struct Count
+{
+  uint32_t allocated;
+  uint32_t freed;
+
+  Count() : allocated{0}, freed{0}
+  {
+    // DO NOTHING
+  }
+};
+
+class Object
+{
+public:
+  Object(Count *count, uint32_t value) : _count{count}, _value{value}
+  {
+    _count->allocated += 1;
+  }
+
+public:
+  ~Object()
+  {
+    _count->freed += 1;
+  }
+
+public:
+  uint32_t value(void) const { return _value; }
+
+private:
+  Count * const _count;
+
+private:
+  uint32_t _value;
+};
+
+struct ObjectManager final : public coco::PtrManager<Object>
+{
+  Object *alloc(Count *count, uint32_t value)
+  {
+    std::unique_ptr<Object> o{new Object{count, value}};
+    return take(std::move(o));
+  }
+};
+}
+
+TEST(ADT_PTR_MANAGER, usecase)
+{
+  Count c;
+
+  ASSERT_EQ(c.allocated, 0);
+  ASSERT_EQ(c.freed, 0);
+
+  {
+    ::ObjectManager mgr;
+
+    auto obj_1 = mgr.alloc(&c, 3);
+    auto obj_2 = mgr.alloc(&c, 4);
+
+    EXPECT_EQ(c.allocated, 2);
+    ASSERT_EQ(c.freed, 0);
+
+    EXPECT_EQ(mgr.size(), 2);
+    EXPECT_EQ(mgr.at(0), obj_1);
+    EXPECT_EQ(mgr.at(1), obj_2);
+  }
+
+  // PtrManger SHOULD destruct all of the allocated object when it is destructed.
+  ASSERT_EQ(c.allocated, 2);
+  ASSERT_EQ(c.freed, 2);
+}
-- 
2.7.4