第3回 Ruby/OpenCV進捗報告

はじめに

連休もそろそろ終わりですが、皆様いかがお過ごしでしょうか？
Ruby/OpenCVの開発はゆるゆる進んでおります。

進捗

デモ

前回までとはちょっと趣向を変えて、OpenCVのGUIを使ったペイントのデモです*3。
マウスのドラッグで線の描画、右クリックで塗りつぶし、ESCキーで終了です。また、キーを入力して線の太さの変更、色名を入力して線の色の変更ができます。
環境はRuby 1.9.2+OpenCV 2.2+Ubuntu 11.04です。

require 'opencv'
include OpenCV

# Windowに描画領域を作成
window = GUI::Window.new('free canvas')
canvas = CvMat.new(500, 500, CV_8U, 3).fill!(CvColor::White)
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

# 線の色と太さ
opt = {
  :color => CvColor::Black,
  :tickness => 1
}

# マウスイベントの処理
point = nil
window.on_mouse{ |m|
  case m.event
  when :left_button_down # マウスドラッグで線の描画
    canvas.line!(m, m, opt)
    point = m
  when :move
    if m.left_button?
      canvas.line!(point, m, opt) if point
      point = m
    end
  when :left_button_up
    point = nil
  when :right_button_down # マウスの右ボタンで塗りつぶし
    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
  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[0]
      puts "change color to #{color}."
      opt[:color] = CvColor::const_get(color)
    end
    color_name = '' if choice.size < 2
  end
end

こんな感じでペイントっぽいことができます。

まとめ

そんなわけで、第3回Ruby/OpenCV進捗報告でした。

リポジトリはこちら

人柱はいつでも大歓迎ですよー。
ser1zw/ruby-opencv - GitHub
https://github.com/ser1zw/ruby-opencv