consistently enter register/number/opcode spans

1 files changed, 44 insertions, 79 deletions

diff --git a/src/protected_mode/display.rs b/src/protected_mode/display.rs
index 1629b4e..db12878 100644
--- a/src/protected_mode/display.rs
+++ b/src/protected_mode/display.rs
@@ -65,6 +65,8 @@ trait DisplaySinkExt {
 
     fn write_opcode(&mut self, opcode: super::Opcode) -> Result<(), core::fmt::Error>;
     fn write_reg(&mut self, reg: RegSpec) -> Result<(), core::fmt::Error>;
+    fn write_displacement(&mut self, disp: i32) -> Result<(), core::fmt::Error>;
+    fn write_scale(&mut self, scale: u8) -> Result<(), core::fmt::Error>;
     fn write_mem_size_label(&mut self, mem_size: u8) -> Result<(), core::fmt::Error>;
     fn write_sae_mode(&mut self, sae: super::SaeMode) -> Result<(), core::fmt::Error>;
 }
@@ -89,6 +91,29 @@ impl<T: DisplaySink> DisplaySinkExt for T {
     }
 
     #[inline(always)]
+    fn write_displacement(&mut self, disp: i32) -> Result<(), core::fmt::Error> {
+        let mut v = disp as u32;
+        if disp < 0 {
+            self.write_fixed_size("- ")?;
+            v = disp.unsigned_abs();
+        } else {
+            self.write_fixed_size("+ ")?;
+        }
+        self.span_start_number();
+        self.write_prefixed_u32(v)?;
+        self.span_end_number();
+        Ok(())
+    }
+
+    #[inline(always)]
+    fn write_scale(&mut self, scale: u8) -> Result<(), core::fmt::Error> {
+        self.span_start_number();
+        self.write_char((0x30 + scale) as char)?; // translate scale=1 to '1', scale=2 to '2', etc
+        self.span_end_number();
+        Ok(())
+    }
+
+    #[inline(always)]
     fn write_mem_size_label(&mut self, mem_size: u8) -> Result<(), core::fmt::Error> {
         let label = mem_size_label(mem_size);
         let _ = Self::MEM_SIZE_LABEL_LT_8;
@@ -343,20 +368,14 @@ impl <T: DisplaySink> super::OperandVisitor for DisplayingOperandVisitor<'_, T>
     }
     #[cfg_attr(feature="profiling", inline(never))]
     fn visit_reg(&mut self, reg: RegSpec) -> Result<Self::Ok, Self::Error> {
-        self.f.span_start_register();
         self.f.write_reg(reg)?;
-        self.f.span_end_register();
         Ok(())
     }
     fn visit_reg_mask_merge(&mut self, spec: RegSpec, mask: RegSpec, merge_mode: MergeMode) -> Result<Self::Ok, Self::Error> {
-        self.f.span_start_register();
         self.f.write_reg(spec)?;
-        self.f.span_end_register();
         if mask.num != 0 {
             self.f.write_fixed_size("{")?;
-            self.f.span_start_register();
             self.f.write_reg(mask)?;
-            self.f.span_end_register();
             self.f.write_fixed_size("}")?;
         }
         if let MergeMode::Zero = merge_mode {
@@ -392,15 +411,19 @@ impl <T: DisplaySink> super::OperandVisitor for DisplayingOperandVisitor<'_, T>
     }
     fn visit_abs_u16(&mut self, imm: u16) -> Result<Self::Ok, Self::Error> {
         self.f.write_fixed_size("[")?;
+        self.f.span_start_address();
         self.f.write_fixed_size("0x")?;
         self.f.write_u16(imm)?;
+        self.f.span_end_address();
         self.f.write_fixed_size("]")?;
         Ok(())
     }
     fn visit_abs_u32(&mut self, imm: u32) -> Result<Self::Ok, Self::Error> {
         self.f.write_fixed_size("[")?;
+        self.f.span_start_address();
         self.f.write_fixed_size("0x")?;
         self.f.write_u32(imm)?;
+        self.f.span_end_address();
         self.f.write_fixed_size("]")?;
         Ok(())
     }
@@ -410,17 +433,7 @@ impl <T: DisplaySink> super::OperandVisitor for DisplayingOperandVisitor<'_, T>
         self.f.write_char('[')?;
         self.f.write_reg(base)?;
         self.f.write_fixed_size(" ")?;
-
-        {
-            let mut v = disp as u32;
-            if disp < 0 {
-                self.f.write_fixed_size("- 0x")?;
-                v = disp.unsigned_abs();
-            } else {
-                self.f.write_fixed_size("+ 0x")?;
-            }
-            self.f.write_u32(v)?;
-        }
+        self.f.write_displacement(disp)?;
         self.f.write_fixed_size("]")
     }
     fn visit_deref(&mut self, base: RegSpec) -> Result<Self::Ok, Self::Error> {
@@ -432,7 +445,7 @@ impl <T: DisplaySink> super::OperandVisitor for DisplayingOperandVisitor<'_, T>
         self.f.write_fixed_size("[")?;
         self.f.write_reg(index)?;
         self.f.write_fixed_size(" * ")?;
-        self.f.write_char((0x30 + scale) as char)?; // translate scale=1 to '1', scale=2 to '2', etc
+        self.f.write_scale(scale)?;
         self.f.write_fixed_size("]")?;
 
         Ok(())
@@ -441,19 +454,9 @@ impl <T: DisplaySink> super::OperandVisitor for DisplayingOperandVisitor<'_, T>
         self.f.write_fixed_size("[")?;
         self.f.write_reg(index)?;
         self.f.write_fixed_size(" * ")?;
-        self.f.write_char((0x30 + scale) as char)?; // translate scale=1 to '1', scale=2 to '2', etc
+        self.f.write_scale(scale)?;
         self.f.write_fixed_size(" ")?;
-
-        {
-            let mut v = disp as u32;
-            if disp < 0 {
-                self.f.write_fixed_size("- 0x")?;
-                v = disp.unsigned_abs();
-            } else {
-                self.f.write_fixed_size("+ 0x")?;
-            }
-            self.f.write_u32(v)?;
-        }
+        self.f.write_displacement(disp)?;
         self.f.write_char(']')
     }
     fn visit_base_index_scale(&mut self, base: RegSpec, index: RegSpec, scale: u8) -> Result<Self::Ok, Self::Error> {
@@ -462,7 +465,7 @@ impl <T: DisplaySink> super::OperandVisitor for DisplayingOperandVisitor<'_, T>
         self.f.write_fixed_size(" + ")?;
         self.f.write_reg(index)?;
         self.f.write_fixed_size(" * ")?;
-        self.f.write_char((0x30 + scale) as char)?; // translate scale=1 to '1', scale=2 to '2', etc
+        self.f.write_scale(scale)?;
         self.f.write_fixed_size("]")
     }
     fn visit_base_index_scale_disp(&mut self, base: RegSpec, index: RegSpec, scale: u8, disp: i32) -> Result<Self::Ok, Self::Error> {
@@ -471,33 +474,16 @@ impl <T: DisplaySink> super::OperandVisitor for DisplayingOperandVisitor<'_, T>
         self.f.write_fixed_size(" + ")?;
         self.f.write_reg(index)?;
         self.f.write_fixed_size(" * ")?;
-        self.f.write_char((0x30 + scale) as char)?; // translate scale=1 to '1', scale=2 to '2', etc
+        self.f.write_scale(scale)?;
         self.f.write_fixed_size(" ")?;
-
-        {
-            let mut v = disp as u32;
-            if disp < 0 {
-                self.f.write_fixed_size("- 0x")?;
-                v = disp.unsigned_abs();
-            } else {
-                self.f.write_fixed_size("+ 0x")?;
-            }
-            self.f.write_u32(v)?;
-        }
+        self.f.write_displacement(disp)?;
         self.f.write_fixed_size("]")
     }
     fn visit_disp_masked(&mut self, base: RegSpec, disp: i32, mask_reg: RegSpec) -> Result<Self::Ok, Self::Error> {
         self.f.write_char('[')?;
         self.f.write_reg(base)?;
         self.f.write_char(' ')?;
-        let mut v = disp as u32;
-        if disp < 0 {
-            self.f.write_fixed_size("- 0x")?;
-            v = disp.unsigned_abs();
-        } else {
-            self.f.write_fixed_size("+ 0x")?;
-        }
-        self.f.write_u32(v)?;
+        self.f.write_displacement(disp)?;
         self.f.write_char(']')?;
         self.f.write_char('{')?;
         self.f.write_reg(mask_reg)?;
@@ -517,7 +503,7 @@ impl <T: DisplaySink> super::OperandVisitor for DisplayingOperandVisitor<'_, T>
         self.f.write_fixed_size("[")?;
         self.f.write_reg(index)?;
         self.f.write_fixed_size(" * ")?;
-        self.f.write_char((0x30 + scale) as char)?; // translate scale=1 to '1', scale=2 to '2', etc
+        self.f.write_scale(scale)?;
         self.f.write_fixed_size("]")?;
         self.f.write_char('{')?;
         self.f.write_reg(mask_reg)?;
@@ -528,16 +514,9 @@ impl <T: DisplaySink> super::OperandVisitor for DisplayingOperandVisitor<'_, T>
         self.f.write_fixed_size("[")?;
         self.f.write_reg(index)?;
         self.f.write_fixed_size(" * ")?;
-        self.f.write_char((0x30 + scale) as char)?; // translate scale=1 to '1', scale=2 to '2', etc
+        self.f.write_scale(scale)?;
         self.f.write_fixed_size(" ")?;
-        let mut v = disp as u32;
-        if disp < 0 {
-            self.f.write_fixed_size("- 0x")?;
-            v = disp.unsigned_abs();
-        } else {
-            self.f.write_fixed_size("+ 0x")?;
-        }
-        self.f.write_u32(v)?;
+        self.f.write_displacement(disp)?;
         self.f.write_char(']')?;
         self.f.write_char('{')?;
         self.f.write_reg(mask_reg)?;
@@ -561,14 +540,7 @@ impl <T: DisplaySink> super::OperandVisitor for DisplayingOperandVisitor<'_, T>
         self.f.write_fixed_size(" + ")?;
         self.f.write_reg(index)?;
         self.f.write_fixed_size(" ")?;
-        let mut v = disp as u32;
-        if disp < 0 {
-            self.f.write_fixed_size("- 0x")?;
-            v = disp.unsigned_abs();
-        } else {
-            self.f.write_fixed_size("+ 0x")?;
-        }
-        self.f.write_u32(v)?;
+        self.f.write_displacement(disp)?;
         self.f.write_char(']')?;
         self.f.write_char('{')?;
         self.f.write_reg(mask_reg)?;
@@ -581,7 +553,7 @@ impl <T: DisplaySink> super::OperandVisitor for DisplayingOperandVisitor<'_, T>
         self.f.write_fixed_size(" + ")?;
         self.f.write_reg(index)?;
         self.f.write_fixed_size(" * ")?;
-        self.f.write_char((0x30 + scale) as char)?; // translate scale=1 to '1', scale=2 to '2', etc
+        self.f.write_scale(scale)?;
         self.f.write_fixed_size("]")?;
         self.f.write_char('{')?;
         self.f.write_reg(mask_reg)?;
@@ -594,16 +566,9 @@ impl <T: DisplaySink> super::OperandVisitor for DisplayingOperandVisitor<'_, T>
         self.f.write_fixed_size(" + ")?;
         self.f.write_reg(index)?;
         self.f.write_fixed_size(" * ")?;
-        self.f.write_char((0x30 + scale) as char)?; // translate scale=1 to '1', scale=2 to '2', etc
+        self.f.write_scale(scale)?;
         self.f.write_char(' ')?;
-        let mut v = disp as u32;
-        if disp < 0 {
-            self.f.write_fixed_size("- 0x")?;
-            v = disp.unsigned_abs();
-        } else {
-            self.f.write_fixed_size("+ 0x")?;
-        }
-        self.f.write_u32(v)?;
+        self.f.write_displacement(disp)?;
         self.f.write_char(']')?;
         self.f.write_char('{')?;
         self.f.write_reg(mask_reg)?;