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 // QMITKPATIENTTABLEMODEL_H