OLD | NEW |
1 # Copyright 2014 The Chromium Authors. All rights reserved. | 1 # Copyright 2014 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 from __future__ import division | 5 from __future__ import division |
6 | 6 |
7 import warnings | 7 import warnings |
8 | 8 |
9 from telemetry.internal.util import external_modules | 9 from telemetry.internal.util import external_modules |
10 from telemetry.util import color_histogram | 10 from telemetry.util import color_histogram |
(...skipping 160 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
171 filtered = image.reshape(-1, 3) | 171 filtered = image.reshape(-1, 3) |
172 if ignore_color is not None: | 172 if ignore_color is not None: |
173 color = np.array([ignore_color.b, ignore_color.g, ignore_color.r]) | 173 color = np.array([ignore_color.b, ignore_color.g, ignore_color.r]) |
174 colorm = np.array(color) - tolerance | 174 colorm = np.array(color) - tolerance |
175 colorp = np.array(color) + tolerance | 175 colorp = np.array(color) + tolerance |
176 in_range = ((filtered[:, 0] < colorm[0]) | (filtered[:, 0] > colorp[0]) | | 176 in_range = ((filtered[:, 0] < colorm[0]) | (filtered[:, 0] > colorp[0]) | |
177 (filtered[:, 1] < colorm[1]) | (filtered[:, 1] > colorp[1]) | | 177 (filtered[:, 1] < colorm[1]) | (filtered[:, 1] > colorp[1]) | |
178 (filtered[:, 2] < colorm[2]) | (filtered[:, 2] > colorp[2])) | 178 (filtered[:, 2] < colorm[2]) | (filtered[:, 2] > colorp[2])) |
179 filtered = np.compress(in_range, filtered, axis=0) | 179 filtered = np.compress(in_range, filtered, axis=0) |
180 if len(filtered[:, 0]) == 0: | 180 if len(filtered[:, 0]) == 0: |
181 return color_histogram.ColorHistogram(np.zeros((256)), np.zeros((256)), | 181 return color_histogram.ColorHistogram( |
182 np.zeros((256)), ignore_color) | 182 np.zeros((256)), np.zeros((256)), |
| 183 np.zeros((256)), ignore_color) |
183 hist_b = np.bincount(filtered[:, 0], minlength=256) | 184 hist_b = np.bincount(filtered[:, 0], minlength=256) |
184 hist_g = np.bincount(filtered[:, 1], minlength=256) | 185 hist_g = np.bincount(filtered[:, 1], minlength=256) |
185 hist_r = np.bincount(filtered[:, 2], minlength=256) | 186 hist_r = np.bincount(filtered[:, 2], minlength=256) |
186 | 187 |
187 return color_histogram.ColorHistogram(hist_r, hist_g, hist_b, ignore_color) | 188 return color_histogram.ColorHistogram(hist_r, hist_g, hist_b, ignore_color) |
OLD | NEW |