summaryrefslogtreecommitdiffstats
path: root/test/Analysis/os_smart_ptr.h
blob: 48a5ef3df042f49cb84b0c49b4db32b704d6f782 (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
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
80
81
82
83
84
85
86
87
88
#ifndef _OS_SMART_POINTER_H
#define _OS_SMART_POINTER_H

#include "os_object_base.h"

namespace os {

template<class T>
struct smart_ptr {
  smart_ptr() : pointer(nullptr) {}

  explicit smart_ptr(T *&p) : pointer(p) {
    if (pointer) {
      _retain(pointer);
    }
  }

  smart_ptr(smart_ptr const &rhs) : pointer(rhs.pointer) {
    if (pointer) {
      _retain(pointer);
    }
  }

  smart_ptr & operator=(T *&rhs) {
    smart_ptr(rhs).swap(*this);
    return *this;
  }

  smart_ptr & operator=(smart_ptr &rhs) {
    smart_ptr(rhs).swap(*this);
    return *this;
  }

  ~smart_ptr() {
    if (pointer) {
      _release(pointer);
    }
  }

  void reset() {
    smart_ptr().swap(*this);
  }

  T *get() const {
    return pointer;
  }

  T ** get_for_out_param() {
    reset();
    return &pointer;
  }

  T * operator->() const {
    return pointer;
  }

  explicit
  operator bool() const {
    return pointer != nullptr;
  }

  inline void
  swap(smart_ptr &p) {
    T *temp = pointer;
    pointer = p.pointer;
    p.pointer = temp;
  }

  static inline void
  _retain(T *obj) {
    obj->retain();
  }

  static inline void
  _release(T *obj) {
    obj->release();
  }

  static inline T *
  _alloc() {
    return new T;
  }

  T *pointer;
};
} // namespace os

#endif /* _OS_SMART_POINTER_H */