niftkDataStorageCheckableComboBox.h

Go to the documentation of this file.

/*=============================================================================

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). All rights reserved.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

  See LICENSE.txt in the top level directory for details.

=============================================================================*/

/*=============================================================================
  This file was copy-pasted from MITK and modified.
  MITK copyright statement is below.
=============================================================================*/

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/

#ifndef niftkDataStorageCheckableComboBox_h
#define niftkDataStorageCheckableComboBox_h

#include "niftkCoreGuiExports.h"
#include <ctkCheckableComboBox.h>
#include <mitkDataStorage.h>
#include <mitkDataNode.h>
#include <mitkWeakPointer.h>
#include <mitkNodePredicateBase.h>
#include <QComboBox>
#include <map>

namespace niftk
{

class NIFTKCOREGUI_EXPORT DataStorageCheckableComboBox : public ctkCheckableComboBox
{
  //#CLASS-MACROS,FRIENDS
  Q_OBJECT

  //#CTORS/DTOR
  public:
    DataStorageCheckableComboBox(QWidget* parent = 0, bool _AutoSelectNewNodes=false);

    DataStorageCheckableComboBox( mitk::DataStorage* _DataStorage,
                                  const mitk::NodePredicateBase* _Predicate,
                                  QWidget* parent = 0, bool _AutoSelectNewNodes = false);

    ~DataStorageCheckableComboBox();

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

  //#PUBLIC GETTER
  public:
    mitk::DataStorage::Pointer GetDataStorage() const;
    const mitk::NodePredicateBase::ConstPointer GetPredicate() const;
    virtual mitk::DataNode::Pointer GetNode(int index) const;
    virtual std::vector<mitk::DataNode*> GetSelectedNodes() const;
    mitk::DataStorage::SetOfObjects::ConstPointer GetNodes() const;
    virtual bool GetAutoSelectNewItems();

  //#PUBLIC SETTER
  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 OnDataNodeDeleteOrModified(const itk::Object *caller, const itk::EventObject &event);

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

  //#PROTECTED GETTER
  protected:
    bool HasIndex(unsigned int index) const;

  //#PROTECTED SETTER
  protected slots:

  //#PUBLIC SETTER
  public slots:
    void SetSelectedNode(mitk::DataNode::Pointer item);

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

    void Init();

    virtual void Reset();

  protected:
    //#PROTECTED MEMBER VARS
    mitk::WeakPointer<mitk::DataStorage> m_DataStorage;

    mitk::NodePredicateBase::ConstPointer m_Predicate;

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

    std::vector<long> m_NodesModifiedObserverTags;

    std::vector<long> m_NodesDeleteObserverTags;

    std::map<mitk::DataNode*, const mitk::BaseProperty*> m_PropertyToNode;

    bool m_BlockEvents;

    bool m_AutoSelectNewNodes;
};

} // end namespace

#endif