Grouping unused gen_server callbacks.

[cellular-automata.git] / 001 / src / life_time.erl

diff --git a/001/src/life_time.erl b/001/src/life_time.erl
index 6f9d859..22d3300 100644 (file)
--- a/001/src/life_time.erl
+++ b/001/src/life_time.erl
@@ -4,7 +4,7 @@
 
 %% API
 -export([start_link/3
-        ,tock/2
+        ,report_state/3
         ]).
 
 %% Callbacks
@@ -28,9 +28,11 @@
                ,y               :: integer()
                ,cells           :: list(atom())
                ,num_cells       :: integer()
+               ,num_dead        :: integer()
+               ,num_alive       :: integer()
                ,state_pairs     :: list(tuple(integer(), integer())) | []
                ,replies_pending :: integer()
-               ,gen_count = 0   :: integer()
+               ,gen_id          :: integer()
                }).
 
 
@@ -45,63 +47,74 @@ start_link(X, Y, Cells) ->
     gen_server:start_link(ServerName, ?MODULE, Args, Opts).
 
 
-tock(CellID, CellState) ->
-    gen_server:cast(?MODULE, {tock, {CellID, CellState}}).
+report_state(CellID, GenID, CellState) ->
+    gen_server:cast(?MODULE, {report_state, {CellID, GenID, CellState}}).
 
 
 %% ============================================================================
-%% Callbacks
+%% Callbacks (unused)
 %% ============================================================================
 
-init([X, Y, Cells]) ->
-    State = #state{x=X
-                  ,y=Y
-                  ,cells=Cells
-                  ,num_cells=length(Cells)
-                  ,state_pairs=[]
-                  ,replies_pending=0
-                  },
-    schedule_next_tick(),
-    {ok, State}.
-
+handle_call(_Msg, _From, State)  -> {reply, ok, State}.
+handle_info(_Msg, State)         -> {noreply, State}.
+code_change(_Old, State, _Other) -> {ok, State}.
+terminate(_Reason, State)        -> {ok, State}.
 
-terminate(_Reason, State) ->
-    {ok, State}.
 
+%% ============================================================================
+%% Callbacks
+%% ============================================================================
 
-code_change(_Old, State, _Other) ->
+init([X, Y, Cells]) ->
+    State = #state{x               = X
+                  ,y               = Y
+                  ,cells           = Cells
+                  ,num_cells       = length(Cells)
+                  ,state_pairs     = []
+                  ,replies_pending = 0
+                  ,gen_id          = 0
+                  },
+    ok = schedule_next_gen(),
     {ok, State}.
 
 
-handle_call(_Msg, _From, State) ->
-    {reply, ok, State}.
-
-
-handle_cast(next_tick,
+handle_cast(next_gen,
     #state{cells=Cells
           ,num_cells=NumCells
           ,state_pairs=[]
+          ,gen_id=GenID
           }=State) ->
 
-    ok = cast_all(Cells, tick),
-    {noreply, State#state{replies_pending=NumCells}};
+    NewGenID = GenID + 1,
+    ok = cast_all(Cells, {next_gen, NewGenID}),
+    NewState = State#state{replies_pending=NumCells
+                          ,gen_id=NewGenID
+                          ,num_dead=0
+                          ,num_alive=0
+                          },
+    {noreply, NewState};
 
-handle_cast({tock, {ID, CellState}},
+handle_cast({report_state, {CellID, GenID, CellState}},
     #state{x=X
           ,y=Y
+          ,num_dead=NDead
+          ,num_alive=NAlive
           ,state_pairs=StatePairs
           ,replies_pending=RepliesPending
-          ,gen_count=GenCount
+          ,gen_id=GenID
           ,num_cells=NumCells
           }=State) ->
 
-    NewStatePairs = [{ID, CellState} | StatePairs],
+    NewStatePairs = [{CellID, CellState} | StatePairs],
     NewRepliesPending = RepliesPending - 1,
-    NewState = State#state{replies_pending=NewRepliesPending},
+    {NewNDead, NewNAlive} = increment_dead_or_alive(CellState, NDead, NAlive),
+    NewState = State#state{replies_pending=NewRepliesPending
+                          ,num_dead=NewNDead
+                          ,num_alive=NewNAlive
+                          },
 
     case NewRepliesPending of
         0 ->
-            NewGenCount = GenCount + 1,
             SortedStatePairs = lists:sort(
                 fun({A, _}, {B, _}) -> A < B end,
                 NewStatePairs
@@ -109,16 +122,14 @@ handle_cast({tock, {ID, CellState}},
             StateChars = [state_to_char(S) || {_, S} <- SortedStatePairs],
 
             ok = io:format(
-                "X: ~b Y: ~b CELLS: ~b GENERATIONS: ~b~n",
-                [X, Y, NumCells, NewGenCount]
+                "X: ~b Y: ~b CELLS: ~b DEAD: ~b ALIVE: ~b GENERATION: ~b~n",
+                [X, Y, NumCells, NewNDead, NewNAlive, GenID]
             ),
             ok = do_print_bar(X),
-
             ok = do_print_state_chars(X, StateChars),
-
             ok = timer:sleep(?INTERVAL),
-            schedule_next_tick(),
-            {noreply, NewState#state{state_pairs=[], gen_count=NewGenCount}};
+            ok = schedule_next_gen(),
+            {noreply, NewState#state{state_pairs=[]}};
 
         _N ->
             {noreply, NewState#state{state_pairs=NewStatePairs}}
@@ -128,16 +139,16 @@ handle_cast(_Msg, State) ->
     {noreply, State}.
 
 
-handle_info(_Msg, State) ->
-    {noreply, State}.
-
-
 %% ============================================================================
 %% Internal
 %% ============================================================================
 
-schedule_next_tick() ->
-    gen_server:cast(?MODULE, next_tick).
+increment_dead_or_alive(0, NDead, NAlive) -> {NDead + 1, NAlive};
+increment_dead_or_alive(1, NDead, NAlive) -> {NDead, NAlive + 1}.
+
+
+schedule_next_gen() ->
+    ok = gen_server:cast(?MODULE, next_gen).
 
 
 cast_all([], _) -> ok;