2014-04-17 25 views
4

为什么在F#中,我能做到这一点...F#:低迷现状,以次来的IEnumerator

let s = seq { for i in 0 .. 4095 do yield i } :?> IEnumerator 

...但是这将引发System.InvalidCastException

let s = Seq.init 4095 (fun i -> i) :?> IEnumerator 
+1

你确定你不想IEnumerable而不是IEnumerator? – ildjarn

+0

我正在为Unity编写协程。必须是IEnumerator。 – MiloDC

回答

6

序列表达式创建了一个实现IEnumerable<T>IEnumerator<T>

let s = seq { for i in 0 .. 4095 do yield i } 
printfn "%b" (s :? IEnumerable<int>) // true 
printfn "%b" (s :? IEnumerator<int>) // true 

Seq.init对象不:

let s = Seq.init 4095 (fun i -> i) 
printfn "%b" (s :? IEnumerable<int>) // true 
printfn "%b" (s :? IEnumerator<int>) // false 

您可以重构代码以使用IEnumerable<T>而不是IEnumerator因为两个构造产生一个IEnumerable<T>

或者,如果你真的想要一个IEnumerator,你可以简单地调用GetEnumerator返回一个EnumeratorEnumerable

let s = (Seq.init 4095 (fun i -> i)).GetEnumerator() 
printfn "%b" (s :? IEnumerable<int>) // false 
printfn "%b" (s :? IEnumerator<int>) // true 
+0

当,偷走了我的答案。好吧,其中的一部分......不知道序列表达式与Seq模块的对应表达式并不完全相同。 – Jwosty

5

如果你看一下the specification,你序列表达式转换为:

Seq.collect (fun pat -> Seq.singleton(pat)) (0 .. 4095) 

如果你看看Seq.collect的定义来源,它是:

let collect f sources = map f sources |> concat 

,如果你看一下定义concat是:

let concat sources = 
      checkNonNull "sources" sources 
      mkConcatSeq sources 

mkConcatSeq被定义为:

let mkConcatSeq (sources: seq<'U :> seq<'T>>) = 
      mkSeq (fun() -> new ConcatEnumerator<_,_>(sources) :> IEnumerator<'T>) 

所以你可以看到,返回的序列实现IEnumerator<'T>,因此IEnumerator

let init count f = 
      if count < 0 then invalidArg "count" (SR.GetString(SR.inputMustBeNonNegative)) 
      mkSeq (fun() -> IEnumerator.upto (Some (count-1)) f) 

mkSeq定义为:

let mkSeq f = 
      { new IEnumerable<'U> with 
       member x.GetEnumerator() = f() 
       interface IEnumerable with 
       member x.GetEnumerator() = (f() :> IEnumerator) } 

所以它仅实现IEnumerable<'T>,而不是IEnumerator

因为现在Seq.init定义。