File: Vector.swift

package info (click to toggle)
swiftlang 6.1.3-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 2,791,644 kB
  • sloc: cpp: 9,901,738; ansic: 2,201,433; asm: 1,091,827; python: 308,252; objc: 82,166; f90: 80,126; lisp: 38,358; pascal: 25,559; sh: 20,429; ml: 5,058; perl: 4,745; makefile: 4,484; awk: 3,535; javascript: 3,018; xml: 918; fortran: 664; cs: 573; ruby: 396
file content (110 lines) | stat: -rw-r--r-- 3,259 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
// RUN: %target-run-simple-swift(-target %module-target-future -enable-experimental-feature ValueGenerics -enable-experimental-feature BuiltinModule -Xfrontend -disable-experimental-parser-round-trip) | %FileCheck %s
// RUN: %target-swift-frontend -target %module-target-future -enable-experimental-feature ValueGenerics -enable-experimental-feature BuiltinModule -disable-experimental-parser-round-trip -emit-module %s -o %t/Vector.swiftmodule

// REQUIRES: executable_test
// REQUIRES: swift_feature_BuiltinModule
// REQUIRES: swift_feature_ValueGenerics
// UNSUPPORTED: use_os_stdlib
// UNSUPPORTED: back_deployment_runtime

// FIXME: miscompile/verifier error after optimization

import Builtin

@frozen
public struct Vector<let Count: Int, Element: ~Copyable>: ~Copyable {
    private var storage: Builtin.FixedArray<Count, Element>

    public init(_ valueForIndex: (Int) -> Element) {
        storage = Builtin.emplace { rawPointer in
            let base = UnsafeMutablePointer<Element>(rawPointer)
            for i in 0..<Count {
                (base + i).initialize(to: valueForIndex(i))
            }
        }
    }

    public subscript(i: Int) -> Element {
        _read {
            assert(i >= 0 && i < Count)
            let rawPointer = Builtin.addressOfBorrow(self)
            let base = UnsafePointer<Element>(rawPointer)
            yield ((base + i).pointee)
        }

        _modify {
            assert(i >= 0 && i < Count)
            let rawPointer = Builtin.addressof(&self)
            let base = UnsafeMutablePointer<Element>(rawPointer)
            yield (&(base + i).pointee)
        }
    }
}
extension Vector: Copyable where Element: Copyable {
    public init(repeating value: Element) {
        self.init { _ in value }
    }
}
extension Vector: BitwiseCopyable where Element: BitwiseCopyable {}

extension Vector: CustomDebugStringConvertible {
    public var debugDescription: String {
        var result = "["
        if Count > 0 {
            result += String(reflecting: self[0])

            for i in 1..<Count {
                result += ", " + String(reflecting: self[i])
            }
        }
        result += "]"
        return result
    }
}

class Canary: CustomDebugStringConvertible {
    var value: String

    init(value: String) { self.value = value }

    var debugDescription: String {
        "<\(self.value)>"
    }

    deinit { print("deinit <\(self.value)>") }
}

func main() {
    var v = Vector<4, Canary>(repeating: Canary(value: ""))

    // CHECK: [<>, <>, <>, <>]
    print(v)

    v[0] = Canary(value: "hello")
    v[1] = Canary(value: "world")
    v[2] = Canary(value: "goodbye")
    v[3] = Canary(value: "moon")
    // CHECK-NEXT: deinit <>

    // CHECK-NEXT: [<hello>, <world>, <goodbye>, <moon>]
    print(v)

    do {
        var w = v
        w[1] = Canary(value: "sun")
        w[3] = Canary(value: "earth")
        // CHECK-NEXT: [<hello>, <sun>, <goodbye>, <earth>]
        print(w)
    }
    // CHECK-NEXT: deinit <sun>
    // CHECK-NEXT: deinit <earth>

    // CHECK-NEXT: [<hello>, <world>, <goodbye>, <moon>]
    print(v)

    // CHECK-NEXT: deinit <hello>
    // CHECK-NEXT: deinit <world>
    // CHECK-NEXT: deinit <goodbye>
    // CHECK-NEXT: deinit <moon>
}
main()