Line data    Source code

       1             : /* Copyright (C) 2022 Wildfire Games.
       2             :  *
       3             :  * Permission is hereby granted, free of charge, to any person obtaining
       4             :  * a copy of this software and associated documentation files (the
       5             :  * "Software"), to deal in the Software without restriction, including
       6             :  * without limitation the rights to use, copy, modify, merge, publish,
       7             :  * distribute, sublicense, and/or sell copies of the Software, and to
       8             :  * permit persons to whom the Software is furnished to do so, subject to
       9             :  * the following conditions:
      10             :  *
      11             :  * The above copyright notice and this permission notice shall be included
      12             :  * in all copies or substantial portions of the Software.
      13             :  *
      14             :  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
      15             :  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
      16             :  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
      17             :  * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
      18             :  * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
      19             :  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
      20             :  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
      21             :  */
      22             : 
      23             : #ifndef INCLUDED_ALLOCATORS_SHARED_PTR
      24             : #define INCLUDED_ALLOCATORS_SHARED_PTR
      25             : 
      26             : #include "lib/alignment.h"
      27             : #include "lib/sysdep/rtl.h" // rtl_AllocateAligned
      28             : 
      29             : struct DummyDeleter
      30             : {
      31             :     template<class T>
      32        2837 :     void operator()(T*)
      33             :     {
      34        2837 :     }
      35             : };
      36             : 
      37             : template<class T>
      38        2837 : inline std::shared_ptr<T> DummySharedPtr(T* ptr)
      39             : {
      40        2837 :     return std::shared_ptr<T>(ptr, DummyDeleter());
      41             : }
      42             : 
      43             : struct ArrayDeleter
      44             : {
      45             :     template<class T>
      46           8 :     void operator()(T* p)
      47             :     {
      48           8 :         delete[] p;
      49           8 :     }
      50             : };
      51             : 
      52             : // (note: uses CheckedArrayDeleter)
      53             : std::shared_ptr<u8> Allocate(size_t size);
      54             : 
      55             : 
      56             : struct AlignedDeleter
      57             : {
      58             :     template<class T>
      59        2995 :     void operator()(T* t)
      60             :     {
      61        2995 :         rtl_FreeAligned(t);
      62        2995 :     }
      63             : };
      64             : 
      65             : template<class T>
      66        2879 : static inline Status AllocateAligned(std::shared_ptr<T>& p, size_t size, size_t alignment = cacheLineSize)
      67             : {
      68        2879 :     void* mem = rtl_AllocateAligned(size, alignment);
      69        2879 :     if(!mem)
      70           0 :         WARN_RETURN(ERR::NO_MEM);
      71        2879 :     p.reset((T*)mem, AlignedDeleter());
      72        2879 :     return INFO::OK;
      73             : }
      74             : 
      75             : #endif  // #ifndef INCLUDED_ALLOCATORS_SHARED_PTR