QmitkMultiLabelTreeModel.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 QmitkMultiLabelTreeModel_h
#define QmitkMultiLabelTreeModel_h
 
#include "mitkLabelSetImage.h"
 
// qt
#include <QAbstractItemModel>
 
#include "MitkSegmentationUIExports.h"
 
 
class QmitkMultiLabelSegTreeItem;
 
class MITKSEGMENTATIONUI_EXPORT QmitkMultiLabelTreeModel : public QAbstractItemModel
{
    Q_OBJECT
 
public:
  using LabelValueType = mitk::LabelSetImage::LabelValueType;
  using GroupIndexType = mitk::LabelSetImage::GroupIndexType;
 
  QmitkMultiLabelTreeModel(QObject *parent = nullptr);
  ~QmitkMultiLabelTreeModel() override;
 
  void SetSegmentation(mitk::LabelSetImage* segmentation);
  const mitk::LabelSetImage* GetSegmentation() const;
 
  Qt::ItemFlags flags(const QModelIndex &index) 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) const override;
  int rowCount(const QModelIndex &parent = QModelIndex()) const override;
  int columnCount(const QModelIndex &parent = QModelIndex()) const override;
 
  QModelIndex index(int row, int column, const QModelIndex &parent = QModelIndex()) const override;
  QModelIndex parent(const QModelIndex &child) const override;
 
  QModelIndex indexOfLabel(mitk::Label::PixelType labelValue) const;
  QModelIndex indexOfGroup(mitk::LabelSetImage::GroupIndexType groupIndex) const;
  QModelIndex ClosestLabelInstanceIndex(const QModelIndex& currentIndex) const;
  QModelIndex FirstLabelInstanceIndex(const QModelIndex& currentIndex) const;
 
  //or instance node). If current index is at the end, an invalid index is returned.*/
  //QModelIndex PrevLabelInstanceIndex(const QModelIndex& currentIndex) const;
 
  std::vector <LabelValueType> GetLabelsInSubTree(const QModelIndex& currentIndex) const;
 
  std::vector <LabelValueType> GetLabelInstancesOfSameLabelClass(const QModelIndex& currentIndex) const;
 
  enum TableColumns
  {
    NAME_COL = 0,
    LOCKED_COL,
    COLOR_COL,
    VISIBLE_COL
  };
 
  enum ItemModelRole
  {
    LabelDataRole = 64,
    LabelValueRole = 65,
    LabelInstanceDataRole = 66,
    LabelInstanceValueRole = 67,
    GroupIDRole = 68
  };
 
  bool GetAllowVisibilityModification() const;
  bool GetAllowLockModification() const;
 
public Q_SLOTS:
  void SetAllowVisibilityModification(bool vmod);
  void SetAllowLockModification(bool lmod);
 
Q_SIGNALS:
  void dataAvailable();
  void modelChanged();
 
protected:
  void OnLabelAdded(LabelValueType labelValue);
  void OnLabelModified(LabelValueType labelValue);
  void OnLabelRemoved(LabelValueType labelValue);
  void OnGroupAdded(GroupIndexType groupIndex);
  void OnGroupModified(GroupIndexType groupIndex);
  void OnGroupRemoved(GroupIndexType groupIndex);
 
private:
  void AddObserver();
  void RemoveObserver();
 
  void UpdateInternalTree();
  void GenerateInternalGroupTree(unsigned int layerID, QmitkMultiLabelSegTreeItem* layerItem);
  QmitkMultiLabelSegTreeItem* GenerateInternalTree();
 
  /* builds a hierarchical tree model for the image statistics
  1. Level: Image
  --> 2. Level: Mask [if exist]
      --> 3. Level: Timestep [if >1 exist] */
  void BuildHierarchicalModel();
 
  mitk::LabelSetImage::Pointer m_Segmentation;
 
  std::mutex m_Mutex;
  std::unique_ptr<QmitkMultiLabelSegTreeItem> m_RootItem;
 
  bool m_Observed;
  bool m_ShowGroups = true;
 
  bool m_ShowVisibility = true;
  bool m_ShowLock = true;
  bool m_ShowOther = false;
 
  bool m_AllowVisibilityModification = true;
  bool m_AllowLockModification = true;
};
 
#endif