diff options
Diffstat (limited to 'sources/shiboken6/ApiExtractor/textstream.cpp')
| -rw-r--r-- | sources/shiboken6/ApiExtractor/textstream.cpp | 210 |
1 files changed, 210 insertions, 0 deletions
diff --git a/sources/shiboken6/ApiExtractor/textstream.cpp b/sources/shiboken6/ApiExtractor/textstream.cpp new file mode 100644 index 000000000..74b43b6da --- /dev/null +++ b/sources/shiboken6/ApiExtractor/textstream.cpp @@ -0,0 +1,210 @@ +/**************************************************************************** +** +** Copyright (C) 2020 The Qt Company Ltd. +** Contact: https://www.qt.io/licensing/ +** +** This file is part of Qt for Python. +** +** $QT_BEGIN_LICENSE:GPL-EXCEPT$ +** Commercial License Usage +** Licensees holding valid commercial Qt licenses may use this file in +** accordance with the commercial license agreement provided with the +** Software or, alternatively, in accordance with the terms contained in +** a written agreement between you and The Qt Company. For licensing terms +** and conditions see https://www.qt.io/terms-conditions. For further +** information use the contact form at https://www.qt.io/contact-us. +** +** GNU General Public License Usage +** Alternatively, this file may be used under the terms of the GNU +** General Public License version 3 as published by the Free Software +** Foundation with exceptions as appearing in the file LICENSE.GPL3-EXCEPT +** included in the packaging of this file. Please review the following +** information to ensure the GNU General Public License requirements will +** be met: https://www.gnu.org/licenses/gpl-3.0.html. +** +** $QT_END_LICENSE$ +** +****************************************************************************/ + +#include "textstream.h" + +#include <cstring> + +TextStream::TextStream(QIODevice *device, Language l) : + m_str(device), m_language(l) +{ +} + +TextStream::TextStream(QString *string, Language l) : + m_str(string), m_language(l) +{ +} + +TextStream::TextStream(QByteArray *array, Language l) : + m_str(array), m_language(l) +{ +} + +TextStream::~TextStream() = default; + +QChar TextStream::lastChar() const +{ + auto s = m_str.string(); + return s != nullptr && !s->isEmpty() ? *(s->crbegin()) : QChar(); +} + +void TextStream::setIndentation(int i) +{ + Q_ASSERT(i >= 0); + m_indentation = i; +} + +void TextStream::outdent(int n) +{ + m_indentation -= n; + Q_ASSERT(m_indentation >= 0); +} + +void TextStream::putRepetitiveChars(char c, int count) +{ + if (count > 0) { + for (int i = 0; i < count; ++i) { + const int ofw = m_str.fieldWidth(); + m_str.setFieldWidth(0); + m_str << c; + m_str.setFieldWidth(ofw); + } + } +} + +void TextStream::setLastCharClass(CharClass c) +{ + m_lastCharClass = c; +} + +void TextStream::writeIndent() +{ + putRepetitiveChars(' ', m_indentation * m_tabWidth); +} + +// Indent handling: If the last character was a new line +// and the upcoming one is none, indent the stream +// Special case for C++ : If the upcoming char is a '#', we don't +// indent (preprocessor directive). + +template <class Char> +static TextStream::CharClass charClassHelper(Char c) +{ + switch (c) { + case '\n': + return TextStream::CharClass::NewLine; + case '#': + return TextStream::CharClass::Hash; + default: + break; + } + return TextStream::CharClass::Other; +} + +static inline TextStream::CharClass charClass(char c) +{ return charClassHelper(c); } + +static inline TextStream::CharClass charClass(QChar c) +{ return charClassHelper(c.unicode()); } + +void TextStream::checkIndent(CharClass upComingCharClass) +{ + if (m_indentationEnabled && m_lastCharClass == CharClass::NewLine + && (upComingCharClass != CharClass::NewLine + && (m_language != Language::Cpp || upComingCharClass != CharClass::Hash))) { + writeIndent(); + } + m_lastCharClass = upComingCharClass; +} + +template <class Char> +void TextStream::putCharHelper(Char c) +{ + checkIndent(charClass(c)); + m_str << c; +} + +void TextStream::putString(QStringView v) +{ + if (v.isEmpty()) + return; + if (v.contains(u'\n')) { + for (auto c : v) + putCharHelper(c); + } else { + // If there is no newline, write as a blob. This is important to make + // field formatting (alignment/width) working, else each char will be + // considered a field. + checkIndent(charClass(*v.cbegin())); + m_str << v; + m_lastCharClass = CharClass::Other; + } +} + +void TextStream::putString(const char *s) +{ + const char firstChar = *s; + if (firstChar == '\0') + return; + if (std::strchr(s, '\n') != nullptr) { // See above + for ( ; *s; ++s) + putCharHelper(*s); + } else { + checkIndent(charClass(firstChar)); + m_str << s; + m_lastCharClass = CharClass::Other; + } +} + +void TextStream::putInt(int t) +{ + checkIndent(CharClass::Other); + m_str << t; +} + +void TextStream::putSizeType(qsizetype t) +{ + checkIndent(CharClass::Other); + m_str << t; +} + +StringStream::StringStream(Language l) : TextStream(&m_buffer, l) +{ +} + +void StringStream::clear() +{ + m_buffer.clear(); + setLastCharClass(CharClass::NewLine); +} + +void indent(TextStream &s) +{ + s.indent(); +} + +void outdent(TextStream &s) +{ + s.outdent(); +} + +void enableIndent(TextStream &s) +{ + s.setIndentationEnabled(true); +} + +void disableIndent(TextStream &s) +{ + s.setIndentationEnabled(false); +} + +void ensureEndl(TextStream &s) +{ + if (s.lastChar() != QLatin1Char('\n')) + s << '\n'; +} |