public struct Memo: CustomDebugStringConvertible, CustomStringConvertible {
public init(unevaluted: () -> A) {
self.init(.Unevaluated(unevaluted))
}
public init(evaluated: A) {
self.init(.Evaluated(evaluated))
}
private init(_ thunk: Thunk) {
_value = MutableBox(thunk)
}
public var value: A {
return _value.value.value()
}
private var _value: MutableBox>
public func map(transform: A -> B) -> Memo {
return Memo { transform(self.value) }
}
public func flatMap(transform: A -> Memo) -> Memo {
return Memo { transform(self.value).value }
}
public var description: String {
return String(value)
}
public var debugDescription: String {
return String(reflecting: value)
}
}
private final class MutableBox {
init(_ value: A) {
self.value = value
}
var value: A
}
private enum Thunk {
case Evaluated(A)
case Unevaluated(() -> A)
mutating func value() -> A {
switch self {
case let .Evaluated(a):
return a
case let .Unevaluated(f):
let a = f()
self = .Evaluated(a)
return a
}
}
}