Home | History | Annotate | Download | only in mask.array.assign
      1 //===----------------------------------------------------------------------===//
      2 //
      3 //                     The LLVM Compiler Infrastructure
      4 //
      5 // This file is dual licensed under the MIT and the University of Illinois Open
      6 // Source Licenses. See LICENSE.TXT for details.
      7 //
      8 //===----------------------------------------------------------------------===//
      9 
     10 // <valarray>
     11 
     12 // template <class T> class mask_array
     13 
     14 // void operator=(const valarray<value_type>& v) const;
     15 
     16 #include <valarray>
     17 #include <cassert>
     18 
     19 int main()
     20 {
     21     int a1[] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15};
     22     const std::size_t N1 = sizeof(a1)/sizeof(a1[0]);
     23     int a2[] = {1, 2, 3, 4, 5};
     24     const std::size_t N2 = sizeof(a2)/sizeof(a2[0]);
     25     bool b[N1] = {true,  false, false, true,  true,  false,
     26                   false, true,  false, false, false, true};
     27     std::valarray<int> v1(a1, N1);
     28     std::valarray<int> v2(a2, N2);
     29     assert(N2 == std::count(b, b+N1, true));
     30     std::valarray<bool> vb(b, N1);
     31     v1[vb] = v2;
     32     assert(v1.size() == 16);
     33     assert(v1[ 0] ==  1);
     34     assert(v1[ 1] ==  1);
     35     assert(v1[ 2] ==  2);
     36     assert(v1[ 3] ==  2);
     37     assert(v1[ 4] ==  3);
     38     assert(v1[ 5] ==  5);
     39     assert(v1[ 6] ==  6);
     40     assert(v1[ 7] ==  4);
     41     assert(v1[ 8] ==  8);
     42     assert(v1[ 9] ==  9);
     43     assert(v1[10] == 10);
     44     assert(v1[11] ==  5);
     45     assert(v1[12] == 12);
     46     assert(v1[13] == 13);
     47     assert(v1[14] == 14);
     48     assert(v1[15] == 15);
     49 }
     50