Files
swift-mirror/test/embedded/keypaths.swift
Kuba Mracek b642d771be [embedded] Compile-time (literal) KeyPaths for Embedded Swift
Enable KeyPath/AnyKeyPath/PartialKeyPath/WritableKeyPath in Embedded Swift, but
for compile-time use only:

- Add keypath optimizations into the mandatory optimizations pipeline
- Allow keypath optimizations to look through begin_borrow, to make them work
  even in OSSA.
- If a use of a KeyPath doesn't optimize away, diagnose in PerformanceDiagnostics
- Make UnsafePointer.pointer(to:) transparent to allow the keypath optimization
  to happen in the callers of UnsafePointer.pointer(to:).
2024-03-20 15:35:46 -07:00

53 lines
1.8 KiB
Swift

// RUN: %target-run-simple-swift( -enable-experimental-feature Embedded -wmo -Xfrontend -disable-access-control -runtime-compatibility-version none) | %FileCheck %s
// RUN: %target-run-simple-swift(-O -enable-experimental-feature Embedded -wmo -Xfrontend -disable-access-control -runtime-compatibility-version none) | %FileCheck %s
// REQUIRES: swift_in_compiler
// REQUIRES: executable_test
// REQUIRES: optimized_stdlib
// REQUIRES: OS=macosx || OS=linux-gnu
struct MyStruct {
var x: Int8
var y: Int16
var z: Int32
}
public func test1() -> Int {
return (\MyStruct.z)._storedInlineOffset!
}
print(test1() == 4 ? "OK!" : "FAIL") // CHECK: OK!
public func test2() -> Int {
return MemoryLayout<MyStruct>.offset(of: \MyStruct.z)!
}
print(test2() == 4 ? "OK!" : "FAIL") // CHECK: OK!
public func test3(input: UnsafePointer<MyStruct>) -> UnsafePointer<Int32> {
return input.pointer(to: \MyStruct.z)!
}
print(UInt(bitPattern: test3(input: UnsafePointer(bitPattern: 0x1000)!)) == 0x1004 ? "OK!" : "FAIL") // CHECK: OK!
@dynamicMemberLookup
struct StructWithDynamicMemberLookup<T>: ~Copyable {
private var value: UnsafeMutablePointer<T>
init(from t: T) { value = UnsafeMutablePointer<T>.allocate(capacity: 1); value.pointee = t }
deinit { value.deallocate() }
subscript<U>(dynamicMember keyPath: KeyPath<T, U>) -> UnsafeMutablePointer<U> {
@_transparent
get {
return UnsafeMutablePointer(mutating: self.value.pointer(to: keyPath).unsafelyUnwrapped)
}
}
}
public func test4(arg: borrowing StructWithDynamicMemberLookup<MyStruct>) -> Int {
return Int(arg.z.pointee) + Int(arg.y.pointee)
}
let str = StructWithDynamicMemberLookup<MyStruct>(from: MyStruct(x: 10, y: 20, z: 30))
print(str.z.pointee == 30 ? "OK!" : "FAIL") // CHECK: OK!