From 5c5b8e62e5f623127c676b98edeebfb75da59c00 Mon Sep 17 00:00:00 2001
From: visualfc <visualfc@gmail.com>
Date: Wed, 1 May 2024 21:56:11 +0800
Subject: [PATCH] implement ssa.index

---
 cl/_testdata/index/in.go  |  25 ++++++++
 cl/_testdata/index/out.ll | 130 ++++++++++++++++++++++++++++++++++++++
 cl/compile.go             |  11 ++++
 ssa/expr.go               |  37 +++++++++++
 4 files changed, 203 insertions(+)
 create mode 100644 cl/_testdata/index/in.go
 create mode 100644 cl/_testdata/index/out.ll

diff --git a/cl/_testdata/index/in.go b/cl/_testdata/index/in.go
new file mode 100644
index 00000000..af7c83e2
--- /dev/null
+++ b/cl/_testdata/index/in.go
@@ -0,0 +1,25 @@
+package main
+
+import "github.com/goplus/llgo/internal/runtime/c"
+
+type point struct {
+	x int
+	y int
+}
+
+func main() {
+	a := [...]point{{1, 2}, {3, 4}, {5, 6}}[2]
+	c.Printf(c.Str("%d %d\n"), a.x, a.y)
+
+	b := [...][2]int{[2]int{1, 2}, [2]int{3, 4}}[1]
+	c.Printf(c.Str("%d %d\n"), b[0], b[1])
+
+	var i int = 2
+	n := [...]int{1, 2, 3, 4, 5}[i]
+	c.Printf(c.Str("%d\n"), n)
+	c.Printf(c.Str("%d\n"), [...]int{1, 2, 3, 4, 5}[i])
+
+	s := "123456"
+	c.Printf(c.Str("%c\n"), s[i])
+	c.Printf(c.Str("%c\n"), "123456"[1])
+}
diff --git a/cl/_testdata/index/out.ll b/cl/_testdata/index/out.ll
new file mode 100644
index 00000000..5a7e3d7a
--- /dev/null
+++ b/cl/_testdata/index/out.ll
@@ -0,0 +1,130 @@
+; ModuleID = 'main'
+source_filename = "main"
+
+%main.point = type { i64, i64 }
+%"github.com/goplus/llgo/internal/runtime.String" = type { ptr, i64 }
+
+@"main.init$guard" = global ptr null
+@0 = private unnamed_addr constant [7 x i8] c"%d %d\0A\00", align 1
+@1 = private unnamed_addr constant [7 x i8] c"%d %d\0A\00", align 1
+@2 = private unnamed_addr constant [4 x i8] c"%d\0A\00", align 1
+@3 = private unnamed_addr constant [4 x i8] c"%d\0A\00", align 1
+@4 = private unnamed_addr constant [4 x i8] c"%c\0A\00", align 1
+@5 = private unnamed_addr constant [7 x i8] c"123456\00", align 1
+@6 = private unnamed_addr constant [4 x i8] c"%c\0A\00", align 1
+@7 = private unnamed_addr constant [7 x i8] c"123456\00", align 1
+
+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 void @main() {
+_llgo_0:
+  call void @"github.com/goplus/llgo/internal/runtime.init"()
+  call void @main.init()
+  %0 = alloca %main.point, align 8
+  %1 = alloca [3 x %main.point], align 8
+  %2 = getelementptr inbounds %main.point, ptr %1, i64 0
+  %3 = getelementptr inbounds %main.point, ptr %2, i32 0, i32 0
+  %4 = getelementptr inbounds %main.point, ptr %2, i32 0, i32 1
+  %5 = getelementptr inbounds %main.point, ptr %1, i64 1
+  %6 = getelementptr inbounds %main.point, ptr %5, i32 0, i32 0
+  %7 = getelementptr inbounds %main.point, ptr %5, i32 0, i32 1
+  %8 = getelementptr inbounds %main.point, ptr %1, i64 2
+  %9 = getelementptr inbounds %main.point, ptr %8, i32 0, i32 0
+  %10 = getelementptr inbounds %main.point, ptr %8, i32 0, i32 1
+  store i64 1, ptr %3, align 4
+  store i64 2, ptr %4, align 4
+  store i64 3, ptr %6, align 4
+  store i64 4, ptr %7, align 4
+  store i64 5, ptr %9, align 4
+  store i64 6, ptr %10, align 4
+  %11 = load [3 x %main.point], ptr %1, align 4
+  %12 = getelementptr inbounds %main.point, ptr %1, i64 2
+  %13 = load %main.point, ptr %12, align 4
+  store %main.point %13, ptr %0, align 4
+  %14 = getelementptr inbounds %main.point, ptr %0, i32 0, i32 0
+  %15 = load i64, ptr %14, align 4
+  %16 = getelementptr inbounds %main.point, ptr %0, i32 0, i32 1
+  %17 = load i64, ptr %16, align 4
+  %18 = call i32 (ptr, ...) @printf(ptr @0, i64 %15, i64 %17)
+  %19 = alloca [2 x i64], align 8
+  %20 = alloca [2 x [2 x i64]], align 8
+  %21 = getelementptr inbounds [2 x i64], ptr %20, i64 0
+  %22 = getelementptr inbounds i64, ptr %21, i64 0
+  %23 = getelementptr inbounds i64, ptr %21, i64 1
+  %24 = getelementptr inbounds [2 x i64], ptr %20, i64 1
+  %25 = getelementptr inbounds i64, ptr %24, i64 0
+  %26 = getelementptr inbounds i64, ptr %24, i64 1
+  store i64 1, ptr %22, align 4
+  store i64 2, ptr %23, align 4
+  store i64 3, ptr %25, align 4
+  store i64 4, ptr %26, align 4
+  %27 = load [2 x [2 x i64]], ptr %20, align 4
+  %28 = getelementptr inbounds [2 x i64], ptr %20, i64 1
+  %29 = load [2 x i64], ptr %28, align 4
+  store [2 x i64] %29, ptr %19, align 4
+  %30 = getelementptr inbounds i64, ptr %19, i64 0
+  %31 = load i64, ptr %30, align 4
+  %32 = getelementptr inbounds i64, ptr %19, i64 1
+  %33 = load i64, ptr %32, align 4
+  %34 = call i32 (ptr, ...) @printf(ptr @1, i64 %31, i64 %33)
+  %35 = alloca [5 x i64], align 8
+  %36 = getelementptr inbounds i64, ptr %35, i64 0
+  %37 = getelementptr inbounds i64, ptr %35, i64 1
+  %38 = getelementptr inbounds i64, ptr %35, i64 2
+  %39 = getelementptr inbounds i64, ptr %35, i64 3
+  %40 = getelementptr inbounds i64, ptr %35, i64 4
+  store i64 1, ptr %36, align 4
+  store i64 2, ptr %37, align 4
+  store i64 3, ptr %38, align 4
+  store i64 4, ptr %39, align 4
+  store i64 5, ptr %40, align 4
+  %41 = load [5 x i64], ptr %35, align 4
+  %42 = getelementptr inbounds i64, ptr %35, i64 2
+  %43 = load i64, ptr %42, align 4
+  %44 = call i32 (ptr, ...) @printf(ptr @2, i64 %43)
+  %45 = alloca [5 x i64], align 8
+  %46 = getelementptr inbounds i64, ptr %45, i64 0
+  %47 = getelementptr inbounds i64, ptr %45, i64 1
+  %48 = getelementptr inbounds i64, ptr %45, i64 2
+  %49 = getelementptr inbounds i64, ptr %45, i64 3
+  %50 = getelementptr inbounds i64, ptr %45, i64 4
+  store i64 1, ptr %46, align 4
+  store i64 2, ptr %47, align 4
+  store i64 3, ptr %48, align 4
+  store i64 4, ptr %49, align 4
+  store i64 5, ptr %50, align 4
+  %51 = load [5 x i64], ptr %45, align 4
+  %52 = getelementptr inbounds i64, ptr %45, i64 2
+  %53 = load i64, ptr %52, align 4
+  %54 = call i32 (ptr, ...) @printf(ptr @3, i64 %53)
+  %55 = call %"github.com/goplus/llgo/internal/runtime.String" @"github.com/goplus/llgo/internal/runtime.NewString"(ptr @5, i64 6)
+  %56 = call ptr @"github.com/goplus/llgo/internal/runtime.StringData"(%"github.com/goplus/llgo/internal/runtime.String" %55)
+  %57 = getelementptr inbounds i8, ptr %56, i64 2
+  %58 = load i8, ptr %57, align 1
+  %59 = call i32 (ptr, ...) @printf(ptr @4, i8 %58)
+  %60 = call %"github.com/goplus/llgo/internal/runtime.String" @"github.com/goplus/llgo/internal/runtime.NewString"(ptr @7, i64 6)
+  %61 = call ptr @"github.com/goplus/llgo/internal/runtime.StringData"(%"github.com/goplus/llgo/internal/runtime.String" %60)
+  %62 = getelementptr inbounds i8, ptr %61, i64 1
+  %63 = load i8, ptr %62, align 1
+  %64 = call i32 (ptr, ...) @printf(ptr @6, i8 %63)
+  ret void
+}
+
+declare void @"github.com/goplus/llgo/internal/runtime.init"()
+
+declare i32 @printf(ptr, ...)
+
+declare %"github.com/goplus/llgo/internal/runtime.String" @"github.com/goplus/llgo/internal/runtime.NewString"(ptr, i64)
+
+declare ptr @"github.com/goplus/llgo/internal/runtime.StringData"(%"github.com/goplus/llgo/internal/runtime.String")
diff --git a/cl/compile.go b/cl/compile.go
index c374a5d5..82fc4d7b 100644
--- a/cl/compile.go
+++ b/cl/compile.go
@@ -416,6 +416,17 @@ func (p *context) compileInstrOrValue(b llssa.Builder, iv instrOrValue, asValue
 		x := p.compileValue(b, vx)
 		idx := p.compileValue(b, v.Index)
 		ret = b.IndexAddr(x, idx)
+	case *ssa.Index:
+		x := p.compileValue(b, v.X)
+		idx := p.compileValue(b, v.Index)
+		ret = b.Index(x, idx, func(e llssa.Expr) (ret llssa.Expr) {
+			if e == x {
+				if n, ok := v.X.(*ssa.UnOp); ok {
+					return p.compileValue(b, n.X)
+				}
+			}
+			panic(fmt.Errorf("todo addr of %v", e))
+		})
 	case *ssa.Slice:
 		vx := v.X
 		if _, ok := p.isVArgs(vx); ok { // varargs: this is a varargs slice
diff --git a/ssa/expr.go b/ssa/expr.go
index c19ee41d..9d780fe4 100644
--- a/ssa/expr.go
+++ b/ssa/expr.go
@@ -438,6 +438,43 @@ func (b Builder) IndexAddr(x, idx Expr) Expr {
 	return Expr{llvm.CreateInBoundsGEP(b.impl, telem.ll, x.impl, indices), pt}
 }
 
+// The Index instruction yields element Index of collection X, an array,
+// string or type parameter containing an array, a string, a pointer to an,
+// array or a slice.
+//
+// Example printed form:
+//
+//	t2 = t0[t1]
+func (b Builder) Index(x, idx Expr, addr func(Expr) Expr) Expr {
+	if debugInstr {
+		log.Printf("Index %v, %v\n", x.impl, idx.impl)
+	}
+	prog := b.Prog
+	var telem Type
+	var ptr Expr
+	switch t := x.t.Underlying().(type) {
+	case *types.Basic:
+		if t.Info()&types.IsString == 0 {
+			panic(fmt.Errorf("invalid operation: cannot index %v", t))
+		}
+		telem = prog.Type(types.Typ[types.Byte])
+		pkg := b.fn.pkg
+		ptr = b.InlineCall(pkg.rtFunc("StringData"), x)
+	case *types.Array:
+		telem = prog.Index(x.Type)
+		if addr != nil {
+			ptr = addr(x)
+		} else {
+			ptr = b.Alloca(prog.IntVal(uint64(t.Len()*b.Prog.sizs.Sizeof(t.Elem())), prog.Index(x.Type)))
+			b.Store(ptr, x)
+		}
+	}
+	pt := prog.Pointer(telem)
+	indices := []llvm.Value{idx.impl}
+	buf := Expr{llvm.CreateInBoundsGEP(b.impl, telem.ll, ptr.impl, indices), pt}
+	return b.Load(buf)
+}
+
 // The Slice instruction yields a slice of an existing string, slice
 // or *array X between optional integer bounds Low and High.
 //