c64-assembler 0.2.0

Commodore 64 assembler, outputs directly to .PRG or Dasm source code
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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147

use std::collections::HashMap;

use crate::{
    instruction::operation::Operation,
    memory::{
        address_mode::{AddressMode, Immediate},
        user_count::UserCount,
        Address,
    },
    validator::AssemblerResult,
    Application, Instructions,
};

pub fn finalize(application: &mut Application) -> AssemblerResult<()> {
    defines_update_user_count(application);
    functions_update_user_count(application);
    update_label_addresses(application)
}

fn defines_update_user_count(application: &mut Application) {
    let mut define_users = HashMap::new();
    for define in &application.defines {
        let user_count = count_users(application, &define.name);
        if user_count > 0 {
            define_users.insert(define.name.clone(), user_count);
        }
    }

    for (define_name, user_count) in define_users {
        let define = application.define_mut(&define_name);
        for _ in 0..user_count {
            define.user_increase();
        }
    }
}

fn functions_update_user_count(application: &mut Application) {
    let mut function_users = HashMap::new();
    for module in &application.modules {
        for function in &module.functions {
            let user_count = count_users(application, &function.name);
            if user_count > 0 {
                function_users.insert(function.name.clone(), user_count);
            }
        }
    }

    for module in &mut application.modules {
        for function in &mut module.functions {
            if let Some(user_count) = function_users.get(&function.name) {
                for _ in 0..*user_count {
                    function.user_increase();
                }
            }
        }
    }
}

fn count_users(application: &Application, name: &String) -> usize {
    let mut result = 0;
    for module in &application.modules {
        result += count_users_instructions(&module.instructions, name);
        for function in &module.functions {
            result += count_users_instructions(&function.instructions, name);
        }
    }
    result
}

fn count_users_instructions(instructions: &Instructions, name: &String) -> usize {
    let mut result = 0;
    for instruction in &instructions.instructions {
        match &instruction.address_mode {
            AddressMode::Absolute(address_reference)
            | AddressMode::AbsoluteX(address_reference)
            | AddressMode::AbsoluteY(address_reference)
            | AddressMode::Indirect(address_reference)
            | AddressMode::IndexedIndirect(address_reference)
            | AddressMode::IndirectIndexed(address_reference)
            | AddressMode::Immediate(Immediate::Low(address_reference))
            | AddressMode::Immediate(Immediate::High(address_reference))
            | AddressMode::Relative(address_reference) => {
                if &address_reference.name == name {
                    result += 1;
                }
            }

            AddressMode::Immediate(Immediate::Byte(_))
            | crate::memory::address_mode::AddressMode::Implied
            | AddressMode::Accumulator => {}
        }
    }
    result
}

fn update_label_addresses(application: &mut Application) -> AssemblerResult<()> {
    // First go over all labels and add dummy addresses to the application address lookup.
    // This will ensure that the correct instruction byte size can be determined.
    let mut label_addresses = HashMap::<String, Address>::default();
    let mut function_addresses = HashMap::<String, Address>::default();

    let mut init_label_addresses = |instructions: &Instructions| {
        for instruction in &instructions.instructions {
            if let Operation::Label(label) = &instruction.operation {
                label_addresses.insert(label.clone(), 0xFFFF);
            }
        }
    };
    for module in &application.modules {
        init_label_addresses(&module.instructions);
        for function in &module.functions {
            function_addresses.insert(function.name.clone(), 0xFFFF);
            init_label_addresses(&function.instructions);
        }
    }
    application.address_lookup.extend(label_addresses);
    application.address_lookup.extend(function_addresses);

    // Now go over all the labels again and determine the correct address.
    let mut label_addresses = HashMap::<String, Address>::default();
    let mut function_addresses = HashMap::<String, Address>::default();
    let mut current_address = application.entry_point;

    let mut update_label_addresses_instructions =
        |current_address: &mut Address, instructions: &Instructions| -> AssemblerResult<()> {
            for instruction in &instructions.instructions {
                if let Operation::Label(label) = &instruction.operation {
                    label_addresses.insert(label.clone(), *current_address);
                }
                let byte_size = instruction.byte_size(application)?;
                *current_address += byte_size;
            }
            Ok(())
        };

    for module in &application.modules {
        update_label_addresses_instructions(&mut current_address, &module.instructions)?;
        for function in &module.functions {
            function_addresses.insert(function.name.clone(), current_address);
            update_label_addresses_instructions(&mut current_address, &function.instructions)?;
        }
    }

    application.address_lookup.extend(label_addresses);
    application.address_lookup.extend(function_addresses);
    Ok(())
}