xref: /freebsd/usr.bin/sort/coll.h (revision f4b37ed0f8b307b1f3f0f630ca725d68f1dff30d)
1 /*	$FreeBSD$	*/
2 
3 /*-
4  * Copyright (C) 2009 Gabor Kovesdan <gabor@FreeBSD.org>
5  * Copyright (C) 2012 Oleg Moskalenko <mom040267@gmail.com>
6  * All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  */
29 
30 #if !defined(__COLL_H__)
31 #define	__COLL_H__
32 
33 #include "bwstring.h"
34 #include "sort.h"
35 
36 /*
37  * Sort hint data for -n
38  */
39 struct n_hint
40 {
41 	unsigned long long	 n1;
42 	unsigned char		 si;
43 	bool			 empty;
44 	bool			 neg;
45 };
46 
47 /*
48  * Sort hint data for -g
49  */
50 struct g_hint
51 {
52 	double			 d;
53 	bool			 nan;
54 	bool			 notnum;
55 };
56 
57 /*
58  * Sort hint data for -M
59  */
60 struct M_hint
61 {
62 	int			 m;
63 };
64 
65 /*
66  * Status of a sort hint object
67  */
68 typedef enum
69 {
70 	HS_ERROR = -1, HS_UNINITIALIZED = 0, HS_INITIALIZED = 1
71 } hint_status;
72 
73 /*
74  * Sort hint object
75  */
76 struct key_hint
77 {
78 	hint_status		status;
79 	union
80 	{
81 		struct n_hint		nh;
82 		struct g_hint		gh;
83 		struct M_hint		Mh;
84 	}			v;
85 };
86 
87 /*
88  * Key value
89  */
90 struct key_value
91 {
92 	struct bwstring		*k; /* key string */
93 	struct key_hint		 hint[0]; /* key sort hint */
94 };
95 
96 /*
97  * Set of keys container object.
98  */
99 struct keys_array
100 {
101 	struct key_value	 key[0];
102 };
103 
104 /*
105  * Parsed -k option data
106  */
107 struct key_specs
108 {
109 	struct sort_mods	 sm;
110 	size_t			 c1;
111 	size_t			 c2;
112 	size_t			 f1;
113 	size_t			 f2;
114 	bool			 pos1b;
115 	bool			 pos2b;
116 };
117 
118 /*
119  * Single entry in sort list.
120  */
121 struct sort_list_item
122 {
123 	struct bwstring		*str;
124 	struct keys_array	 ka;
125 };
126 
127 /*
128  * Function type, used to compare two list objects
129  */
130 typedef int (*listcoll_t)(struct sort_list_item **ss1, struct sort_list_item **ss2);
131 
132 extern struct key_specs *keys;
133 extern size_t keys_num;
134 
135 /*
136  * Main localised symbols. These must be wint_t as they may hold WEOF.
137  */
138 extern wint_t symbol_decimal_point;
139 extern wint_t symbol_thousands_sep;
140 extern wint_t symbol_negative_sign;
141 extern wint_t symbol_positive_sign;
142 
143 /* funcs */
144 
145 cmpcoll_t get_sort_func(struct sort_mods *sm);
146 
147 struct keys_array *keys_array_alloc(void);
148 size_t keys_array_size(void);
149 void set_key_on_keys_array(struct keys_array *ka, struct bwstring *s, size_t ind);
150 void clean_keys_array(const struct bwstring *s, struct keys_array *ka);
151 
152 struct sort_list_item *sort_list_item_alloc(void);
153 void sort_list_item_set(struct sort_list_item *si, struct bwstring *str);
154 void sort_list_item_clean(struct sort_list_item *si);
155 size_t sort_list_item_size(struct sort_list_item *si);
156 
157 int preproc(struct bwstring *s, struct keys_array *ka);
158 int top_level_str_coll(const struct bwstring *, const struct bwstring *);
159 int key_coll(struct keys_array *ks1, struct keys_array *ks2, size_t offset);
160 int str_list_coll(struct bwstring *str1, struct sort_list_item **ss2);
161 int list_coll_by_str_only(struct sort_list_item **ss1, struct sort_list_item **ss2);
162 int list_coll(struct sort_list_item **ss1, struct sort_list_item **ss2);
163 int list_coll_offset(struct sort_list_item **ss1, struct sort_list_item **ss2, size_t offset);
164 
165 listcoll_t get_list_call_func(size_t offset);
166 
167 #endif /* __COLL_H__ */
168