QmitkLesionTreeItem.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 QMITKLESIONTREEITEM_H
#define QMITKLESIONTREEITEM_H

// mitk semantic relations UI
#include "mitkLesionData.h"

// mitk semantic relations
#include <mitkSemanticTypes.h>

// qt
#include <QVariant>

// c++
#include <memory>
#include <vector>

/*
* @brief This class is used by custom tree models to create their tree items.
*        It provides functions to traverse and modify the tree.
*        Additionally it holds some 'LesionData' that is used to display lesion properties inside a tree view.
*/
class QmitkLesionTreeItem : public std::enable_shared_from_this<QmitkLesionTreeItem>
{
public:

  using ChildPointer = std::shared_ptr<QmitkLesionTreeItem>;
  using ChildConstPointer = std::shared_ptr<const QmitkLesionTreeItem>;
  using ParentPointer = std::weak_ptr<QmitkLesionTreeItem>;

  QmitkLesionTreeItem(mitk::LesionData lesionData = mitk::LesionData());

  ChildPointer GetChildInRow(int row) const { return m_Children.at(row); };
  ParentPointer GetParent() const { return m_ParentItem; };
  void SetParent(ParentPointer parent);
  mitk::LesionData& GetData() { return m_ItemData; };
  int GetRow() const;
  int GetRowOfChild(ChildConstPointer child) const;
  size_t ChildCount() const { return m_Children.size(); };
  void AddChild(ChildPointer child);
  void RemoveChild(ChildPointer child);
  void SetData(const mitk::LesionData& lesionData);

private:

  ParentPointer m_ParentItem;
  std::vector<ChildPointer> m_Children;
  mitk::LesionData m_ItemData;

};

Q_DECLARE_METATYPE(QmitkLesionTreeItem)
Q_DECLARE_METATYPE(QmitkLesionTreeItem*)

#endif // QMITKLESIONTREEITEM_H