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

ssa: PyVal support pointer to long

Browse Source
This commit is contained in:
visualfc
2025-07-30 15:42:34 +08:00
parent f43e2c0dee
commit 54dfd3c701
3 changed files with 70 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
cl/_testpy/list/in.go
View File

@@ -1,14 +1,17 @@
package main package main
import ( import (
"unsafe"
"github.com/goplus/lib/py" "github.com/goplus/lib/py"
"github.com/goplus/lib/py/math" "github.com/goplus/lib/py/math"
"github.com/goplus/lib/py/std" "github.com/goplus/lib/py/std"
) )
func main() { func main() {
v := 100
x := py.List(true, false, 1, float32(2.1), 3.1, uint(4), 1+2i, complex64(3+4i), x := py.List(true, false, 1, float32(2.1), 3.1, uint(4), 1+2i, complex64(3+4i),
"hello", []byte("world"), [...]byte{1, 2, 3}, [...]byte{}) "hello", []byte("world"), [...]byte{1, 2, 3}, [...]byte{}, &v, unsafe.Pointer(&v))
y := py.List(std.Abs, std.Print, math.Pi) y := py.List(std.Abs, std.Print, math.Pi)
std.Print(x, y) std.Print(x, y)
} }

112
cl/_testpy/list/out.ll
View File

@@ -34,57 +34,65 @@ _llgo_2: ; preds = %_llgo_1, %_llgo_0
define void @"github.com/goplus/llgo/cl/_testpy/list.main"() { define void @"github.com/goplus/llgo/cl/_testpy/list.main"() {
_llgo_0: _llgo_0:
%0 = call %"github.com/goplus/llgo/runtime/internal/runtime.Slice" @"github.com/goplus/llgo/runtime/internal/runtime.StringToBytes"(%"github.com/goplus/llgo/runtime/internal/runtime.String" { ptr @0, i64 5 }) %0 = call ptr @"github.com/goplus/llgo/runtime/internal/runtime.AllocZ"(i64 8)
%1 = alloca [3 x i8], align 1 store i64 100, ptr %0, align 4
call void @llvm.memset(ptr %1, i8 0, i64 3, i1 false) %1 = call %"github.com/goplus/llgo/runtime/internal/runtime.Slice" @"github.com/goplus/llgo/runtime/internal/runtime.StringToBytes"(%"github.com/goplus/llgo/runtime/internal/runtime.String" { ptr @0, i64 5 })
%2 = getelementptr inbounds i8, ptr %1, i64 0 %2 = alloca [3 x i8], align 1
%3 = getelementptr inbounds i8, ptr %1, i64 1 call void @llvm.memset(ptr %2, i8 0, i64 3, i1 false)
%4 = getelementptr inbounds i8, ptr %1, i64 2 %3 = getelementptr inbounds i8, ptr %2, i64 0
store i8 1, ptr %2, align 1 %4 = getelementptr inbounds i8, ptr %2, i64 1
store i8 2, ptr %3, align 1 %5 = getelementptr inbounds i8, ptr %2, i64 2
store i8 3, ptr %4, align 1 store i8 1, ptr %3, align 1
%5 = load [3 x i8], ptr %1, align 1 store i8 2, ptr %4, align 1
%6 = call ptr @PyList_New(i64 12) store i8 3, ptr %5, align 1
%7 = call ptr @PyBool_FromLong(i32 -1) %6 = load [3 x i8], ptr %2, align 1
%8 = call i32 @PyList_SetItem(ptr %6, i64 0, ptr %7) %7 = call ptr @PyList_New(i64 14)
%9 = call ptr @PyBool_FromLong(i32 0) %8 = call ptr @PyBool_FromLong(i32 -1)
%10 = call i32 @PyList_SetItem(ptr %6, i64 1, ptr %9) %9 = call i32 @PyList_SetItem(ptr %7, i64 0, ptr %8)
%11 = call ptr @PyLong_FromLongLong(i64 1) %10 = call ptr @PyBool_FromLong(i32 0)
%12 = call i32 @PyList_SetItem(ptr %6, i64 2, ptr %11) %11 = call i32 @PyList_SetItem(ptr %7, i64 1, ptr %10)
%13 = call ptr @PyFloat_FromDouble(double 0x4000CCCCC0000000) %12 = call ptr @PyLong_FromLongLong(i64 1)
%14 = call i32 @PyList_SetItem(ptr %6, i64 3, ptr %13) %13 = call i32 @PyList_SetItem(ptr %7, i64 2, ptr %12)
%15 = call ptr @PyFloat_FromDouble(double 3.100000e+00) %14 = call ptr @PyFloat_FromDouble(double 0x4000CCCCC0000000)
%16 = call i32 @PyList_SetItem(ptr %6, i64 4, ptr %15) %15 = call i32 @PyList_SetItem(ptr %7, i64 3, ptr %14)
%17 = call ptr @PyLong_FromUnsignedLongLong(i64 4) %16 = call ptr @PyFloat_FromDouble(double 3.100000e+00)
%18 = call i32 @PyList_SetItem(ptr %6, i64 5, ptr %17) %17 = call i32 @PyList_SetItem(ptr %7, i64 4, ptr %16)
%19 = call ptr @PyComplex_FromDoubles(double 1.000000e+00, double 2.000000e+00) %18 = call ptr @PyLong_FromUnsignedLongLong(i64 4)
%20 = call i32 @PyList_SetItem(ptr %6, i64 6, ptr %19) %19 = call i32 @PyList_SetItem(ptr %7, i64 5, ptr %18)
%21 = call ptr @PyComplex_FromDoubles(double 3.000000e+00, double 4.000000e+00) %20 = call ptr @PyComplex_FromDoubles(double 1.000000e+00, double 2.000000e+00)
%22 = call i32 @PyList_SetItem(ptr %6, i64 7, ptr %21) %21 = call i32 @PyList_SetItem(ptr %7, i64 6, ptr %20)
%23 = call ptr @PyUnicode_FromStringAndSize(ptr @1, i64 5) %22 = call ptr @PyComplex_FromDoubles(double 3.000000e+00, double 4.000000e+00)
%24 = call i32 @PyList_SetItem(ptr %6, i64 8, ptr %23) %23 = call i32 @PyList_SetItem(ptr %7, i64 7, ptr %22)
%25 = extractvalue %"github.com/goplus/llgo/runtime/internal/runtime.Slice" %0, 0 %24 = call ptr @PyUnicode_FromStringAndSize(ptr @1, i64 5)
%26 = extractvalue %"github.com/goplus/llgo/runtime/internal/runtime.Slice" %0, 1 %25 = call i32 @PyList_SetItem(ptr %7, i64 8, ptr %24)
%27 = call ptr @PyByteArray_FromStringAndSize(ptr %25, i64 %26) %26 = extractvalue %"github.com/goplus/llgo/runtime/internal/runtime.Slice" %1, 0
%28 = call i32 @PyList_SetItem(ptr %6, i64 9, ptr %27) %27 = extractvalue %"github.com/goplus/llgo/runtime/internal/runtime.Slice" %1, 1
%29 = alloca [3 x i8], align 1 %28 = call ptr @PyByteArray_FromStringAndSize(ptr %26, i64 %27)
call void @llvm.memset(ptr %29, i8 0, i64 3, i1 false) %29 = call i32 @PyList_SetItem(ptr %7, i64 9, ptr %28)
store [3 x i8] %5, ptr %29, align 1 %30 = alloca [3 x i8], align 1
%30 = getelementptr inbounds ptr, ptr %29, i64 0 call void @llvm.memset(ptr %30, i8 0, i64 3, i1 false)
%31 = call ptr @PyBytes_FromStringAndSize(ptr %30, i64 3) store [3 x i8] %6, ptr %30, align 1
%32 = call i32 @PyList_SetItem(ptr %6, i64 10, ptr %31) %31 = getelementptr inbounds ptr, ptr %30, i64 0
%33 = call ptr @PyBytes_FromStringAndSize(ptr null, i64 0) %32 = call ptr @PyBytes_FromStringAndSize(ptr %31, i64 3)
%34 = call i32 @PyList_SetItem(ptr %6, i64 11, ptr %33) %33 = call i32 @PyList_SetItem(ptr %7, i64 10, ptr %32)
%35 = load ptr, ptr @__llgo_py.math, align 8 %34 = call ptr @PyBytes_FromStringAndSize(ptr null, i64 0)
%36 = call ptr @PyObject_GetAttrString(ptr %35, ptr @2) %35 = call i32 @PyList_SetItem(ptr %7, i64 11, ptr %34)
%37 = call ptr @PyList_New(i64 3) %36 = ptrtoint ptr %0 to i64
%38 = load ptr, ptr @__llgo_py.builtins.abs, align 8 %37 = call ptr @PyLong_FromUnsignedLongLong(i64 %36)
%39 = call i32 @PyList_SetItem(ptr %37, i64 0, ptr %38) %38 = call i32 @PyList_SetItem(ptr %7, i64 12, ptr %37)
%40 = load ptr, ptr @__llgo_py.builtins.print, align 8 %39 = ptrtoint ptr %0 to i64
%41 = call i32 @PyList_SetItem(ptr %37, i64 1, ptr %40) %40 = call ptr @PyLong_FromUnsignedLongLong(i64 %39)
%42 = call i32 @PyList_SetItem(ptr %37, i64 2, ptr %36) %41 = call i32 @PyList_SetItem(ptr %7, i64 13, ptr %40)
%43 = load ptr, ptr @__llgo_py.builtins.print, align 8 %42 = load ptr, ptr @__llgo_py.math, align 8
%44 = call ptr (ptr, ...) @PyObject_CallFunctionObjArgs(ptr %43, ptr %6, ptr %37, ptr null) %43 = call ptr @PyObject_GetAttrString(ptr %42, ptr @2)
%44 = call ptr @PyList_New(i64 3)
%45 = load ptr, ptr @__llgo_py.builtins.abs, align 8
%46 = call i32 @PyList_SetItem(ptr %44, i64 0, ptr %45)
%47 = load ptr, ptr @__llgo_py.builtins.print, align 8
%48 = call i32 @PyList_SetItem(ptr %44, i64 1, ptr %47)
%49 = call i32 @PyList_SetItem(ptr %44, i64 2, ptr %43)
%50 = load ptr, ptr @__llgo_py.builtins.print, align 8
%51 = call ptr (ptr, ...) @PyObject_CallFunctionObjArgs(ptr %50, ptr %7, ptr %44, ptr null)
ret void ret void
} }
@@ -92,6 +100,8 @@ declare void @"github.com/goplus/lib/py/math.init"()
declare void @"github.com/goplus/lib/py/std.init"() declare void @"github.com/goplus/lib/py/std.init"()
declare ptr @"github.com/goplus/llgo/runtime/internal/runtime.AllocZ"(i64)
declare %"github.com/goplus/llgo/runtime/internal/runtime.Slice" @"github.com/goplus/llgo/runtime/internal/runtime.StringToBytes"(%"github.com/goplus/llgo/runtime/internal/runtime.String") declare %"github.com/goplus/llgo/runtime/internal/runtime.Slice" @"github.com/goplus/llgo/runtime/internal/runtime.StringToBytes"(%"github.com/goplus/llgo/runtime/internal/runtime.String")
; Function Attrs: nocallback nofree nounwind willreturn memory(argmem: write) ; Function Attrs: nocallback nofree nounwind willreturn memory(argmem: write)

5
ssa/python.go
View File

@@ -455,6 +455,9 @@ func (b Builder) PyVal(v Expr) (ret Expr) {
return b.PyComplex64(v) return b.PyComplex64(v)
case types.Complex128: case types.Complex128:
return b.PyComplex128(v) return b.PyComplex128(v)
case types.UnsafePointer:
typ := b.Prog.Uint64()
return b.PyUint64(Expr{llvm.CreatePtrToInt(b.impl, v.impl, typ.ll), typ})
} }
case *types.Slice: case *types.Slice:
if elem, ok := t.Elem().Underlying().(*types.Basic); ok && elem.Kind() == types.Byte { if elem, ok := t.Elem().Underlying().(*types.Basic); ok && elem.Kind() == types.Byte {
@@ -471,6 +474,8 @@ func (b Builder) PyVal(v Expr) (ret Expr) {
if v.kind == vkPyFuncRef { if v.kind == vkPyFuncRef {
return b.Load(v) return b.Load(v)
} }
typ := b.Prog.Uint64()
return b.PyUint64(Expr{llvm.CreatePtrToInt(b.impl, v.impl, typ.ll), typ})
} }
panic("PyVal: todo " + v.raw.Type.String()) panic("PyVal: todo " + v.raw.Type.String())
} }