admin/llgo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

llgo/ssa: abiMapOf

Browse Source
This commit is contained in:
xushiwei
2024-06-14 22:07:48 +08:00
parent 47b20b01d0
commit 33d73eaecd
5 changed files with 63 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
cl/_testrt/map/out.ll
View File

@@ -4,6 +4,8 @@ source_filename = "main"
@"main.init$guard" = global ptr null
@__llgo_argc = global ptr null
@__llgo_argv = global ptr null
@"map[_llgo_int]_llgo_int" = linkonce global ptr null
@_llgo_int = linkonce global ptr null
@0 = private unnamed_addr constant [10 x i8] c"Hello %d\0A\00", align 1
define void @main.init() {
@@ -13,6 +15,7 @@ _llgo_0:
_llgo_1: ; preds = %_llgo_0
store i1 true, ptr @"main.init$guard", align 1
call void @"main.init$after"()
br label %_llgo_2
_llgo_2: ; preds = %_llgo_1, %_llgo_0
@@ -26,7 +29,13 @@ _llgo_0:
call void @"github.com/goplus/llgo/internal/runtime.init"()
call void @main.init()
%2 = call ptr @"github.com/goplus/llgo/internal/runtime.MakeSmallMap"()
%3 = call i32 (ptr, ...) @printf(ptr @0, <null operand!>)
%3 = load ptr, ptr @"map[_llgo_int]_llgo_int", align 8
%4 = call ptr @"github.com/goplus/llgo/internal/runtime.MapAssign"(ptr %3, ptr %2, i64 23)
store i64 100, ptr %4, align 4
%5 = load ptr, ptr @"map[_llgo_int]_llgo_int", align 8
%6 = call ptr @"github.com/goplus/llgo/internal/runtime.MapAssign"(ptr %5, ptr %2, i64 7)
store i64 29, ptr %6, align 4
%7 = call i32 (ptr, ...) @printf(ptr @0, <null operand!>)
ret i32 0
}
@@ -34,4 +43,37 @@ declare void @"github.com/goplus/llgo/internal/runtime.init"()
declare ptr @"github.com/goplus/llgo/internal/runtime.MakeSmallMap"()
define void @"main.init$after"() {
_llgo_0:
%0 = load ptr, ptr @_llgo_int, align 8
%1 = icmp eq ptr %0, null
br i1 %1, label %_llgo_1, label %_llgo_2
_llgo_1: ; preds = %_llgo_0
%2 = call ptr @"github.com/goplus/llgo/internal/runtime.Basic"(i64 2)
store ptr %2, ptr @_llgo_int, align 8
br label %_llgo_2
_llgo_2: ; preds = %_llgo_1, %_llgo_0
%3 = load ptr, ptr @_llgo_int, align 8
%4 = load ptr, ptr @_llgo_int, align 8
%5 = load ptr, ptr @"map[_llgo_int]_llgo_int", align 8
%6 = icmp eq ptr %5, null
br i1 %6, label %_llgo_3, label %_llgo_4
_llgo_3: ; preds = %_llgo_2
%7 = call ptr @"github.com/goplus/llgo/internal/runtime.MapOf"(ptr %3, ptr %4)
store ptr %7, ptr @"map[_llgo_int]_llgo_int", align 8
br label %_llgo_4
_llgo_4: ; preds = %_llgo_3, %_llgo_2
ret void
}
declare ptr @"github.com/goplus/llgo/internal/runtime.Basic"(i64)
declare ptr @"github.com/goplus/llgo/internal/runtime.MapOf"(ptr, ptr)
declare ptr @"github.com/goplus/llgo/internal/runtime.MapAssign"(ptr, ptr, ptr)
declare i32 @printf(ptr, ...)

6
internal/runtime/z_slice.go
View File

@@ -24,10 +24,10 @@ import (
// -----------------------------------------------------------------------------
// Slice is the runtime representation of a slice.
type Slice = slice
// type Slice = slice
type slice struct {
// Slice is the runtime representation of a slice.
type Slice struct {
data unsafe.Pointer
len int
cap int

4
ssa/abi/abi.go
View File

@@ -159,6 +159,10 @@ func (b *Builder) TypeName(t types.Type) (ret string, pub bool) {
return "_llgo_any", true
}
return b.InterfaceName(t)
case *types.Map:
key, pub1 := b.TypeName(t.Key())
elem, pub2 := b.TypeName(t.Elem())
return fmt.Sprintf("map[%s]%s", key, elem), pub1 && pub2
}
log.Panicf("todo: %T\n", t)
return

10
ssa/abitype.go
View File

@@ -66,6 +66,8 @@ func (b Builder) abiTypeOf(t types.Type) func() Expr {
return b.abiFuncOf(t)
case *types.Slice:
return b.abiSliceOf(t)
case *types.Map:
return b.abiMapOf(t)
case *types.Array:
return b.abiArrayOf(t)
}
@@ -244,6 +246,14 @@ func (b Builder) abiPointerOf(t *types.Pointer) func() Expr {
}
}
func (b Builder) abiMapOf(t *types.Map) func() Expr {
key := b.abiType(t.Key())
elem := b.abiType(t.Elem())
return func() Expr {
return b.Call(b.Pkg.rtFunc("MapOf"), key, elem)
}
}
func (b Builder) abiSliceOf(t *types.Slice) func() Expr {
elem := b.abiType(t.Elem())
return func() Expr {

4
ssa/datastruct.go
View File

@@ -365,7 +365,9 @@ func (b Builder) MapUpdate(m, k, v Expr) {
panic("TODO: not a map")
}
tabi := b.abiType(t.raw.Type)
ptr := b.InlineCall(b.Pkg.rtFunc("MapAssign"), tabi, m, k)
prog := b.Prog
ptrimpl := b.InlineCall(b.Pkg.rtFunc("MapAssign"), tabi, m, k).impl
ptr := Expr{ptrimpl, prog.Pointer(v.Type)}
b.Store(ptr, v) // TODO(xsw): indirect store
}