swift-mirror/stdlib/core/Existential.swift

//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2015 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See http://swift.org/LICENSE.txt for license information
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

//////////////////////////////////////////
// FIXME: Workaround for inability to create existentials of protocols
// with associated types <rdar://problem/11689181>

// This file contains "existentials" for the protocols defined in
// Policy.swift.  Similar components should usually be defined next to
// their respective protocols.
@public struct GeneratorOf<T> : Generator, Sequence {
  @public init(_ next: ()->T?) {
    self._next = next
  }
  
  @public init<G: Generator where G.Element == T>(var _ self_: G) {
    self._next = { self_.next() }
  }
  
  @public mutating func next() -> T? {
    return _next()
  }

  @public func generate() -> GeneratorOf {
    return self
  }
  let _next: ()->T?
}

@public struct SequenceOf<T> : Sequence {
  @public init<G: Generator where G.Element == T>(_ generate: ()->G) {
    _generate = { GeneratorOf(generate()) }
  }
  @public init<S: Sequence where S.GeneratorType.Element == T>(_ self_: S) {
    self = SequenceOf({ self_.generate() })
  }

  @public func generate() -> GeneratorOf<T> {
    return _generate()
  }
  
  let _generate: ()->GeneratorOf<T>
}

@internal struct _CollectionOf<IndexType_ : ForwardIndex, T> : Collection {
  init(startIndex: IndexType_, endIndex: IndexType_,
      _ subscriptImpl: (IndexType_)->T) {
    self.startIndex = startIndex
    self.endIndex = endIndex
    _subscriptImpl = subscriptImpl
  }

  func generate() -> GeneratorOf<T> {
    var index = startIndex
    return GeneratorOf {
      () -> T? in
      if _fastPath(index != self.endIndex) {
        ++index
        return self._subscriptImpl(index)
      }
      return .None
    }
  }

  let startIndex: IndexType_
  let endIndex: IndexType_

  subscript(i: IndexType_) -> T {
    return _subscriptImpl(i)
  }

  let _subscriptImpl: (IndexType_)->T
}

@public struct SinkOf<T> : Sink {
  @public init(_ put: (T)->()) {
    _put = put
  }

  @public init<S: Sink where S.Element == T>(var _ base: S) {
    _put = { base.put($0) }
  }
  @public func put(x: T) {
    _put(x)
  }
  let _put: (T)->()
}