From ea830c8f7b42cc7aebf5f714699011b3e5fae850 Mon Sep 17 00:00:00 2001
From: Jon Palmisciano <jp@jonpalmisc.com>
Date: Thu, 3 Jun 2021 10:54:44 -0400
Subject: Function: Added `GetILVariables()` method

---
 function.cpp | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'function.cpp')

diff --git a/function.cpp b/function.cpp
index 2c1b34ea..58968938 100644
--- a/function.cpp
+++ b/function.cpp
@@ -1107,6 +1107,23 @@ map<Variable, VariableNameAndType> Function::GetVariables()
 }
 
 
+set<Variable> Function::GetILVariables(BNFunctionGraphType ilType)
+{
+	size_t count;
+	auto* vars = BNGetFunctionILVariables(m_object, ilType, &count);
+
+	set<Variable> result;
+	for (size_t i = 0; i < count; i++)
+	{
+		Variable v(vars[i]);
+		result.insert(v);
+	}
+
+	BNFreeVariableList(vars);
+	return result;
+}
+
+
 void Function::CreateAutoVariable(const Variable& var, const Confidence<Ref<Type>>& type,
 	const string& name, bool ignoreDisjointUses)
 {
-- 
cgit v1.3.1