aboutsummaryrefslogtreecommitdiffstats
path: root/src/primes.ml
blob: 592fdc91541c8ddc3b99aabcf00fb022ab522755 (plain) (blame)
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

open Util

module Primes (K : Kahn.S) = struct
  module K = K
  module Lib = Kahn.Lib(K)
  open K
  open Lib

  let integers nmax (qo : int out_port) : unit process =
    let rec loop n =
      if n > nmax then
        put (-1) qo
      else
        (put n qo) >>= (fun () -> loop (n+1))
    in
    loop 2

  let filter n (qi : int in_port) (qo : int out_port) : unit process =
    let rec loop () =
      (get qi) >>= (fun v ->
        if v <> -1 then
          (if v mod n = 0 then return () else put v qo) >>= loop
        else
          put v qo)
    in loop()

  let rec primes (qi : int in_port) : unit process =
      (get qi) >>= (fun v ->
        if v <> -1 then
          begin
            K.output @@ Format.sprintf "%d@." v;
            (delay new_channel ()) >>=
            (fun (qi2, qo2) -> doco [ filter v qi qo2 ; primes qi2 ])
          end 
        else return ())

  let main : int process =
    (delay new_channel ()) >>=
    (fun (q_in, q_out) -> doco [ integers 500 q_out ; primes q_in ])
    >>= (fun () -> return 42)

end

module Eng = Kahn_stdio.ProtoKahn
module P = Primes(Eng)

let () =
    let r = P.K.run P.main in
    assert (r = 42);
    Format.eprintf "Primes finished (%d \\o/).@." r