1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

use crate::{ProcedureOperation, COLUMN_SEPARATOR, FRAME_SEPARATOR};
use stak_vm::{Cons, PrimitiveSet, Profiler, StackSlot, Vm};
use std::{io::Write, time::Instant};

pub struct StackProfiler<T: Write> {
    writer: T,
    start_time: Instant,
}

impl<T: Write> StackProfiler<T> {
    pub fn new(writer: T) -> Self {
        Self {
            writer,
            start_time: Instant::now(),
        }
    }

    fn write_column_separator(&mut self) {
        write!(self.writer, "{COLUMN_SEPARATOR}").unwrap();
    }

    fn write_frame_separator(&mut self) {
        write!(self.writer, "{FRAME_SEPARATOR}").unwrap();
    }

    fn write_time(&mut self) {
        writeln!(
            &mut self.writer,
            "{}",
            Instant::now().duration_since(self.start_time).as_nanos()
        )
        .unwrap();
    }

    fn write_procedure<P: PrimitiveSet>(&mut self, vm: &Vm<P>, code: Cons) {
        let operand = vm.car(code);

        if let Some(symbol) = operand.to_cons() {
            let mut string = vm.car_value(vm.car(symbol)).assume_cons();

            while string != vm.null() {
                write!(
                    self.writer,
                    "{}",
                    char::from_u32(vm.car(string).assume_number().to_i64() as _).unwrap_or('�')
                )
                .unwrap();
                string = vm.cdr(string).assume_cons();
            }
        }
    }

    fn write_stack<P: PrimitiveSet>(&mut self, vm: &Vm<P>) {
        let mut stack = vm.stack();
        let mut first = true;

        while stack != vm.null() {
            stack = if vm.cdr(stack).tag() == StackSlot::Frame as _ {
                if !first {
                    self.write_frame_separator();
                }

                first = false;

                self.write_procedure(vm, vm.car_value(vm.car(stack)).assume_cons());

                vm.cdr_value(vm.car(stack)).assume_cons()
            } else {
                vm.cdr(stack).assume_cons()
            };
        }
    }
}

impl<T: Write, P: PrimitiveSet> Profiler<P> for StackProfiler<T> {
    fn profile_call(&mut self, vm: &Vm<P>, call_code: Cons, r#return: bool) {
        write!(
            self.writer,
            "{}",
            if r#return {
                ProcedureOperation::ReturnCall
            } else {
                ProcedureOperation::Call
            }
        )
        .unwrap();
        self.write_column_separator();
        self.write_procedure(vm, call_code);
        self.write_frame_separator();
        self.write_stack(vm);
        self.write_column_separator();
        self.write_time();
    }

    fn profile_return(&mut self, vm: &Vm<P>) {
        write!(self.writer, "{}", ProcedureOperation::Return).unwrap();
        self.write_column_separator();
        self.write_stack(vm);
        self.write_column_separator();
        self.write_time();
    }
}