FilteredTreeModel.cpp

Go to the documentation of this file.

// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2018 ISIS Rutherford Appleton Laboratory UKRI,
//   NScD Oak Ridge National Laboratory, European Spallation Source,
//   Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS
// SPDX - License - Identifier: GPL - 3.0 +
#include "MantidQtWidgets/Common/Batch/FilteredTreeModel.h"
#include "MantidQtWidgets/Common/Batch/StrictQModelIndices.h"
namespace MantidQt::MantidWidgets::Batch {
 
FilteredTreeModel::FilteredTreeModel(RowLocationAdapter rowLocationAdapter, QObject *parent)
    : QSortFilterProxyModel(parent), m_rowLocation(rowLocationAdapter) {
  resetPredicate();
}
 
void FilteredTreeModel::resetPredicate() {
  m_predicate = nullptr;
  invalidate();
}
 
bool FilteredTreeModel::isReset() const { return m_predicate == nullptr; }
 
void FilteredTreeModel::setPredicate(std::unique_ptr<RowPredicate> predicate) {
  m_predicate = std::move(predicate);
  invalidate();
}
 
RowLocation FilteredTreeModel::rowLocationAt(QModelIndex const &index) const {
  return m_rowLocation.atIndex(fromMainModel(index, *sourceModel()));
}
 
bool FilteredTreeModel::filterAcceptsRow(int row, const QModelIndex &parent) const {
  auto index = sourceModel()->index(row, 0, parent);
  if (index.isValid()) {
    if (m_predicate == nullptr || (*m_predicate)(rowLocationAt(index))) {
      return true;
    } else {
      int rows = sourceModel()->rowCount(index);
      for (auto r = 0; r < rows; r++)
        if (filterAcceptsRow(r, index))
          return true;
      return false;
    }
  } else {
    return false;
  }
}
} // namespace MantidQt::MantidWidgets::Batch