pqFlatTreeView.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Kitware Inc.
// SPDX-FileCopyrightText: Copyright (c) Sandia Corporation
// SPDX-License-Identifier: BSD-3-Clause

#ifndef pqFlatTreeView_h
#define pqFlatTreeView_h

#include "pqWidgetsModule.h"
#include <QAbstractScrollArea>
#include <QModelIndex>          // Needed for return type
#include <QStyleOptionViewItem> // Needed for return type

class pqFlatTreeViewItem;
class pqFlatTreeViewItemRows;
class pqFlatTreeViewInternal;

class QAbstractItemModel;
class QColor;
class QFontMetrics;
class QHeaderView;
class QItemSelection;
class QItemSelectionModel;
class QPoint;
class QRect;

class PQWIDGETS_EXPORT pqFlatTreeView : public QAbstractScrollArea
{
  Q_OBJECT

public:
  enum SelectionBehavior
  {
    SelectItems,
    SelectRows,
    SelectColumns
  };

  enum SelectionMode
  {
    NoSelection,
    SingleSelection,
    ExtendedSelection
  };

  pqFlatTreeView(QWidget* parent = nullptr);
  ~pqFlatTreeView() override;

  bool eventFilter(QObject* object, QEvent* e) override;

  QAbstractItemModel* getModel() const { return this->Model; }
  void setModel(QAbstractItemModel* model);

  QModelIndex getRootIndex() const;
  void setRootIndex(const QModelIndex& index);

  QItemSelectionModel* getSelectionModel() const { return this->Selection; }
  void setSelectionModel(QItemSelectionModel* selectionModel);

  SelectionBehavior getSelectionBehavior() const { return this->Behavior; }
  void setSelectionBehavior(SelectionBehavior behavior);

  SelectionMode getSelectionMode() const { return this->Mode; }
  void setSelectionMode(SelectionMode mode);

  QHeaderView* getHeader() const { return this->HeaderView; }
  void setHeader(QHeaderView* headerView);

  bool isColumnSizeManaged() const { return this->ManageSizes; }

  void setColumnSizeManaged(bool managed);

  int getIconSize() const;
  void setIconSize(int iconSize);

  bool isIndexHidden(const QModelIndex& index) const;
  void getVisibleRect(const QModelIndex& index, QRect& area) const;
  QModelIndex getIndexVisibleAt(const QPoint& point) const;
  QModelIndex getIndexCellAt(const QPoint& point) const;
  void getSelectionIn(const QRect& rect, QItemSelection& items) const;

  bool isIndexExpanded(const QModelIndex& index) const;
  QModelIndex getNextVisibleIndex(
    const QModelIndex& index, const QModelIndex& root = QModelIndex()) const;
  QModelIndex getRelativeIndex(const QString& id, const QModelIndex& root = QModelIndex()) const;
  void getRelativeIndexId(
    const QModelIndex& index, QString& id, const QModelIndex& root = QModelIndex()) const;

  bool startEditing(const QModelIndex& index);
  void finishEditing();
  void cancelEditing();

Q_SIGNALS:
  void activated(const QModelIndex& index);
  void clicked(const QModelIndex& index);

public Q_SLOTS:
  void reset();
  void selectAll();
  void setCurrentIndex(const QModelIndex& index);
  void expandAll();
  void expand(const QModelIndex& index);
  void collapse(const QModelIndex& index);
  void scrollTo(const QModelIndex& index);

protected Q_SLOTS:
  void insertRows(const QModelIndex& parent, int start, int end);
  void startRowRemoval(const QModelIndex& parent, int start, int end);
  void finishRowRemoval(const QModelIndex& parent, int start, int end);
  void insertColumns(const QModelIndex& parent, int start, int end);
  void startColumnRemoval(const QModelIndex& parent, int start, int end);
  void finishColumnRemoval(const QModelIndex& parent, int start, int end);
  void updateData(const QModelIndex& topLeft, const QModelIndex& bottomRight);

protected: // NOLINT(readability-redundant-access-specifiers)
  void keyPressEvent(QKeyEvent* e) override;
  void keyboardSearch(const QString& search);

  void mousePressEvent(QMouseEvent* e) override;
  void mouseMoveEvent(QMouseEvent* e) override;
  void mouseReleaseEvent(QMouseEvent* e) override;
  void mouseDoubleClickEvent(QMouseEvent* e) override;

  bool event(QEvent* e) override;

  int horizontalOffset() const;
  int verticalOffset() const;

  void resizeEvent(QResizeEvent* e) override;
  bool viewportEvent(QEvent* e) override;
  void paintEvent(QPaintEvent* e) override;
  QStyleOptionViewItem getViewOptions() const;

private Q_SLOTS:
  void handleSectionResized(int index, int oldSize, int newSize);
  void handleSectionMoved(int index, int oldVisual, int newVisual);

  void changeCurrent(const QModelIndex& current, const QModelIndex& previous);
  void changeCurrentRow(const QModelIndex& current, const QModelIndex& previous);
  void changeCurrentColumn(const QModelIndex& current, const QModelIndex& previous);
  void changeSelection(const QItemSelection& selected, const QItemSelection& deselected);

private: // NOLINT(readability-redundant-access-specifiers)
  void resetRoot();
  void resetPreferredSizes();

  void layoutEditor();
  void layoutItems();
  void layoutItem(pqFlatTreeViewItem* item, int& point, const QFontMetrics& fm);
  int getDataWidth(const QModelIndex& index, const QFontMetrics& fm) const;
  int getWidthSum(pqFlatTreeViewItem* item, int column) const;
  bool updateContentsWidth();
  void updateScrollBars();
  void addChildItems(pqFlatTreeViewItem* item, int parentChildCount);

  bool getIndexRowList(const QModelIndex& index, pqFlatTreeViewItemRows& rowList) const;
  pqFlatTreeViewItem* getItem(const pqFlatTreeViewItemRows& rowList) const;
  pqFlatTreeViewItem* getItem(const QModelIndex& index) const;
  pqFlatTreeViewItem* getItemAt(int contentsY) const;
  pqFlatTreeViewItem* getNextItem(pqFlatTreeViewItem* item) const;
  pqFlatTreeViewItem* getNextVisibleItem(pqFlatTreeViewItem* item) const;
  pqFlatTreeViewItem* getPreviousVisibleItem(pqFlatTreeViewItem* item) const;
  pqFlatTreeViewItem* getLastVisibleItem() const;

  void expandItem(pqFlatTreeViewItem* item);

  void getSelectionIn(
    const QModelIndex& topLeft, const QModelIndex& bottomRight, QItemSelection& items) const;

  void drawBranches(QPainter& painter, pqFlatTreeViewItem* item, int halfIndent,
    const QColor& branchColor, const QColor& expandColor, const QStyleOptionViewItem& options);
  bool drawDecoration(QPainter& painter, int px, int py, const QModelIndex& index,
    const QStyleOptionViewItem& options, int itemHeight);
  void drawData(QPainter& painter, int px, int py, const QModelIndex& index,
    const QStyleOptionViewItem& options, int itemHeight, int itemWidth, int columnWidth,
    bool selected);
  void drawFocus(
    QPainter& painter, const QRect& cell, const QStyleOptionViewItem& options, bool selected);

  QAbstractItemModel* Model;
  QItemSelectionModel* Selection;
  SelectionBehavior Behavior;
  SelectionMode Mode;
  QHeaderView* HeaderView;
  pqFlatTreeViewItem* Root;
  pqFlatTreeViewInternal* Internal;
  int IconSize;
  int IndentWidth;
  int ContentsWidth;
  int ContentsHeight;
  int TextMargin;
  int DoubleTextMargin;
  bool FontChanged;
  bool ManageSizes;
  bool InUpdateWidth;
  bool HeaderOwned;
  bool SelectionOwned;

  static int PipeLength;
};

#endif