Merge remote branch 'gvdb/master'
[platform/upstream/glib.git] / tests / unicode-collate.c
1 #undef G_DISABLE_ASSERT
2 #undef G_LOG_DOMAIN
3
4 #include <glib.h>
5 #include <stdio.h>
6 #include <stdlib.h>
7 #include <string.h>
8 #include <locale.h>
9
10 typedef struct {
11   const char *key;
12   const char *str;
13 } Line;
14   
15
16 int 
17 compare_collate (const void *a, const void *b)
18 {
19   const Line *line_a = a;
20   const Line *line_b = b;
21
22   return g_utf8_collate (line_a->str, line_b->str);
23 }
24
25 int 
26 compare_key (const void *a, const void *b)
27 {
28   const Line *line_a = a;
29   const Line *line_b = b;
30
31   return strcmp (line_a->key, line_b->key);
32 }
33
34 int main (int argc, char **argv)
35 {
36   GIOChannel *in;
37   GError *error = NULL;
38   GArray *line_array = g_array_new (FALSE, FALSE, sizeof(Line));
39   guint i;
40   gboolean do_key = FALSE;
41   gboolean do_file = FALSE;
42   gchar *locale;
43
44   /* FIXME: need to modify environment here,
45    * since g_utf8_collate_key calls setlocal (LC_COLLATE, "")
46    */
47   g_setenv ("LC_ALL", "en_US", TRUE);
48   locale = setlocale (LC_ALL, "");
49   if (locale == NULL || strcmp (locale, "en_US") != 0)
50     {
51       fprintf (stderr, "No suitable locale, skipping test\n"); 
52       return 2;
53     }
54
55   if (argc != 1 && argc != 2 && argc != 3)
56     {
57       fprintf (stderr, "Usage: unicode-collate [--key|--file] [FILE]\n");
58       return 1;
59     }
60
61   i = 1;
62   if (argc > 1)
63     {
64       if (strcmp (argv[1], "--key") == 0)
65         {
66           do_key = TRUE;
67           i = 2;
68         }
69       else if (strcmp (argv[1], "--file") == 0)
70         {
71           do_key = TRUE;
72           do_file = TRUE;
73           i = 2;
74         }
75     }
76
77  if (argc > i)
78     {
79       in = g_io_channel_new_file (argv[i], "r", &error);
80       if (!in)
81         {
82           fprintf (stderr, "Cannot open %s: %s\n", argv[i], error->message);
83           return 1;
84         }
85     }
86   else
87     {
88       in = g_io_channel_unix_new (fileno (stdin));
89     }
90
91   while (TRUE)
92     {
93       gsize term_pos;
94       gchar *str;
95       Line line;
96
97       if (g_io_channel_read_line (in, &str, NULL, &term_pos, &error) != G_IO_STATUS_NORMAL)
98         break;
99
100       str[term_pos] = '\0';
101
102       if (do_file)
103         line.key = g_utf8_collate_key_for_filename (str, -1);
104       else
105         line.key = g_utf8_collate_key (str, -1);
106       line.str = str;
107
108       g_array_append_val (line_array, line);
109     }
110
111   if (error)
112     {
113       fprintf (stderr, "Error reading test file, %s\n", error->message);
114       return 1;
115     }
116
117   qsort (line_array->data, line_array->len, sizeof (Line), do_key ? compare_key : compare_collate);
118   for (i = 0; i < line_array->len; i++)
119     printf ("%s\n", g_array_index (line_array, Line, i).str);
120
121   g_io_channel_unref (in);
122
123   return 0;
124 }