1 /*************************************************************************
2 *
3 * © 2016 and later: Unicode, Inc. and others.
4 * License & terms of use: http://www.unicode.org/copyright.html
5 *
6 *************************************************************************
7 *************************************************************************
8 * COPYRIGHT:
9 * Copyright (C) 2002-2006 IBM, Inc. All Rights Reserved.
10 *
11 *************************************************************************/
12
13 /**
14 * This program demos string collation
15 */
16
17 const char gHelpString[] =
18 "usage: coll [options*] -source source_string -target target_string\n"
19 "-help Display this message.\n"
20 "-locale name ICU locale to use. Default is en_US\n"
21 "-rules rule Collation rules file (overrides locale)\n"
22 "-french French accent ordering\n"
23 "-norm Normalizing mode on\n"
24 "-shifted Shifted mode\n"
25 "-lower Lower case first\n"
26 "-upper Upper case first\n"
27 "-case Enable separate case level\n"
28 "-level n Sort level, 1 to 5, for Primary, Secondary, Tertiary, Quaternary, Identical\n"
29 "-source string Source string for comparison\n"
30 "-target string Target string for comparison\n"
31 "Example coll -rules \\u0026b\\u003ca -source a -target b\n"
32 "The format \\uXXXX is supported for the rules and comparison strings\n"
33 ;
34
35 #include <stdio.h>
36 #include <string.h>
37 #include <stdlib.h>
38
39 #include <unicode/utypes.h>
40 #include <unicode/ucol.h>
41 #include <unicode/ustring.h>
42
43 /**
44 * Command line option variables
45 * These global variables are set according to the options specified
46 * on the command line by the user.
47 */
48 char * opt_locale = "en_US";
49 char * opt_rules = 0;
50 UBool opt_help = false;
51 UBool opt_norm = false;
52 UBool opt_french = false;
53 UBool opt_shifted = false;
54 UBool opt_lower = false;
55 UBool opt_upper = false;
56 UBool opt_case = false;
57 int opt_level = 0;
58 char * opt_source = "abc";
59 char * opt_target = "abd";
60 UCollator * collator = 0;
61
62 /**
63 * Definitions for the command line options
64 */
65 struct OptSpec {
66 const char *name;
67 enum {FLAG, NUM, STRING} type;
68 void *pVar;
69 };
70
71 OptSpec opts[] = {
72 {"-locale", OptSpec::STRING, &opt_locale},
73 {"-rules", OptSpec::STRING, &opt_rules},
74 {"-source", OptSpec::STRING, &opt_source},
75 {"-target", OptSpec::STRING, &opt_target},
76 {"-norm", OptSpec::FLAG, &opt_norm},
77 {"-french", OptSpec::FLAG, &opt_french},
78 {"-shifted", OptSpec::FLAG, &opt_shifted},
79 {"-lower", OptSpec::FLAG, &opt_lower},
80 {"-upper", OptSpec::FLAG, &opt_upper},
81 {"-case", OptSpec::FLAG, &opt_case},
82 {"-level", OptSpec::NUM, &opt_level},
83 {"-help", OptSpec::FLAG, &opt_help},
84 {"-?", OptSpec::FLAG, &opt_help},
85 {0, OptSpec::FLAG, 0}
86 };
87
88 /**
89 * processOptions() Function to read the command line options.
90 */
processOptions(int argc,const char ** argv,OptSpec opts[])91 UBool processOptions(int argc, const char **argv, OptSpec opts[])
92 {
93 for (int argNum = 1; argNum < argc; argNum ++) {
94 const char *pArgName = argv[argNum];
95 OptSpec *pOpt;
96 for (pOpt = opts; pOpt->name != 0; pOpt ++) {
97 if (strcmp(pOpt->name, pArgName) == 0) {
98 switch (pOpt->type) {
99 case OptSpec::FLAG:
100 *(UBool *)(pOpt->pVar) = true;
101 break;
102 case OptSpec::STRING:
103 argNum ++;
104 if (argNum >= argc) {
105 fprintf(stderr, "value expected for \"%s\" option.\n",
106 pOpt->name);
107 return false;
108 }
109 *(const char **)(pOpt->pVar) = argv[argNum];
110 break;
111 case OptSpec::NUM:
112 argNum ++;
113 if (argNum >= argc) {
114 fprintf(stderr, "value expected for \"%s\" option.\n",
115 pOpt->name);
116 return false;
117 }
118 char *endp;
119 int i = strtol(argv[argNum], &endp, 0);
120 if (endp == argv[argNum]) {
121 fprintf(stderr,
122 "integer value expected for \"%s\" option.\n",
123 pOpt->name);
124 return false;
125 }
126 *(int *)(pOpt->pVar) = i;
127 }
128 break;
129 }
130 }
131 if (pOpt->name == 0)
132 {
133 fprintf(stderr, "Unrecognized option \"%s\"\n", pArgName);
134 return false;
135 }
136 }
137 return true;
138 }
139
140 /**
141 * ICU string comparison
142 */
strcmp()143 int strcmp()
144 {
145 UChar source[100];
146 UChar target[100];
147 u_unescape(opt_source, source, 100);
148 u_unescape(opt_target, target, 100);
149 UCollationResult result = ucol_strcoll(collator, source, -1, target, -1);
150 if (result == UCOL_LESS) {
151 return -1;
152 }
153 else if (result == UCOL_GREATER) {
154 return 1;
155 }
156 return 0;
157 }
158
159 /**
160 * Creates a collator
161 */
processCollator()162 UBool processCollator()
163 {
164 // Set up an ICU collator
165 UErrorCode status = U_ZERO_ERROR;
166 UChar rules[100];
167
168 if (opt_rules != 0) {
169 u_unescape(opt_rules, rules, 100);
170 collator = ucol_openRules(rules, -1, UCOL_OFF, UCOL_TERTIARY,
171 NULL, &status);
172 }
173 else {
174 collator = ucol_open(opt_locale, &status);
175 }
176 if (U_FAILURE(status)) {
177 fprintf(stderr, "Collator creation failed.: %d\n", status);
178 return false;
179 }
180 if (status == U_USING_DEFAULT_WARNING) {
181 fprintf(stderr, "Warning, U_USING_DEFAULT_WARNING for %s\n",
182 opt_locale);
183 }
184 if (status == U_USING_FALLBACK_WARNING) {
185 fprintf(stderr, "Warning, U_USING_FALLBACK_ERROR for %s\n",
186 opt_locale);
187 }
188 if (opt_norm) {
189 ucol_setAttribute(collator, UCOL_NORMALIZATION_MODE, UCOL_ON, &status);
190 }
191 if (opt_french) {
192 ucol_setAttribute(collator, UCOL_FRENCH_COLLATION, UCOL_ON, &status);
193 }
194 if (opt_lower) {
195 ucol_setAttribute(collator, UCOL_CASE_FIRST, UCOL_LOWER_FIRST,
196 &status);
197 }
198 if (opt_upper) {
199 ucol_setAttribute(collator, UCOL_CASE_FIRST, UCOL_UPPER_FIRST,
200 &status);
201 }
202 if (opt_case) {
203 ucol_setAttribute(collator, UCOL_CASE_LEVEL, UCOL_ON, &status);
204 }
205 if (opt_shifted) {
206 ucol_setAttribute(collator, UCOL_ALTERNATE_HANDLING, UCOL_SHIFTED,
207 &status);
208 }
209 if (opt_level != 0) {
210 switch (opt_level) {
211 case 1:
212 ucol_setAttribute(collator, UCOL_STRENGTH, UCOL_PRIMARY, &status);
213 break;
214 case 2:
215 ucol_setAttribute(collator, UCOL_STRENGTH, UCOL_SECONDARY,
216 &status);
217 break;
218 case 3:
219 ucol_setAttribute(collator, UCOL_STRENGTH, UCOL_TERTIARY, &status);
220 break;
221 case 4:
222 ucol_setAttribute(collator, UCOL_STRENGTH, UCOL_QUATERNARY,
223 &status);
224 break;
225 case 5:
226 ucol_setAttribute(collator, UCOL_STRENGTH, UCOL_IDENTICAL,
227 &status);
228 break;
229 default:
230 fprintf(stderr, "-level param must be between 1 and 5\n");
231 return false;
232 }
233 }
234 if (U_FAILURE(status)) {
235 fprintf(stderr, "Collator attribute setting failed.: %d\n", status);
236 return false;
237 }
238 return true;
239 }
240
241 /**
242 * Main -- process command line, read in and pre-process the test file,
243 * call other functions to do the actual tests.
244 */
main(int argc,const char ** argv)245 int main(int argc, const char** argv)
246 {
247 if (!static_cast<bool>(processOptions(argc, argv, opts)) || static_cast<bool>(opt_help)) {
248 printf(gHelpString);
249 return -1;
250 }
251
252 if (!static_cast<bool>(processCollator())) {
253 fprintf(stderr, "Error creating collator for comparison\n");
254 return -1;
255 }
256
257 fprintf(stdout, "Comparing source=%s and target=%s\n", opt_source,
258 opt_target);
259 int result = strcmp();
260 if (result == 0) {
261 fprintf(stdout, "source is equals to target\n");
262 }
263 else if (result < 0) {
264 fprintf(stdout, "source is less than target\n");
265 }
266 else {
267 fprintf(stdout, "source is greater than target\n");
268 }
269
270 ucol_close(collator);
271 return 0;
272 }
273