ExternalAllocator.h

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
#ifndef TRACKINGITSU_INCLUDE_EXTERNALALLOCATOR_H_
#define TRACKINGITSU_INCLUDE_EXTERNALALLOCATOR_H_
 
#include <memory_resource>
 
namespace o2::its
{
 
class ExternalAllocator
{
 public:
  virtual void* allocate(size_t) = 0;
  virtual void deallocate(char*, size_t) = 0;
};
 
class ExternalAllocatorAdaptor final : public std::pmr::memory_resource
{
 public:
  explicit ExternalAllocatorAdaptor(ExternalAllocator* alloc) : mAlloc(alloc) {}
 
 protected:
  void* do_allocate(size_t bytes, size_t alignment) override
  {
    void* p = mAlloc->allocate(bytes);
    if (!p) {
      throw std::bad_alloc();
    }
    return p;
  }
 
  void do_deallocate(void* p, size_t bytes, size_t) override
  {
    mAlloc->deallocate(static_cast<char*>(p), bytes);
  }
 
  bool do_is_equal(const std::pmr::memory_resource& other) const noexcept override
  {
    return this == &other;
  }
 
 private:
  ExternalAllocator* mAlloc;
};
 
} // namespace o2::its
 
#endif