Changes related to display as/type toggling.

1 files changed, 13 insertions, 0 deletions

diff --git a/rust/src/types.rs b/rust/src/types.rs
index f45d93bc..e6c16f6b 100644
--- a/rust/src/types.rs
+++ b/rust/src/types.rs
@@ -171,6 +171,11 @@ impl TypeBuilder {
         self
     }
 
+    pub fn set_integer_display_type(&self, display_type: IntegerDisplayType) -> &Self {
+        unsafe { BNSetIntegerTypeDisplayType(self.handle, display_type) };
+        self
+    }
+
     // Readable properties
 
     pub fn type_class(&self) -> TypeClass {
@@ -189,6 +194,10 @@ impl TypeBuilder {
         unsafe { BNIsTypeBuilderSigned(self.handle).into() }
     }
 
+    pub fn integer_display_type(&self) -> IntegerDisplayType {
+        self.finalize().integer_display_type()
+    }
+
     pub fn is_const(&self) -> Conf<bool> {
         unsafe { BNIsTypeBuilderConst(self.handle).into() }
     }
@@ -689,6 +698,10 @@ impl Type {
         unsafe { BNIsTypeSigned(self.handle).into() }
     }
 
+    pub fn integer_display_type(&self) -> IntegerDisplayType {
+        unsafe { BNGetIntegerTypeDisplayType(self.handle) }
+    }
+
     pub fn is_const(&self) -> Conf<bool> {
         unsafe { BNIsTypeConst(self.handle).into() }
     }