From 54dfd3c701eb3850540dc39ceeae69b18891e597 Mon Sep 17 00:00:00 2001 From: visualfc Date: Wed, 30 Jul 2025 15:42:34 +0800 Subject: [PATCH] ssa: PyVal support pointer to long --- cl/_testpy/list/in.go | 5 +- cl/_testpy/list/out.ll | 112 ++++++++++++++++++++++------------------- ssa/python.go | 5 ++ 3 files changed, 70 insertions(+), 52 deletions(-) diff --git a/cl/_testpy/list/in.go b/cl/_testpy/list/in.go index f963d0a5..3cbc8604 100644 --- a/cl/_testpy/list/in.go +++ b/cl/_testpy/list/in.go @@ -1,14 +1,17 @@ package main import ( + "unsafe" + "github.com/goplus/lib/py" "github.com/goplus/lib/py/math" "github.com/goplus/lib/py/std" ) func main() { + v := 100 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) std.Print(x, y) } diff --git a/cl/_testpy/list/out.ll b/cl/_testpy/list/out.ll index 1552fc5e..2841773a 100644 --- a/cl/_testpy/list/out.ll +++ b/cl/_testpy/list/out.ll @@ -34,57 +34,65 @@ _llgo_2: ; preds = %_llgo_1, %_llgo_0 define void @"github.com/goplus/llgo/cl/_testpy/list.main"() { _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 }) - %1 = alloca [3 x i8], align 1 - call void @llvm.memset(ptr %1, i8 0, i64 3, i1 false) - %2 = getelementptr inbounds i8, ptr %1, i64 0 - %3 = getelementptr inbounds i8, ptr %1, i64 1 - %4 = getelementptr inbounds i8, ptr %1, i64 2 - store i8 1, ptr %2, align 1 - store i8 2, ptr %3, align 1 - store i8 3, ptr %4, align 1 - %5 = load [3 x i8], ptr %1, align 1 - %6 = call ptr @PyList_New(i64 12) - %7 = call ptr @PyBool_FromLong(i32 -1) - %8 = call i32 @PyList_SetItem(ptr %6, i64 0, ptr %7) - %9 = call ptr @PyBool_FromLong(i32 0) - %10 = call i32 @PyList_SetItem(ptr %6, i64 1, ptr %9) - %11 = call ptr @PyLong_FromLongLong(i64 1) - %12 = call i32 @PyList_SetItem(ptr %6, i64 2, ptr %11) - %13 = call ptr @PyFloat_FromDouble(double 0x4000CCCCC0000000) - %14 = call i32 @PyList_SetItem(ptr %6, i64 3, ptr %13) - %15 = call ptr @PyFloat_FromDouble(double 3.100000e+00) - %16 = call i32 @PyList_SetItem(ptr %6, i64 4, ptr %15) - %17 = call ptr @PyLong_FromUnsignedLongLong(i64 4) - %18 = call i32 @PyList_SetItem(ptr %6, i64 5, ptr %17) - %19 = call ptr @PyComplex_FromDoubles(double 1.000000e+00, double 2.000000e+00) - %20 = call i32 @PyList_SetItem(ptr %6, i64 6, ptr %19) - %21 = call ptr @PyComplex_FromDoubles(double 3.000000e+00, double 4.000000e+00) - %22 = call i32 @PyList_SetItem(ptr %6, i64 7, ptr %21) - %23 = call ptr @PyUnicode_FromStringAndSize(ptr @1, i64 5) - %24 = call i32 @PyList_SetItem(ptr %6, i64 8, ptr %23) - %25 = extractvalue %"github.com/goplus/llgo/runtime/internal/runtime.Slice" %0, 0 - %26 = extractvalue %"github.com/goplus/llgo/runtime/internal/runtime.Slice" %0, 1 - %27 = call ptr @PyByteArray_FromStringAndSize(ptr %25, i64 %26) - %28 = call i32 @PyList_SetItem(ptr %6, i64 9, ptr %27) - %29 = alloca [3 x i8], align 1 - call void @llvm.memset(ptr %29, i8 0, i64 3, i1 false) - store [3 x i8] %5, ptr %29, align 1 - %30 = getelementptr inbounds ptr, ptr %29, i64 0 - %31 = call ptr @PyBytes_FromStringAndSize(ptr %30, i64 3) - %32 = call i32 @PyList_SetItem(ptr %6, i64 10, ptr %31) - %33 = call ptr @PyBytes_FromStringAndSize(ptr null, i64 0) - %34 = call i32 @PyList_SetItem(ptr %6, i64 11, ptr %33) - %35 = load ptr, ptr @__llgo_py.math, align 8 - %36 = call ptr @PyObject_GetAttrString(ptr %35, ptr @2) - %37 = call ptr @PyList_New(i64 3) - %38 = load ptr, ptr @__llgo_py.builtins.abs, align 8 - %39 = call i32 @PyList_SetItem(ptr %37, i64 0, ptr %38) - %40 = load ptr, ptr @__llgo_py.builtins.print, align 8 - %41 = call i32 @PyList_SetItem(ptr %37, i64 1, ptr %40) - %42 = call i32 @PyList_SetItem(ptr %37, i64 2, ptr %36) - %43 = load ptr, ptr @__llgo_py.builtins.print, align 8 - %44 = call ptr (ptr, ...) @PyObject_CallFunctionObjArgs(ptr %43, ptr %6, ptr %37, ptr null) + %0 = call ptr @"github.com/goplus/llgo/runtime/internal/runtime.AllocZ"(i64 8) + store i64 100, ptr %0, align 4 + %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 = alloca [3 x i8], align 1 + call void @llvm.memset(ptr %2, i8 0, i64 3, i1 false) + %3 = getelementptr inbounds i8, ptr %2, i64 0 + %4 = getelementptr inbounds i8, ptr %2, i64 1 + %5 = getelementptr inbounds i8, ptr %2, i64 2 + store i8 1, ptr %3, align 1 + store i8 2, ptr %4, align 1 + store i8 3, ptr %5, align 1 + %6 = load [3 x i8], ptr %2, align 1 + %7 = call ptr @PyList_New(i64 14) + %8 = call ptr @PyBool_FromLong(i32 -1) + %9 = call i32 @PyList_SetItem(ptr %7, i64 0, ptr %8) + %10 = call ptr @PyBool_FromLong(i32 0) + %11 = call i32 @PyList_SetItem(ptr %7, i64 1, ptr %10) + %12 = call ptr @PyLong_FromLongLong(i64 1) + %13 = call i32 @PyList_SetItem(ptr %7, i64 2, ptr %12) + %14 = call ptr @PyFloat_FromDouble(double 0x4000CCCCC0000000) + %15 = call i32 @PyList_SetItem(ptr %7, i64 3, ptr %14) + %16 = call ptr @PyFloat_FromDouble(double 3.100000e+00) + %17 = call i32 @PyList_SetItem(ptr %7, i64 4, ptr %16) + %18 = call ptr @PyLong_FromUnsignedLongLong(i64 4) + %19 = call i32 @PyList_SetItem(ptr %7, i64 5, ptr %18) + %20 = call ptr @PyComplex_FromDoubles(double 1.000000e+00, double 2.000000e+00) + %21 = call i32 @PyList_SetItem(ptr %7, i64 6, ptr %20) + %22 = call ptr @PyComplex_FromDoubles(double 3.000000e+00, double 4.000000e+00) + %23 = call i32 @PyList_SetItem(ptr %7, i64 7, ptr %22) + %24 = call ptr @PyUnicode_FromStringAndSize(ptr @1, i64 5) + %25 = call i32 @PyList_SetItem(ptr %7, i64 8, ptr %24) + %26 = extractvalue %"github.com/goplus/llgo/runtime/internal/runtime.Slice" %1, 0 + %27 = extractvalue %"github.com/goplus/llgo/runtime/internal/runtime.Slice" %1, 1 + %28 = call ptr @PyByteArray_FromStringAndSize(ptr %26, i64 %27) + %29 = call i32 @PyList_SetItem(ptr %7, i64 9, ptr %28) + %30 = alloca [3 x i8], align 1 + call void @llvm.memset(ptr %30, i8 0, i64 3, i1 false) + store [3 x i8] %6, ptr %30, align 1 + %31 = getelementptr inbounds ptr, ptr %30, i64 0 + %32 = call ptr @PyBytes_FromStringAndSize(ptr %31, i64 3) + %33 = call i32 @PyList_SetItem(ptr %7, i64 10, ptr %32) + %34 = call ptr @PyBytes_FromStringAndSize(ptr null, i64 0) + %35 = call i32 @PyList_SetItem(ptr %7, i64 11, ptr %34) + %36 = ptrtoint ptr %0 to i64 + %37 = call ptr @PyLong_FromUnsignedLongLong(i64 %36) + %38 = call i32 @PyList_SetItem(ptr %7, i64 12, ptr %37) + %39 = ptrtoint ptr %0 to i64 + %40 = call ptr @PyLong_FromUnsignedLongLong(i64 %39) + %41 = call i32 @PyList_SetItem(ptr %7, i64 13, ptr %40) + %42 = load ptr, ptr @__llgo_py.math, align 8 + %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 } @@ -92,6 +100,8 @@ declare void @"github.com/goplus/lib/py/math.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") ; Function Attrs: nocallback nofree nounwind willreturn memory(argmem: write) diff --git a/ssa/python.go b/ssa/python.go index a0e29bca..e6718181 100644 --- a/ssa/python.go +++ b/ssa/python.go @@ -455,6 +455,9 @@ func (b Builder) PyVal(v Expr) (ret Expr) { return b.PyComplex64(v) case types.Complex128: 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: 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 { 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()) }