|
| 1 | +-module(example). |
| 2 | + |
| 3 | +-export([prime/1]). |
| 4 | + |
| 5 | + |
| 6 | +prime(N) when is_integer(N), N>0 -> |
| 7 | + Generator=prime_generator(), |
| 8 | + prime(Generator, N, []). |
| 9 | + |
| 10 | +prime(Generator, N, KnownPrimes0) when length(KnownPrimes0)<N -> |
| 11 | + Next=next_candidate(Generator), |
| 12 | + KnownPrimes1=case is_prime(Next, KnownPrimes0) of |
| 13 | + true -> KnownPrimes0++[Next]; |
| 14 | + false -> KnownPrimes0 |
| 15 | + end, |
| 16 | + prime(Generator, N, KnownPrimes1); |
| 17 | +prime(_, _, Primes) -> |
| 18 | + lists:last(Primes). |
| 19 | + |
| 20 | +is_prime(Candidate, KnownPrimes) -> |
| 21 | + SqrtCandidate=math:sqrt(Candidate), |
| 22 | + is_prime(Candidate, SqrtCandidate, KnownPrimes). |
| 23 | + |
| 24 | +is_prime(_, _, []) -> |
| 25 | + true; |
| 26 | +is_prime(_, SqrtCandidate, [KnownPrime|_]) when KnownPrime>SqrtCandidate -> |
| 27 | + true; |
| 28 | +is_prime(Candidate, _, [KnownPrime|_]) when Candidate rem KnownPrime=:=0 -> |
| 29 | + false; |
| 30 | +is_prime(Candidate, SqrtCandidate, [_|KnownPrimes]) -> |
| 31 | + is_prime(Candidate, SqrtCandidate, KnownPrimes). |
| 32 | + |
| 33 | +next_candidate(Generator) -> |
| 34 | + Candidate=receive {Generator, Candidate1} -> Candidate1 end, |
| 35 | + Generator ! {self(), next}, |
| 36 | + Candidate. |
| 37 | + |
| 38 | +prime_generator() -> |
| 39 | + Generator=spawn_link( |
| 40 | + fun () -> prime_generator_loop() end |
| 41 | + ), |
| 42 | + Generator ! {self(), next}, |
| 43 | + Generator. |
| 44 | + |
| 45 | +prime_generator_loop() -> |
| 46 | + prime_generator_loop(2). |
| 47 | + |
| 48 | +prime_generator_loop(2) -> |
| 49 | + receive {Reply, next} -> Reply ! {self(), 2} end, |
| 50 | + prime_generator_loop(3); |
| 51 | +prime_generator_loop(3) -> |
| 52 | + receive {Reply, next} -> Reply ! {self(), 3} end, |
| 53 | + prime_generator_loop(6); |
| 54 | +prime_generator_loop(N) -> |
| 55 | + receive {Reply1, next} -> Reply1 ! {self(), N-1} end, |
| 56 | + receive {Reply2, next} -> Reply2 ! {self(), N+1} end, |
| 57 | + prime_generator_loop(N+6). |
0 commit comments