module Seq = type [< RequireQualifiedAccess >] SplitByOption = Exclude | IncludeInFirst | IncludeInSecond type [< RequireQualifiedAccess >] private SplitSubUnfoldState<'T> = | PostValue of 'T * SplitSubUnfoldState<'T> | Start of seqNo: int * start: int | Started of tryNext: (unit -> 'T option) * bingo: ('T -> unit) * finish: (unit -> unit) | Finish type [< RequireQualifiedAccess >] private SplitUnfoldState<'T> = { enumerator : System.Collections.Generic.IEnumerator<'T> mutable currentPos : int mutable isDone : bool mutable splitterO : 'T option mutable currentSeqO : 'T seq option } /// Straight scan through is efficient, reusing seqs causes rescan from beginning let splitBy f opt (input: 'a seq) = let getEnumerator() = input.GetEnumerator() let startingState() = { SplitUnfoldState.enumerator = getEnumerator() SplitUnfoldState.currentPos = 0 SplitUnfoldState.isDone = false SplitUnfoldState.splitterO = None SplitUnfoldState.currentSeqO = None }, 0 Seq.unfold(fun stateO -> let state, currentSeqNo = stateO |> Option.defaultWith startingState let tryNextMain() : 'a option= if state.isDone then None else if state.enumerator.MoveNext() then state.currentPos <- state.currentPos + 1 Some state.enumerator.Current else state.enumerator.Dispose() None let bingo v = state.splitterO <- Some v ; state.currentSeqO <- None let finish () = state.splitterO <- None ; state.currentSeqO <- None ; state.isDone <- true let tryNextSub start = let en = getEnumerator() for i in 0..start - 1 do (en.MoveNext() |> ignore) fun () -> if en.MoveNext() then Some en.Current else en.Dispose() None let subUnFold(tryNext: unit -> 'a option, bingo, finish) = match tryNext() with | None -> finish() None | Some v -> if f v then bingo v if opt = SplitByOption.IncludeInFirst then Some( v, SplitSubUnfoldState.Finish ) else None else Some( v, SplitSubUnfoldState.Started(tryNext, bingo, finish) ) //printfn "Unfold %A" currentSeqNo while state.currentSeqO |> Option.isSome do //printfn "skipping" subUnFold(tryNextMain, bingo , finish) |> ignore if state.isDone then None else if opt <> SplitByOption.IncludeInSecond then state.splitterO <- None let s0 = let start = SplitSubUnfoldState.Start(currentSeqNo, state.currentPos) match state.splitterO, opt with | Some v, SplitByOption.IncludeInSecond -> SplitSubUnfoldState.PostValue(v, start) | _ -> start |> Seq.unfold(function | SplitSubUnfoldState.PostValue(v, next) -> Some(v, next) | SplitSubUnfoldState.Finish -> None | SplitSubUnfoldState.Started(tryNext, bingo, finish) -> subUnFold(tryNext, bingo, finish) | SplitSubUnfoldState.Start(seqNo, myStart) -> //printfn "Starting %d at %d = %d" seqNo myStart state.currentPos if state.currentPos = myStart // && not state.isDone then subUnFold(tryNextMain , bingo , finish) else subUnFold(tryNextSub myStart, ignore, ignore) ) state.currentSeqO <- Some s0 Some (s0, Some(state, currentSeqNo + 1) ) ) None let splitAt n s = s |> Seq.mapi(fun i v -> i,v) |> splitBy (fst >> ((=) n )) SplitByOption.IncludeInSecond |> Seq.map (Seq.map snd) |> Seq.truncate 2 let tryHeadTail fhead ftail s = ( Choice1Of3(), splitAt 1 s ) ||> Seq.fold(function | Choice1Of3 ( ) -> Seq.tryHead >> Option.map fhead >> Choice2Of3 | Choice2Of3 (Some headv) -> ftail headv >> Choice3Of3 | result -> fun _ -> result ) |> function | Choice2Of3 (Some headv) -> Seq.empty |> ftail headv |> Some | Choice3Of3 v -> Some v | _ -> None /////////// Sample Usage [ -1 ; 0 ; 1; 2; 3; -1; 3; 5; 7; -1; 2; 3; 9 ; -1] |> Seq.map (fun x -> printfn "---> %A" x; x) |> splitBy ((=) -1) SplitByOption.Exclude |> Seq.skip 2 //|> Seq.take 2 |> Seq.iter (fun s -> let s = Seq.cache s Seq.length s |> printfn " ==> %A %A" (Seq.toList s) ) "Hello friend how are you? Good " |> Seq.toArray |> Seq.map (fun x -> printfn "---> %A" x; x) |> splitBy ((=) ' ') SplitByOption.IncludeInFirst //|> Seq.iter (printfn "%A") |> Seq.skip 3 //|> Seq.take 2 |> Seq.iter (fun s -> let s = Seq.cache s Seq.length s |> printfn " ==> %A %A" (System.String(Seq.toArray s)) ) [ 1 ; 2 ; 3 ; 4 ; 5 ; 6 ; 7 ; 8 ] |> Seq.map (fun x -> printfn "---> %A" x; x) |> splitAt 2 |> printfn "%A" [ 1 ; 2 ; 3 ; 4 ; 5 ; 6 ; 7 ; 8 ] |> Seq.map (fun x -> printfn "---> %A" x; x) |> tryHeadTail id (fun head tail -> printfn "head = %A, tail = %A" head tail ) |> Option.defaultWith (fun () -> printfn "No Head Tail" ) // The following example uses Seq.splitBy and Seq.tryHeadTail // twice to split the input file first in 2 and then into several pieces // collecting the output file names in the first section and the content in the subsequent pieces // it can split a huge 2.2GB file into several subfiles without having to rescan the file // or cache in memory any portions of the file apart from the current line // the first Seq.tryHeadTail operates on a seq>. Because of the nested seq it must use 2 // separate functions to process first the head and then the tail without backtracking, // the result from the first function is passed to the second: // |> Seq.tryHeadTail (fun head -> ...; filelist) (fun filelist tail -> ...) // the second Seq.tryHeadTail operates on a seq, because it is not nested seqs it can be called simply: // |> Seq.tryHeadTail id (fun head tail -> ...) open System.IO let splitFile path file = File.ReadLines file //|> Seq.mapi (fun i v -> printfn "%d -> %s" i v; v) |> Seq.skip 1 |> Seq.splitBy (fun l -> l.EndsWith "---*/") Seq.SplitByOption.IncludeInSecond |> Seq.tryHeadTail(fun head -> let filelist = head |> Seq.map (fun s-> s.Split('.') |> Seq.last) |> Seq.toArray for file in filelist do printfn "file: %A" file printfn "--------" filelist ) (fun filelist tail -> tail |> Seq.collect id |> Seq.splitBy (fun l -> l.EndsWith " rows affected)" || l = "(1 row affected)") Seq.SplitByOption.IncludeInFirst |> Seq.map (Seq.skip 1) |> Seq.iteri(fun i s -> if i < filelist.Length then s |> Seq.tryHeadTail id (fun head tail -> printfn "%s => %s" filelist.[i] head File.WriteAllLines(path + filelist.[i] + ".rpt", Seq.append [head] tail) ) |> Option.defaultValue () ) ) |> Option.defaultValue ()