-
Notifications
You must be signed in to change notification settings - Fork 0
/
master.erl
45 lines (40 loc) · 1.42 KB
/
master.erl
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
-module(master).
-export ([master/2]).
% Map exit signal to all ants within an array
exterminate(Ants) ->
lists:map(fun(A) -> A ! {die} end, Ants).
% Map init message to all ants within an array
awakening(Ants) ->
lists:map(fun(A) -> A ! {init} end, Ants).
wakeUp(0, _, _, _, Ants) when is_list(Ants) -> Ants;
wakeUp(N, Source, Nodes, Target, Ants) when
is_integer(N),
is_integer(Nodes),
is_list(Ants), N > 0 ->
A = spawn(ant, ant, [Source, [], Nodes, Target]),
wakeUp(N-1, Source, Nodes, Target, [A|Ants]);
wakeUp(_, _, _, _, _) -> throw("Cannot awake a non natural number of ants").
master(N, NodeList) -> master(N, NodeList, []).
master(N, NodeList, Ants) ->
receive
{init} ->
awakening(Ants),
master(N, NodeList, Ants);
{createAnts, N_ants, Source, Target} ->
exterminate(Ants),
NewAnts = wakeUp(N_ants, Source, N, Target, []),
master(N, NodeList, NewAnts);
{killAnts} ->
exterminate(Ants),
master(N, NodeList, []);
{printNodes} ->
io:format("Graph: ", []),
lists:map(fun (Nd) -> Nd ! {print} end, NodeList),
master(N, NodeList, Ants);
{printAnts} ->
io:format("Ants: ", []),
lists:map(fun (A) -> A ! {print} end, Ants),
master(N, NodeList, Ants);
_ ->
master(N, NodeList, Ants)
end.