py.Str; py.FromCStr/FromCStrAndLen/FromGoString

README.md

@@ -73,24 +73,33 @@ And you can import any Python library into `llgo` through a program called `llpy
 
 Note: For third-party libraries (such as pandas and pytorch), you still need to install the library files.
 
-Here is an example using the Python `math` library:
+Here is an example:
 
 ```go
 package main
 
 import (
-	"github.com/goplus/llgo/c"
 	"github.com/goplus/llgo/py"
 	"github.com/goplus/llgo/py/math"
+	"github.com/goplus/llgo/py/std"
 )
 
 func main() {
-	x := math.Sqrt(py.Float(2))
+	x := math.Sqrt(py.Float(2))       // x = sqrt(2)
-	c.Printf(c.Str("sqrt(2) = %f\n"), x.Float64())
+	std.Print(py.Str("sqrt(2) ="), x) // print("sqrt(2) =", x)
 }
 ```
 
-Here, We call `py.Float(2)` to create a Python number 2, and pass it to Python’s `math.sqrt` to get `x`. Then use `x.Float64()` to convert x to Go's `float64` type, and print the value through the C `printf` function.
+It is equivalent to the following Python code:
+
+```py
+import math
+
+x = math.sqrt(2)
+print("sqrt =", x)
+```
+
+Here, We call `py.Float(2)` to create a Python number 2, and pass it to Python’s `math.sqrt` to get `x`. Then we call `std.Print` to print the result.
 
 Let's look at a slightly more complex example. For example, we use `numpy` to calculate:
 
@@ -98,9 +107,9 @@ Let's look at a slightly more complex example. For example, we use `numpy` to ca
 package main
 
 import (
-	"github.com/goplus/llgo/c"
 	"github.com/goplus/llgo/py"
 	"github.com/goplus/llgo/py/numpy"
+	"github.com/goplus/llgo/py/std"
 )
 
 func main() {
@@ -115,7 +124,7 @@ func main() {
 		py.List(3.0, 2.0, 1.0),
 	)
 	x := numpy.Add(a, b)
-	c.Printf(c.Str("a+b = %s\n"), x.Str().CStr())
+	std.Print(py.Str("a+b ="), x)
 }
 ```
 
@@ -246,9 +255,9 @@ Here are the Go packages that can be imported correctly:
 - [Clang 17](https://clang.llvm.org)
 - [pkg-config 0.29+](https://www.freedesktop.org/wiki/Software/pkg-config/)
 - [bdwgc/libgc 8.0+](https://www.hboehm.info/gc/)
-- [cJSON 1.7+](https://github.com/DaveGamble/cJSON) (optional, for [`github.com/goplus/llgo/c/cjson`](https://pkg.go.dev/github.com/goplus/llgo/c/cjson))
+- [cJSON 1.7+](https://github.com/DaveGamble/cJSON) (optional, for [github.com/goplus/llgo/c/cjson](https://pkg.go.dev/github.com/goplus/llgo/c/cjson))
-- [SQLite 3](https://www.sqlite.org) (optional, for [`github.com/goplus/llgo/c/sqlite`](https://pkg.go.dev/github.com/goplus/llgo/c/sqlite))
+- [SQLite 3](https://www.sqlite.org) (optional, for [github.com/goplus/llgo/c/sqlite](https://pkg.go.dev/github.com/goplus/llgo/c/sqlite))
-- [Python 3.11+](https://www.python.org) (optional, for [`github.com/goplus/llgo/py`](https://pkg.go.dev/github.com/goplus/llgo/py))
+- [Python 3.11+](https://www.python.org) (optional, for [github.com/goplus/llgo/py](https://pkg.go.dev/github.com/goplus/llgo/py))
 
 ## How to install
 

_pydemo/callpy/callpy.go

@@ -1,12 +1,12 @@
 package main
 
 import (
-	"github.com/goplus/llgo/c"
 	"github.com/goplus/llgo/py"
 	"github.com/goplus/llgo/py/math"
+	"github.com/goplus/llgo/py/std"
 )
 
 func main() {
-	x := math.Sqrt(py.Float(2))
+	x := math.Sqrt(py.Float(2))       // x = sqrt(2)
-	c.Printf(c.Str("sqrt(2) = %f\n"), x.Float64())
+	std.Print(py.Str("sqrt(2) ="), x) // print("sqrt(2) =", x)
 }

_pydemo/matrix/matrix.go

@@ -1,9 +1,9 @@
 package main
 
 import (
-	"github.com/goplus/llgo/c"
 	"github.com/goplus/llgo/py"
 	"github.com/goplus/llgo/py/numpy"
+	"github.com/goplus/llgo/py/std"
 )
 
 func main() {
@@ -18,5 +18,5 @@ func main() {
 		py.List(3.0, 2.0, 1.0),
 	)
 	x := numpy.Add(a, b)
-	c.Printf(c.Str("a+b = %s\n"), x.Str().CStr())
+	std.Print(py.Str("a+b ="), x)
 }

cl/builtin_test.go

@@ -144,6 +144,8 @@ func TestErrBuiltin(t *testing.T) {
 	test("siglongjmp", func(ctx *context) { ctx.siglongjmp(nil, nil) })
 	test("cstr(NoArgs)", func(ctx *context) { cstr(nil, nil) })
 	test("cstr(Nonconst)", func(ctx *context) { cstr(nil, []ssa.Value{&ssa.Parameter{}}) })
+	test("pystr(NoArgs)", func(ctx *context) { pystr(nil, nil) })
+	test("pystr(Nonconst)", func(ctx *context) { pystr(nil, []ssa.Value{&ssa.Parameter{}}) })
 	test("atomic", func(ctx *context) { ctx.atomic(nil, 0, nil) })
 	test("atomicLoad", func(ctx *context) { ctx.atomicLoad(nil, nil) })
 	test("atomicStore", func(ctx *context) { ctx.atomicStore(nil, nil) })

cl/import.go

@@ -391,6 +391,7 @@ const (
 	llgoSiglongjmp = llgoInstrBase + 0xc
 
 	llgoPyList = llgoInstrBase + 0x10
+	llgoPyStr  = llgoInstrBase + 0x11
 
 	llgoAtomicLoad    = llgoInstrBase + 0x1d
 	llgoAtomicStore   = llgoInstrBase + 0x1e

cl/instr.go

@@ -28,6 +28,19 @@ import (
 
 // -----------------------------------------------------------------------------
 
+// func pystr(string) *py.Object
+func pystr(b llssa.Builder, args []ssa.Value) (ret llssa.Expr) {
+	if len(args) == 1 {
+		if c, ok := args[0].(*ssa.Const); ok {
+			if v := c.Value; v.Kind() == constant.String {
+				sv := constant.StringVal(v)
+				return b.PyStr(sv)
+			}
+		}
+	}
+	panic("pystr(<string-literal>): invalid arguments")
+}
+
 // func cstr(string) *int8
 func cstr(b llssa.Builder, args []ssa.Value) (ret llssa.Expr) {
 	if len(args) == 1 {
@@ -175,6 +188,7 @@ var llgoInstrs = map[string]int{
 	"string":      llgoString,
 	"stringData":  llgoStringData,
 	"funcAddr":    llgoFuncAddr,
+	"pystr":       llgoPyStr,
 	"pyList":      llgoPyList,
 	"sigjmpbuf":   llgoSigjmpbuf,
 	"sigsetjmp":   llgoSigsetjmp,
@@ -314,6 +328,8 @@ func (p *context) call(b llssa.Builder, act llssa.DoAction, call *ssa.CallCommon
 		case llgoPyList:
 			args := p.compileValues(b, args, fnHasVArg)
 			ret = b.PyList(args...)
+		case llgoPyStr:
+			ret = pystr(b, args)
 		case llgoCstr:
 			ret = cstr(b, args)
 		case llgoAdvance:

py/func.go

@@ -46,7 +46,7 @@ func NewFuncWithQualName(code, globals, qualname *Object) *Object
 // Return true if o is a function object (has type PyFunction_Type). The
 // parameter must not be nil. This function always succeeds.
 //
-// llgo:link (*Object).FuncCheck C.PyFunction_Check
+//- llgo:link (*Object).FuncCheck C.PyFunction_Check
 func (o *Object) FuncCheck() c.Int { return 0 }
 */
 

py/type.go

@@ -49,5 +49,5 @@ func (t *Object) TypeFlags() uint32 { return 0 }
 // llgo:link (*Object).TypeModule C.PyType_GetModule
 func (t *Object) TypeModule() *Object { return nil }
 
-// llgo:link (*Object).TypeModuleByDef C.PyType_GetModuleByDef
+// -llgo:link (*Object).TypeModuleByDef C.PyType_GetModuleByDef
 // func (t *Object) TypeModuleByDef(def *ModuleDef) *Object { return nil }

py/unicode.go

@@ -24,6 +24,20 @@ import (
 
 // https://docs.python.org/3/c-api/unicode.html
 
+//go:linkname Str llgo.pystr
+func Str(s string) *Object
+
+//go:linkname FromCStr C.PyUnicode_FromString
+func FromCStr(str *c.Char) *Object
+
+//go:linkname FromCStrAndLen C.PyUnicode_FromStringAndSize
+func FromCStrAndLen(str *c.Char, n int) *Object
+
+// FromGoString returns a new Unicode object from a Go string.
+func FromGoString(s string) *Object {
+	return FromCStrAndLen(c.GoStringData(s), len(s))
+}
+
 // Return a pointer to the UTF-8 encoding of the Unicode object, and store the
 // size of the encoded representation (in bytes) in size. The size argument can
 // be nil; in this case no size will be stored. The returned buffer always has

ssa/package.go

@@ -174,6 +174,7 @@ type aProgram struct {
 	callFOArgs   *types.Signature
 	loadPyModS   *types.Signature
 	getAttrStr   *types.Signature
+	pyUniStr     *types.Signature
 
 	mallocTy *types.Signature
 	freeTy   *types.Signature

ssa/python.go

@@ -184,6 +184,17 @@ func (p Program) tyLoadPyModSyms() *types.Signature {
 	return p.loadPyModS
 }
 
+// func(*char) *Object
+func (p Program) tyPyUnicodeFromString() *types.Signature {
+	if p.pyUniStr == nil {
+		charPtr := types.NewPointer(types.Typ[types.Int8])
+		params := types.NewTuple(types.NewParam(token.NoPos, nil, "", charPtr))
+		results := types.NewTuple(p.paramObjPtr())
+		p.pyUniStr = types.NewSignatureType(nil, nil, nil, params, results, false)
+	}
+	return p.pyUniStr
+}
+
 // func(*Objecg, *char) *Object
 func (p Program) tyGetAttrString() *types.Signature {
 	if p.getAttrStr == nil {
@@ -332,6 +343,12 @@ func (b Builder) callPyInit() (ret Expr) {
 	return b.Call(fn)
 }
 
+// PyStr returns a py-style string constant expression.
+func (b Builder) PyStr(v string) Expr {
+	fn := b.Pkg.pyFunc("PyUnicode_FromString", b.Prog.tyPyUnicodeFromString())
+	return b.Call(fn, b.CStr(v))
+}
+
 // -----------------------------------------------------------------------------
 
 type aPyGlobal struct {