mitkDataStorageSelection.h

Go to the documentation of this file.

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

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 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:
    virtual ~DataStorageSelection();

    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 // mitkDataStorageSelection_h