QmitkDataStorageListModel.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 QmitkDataStorageListModel_h
#define QmitkDataStorageListModel_h

#include <MitkQtWidgetsExports.h>

// MITK
#include "mitkDataStorage.h"
#include "mitkNodePredicateBase.h"

// Qt
#include <QAbstractListModel>


class MITKQTWIDGETS_EXPORT QmitkDataStorageListModel : public QAbstractListModel
{
public:
  QmitkDataStorageListModel(mitk::DataStorage *dataStorage = nullptr,
                            mitk::NodePredicateBase::Pointer pred = nullptr,
                            QObject *parent = nullptr);

  ~QmitkDataStorageListModel() override;

  void SetDataStorage(mitk::DataStorage::Pointer dataStorage);

  mitk::DataStorage *GetDataStorage() const;

  void SetPredicate(mitk::NodePredicateBase *pred);

  mitk::NodePredicateBase *GetPredicate() const;

  std::vector<mitk::DataNode *> GetDataNodes() const;

  mitk::DataNode::Pointer getNode(const QModelIndex &index) const;

  QModelIndex getIndex(const mitk::DataNode *node) const;

  Qt::ItemFlags flags(const QModelIndex &index) const override;

  QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;

  QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const override;

  int rowCount(const QModelIndex &parent = QModelIndex()) const override;

  virtual void OnDataStorageNodeAdded(const mitk::DataNode *node);

  virtual void OnDataStorageNodeRemoved(const mitk::DataNode *node);

  virtual void OnDataNodeModified(const itk::Object *caller, const itk::EventObject &event);

  virtual void OnDataModified(const itk::Object *caller, const itk::EventObject &event);

  virtual void OnDataStorageDeleted(const itk::Object *caller, const itk::EventObject &event);

protected:
  void reset();

  void AddNodeToInternalList(mitk::DataNode *node);

  void RemoveNodeFromInternalList(mitk::DataNode *node);

  void ClearInternalNodeList();

private:
  enum OBSERVER_TUPLE_NAMES
  {
    NODE = 0,
    NODE_OBSERVER = 1,
    DATA_OBSERVER = 2,
  };

  mitk::NodePredicateBase::Pointer m_NodePredicate;

  mitk::DataStorage *m_DataStorage;

  unsigned long m_DataStorageDeleteObserverTag;

  std::vector<std::tuple<mitk::DataNode *, unsigned long, unsigned long>> m_NodesAndObserverTags;

  bool m_BlockEvents;
};

#endif /* QMITKDATASTORAGELISTMODEL_H_ */