#!/usr/bin/env ruby
# paint.rb
require "rubygems"
require "opencv"

include OpenCV

window = GUI::Window.new("free canvas")
canvas = CvMat.new(500, 500, CV_8U, 3).fill!(CvColor::White) # create white canvas
window.show canvas

colors = CvColor::constants.collect{ |i| i.to_s }

usage =<<USAGE
[mouse]
drag            - draw
right button    - fill by color
[keyborad]
1 to 9          - change thickness of line
type color name - change color
esc             - exit
USAGE
puts usage

# drawing option
opt = {
  :color => CvColor::Black,
  :tickness => 1
}

point = nil
window.on_mouse{ |m|
  case m.event
  when :move
    if m.left_button?
      canvas.line!(point, m, opt) if point
      point = m
    end
  when :left_button_down
    canvas.line!(m, m, opt)
    point = m
  when :left_button_up
    point = nil
  when :right_button_down
    mask = canvas.flood_fill!(m, opt[:color])    
  end
  window.show canvas
}

color_name = ''
while key = GUI.wait_key
  next if key < 0 or key > 255
  case key.chr
  when "\e" # [esc] - exit
    exit
  when '1'..'9'
    puts "change thickness to #{key.chr.to_i}."
    opt[:thickness] = key.chr.to_i
  when /[A-Za-z]/
    color_name << key.chr
    choice = colors.find_all{ |i| i =~ /\A#{color_name}/i }
    if choice.size == 1
      color,= choice
      puts "change color to #{color}."
      opt[:color] = CvColor::const_get(color)
    end
    color_name = '' if choice.size < 2
  end
end