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
(** Streams equipped with a (non-canonical) location function *)
type ('e,'a) t = {
strm : ('e,'a) Stream.t;
fun_loc : int -> Loc.t;
mutable max_peek : int;
}
let from ?(loc=Loc.(initial ToplevelInput)) f =
let loct = Hashtbl.create 207 in
let loct_func loct i = Hashtbl.find loct i in
let loct_add loct i loc = Hashtbl.add loct i loc in
let strm =
let i = ref 0 in
Stream.from
(fun e ->
match f e with
| None -> None
| Some (a,loc) ->
loct_add loct !i loc; incr i; Some a) in
let fun_loc i = if i = 0 then loc else loct_func loct (i - 1) in
{ strm; max_peek = 0; fun_loc }
let count strm = Stream.count strm.strm
let current_loc strm =
strm.fun_loc (Stream.count strm.strm)
let max_peek_loc strm =
strm.fun_loc strm.max_peek
let interval_loc bp ep strm =
assert (bp <= ep);
if ep > strm.max_peek then failwith "Not peeked position";
if bp == ep then
Loc.after (strm.fun_loc bp) 0 0
else
let loc1 = strm.fun_loc (bp + 1) in
let loc2 = strm.fun_loc ep in
Loc.merge loc1 loc2
let get_loc n strm =
strm.fun_loc (n + 1)
let peek e strm =
let a = Stream.peek e strm.strm in
if Option.has_some a then strm.max_peek <- max (Stream.count strm.strm + 1) strm.max_peek;
a
let npeek e n strm =
let l = Stream.npeek e n strm.strm in
strm.max_peek <- max (Stream.count strm.strm + List.length l) strm.max_peek;
l
let peek_nth e n strm =
let list = Stream.npeek e (n + 1) strm.strm in
let rec loop list p =
match list, p with
x :: _, 0 -> strm.max_peek <- Stream.count strm.strm + n + 1; x
| _ :: l, p -> loop l (p - 1)
| [], p -> strm.max_peek <- Stream.count strm.strm + (n - p); raise Stream.Failure
in
loop list n
let junk e strm = Stream.junk e strm.strm
let njunk e len strm = Stream.njunk e len strm.strm
let next e strm = Stream.next e strm.strm