mitkReferenceCountWatcher.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 mitkReferenceCountWatcher_h
#define mitkReferenceCountWatcher_h
 
#include <itkCommand.h>
#include <MitkCoreExports.h>
#include <mitkCommon.h>
 
namespace mitk
{
  class ReferenceCountWatcher : public itk::Object
  {
  public:
    typedef itk::SimpleMemberCommand<ReferenceCountWatcher> CommandType;
 
    mitkClassMacroItkParent(ReferenceCountWatcher, itk::Object);
 
  protected:
    itk::Object *m_Object;
 
    std::string m_Comment;
 
    bool m_Deleted;
 
    CommandType::Pointer m_DeleteCommand;
 
  public:
    ReferenceCountWatcher(itk::Object *o, const char *comment = "")
      : m_Object(o), m_Comment(comment), m_Deleted(false), m_ObserverTag(0)
    {
      m_DeleteCommand = CommandType::New();
      m_DeleteCommand->SetCallbackFunction(this, &ReferenceCountWatcher::DeleteObserver);
      if (m_Object != nullptr)
        m_ObserverTag = m_Object->AddObserver(itk::DeleteEvent(), m_DeleteCommand);
      m_ReferenceCount = 0;
    }
 
    ~ReferenceCountWatcher() override
    {
      if ((m_Deleted == false) && (m_Object != nullptr))
      {
        m_Object->RemoveObserver(m_ObserverTag);
      }
    }
 
    int GetReferenceCount() const override
    {
      if (m_Object == nullptr)
        return -1;
      if (m_Deleted)
        return 0;
      return m_Object->GetReferenceCount();
    }
 
    itkGetStringMacro(Comment);
 
  protected:
    void DeleteObserver() { m_Deleted = true; }
    unsigned long m_ObserverTag;  
  };
}
 
#endif