mirror of
https://github.com/apple/swift.git
synced 2026-10-28 21:15:56 +01:00
Doug Gregor
e2eb66fedb
Reimplement swift_readAtKeyPath, swift_modifyAtWritableKeyPath, and swift_modifyAtReferenceWritableKeyPath in (Embedded) Swift. The existing ABI for non-Embedded Swift cannot be changed, and isn't directly implementable on the Swift side (where we don't have a way to write a thin yield-once coroutine function), so we diverge the Embedded Swift ABI slightly to implement the coroutines directly as subscripts.
64 lines
2.7 KiB
Swift
64 lines
2.7 KiB
Swift
// RUN: %target-swift-emit-sil %s -enable-experimental-feature Embedded -enable-experimental-feature EmbeddedKeyPaths -wmo -o - | %FileCheck -check-prefix=CHECK-SIL %s
|
|
// RUN: %target-swift-emit-ir %s -enable-experimental-feature Embedded -enable-experimental-feature EmbeddedKeyPaths -wmo -o - | %FileCheck -check-prefix=CHECK-IR %s
|
|
|
|
// REQUIRES: optimized_stdlib
|
|
// REQUIRES: swift_feature_Embedded
|
|
// REQUIRES: swift_feature_EmbeddedKeyPaths
|
|
|
|
public struct MyStruct {
|
|
var x: Int32
|
|
var y: Int32
|
|
|
|
var max: Int32 {
|
|
x > y ? x : y
|
|
}
|
|
}
|
|
|
|
@inline(never)
|
|
public func getValueIn<Root, Value>(_ root: Root, keyPath: KeyPath<Root, Value>) -> Value {
|
|
return root[keyPath: keyPath]
|
|
}
|
|
|
|
@inline(never)
|
|
public func getValueAsAnyIn<Root>(_ root: Root, keyPath: PartialKeyPath<Root>) -> Any {
|
|
return root[keyPath: keyPath]
|
|
}
|
|
|
|
@inline(never)
|
|
public func getValueAsAnyIn<Root>(anyRoot root: Root, keyPath: AnyKeyPath) -> Any? {
|
|
return root[keyPath: keyPath]
|
|
}
|
|
|
|
@inline(never)
|
|
public func updateValueIn<Root, Value>(_ root: inout Root, keyPath: WritableKeyPath<Root, Value>, body: (inout Value) -> Void) {
|
|
body(&root[keyPath: keyPath])
|
|
}
|
|
|
|
public func getKeyPaths(ms: MyStruct) {
|
|
_ = getValueIn(ms, keyPath: \.x)
|
|
_ = getValueAsAnyIn(ms, keyPath: \.x)
|
|
_ = getValueAsAnyIn(anyRoot: ms, keyPath: \MyStruct.x)
|
|
}
|
|
|
|
public func updateKeyPaths(ms: MyStruct) {
|
|
var ms = ms
|
|
|
|
updateValueIn(&ms, keyPath: \.x) { $0 += 1 }
|
|
}
|
|
|
|
// CHECK-IR-LABEL: define linkonce_odr hidden swiftcc i32 @"$e9keypaths510getValueIn_7keyPathq_x_s03KeyF0Cyxq_Gtr0_lFAA8MyStructV_s5Int32VTg5"
|
|
// CHECK-IR-NOT: ret void
|
|
// CHECK-IR: call swiftcc i32 @"$e18swift_getAtKeyPath9keypaths58MyStructV_s5Int32VTg5"
|
|
|
|
// CHECK-IR-LABEL: define linkonce_odr hidden swiftcc void @"$e9keypaths515getValueAsAnyIn_7keyPathypx_s010PartialKeyH0CyxGtlFAA8MyStructV_Tg5"
|
|
// CHECK-IR-NOT: ret void
|
|
// CHECK-IR: call swiftcc void @"$e25swift_getAtPartialKeyPath9keypaths58MyStructV_Tg5"
|
|
|
|
// CHECK-IR-LABEL: define linkonce_odr hidden swiftcc void @"$e9keypaths515getValueAsAnyIn7anyRoot7keyPathypSgx_s0e3KeyJ0CtlFAA8MyStructV_Tg5"
|
|
// CHECK-IR-NOT: ret void
|
|
// CHECK-IR: call swiftcc void @"$e21swift_getAtAnyKeyPath9keypaths58MyStructV_Tg5"
|
|
|
|
// CHECK-SIL-LABEL: sil shared [noinline] @$e9keypaths513updateValueIn_7keyPath4bodyyxz_s011WritableKeyF0Cyxq_Gyq_zXEtr0_lFAA8MyStructV_s5Int32VTg5 : $@convention(thin) (@inout MyStruct, @guaranteed WritableKeyPath<MyStruct, Int32>, @guaranteed @noescape @callee_guaranteed @substituted <τ_0_0> (@inout τ_0_0) -> () for <Int32>) -> () {
|
|
// CHECK-SIL-NOT: return
|
|
// CHECK-SIL: function_ref @$e29swift_modifyAtWritableKeyPath9keypaths58MyStructV_s5Int32VTg5 : $@yield_once @convention(method) (MyStruct, @guaranteed WritableKeyPath<MyStruct, Int32>) -> @yields @inout Int32
|