AdvancedMathExpressionTree/CMySmartPointer.h at master · MarczukMobbyn/AdvancedMathExpressionTree · GitHub

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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
#ifndef CMYSMARTPOINTER_H
#define CMYSMARTPOINTER_H
#include "CRefCounter.h"

template <typename T>
class CMySmartPointer
{
public:
    CMySmartPointer (T *pcPointer)
    {
        pc_pointer = pcPointer;
        pc_counter = new CRefCounter<T>();
        pc_counter->iAdd();
        pc_counter->v_add_pointer(this);
    }

    CMySmartPointer (const CMySmartPointer &pcOther)
    {
        pc_pointer = pcOther.pc_pointer;
        pc_counter = pcOther.pc_counter;
        pc_counter->iAdd();
        pc_counter->v_add_pointer(this);
    }
    ~CMySmartPointer()
    {
        pc_counter->v_remove_pointer(this);

        if (pc_counter->iDec() == 0)
        {
            delete pc_pointer;
            delete pc_counter;
        }
    }

    CMySmartPointer& operator=(const CMySmartPointer &pcOther)
    {
        if (this == &pcOther)
        {
            return *this;
        }

        pc_counter->v_remove_pointer(this);

        if (pc_counter->iDec() == 0)
        {
            // Był to ostatni wskaźnik, musimy zwolnić pamięć.
            delete pc_pointer;
            delete pc_counter;
        }

        pc_pointer = pcOther.pc_pointer;
        pc_counter = pcOther.pc_counter;

        pc_counter->v_add_pointer(this);

        pc_counter->iAdd();

        return *this;
    }

    int iGetCount() const {
        return pc_counter->iGet();
    }

    T& operator*() { return (*pc_pointer); }
    T* operator->() { return (pc_pointer); }

    std::vector<CMySmartPointer<T>*> pv_get_all_this() {
        return pc_counter->pv_get_all_pointers();
    }


private:
    CRefCounter<T> *pc_counter;
    T *pc_pointer;
};

#endif