blob: ce1f4d1d8ec95368dcc79ad897380ce21be74532 [file] [log] [blame]
Howard Hinnantc52f43e2010-08-22 00:59:46 +00001//===----------------------------------------------------------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
10// <memory>
11
12// unique_ptr
13
14// Test unique_ptr converting move ctor
15
16#include <memory>
17#include <cassert>
18
19// test converting move ctor. Should only require a MoveConstructible deleter, or if
20// deleter is a reference, not even that.
21// Implicit version
22
23struct A
24{
25 static int count;
26 A() {++count;}
27 A(const A&) {++count;}
28 virtual ~A() {--count;}
29};
30
31int A::count = 0;
32
33struct B
34 : public A
35{
36 static int count;
37 B() {++count;}
38 B(const B&) {++count;}
39 virtual ~B() {--count;}
40};
41
42int B::count = 0;
43
44template <class T>
45class Deleter
46{
47 int state_;
48
Howard Hinnant73d21a42010-09-04 23:28:19 +000049#ifndef _LIBCPP_HAS_NO_RVALUE_REFERENCES
Howard Hinnantc52f43e2010-08-22 00:59:46 +000050 Deleter(const Deleter&);
51 Deleter& operator=(const Deleter&);
Howard Hinnant73d21a42010-09-04 23:28:19 +000052#else // _LIBCPP_HAS_NO_RVALUE_REFERENCES
Howard Hinnantc52f43e2010-08-22 00:59:46 +000053 Deleter(Deleter&);
54 Deleter& operator=(Deleter&);
Howard Hinnant73d21a42010-09-04 23:28:19 +000055#endif // _LIBCPP_HAS_NO_RVALUE_REFERENCES
Howard Hinnantc52f43e2010-08-22 00:59:46 +000056
57public:
Howard Hinnant73d21a42010-09-04 23:28:19 +000058#ifndef _LIBCPP_HAS_NO_RVALUE_REFERENCES
Howard Hinnantc52f43e2010-08-22 00:59:46 +000059 Deleter(Deleter&& r) : state_(r.state_) {r.state_ = 0;}
60 Deleter& operator=(Deleter&& r)
61 {
62 state_ = r.state_;
63 r.state_ = 0;
64 return *this;
65 }
Howard Hinnant73d21a42010-09-04 23:28:19 +000066#else // _LIBCPP_HAS_NO_RVALUE_REFERENCES
Howard Hinnantc52f43e2010-08-22 00:59:46 +000067 operator std::__rv<Deleter>() {return std::__rv<Deleter>(*this);}
68 Deleter(std::__rv<Deleter> r) : state_(r->state_) {r->state_ = 0;}
69 Deleter& operator=(std::__rv<Deleter> r)
70 {
71 state_ = r->state_;
72 r->state_ = 0;
73 return *this;
74 }
Howard Hinnant73d21a42010-09-04 23:28:19 +000075#endif // _LIBCPP_HAS_NO_RVALUE_REFERENCES
Howard Hinnantc52f43e2010-08-22 00:59:46 +000076
77 Deleter() : state_(5) {}
78
Howard Hinnant73d21a42010-09-04 23:28:19 +000079#ifndef _LIBCPP_HAS_NO_RVALUE_REFERENCES
Howard Hinnantc52f43e2010-08-22 00:59:46 +000080 template <class U>
81 Deleter(Deleter<U>&& d,
82 typename std::enable_if<!std::is_same<U, T>::value>::type* = 0)
83 : state_(d.state()) {d.set_state(0);}
84
85private:
86 template <class U>
87 Deleter(const Deleter<U>& d,
88 typename std::enable_if<!std::is_same<U, T>::value>::type* = 0);
Howard Hinnant73d21a42010-09-04 23:28:19 +000089#else // _LIBCPP_HAS_NO_RVALUE_REFERENCES
Howard Hinnantc52f43e2010-08-22 00:59:46 +000090 template <class U>
91 Deleter(Deleter<U> d,
92 typename std::enable_if<!std::is_same<U, T>::value>::type* = 0)
93 : state_(d.state()) {}
Howard Hinnant73d21a42010-09-04 23:28:19 +000094#endif // _LIBCPP_HAS_NO_RVALUE_REFERENCES
Howard Hinnantc52f43e2010-08-22 00:59:46 +000095public:
96 int state() const {return state_;}
97 void set_state(int i) {state_ = i;}
98
99 void operator()(T* p) {delete p;}
100};
101
102int main()
103{
104 {
105 const std::unique_ptr<B, Deleter<B> > s(new B);
106 A* p = s.get();
107 std::unique_ptr<A, Deleter<A> > s2 = s;
108 assert(s2.get() == p);
109 assert(s.get() == 0);
110 assert(A::count == 1);
111 assert(B::count == 1);
112 assert(s2.get_deleter().state() == 5);
113 assert(s.get_deleter().state() == 0);
114 }
115 assert(A::count == 0);
116 assert(B::count == 0);
117}