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