Index: src/opts/SkMorphologyImageFilter_opts.h |
diff --git a/src/opts/SkMorphologyImageFilter_opts.h b/src/opts/SkMorphologyImageFilter_opts.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..e30a9e49736fa96df7edd3c4df15f2bdc199cc5d |
--- /dev/null |
+++ b/src/opts/SkMorphologyImageFilter_opts.h |
@@ -0,0 +1,137 @@ |
+/* |
+ * Copyright 2015 Google Inc. |
+ * |
+ * Use of this source code is governed by a BSD-style license that can be |
+ * found in the LICENSE file. |
+ */ |
+ |
+#ifndef SkMorphologyImageFilter_opts_DEFINED |
+#define SkMorphologyImageFilter_opts_DEFINED |
+ |
+namespace SK_OPTS_NS { |
+ |
+enum MorphType { kDilate, kErode }; |
+enum class MorphDirection { kX, kY }; |
+ |
+#if SK_CPU_SSE_LEVEL >= SK_CPU_SSE_LEVEL_SSE2 |
+template<MorphType type, MorphDirection direction> |
+static void morph(const SkPMColor* src, SkPMColor* dst, |
+ int radius, int width, int height, int srcStride, int dstStride) { |
+ const int srcStrideX = direction == MorphDirection::kX ? 1 : srcStride; |
+ const int dstStrideX = direction == MorphDirection::kX ? 1 : dstStride; |
+ const int srcStrideY = direction == MorphDirection::kX ? srcStride : 1; |
+ const int dstStrideY = direction == MorphDirection::kX ? dstStride : 1; |
+ radius = SkMin32(radius, width - 1); |
+ const SkPMColor* upperSrc = src + radius * srcStrideX; |
+ for (int x = 0; x < width; ++x) { |
+ const SkPMColor* lp = src; |
+ const SkPMColor* up = upperSrc; |
+ SkPMColor* dptr = dst; |
+ for (int y = 0; y < height; ++y) { |
+ __m128i extreme = (type == kDilate) ? _mm_setzero_si128() |
+ : _mm_set1_epi32(0xFFFFFFFF); |
+ for (const SkPMColor* p = lp; p <= up; p += srcStrideX) { |
+ __m128i src_pixel = _mm_cvtsi32_si128(*p); |
+ extreme = (type == kDilate) ? _mm_max_epu8(src_pixel, extreme) |
+ : _mm_min_epu8(src_pixel, extreme); |
+ } |
+ *dptr = _mm_cvtsi128_si32(extreme); |
+ dptr += dstStrideY; |
+ lp += srcStrideY; |
+ up += srcStrideY; |
+ } |
+ if (x >= radius) { src += srcStrideX; } |
+ if (x + radius < width - 1) { upperSrc += srcStrideX; } |
+ dst += dstStrideX; |
+ } |
+} |
+ |
+#elif defined(SK_ARM_HAS_NEON) |
+template<MorphType type, MorphDirection direction> |
+static void morph(const SkPMColor* src, SkPMColor* dst, |
+ int radius, int width, int height, int srcStride, int dstStride) { |
+ const int srcStrideX = direction == MorphDirection::kX ? 1 : srcStride; |
+ const int dstStrideX = direction == MorphDirection::kX ? 1 : dstStride; |
+ const int srcStrideY = direction == MorphDirection::kX ? srcStride : 1; |
+ const int dstStrideY = direction == MorphDirection::kX ? dstStride : 1; |
+ radius = SkMin32(radius, width - 1); |
+ const SkPMColor* upperSrc = src + radius * srcStrideX; |
+ for (int x = 0; x < width; ++x) { |
+ const SkPMColor* lp = src; |
+ const SkPMColor* up = upperSrc; |
+ SkPMColor* dptr = dst; |
+ for (int y = 0; y < height; ++y) { |
+ uint8x8_t extreme = vdup_n_u8(type == kDilate ? 0 : 255); |
+ for (const SkPMColor* p = lp; p <= up; p += srcStrideX) { |
+ uint8x8_t src_pixel = vreinterpret_u8_u32(vdup_n_u32(*p)); |
+ extreme = (type == kDilate) ? vmax_u8(src_pixel, extreme) |
+ : vmin_u8(src_pixel, extreme); |
+ } |
+ *dptr = vget_lane_u32(vreinterpret_u32_u8(extreme), 0); |
+ dptr += dstStrideY; |
+ lp += srcStrideY; |
+ up += srcStrideY; |
+ } |
+ if (x >= radius) src += srcStrideX; |
+ if (x + radius < width - 1) upperSrc += srcStrideX; |
+ dst += dstStrideX; |
+ } |
+} |
+ |
+#else |
+template<MorphType type, MorphDirection direction> |
+static void morph(const SkPMColor* src, SkPMColor* dst, |
+ int radius, int width, int height, int srcStride, int dstStride) { |
+ const int srcStrideX = direction == MorphDirection::kX ? 1 : srcStride; |
+ const int dstStrideX = direction == MorphDirection::kX ? 1 : dstStride; |
+ const int srcStrideY = direction == MorphDirection::kX ? srcStride : 1; |
+ const int dstStrideY = direction == MorphDirection::kX ? dstStride : 1; |
+ radius = SkMin32(radius, width - 1); |
+ const SkPMColor* upperSrc = src + radius * srcStrideX; |
+ for (int x = 0; x < width; ++x) { |
+ const SkPMColor* lp = src; |
+ const SkPMColor* up = upperSrc; |
+ SkPMColor* dptr = dst; |
+ for (int y = 0; y < height; ++y) { |
+ // If we're maxing (dilate), start from 0; if minning (erode), start from 255. |
+ const int start = (type == kDilate) ? 0 : 255; |
+ int B = start, G = start, R = start, A = start; |
+ for (const SkPMColor* p = lp; p <= up; p += srcStrideX) { |
+ int b = SkGetPackedB32(*p), |
+ g = SkGetPackedG32(*p), |
+ r = SkGetPackedR32(*p), |
+ a = SkGetPackedA32(*p); |
+ if (type == kDilate) { |
+ B = SkTMax(b, B); |
+ G = SkTMax(g, G); |
+ R = SkTMax(r, R); |
+ A = SkTMax(a, A); |
+ } else { |
+ B = SkTMin(b, B); |
+ G = SkTMin(g, G); |
+ R = SkTMin(r, R); |
+ A = SkTMin(a, A); |
+ } |
+ } |
+ *dptr = SkPackARGB32(A, R, G, B); |
+ dptr += dstStrideY; |
+ lp += srcStrideY; |
+ up += srcStrideY; |
+ } |
+ if (x >= radius) { src += srcStrideX; } |
+ if (x + radius < width - 1) { upperSrc += srcStrideX; } |
+ dst += dstStrideX; |
+ } |
+} |
+ |
+#endif |
+ |
+static auto dilate_x = &morph<kDilate, MorphDirection::kX>, |
+ dilate_y = &morph<kDilate, MorphDirection::kY>, |
+ erode_x = &morph<kErode, MorphDirection::kX>, |
+ erode_y = &morph<kErode, MorphDirection::kY>; |
+ |
+} // namespace SK_OPTS_NS |
+ |
+#endif//SkMorphologyImageFilter_opts_DEFINED |
+ |