atlas_alloc_cpp.hpp

Go to the documentation of this file.

/*
 * (c) Copyright 2016 Hewlett Packard Enterprise Development LP
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version. This program 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 Lesser
 * General Public License along with this program. If not, see
 * <http://www.gnu.org/licenses/>.
 */


#ifndef ATLAS_ALLOC_CPP_H
#define ATLAS_ALLOC_CPP_H

// Forward declarations
namespace Atlas 
{
    class PRegion;
}

Atlas::PRegion *NVM_GetRegion(uint32_t rid);

void* operator new(size_t sz, Atlas::PRegion *rgn) noexcept;

void* operator new[](size_t sz, Atlas::PRegion *rgn) noexcept;

template <class T> static inline void NVM_Destroy(T *ptr)
{
    if (!ptr) return;
    if (!NVM_IsInRegion(ptr, 1 /* dummy, since size unknown */)) {
        delete ptr;
        return;
    }
    ptr->~T();
    void nvm_delete(void*);
    nvm_delete(ptr);
}

template <class T> static inline void NVM_Destroy_Array(T *ptr)
{
    if (!ptr) return;
    if (!NVM_IsInRegion(ptr, 1 /* dummy, since size unknown */)) {
        delete [] ptr;
        return;
    }
    char *delete_ptr = reinterpret_cast<char*>(ptr) - sizeof(size_t);
    size_t count = *reinterpret_cast<size_t*>(delete_ptr);
    for (int i=count-1; i>=0; --i) (ptr+i)->~T();
    void nvm_delete(void*);
    nvm_delete(delete_ptr);
}

#endif