cli/q
Code Issues 1 Pull Requests Activity

Reduced memory usage

Browse Source
This commit is contained in:
Eduard Urbach
2024-06-28 10:28:23 +02:00
parent 668971808b
commit c8b25dd5b7
5 changed files with 42 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/build/compile.go
View File

@ -39,8 +39,6 @@ func compile(functions <-chan *Function, errors <-chan error) (Result, error) {
if function.Error != nil {
return result, function.Error
}
result.instructionCount += len(function.Assembler.Instructions)
}
main, exists := result.Unused["main"]
@ -49,8 +47,8 @@ func compile(functions <-chan *Function, errors <-chan error) (Result, error) {
return result, fail.MissingMainFunction
}
result.Used = append(result.Used, main)
delete(result.Unused, "main")
result.Used = make([]*Function, 0, len(result.Unused))
result.markAlive(main)
result.findAliveCode(main)
return result, nil
@ -73,21 +71,27 @@ func compileFunctions(functions map[string]*Function) {
}
// findAliveCode recursively finds all the calls to external functions and marks them as required.
func (result *Result) findAliveCode(f *Function) {
for _, x := range f.Assembler.Instructions {
func (result *Result) findAliveCode(caller *Function) {
for _, x := range caller.Assembler.Instructions {
if x.Mnemonic != asm.CALL {
continue
}
name := x.Data.(*asm.Label).Name
called, exists := result.Unused[name]
callee, exists := result.Unused[name]
if !exists {
continue
}
result.Used = append(result.Used, called)
delete(result.Unused, name)
result.findAliveCode(called)
result.markAlive(callee)
result.findAliveCode(callee)
}
}
// markAlive marks a function as being alive.
func (result *Result) markAlive(f *Function) {
result.Used = append(result.Used, f)
result.InstructionCount += len(f.Assembler.Instructions)
delete(result.Unused, f.Name)
}