MCThreadLocalData.hpp

/*
 *  This file is part of the AiBO+ project
 *
 *  Copyright (C) 2005-2016 Csaba Kertész (csaba.kertesz@gmail.com)
 *
 *  AiBO+ is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  AiBO+ is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Street #330, Boston, MA 02111-1307, USA.
 *
 */

#pragma once

#include <cstddef>

#if defined(__MINGW32__) || defined(AP_COVERITY) || (!defined(__AIBO_BUILD__) && defined(__unix__) && !defined(__LP64__))
#include <boost/thread/tss.hpp>
#elif !defined(__AIBO_BUILD__) && !defined(__ANDROID__) && !defined(__APPLE__)

// This hack is needed for testing
#if defined(protected)
#undef protected
#undef private
#include <folly/ThreadLocal.h> // krazy:exclude=includes
#define protected public
#define private public
#else
#include <folly/ThreadLocal.h>
#endif

#endif

#if defined(__MINGW32__) || defined(AP_COVERITY) || (!defined(__AIBO_BUILD__) && defined(__unix__) && !defined(__LP64__))
template <typename T>
class MCThreadLocalData : public boost::thread_specific_ptr<T>
{
public:
  explicit MCThreadLocalData(bool cleanup = false) : boost::thread_specific_ptr<T>(), Cleanup(cleanup)
  { }

  ~MCThreadLocalData()
  {
    // The objects are only destroyed if the thread local storage is released finally
    if (!Cleanup)
      boost::thread_specific_ptr<T>::release();
  }

private:
  bool Cleanup;
};
#elif !defined(__AIBO_BUILD__) && !defined(__ANDROID__) && !defined(__APPLE__)
template <typename T>
class MCThreadLocalData : public folly::ThreadLocalPtr<T>
{
public:
  explicit MCThreadLocalData(bool cleanup = false) : folly::ThreadLocalPtr<T>(), Cleanup(cleanup)
  { }

  ~MCThreadLocalData()
  {
    // The objects are only destroyed if the thread local storage is released finally
    if (!Cleanup)
      folly::ThreadLocalPtr<T>::release();
  }

private:
  bool Cleanup;
};
#else
template <typename T>
class MCThreadLocalData
{
public:
  explicit MCThreadLocalData(bool) : DataPtr(nullptr)
  {
  }

  ~MCThreadLocalData()
  {
    delete DataPtr;
    DataPtr = nullptr;
  }

  T* get() const
  {
    return DataPtr;
  }

  T* operator->() const
  {
    return DataPtr;
  }

  T& operator*() const
  {
    return *DataPtr;
  }

  void reset(T* new_data_ptr)
  {
    delete DataPtr;
    DataPtr = new_data_ptr;
  }

  T* release()
  {
    T* TempPtr = DataPtr;

    DataPtr = nullptr;
    return TempPtr;
  }

  T* DataPtr;
};
#endif