QmitkDataStorageComboBox.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 QmitkDataStorageComboBox_h
#define QmitkDataStorageComboBox_h

#include <MitkQtWidgetsExports.h>

// Own Includes
#include "mitkDataNode.h"
#include "mitkDataStorage.h"
#include "mitkNodePredicateBase.h"
#include "mitkWeakPointer.h"

// Toolkit Includes
#include <QComboBox>
#include <map>

class MITKQTWIDGETS_EXPORT QmitkDataStorageComboBox : public QComboBox
{
  Q_OBJECT

public:
  QmitkDataStorageComboBox(QWidget *parent = nullptr, bool autoSelectNewNodes = false);

  QmitkDataStorageComboBox(mitk::DataStorage *dataStorage,
                           const mitk::NodePredicateBase *predicate,
                           QWidget *parent = nullptr,
                           bool autoSelectNewNodes = false);

  ~QmitkDataStorageComboBox() override;

  virtual int Find(const mitk::DataNode *dataNode) const;

public:
  mitk::DataStorage::Pointer GetDataStorage() const;
  const mitk::NodePredicateBase::ConstPointer GetPredicate() const;
  virtual mitk::DataNode::Pointer GetNode(int index) const;
  virtual mitk::DataNode::Pointer GetSelectedNode() const;
  mitk::DataStorage::SetOfObjects::ConstPointer GetNodes() const;
  virtual bool GetAutoSelectNewItems();

public:
  void SetDataStorage(mitk::DataStorage *dataStorage);
  void SetPredicate(const mitk::NodePredicateBase *predicate);
  virtual void AddNode(const mitk::DataNode *dataNode);
  virtual void RemoveNode(int index);
  virtual void RemoveNode(const mitk::DataNode *dataNode);
  virtual void SetNode(int index, const mitk::DataNode *dataNode);
  virtual void SetNode(const mitk::DataNode *dataNode, const mitk::DataNode *otherDataNode);
  virtual void SetAutoSelectNewItems(bool autoSelectNewItems);
  virtual void OnPropertyListChanged(const itk::Object *caller, const itk::EventObject &event);

signals:
  void OnSelectionChanged(const mitk::DataNode *);

protected:
  bool HasIndex(unsigned int index) const;

protected slots:
  void OnCurrentIndexChanged(int);

public slots:
  void SetSelectedNode(const mitk::DataNode::Pointer& node);

protected:
  virtual void InsertNode(int index, const mitk::DataNode *dataNode);

  void Init();

  virtual void Reset();

  void RemoveNodeAndPropertyLists(int index);

  virtual void UpdateComboBoxText(const mitk::PropertyList*);

protected:
  mitk::WeakPointer<mitk::DataStorage> m_DataStorage;

  mitk::NodePredicateBase::ConstPointer m_Predicate;

  std::vector<mitk::DataNode *> m_Nodes;

  std::vector<long> m_DataNodePropertyListObserverTags;

  std::vector<long> m_BaseDatapropertyListObserverTags;

  bool m_BlockEvents;

  bool m_AutoSelectNewNodes;
};

#endif // QmitkDataStorageComboBox_h