Adds TypeSystem::findBlockContext()

Issue: #17 Issue: #92
0x7CFE · Jul 21, 2016 · a532ac0 · a532ac0
1 parent ee7d469
commit a532ac0
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/include/inference.h b/include/inference.h
@@ -391,6 +391,8 @@ class TypeSystem {
 
     typedef TSymbol* TSelector;
 
+    InferContext* findBlockContext(std::size_t contextIndex) const;
+
     InferContext* inferMessage(
         TSelector selector,
         const Type& arguments,

diff --git a/src/TypeAnalyzer.cpp b/src/TypeAnalyzer.cpp
@@ -1485,3 +1485,15 @@ void type::TypeSystem::drawCallGraph() const {
     stream << "}\n";
     stream.close();
 }
+
+type::InferContext* type::TypeSystem::findBlockContext(std::size_t contextIndex) const
+{
+    TBlockCache::const_iterator iContext = m_blockCache.begin();
+    for (; iContext != m_blockCache.end(); ++iContext) {
+        InferContext* const candidate = iContext->second;
+        if (candidate->getIndex() == contextIndex)
+            return candidate;
+    }
+
+    return 0;
+}