Adds const cast for BranchNode

Issue: #17
0x7CFE · May 21, 2016 · aa97c11 · aa97c11
1 parent 7f43e3f
commit aa97c11
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/include/analysis.h b/include/analysis.h
@@ -105,6 +105,7 @@ class ControlNode {
     // Dynamically cast node to a specified type.
     // If type does not match null is returned.
     template<class T> T* cast();
+    template<class T> const T* cast() const;
 
     uint32_t getIndex() const { return m_index; }
 
@@ -484,6 +485,7 @@ template<> PushBlockNode* ControlNode::cast<PushBlockNode>();
 template<> PushBlockNode* ControlGraph::newNode<PushBlockNode>();
 
 template<> BranchNode* ControlNode::cast<BranchNode>();
+template<> const BranchNode* ControlNode::cast<BranchNode>() const;
 template<> BranchNode* ControlGraph::newNode<BranchNode>();
 
 class DomainVisitor {

diff --git a/src/ControlGraph.cpp b/src/ControlGraph.cpp
@@ -57,6 +57,10 @@ template<> BranchNode* ControlNode::cast<BranchNode>() {
     return 0;
 }
 
+template<> const BranchNode* ControlNode::cast<BranchNode>() const {
+    return const_cast<ControlNode*>(this)->cast<BranchNode>();
+}
+
 template<> BranchNode* ControlGraph::newNode<BranchNode>() {
     BranchNode* const node = new BranchNode(m_lastNodeIndex++);
     m_nodes.push_back(node);