Elem, ElemRef

Author: xushiwei

codebuild.go

@@ -962,6 +962,36 @@ func (p *CodeBuilder) pushVal(v interface{}) *CodeBuilder {
 	return p
 }
 
+// Elem func
+func (p *CodeBuilder) Elem() *CodeBuilder {
+	if debug {
+		log.Println("Elem")
+	}
+	arg := p.stk.Get(-1)
+	t, ok := arg.Type.(*types.Pointer)
+	if !ok {
+		log.Panicln("TODO: not a pointer")
+	}
+	p.stk.Ret(1, internal.Elem{Val: &ast.StarExpr{X: arg.Val}, Type: t.Elem()})
+	return p
+}
+
+// Elem func
+func (p *CodeBuilder) ElemRef() *CodeBuilder {
+	if debug {
+		log.Println("ElemRef")
+	}
+	arg := p.stk.Get(-1)
+	t, ok := arg.Type.(*types.Pointer)
+	if !ok {
+		log.Panicln("TODO: not a pointer")
+	}
+	p.stk.Ret(1, internal.Elem{
+		Val: &ast.StarExpr{X: arg.Val}, Type: &refType{typ: t.Elem()},
+	})
+	return p
+}
+
 // MemberRef func
 func (p *CodeBuilder) MemberRef(name string) *CodeBuilder {
 	if debug {

package_test.go

@@ -1666,6 +1666,8 @@ func TestIndexRef(t *testing.T) {
 		Val(x).Val(0).IndexRef(1).Val(1).Assign(1).
 		Val(y).Val(1).IndexRef(1).Val(2).Assign(1).
 		Val(z).Val("a").IndexRef(1).Val(3).Assign(1).
+		Val(y).ElemRef().Val(x).Assign(1).
+		VarRef(x).Val(y).Elem().Assign(1).
 		End()
 
 	pkg.NewFunc(nil, "main", nil, nil, false).BodyStart(pkg).End()
@@ -1675,6 +1677,8 @@ func foo(x [10]int, y *[10]int, z map[string]int) {
 	x[0] = 1
 	y[1] = 2
 	z["a"] = 3
+	*y = x
+	x = *y
 }
 func main() {
 }