about summary refs log tree commit diff
path: root/tvix/eval
diff options
context:
space:
mode:
Diffstat (limited to 'tvix/eval')
-rw-r--r--tvix/eval/src/chunk.rs9
-rw-r--r--tvix/eval/src/disassembler.rs7
2 files changed, 12 insertions, 4 deletions
diff --git a/tvix/eval/src/chunk.rs b/tvix/eval/src/chunk.rs
index defa409f18d6..c89e6ef84810 100644
--- a/tvix/eval/src/chunk.rs
+++ b/tvix/eval/src/chunk.rs
@@ -77,4 +77,13 @@ impl Chunk {
 
         panic!("compiler error: chunk missing span for offset {}", offset.0);
     }
+
+    /// Retrieve the line from which the instruction at `offset` was
+    /// compiled. Only available when the chunk carries a codemap,
+    /// i.e. when the disassembler is enabled.
+    #[cfg(feature = "disassembler")]
+    pub fn get_line(&self, offset: CodeIdx) -> usize {
+        let span = self.get_span(offset);
+        self.codemap.look_up_span(span).begin.line + 1
+    }
 }
diff --git a/tvix/eval/src/disassembler.rs b/tvix/eval/src/disassembler.rs
index b089797f8ab1..7948977e4b55 100644
--- a/tvix/eval/src/disassembler.rs
+++ b/tvix/eval/src/disassembler.rs
@@ -44,13 +44,12 @@ impl Drop for Tracer {
 fn disassemble_op(tw: &mut TabWriter<Stderr>, chunk: &Chunk, width: usize, offset: usize) {
     let _ = write!(tw, "{:0width$}\t ", offset, width = width);
 
-    let span = chunk.get_span(CodeIdx(offset));
+    let line = chunk.get_line(CodeIdx(offset));
 
-    if offset > 0 && chunk.get_span(CodeIdx(offset - 1)) == span {
+    if offset > 0 && chunk.get_line(CodeIdx(offset - 1)) == line {
         write!(tw, "   |\t").unwrap();
     } else {
-        let loc = chunk.codemap.look_up_span(span);
-        write!(tw, "{:4}\t", loc.begin.line + 1).unwrap();
+        write!(tw, "{:4}\t", line).unwrap();
     }
 
     let _ = match chunk.code[offset] {