QmitkPatientTableModel.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 QmitkPatientTableModel_h
#define QmitkPatientTableModel_h
 
// semantic relations UI module
#include "QmitkAbstractSemanticRelationsStorageModel.h"
 
// semantic relations module
#include <mitkSemanticTypes.h>
#include <mitkSemanticRelationsDataStorageAccess.h>
 
// mitk core
#include <mitkDataNode.h>
 
// qt
#include <QPixmap>
#include <QStandardItemModel>
 
class QmitkPatientTableModel : public QmitkAbstractSemanticRelationsStorageModel
{
  Q_OBJECT
 
public:
 
  QmitkPatientTableModel(QObject* parent = nullptr);
  ~QmitkPatientTableModel() override;
 
  // overridden functions 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;
 
  QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const override;
  Qt::ItemFlags flags(const QModelIndex& index) const override;
 
  void SetNodeType(const std::string& nodeType);
 
protected:
 
  // the following functions have to be overridden...
  void NodePredicateChanged() override;
  // but are not implemented in this model
  void NodeAdded(const mitk::DataNode*) override { }
  void NodeChanged(const mitk::DataNode*) override { }
  void NodeRemoved(const mitk::DataNode*) override { }
  void SetData() override;
 
private:
 
  void SetHeaderModel();
  void SetPixmaps();
  void SetLesionPresences();
 
  void SetPixmapOfNode(const mitk::DataNode* dataNode, QPixmap* pixmapFromImage);
  void SetLesionPresenceOfNode(const mitk::DataNode* dataNode, bool lesionPresence);
  mitk::DataNode* GetCurrentDataNode(const QModelIndex &index) const;
 
  std::map<mitk::DataNode::ConstPointer, QPixmap> m_PixmapMap;
  std::map<mitk::DataNode::ConstPointer, bool> m_LesionPresence;
 
  mitk::SemanticTypes::ExaminationPeriodVector m_ExaminationPeriods;
  mitk::SemanticTypes::InformationTypeVector m_InformationTypes;
  mitk::SemanticRelationsDataStorageAccess::DataNodeVector m_CurrentDataNodes;
  std::string m_SelectedNodeType;
 
  QStandardItemModel* m_HeaderModel;
};
 
#endif