Rochester.rb LCD Number Solution

lcd.rb

#!/usr/bin/env ruby

@numbers = [
  ["-", "|", "|", " ", "|", "|", "-"],
  [" ", " ", "|", " ", " ", "|", " "],
  ["-", " ", "|", "-", "|", " ", "-"],
  ["-", " ", "|", "-", " ", "|", "-"],
  [" ", "|", "|", "-", " ", "|", " "],
  ["-", "|", " ", "-", " ", "|", "-"],
  ["-", "|", " ", "-", "|", "|", "-"],
  ["-", " ", "|", " ", " ", "|", " "],
  ["-", "|", "|", "-", "|", "|", "-"],
  ["-", "|", "|", "-", " ", "|", "-"]
]

def number_to_lcd(i, size = 2)
  n = @numbers[i.to_i]
  rows = []
  rows[0] = " " + (n[0] * size) + " "
  size.times do
    rows << n[1] + (" " * size) + n[2]
  end
  rows << " " + (n[3] * size) + " "
  size.times do
    rows << n[4] + (" " * size) + n[5]
  end
  rows << " " + (n[6] * size) + " "
end

rows = []
size = 2

input = ARGV[0]

if ARGV[0] == "-s"
  size = ARGV[1].to_i
  input = ARGV[2]
end

input.chomp.split("").each do |char|
  number_to_lcd(char, size).each_with_index do |num_row, i|
    rows[i] ||= []
    rows[i] << num_row
  end
end

puts rows.map{|r| r.join(" ")}