151 lines
3.5 KiB
Go
151 lines
3.5 KiB
Go
/*
|
|
* Copyright (c) 2024 The GoPlus Authors (goplus.org). All rights reserved.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package ssa
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"go/types"
|
|
"log"
|
|
|
|
"github.com/goplus/llvm"
|
|
)
|
|
|
|
// -----------------------------------------------------------------------------
|
|
|
|
type aBasicBlock struct {
|
|
impl llvm.BasicBlock
|
|
fn Function
|
|
idx int
|
|
}
|
|
|
|
// BasicBlock represents a basic block in a function.
|
|
type BasicBlock = *aBasicBlock
|
|
|
|
// Parent returns the function to which the basic block belongs.
|
|
func (p BasicBlock) Parent() Function {
|
|
return p.fn
|
|
}
|
|
|
|
// Index returns the index of the basic block in the parent function.
|
|
func (p BasicBlock) Index() int {
|
|
return p.idx
|
|
}
|
|
|
|
// -----------------------------------------------------------------------------
|
|
|
|
type aBuilder struct {
|
|
impl llvm.Builder
|
|
Func Function
|
|
Prog Program
|
|
}
|
|
|
|
// Builder represents a builder for creating instructions in a function.
|
|
type Builder = *aBuilder
|
|
|
|
// SetBlock sets the current block to the specified basic block.
|
|
func (b Builder) SetBlock(blk BasicBlock) Builder {
|
|
if b.Func != blk.fn {
|
|
panic("mismatched function")
|
|
}
|
|
if debugInstr {
|
|
log.Printf("Block _llgo_%v:\n", blk.idx)
|
|
}
|
|
b.impl.SetInsertPointAtEnd(blk.impl)
|
|
return b
|
|
}
|
|
|
|
// Panic emits a panic instruction.
|
|
func (b Builder) Panic(v Expr) {
|
|
if debugInstr {
|
|
log.Printf("Panic %v\n", v.impl)
|
|
}
|
|
pkg := b.Func.Pkg
|
|
b.Call(pkg.rtFunc("TracePanic"), v)
|
|
b.impl.CreateUnreachable()
|
|
}
|
|
|
|
// Unreachable emits an unreachable instruction.
|
|
func (b Builder) Unreachable() {
|
|
b.impl.CreateUnreachable()
|
|
}
|
|
|
|
// Return emits a return instruction.
|
|
func (b Builder) Return(results ...Expr) {
|
|
if debugInstr {
|
|
var b bytes.Buffer
|
|
fmt.Fprint(&b, "Return ")
|
|
for i, arg := range results {
|
|
if i > 0 {
|
|
fmt.Fprint(&b, ", ")
|
|
}
|
|
fmt.Fprint(&b, arg.impl)
|
|
}
|
|
log.Println(b.String())
|
|
}
|
|
switch n := len(results); n {
|
|
case 0:
|
|
b.impl.CreateRetVoid()
|
|
case 1:
|
|
b.impl.CreateRet(results[0].impl)
|
|
default:
|
|
tret := b.Func.raw.Type.(*types.Signature).Results()
|
|
b.impl.CreateAggregateRet(llvmParams(results, tret, b))
|
|
}
|
|
}
|
|
|
|
// Jump emits a jump instruction.
|
|
func (b Builder) Jump(jmpb BasicBlock) {
|
|
if b.Func != jmpb.fn {
|
|
panic("mismatched function")
|
|
}
|
|
if debugInstr {
|
|
log.Printf("Jump _llgo_%v\n", jmpb.idx)
|
|
}
|
|
b.impl.CreateBr(jmpb.impl)
|
|
}
|
|
|
|
// If emits an if instruction.
|
|
func (b Builder) If(cond Expr, thenb, elseb BasicBlock) {
|
|
if b.Func != thenb.fn || b.Func != elseb.fn {
|
|
panic("mismatched function")
|
|
}
|
|
if debugInstr {
|
|
log.Printf("If %v, _llgo_%v, _llgo_%v\n", cond.impl, thenb.idx, elseb.idx)
|
|
}
|
|
b.impl.CreateCondBr(cond.impl, thenb.impl, elseb.impl)
|
|
}
|
|
|
|
// The MapUpdate instruction updates the association of Map[Key] to
|
|
// Value.
|
|
//
|
|
// Pos() returns the ast.KeyValueExpr.Colon or ast.IndexExpr.Lbrack,
|
|
// if explicit in the source.
|
|
//
|
|
// Example printed form:
|
|
//
|
|
// t0[t1] = t2
|
|
func (b Builder) MapUpdate(m, k, v Expr) {
|
|
if debugInstr {
|
|
log.Printf("MapUpdate %v[%v] = %v\n", m.impl, k.impl, v.impl)
|
|
}
|
|
// TODO(xsw)
|
|
// panic("todo")
|
|
}
|
|
|
|
// -----------------------------------------------------------------------------
|