Class: Capybara::Screenshot::Diff::Drivers::ChunkyPNGDriver::DifferenceRegionFinder
- Inherits:
-
Object
- Object
- Capybara::Screenshot::Diff::Drivers::ChunkyPNGDriver::DifferenceRegionFinder
- Defined in:
- lib/capybara/screenshot/diff/drivers/chunky_png_driver.rb
Instance Attribute Summary collapse
-
#color_distance_limit ⇒ Object
Returns the value of attribute color_distance_limit.
-
#shift_distance_limit ⇒ Object
Returns the value of attribute shift_distance_limit.
-
#skip_area ⇒ Object
Returns the value of attribute skip_area.
Instance Method Summary collapse
- #color_distance_at(new_img, old_img, x, y, shift_distance_limit:) ⇒ Object
- #color_matches(new_img, org_color, x, y, color_distance_limit) ⇒ Object
- #difference_level(_diff_mask, base_image, region) ⇒ Object
- #find_bottom(old_img, new_img, left, right, bottom, cache:) ⇒ Object
- #find_diff_rectangle(org_img, new_img, area_coordinates, cache:) ⇒ Object
- #find_difference_region(comparison) ⇒ Object
- #find_left_right_and_top(old_img, new_img, region, cache:) ⇒ Object
- #find_top(old_img, new_img, cache:) ⇒ Object
-
#initialize(comparison, driver = nil) ⇒ DifferenceRegionFinder
constructor
A new instance of DifferenceRegionFinder.
- #perform ⇒ Object
- #same_color?(old_img, new_img, x, y, cache:) ⇒ Boolean
- #shift_distance_at(new_img, old_img, x, y, color_distance_limit:) ⇒ Object
- #skipped_region?(x, y) ⇒ Boolean
Constructor Details
#initialize(comparison, driver = nil) ⇒ DifferenceRegionFinder
Returns a new instance of DifferenceRegionFinder.
79 80 81 82 83 84 85 86 |
# File 'lib/capybara/screenshot/diff/drivers/chunky_png_driver.rb', line 79 def initialize(comparison, driver = nil) @comparison = comparison @driver = driver @color_distance_limit = comparison.[:color_distance_limit] @shift_distance_limit = comparison.[:shift_distance_limit] @skip_area = comparison.[:skip_area] end |
Instance Attribute Details
#color_distance_limit ⇒ Object
Returns the value of attribute color_distance_limit.
77 78 79 |
# File 'lib/capybara/screenshot/diff/drivers/chunky_png_driver.rb', line 77 def color_distance_limit @color_distance_limit end |
#shift_distance_limit ⇒ Object
Returns the value of attribute shift_distance_limit.
77 78 79 |
# File 'lib/capybara/screenshot/diff/drivers/chunky_png_driver.rb', line 77 def shift_distance_limit @shift_distance_limit end |
#skip_area ⇒ Object
Returns the value of attribute skip_area.
77 78 79 |
# File 'lib/capybara/screenshot/diff/drivers/chunky_png_driver.rb', line 77 def skip_area @skip_area end |
Instance Method Details
#color_distance_at(new_img, old_img, x, y, shift_distance_limit:) ⇒ Object
213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 |
# File 'lib/capybara/screenshot/diff/drivers/chunky_png_driver.rb', line 213 def color_distance_at(new_img, old_img, x, y, shift_distance_limit:) org_color = old_img[x, y] if shift_distance_limit start_x = [0, x - shift_distance_limit].max end_x = [x + shift_distance_limit, new_img.width - 1].min xs = (start_x..end_x).to_a start_y = [0, y - shift_distance_limit].max end_y = [y + shift_distance_limit, new_img.height - 1].min ys = (start_y..end_y).to_a new_pixels = xs.product(ys) distances = new_pixels.map do |dx, dy| ChunkyPNG::Color.euclidean_distance_rgba(org_color, new_img[dx, dy]) end distances.min else ChunkyPNG::Color.euclidean_distance_rgba(org_color, new_img[x, y]) end end |
#color_matches(new_img, org_color, x, y, color_distance_limit) ⇒ Object
286 287 288 289 290 291 292 |
# File 'lib/capybara/screenshot/diff/drivers/chunky_png_driver.rb', line 286 def color_matches(new_img, org_color, x, y, color_distance_limit) new_color = new_img[x, y] return new_color == org_color unless color_distance_limit color_distance = ChunkyPNG::Color.euclidean_distance_rgba(org_color, new_color) color_distance <= color_distance_limit end |
#difference_level(_diff_mask, base_image, region) ⇒ Object
119 120 121 122 123 124 |
# File 'lib/capybara/screenshot/diff/drivers/chunky_png_driver.rb', line 119 def difference_level(_diff_mask, base_image, region) image_area_size = @driver.image_area_size(base_image) return nil if image_area_size.zero? region.size.to_f / image_area_size end |
#find_bottom(old_img, new_img, left, right, bottom, cache:) ⇒ Object
171 172 173 174 175 176 177 178 179 180 181 |
# File 'lib/capybara/screenshot/diff/drivers/chunky_png_driver.rb', line 171 def find_bottom(old_img, new_img, left, right, bottom, cache:) if bottom (old_img.height - 1).step(bottom + 1, -1).find do |y| (left..right).find do |x| bottom = y unless same_color?(old_img, new_img, x, y, cache: cache) end end end bottom end |
#find_diff_rectangle(org_img, new_img, area_coordinates, cache:) ⇒ Object
126 127 128 129 130 131 |
# File 'lib/capybara/screenshot/diff/drivers/chunky_png_driver.rb', line 126 def find_diff_rectangle(org_img, new_img, area_coordinates, cache:) left, top, right, bottom = find_left_right_and_top(org_img, new_img, area_coordinates, cache: cache) bottom = find_bottom(org_img, new_img, left, right, bottom, cache: cache) Region.from_edge_coordinates(left, top, right, bottom) end |
#find_difference_region(comparison) ⇒ Object
92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 |
# File 'lib/capybara/screenshot/diff/drivers/chunky_png_driver.rb', line 92 def find_difference_region(comparison) new_image, base_image, = comparison.new_image, comparison.base_image = {} [:max_color_distance] = 0 [:max_shift_distance] = 0 if shift_distance_limit region = find_top(base_image, new_image, cache: ) region = if region.nil? || region[1].nil? nil else find_diff_rectangle(base_image, new_image, region, cache: ) end result = Difference.new(region, , comparison) unless result.blank? [:max_color_distance] = [:max_color_distance].ceil(1) if [:max_color_distance] if comparison.[:tolerance] [:difference_level] = difference_level(nil, base_image, region) end end result end |
#find_left_right_and_top(old_img, new_img, region, cache:) ⇒ Object
142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 |
# File 'lib/capybara/screenshot/diff/drivers/chunky_png_driver.rb', line 142 def find_left_right_and_top(old_img, new_img, region, cache:) region = region.is_a?(Region) ? region.to_edge_coordinates : region left = region[0] || old_img.width - 1 top = region[1] right = region[2] || 0 bottom = region[3] old_img.height.times do |y| (0...left).find do |x| next if same_color?(old_img, new_img, x, y, cache: cache) top ||= y bottom = y left = x right = x if x > right x end (old_img.width - 1).step(right + 1, -1).find do |x| unless same_color?(old_img, new_img, x, y, cache: cache) bottom = y right = x end end end [left, top, right, bottom] end |
#find_top(old_img, new_img, cache:) ⇒ Object
133 134 135 136 137 138 139 140 |
# File 'lib/capybara/screenshot/diff/drivers/chunky_png_driver.rb', line 133 def find_top(old_img, new_img, cache:) old_img.height.times do |y| old_img.width.times do |x| return [x, y, x, y] unless same_color?(old_img, new_img, x, y, cache: cache) end end nil end |
#perform ⇒ Object
88 89 90 |
# File 'lib/capybara/screenshot/diff/drivers/chunky_png_driver.rb', line 88 def perform find_difference_region(@comparison) end |
#same_color?(old_img, new_img, x, y, cache:) ⇒ Boolean
183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 |
# File 'lib/capybara/screenshot/diff/drivers/chunky_png_driver.rb', line 183 def same_color?(old_img, new_img, x, y, cache:) return true if skipped_region?(x, y) color_distance = color_distance_at(new_img, old_img, x, y, shift_distance_limit: @shift_distance_limit) if color_distance > cache[:max_color_distance] cache[:max_color_distance] = color_distance end color_matches = color_distance == 0 || (!!@color_distance_limit && @color_distance_limit > 0 && color_distance <= @color_distance_limit) return color_matches if !@shift_distance_limit || cache[:max_shift_distance] == Float::INFINITY shift_distance = (color_matches && 0) || shift_distance_at(new_img, old_img, x, y, color_distance_limit: @color_distance_limit) if shift_distance && (cache[:max_shift_distance].nil? || shift_distance > cache[:max_shift_distance]) cache[:max_shift_distance] = shift_distance end color_matches end |
#shift_distance_at(new_img, old_img, x, y, color_distance_limit:) ⇒ Object
233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 |
# File 'lib/capybara/screenshot/diff/drivers/chunky_png_driver.rb', line 233 def shift_distance_at(new_img, old_img, x, y, color_distance_limit:) org_color = old_img[x, y] shift_distance = 0 loop do bounds_breached = 0 top_row = y - shift_distance if top_row >= 0 # top ([0, x - shift_distance].max..[x + shift_distance, new_img.width - 1].min).each do |dx| if color_matches(new_img, org_color, dx, top_row, color_distance_limit) return shift_distance end end else bounds_breached += 1 end if shift_distance > 0 if (x - shift_distance) >= 0 # left ([0, top_row + 1].max..[y + shift_distance, new_img.height - 2].min) .each do |dy| if color_matches(new_img, org_color, x - shift_distance, dy, color_distance_limit) return shift_distance end end else bounds_breached += 1 end if (y + shift_distance) < new_img.height # bottom ([0, x - shift_distance].max..[x + shift_distance, new_img.width - 1].min).each do |dx| if color_matches(new_img, org_color, dx, y + shift_distance, color_distance_limit) return shift_distance end end else bounds_breached += 1 end if (x + shift_distance) < new_img.width # right ([0, top_row + 1].max..[y + shift_distance, new_img.height - 2].min) .each do |dy| if color_matches(new_img, org_color, x + shift_distance, dy, color_distance_limit) return shift_distance end end else bounds_breached += 1 end end break if bounds_breached == 4 shift_distance += 1 end Float::INFINITY end |
#skipped_region?(x, y) ⇒ Boolean
207 208 209 210 211 |
# File 'lib/capybara/screenshot/diff/drivers/chunky_png_driver.rb', line 207 def skipped_region?(x, y) return false unless @skip_area @skip_area.any? { |region| region.cover?(x, y) } end |