File indexing completed on 2024-05-19 15:46:52

 /*
     SPDX-FileCopyrightText: 2007 David Nolden <david.nolden.kdevelop@art-master.de>
 
     SPDX-License-Identifier: LGPL-2.0-or-later
 */
 
 #include "expandingtree.h"
 
 #include <QAbstractProxyModel>
 #include <QAbstractTextDocumentLayout>
 #include <QPainter>
 #include "expandingwidgetmodel.h"
 
 #include <util/path.h>
 #include <util/widgetcolorizer.h>
 
 using namespace KDevelop;
 
 ExpandingTree::ExpandingTree(QWidget* parent) : QTreeView(parent)
 {
     m_drawText.documentLayout()->setPaintDevice(this);
     setUniformRowHeights(false);
 }
 
 void ExpandingTree::setModel(QAbstractItemModel* model)
 {
     Q_ASSERT(!model || qobject_cast<const ExpandingWidgetModel*>(
                  qobject_cast<const QAbstractProxyModel*>(model)->sourceModel())
              );
     QTreeView::setModel(model);
 }
 
 void ExpandingTree::drawRow(QPainter* painter, const QStyleOptionViewItem& option, const QModelIndex& index) const
 {
     QTreeView::drawRow(painter, option, index);
 
     const auto* eModel = qobject_cast<const ExpandingWidgetModel*>(qobject_cast<const QAbstractProxyModel*>(model())->sourceModel());
     Q_ASSERT(eModel);
     const QModelIndex sourceIndex = eModel->mapToSource(index);
     if (eModel->isPartiallyExpanded(sourceIndex) != ExpandingWidgetModel::ExpansionType::NotExpanded) {
         QRect rect = eModel->partialExpandRect(sourceIndex);
         if (rect.isValid()) {
             QStyleOption opt;
             QAbstractTextDocumentLayout::PaintContext ctx;
 
             opt.rect = rect;
             style()->drawPrimitive(QStyle::PE_FrameLineEdit, &opt, painter);
 
             ctx.clip = QRectF(0, 0, rect.width(), rect.height());
             painter->setViewTransformEnabled(true);
             painter->translate(rect.left(), rect.top());
 
             m_drawText.setHtml(eModel->partialExpandText(sourceIndex));
             WidgetColorizer::convertDocumentToDarkTheme(&m_drawText);
             m_drawText.setPageSize(QSizeF(rect.width(), rect.height()));
             m_drawText.documentLayout()->draw(painter, ctx);
 
             painter->translate(-rect.left(), -rect.top());
         }
     }
 }
 
 int ExpandingTree::sizeHintForColumn(int column) const
 {
     return columnWidth(column);
 }
 
 void ExpandingTree::drawBranches(QPainter* painter, const QRect& rect, const QModelIndex& index) const
 {
     const auto& path = index.data(ProjectPathRole).value<Path>();
     if (path.isValid()) {
         const auto color = WidgetColorizer::colorForId(qHash(path), palette(), true);
         WidgetColorizer::drawBranches(this, painter, rect, index, color);
     }
     QTreeView::drawBranches(painter, rect, index);
 }
 
 #include "moc_expandingtree.cpp"