heartbeat: multi-color support

diff --git a/draw-heartbeat.lua b/draw-heartbeat.lua
index 50a50983b38e55d58d67e79c003e5105e268ecb7..845fd7ac49d6b30c497a9ea76325643372f78c3c 100644 (file)
--- a/draw-heartbeat.lua
+++ b/draw-heartbeat.lua
@@ -1,9 +1,11 @@
 return function(t,fb,p)
   local c = p[1]
-  local ft = { [1] = function() fb:set(11,c)                      fb:set(13,c)     end, -- side chambers
-               [2] = function() fb:set(11,0,0,0) fb:set(20,c)     fb:set(13,0,0,0) end, -- bottom chamber
-               [3] = function()                  fb:set(20,0,0,0)                  end, -- empty
-               [4] = function()                                                    end  -- stay empty
+  local c2 = p[2] or c
+  local z = string.char(0,0,0)
+  local ft = { [1] = function() fb:set(11,c2)               fb:set(13,c2) end, -- side chambers
+               [2] = function() fb:set(11,z)  fb:set(20,c2) fb:set(13,z)  end, -- bottom chamber
+               [3] = function()               fb:set(20,z)                end, -- empty
+               [4] = function()                                           end  -- stay empty
              }
 
   fb:fill(0,0,0)