dox/Common/Core/vtkSMPThreadLocalObject.h

Go to the documentation of this file.

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

  Program:   Visualization Toolkit
  Module:    vtkSMPThreadLocalObject.h

  Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
  All rights reserved.
  See Copyright.txt or http://www.kitware.com/Copyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notice for more information.

=========================================================================*/
#ifndef __vtkSMPThreadLocalObject_h
#define __vtkSMPThreadLocalObject_h

#include <vtkSMPThreadLocal.h>

template <typename T>
class vtkSMPThreadLocalObject
{
  typedef vtkSMPThreadLocal<T*> TLS;
  typedef typename vtkSMPThreadLocal<T*>::iterator TLSIter;

  // Hide the copy constructor for now and assignment
  // operator for now.
  vtkSMPThreadLocalObject(const vtkSMPThreadLocalObject&);
  void operator=(const vtkSMPThreadLocalObject&);

public:

  vtkSMPThreadLocalObject() : Internal(0)
    {
    }

  virtual ~vtkSMPThreadLocalObject()
    {
      iterator iter = this->begin();
      while (iter != this->end())
        {
        if (*iter)
          {
          (*iter)->Delete();
          }
        ++iter;
        }
    }


  T*& Local()
    {
      T*& vtkobject = this->Internal.Local();
      if (!vtkobject)
        {
        vtkobject = T::New();
        }
      return vtkobject;
    }


  class iterator
  {
  public:
    iterator& operator++()
      {
        ++this->Iter;
        return *this;
      }

    bool operator!=(const iterator& other)
      {
        return this->Iter != other.Iter;
      }

    T*& operator*()
      {
        return *this->Iter;
      }

  private:
    TLSIter Iter;

    friend class vtkSMPThreadLocalObject<T>;
  };

  iterator begin()
    {
      iterator iter;
      iter.Iter = this->Internal.begin();
      return iter;
    };

  iterator end()
    {
      iterator iter;
      iter.Iter = this->Internal.end();
      return iter;
    }

private:
  TLS Internal;
};

#endif
// VTK-HeaderTest-Exclude: vtkSMPThreadLocalObject.h