1 files changed, 58 insertions, 0 deletions
diff --git a/src/modelutils.cpp b/src/modelutils.cpp
new file mode 100644
index 00000000..43aa6b99
--- /dev/null
+++ b/src/modelutils.cpp
@@ -0,0 +1,58 @@
+#include "modelutils.h"
+
+#include <QAbstractProxyModel>
+
+QModelIndex indexModelToView(const QModelIndex& index, const QAbstractItemView* view)
+{
+  // we need to stack the proxy
+  std::vector<QAbstractProxyModel*> proxies;
+  {
+    auto* currentModel = view->model();
+    while (auto* proxy = qobject_cast<QAbstractProxyModel*>(currentModel)) {
+      proxies.push_back(proxy);
+      currentModel = proxy->sourceModel();
+    }
+  }
+
+  if (proxies.empty() || proxies.back()->sourceModel() != index.model()) {
+    return QModelIndex();
+  }
+
+  auto qindex = index;
+  for (auto rit = proxies.rbegin(); rit != proxies.rend(); ++rit) {
+    qindex = (*rit)->mapFromSource(qindex);
+  }
+
+  return qindex;
+}
+
+QModelIndexList indexModelToView(const QModelIndexList& index, const QAbstractItemView* view)
+{
+  QModelIndexList result;
+  for (auto& idx : index) {
+    result.append(indexModelToView(idx, view));
+  }
+  return result;
+}
+
+QModelIndex indexViewToModel(const QModelIndex& index, const QAbstractItemModel* model)
+{
+  if (index.model() == model) {
+    return index;
+  }
+  else if (auto* proxy = qobject_cast<const QAbstractProxyModel*>(index.model())) {
+    return indexViewToModel(proxy->mapToSource(index), model);
+  }
+  else {
+    return QModelIndex();
+  }
+}
+
+QModelIndexList indexViewToModel(const QModelIndexList& index, const QAbstractItemModel* model)
+{
+  QModelIndexList result;
+  for (auto& idx : index) {
+    result.append(indexViewToModel(idx, model));
+  }
+  return result;
+}