QmitkDataStorageSimpleTreeModel.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef QmitkDataStorageSimpleTreeModel_h
#define QmitkDataStorageSimpleTreeModel_h
 
#include <MitkQtWidgetsExports.h>
 
// qt widgets module
#include <QmitkAbstractDataStorageModel.h>
 
class QmitkDataStorageTreeModelInternalItem;
 
class MITKQTWIDGETS_EXPORT QmitkDataStorageSimpleTreeModel : public QmitkAbstractDataStorageModel
{
  Q_OBJECT
 
public:
  QmitkDataStorageSimpleTreeModel(QObject *parent);
  ~QmitkDataStorageSimpleTreeModel() override;
 
  // override from 'QmitkAbstractDataStorageModel'
  /*
   * @brief See 'QmitkAbstractDataStorageModel'
   */
  void DataStorageChanged() override;
  /*
   * @brief See 'QmitkAbstractDataStorageModel'
   */
  void NodePredicateChanged() override;
  /*
   * @brief See 'QmitkAbstractDataStorageModel'
   */
  void NodeAdded(const mitk::DataNode *node) override;
  /*
   * @brief See 'QmitkAbstractDataStorageModel'
   */
  void NodeChanged(const mitk::DataNode *node) override;
  /*
   * @brief See 'QmitkAbstractDataStorageModel'
   */
  void NodeRemoved(const mitk::DataNode *node) override;
 
  // override pure virtual from 'QAbstractItemModel'
  QModelIndex index(int row, int column, const QModelIndex &parent = QModelIndex()) const override;
  QModelIndex parent(const QModelIndex &child) const override;
 
  int rowCount(const QModelIndex &parent = QModelIndex()) const override;
  int columnCount(const QModelIndex &parent = QModelIndex()) const override;
 
  QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
 
  bool setData(const QModelIndex &index, const QVariant &value, int role = Qt::EditRole) override;
 
  QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const override;
  Qt::ItemFlags flags(const QModelIndex &index) const override;
 
protected:
  using TreeItem = QmitkDataStorageTreeModelInternalItem;
 
private:
  void UpdateModelData();
  void AddNodeInternal(const mitk::DataNode *node);
 
  mitk::DataNode *GetParentNode(const mitk::DataNode *node) const;
 
  TreeItem *TreeItemFromIndex(const QModelIndex &index) const;
  QModelIndex IndexFromTreeItem(TreeItem *item) const;
 
  void ResetTree();
 
  TreeItem *m_Root;
 
  std::list<const TreeItem*> m_TreeItems;
};
 
#endif