mitkDataStorageSelection.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 mitkDataStorageSelection_h
#define mitkDataStorageSelection_h
 
#include <MitkDataTypesExtExports.h>
 
#include <mitkCommon.h>
#include <mitkDataStorage.h>
#include <mitkNodePredicateBase.h>
 
#include <set>
 
namespace mitk
{
  class BaseProperty;
  class PropertyList;
 
  class MITKDATATYPESEXT_EXPORT DataStorageSelection : public itk::Object
  {
  public:
    typedef std::vector<mitk::DataNode *> Nodes;
    typedef Message1<const mitk::DataNode *> DataNodeEvent;
 
    DataNodeEvent NodeChanged;
    DataNodeEvent NodeAdded;
    DataNodeEvent NodeRemoved;
    Message2<const mitk::DataNode *, const mitk::BaseProperty *> PropertyChanged;
 
    mitkClassMacroItkParent(DataStorageSelection, itk::Object);
 
    mitkNewMacro2Param(DataStorageSelection, DataStorage *, bool);
    mitkNewMacro3Param(DataStorageSelection, DataStorage *, NodePredicateBase *, bool);
 
  protected:
    DataStorageSelection(mitk::DataStorage *_DataStorage, bool _AutoAddNodes);
    DataStorageSelection(mitk::DataStorage *_DataStorage, mitk::NodePredicateBase *_Predicate, bool _AutoAddNodes);
 
  public:
    ~DataStorageSelection() override;
 
    mitk::DataStorage::Pointer GetDataStorage() const;
 
    mitk::NodePredicateBase::Pointer GetPredicate() const;
 
    unsigned int GetSize() const;
 
    mitk::DataNode::Pointer GetNode(unsigned int index) const;
 
    mitk::DataNode::Pointer GetNode() const;
 
    std::vector<mitk::DataNode *> GetNodes() const;
 
    bool DoesAutoAddNodes() const;
 
  public:
    DataStorageSelection &operator=(mitk::DataNode *node);
 
    DataStorageSelection &operator=(mitk::DataNode::Pointer node);
 
    virtual void SetDataStorage(mitk::DataStorage *_DataStorage);
 
    virtual void SetPredicate(mitk::NodePredicateBase *_Predicate);
 
    virtual void AddNode(const mitk::DataNode *node);
 
    virtual void RemoveNode(const mitk::DataNode *node);
 
    virtual void RemoveAllNodes();
 
    virtual void ObjectChanged(const itk::Object *caller, const itk::EventObject &event);
 
  protected:
    mitk::DataNode::Pointer FindNode(const mitk::BaseProperty *prop) const;
    mitk::DataNode::Pointer FindNode(const mitk::PropertyList *propList) const;
    void Reset();
    void RemoveListener(mitk::DataNode *node);
    void AddListener(mitk::DataNode *node);
 
  protected:
    mitk::DataStorage *m_DataStorage;
    unsigned int m_DataStorageDeletedTag;
    mitk::NodePredicateBase::Pointer m_Predicate;
    std::vector<mitk::DataNode *> m_Nodes;
    std::map<mitk::DataNode *, unsigned long> m_NodeModifiedObserverTags;
    std::map<mitk::PropertyList *, unsigned long> m_PropertyListModifiedObserverTags;
    std::map<mitk::PropertyList *, unsigned long> m_PropertyListDeletedObserverTags;
    std::map<mitk::BaseProperty *, unsigned long> m_PropertyModifiedObserverTags;
    std::map<mitk::BaseProperty *, unsigned long> m_PropertyDeletedObserverTags;
    bool m_SelfCall;
    bool m_AutoAddNodes;
  };
}
 
#endif