[loco] Introduce Node.opnum API (#4004)

This opnum API allows us to check the type of nodes without using
dynamic_cast.

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

diff --git a/contrib/loco/include/loco/IR/CanonicalNode.h b/contrib/loco/include/loco/IR/CanonicalNode.h
index 99a13e9..38a24a2 100644 (file)
--- a/contrib/loco/include/loco/IR/CanonicalNode.h
+++ b/contrib/loco/include/loco/IR/CanonicalNode.h
@@ -40,6 +40,7 @@ template <CanonicalOpcode Code> struct CanonicalNodeImpl : public CanonicalNode
 {
   virtual ~CanonicalNodeImpl() = default;
 
+  uint32_t opnum(void) const final { return static_cast<uint32_t>(Code); }
   CanonicalOpcode opcode(void) const final { return Code; }
 };
 

diff --git a/contrib/loco/include/loco/IR/Node.h b/contrib/loco/include/loco/IR/Node.h
index f8219e3..c436478 100644 (file)
--- a/contrib/loco/include/loco/IR/Node.h
+++ b/contrib/loco/include/loco/IR/Node.h
@@ -70,6 +70,8 @@ public:
    */
   virtual const Dialect *dialect(void) const = 0;
 
+  virtual uint32_t opnum(void) const = 0;
+
 public:
   /// @brief Return the number of arguments
   virtual uint32_t arity(void) const = 0;

diff --git a/contrib/loco/src/IR/MockupNode.h b/contrib/loco/src/IR/MockupNode.h
index 184c30f..ec56c90 100644 (file)
--- a/contrib/loco/src/IR/MockupNode.h
+++ b/contrib/loco/src/IR/MockupNode.h
@@ -40,6 +40,7 @@ public:
 
 public:
   const loco::Dialect *dialect(void) const final { return MockDialect::get(); }
+  uint32_t opnum(void) const final { return 0; }
 
   uint32_t arity(void) const final { return 1; }
   Node *arg(uint32_t N) const final { return _arg.node(); }

diff --git a/contrib/loco/src/IR/Node.test.cpp b/contrib/loco/src/IR/Node.test.cpp
index cc879e7..61d0931 100644 (file)
--- a/contrib/loco/src/IR/Node.test.cpp
+++ b/contrib/loco/src/IR/Node.test.cpp
@@ -81,6 +81,7 @@ TEST(FixedArityNodeTest, constructor)
   struct DerivedNode final : public loco::FixedArityNode<1, loco::Node>
   {
     loco::Dialect *dialect(void) const final { return MockDialect::get(); }
+    uint32_t opnum(void) const final { return 0; }
   };
 
   DerivedNode node;