Simon!

lib/rgb_matrix/animation.ex

@@ -38,6 +38,7 @@ defmodule RGBMatrix.Animation do
           | __MODULE__.SolidColor
           | __MODULE__.Breathing
           | __MODULE__.SolidReactive
+          | __MODULE__.Simon
 
   @doc """
   Returns a list of the available types of animations.
@@ -52,7 +53,8 @@ defmodule RGBMatrix.Animation do
       __MODULE__.RandomKeypresses,
       __MODULE__.SolidColor,
       __MODULE__.Breathing,
-      __MODULE__.SolidReactive
+      __MODULE__.SolidReactive,
+      __MODULE__.Simon
     ]
   end
 

lib/rgb_matrix/animation/simon.ex

@@ -0,0 +1,158 @@
+defmodule RGBMatrix.Animation.Simon do
+  @moduledoc """
+  An interactive "Simon" game.
+  """
+
+  alias Chameleon.HSV
+  alias RGBMatrix.Animation
+
+  use Animation
+
+  defmodule Config do
+    @moduledoc false
+    use RGBMatrix.Animation.Config
+  end
+
+  defmodule State do
+    @moduledoc false
+    defstruct [:leds, :simon_sequence, :state, :colors]
+  end
+
+  @black Chameleon.RGB.new(0, 0, 0)
+  @red Chameleon.RGB.new(255, 0, 0)
+
+  @impl true
+  def new(leds, _config) do
+    state =
+      %State{leds: leds}
+      |> init_colors()
+      |> init_sequence()
+
+    {0, state}
+  end
+
+  @impl true
+  def render(%{state: :start_sequence} = state, _config) do
+    colors = state.leds |> Enum.map(&{&1.id, @black})
+
+    state = %{state | state: {:playing_sequence, state.simon_sequence}}
+
+    {1_000, colors, state}
+  end
+
+  @impl true
+  def render(%{state: {:black_flash, current_sequence}} = state, _config) do
+    colors = state.leds |> Enum.map(&{&1.id, @black})
+
+    state = %{state | state: {:playing_sequence, current_sequence}}
+
+    {150, colors, state}
+  end
+
+  @impl true
+  def render(%{state: {:playing_sequence, []}} = state, _config) do
+    colors = state.leds |> Enum.map(fn led -> {led.id, @black} end)
+
+    state = %{state | state: {:expecting_input, state.simon_sequence}}
+
+    {:never, colors, state}
+  end
+
+  @impl true
+  def render(%{state: {:playing_sequence, [{led, color} | rest]}} = state, _config) do
+    colors =
+      Enum.map(state.leds, fn
+        ^led -> {led.id, color}
+        other_led -> {other_led.id, @black}
+      end)
+
+    state = %{state | state: {:black_flash, rest}}
+
+    {850, colors, state}
+  end
+
+  @impl true
+  def render(%{state: {:feedback, [{led, color} | rest]}} = state, _config) do
+    colors =
+      Enum.map(state.leds, fn
+        ^led -> {led.id, color}
+        other_led -> {other_led.id, @black}
+      end)
+
+    state =
+      case rest do
+        [] -> extend_sequence(state)
+        rest -> %{state | state: {:expecting_input, rest}}
+      end
+
+    {500, colors, state}
+  end
+
+  @impl true
+  def render(%{state: :lost} = state, _config) do
+    colors = state.leds |> Enum.map(fn led -> {led.id, @red} end)
+
+    state =
+      state
+      |> init_sequence()
+
+    {2_000, colors, state}
+  end
+
+  @impl true
+  def render(state, _config) do
+    colors = state.leds |> Enum.map(fn led -> {led.id, @black} end)
+
+    {:never, colors, state}
+  end
+
+  @impl true
+  def interact(
+        %{state: {:expecting_input, [{led, _color} | _rest] = sequence}} = state,
+        _config,
+        led
+      ) do
+    state = %{state | state: {:feedback, sequence}}
+
+    {0, state}
+  end
+
+  @impl true
+  def interact(%{state: {:expecting_input, [_x | _rest]}} = state, _config, _y) do
+    state = %{state | state: :lost}
+
+    {0, state}
+  end
+
+  @impl true
+  def interact(state, _config, _led) do
+    {:ignore, state}
+  end
+
+  defp random_color do
+    HSV.new((:rand.uniform() * 360) |> trunc(), 100, 100)
+  end
+
+  defp init_colors(state) do
+    colors = for led <- state.leds, into: %{}, do: {led.id, random_color()}
+    %{state | colors: colors}
+  end
+
+  defp init_sequence(state) do
+    led = Enum.random(state.leds)
+    color = state.colors[led.id]
+    %State{state | simon_sequence: [{led, color}], state: :start_sequence}
+  end
+
+  defp extend_sequence(state) do
+    led = Enum.random(state.leds)
+    color = state.colors[led.id]
+    sequence = state.simon_sequence ++ [{led, color}]
+
+    %{
+      state
+      | simon_sequence: sequence,
+        state: :start_sequence
+    }
+  end
+end