54 lines
971 B
Go
54 lines
971 B
Go
package core
|
|
|
|
import (
|
|
"git.akyoto.dev/cli/q/src/asm"
|
|
"git.akyoto.dev/cli/q/src/ast"
|
|
)
|
|
|
|
// CompileSwitch compiles a multi-branch instruction.
|
|
func (f *Function) CompileSwitch(s *ast.Switch) error {
|
|
f.count.multiBranch++
|
|
end := f.CreateLabel("switch end", f.count.multiBranch)
|
|
|
|
for _, branch := range s.Cases {
|
|
if branch.Condition == nil {
|
|
f.PushScope(branch.Body, f.File.Bytes)
|
|
err := f.CompileAST(branch.Body)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
f.PopScope()
|
|
break
|
|
}
|
|
|
|
f.count.branch++
|
|
|
|
var (
|
|
success = f.CreateLabel("case true", f.count.branch)
|
|
fail = f.CreateLabel("case false", 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
|
|
}
|
|
|
|
f.Jump(asm.JUMP, end)
|
|
f.PopScope()
|
|
f.AddLabel(fail)
|
|
}
|
|
|
|
f.AddLabel(end)
|
|
return nil
|
|
}
|