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