Skip to content

Commit

Permalink
build: instantiate generics
Browse files Browse the repository at this point in the history
  • Loading branch information
@visualfc
visualfc committed Jun 18, 2024
1 parent e35d70f commit 8d34622
Show file tree
Hide file tree
Showing 12 changed files with 561 additions and 7 deletions.
18 changes: 18 additions & 0 deletions cl/_testgo/tpindex/in.go
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
package main

// The index function returns the index of the first occurrence of v in s,
// or -1 if not present.
func index[E comparable](s []E, v E) int {
for i, vs := range s {
if v == vs {
return i
}
}
return -1
}

func main() {
s := []int{1, 3, 5, 2, 4}
println(index(s, 3))
println(index(s, 6))
}
92 changes: 92 additions & 0 deletions cl/_testgo/tpindex/out.ll
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,92 @@
; ModuleID = 'main'
source_filename = "main"

%"github.com/goplus/llgo/internal/runtime.Slice" = type { ptr, i64, i64 }

@"main.init$guard" = global i1 false, align 1
@__llgo_argc = global i32 0, align 4
@__llgo_argv = global ptr null, align 8

define void @main.init() {
_llgo_0:
%0 = load i1, ptr @"main.init$guard", align 1
br i1 %0, label %_llgo_2, label %_llgo_1

_llgo_1: ; preds = %_llgo_0
store i1 true, ptr @"main.init$guard", align 1
br label %_llgo_2

_llgo_2: ; preds = %_llgo_1, %_llgo_0
ret void
}

define i32 @main(i32 %0, ptr %1) {
_llgo_0:
store i32 %0, ptr @__llgo_argc, align 4
store ptr %1, ptr @__llgo_argv, align 8
call void @"github.com/goplus/llgo/internal/runtime.init"()
call void @main.init()
%2 = call ptr @"github.com/goplus/llgo/internal/runtime.AllocZ"(i64 40)
%3 = getelementptr inbounds i64, ptr %2, i64 0
store i64 1, ptr %3, align 4
%4 = getelementptr inbounds i64, ptr %2, i64 1
store i64 3, ptr %4, align 4
%5 = getelementptr inbounds i64, ptr %2, i64 2
store i64 5, ptr %5, align 4
%6 = getelementptr inbounds i64, ptr %2, i64 3
store i64 2, ptr %6, align 4
%7 = getelementptr inbounds i64, ptr %2, i64 4
store i64 4, ptr %7, align 4
%8 = alloca %"github.com/goplus/llgo/internal/runtime.Slice", align 8
%9 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.Slice", ptr %8, i32 0, i32 0
store ptr %2, ptr %9, align 8
%10 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.Slice", ptr %8, i32 0, i32 1
store i64 5, ptr %10, align 4
%11 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.Slice", ptr %8, i32 0, i32 2
store i64 5, ptr %11, align 4
%12 = load %"github.com/goplus/llgo/internal/runtime.Slice", ptr %8, align 8
%13 = call i64 @"main.index[int]"(%"github.com/goplus/llgo/internal/runtime.Slice" %12, i64 3)
call void @"github.com/goplus/llgo/internal/runtime.PrintInt"(i64 %13)
call void @"github.com/goplus/llgo/internal/runtime.PrintByte"(i8 10)
%14 = call i64 @"main.index[int]"(%"github.com/goplus/llgo/internal/runtime.Slice" %12, i64 6)
call void @"github.com/goplus/llgo/internal/runtime.PrintInt"(i64 %14)
call void @"github.com/goplus/llgo/internal/runtime.PrintByte"(i8 10)
ret i32 0
}

declare void @"github.com/goplus/llgo/internal/runtime.init"()

declare ptr @"github.com/goplus/llgo/internal/runtime.AllocZ"(i64)

define i64 @"main.index[int]"(%"github.com/goplus/llgo/internal/runtime.Slice" %0, i64 %1) {
_llgo_0:
%2 = extractvalue %"github.com/goplus/llgo/internal/runtime.Slice" %0, 1
br label %_llgo_1

_llgo_1: ; preds = %_llgo_2, %_llgo_0
%3 = phi i64 [ -1, %_llgo_0 ], [ %4, %_llgo_2 ]
%4 = add i64 %3, 1
%5 = icmp slt i64 %4, %2
br i1 %5, label %_llgo_2, label %_llgo_3

_llgo_2: ; preds = %_llgo_1
%6 = icmp slt i64 %4, 0
call void @"github.com/goplus/llgo/internal/runtime.AssertIndexRange"(i1 %6)
%7 = extractvalue %"github.com/goplus/llgo/internal/runtime.Slice" %0, 0
%8 = getelementptr inbounds i64, ptr %7, i64 %4
%9 = load i64, ptr %8, align 4
%10 = icmp eq i64 %1, %9
br i1 %10, label %_llgo_4, label %_llgo_1

_llgo_3: ; preds = %_llgo_1
ret i64 -1

_llgo_4: ; preds = %_llgo_2
ret i64 %4
}

declare void @"github.com/goplus/llgo/internal/runtime.PrintInt"(i64)

declare void @"github.com/goplus/llgo/internal/runtime.PrintByte"(i8)

declare void @"github.com/goplus/llgo/internal/runtime.AssertIndexRange"(i1)
36 changes: 36 additions & 0 deletions cl/_testgo/tprecur/in.go
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,36 @@
package main

func main() {
recursive()
}

func recursive() {
type T int
if got, want := recur1[T](5), T(110); got != want {
panic("error")
}
}

type Integer interface {
~int | ~int32 | ~int64
}

func recur1[T Integer](n T) T {
if n == 0 || n == 1 {
return T(1)
} else {
return n * recur2(n-1)
}
}

func recur2[T Integer](n T) T {
list := make([]T, n)
for i, _ := range list {
list[i] = T(i + 1)
}
var sum T
for _, elt := range list {
sum += elt
}
return sum + recur1(n-1)
}
168 changes: 168 additions & 0 deletions cl/_testgo/tprecur/out.ll
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,168 @@
; ModuleID = 'main'
source_filename = "main"

%"github.com/goplus/llgo/internal/runtime.String" = type { ptr, i64 }
%"github.com/goplus/llgo/internal/runtime.eface" = type { ptr, ptr }
%"github.com/goplus/llgo/internal/runtime.Slice" = type { ptr, i64, i64 }

@"main.init$guard" = global i1 false, align 1
@__llgo_argc = global i32 0, align 4
@__llgo_argv = global ptr null, align 8
@0 = private unnamed_addr constant [5 x i8] c"error", align 1
@_llgo_string = linkonce global ptr null, align 8

define void @main.init() {
_llgo_0:
%0 = load i1, ptr @"main.init$guard", align 1
br i1 %0, label %_llgo_2, label %_llgo_1

_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
ret void
}

define i32 @main(i32 %0, ptr %1) {
_llgo_0:
store i32 %0, ptr @__llgo_argc, align 4
store ptr %1, ptr @__llgo_argv, align 8
call void @"github.com/goplus/llgo/internal/runtime.init"()
call void @main.init()
call void @main.recursive()
ret i32 0
}

define void @main.recursive() {
_llgo_0:
%0 = call i64 @"main.recur1[command-line-arguments.T]"(i64 5)
%1 = icmp ne i64 %0, 110
br i1 %1, label %_llgo_1, label %_llgo_2

_llgo_1: ; preds = %_llgo_0
%2 = alloca %"github.com/goplus/llgo/internal/runtime.String", align 8
%3 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %2, i32 0, i32 0
store ptr @0, ptr %3, align 8
%4 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %2, i32 0, i32 1
store i64 5, ptr %4, align 4
%5 = load %"github.com/goplus/llgo/internal/runtime.String", ptr %2, align 8
%6 = load ptr, ptr @_llgo_string, align 8
%7 = call ptr @"github.com/goplus/llgo/internal/runtime.AllocU"(i64 16)
store %"github.com/goplus/llgo/internal/runtime.String" %5, ptr %7, align 8
%8 = alloca %"github.com/goplus/llgo/internal/runtime.eface", align 8
%9 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.eface", ptr %8, i32 0, i32 0
store ptr %6, ptr %9, align 8
%10 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.eface", ptr %8, i32 0, i32 1
store ptr %7, ptr %10, align 8
%11 = load %"github.com/goplus/llgo/internal/runtime.eface", ptr %8, align 8
call void @"github.com/goplus/llgo/internal/runtime.Panic"(%"github.com/goplus/llgo/internal/runtime.eface" %11)
unreachable

_llgo_2: ; preds = %_llgo_0
ret void
}

declare void @"github.com/goplus/llgo/internal/runtime.init"()

define i64 @"main.recur1[command-line-arguments.T]"(i64 %0) {
_llgo_0:
%1 = icmp eq i64 %0, 0
br i1 %1, label %_llgo_1, label %_llgo_3

_llgo_1: ; preds = %_llgo_3, %_llgo_0
ret i64 1

_llgo_2: ; preds = %_llgo_3
%2 = sub i64 %0, 1
%3 = call i64 @"main.recur2[command-line-arguments.T]"(i64 %2)
%4 = mul i64 %0, %3
ret i64 %4

_llgo_3: ; preds = %_llgo_0
%5 = icmp eq i64 %0, 1
br i1 %5, label %_llgo_1, label %_llgo_2
}

define void @"main.init$after"() {
_llgo_0:
%0 = load ptr, ptr @_llgo_string, 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 24)
store ptr %2, ptr @_llgo_string, align 8
br label %_llgo_2

_llgo_2: ; preds = %_llgo_1, %_llgo_0
ret void
}

declare ptr @"github.com/goplus/llgo/internal/runtime.Basic"(i64)

declare ptr @"github.com/goplus/llgo/internal/runtime.AllocU"(i64)

declare void @"github.com/goplus/llgo/internal/runtime.Panic"(%"github.com/goplus/llgo/internal/runtime.eface")

define i64 @"main.recur2[command-line-arguments.T]"(i64 %0) {
_llgo_0:
%1 = mul i64 %0, 8
%2 = call ptr @"github.com/goplus/llgo/internal/runtime.AllocZ"(i64 %1)
%3 = alloca %"github.com/goplus/llgo/internal/runtime.Slice", align 8
%4 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.Slice", ptr %3, i32 0, i32 0
store ptr %2, ptr %4, align 8
%5 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.Slice", ptr %3, i32 0, i32 1
store i64 %0, ptr %5, align 4
%6 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.Slice", ptr %3, i32 0, i32 2
store i64 %0, ptr %6, align 4
%7 = load %"github.com/goplus/llgo/internal/runtime.Slice", ptr %3, align 8
%8 = extractvalue %"github.com/goplus/llgo/internal/runtime.Slice" %7, 1
br label %_llgo_1

_llgo_1: ; preds = %_llgo_2, %_llgo_0
%9 = phi i64 [ -1, %_llgo_0 ], [ %10, %_llgo_2 ]
%10 = add i64 %9, 1
%11 = icmp slt i64 %10, %8
br i1 %11, label %_llgo_2, label %_llgo_3

_llgo_2: ; preds = %_llgo_1
%12 = add i64 %10, 1
%13 = icmp slt i64 %10, 0
call void @"github.com/goplus/llgo/internal/runtime.AssertIndexRange"(i1 %13)
%14 = extractvalue %"github.com/goplus/llgo/internal/runtime.Slice" %7, 0
%15 = getelementptr inbounds i64, ptr %14, i64 %10
store i64 %12, ptr %15, align 4
br label %_llgo_1

_llgo_3: ; preds = %_llgo_1
%16 = extractvalue %"github.com/goplus/llgo/internal/runtime.Slice" %7, 1
br label %_llgo_4

_llgo_4: ; preds = %_llgo_5, %_llgo_3
%17 = phi i64 [ 0, %_llgo_3 ], [ %25, %_llgo_5 ]
%18 = phi i64 [ -1, %_llgo_3 ], [ %19, %_llgo_5 ]
%19 = add i64 %18, 1
%20 = icmp slt i64 %19, %16
br i1 %20, label %_llgo_5, label %_llgo_6

_llgo_5: ; preds = %_llgo_4
%21 = icmp slt i64 %19, 0
call void @"github.com/goplus/llgo/internal/runtime.AssertIndexRange"(i1 %21)
%22 = extractvalue %"github.com/goplus/llgo/internal/runtime.Slice" %7, 0
%23 = getelementptr inbounds i64, ptr %22, i64 %19
%24 = load i64, ptr %23, align 4
%25 = add i64 %17, %24
br label %_llgo_4

_llgo_6: ; preds = %_llgo_4
%26 = sub i64 %0, 1
%27 = call i64 @"main.recur1[command-line-arguments.T]"(i64 %26)
%28 = add i64 %17, %27
ret i64 %28
}

declare ptr @"github.com/goplus/llgo/internal/runtime.AllocZ"(i64)

declare void @"github.com/goplus/llgo/internal/runtime.AssertIndexRange"(i1)
61 changes: 61 additions & 0 deletions cl/_testgo/tptypes/in.go
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,61 @@
package main

type Data[T any] struct {
v T
}

func (p *Data[T]) Set(v T) {
p.v = v
}

func (p *(Data[T1])) Set2(v T1) {
p.v = v
}

type sliceOf[E any] interface {
~[]E
}

type Slice[S sliceOf[T], T any] struct {
Data S
}

func (p *Slice[S, T]) Append(t ...T) S {
p.Data = append(p.Data, t...)
return p.Data
}

func (p *Slice[S1, T1]) Append2(t ...T1) S1 {
p.Data = append(p.Data, t...)
return p.Data
}

type (
DataInt = Data[int]
SliceInt = Slice[[]int, int]
DataString = Data[string]
SliceString = Slice[[]string, string]
)

func main() {
println(DataInt{1}.v)
println(DataString{"hello"}.v)
println(Data[int]{100}.v)
println(Data[string]{"hello"}.v)

// TODO
// println(Data[struct {
// X int
// Y int
// }]{}.v.X)

v1 := SliceInt{}
v2 := SliceString{}
v3 := Slice[[]int, int]{}
v3.Append([]int{1, 2, 3, 4}...)
v3.Append2([]int{1, 2, 3, 4}...)

println(v1.Data)
println(v2.Data)
println(v3.Data)
}
Loading

0 comments on commit 8d34622

Please sign in to comment.