mitkLocalStorageHandler.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 LOCALSTORAGEHANDLER_H_HEADER_INCLUDED_C1E6EA08
#define LOCALSTORAGEHANDLER_H_HEADER_INCLUDED_C1E6EA08

#include "mitkBaseRenderer.h"
#include "mitkCommon.h"
#include "mitkLevelWindow.h"
#include "mitkVtkPropRenderer.h"
#include <MitkCoreExports.h>

#include <itkObject.h>
#include <itkWeakPointer.h>

// Just included to get VTK version
#include <vtkConfigure.h>

class vtkWindow;
class vtkProp;

namespace mitk
{
  class BaseLocalStorageHandler
  {
  public:
    virtual ~BaseLocalStorageHandler() {}
    virtual void ClearLocalStorage(mitk::BaseRenderer *renderer, bool unregisterFromBaseRenderer = true) = 0;
  };

  template <class L>
  class LocalStorageHandler : public mitk::BaseLocalStorageHandler
  {
  protected:
    std::map<mitk::BaseRenderer *, L *> m_BaseRenderer2LS;

  public:
    void ClearLocalStorage(mitk::BaseRenderer *renderer, bool unregisterFromBaseRenderer = true) override
    {
      // MITK_INFO << "deleting a localstorage on a mapper request";
      if (unregisterFromBaseRenderer)
        renderer->UnregisterLocalStorageHandler(this);
      L *l = m_BaseRenderer2LS[renderer];
      m_BaseRenderer2LS.erase(renderer);
      delete l;
    }

    std::vector<mitk::BaseRenderer *> GetRegisteredBaseRenderer()
    {
      std::vector<mitk::BaseRenderer *> baserenderers;
      typename std::map<mitk::BaseRenderer *, L *>::iterator it;
      for (it = m_BaseRenderer2LS.begin(); it != m_BaseRenderer2LS.end(); ++it)
      {
        baserenderers.push_back(it->first);
      }
      return baserenderers;
    }

    L *GetLocalStorage(mitk::BaseRenderer *forRenderer)
    {
      L *l = m_BaseRenderer2LS[forRenderer];
      if (!l)
      {
        // MITK_INFO << "creating new localstorage";
        l = new L;
        m_BaseRenderer2LS[forRenderer] = l;
        forRenderer->RegisterLocalStorageHandler(this);
      }
      return l;
    }

    ~LocalStorageHandler() override
    {
      typename std::map<mitk::BaseRenderer *, L *>::iterator it;

      for (it = m_BaseRenderer2LS.begin(); it != m_BaseRenderer2LS.end(); it++)
      {
        (*it).first->UnregisterLocalStorageHandler(this);
        delete (*it).second;
      }

      m_BaseRenderer2LS.clear();
    }
  };

} // namespace mitk

#endif /* LOCALSTORAGEHANDLER_H_HEADER_INCLUDED_C1E6EA08 */