[nest] Overload 'operator+' over nest expressions (#712)

This commit introduce 'operator+' overloading to make it easy to
construct 'ADD' expression in nest IR.

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

diff --git a/contrib/nest/include/nest/Expr.h b/contrib/nest/include/nest/Expr.h
index 9fc09cb..8fef2b1 100644 (file)
--- a/contrib/nest/include/nest/Expr.h
+++ b/contrib/nest/include/nest/Expr.h
@@ -17,4 +17,6 @@ using Expr = std::shared_ptr<nest::expr::Node>;
 
 } // namespace nest
 
+nest::Expr operator+(const nest::Expr &, const nest::Expr &);
+
 #endif // __NEST_EXPR_H__

diff --git a/contrib/nest/src/Expr.cpp b/contrib/nest/src/Expr.cpp
new file mode 100644 (file)
index 0000000..46e9b41
--- /dev/null
+++ b/contrib/nest/src/Expr.cpp
@@ -0,0 +1,6 @@
+#include "nest/Expr.h"
+
+nest::Expr operator+(const nest::Expr &lhs, const nest::Expr &rhs)
+{
+  return std::make_shared<nest::expr::AddNode>(lhs, rhs);
+}

diff --git a/contrib/nest/src/Expr.test.cpp b/contrib/nest/src/Expr.test.cpp
new file mode 100644 (file)
index 0000000..de469ca
--- /dev/null
+++ b/contrib/nest/src/Expr.test.cpp
@@ -0,0 +1,27 @@
+#include "nest/Expr.h"
+
+#include <memory>
+
+#include <gtest/gtest.h>
+
+namespace
+{
+struct DummyNode final : public nest::expr::Node
+{
+};
+}
+
+TEST(EXPR, operator_sum)
+{
+  auto left = std::make_shared<DummyNode>();
+  auto right = std::make_shared<DummyNode>();
+
+  auto expr = left + right;
+
+  ASSERT_NE(expr->asAdd(), nullptr);
+
+  auto add = expr->asAdd();
+
+  ASSERT_EQ(add->lhs().get(), left.get());
+  ASSERT_EQ(add->rhs().get(), right.get());
+}