1 | // Copyright (C) 2016 The Qt Company Ltd. |
2 | // SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0 |
3 | |
4 | #include "compress.h" |
5 | |
6 | #include <QtCore/qdebug.h> |
7 | #include <QtCore/qlist.h> |
8 | |
9 | #include <algorithm> |
10 | #include <iterator> |
11 | #include <iostream> |
12 | |
13 | #define QLALR_NO_CHECK_SORTED_TABLE |
14 | |
15 | struct _Fit |
16 | { |
17 | inline bool operator () (int a, int b) const |
18 | { |
19 | return a == 0 || b == 0 || a == b; |
20 | } |
21 | }; |
22 | |
23 | struct _PerfectMatch |
24 | { |
25 | inline bool operator () (int a, int b) const |
26 | { return a == b; } |
27 | }; |
28 | |
29 | struct _GenerateCheck |
30 | { |
31 | QList<int>::const_iterator iterator; |
32 | int initial; |
33 | |
34 | _GenerateCheck(QList<int>::const_iterator it, int i) : iterator(it), initial(i) { } |
35 | |
36 | inline int operator()() |
37 | { |
38 | int check = initial++; |
39 | return *iterator++ ? check : -1; |
40 | } |
41 | }; |
42 | |
43 | class UncompressedRow |
44 | { |
45 | public: |
46 | typedef const int *const_iterator; |
47 | typedef const int *iterator; |
48 | |
49 | public: |
50 | inline UncompressedRow (): |
51 | _M_index (0), |
52 | _M_begin (nullptr), |
53 | _M_end (nullptr), |
54 | _M_beginNonZeros (nullptr), |
55 | _M_endNonZeros (nullptr) {} |
56 | |
57 | inline UncompressedRow (int index, const_iterator begin, const_iterator end) |
58 | { assign (index, begin, end); } |
59 | |
60 | inline int index () const { return _M_index; } |
61 | inline const_iterator begin () const { return _M_begin; } |
62 | inline const_iterator end () const { return _M_end; } |
63 | |
64 | inline void assign (int index, const_iterator begin, const_iterator end) |
65 | { |
66 | _M_index = index; |
67 | _M_begin = begin; |
68 | _M_end = end; |
69 | |
70 | _M_beginNonZeros = _M_begin; |
71 | _M_endNonZeros = _M_end; |
72 | |
73 | for (_M_beginNonZeros = _M_begin; _M_beginNonZeros != _M_end && ! _M_beginNonZeros [0]; ++_M_beginNonZeros) |
74 | /*continue*/ ; |
75 | |
76 | #if 0 |
77 | for (_M_endNonZeros = _M_end; _M_endNonZeros != _M_beginNonZeros && ! _M_endNonZeros [-1]; --_M_endNonZeros) |
78 | /*continue*/ ; |
79 | #endif |
80 | } |
81 | |
82 | inline int at (int index) const |
83 | { return _M_begin [index]; } |
84 | |
85 | inline bool isEmpty () const |
86 | { return _M_begin == _M_end; } |
87 | |
88 | inline int size () const |
89 | { return _M_end - _M_begin; } |
90 | |
91 | inline int nonZeroElements () const |
92 | { return _M_endNonZeros - _M_beginNonZeros; } |
93 | |
94 | inline int count (int value) const |
95 | { return std::count (first: begin (), last: end (), value: value); } |
96 | |
97 | inline const_iterator beginNonZeros () const |
98 | { return _M_beginNonZeros; } |
99 | |
100 | inline const_iterator endNonZeros () const |
101 | { return _M_endNonZeros; } |
102 | |
103 | private: |
104 | int _M_index; |
105 | const_iterator _M_begin; |
106 | const_iterator _M_end; |
107 | const_iterator _M_beginNonZeros; |
108 | const_iterator _M_endNonZeros; |
109 | }; |
110 | |
111 | struct _SortUncompressedRow |
112 | { |
113 | inline bool operator () (const UncompressedRow &a, const UncompressedRow &b) const |
114 | { return a.count (value: 0) > b.count (value: 0); } |
115 | }; |
116 | |
117 | Compress::Compress () |
118 | { |
119 | } |
120 | |
121 | void Compress::operator () (int *table, int row_count, int column_count) |
122 | { |
123 | index.clear (); |
124 | info.clear (); |
125 | check.clear (); |
126 | |
127 | QList<UncompressedRow> sortedTable(row_count); |
128 | |
129 | for (int i = 0; i < row_count; ++i) |
130 | { |
131 | int *begin = &table [i * column_count]; |
132 | int *end = begin + column_count; |
133 | |
134 | sortedTable [i].assign (index: i, begin, end); |
135 | } |
136 | |
137 | std::sort (first: sortedTable.begin (), last: sortedTable.end (), comp: _SortUncompressedRow ()); |
138 | |
139 | #ifndef QLALR_NO_CHECK_SORTED_TABLE |
140 | int previous_zeros = INT_MAX; |
141 | |
142 | for (const UncompressedRow &row : std::as_const(sortedTable)) |
143 | { |
144 | int zeros = row.count (0); |
145 | |
146 | Q_ASSERT (zeros <= previous_zeros); |
147 | zeros = previous_zeros; |
148 | qDebug () << "OK!" ; |
149 | } |
150 | #endif |
151 | |
152 | index.fill (t: -999999, newSize: row_count); |
153 | |
154 | for (const UncompressedRow &row : std::as_const(t&: sortedTable)) |
155 | { |
156 | int first_token = std::distance (first: row.begin (), last: row.beginNonZeros ()); |
157 | QList<int>::iterator pos = info.begin(); |
158 | |
159 | while (pos != info.end ()) |
160 | { |
161 | if (pos == info.begin ()) |
162 | { |
163 | // try to find a perfect match |
164 | QList<int>::iterator pm = std::search(first1: pos, last1: info.end(), first2: row.beginNonZeros(), |
165 | last2: row.endNonZeros(), predicate: _PerfectMatch()); |
166 | |
167 | if (pm != info.end ()) |
168 | { |
169 | pos = pm; |
170 | break; |
171 | } |
172 | } |
173 | |
174 | pos = std::search (first1: pos, last1: info.end (), first2: row.beginNonZeros (), last2: row.endNonZeros (), predicate: _Fit ()); |
175 | |
176 | if (pos == info.end ()) |
177 | break; |
178 | |
179 | int idx = std::distance (first: info.begin (), last: pos) - first_token; |
180 | bool conflict = false; |
181 | |
182 | for (int j = 0; ! conflict && j < row.size (); ++j) |
183 | { |
184 | if (row.at (index: j) == 0) |
185 | conflict |= idx + j >= 0 && check [idx + j] == j; |
186 | |
187 | else |
188 | conflict |= check [idx + j] == j; |
189 | } |
190 | |
191 | if (! conflict) |
192 | break; |
193 | |
194 | ++pos; |
195 | } |
196 | |
197 | if (pos == info.end ()) |
198 | { |
199 | int size = info.size (); |
200 | |
201 | info.resize (size: info.size () + row.nonZeroElements ()); |
202 | check.resize (size: info.size ()); |
203 | |
204 | std::fill (first: check.begin () + size, last: check.end (), value: -1); |
205 | pos = info.begin () + size; |
206 | } |
207 | |
208 | int offset = std::distance (first: info.begin (), last: pos); |
209 | index [row.index ()] = offset - first_token; |
210 | |
211 | for (const int *it = row.beginNonZeros (); it != row.endNonZeros (); ++it, ++pos) |
212 | { |
213 | if (*it) |
214 | *pos = *it; |
215 | } |
216 | |
217 | int i = row.index (); |
218 | |
219 | for (int j = 0; j < row.size (); ++j) |
220 | { |
221 | if (row.at (index: j) == 0) |
222 | continue; |
223 | |
224 | check [index [i] + j] = j; |
225 | } |
226 | } |
227 | |
228 | #if 0 |
229 | for (const UncompressedRow &row : std::as_const(sortedTable)) |
230 | { |
231 | int i = row.index (); |
232 | Q_ASSERT (i < sortedTable.size ()); |
233 | |
234 | for (int j = 0; j < row.size (); ++j) |
235 | { |
236 | if (row.at (j) == 0) |
237 | { |
238 | Q_ASSERT (index [i] + j < 0 || check [index [i] + j] != j); |
239 | continue; |
240 | } |
241 | |
242 | Q_ASSERT ( info [index [i] + j] == row.at (j)); |
243 | Q_ASSERT (check [index [i] + j] == j); |
244 | } |
245 | } |
246 | #endif |
247 | } |
248 | |