Class: Artoo::Drivers::Opencv

Inherits:
Driver
  • Object
show all
Defined in:
lib/artoo/drivers/opencv.rb

Overview

The opencv driver behaviors

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(image) ⇒ Opencv

Returns a new instance of Opencv.



9
10
11
 
# File 'lib/artoo/drivers/opencv.rb', line 9

def initialize image
  @image = image
end

Instance Attribute Details

#image ⇒ Object

Returns the value of attribute image.



7
8
9
 
# File 'lib/artoo/drivers/opencv.rb', line 7

def image
  @image
end

Instance Method Details

#detect_circles(lower, upper) ⇒ Object 



13
14
15
16
17
 
# File 'lib/artoo/drivers/opencv.rb', line 13

def detect_circles(lower, upper)
  tmp_image = ::OpenCV::BGR2HSV(@image)
  tmp_image = tmp_image.in_range(::OpenCV::CvScalar.new(lower[:b], lower[:g], lower[:r]), ::OpenCV::CvScalar.new(upper[:b], upper[:g], upper[:r]))
  return tmp_image.hough_circles(::OpenCV::CV_HOUGH_GRADIENT, 2.0, 10, 200, 50)
end

#detect_faces(haar) ⇒ Object 



25
26
27
 
# File 'lib/artoo/drivers/opencv.rb', line 25

def detect_faces(haar)
  ::OpenCV::CvHaarClassifierCascade::load(haar).detect_objects(@image)
end

#draw_circles!(circles = []) ⇒ Object 



19
20
21
22
23
 
# File 'lib/artoo/drivers/opencv.rb', line 19

def draw_circles!(circles=[])
  circles.each do |circle|
    @image.circle! circle.center, circle.radius, :color => ::OpenCV::CvColor::Red, :thickness => 3
  end
end

#draw_rectangles!(rectangles = []) ⇒ Object 



29
30
31
32
33
 
# File 'lib/artoo/drivers/opencv.rb', line 29

def draw_rectangles!(rectangles=[])
  rectangles.each { |rect|
    @image.rectangle! rect.top_left, rect.bottom_right, :color => ::OpenCV::CvColor::Red, :thickness => 3
  }
end

#refresh ⇒ Object 



35
36
37
 
# File 'lib/artoo/drivers/opencv.rb', line 35

def refresh
  connection.connect
end