niftkDataNodeVisibilityTracker.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.

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

#ifndef niftkDataStorageVisibilityTracker_h
#define niftkDataStorageVisibilityTracker_h

#include "niftkCoreExports.h"

#include <itkObject.h>

#include <mitkDataNode.h>
#include <mitkDataStorage.h>

#include "niftkDataNodePropertyListener.h"

namespace mitk
{
class BaseRenderer;
}


namespace niftk
{

class NIFTKCORE_EXPORT DataNodeVisibilityTracker : public DataNodePropertyListener
{

public:

  mitkClassMacro(DataNodeVisibilityTracker, DataNodePropertyListener)
  mitkNewMacro1Param(DataNodeVisibilityTracker, const mitk::DataStorage::Pointer)

  
  void SetTrackedRenderer(const mitk::BaseRenderer* trackedRenderer);

  void SetManagedRenderers(const std::vector<const mitk::BaseRenderer*>& managedRenderers);

  void SetNodesToIgnore(const std::vector<mitk::DataNode*>& nodesToIgnore);

  bool IsIgnored(mitk::DataNode* node);

protected:

  DataNodeVisibilityTracker(const mitk::DataStorage::Pointer dataStorage);
  virtual ~DataNodeVisibilityTracker();

  DataNodeVisibilityTracker(const DataNodeVisibilityTracker&); // Purposefully not implemented.
  DataNodeVisibilityTracker& operator=(const DataNodeVisibilityTracker&); // Purposefully not implemented.

  virtual void OnNodeAdded(mitk::DataNode* node) override;

  virtual void OnPropertyChanged(mitk::DataNode* node, const mitk::BaseRenderer* renderer) override;

private:

  const mitk::BaseRenderer* m_TrackedRenderer;
  std::vector<const mitk::BaseRenderer*> m_ManagedRenderers;
  std::vector<mitk::DataNode*> m_NodesToIgnore;
};

}

#endif