1*4684ddb6SLionel Sambuc //===----------------------------------------------------------------------===//
2*4684ddb6SLionel Sambuc //
3*4684ddb6SLionel Sambuc // The LLVM Compiler Infrastructure
4*4684ddb6SLionel Sambuc //
5*4684ddb6SLionel Sambuc // This file is dual licensed under the MIT and the University of Illinois Open
6*4684ddb6SLionel Sambuc // Source Licenses. See LICENSE.TXT for details.
7*4684ddb6SLionel Sambuc //
8*4684ddb6SLionel Sambuc //===----------------------------------------------------------------------===//
9*4684ddb6SLionel Sambuc
10*4684ddb6SLionel Sambuc // <complex>
11*4684ddb6SLionel Sambuc
12*4684ddb6SLionel Sambuc // template<class T>
13*4684ddb6SLionel Sambuc // T
14*4684ddb6SLionel Sambuc // arg(const complex<T>& x);
15*4684ddb6SLionel Sambuc
16*4684ddb6SLionel Sambuc #include <complex>
17*4684ddb6SLionel Sambuc #include <cassert>
18*4684ddb6SLionel Sambuc
19*4684ddb6SLionel Sambuc #include "../cases.h"
20*4684ddb6SLionel Sambuc
21*4684ddb6SLionel Sambuc template <class T>
22*4684ddb6SLionel Sambuc void
test()23*4684ddb6SLionel Sambuc test()
24*4684ddb6SLionel Sambuc {
25*4684ddb6SLionel Sambuc std::complex<T> z(1, 0);
26*4684ddb6SLionel Sambuc assert(arg(z) == 0);
27*4684ddb6SLionel Sambuc }
28*4684ddb6SLionel Sambuc
test_edges()29*4684ddb6SLionel Sambuc void test_edges()
30*4684ddb6SLionel Sambuc {
31*4684ddb6SLionel Sambuc const double pi = std::atan2(+0., -0.);
32*4684ddb6SLionel Sambuc const unsigned N = sizeof(x) / sizeof(x[0]);
33*4684ddb6SLionel Sambuc for (unsigned i = 0; i < N; ++i)
34*4684ddb6SLionel Sambuc {
35*4684ddb6SLionel Sambuc double r = arg(x[i]);
36*4684ddb6SLionel Sambuc if (std::isnan(x[i].real()) || std::isnan(x[i].imag()))
37*4684ddb6SLionel Sambuc assert(std::isnan(r));
38*4684ddb6SLionel Sambuc else
39*4684ddb6SLionel Sambuc {
40*4684ddb6SLionel Sambuc switch (classify(x[i]))
41*4684ddb6SLionel Sambuc {
42*4684ddb6SLionel Sambuc case zero:
43*4684ddb6SLionel Sambuc if (std::signbit(x[i].real()))
44*4684ddb6SLionel Sambuc {
45*4684ddb6SLionel Sambuc if (std::signbit(x[i].imag()))
46*4684ddb6SLionel Sambuc is_about(r, -pi);
47*4684ddb6SLionel Sambuc else
48*4684ddb6SLionel Sambuc is_about(r, pi);
49*4684ddb6SLionel Sambuc }
50*4684ddb6SLionel Sambuc else
51*4684ddb6SLionel Sambuc {
52*4684ddb6SLionel Sambuc assert(std::signbit(x[i].imag()) == std::signbit(r));
53*4684ddb6SLionel Sambuc }
54*4684ddb6SLionel Sambuc break;
55*4684ddb6SLionel Sambuc case non_zero:
56*4684ddb6SLionel Sambuc if (x[i].real() == 0)
57*4684ddb6SLionel Sambuc {
58*4684ddb6SLionel Sambuc if (x[i].imag() < 0)
59*4684ddb6SLionel Sambuc is_about(r, -pi/2);
60*4684ddb6SLionel Sambuc else
61*4684ddb6SLionel Sambuc is_about(r, pi/2);
62*4684ddb6SLionel Sambuc }
63*4684ddb6SLionel Sambuc else if (x[i].imag() == 0)
64*4684ddb6SLionel Sambuc {
65*4684ddb6SLionel Sambuc if (x[i].real() < 0)
66*4684ddb6SLionel Sambuc {
67*4684ddb6SLionel Sambuc if (std::signbit(x[i].imag()))
68*4684ddb6SLionel Sambuc is_about(r, -pi);
69*4684ddb6SLionel Sambuc else
70*4684ddb6SLionel Sambuc is_about(r, pi);
71*4684ddb6SLionel Sambuc }
72*4684ddb6SLionel Sambuc else
73*4684ddb6SLionel Sambuc {
74*4684ddb6SLionel Sambuc assert(r == 0);
75*4684ddb6SLionel Sambuc assert(std::signbit(x[i].imag()) == std::signbit(r));
76*4684ddb6SLionel Sambuc }
77*4684ddb6SLionel Sambuc }
78*4684ddb6SLionel Sambuc else if (x[i].imag() > 0)
79*4684ddb6SLionel Sambuc assert(r > 0);
80*4684ddb6SLionel Sambuc else
81*4684ddb6SLionel Sambuc assert(r < 0);
82*4684ddb6SLionel Sambuc break;
83*4684ddb6SLionel Sambuc case inf:
84*4684ddb6SLionel Sambuc if (std::isinf(x[i].real()) && std::isinf(x[i].imag()))
85*4684ddb6SLionel Sambuc {
86*4684ddb6SLionel Sambuc if (x[i].real() < 0)
87*4684ddb6SLionel Sambuc {
88*4684ddb6SLionel Sambuc if (x[i].imag() > 0)
89*4684ddb6SLionel Sambuc is_about(r, 0.75 * pi);
90*4684ddb6SLionel Sambuc else
91*4684ddb6SLionel Sambuc is_about(r, -0.75 * pi);
92*4684ddb6SLionel Sambuc }
93*4684ddb6SLionel Sambuc else
94*4684ddb6SLionel Sambuc {
95*4684ddb6SLionel Sambuc if (x[i].imag() > 0)
96*4684ddb6SLionel Sambuc is_about(r, 0.25 * pi);
97*4684ddb6SLionel Sambuc else
98*4684ddb6SLionel Sambuc is_about(r, -0.25 * pi);
99*4684ddb6SLionel Sambuc }
100*4684ddb6SLionel Sambuc }
101*4684ddb6SLionel Sambuc else if (std::isinf(x[i].real()))
102*4684ddb6SLionel Sambuc {
103*4684ddb6SLionel Sambuc if (x[i].real() < 0)
104*4684ddb6SLionel Sambuc {
105*4684ddb6SLionel Sambuc if (std::signbit(x[i].imag()))
106*4684ddb6SLionel Sambuc is_about(r, -pi);
107*4684ddb6SLionel Sambuc else
108*4684ddb6SLionel Sambuc is_about(r, pi);
109*4684ddb6SLionel Sambuc }
110*4684ddb6SLionel Sambuc else
111*4684ddb6SLionel Sambuc {
112*4684ddb6SLionel Sambuc assert(r == 0);
113*4684ddb6SLionel Sambuc assert(std::signbit(r) == std::signbit(x[i].imag()));
114*4684ddb6SLionel Sambuc }
115*4684ddb6SLionel Sambuc }
116*4684ddb6SLionel Sambuc else
117*4684ddb6SLionel Sambuc {
118*4684ddb6SLionel Sambuc if (x[i].imag() < 0)
119*4684ddb6SLionel Sambuc is_about(r, -pi/2);
120*4684ddb6SLionel Sambuc else
121*4684ddb6SLionel Sambuc is_about(r, pi/2);
122*4684ddb6SLionel Sambuc }
123*4684ddb6SLionel Sambuc break;
124*4684ddb6SLionel Sambuc }
125*4684ddb6SLionel Sambuc }
126*4684ddb6SLionel Sambuc }
127*4684ddb6SLionel Sambuc }
128*4684ddb6SLionel Sambuc
main()129*4684ddb6SLionel Sambuc int main()
130*4684ddb6SLionel Sambuc {
131*4684ddb6SLionel Sambuc test<float>();
132*4684ddb6SLionel Sambuc test<double>();
133*4684ddb6SLionel Sambuc test<long double>();
134*4684ddb6SLionel Sambuc test_edges();
135*4684ddb6SLionel Sambuc }
136