From 64706bbd526c22e38b0be34f5d2b5a75766a6c02 Mon Sep 17 00:00:00 2001
From: Glenn Smith <glenn@vector35.com>
Date: Fri, 16 May 2025 16:24:46 -0400
Subject: [Python] Support HLIL collapsing and update PseudoPython

Fixes Vector35/binaryninja-api#6679
---
 python/function.py | 53 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)

(limited to 'python/function.py')

diff --git a/python/function.py b/python/function.py
index ec2fbab8..b101a44f 100644
--- a/python/function.py
+++ b/python/function.py
@@ -3337,6 +3337,59 @@ class Function:
 			bc.confidence = core.max_confidence
 		core.BNSetUserFunctionInlinedDuringAnalysis(self.handle, bc)
 
+	def toggle_region(self, hash):
+		"""
+		Toggle the collapsed state of a region during rendering, by hash value
+		:param hash: Hash value of region
+		"""
+		core.BNFunctionToggleRegion(self.handle, hash)
+
+	def collapse_region(self, hash):
+		"""
+		Collapse a region during rendering
+		:param hash: Hash value of region
+		"""
+		core.BNFunctionCollapseRegion(self.handle, hash)
+
+	def expand_region(self, hash):
+		"""
+		Un-collapse a region during rendering
+		:param hash: Hash value of region
+		"""
+		core.BNFunctionExpandRegion(self.handle, hash)
+
+	def expand_all(self):
+		"""
+		Expand all regions in the function
+		"""
+		core.BNFunctionExpandAll(self.handle)
+
+	@property
+	def is_collapsed(self):
+		"""If the entire function is collapsed during rendering."""
+		return self.is_region_collapsed(self.start)
+
+	def is_instruction_collapsed(
+		self,
+		instr: 'highlevelil.HighLevelILInstruction',
+		discriminator: int = 0
+	) -> bool:
+		"""
+		Determine if a given HLIL instruction (with discriminator) is collapsed during rendering.
+		:param instr: Instruction which might be collapsed
+		:param discriminator: Unique discriminator id for the region
+		:return: True if the instruction should be rendered as collapsed
+		"""
+		return self.is_region_collapsed(instr.get_instruction_hash(discriminator))
+
+	def is_region_collapsed(self, hash) -> bool:
+		"""
+		Determine if a given region is collapsed during rendering.
+		:param hash: Hash value of region
+		:return: True if the region should be rendered as collapsed
+		"""
+		return core.BNFunctionIsRegionCollapsed(self.handle, hash)
+
 
 class AdvancedFunctionAnalysisDataRequestor:
 	def __init__(self, func: Optional['Function'] = None):
-- 
cgit v1.3.1