SelfReferenceCountingObject.hpp

#pragma once
 
#include <cstddef>
 
#ifdef DELPI_THREAD_SAFE
#include <atomic>
#endif
 
namespace delpi {

class SelfReferenceCountingObject {
 public:
  void AddRef() {
#ifdef DELPI_THREAD_SAFE
    ref_count_.fetch_add(1);
#else
    ref_count_++;
#endif
  }
  void Release() {
#ifdef DELPI_THREAD_SAFE
    if (ref_count_.fetch_sub(1) == 1) {
#else
    if (--ref_count_ == 0) {
#endif
      delete this;
    }
  }

  [[nodiscard]] std::size_t use_count() const noexcept {
#ifdef DELPI_THREAD_SAFE
    return ref_count_.load();
#else
    return ref_count_;
#endif
  }
 
 protected:
  virtual ~SelfReferenceCountingObject() = default;
 
 private:
#ifdef DELPI_THREAD_SAFE
  std::atomic<std::size_t> ref_count_;  
#else
  std::size_t ref_count_{0};  
#endif
};  // NOLINT(readability/braces) per C++ standard concept definition
 
}  // namespace delpi