AutoRef.h

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2025 CEA (www.cea.fr) IFPEN (www.ifpenergiesnouvelles.com)
// See the top-level COPYRIGHT file for details.
// SPDX-License-Identifier: Apache-2.0
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* AutoRef.h                                                   (C) 2000-2025 */
/*                                                                           */
/* Encapsulation d'un pointeur avec compteur de référence.                   */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_UTILS_AUTOREF_H
#define ARCANE_UTILS_AUTOREF_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/utils/Ptr.h"
#include "arccore/base/AutoRef2.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template <class T>
class AutoRefT
: public PtrT<T>
{
 public:

  using BaseClass = PtrT<T>;
 
  using BaseClass::m_value;
 
 public:

  AutoRefT() = default;
  explicit AutoRefT(T* t)
  : BaseClass()
  {
    _changeValue(t);
  }

  AutoRefT(const AutoRefT<T>& from)
  : BaseClass()
  {
    _changeValue(from.m_value);
  }

  AutoRefT<T>& operator=(const AutoRefT<T>& from)
  {
    _changeValue(from.m_value);
    return (*this);
  }

  AutoRefT<T>& operator=(T* new_value)
  {
    _changeValue(new_value);
    return (*this);
  }

  ~AutoRefT() { _removeRef(); }
 
 private:

  void _addRef()
  {
    if (m_value)
      m_value->addRef();
  }

  void _removeRef()
  {
    if (m_value)
      m_value->removeRef();
  }

  void _changeValue(T* new_value)
  {
    if (m_value == new_value)
      return;
    _removeRef();
    m_value = new_value;
    _addRef();
  }
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // namespace Arcane
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif