summaryrefslogtreecommitdiff
path: root/include/util/ptr.hpp
blob: 0e8be3f8ca84f9d901e6baaaa3016edc7faed188 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
// Display images inside a terminal
// Copyright (C) 2023  JustKidding
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU 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 General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.

#ifndef UTIL_PTR_H
#define UTIL_PTR_H

#include <memory>

template <auto fn>
struct deleter_from_fn {
    template <typename T>
    constexpr void operator()(T *arg) const
    {
        fn(const_cast<std::remove_const_t<T> *>(arg));
    }
};

template <auto fn>
struct deleter_from_fn_null {
    template <typename T>
    constexpr void operator()(T *arg) const
    {
        if (arg != nullptr) {
            fn(const_cast<std::remove_const_t<T> *>(arg));
        }
    }
};

// custom unique pointer
template <typename T, auto fn>
using c_unique_ptr = std::unique_ptr<T, deleter_from_fn<fn>>;

// custom unique pointer that checks for null before deleting
template <typename T, auto fn>
using cn_unique_ptr = std::unique_ptr<T, deleter_from_fn_null<fn>>;

template <typename T>
using unique_C_ptr = std::unique_ptr<T, deleter_from_fn<std::free>>;

#endif