1 // RUN: %libomp-cxx20-compile-and-run | FileCheck %s --match-full-lines
2
3 #ifndef HEADER
4 #define HEADER
5
6 #include <cstdlib>
7 #include <cstdarg>
8 #include <cstdio>
9 #include <vector>
10
11 struct Reporter {
12 const char *name;
13
ReporterReporter14 Reporter(const char *name) : name(name) { print("ctor"); }
15
ReporterReporter16 Reporter() : name("<anon>") { print("ctor"); }
17
ReporterReporter18 Reporter(const Reporter &that) : name(that.name) { print("copy ctor"); }
19
ReporterReporter20 Reporter(Reporter &&that) : name(that.name) { print("move ctor"); }
21
~ReporterReporter22 ~Reporter() { print("dtor"); }
23
operator =Reporter24 const Reporter &operator=(const Reporter &that) {
25 print("copy assign");
26 this->name = that.name;
27 return *this;
28 }
29
operator =Reporter30 const Reporter &operator=(Reporter &&that) {
31 print("move assign");
32 this->name = that.name;
33 return *this;
34 }
35
36 struct Iterator {
37 const Reporter *owner;
38 int pos;
39
IteratorReporter::Iterator40 Iterator(const Reporter *owner, int pos) : owner(owner), pos(pos) {}
41
IteratorReporter::Iterator42 Iterator(const Iterator &that) : owner(that.owner), pos(that.pos) {
43 owner->print("iterator copy ctor");
44 }
45
IteratorReporter::Iterator46 Iterator(Iterator &&that) : owner(that.owner), pos(that.pos) {
47 owner->print("iterator move ctor");
48 }
49
~IteratorReporter::Iterator50 ~Iterator() { owner->print("iterator dtor"); }
51
operator =Reporter::Iterator52 const Iterator &operator=(const Iterator &that) {
53 owner->print("iterator copy assign");
54 this->owner = that.owner;
55 this->pos = that.pos;
56 return *this;
57 }
58
operator =Reporter::Iterator59 const Iterator &operator=(Iterator &&that) {
60 owner->print("iterator move assign");
61 this->owner = that.owner;
62 this->pos = that.pos;
63 return *this;
64 }
65
operator ==Reporter::Iterator66 bool operator==(const Iterator &that) const {
67 owner->print("iterator %d == %d", 2 - this->pos, 2 - that.pos);
68 return this->pos == that.pos;
69 }
70
operator !=Reporter::Iterator71 bool operator!=(const Iterator &that) const {
72 owner->print("iterator %d != %d", 2 - this->pos, 2 - that.pos);
73 return this->pos != that.pos;
74 }
75
operator ++Reporter::Iterator76 Iterator &operator++() {
77 owner->print("iterator prefix ++");
78 pos -= 1;
79 return *this;
80 }
81
operator ++Reporter::Iterator82 Iterator operator++(int) {
83 owner->print("iterator postfix ++");
84 auto result = *this;
85 pos -= 1;
86 return result;
87 }
88
operator *Reporter::Iterator89 int operator*() const {
90 int result = 2 - pos;
91 owner->print("iterator deref: %i", result);
92 return result;
93 }
94
operator -Reporter::Iterator95 size_t operator-(const Iterator &that) const {
96 int result = (2 - this->pos) - (2 - that.pos);
97 owner->print("iterator distance: %d", result);
98 return result;
99 }
100
operator +Reporter::Iterator101 Iterator operator+(int steps) const {
102 owner->print("iterator advance: %i += %i", 2 - this->pos, steps);
103 return Iterator(owner, pos - steps);
104 }
105
printReporter::Iterator106 void print(const char *msg) const { owner->print(msg); }
107 };
108
beginReporter109 Iterator begin() const {
110 print("begin()");
111 return Iterator(this, 2);
112 }
113
endReporter114 Iterator end() const {
115 print("end()");
116 return Iterator(this, -1);
117 }
118
printReporter119 void print(const char *msg, ...) const {
120 va_list args;
121 va_start(args, msg);
122 printf("[%s] ", name);
123 vprintf(msg, args);
124 printf("\n");
125 va_end(args);
126 }
127 };
128
main()129 int main() {
130 printf("do\n");
131 #pragma omp parallel for collapse(2) num_threads(1)
132 for (int i = 0; i < 3; ++i)
133 #pragma omp unroll partial(2)
134 for (Reporter c{"init-stmt"}; auto &&v : Reporter("range"))
135 printf("i=%d v=%d\n", i, v);
136 printf("done\n");
137 return EXIT_SUCCESS;
138 }
139
140 #endif /* HEADER */
141
142 // CHECK: do
143 // CHECK-NEXT: [init-stmt] ctor
144 // CHECK-NEXT: [range] ctor
145 // CHECK-NEXT: [range] end()
146 // CHECK-NEXT: [range] begin()
147 // CHECK-NEXT: [range] begin()
148 // CHECK-NEXT: [range] iterator distance: 3
149 // CHECK-NEXT: [range] iterator advance: 0 += 0
150 // CHECK-NEXT: [range] iterator move assign
151 // CHECK-NEXT: [range] iterator deref: 0
152 // CHECK-NEXT: i=0 v=0
153 // CHECK-NEXT: [range] iterator dtor
154 // CHECK-NEXT: [range] iterator advance: 0 += 1
155 // CHECK-NEXT: [range] iterator move assign
156 // CHECK-NEXT: [range] iterator deref: 1
157 // CHECK-NEXT: i=0 v=1
158 // CHECK-NEXT: [range] iterator dtor
159 // CHECK-NEXT: [range] iterator advance: 0 += 2
160 // CHECK-NEXT: [range] iterator move assign
161 // CHECK-NEXT: [range] iterator deref: 2
162 // CHECK-NEXT: i=0 v=2
163 // CHECK-NEXT: [range] iterator dtor
164 // CHECK-NEXT: [range] iterator advance: 0 += 0
165 // CHECK-NEXT: [range] iterator move assign
166 // CHECK-NEXT: [range] iterator deref: 0
167 // CHECK-NEXT: i=1 v=0
168 // CHECK-NEXT: [range] iterator dtor
169 // CHECK-NEXT: [range] iterator advance: 0 += 1
170 // CHECK-NEXT: [range] iterator move assign
171 // CHECK-NEXT: [range] iterator deref: 1
172 // CHECK-NEXT: i=1 v=1
173 // CHECK-NEXT: [range] iterator dtor
174 // CHECK-NEXT: [range] iterator advance: 0 += 2
175 // CHECK-NEXT: [range] iterator move assign
176 // CHECK-NEXT: [range] iterator deref: 2
177 // CHECK-NEXT: i=1 v=2
178 // CHECK-NEXT: [range] iterator dtor
179 // CHECK-NEXT: [range] iterator advance: 0 += 0
180 // CHECK-NEXT: [range] iterator move assign
181 // CHECK-NEXT: [range] iterator deref: 0
182 // CHECK-NEXT: i=2 v=0
183 // CHECK-NEXT: [range] iterator dtor
184 // CHECK-NEXT: [range] iterator advance: 0 += 1
185 // CHECK-NEXT: [range] iterator move assign
186 // CHECK-NEXT: [range] iterator deref: 1
187 // CHECK-NEXT: i=2 v=1
188 // CHECK-NEXT: [range] iterator dtor
189 // CHECK-NEXT: [range] iterator advance: 0 += 2
190 // CHECK-NEXT: [range] iterator move assign
191 // CHECK-NEXT: [range] iterator deref: 2
192 // CHECK-NEXT: i=2 v=2
193 // CHECK-NEXT: [range] iterator dtor
194 // CHECK-NEXT: [range] iterator dtor
195 // CHECK-NEXT: [range] iterator dtor
196 // CHECK-NEXT: [range] iterator dtor
197 // CHECK-NEXT: [range] dtor
198 // CHECK-NEXT: [init-stmt] dtor
199 // CHECK-NEXT: done
200