1 /* SearchMode.cpp */
2 
3 /* Copyright (C) 2011-2020 Michael Lugmair (Lucio Carreras)
4  *
5  * This file is part of sayonara player
6  *
7  * This program is free software: you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation, either version 3 of the License, or
10  * (at your option) any later version.
11 
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16 
17  * You should have received a copy of the GNU General Public License
18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
19  */
20 
21 #include "SearchMode.h"
22 #include "Utils/Settings/Settings.h"
23 #include "Utils/Utils.h"
24 
25 #include <QCache>
26 #include <QMap>
27 #include <QList>
28 #include <QString>
29 
30 #include <array>
31 
32 using DiacMap = QMap<QString, QString>;
33 
34 using ConvertCacheEntry = QCache<QString, QString>;
35 using ConvertCache = std::array<ConvertCacheEntry, +Library::SearchModeMaskSize>;
36 
37 Q_GLOBAL_STATIC(DiacMap, diacriticChars)
38 Q_GLOBAL_STATIC(ConvertCache, convertCache)
39 
40 namespace
41 {
initDiacriticChars()42 	void initDiacriticChars()
43 	{
44 		diacriticChars->insert(QString::fromUtf8("Š"), "S");
45 		diacriticChars->insert(QString::fromUtf8("Œ"), "OE");
46 		diacriticChars->insert(QString::fromUtf8("Ž"), "Z");
47 		diacriticChars->insert(QString::fromUtf8("š"), "s");
48 		diacriticChars->insert(QString::fromUtf8("œ"), "oe");
49 		diacriticChars->insert(QString::fromUtf8("ž"), "z");
50 		diacriticChars->insert(QString::fromUtf8("Ÿ"), "Y");
51 		diacriticChars->insert(QString::fromUtf8("¥"), "Y");
52 		diacriticChars->insert(QString::fromUtf8("µ"), "u");
53 		diacriticChars->insert(QString::fromUtf8("À"), "A");
54 		diacriticChars->insert(QString::fromUtf8("Á"), "A");
55 		diacriticChars->insert(QString::fromUtf8("Â"), "A");
56 		diacriticChars->insert(QString::fromUtf8("Ä"), "A");
57 		diacriticChars->insert(QString::fromUtf8("Å"), "A");
58 		diacriticChars->insert(QString::fromUtf8("Æ"), "AE");
59 		diacriticChars->insert(QString::fromUtf8("Ç"), "C");
60 		diacriticChars->insert(QString::fromUtf8("È"), "E");
61 		diacriticChars->insert(QString::fromUtf8("É"), "E");
62 		diacriticChars->insert(QString::fromUtf8("Ê"), "E");
63 		diacriticChars->insert(QString::fromUtf8("Ë"), "E");
64 		diacriticChars->insert(QString::fromUtf8("Ì"), "I");
65 		diacriticChars->insert(QString::fromUtf8("Í"), "I");
66 		diacriticChars->insert(QString::fromUtf8("Î"), "I");
67 		diacriticChars->insert(QString::fromUtf8("Ï"), "I");
68 		diacriticChars->insert(QString::fromUtf8("Ð"), "D");
69 		diacriticChars->insert(QString::fromUtf8("Ñ"), "N");
70 		diacriticChars->insert(QString::fromUtf8("Ò"), "O");
71 		diacriticChars->insert(QString::fromUtf8("Ó"), "O");
72 		diacriticChars->insert(QString::fromUtf8("Ô"), "O");
73 		diacriticChars->insert(QString::fromUtf8("Õ"), "O");
74 		diacriticChars->insert(QString::fromUtf8("Ö"), "O");
75 		diacriticChars->insert(QString::fromUtf8("Ø"), "O");
76 		diacriticChars->insert(QString::fromUtf8("Ù"), "U");
77 		diacriticChars->insert(QString::fromUtf8("Ú"), "U");
78 		diacriticChars->insert(QString::fromUtf8("Û"), "U");
79 		diacriticChars->insert(QString::fromUtf8("Ü"), "U");
80 		diacriticChars->insert(QString::fromUtf8("Ý"), "Y");
81 		diacriticChars->insert(QString::fromUtf8("ß"), "ss");
82 		diacriticChars->insert(QString::fromUtf8("à"), "a");
83 		diacriticChars->insert(QString::fromUtf8("á"), "a");
84 		diacriticChars->insert(QString::fromUtf8("â"), "a");
85 		diacriticChars->insert(QString::fromUtf8("ã"), "a");
86 		diacriticChars->insert(QString::fromUtf8("ä"), "a");
87 		diacriticChars->insert(QString::fromUtf8("å"), "a");
88 		diacriticChars->insert(QString::fromUtf8("æ"), "ae");
89 		diacriticChars->insert(QString::fromUtf8("ç"), "c");
90 		diacriticChars->insert(QString::fromUtf8("è"), "e");
91 		diacriticChars->insert(QString::fromUtf8("é"), "e");
92 		diacriticChars->insert(QString::fromUtf8("ê"), "e");
93 		diacriticChars->insert(QString::fromUtf8("ë"), "e");
94 		diacriticChars->insert(QString::fromUtf8("ì"), "i");
95 		diacriticChars->insert(QString::fromUtf8("í"), "i");
96 		diacriticChars->insert(QString::fromUtf8("î"), "i");
97 		diacriticChars->insert(QString::fromUtf8("ï"), "i");
98 		diacriticChars->insert(QString::fromUtf8("ð"), "o");
99 		diacriticChars->insert(QString::fromUtf8("ñ"), "n");
100 		diacriticChars->insert(QString::fromUtf8("ò"), "o");
101 		diacriticChars->insert(QString::fromUtf8("ó"), "o");
102 		diacriticChars->insert(QString::fromUtf8("ô"), "o");
103 		diacriticChars->insert(QString::fromUtf8("õ"), "o");
104 		diacriticChars->insert(QString::fromUtf8("ö"), "o");
105 		diacriticChars->insert(QString::fromUtf8("ø"), "o");
106 		diacriticChars->insert(QString::fromUtf8("ù"), "u");
107 		diacriticChars->insert(QString::fromUtf8("ú"), "u");
108 		diacriticChars->insert(QString::fromUtf8("û"), "u");
109 		diacriticChars->insert(QString::fromUtf8("ü"), "u");
110 		diacriticChars->insert(QString::fromUtf8("ý"), "y");
111 		diacriticChars->insert(QString::fromUtf8("ÿ"), "y");
112 	}
113 }
114 
convertSearchstring(const QString & originalString,Library::SearchModeMask mode,const QList<QChar> & ignoredChars)115 QString Library::Utils::convertSearchstring(const QString& originalString, Library::SearchModeMask mode,
116                                             const QList<QChar>& ignoredChars)
117 {
118 	if(mode == Library::SearchMode::None)
119 	{
120 		return originalString;
121 	}
122 
123 	if(originalString.isEmpty())
124 	{
125 		return ::Util::convertNotNull(originalString);
126 	}
127 
128 	auto& qCache = convertCache->at(+mode);
129 	if(qCache.contains(originalString))
130 	{
131 		return *(qCache[originalString]);
132 	}
133 
134 	if(diacriticChars->isEmpty())
135 	{
136 		initDiacriticChars();
137 	}
138 
139 	auto convertedString = originalString;
140 	if(mode & Library::CaseInsensitve)
141 	{
142 		convertedString = originalString.toLower();
143 	}
144 
145 	if(mode & Library::NoSpecialChars)
146 	{
147 		auto withoutSpecialChars = QString {};
148 		for(const auto& c : convertedString)
149 		{
150 			if(ignoredChars.contains(c) || c.isLetterOrNumber())
151 			{
152 				withoutSpecialChars.append(c);
153 			}
154 		}
155 
156 		convertedString = withoutSpecialChars;
157 	}
158 
159 	if(mode & Library::NoDiacriticChars)
160 	{
161 		auto withoutDiacriticChars = QString {};
162 
163 		for(const auto& c : convertedString)
164 		{
165 			const auto seq = QString(c);
166 			if(diacriticChars->contains(seq))
167 			{
168 				const auto& replacement = diacriticChars->value(seq);
169 				if(mode & Library::CaseInsensitve)
170 				{
171 					withoutDiacriticChars.append(replacement.toLower());
172 				}
173 
174 				else
175 				{
176 					withoutDiacriticChars.append(replacement);
177 				}
178 			}
179 
180 			else
181 			{
182 				withoutDiacriticChars.append(c);
183 			}
184 		}
185 
186 		convertedString = withoutDiacriticChars;
187 	}
188 
189 	const auto result = ::Util::convertNotNull(convertedString).trimmed();
190 
191 	if(qCache.isEmpty()){
192 		qCache.setMaxCost(10000);
193 	}
194 
195 	qCache.insert(originalString, new QString(result));
196 
197 	return result;
198 }
199 
convertSearchstring(const QString & str,Library::SearchModeMask mode)200 QString Library::Utils::convertSearchstring(const QString& str, Library::SearchModeMask mode)
201 {
202 	return convertSearchstring(str, mode, QList<QChar>());
203 }
204 
convertSearchstring(const QString & str)205 QString Library::Utils::convertSearchstring(const QString& str)
206 {
207 	return convertSearchstring(str, GetSetting(Set::Lib_SearchMode));
208 }
209