| OLD | NEW |
| (Empty) |
| 1 /* | |
| 2 * Copyright (C) 2008 Apple Inc. All rights reserved. | |
| 3 * | |
| 4 * Redistribution and use in source and binary forms, with or without | |
| 5 * modification, are permitted provided that the following conditions | |
| 6 * are met: | |
| 7 * | |
| 8 * 1. Redistributions of source code must retain the above copyright | |
| 9 * notice, this list of conditions and the following disclaimer. | |
| 10 * 2. Redistributions in binary form must reproduce the above copyright | |
| 11 * notice, this list of conditions and the following disclaimer in the | |
| 12 * documentation and/or other materials provided with the distribution. | |
| 13 * 3. Neither the name of Apple Computer, Inc. ("Apple") nor the names of | |
| 14 * its contributors may be used to endorse or promote products derived | |
| 15 * from this software without specific prior written permission. | |
| 16 * | |
| 17 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY | |
| 18 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED | |
| 19 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE | |
| 20 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY | |
| 21 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES | |
| 22 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; | |
| 23 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND | |
| 24 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | |
| 25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF | |
| 26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
| 27 */ | |
| 28 | |
| 29 #include "wtf/text/Collator.h" | |
| 30 | |
| 31 #include "wtf/Assertions.h" | |
| 32 #include "wtf/PtrUtil.h" | |
| 33 #include "wtf/StdLibExtras.h" | |
| 34 #include "wtf/StringExtras.h" | |
| 35 #include "wtf/Threading.h" | |
| 36 #include "wtf/ThreadingPrimitives.h" | |
| 37 #include <memory> | |
| 38 #include <stdlib.h> | |
| 39 #include <string.h> | |
| 40 #include <unicode/ucol.h> | |
| 41 | |
| 42 namespace WTF { | |
| 43 | |
| 44 static UCollator* cachedCollator; | |
| 45 static char cachedEquivalentLocale[Collator::ulocFullnameCapacity]; | |
| 46 static Mutex& cachedCollatorMutex() { | |
| 47 DEFINE_THREAD_SAFE_STATIC_LOCAL(Mutex, mutex, new Mutex); | |
| 48 return mutex; | |
| 49 } | |
| 50 | |
| 51 Collator::Collator(const char* locale) | |
| 52 : m_collator(0), | |
| 53 m_locale(locale ? strdup(locale) : 0), | |
| 54 m_lowerFirst(false) { | |
| 55 setEquivalentLocale(m_locale, m_equivalentLocale); | |
| 56 } | |
| 57 | |
| 58 std::unique_ptr<Collator> Collator::userDefault() { | |
| 59 return WTF::wrapUnique(new Collator(0)); | |
| 60 } | |
| 61 | |
| 62 Collator::~Collator() { | |
| 63 releaseCollator(); | |
| 64 free(m_locale); | |
| 65 } | |
| 66 | |
| 67 void Collator::setOrderLowerFirst(bool lowerFirst) { | |
| 68 m_lowerFirst = lowerFirst; | |
| 69 } | |
| 70 | |
| 71 Collator::Result Collator::collate(const UChar* lhs, | |
| 72 size_t lhsLength, | |
| 73 const UChar* rhs, | |
| 74 size_t rhsLength) const { | |
| 75 if (!m_collator) | |
| 76 createCollator(); | |
| 77 | |
| 78 return static_cast<Result>( | |
| 79 ucol_strcoll(m_collator, lhs, lhsLength, rhs, rhsLength)); | |
| 80 } | |
| 81 | |
| 82 void Collator::createCollator() const { | |
| 83 DCHECK(!m_collator); | |
| 84 UErrorCode status = U_ZERO_ERROR; | |
| 85 | |
| 86 { | |
| 87 Locker<Mutex> lock(cachedCollatorMutex()); | |
| 88 if (cachedCollator) { | |
| 89 UColAttributeValue cachedCollatorLowerFirst = | |
| 90 ucol_getAttribute(cachedCollator, UCOL_CASE_FIRST, &status); | |
| 91 DCHECK(U_SUCCESS(status)); | |
| 92 | |
| 93 if (0 == strcmp(cachedEquivalentLocale, m_equivalentLocale) && | |
| 94 ((UCOL_LOWER_FIRST == cachedCollatorLowerFirst && m_lowerFirst) || | |
| 95 (UCOL_UPPER_FIRST == cachedCollatorLowerFirst && !m_lowerFirst))) { | |
| 96 m_collator = cachedCollator; | |
| 97 cachedCollator = 0; | |
| 98 cachedEquivalentLocale[0] = 0; | |
| 99 return; | |
| 100 } | |
| 101 } | |
| 102 } | |
| 103 | |
| 104 m_collator = ucol_open(m_locale, &status); | |
| 105 if (U_FAILURE(status)) { | |
| 106 status = U_ZERO_ERROR; | |
| 107 m_collator = | |
| 108 ucol_open("", &status); // Fallback to Unicode Collation Algorithm. | |
| 109 } | |
| 110 DCHECK(U_SUCCESS(status)); | |
| 111 | |
| 112 ucol_setAttribute(m_collator, UCOL_CASE_FIRST, | |
| 113 m_lowerFirst ? UCOL_LOWER_FIRST : UCOL_UPPER_FIRST, | |
| 114 &status); | |
| 115 DCHECK(U_SUCCESS(status)); | |
| 116 | |
| 117 ucol_setAttribute(m_collator, UCOL_NORMALIZATION_MODE, UCOL_ON, &status); | |
| 118 DCHECK(U_SUCCESS(status)); | |
| 119 } | |
| 120 | |
| 121 void Collator::releaseCollator() { | |
| 122 { | |
| 123 Locker<Mutex> lock(cachedCollatorMutex()); | |
| 124 if (cachedCollator) | |
| 125 ucol_close(cachedCollator); | |
| 126 cachedCollator = m_collator; | |
| 127 strncpy(cachedEquivalentLocale, m_equivalentLocale, ulocFullnameCapacity); | |
| 128 m_collator = 0; | |
| 129 } | |
| 130 m_collator = 0; | |
| 131 } | |
| 132 | |
| 133 void Collator::setEquivalentLocale(const char* locale, char* equivalentLocale) { | |
| 134 UErrorCode status = U_ZERO_ERROR; | |
| 135 UBool isAvailable; | |
| 136 ucol_getFunctionalEquivalent(equivalentLocale, ulocFullnameCapacity, | |
| 137 "collation", locale, &isAvailable, &status); | |
| 138 if (U_FAILURE(status)) | |
| 139 strcpy(equivalentLocale, "root"); | |
| 140 } | |
| 141 | |
| 142 } // namespace WTF | |
| OLD | NEW |