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 |