60 lines
1.0 KiB
Go
60 lines
1.0 KiB
Go
package core
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"git.akyoto.dev/cli/q/src/asm"
|
|
"git.akyoto.dev/cli/q/src/ast"
|
|
)
|
|
|
|
// CompileIf compiles a branch instruction.
|
|
func (f *Function) CompileIf(branch *ast.If) error {
|
|
for _, register := range f.CPU.Input {
|
|
f.SaveRegister(register)
|
|
}
|
|
|
|
f.count.branch++
|
|
|
|
var (
|
|
end string
|
|
success = fmt.Sprintf("%s_if_%d_true", f.UniqueName, f.count.branch)
|
|
fail = fmt.Sprintf("%s_if_%d_false", f.UniqueName, f.count.branch)
|
|
err = f.CompileCondition(branch.Condition, success, fail)
|
|
)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
f.AddLabel(success)
|
|
f.PushScope(branch.Body, f.File.Bytes)
|
|
err = f.CompileAST(branch.Body)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if branch.Else != nil {
|
|
end = fmt.Sprintf("%s_if_%d_end", f.UniqueName, f.count.branch)
|
|
f.Jump(asm.JUMP, end)
|
|
}
|
|
|
|
f.PopScope()
|
|
f.AddLabel(fail)
|
|
|
|
if branch.Else != nil {
|
|
f.PushScope(branch.Else, f.File.Bytes)
|
|
err = f.CompileAST(branch.Else)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
f.PopScope()
|
|
f.AddLabel(end)
|
|
return nil
|
|
}
|
|
|
|
return nil
|
|
}
|