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