[coco] Introduce UseHook (#1042)

This commit introduces UseHook, which encapsulates ObjectInfo update
routines on Object usage changes.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>

diff --git a/contrib/coco/core/include/coco/IR/UseHook.h b/contrib/coco/core/include/coco/IR/UseHook.h
new file mode 100644 (file)
index 0000000..13361e5
--- /dev/null
+++ b/contrib/coco/core/include/coco/IR/UseHook.h
@@ -0,0 +1,35 @@
+#ifndef __COCO_IR_USE_HOOK_H__
+#define __COCO_IR_USE_HOOK_H__
+
+#include "coco/IR/Object.h"
+#include "coco/IR/ObjectInfo.forward.h"
+
+#include "coco/ADT/PtrLink.h"
+
+namespace coco
+{
+
+class UseHook final
+{
+public:
+  UseHook(const PtrLink<Object, ObjectInfo> *obj_link, Object::Use *use)
+      : _obj_link{obj_link}, _use{use}
+  {
+    // DO NOTHING
+  }
+
+private:
+  ObjectInfo *info(Object *o) const;
+
+public:
+  void onTake(Object *o);
+  void onRelease(Object *o);
+
+private:
+  const PtrLink<Object, ObjectInfo> *const _obj_link;
+  Object::Use *const _use;
+};
+
+} // namespace coco
+
+#endif // __COCO_IR_USE_HOOK_H__

diff --git a/contrib/coco/core/src/IR/UseHook.cpp b/contrib/coco/core/src/IR/UseHook.cpp
new file mode 100644 (file)
index 0000000..9538e0a
--- /dev/null
+++ b/contrib/coco/core/src/IR/UseHook.cpp
@@ -0,0 +1,19 @@
+#include "coco/IR/UseHook.h"
+#include "coco/IR/ObjectInfo.h"
+
+#include <cassert>
+
+namespace coco
+{
+
+ObjectInfo *UseHook::info(Object *o) const
+{
+  auto info = _obj_link->find(o);
+  assert(info != nullptr);
+  return info;
+}
+
+void UseHook::onTake(Object *o) { info(o)->user()->insert(_use); }
+void UseHook::onRelease(Object *o) { info(o)->user()->erase(_use); }
+
+} // namespace coco

diff --git a/contrib/coco/core/src/IR/UseHook.test.cpp b/contrib/coco/core/src/IR/UseHook.test.cpp
new file mode 100644 (file)
index 0000000..5e9fb78
--- /dev/null
+++ b/contrib/coco/core/src/IR/UseHook.test.cpp
@@ -0,0 +1,41 @@
+#include "coco/IR/UseHook.h"
+#include "coco/IR/ObjectInfo.h"
+#include "coco/IR/ObjectManager.h"
+
+#include "coco/IR/FeatureObject.h"
+
+#include <gtest/gtest.h>
+
+namespace
+{
+namespace mock
+{
+struct Use final : public coco::Object::Use
+{
+};
+} // namespace mock
+} // namespace
+
+namespace
+{
+class UseHookTest : public ::testing::Test
+{
+protected:
+  coco::PtrLink<coco::Bag, coco::BagInfo> bag_link;
+  coco::PtrLink<coco::Object, coco::ObjectInfo> obj_link;
+
+  coco::ObjectManager obj_mgr{&obj_link, &bag_link};
+};
+} // namespace
+
+TEST_F(UseHookTest, TakeAndRelease)
+{
+  auto o = obj_mgr.create(nncc::core::ADT::feature::Shape{1, 1, 1});
+
+  ::mock::Use use;
+
+  coco::UseHook hook{&obj_link, &use};
+
+  hook.onTake(o);
+  hook.onRelease(o);
+}