1*e4b17023SJohn Marino // The template and inlines for the -*- C++ -*- mask_array class. 2*e4b17023SJohn Marino 3*e4b17023SJohn Marino // Copyright (C) 1997, 1998, 1999, 2000, 2001, 2002, 2004, 2005, 2009, 2010 4*e4b17023SJohn Marino // Free Software Foundation, Inc. 5*e4b17023SJohn Marino // 6*e4b17023SJohn Marino // This file is part of the GNU ISO C++ Library. This library is free 7*e4b17023SJohn Marino // software; you can redistribute it and/or modify it under the 8*e4b17023SJohn Marino // terms of the GNU General Public License as published by the 9*e4b17023SJohn Marino // Free Software Foundation; either version 3, or (at your option) 10*e4b17023SJohn Marino // any later version. 11*e4b17023SJohn Marino 12*e4b17023SJohn Marino // This library is distributed in the hope that it will be useful, 13*e4b17023SJohn Marino // but WITHOUT ANY WARRANTY; without even the implied warranty of 14*e4b17023SJohn Marino // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 15*e4b17023SJohn Marino // GNU General Public License for more details. 16*e4b17023SJohn Marino 17*e4b17023SJohn Marino // Under Section 7 of GPL version 3, you are granted additional 18*e4b17023SJohn Marino // permissions described in the GCC Runtime Library Exception, version 19*e4b17023SJohn Marino // 3.1, as published by the Free Software Foundation. 20*e4b17023SJohn Marino 21*e4b17023SJohn Marino // You should have received a copy of the GNU General Public License and 22*e4b17023SJohn Marino // a copy of the GCC Runtime Library Exception along with this program; 23*e4b17023SJohn Marino // see the files COPYING3 and COPYING.RUNTIME respectively. If not, see 24*e4b17023SJohn Marino // <http://www.gnu.org/licenses/>. 25*e4b17023SJohn Marino 26*e4b17023SJohn Marino /** @file bits/mask_array.h 27*e4b17023SJohn Marino * This is an internal header file, included by other library headers. 28*e4b17023SJohn Marino * Do not attempt to use it directly. @headername{valarray} 29*e4b17023SJohn Marino */ 30*e4b17023SJohn Marino 31*e4b17023SJohn Marino // Written by Gabriel Dos Reis <Gabriel.Dos-Reis@DPTMaths.ENS-Cachan.Fr> 32*e4b17023SJohn Marino 33*e4b17023SJohn Marino #ifndef _MASK_ARRAY_H 34*e4b17023SJohn Marino #define _MASK_ARRAY_H 1 35*e4b17023SJohn Marino 36*e4b17023SJohn Marino #pragma GCC system_header 37*e4b17023SJohn Marino 38*e4b17023SJohn Marino namespace std _GLIBCXX_VISIBILITY(default) 39*e4b17023SJohn Marino { 40*e4b17023SJohn Marino _GLIBCXX_BEGIN_NAMESPACE_VERSION 41*e4b17023SJohn Marino 42*e4b17023SJohn Marino /** 43*e4b17023SJohn Marino * @addtogroup numeric_arrays 44*e4b17023SJohn Marino * @{ 45*e4b17023SJohn Marino */ 46*e4b17023SJohn Marino 47*e4b17023SJohn Marino /** 48*e4b17023SJohn Marino * @brief Reference to selected subset of an array. 49*e4b17023SJohn Marino * 50*e4b17023SJohn Marino * A mask_array is a reference to the actual elements of an array specified 51*e4b17023SJohn Marino * by a bitmask in the form of an array of bool. The way to get a 52*e4b17023SJohn Marino * mask_array is to call operator[](valarray<bool>) on a valarray. The 53*e4b17023SJohn Marino * returned mask_array then permits carrying operations out on the 54*e4b17023SJohn Marino * referenced subset of elements in the original valarray. 55*e4b17023SJohn Marino * 56*e4b17023SJohn Marino * For example, if a mask_array is obtained using the array (false, true, 57*e4b17023SJohn Marino * false, true) as an argument, the mask array has two elements referring 58*e4b17023SJohn Marino * to array[1] and array[3] in the underlying array. 59*e4b17023SJohn Marino * 60*e4b17023SJohn Marino * @param Tp Element type. 61*e4b17023SJohn Marino */ 62*e4b17023SJohn Marino template <class _Tp> 63*e4b17023SJohn Marino class mask_array 64*e4b17023SJohn Marino { 65*e4b17023SJohn Marino public: 66*e4b17023SJohn Marino typedef _Tp value_type; 67*e4b17023SJohn Marino 68*e4b17023SJohn Marino // _GLIBCXX_RESOLVE_LIB_DEFECTS 69*e4b17023SJohn Marino // 253. valarray helper functions are almost entirely useless 70*e4b17023SJohn Marino 71*e4b17023SJohn Marino /// Copy constructor. Both slices refer to the same underlying array. 72*e4b17023SJohn Marino mask_array (const mask_array&); 73*e4b17023SJohn Marino 74*e4b17023SJohn Marino /// Assignment operator. Assigns elements to corresponding elements 75*e4b17023SJohn Marino /// of @a a. 76*e4b17023SJohn Marino mask_array& operator=(const mask_array&); 77*e4b17023SJohn Marino 78*e4b17023SJohn Marino void operator=(const valarray<_Tp>&) const; 79*e4b17023SJohn Marino /// Multiply slice elements by corresponding elements of @a v. 80*e4b17023SJohn Marino void operator*=(const valarray<_Tp>&) const; 81*e4b17023SJohn Marino /// Divide slice elements by corresponding elements of @a v. 82*e4b17023SJohn Marino void operator/=(const valarray<_Tp>&) const; 83*e4b17023SJohn Marino /// Modulo slice elements by corresponding elements of @a v. 84*e4b17023SJohn Marino void operator%=(const valarray<_Tp>&) const; 85*e4b17023SJohn Marino /// Add corresponding elements of @a v to slice elements. 86*e4b17023SJohn Marino void operator+=(const valarray<_Tp>&) const; 87*e4b17023SJohn Marino /// Subtract corresponding elements of @a v from slice elements. 88*e4b17023SJohn Marino void operator-=(const valarray<_Tp>&) const; 89*e4b17023SJohn Marino /// Logical xor slice elements with corresponding elements of @a v. 90*e4b17023SJohn Marino void operator^=(const valarray<_Tp>&) const; 91*e4b17023SJohn Marino /// Logical and slice elements with corresponding elements of @a v. 92*e4b17023SJohn Marino void operator&=(const valarray<_Tp>&) const; 93*e4b17023SJohn Marino /// Logical or slice elements with corresponding elements of @a v. 94*e4b17023SJohn Marino void operator|=(const valarray<_Tp>&) const; 95*e4b17023SJohn Marino /// Left shift slice elements by corresponding elements of @a v. 96*e4b17023SJohn Marino void operator<<=(const valarray<_Tp>&) const; 97*e4b17023SJohn Marino /// Right shift slice elements by corresponding elements of @a v. 98*e4b17023SJohn Marino void operator>>=(const valarray<_Tp>&) const; 99*e4b17023SJohn Marino /// Assign all slice elements to @a t. 100*e4b17023SJohn Marino void operator=(const _Tp&) const; 101*e4b17023SJohn Marino 102*e4b17023SJohn Marino // ~mask_array (); 103*e4b17023SJohn Marino 104*e4b17023SJohn Marino template<class _Dom> 105*e4b17023SJohn Marino void operator=(const _Expr<_Dom,_Tp>&) const; 106*e4b17023SJohn Marino template<class _Dom> 107*e4b17023SJohn Marino void operator*=(const _Expr<_Dom,_Tp>&) const; 108*e4b17023SJohn Marino template<class _Dom> 109*e4b17023SJohn Marino void operator/=(const _Expr<_Dom,_Tp>&) const; 110*e4b17023SJohn Marino template<class _Dom> 111*e4b17023SJohn Marino void operator%=(const _Expr<_Dom,_Tp>&) const; 112*e4b17023SJohn Marino template<class _Dom> 113*e4b17023SJohn Marino void operator+=(const _Expr<_Dom,_Tp>&) const; 114*e4b17023SJohn Marino template<class _Dom> 115*e4b17023SJohn Marino void operator-=(const _Expr<_Dom,_Tp>&) const; 116*e4b17023SJohn Marino template<class _Dom> 117*e4b17023SJohn Marino void operator^=(const _Expr<_Dom,_Tp>&) const; 118*e4b17023SJohn Marino template<class _Dom> 119*e4b17023SJohn Marino void operator&=(const _Expr<_Dom,_Tp>&) const; 120*e4b17023SJohn Marino template<class _Dom> 121*e4b17023SJohn Marino void operator|=(const _Expr<_Dom,_Tp>&) const; 122*e4b17023SJohn Marino template<class _Dom> 123*e4b17023SJohn Marino void operator<<=(const _Expr<_Dom,_Tp>&) const; 124*e4b17023SJohn Marino template<class _Dom> 125*e4b17023SJohn Marino void operator>>=(const _Expr<_Dom,_Tp>&) const; 126*e4b17023SJohn Marino 127*e4b17023SJohn Marino private: 128*e4b17023SJohn Marino mask_array(_Array<_Tp>, size_t, _Array<bool>); 129*e4b17023SJohn Marino friend class valarray<_Tp>; 130*e4b17023SJohn Marino 131*e4b17023SJohn Marino const size_t _M_sz; 132*e4b17023SJohn Marino const _Array<bool> _M_mask; 133*e4b17023SJohn Marino const _Array<_Tp> _M_array; 134*e4b17023SJohn Marino 135*e4b17023SJohn Marino // not implemented 136*e4b17023SJohn Marino mask_array(); 137*e4b17023SJohn Marino }; 138*e4b17023SJohn Marino 139*e4b17023SJohn Marino template<typename _Tp> 140*e4b17023SJohn Marino inline mask_array<_Tp>::mask_array(const mask_array<_Tp>& a) 141*e4b17023SJohn Marino : _M_sz(a._M_sz), _M_mask(a._M_mask), _M_array(a._M_array) {} 142*e4b17023SJohn Marino 143*e4b17023SJohn Marino template<typename _Tp> 144*e4b17023SJohn Marino inline 145*e4b17023SJohn Marino mask_array<_Tp>::mask_array(_Array<_Tp> __a, size_t __s, _Array<bool> __m) 146*e4b17023SJohn Marino : _M_sz(__s), _M_mask(__m), _M_array(__a) {} 147*e4b17023SJohn Marino 148*e4b17023SJohn Marino template<typename _Tp> 149*e4b17023SJohn Marino inline mask_array<_Tp>& 150*e4b17023SJohn Marino mask_array<_Tp>::operator=(const mask_array<_Tp>& __a) 151*e4b17023SJohn Marino { 152*e4b17023SJohn Marino std::__valarray_copy(__a._M_array, __a._M_mask, 153*e4b17023SJohn Marino _M_sz, _M_array, _M_mask); 154*e4b17023SJohn Marino return *this; 155*e4b17023SJohn Marino } 156*e4b17023SJohn Marino 157*e4b17023SJohn Marino template<typename _Tp> 158*e4b17023SJohn Marino inline void 159*e4b17023SJohn Marino mask_array<_Tp>::operator=(const _Tp& __t) const 160*e4b17023SJohn Marino { std::__valarray_fill(_M_array, _M_sz, _M_mask, __t); } 161*e4b17023SJohn Marino 162*e4b17023SJohn Marino template<typename _Tp> 163*e4b17023SJohn Marino inline void 164*e4b17023SJohn Marino mask_array<_Tp>::operator=(const valarray<_Tp>& __v) const 165*e4b17023SJohn Marino { std::__valarray_copy(_Array<_Tp>(__v), __v.size(), _M_array, _M_mask); } 166*e4b17023SJohn Marino 167*e4b17023SJohn Marino template<typename _Tp> 168*e4b17023SJohn Marino template<class _Ex> 169*e4b17023SJohn Marino inline void 170*e4b17023SJohn Marino mask_array<_Tp>::operator=(const _Expr<_Ex, _Tp>& __e) const 171*e4b17023SJohn Marino { std::__valarray_copy(__e, __e.size(), _M_array, _M_mask); } 172*e4b17023SJohn Marino 173*e4b17023SJohn Marino #undef _DEFINE_VALARRAY_OPERATOR 174*e4b17023SJohn Marino #define _DEFINE_VALARRAY_OPERATOR(_Op, _Name) \ 175*e4b17023SJohn Marino template<typename _Tp> \ 176*e4b17023SJohn Marino inline void \ 177*e4b17023SJohn Marino mask_array<_Tp>::operator _Op##=(const valarray<_Tp>& __v) const \ 178*e4b17023SJohn Marino { \ 179*e4b17023SJohn Marino _Array_augmented_##_Name(_M_array, _M_mask, \ 180*e4b17023SJohn Marino _Array<_Tp>(__v), __v.size()); \ 181*e4b17023SJohn Marino } \ 182*e4b17023SJohn Marino \ 183*e4b17023SJohn Marino template<typename _Tp> \ 184*e4b17023SJohn Marino template<class _Dom> \ 185*e4b17023SJohn Marino inline void \ 186*e4b17023SJohn Marino mask_array<_Tp>::operator _Op##=(const _Expr<_Dom, _Tp>& __e) const\ 187*e4b17023SJohn Marino { \ 188*e4b17023SJohn Marino _Array_augmented_##_Name(_M_array, _M_mask, __e, __e.size()); \ 189*e4b17023SJohn Marino } 190*e4b17023SJohn Marino 191*e4b17023SJohn Marino _DEFINE_VALARRAY_OPERATOR(*, __multiplies) 192*e4b17023SJohn Marino _DEFINE_VALARRAY_OPERATOR(/, __divides) 193*e4b17023SJohn Marino _DEFINE_VALARRAY_OPERATOR(%, __modulus) 194*e4b17023SJohn Marino _DEFINE_VALARRAY_OPERATOR(+, __plus) 195*e4b17023SJohn Marino _DEFINE_VALARRAY_OPERATOR(-, __minus) 196*e4b17023SJohn Marino _DEFINE_VALARRAY_OPERATOR(^, __bitwise_xor) 197*e4b17023SJohn Marino _DEFINE_VALARRAY_OPERATOR(&, __bitwise_and) 198*e4b17023SJohn Marino _DEFINE_VALARRAY_OPERATOR(|, __bitwise_or) 199*e4b17023SJohn Marino _DEFINE_VALARRAY_OPERATOR(<<, __shift_left) 200*e4b17023SJohn Marino _DEFINE_VALARRAY_OPERATOR(>>, __shift_right) 201*e4b17023SJohn Marino 202*e4b17023SJohn Marino #undef _DEFINE_VALARRAY_OPERATOR 203*e4b17023SJohn Marino 204*e4b17023SJohn Marino // @} group numeric_arrays 205*e4b17023SJohn Marino 206*e4b17023SJohn Marino _GLIBCXX_END_NAMESPACE_VERSION 207*e4b17023SJohn Marino } // namespace 208*e4b17023SJohn Marino 209*e4b17023SJohn Marino #endif /* _MASK_ARRAY_H */ 210