caryclark@google.com | d88e089 | 2012-03-27 13:23:51 +0000 | [diff] [blame] | 1 | #ifndef TSearch_DEFINED |
| 2 | #define TSearch_DEFINED |
| 3 | |
caryclark@google.com | c682590 | 2012-02-03 22:07:47 +0000 | [diff] [blame] | 4 | #include "SkTypes.h" |
| 5 | |
| 6 | // FIXME: Move this templated version into SKTSearch.h |
| 7 | |
| 8 | template <typename T> |
caryclark@google.com | cd4421d | 2012-03-01 19:16:31 +0000 | [diff] [blame] | 9 | static T** QSort_Partition(T** left, T** right, T** pivot) |
caryclark@google.com | c682590 | 2012-02-03 22:07:47 +0000 | [diff] [blame] | 10 | { |
caryclark@google.com | cd4421d | 2012-03-01 19:16:31 +0000 | [diff] [blame] | 11 | T* pivotValue = *pivot; |
| 12 | SkTSwap(*pivot, *right); |
| 13 | T** newPivot = left; |
| 14 | while (left < right) { |
| 15 | if (**left < *pivotValue) { |
| 16 | SkTSwap(*left, *newPivot); |
| 17 | newPivot += 1; |
caryclark@google.com | c682590 | 2012-02-03 22:07:47 +0000 | [diff] [blame] | 18 | } |
caryclark@google.com | cd4421d | 2012-03-01 19:16:31 +0000 | [diff] [blame] | 19 | left += 1; |
caryclark@google.com | c682590 | 2012-02-03 22:07:47 +0000 | [diff] [blame] | 20 | } |
caryclark@google.com | cd4421d | 2012-03-01 19:16:31 +0000 | [diff] [blame] | 21 | SkTSwap(*newPivot, *right); |
| 22 | return newPivot; |
caryclark@google.com | c682590 | 2012-02-03 22:07:47 +0000 | [diff] [blame] | 23 | } |
| 24 | |
| 25 | template <typename T> |
caryclark@google.com | cd4421d | 2012-03-01 19:16:31 +0000 | [diff] [blame] | 26 | void QSort(T** left, T** right) |
caryclark@google.com | c682590 | 2012-02-03 22:07:47 +0000 | [diff] [blame] | 27 | { |
caryclark@google.com | cd4421d | 2012-03-01 19:16:31 +0000 | [diff] [blame] | 28 | if (left >= right) { |
caryclark@google.com | c682590 | 2012-02-03 22:07:47 +0000 | [diff] [blame] | 29 | return; |
| 30 | } |
caryclark@google.com | cd4421d | 2012-03-01 19:16:31 +0000 | [diff] [blame] | 31 | T** pivot = left + (right - left >> 1); |
| 32 | pivot = QSort_Partition(left, right, pivot); |
| 33 | QSort(left, pivot - 1); |
| 34 | QSort(pivot + 1, right); |
caryclark@google.com | c682590 | 2012-02-03 22:07:47 +0000 | [diff] [blame] | 35 | } |
| 36 | |
caryclark@google.com | f8b000d | 2012-02-09 22:04:27 +0000 | [diff] [blame] | 37 | template <typename S, typename T> |
caryclark@google.com | cef7e3f | 2012-02-28 16:57:05 +0000 | [diff] [blame] | 38 | static T* QSort_Partition(S& context, T* left, T* right, T* pivot, |
| 39 | bool (*lessThan)(S&, const T, const T)) |
caryclark@google.com | c682590 | 2012-02-03 22:07:47 +0000 | [diff] [blame] | 40 | { |
caryclark@google.com | cef7e3f | 2012-02-28 16:57:05 +0000 | [diff] [blame] | 41 | T pivotValue = *pivot; |
| 42 | SkTSwap(*pivot, *right); |
| 43 | T* newPivot = left; |
| 44 | while (left < right) { |
| 45 | if (lessThan(context, *left, pivotValue)) { |
| 46 | SkTSwap(*left, *newPivot); |
| 47 | newPivot += 1; |
caryclark@google.com | c682590 | 2012-02-03 22:07:47 +0000 | [diff] [blame] | 48 | } |
caryclark@google.com | cef7e3f | 2012-02-28 16:57:05 +0000 | [diff] [blame] | 49 | left += 1; |
caryclark@google.com | c682590 | 2012-02-03 22:07:47 +0000 | [diff] [blame] | 50 | } |
caryclark@google.com | cef7e3f | 2012-02-28 16:57:05 +0000 | [diff] [blame] | 51 | SkTSwap(*newPivot, *right); |
| 52 | return newPivot; |
caryclark@google.com | c682590 | 2012-02-03 22:07:47 +0000 | [diff] [blame] | 53 | } |
| 54 | |
caryclark@google.com | f8b000d | 2012-02-09 22:04:27 +0000 | [diff] [blame] | 55 | template <typename S, typename T> |
caryclark@google.com | cef7e3f | 2012-02-28 16:57:05 +0000 | [diff] [blame] | 56 | void QSort(S& context, T* left, T* right, |
| 57 | bool (*lessThan)(S& , const T, const T)) |
caryclark@google.com | c682590 | 2012-02-03 22:07:47 +0000 | [diff] [blame] | 58 | { |
caryclark@google.com | cef7e3f | 2012-02-28 16:57:05 +0000 | [diff] [blame] | 59 | if (left >= right) { |
caryclark@google.com | c682590 | 2012-02-03 22:07:47 +0000 | [diff] [blame] | 60 | return; |
| 61 | } |
caryclark@google.com | cef7e3f | 2012-02-28 16:57:05 +0000 | [diff] [blame] | 62 | T* pivot = left + (right - left >> 1); |
| 63 | pivot = QSort_Partition(context, left, right, pivot, lessThan); |
| 64 | QSort(context, left, pivot - 1, lessThan); |
| 65 | QSort(context, pivot + 1, right, lessThan); |
caryclark@google.com | c682590 | 2012-02-03 22:07:47 +0000 | [diff] [blame] | 66 | } |
caryclark@google.com | d88e089 | 2012-03-27 13:23:51 +0000 | [diff] [blame] | 67 | |
| 68 | #endif |