sim.rb

require 'ncurses'
require 'optparse'

@mode = :WS2801

def get_color(r, g, b)
  return Ncurses::COLOR_PAIR("#{r > 0 ? 1 : 0}#{g > 0 ? 1: 0}#{b > 0 ? 1 : 0}".to_i(2))
end

def set_pixel(x, y, r, g, b)
  Ncurses::attrset(get_color(r, g, b))
  Ncurses::mvaddstr(y, x, " ")
end

def set_buffer(x, y, r, g, b)
  if @buffer.nil?
    @buffer = []
  end
  @buffer.push({ coord: [x, y], color: [r, g, b] })
end

def pixelx(i)
  if ((i - (i % 12)) / 12) % 2 == 0
    i % 12
  else
    11 - (i % 12)
  end
end

def pixely(i)
  (i - (i % 12)) / 12
end

def draw_pixel(buf)
  if buf.map(&:nil?).inject(false) { |n,p| n || p }
    puts "Not enough bytes"
  else
    if @mode != :WS2801
      set_pixel(pixelx(buf[0]), pixely(buf[0]), buf[1], buf[2], buf[3])
    else
      set_buffer(pixelx(buf[0]), pixely(buf[0]), buf[1], buf[2], buf[3])
    end
  end
  buf.clear
end

def draw_screen()
  return if @buffer.nil?
  @buffer.each do |data|
    set_pixel(data[:coord][0], data[:coord][1], data[:color][0], data[:color][1], data[:color][2])
  end
  @buffer = []
end

def process_byte(io, buf)
  byte = io.getbyte
  case byte
    when 0xFF then draw_pixel(buf)
    when 0xFE then draw_screen()
    else
      buf.push(byte)
  end
end

options = {}
parser = OptionParser.new do |opts|
  opts.banner = "Usage: sim.rb [options] PIPE"

  opts.on("-m", "--mode [WS2801]", "Select simulation mode (WS2801|WS2812)") do |m|
    raise "Unknown mode: #{m}" unless [ "WS2801", "WS2812" ].include?(m)
    @mode = m.to_sym
  end
end

parser.parse!(ARGV)

if ARGV[0].nil?
  puts parser.banner
  exit 1
end
serial_pipe = ARGV[0]

if !File.exists?(serial_pipe) or !File.pipe?(serial_pipe)
  puts "File #{serial_pipe} is not a valid pipe or does not exist"
  exit 1
end

begin
  Ncurses::initscr()
  Ncurses::noecho()
  Ncurses::cbreak()
  Ncurses::nodelay(Ncurses::stdscr, TRUE)
  Ncurses::curs_set(0)

  if !(Ncurses::has_colors?())
    raise "Colours no available :("
  end

  if (Ncurses.COLS < 12) or (Ncurses.LINES < 14)
    raise "Screen too small!"
  end

  Ncurses::start_color()
  Ncurses::init_pair(0, Ncurses::COLOR_BLACK, Ncurses::COLOR_BLACK)
  Ncurses::init_pair(1, Ncurses::COLOR_BLUE, Ncurses::COLOR_BLUE)
  Ncurses::init_pair(2, Ncurses::COLOR_GREEN, Ncurses::COLOR_GREEN)
  Ncurses::init_pair(3, Ncurses::COLOR_CYAN, Ncurses::COLOR_CYAN)
  Ncurses::init_pair(4, Ncurses::COLOR_RED, Ncurses::COLOR_RED)
  Ncurses::init_pair(5, Ncurses::COLOR_MAGENTA, Ncurses::COLOR_MAGENTA)
  Ncurses::init_pair(6, Ncurses::COLOR_YELLOW, Ncurses::COLOR_YELLOW)
  Ncurses::init_pair(7, Ncurses::COLOR_WHITE, Ncurses::COLOR_WHITE)

  io = File.open(serial_pipe, "r+")
  buf = []

  while (:restart)
    process_byte(io, buf)

    if ((ch = Ncurses::getch()) != Ncurses::ERR)
      if (ch == Ncurses::KEY_RESIZE)
        Ncurses::erase()
        window_size_changed = true
      else
        break
      end
    end
  end
ensure
  Ncurses::curs_set(1)
  Ncurses::endwin()
end