1 | // Copyright (C) 2018 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 "language.h" |
5 | |
6 | #include <QtCore/qtextstream.h> |
7 | #include <QtCore/QList> |
8 | |
9 | namespace language { |
10 | |
11 | using namespace Qt::StringLiterals; |
12 | |
13 | static Encoding encoding = Encoding::Utf8; |
14 | static Language _language = Language::Cpp; |
15 | |
16 | Language language() { return _language; } |
17 | |
18 | void setLanguage(Language l) |
19 | { |
20 | _language = l; |
21 | switch (_language) { |
22 | case Language::Cpp: |
23 | derefPointer = u"->"_s ; |
24 | listStart = '{'; |
25 | listEnd = '}'; |
26 | nullPtr = u"nullptr"_s ; |
27 | operatorNew = u"new "_s ; |
28 | qtQualifier = u"Qt::"_s ; |
29 | qualifier = u"::"_s ; |
30 | self = u""_s ; // for testing: change to "this->"; |
31 | eol = u";\n"_s ; |
32 | emptyString = u"QString()"_s ; |
33 | encoding = Encoding::Utf8; |
34 | break; |
35 | case Language::Python: |
36 | derefPointer = u"."_s ; |
37 | listStart = '['; |
38 | listEnd = ']'; |
39 | nullPtr = u"None"_s ; |
40 | operatorNew = u""_s ; |
41 | qtQualifier = u"Qt."_s ; |
42 | qualifier = u"."_s ; |
43 | self = u"self."_s ; |
44 | eol = u"\n"_s ; |
45 | emptyString = u"\"\""_s ; |
46 | encoding = Encoding::Unicode; |
47 | break; |
48 | } |
49 | } |
50 | |
51 | QString derefPointer; |
52 | char listStart; |
53 | char listEnd; |
54 | QString nullPtr; |
55 | QString operatorNew; |
56 | QString qtQualifier; |
57 | QString qualifier; |
58 | QString self; |
59 | QString eol; |
60 | QString emptyString; |
61 | |
62 | QString cppQualifier = u"::"_s ; |
63 | QString cppTrue = u"true"_s ; |
64 | QString cppFalse = u"false"_s ; |
65 | |
66 | QTextStream &operator<<(QTextStream &str, const qtConfig &c) |
67 | { |
68 | str << "QT_CONFIG(" << c.parameter() << ')'; |
69 | return str; |
70 | } |
71 | |
72 | QTextStream &operator<<(QTextStream &str, const openQtConfig &c) |
73 | { |
74 | str << "#if " << qtConfig(c.parameter()) << '\n'; |
75 | return str; |
76 | } |
77 | |
78 | QTextStream &operator<<(QTextStream &str, const closeQtConfig &c) |
79 | { |
80 | str << "#endif // " << qtConfig(c.parameter()) << '\n'; |
81 | return str; |
82 | } |
83 | |
84 | struct EnumLookup |
85 | { |
86 | int value; |
87 | QLatin1StringView valueString; |
88 | }; |
89 | |
90 | template <int N> |
91 | QLatin1StringView lookupEnum(const EnumLookup(&array)[N], int value, int defaultIndex = 0) |
92 | { |
93 | for (int i = 0; i < N; ++i) { |
94 | if (value == array[i].value) |
95 | return array[i].valueString; |
96 | } |
97 | auto defaultValue = array[defaultIndex].valueString; |
98 | qWarning("uic: Warning: Invalid enumeration value %d, defaulting to %s" , |
99 | value, defaultValue.data()); |
100 | return defaultValue; |
101 | } |
102 | |
103 | QString fixClassName(QString className) |
104 | { |
105 | if (language() == Language::Python) |
106 | className.replace(before: cppQualifier, after: "_"_L1 ); |
107 | return className; |
108 | } |
109 | |
110 | QLatin1StringView toolbarArea(int v) |
111 | { |
112 | static const EnumLookup toolBarAreas[] = |
113 | { |
114 | {.value: 0, .valueString: "NoToolBarArea"_L1 }, |
115 | {.value: 0x1, .valueString: "LeftToolBarArea"_L1 }, |
116 | {.value: 0x2, .valueString: "RightToolBarArea"_L1 }, |
117 | {.value: 0x4, .valueString: "TopToolBarArea"_L1 }, |
118 | {.value: 0x8, .valueString: "BottomToolBarArea"_L1 }, |
119 | {.value: 0xf, .valueString: "AllToolBarAreas"_L1 } |
120 | }; |
121 | return lookupEnum(array: toolBarAreas, value: v); |
122 | } |
123 | |
124 | QLatin1StringView sizePolicy(int v) |
125 | { |
126 | static const EnumLookup sizePolicies[] = |
127 | { |
128 | {.value: 0, .valueString: "Fixed"_L1 }, |
129 | {.value: 0x1, .valueString: "Minimum"_L1 }, |
130 | {.value: 0x4, .valueString: "Maximum"_L1 }, |
131 | {.value: 0x5, .valueString: "Preferred"_L1 }, |
132 | {.value: 0x3, .valueString: "MinimumExpanding"_L1 }, |
133 | {.value: 0x7, .valueString: "Expanding"_L1 }, |
134 | {.value: 0xD, .valueString: "Ignored"_L1 } |
135 | }; |
136 | return lookupEnum(array: sizePolicies, value: v, defaultIndex: 3); |
137 | } |
138 | |
139 | QLatin1StringView dockWidgetArea(int v) |
140 | { |
141 | static const EnumLookup dockWidgetAreas[] = |
142 | { |
143 | {.value: 0, .valueString: "NoDockWidgetArea"_L1 }, |
144 | {.value: 0x1, .valueString: "LeftDockWidgetArea"_L1 }, |
145 | {.value: 0x2, .valueString: "RightDockWidgetArea"_L1 }, |
146 | {.value: 0x4, .valueString: "TopDockWidgetArea"_L1 }, |
147 | {.value: 0x8, .valueString: "BottomDockWidgetArea"_L1 }, |
148 | {.value: 0xf, .valueString: "AllDockWidgetAreas"_L1 } |
149 | }; |
150 | return lookupEnum(array: dockWidgetAreas, value: v); |
151 | } |
152 | |
153 | QLatin1StringView paletteColorRole(int v) |
154 | { |
155 | static const EnumLookup colorRoles[] = |
156 | { |
157 | {.value: 0, .valueString: "WindowText"_L1 }, |
158 | {.value: 1, .valueString: "Button"_L1 }, |
159 | {.value: 2, .valueString: "Light"_L1 }, |
160 | {.value: 3, .valueString: "Midlight"_L1 }, |
161 | {.value: 4, .valueString: "Dark"_L1 }, |
162 | {.value: 5, .valueString: "Mid"_L1 }, |
163 | {.value: 6, .valueString: "Text"_L1 }, |
164 | {.value: 7, .valueString: "BrightText"_L1 }, |
165 | {.value: 8, .valueString: "ButtonText"_L1 }, |
166 | {.value: 9, .valueString: "Base"_L1 }, |
167 | {.value: 10, .valueString: "Window"_L1 }, |
168 | {.value: 11, .valueString: "Shadow"_L1 }, |
169 | {.value: 12, .valueString: "Highlight"_L1 }, |
170 | {.value: 13, .valueString: "HighlightedText"_L1 }, |
171 | {.value: 14, .valueString: "Link"_L1 }, |
172 | {.value: 15, .valueString: "LinkVisited"_L1 }, |
173 | {.value: 16, .valueString: "AlternateBase"_L1 }, |
174 | {.value: 17, .valueString: "NoRole"_L1 }, |
175 | {.value: 18, .valueString: "ToolTipBase"_L1 }, |
176 | {.value: 19, .valueString: "ToolTipText"_L1 }, |
177 | {.value: 20, .valueString: "PlaceholderText"_L1 }, |
178 | }; |
179 | return lookupEnum(array: colorRoles, value: v); |
180 | } |
181 | |
182 | // Helpers for formatting a character sequences |
183 | |
184 | // Format a special character like '\x0a' |
185 | static int formatEscapedNumber(QTextStream &str, ushort value, int base, int width, |
186 | char prefix = 0) |
187 | { |
188 | int length = 1 + width; |
189 | str << '\\'; |
190 | if (prefix) { |
191 | str << prefix; |
192 | ++length; |
193 | } |
194 | const auto oldPadChar = str.padChar(); |
195 | const auto oldFieldWidth = str.fieldWidth(); |
196 | const auto oldFieldAlignment = str.fieldAlignment(); |
197 | const auto oldIntegerBase = str.integerBase(); |
198 | str.setPadChar(u'0'); |
199 | str.setFieldWidth(width); |
200 | str.setFieldAlignment(QTextStream::AlignRight); |
201 | str.setIntegerBase(base); |
202 | str << value; |
203 | str.setIntegerBase(oldIntegerBase); |
204 | str.setFieldAlignment(oldFieldAlignment); |
205 | str.setFieldWidth(oldFieldWidth); |
206 | str.setPadChar(oldPadChar); |
207 | return length; |
208 | } |
209 | |
210 | static int formatSpecialCharacter(QTextStream &str, ushort value) |
211 | { |
212 | int length = 0; |
213 | switch (value) { |
214 | case '\\': |
215 | str << "\\\\" ; |
216 | length += 2; |
217 | break; |
218 | case '\"': |
219 | str << "\\\"" ; |
220 | length += 2; |
221 | break; |
222 | case '\n': |
223 | str << "\\n\"\n\"" ; |
224 | length += 5; |
225 | break; |
226 | default: |
227 | break; |
228 | } |
229 | return length; |
230 | } |
231 | |
232 | // Format a sequence of characters for C++ with special characters numerically |
233 | // escaped (non-raw string literals), wrappped at maxSegmentSize. FormattingTraits |
234 | // are used to transform characters into (unsigned) codes, which can be used |
235 | // for either normal escapes or Unicode code points as used in Unicode literals. |
236 | |
237 | enum : int { maxSegmentSize = 1024 }; |
238 | |
239 | template <Encoding e> |
240 | struct FormattingTraits |
241 | { |
242 | }; |
243 | |
244 | template <> |
245 | struct FormattingTraits<Encoding::Utf8> |
246 | { |
247 | static ushort code(char c) { return uchar(c); } |
248 | }; |
249 | |
250 | template <> |
251 | struct FormattingTraits<Encoding::Unicode> |
252 | { |
253 | static ushort code(QChar c) { return c.unicode(); } |
254 | }; |
255 | |
256 | template <Encoding e, class Iterator> |
257 | static void formatStringSequence(QTextStream &str, Iterator it, Iterator end, |
258 | const QString &indent, |
259 | int escapeIntegerBase, int escapeWidth, |
260 | char escapePrefix = 0) |
261 | { |
262 | str << '"'; |
263 | int length = 0; |
264 | while (it != end) { |
265 | const auto code = FormattingTraits<e>::code(*it); |
266 | if (code >= 0x80) { |
267 | length += formatEscapedNumber(str, code, escapeIntegerBase, escapeWidth, escapePrefix); |
268 | } else if (const int l = formatSpecialCharacter(str, code)) { |
269 | length += l; |
270 | } else if (code != '\r') { |
271 | str << *it; |
272 | ++length; |
273 | } |
274 | ++it; |
275 | if (it != end && length > maxSegmentSize) { |
276 | str << "\"\n" << indent << indent << '"'; |
277 | length = 0; |
278 | } |
279 | } |
280 | str << '"'; |
281 | } |
282 | |
283 | void _formatString(QTextStream &str, const QString &value, const QString &indent, |
284 | bool qString) |
285 | { |
286 | switch (encoding) { |
287 | // Special characters as 3 digit octal escapes (u8"\303\234mlaut") |
288 | case Encoding::Utf8: { |
289 | if (qString && _language == Language::Cpp) |
290 | str << "QString::fromUtf8(" ; |
291 | const QByteArray utf8 = value.toUtf8(); |
292 | formatStringSequence<Encoding::Utf8>(str, it: utf8.cbegin(), end: utf8.cend(), indent, |
293 | escapeIntegerBase: 8, escapeWidth: 3); |
294 | if (qString && _language == Language::Cpp) |
295 | str << ')'; |
296 | } |
297 | break; |
298 | // Special characters as 4 digit hex Unicode points (u8"\u00dcmlaut") |
299 | case Encoding::Unicode: |
300 | str << 'u'; // Python Unicode literal (would be UTF-16 in C++) |
301 | formatStringSequence<Encoding::Unicode>(str, it: value.cbegin(), end: value.cend(), indent, |
302 | escapeIntegerBase: 16, escapeWidth: 4, escapePrefix: 'u'); |
303 | break; |
304 | } |
305 | } |
306 | |
307 | QTextStream &operator<<(QTextStream &str, const repeat &r) |
308 | { |
309 | for (int i = 0; i < r.m_count; ++i) |
310 | str << r.m_char; |
311 | return str; |
312 | } |
313 | |
314 | startFunctionDefinition1::startFunctionDefinition1(const char *name, const QString ¶meterType, |
315 | const QString ¶meterName, |
316 | const QString &indent, |
317 | const char *returnType) : |
318 | m_name(name), m_parameterType(parameterType), m_parameterName(parameterName), |
319 | m_indent(indent), m_return(returnType) |
320 | { |
321 | } |
322 | |
323 | QTextStream &operator<<(QTextStream &str, const startFunctionDefinition1 &f) |
324 | { |
325 | switch (language()) { |
326 | case Language::Cpp: |
327 | str << (f.m_return ? f.m_return : "void" ) << ' ' << f.m_name << '(' |
328 | << f.m_parameterType; |
329 | if (f.m_parameterType.cend()->isLetter()) |
330 | str << ' '; |
331 | str << f.m_parameterName << ')' << '\n' << f.m_indent << "{\n" ; |
332 | break; |
333 | case Language::Python: |
334 | str << "def " << f.m_name << "(self, " << f.m_parameterName << "):\n" ; |
335 | break; |
336 | } |
337 | return str; |
338 | } |
339 | |
340 | endFunctionDefinition::endFunctionDefinition(const char *name) : m_name(name) |
341 | { |
342 | } |
343 | |
344 | QTextStream &operator<<(QTextStream &str, const endFunctionDefinition &f) |
345 | { |
346 | switch (language()) { |
347 | case Language::Cpp: |
348 | str << "} // " << f.m_name << "\n\n" ; |
349 | break; |
350 | case Language::Python: |
351 | str << "# " << f.m_name << "\n\n" ; |
352 | break; |
353 | } |
354 | return str; |
355 | } |
356 | |
357 | void _formatStackVariable(QTextStream &str, const char *className, QStringView varName, |
358 | bool withInitParameters) |
359 | { |
360 | switch (language()) { |
361 | case Language::Cpp: |
362 | str << className << ' ' << varName; |
363 | if (withInitParameters) |
364 | str << '('; |
365 | break; |
366 | case Language::Python: |
367 | str << varName << " = " << className << '('; |
368 | if (!withInitParameters) |
369 | str << ')'; |
370 | break; |
371 | } |
372 | } |
373 | |
374 | enum class OverloadUse { |
375 | Always, |
376 | WhenAmbiguousOrEmpty, // Use overload if |
377 | // - signal/slot is ambiguous |
378 | // - argument list is empty (chance of connecting mismatching T against const T &) |
379 | Never, |
380 | }; |
381 | |
382 | // Format a member function for a signal slot connection |
383 | static bool isConstRef(const QStringView &arg) |
384 | { |
385 | return arg.startsWith(c: u'Q') && arg != "QPoint"_L1 && arg != "QSize"_L1 ; |
386 | } |
387 | |
388 | static QString formatOverload(const QStringView ¶meters) |
389 | { |
390 | QString result = "qOverload<"_L1 ; |
391 | const auto args = QStringView{parameters}.split(sep: u','); |
392 | for (qsizetype i = 0, size = args.size(); i < size; ++i) { |
393 | const auto &arg = args.at(i); |
394 | if (i > 0) |
395 | result += u','; |
396 | const bool constRef = isConstRef(arg); |
397 | if (constRef) |
398 | result += "const "_L1 ; |
399 | result += arg; |
400 | if (constRef) |
401 | result += u'&'; |
402 | } |
403 | result += u'>'; |
404 | return result; |
405 | } |
406 | |
407 | static void formatMemberFnPtr(QTextStream &str, const SignalSlot &s, OverloadUse useQOverload) |
408 | { |
409 | const qsizetype parenPos = s.signature.indexOf(ch: u'('); |
410 | Q_ASSERT(parenPos >= 0); |
411 | const auto functionName = QStringView{s.signature}.left(n: parenPos); |
412 | |
413 | const auto parameters = QStringView{s.signature}.mid(pos: parenPos + 1, |
414 | n: s.signature.size() - parenPos - 2); |
415 | |
416 | const bool isAmbiguous = s.options.testFlag(flag: SignalSlotOption::Ambiguous); |
417 | bool withOverload = false; // just to silence the compiler |
418 | |
419 | switch (useQOverload) { |
420 | case OverloadUse::Always: |
421 | withOverload = true; |
422 | break; |
423 | case OverloadUse::Never: |
424 | withOverload = false; |
425 | break; |
426 | case OverloadUse::WhenAmbiguousOrEmpty: |
427 | withOverload = parameters.empty() || isAmbiguous; |
428 | break; |
429 | } |
430 | |
431 | if (withOverload) |
432 | str << formatOverload(parameters) << '('; |
433 | |
434 | str << '&' << s.className << "::" << functionName; |
435 | |
436 | if (withOverload) |
437 | str << ')'; |
438 | } |
439 | |
440 | static void formatMemberFnPtrConnection(QTextStream &str, |
441 | const SignalSlot &sender, |
442 | const SignalSlot &receiver) |
443 | { |
444 | str << "QObject::connect(" << sender.name << ", " ; |
445 | formatMemberFnPtr(str, s: sender, useQOverload: OverloadUse::Never); |
446 | str << ", " << receiver.name << ", " ; |
447 | formatMemberFnPtr(str, s: receiver, useQOverload: OverloadUse::WhenAmbiguousOrEmpty); |
448 | str << ')'; |
449 | } |
450 | |
451 | static void formatStringBasedConnection(QTextStream &str, |
452 | const SignalSlot &sender, |
453 | const SignalSlot &receiver) |
454 | { |
455 | str << "QObject::connect(" << sender.name << ", SIGNAL(" << sender.signature |
456 | << "), " << receiver.name << ", SLOT(" << receiver.signature << "))" ; |
457 | } |
458 | |
459 | void formatConnection(QTextStream &str, const SignalSlot &sender, const SignalSlot &receiver, |
460 | ConnectionSyntax connectionSyntax) |
461 | { |
462 | switch (language()) { |
463 | case Language::Cpp: |
464 | switch (connectionSyntax) { |
465 | case ConnectionSyntax::MemberFunctionPtr: |
466 | formatMemberFnPtrConnection(str, sender, receiver); |
467 | break; |
468 | case ConnectionSyntax::StringBased: |
469 | formatStringBasedConnection(str, sender, receiver); |
470 | break; |
471 | } |
472 | break; |
473 | case Language::Python: { |
474 | const auto paren = sender.signature.indexOf(ch: u'('); |
475 | auto senderSignature = QStringView{sender.signature}; |
476 | str << sender.name << '.' << senderSignature.left(n: paren); |
477 | // Signals like "QAbstractButton::clicked(checked=false)" require |
478 | // the parameter if it is used. |
479 | if (sender.options.testFlag(flag: SignalSlotOption::Ambiguous)) { |
480 | const QStringView parameters = |
481 | senderSignature.mid(pos: paren + 1, n: senderSignature.size() - paren - 2); |
482 | if (!parameters.isEmpty() && !parameters.contains(c: u',')) |
483 | str << "[\"" << parameters << "\"]" ; |
484 | } |
485 | str << ".connect(" << receiver.name << '.' |
486 | << QStringView{receiver.signature}.left(n: receiver.signature.indexOf(ch: u'(')) |
487 | << ')'; |
488 | } |
489 | break; |
490 | } |
491 | } |
492 | |
493 | QString boolValue(bool v) |
494 | { |
495 | switch (language()) { |
496 | case Language::Cpp: |
497 | return v ? cppTrue : cppFalse; |
498 | case Language::Python: |
499 | return v ? QStringLiteral("True" ) : QStringLiteral("False" ); |
500 | } |
501 | Q_UNREACHABLE(); |
502 | } |
503 | |
504 | static inline QString dot() { return QStringLiteral("." ); } |
505 | |
506 | QString enumValue(const QString &value) |
507 | { |
508 | if (language() == Language::Cpp || !value.contains(s: cppQualifier)) |
509 | return value; |
510 | QString fixed = value; |
511 | fixed.replace(before: cppQualifier, after: dot()); |
512 | return fixed; |
513 | } |
514 | |
515 | } // namespace language |
516 | |