1 //===----------------------------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 // <valarray>
10 
11 // template<class T> class valarray;
12 
13 // [valarray.syn]/3
14 //   Any function returning a valarray<T> is permitted to return an object of
15 //   another type, provided all the const member functions of valarray<T> are
16 //   also applicable to this type.
17 //
18 // Libc++ uses this and returns __val_expr<_Expr> for several operations.
19 //
20 // The const overloads of
21 //   valarray::operator[](...) const
22 // return propxy objects. These proxies are implicitly convertible to
23 // std::valarray.
24 //
25 // valarray& operator*=(const valarray& v);
26 
27 #include <valarray>
28 #include <cassert>
29 #include <cstddef>
30 
31 #include "test_macros.h"
32 
33 template <class A>
test(const A & rhs)34 void test(const A& rhs) {
35   int input[]      = {1, 2, 3, 4, 5};
36   int expected[]   = {6, 14, 24, 36, 50};
37   const unsigned N = sizeof(input) / sizeof(input[0]);
38   std::valarray<int> value(input, N);
39 
40   value *= rhs;
41 
42   assert(value.size() == N);
43   for (std::size_t i = 0; i < value.size(); ++i)
44     assert(value[i] == expected[i]);
45 }
46 
main(int,char **)47 int main(int, char**) {
48   int input[]      = {6, 7, 8, 9, 10};
49   const unsigned N = sizeof(input) / sizeof(input[0]);
50 
51   std::valarray<bool> mask(true, N);
52   std::size_t indices[] = {0, 1, 2, 3, 4};
53   std::valarray<std::size_t> indirect(indices, N);
54 
55   std::valarray<int> zero(0, N);
56 
57   {
58     std::valarray<int> value(input, N);
59 
60     test(value);
61     test(value[std::slice(0, N, 1)]);
62     test(value[std::gslice(0, std::valarray<std::size_t>(N, 1), std::valarray<std::size_t>(1, 1))]);
63     test(value[mask]);
64     test(value[indirect]);
65     test(value + zero);
66   }
67 
68   {
69     const std::valarray<int> value(input, N);
70 
71     test(value);
72     test(value[std::slice(0, N, 1)]);
73     test(value[std::gslice(0, std::valarray<std::size_t>(N, 1), std::valarray<std::size_t>(1, 1))]);
74     test(value[mask]);
75     test(value[indirect]);
76     test(value + zero);
77   }
78 
79   return 0;
80 }
81