83 lines · cpp
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-exception6//7//===----------------------------------------------------------------------===//8 9// <valarray>10 11// template<class T> class valarray;12 13// gslice_array<value_type> operator[](const gslice& gs);14 15#include <valarray>16#include <cassert>17 18#include "test_macros.h"19 20int main(int, char**)21{22 {23 int a1[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11,24 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23,25 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35,26 36, 37, 38, 39, 40};27 int a2[] = { -0, -1, -2, -3, -4, -5, -6, -7, -8, -9, -10, -11,28 -12, -13, -14, -15, -16, -17, -18, -19, -20, -21, -22, -23};29 std::valarray<int> v1(a1, sizeof(a1)/sizeof(a1[0]));30 std::valarray<int> v2(a2, sizeof(a2)/sizeof(a2[0]));31 std::size_t sz[] = {2, 4, 3};32 std::size_t st[] = {19, 4, 1};33 typedef std::valarray<std::size_t> sizes;34 typedef std::valarray<std::size_t> strides;35 v1[std::gslice(3, sizes(sz, sizeof(sz)/sizeof(sz[0])),36 strides(st, sizeof(st)/sizeof(st[0])))] = v2;37 assert(v1.size() == 41);38 assert(v1[ 0] == 0);39 assert(v1[ 1] == 1);40 assert(v1[ 2] == 2);41 assert(v1[ 3] == 0);42 assert(v1[ 4] == -1);43 assert(v1[ 5] == -2);44 assert(v1[ 6] == 6);45 assert(v1[ 7] == -3);46 assert(v1[ 8] == -4);47 assert(v1[ 9] == -5);48 assert(v1[10] == 10);49 assert(v1[11] == -6);50 assert(v1[12] == -7);51 assert(v1[13] == -8);52 assert(v1[14] == 14);53 assert(v1[15] == -9);54 assert(v1[16] == -10);55 assert(v1[17] == -11);56 assert(v1[18] == 18);57 assert(v1[19] == 19);58 assert(v1[20] == 20);59 assert(v1[21] == 21);60 assert(v1[22] == -12);61 assert(v1[23] == -13);62 assert(v1[24] == -14);63 assert(v1[25] == 25);64 assert(v1[26] == -15);65 assert(v1[27] == -16);66 assert(v1[28] == -17);67 assert(v1[29] == 29);68 assert(v1[30] == -18);69 assert(v1[31] == -19);70 assert(v1[32] == -20);71 assert(v1[33] == 33);72 assert(v1[34] == -21);73 assert(v1[35] == -22);74 assert(v1[36] == -23);75 assert(v1[37] == 37);76 assert(v1[38] == 38);77 assert(v1[39] == 39);78 assert(v1[40] == 40);79 }80 81 return 0;82}83