init

杨锴

2024-08-14 909e20941e45f8712c012db602034b47da0bfdb0

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
//
//  CryptoSwift
//
//  Copyright (C) 2014-2022 Marcin Krzyżanowski <marcin@krzyzanowskim.com>
//  This software is provided 'as-is', without any express or implied warranty.
//
//  In no event will the authors be held liable for any damages arising from the use of this software.
//
//  Permission is granted to anyone to use this software for any purpose,including commercial applications, and to alter it and redistribute it freely, subject to the following restrictions:
//
//  - The origin of this software must not be misrepresented; you must not claim that you wrote the original software. If you use this software in a product, an acknowledgment in the product documentation is required.
//  - Altered source versions must be plainly marked as such, and must not be misrepresented as being the original software.
//  - This notice may not be removed or altered from any source or binary distribution.
//
 
@usableFromInline
struct BatchedCollectionIndex<Base: Collection> {
  let range: Range<Base.Index>
}
 
extension BatchedCollectionIndex: Comparable {
  @usableFromInline
  static func == <BaseCollection>(lhs: BatchedCollectionIndex<BaseCollection>, rhs: BatchedCollectionIndex<BaseCollection>) -> Bool {
    lhs.range.lowerBound == rhs.range.lowerBound
  }
 
  @usableFromInline
  static func < <BaseCollection>(lhs: BatchedCollectionIndex<BaseCollection>, rhs: BatchedCollectionIndex<BaseCollection>) -> Bool {
    lhs.range.lowerBound < rhs.range.lowerBound
  }
}
 
protocol BatchedCollectionType: Collection {
  associatedtype Base: Collection
}
 
@usableFromInline
struct BatchedCollection<Base: Collection>: Collection {
  let base: Base
  let size: Int
 
  @usableFromInline
  init(base: Base, size: Int) {
    self.base = base
    self.size = size
  }
 
  @usableFromInline
  typealias Index = BatchedCollectionIndex<Base>
 
  private func nextBreak(after idx: Base.Index) -> Base.Index {
    self.base.index(idx, offsetBy: self.size, limitedBy: self.base.endIndex) ?? self.base.endIndex
  }
 
  @usableFromInline
  var startIndex: Index {
    Index(range: self.base.startIndex..<self.nextBreak(after: self.base.startIndex))
  }
 
  @usableFromInline
  var endIndex: Index {
    Index(range: self.base.endIndex..<self.base.endIndex)
  }
 
  @usableFromInline
  func index(after idx: Index) -> Index {
    Index(range: idx.range.upperBound..<self.nextBreak(after: idx.range.upperBound))
  }
 
  @usableFromInline
  subscript(idx: Index) -> Base.SubSequence {
    self.base[idx.range]
  }
}
 
extension Collection {
  @inlinable
  func batched(by size: Int) -> BatchedCollection<Self> {
    BatchedCollection(base: self, size: size)
  }
}