a2kit 4.4.2

Retro disk image and language utility
Documentation 
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
103
104
105
106

use tree_sitter;
use a2_memory_map as a2memory;
use crate::lang::node_integer;
use std::collections::HashMap;
use std::format;
use num_traits::abs;

pub struct AddressHovers {
    amap: HashMap<u16,String>
}

const OFFSET_NAMES: [(&str,[&str;6]);4] = [
    ("word",["low byte","high byte","","","",""]),
    ("vector",["opcode","low addr","high addr","","",""]),
    ("float",["1","2","3","4","5",""]),
    ("unpacked float",["1","2","3","4","5","6"])
];

fn create_hover(addr_base: u16,offset: u16,info: &a2memory::AddressInfo) -> String {
    let offset_names = HashMap::from(OFFSET_NAMES);
    let addr_unsigned = addr_base as i32 + offset as i32;
    let addr_signed = addr_unsigned - 1 - u16::MAX as i32;
    let mut ans = String::new();
    if let Some(label) = &info.label {
        ans += &format!("`{}`\n\n",label);
    }
    let mut addr_type = info.typ.clone();
    if let Some(subtype) = offset_names.get(info.typ.as_str()) {
        addr_type += ", ";
        addr_type += subtype[offset as usize];
    }
    if addr_unsigned >= (1<<15) {
        ans += &format!("Special address: **{}** ({} | {} | ${:X})\n\n",addr_type,addr_unsigned,addr_signed,addr_unsigned);
    } else {
        ans += &format!("Special address: **{}** ({} | ${:X})\n\n",addr_type,addr_unsigned,addr_unsigned);
    }
    ans += &info.desc;
    ans += "\n\n";
    if let Some(ctx) = &info.ctx {
        ans += &format!("Context limitation: {}\n\n",ctx);
    }
    if let Some(note) = &info.note {
        ans += &format!("Note: {}\n\n",note);
    }
    ans
}

impl AddressHovers {
    pub fn new() -> Self {
        let mut hov = Self {
            amap: HashMap::new()
        };
        let main_map = a2memory::MemoryMap::new();
        let raw_map = main_map.get_all();
        for (addr,info) in raw_map {
            let bytes = match info.typ.as_str() {
                "word" => 2,
                "vector" => 3,
                "float" => 5,
                "unpacked float" => 6,
                _ => 1
            };
            for offset in 0..bytes {
                let hov_str = create_hover(*addr, offset, info);
                hov.amap.insert(*addr + offset,hov_str);
            }
        }
        hov
    }
    /// OK to pass out of range number; negative address will be translated if possible
    pub fn get(&self,addr: i64) -> Option<String> {
        if abs(addr) > u16::MAX as i64 {
            return None;
        }
        let addr16 = match addr >= 0 {
            true => addr as u16,
            false => (addr + 1 + u16::MAX as i64) as u16
        };
        self.amap.get(&addr16).cloned()
    }
    pub fn get_from_node(&self,curs: &tree_sitter::TreeCursor,line: &str) -> Option<String> {
        let mut mul = 1;
        if curs.node().kind() == "int" {
            let mut maybe_cmd = curs.node().prev_named_sibling();
            let maybe_prev = curs.node().prev_named_sibling();
            if let Some(prev) = maybe_prev {
                if prev.kind() == "tok_minus" || prev.kind() == "tok_plus" {
                    if let Some(parent) = prev.parent() {
                        if parent.kind()=="unary_aexpr" {
                            mul *= match prev.kind()=="tok_minus" { true => -1, false => 1 };
                            maybe_cmd = parent.prev_named_sibling();
                        }
                    }
                }
            }
            if let Some(cmd) = maybe_cmd {
                if cmd.kind() == "tok_poke" || cmd.kind() == "tok_peek" || cmd.kind() == "tok_call" {
                    if let Some(num) = node_integer::<i64>(&curs.node(), line) {
                        return self.get(num*mul);
                    }
                }
            }
        }
        None
    }
}