00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038 #ifndef _CPP_BITS_MASK_ARRAY_H
00039 #define _CPP_BITS_MASK_ARRAY_H 1
00040
00041 #pragma GCC system_header
00042
00043 namespace std {
00044
00045 template <class _Tp>
00046 class mask_array
00047 {
00048 public:
00049 typedef _Tp value_type;
00050
00051 void operator=(const valarray<_Tp>&) const;
00052 void operator*=(const valarray<_Tp>&) const;
00053 void operator/=(const valarray<_Tp>&) const;
00054 void operator%=(const valarray<_Tp>&) const;
00055 void operator+=(const valarray<_Tp>&) const;
00056 void operator-=(const valarray<_Tp>&) const;
00057 void operator^=(const valarray<_Tp>&) const;
00058 void operator&=(const valarray<_Tp>&) const;
00059 void operator|=(const valarray<_Tp>&) const;
00060 void operator<<=(const valarray<_Tp>&) const;
00061 void operator>>=(const valarray<_Tp>&) const;
00062 void operator=(const _Tp&) const;
00063
00064
00065
00066 template<class _Dom>
00067 void operator=(const _Expr<_Dom,_Tp>&) const;
00068 template<class _Dom>
00069 void operator*=(const _Expr<_Dom,_Tp>&) const;
00070 template<class _Dom>
00071 void operator/=(const _Expr<_Dom,_Tp>&) const;
00072 template<class _Dom>
00073 void operator%=(const _Expr<_Dom,_Tp>&) const;
00074 template<class _Dom>
00075 void operator+=(const _Expr<_Dom,_Tp>&) const;
00076 template<class _Dom>
00077 void operator-=(const _Expr<_Dom,_Tp>&) const;
00078 template<class _Dom>
00079 void operator^=(const _Expr<_Dom,_Tp>&) const;
00080 template<class _Dom>
00081 void operator&=(const _Expr<_Dom,_Tp>&) const;
00082 template<class _Dom>
00083 void operator|=(const _Expr<_Dom,_Tp>&) const;
00084 template<class _Dom>
00085 void operator<<=(const _Expr<_Dom,_Tp>&) const;
00086 template<class _Dom>
00087 void operator>>=(const _Expr<_Dom,_Tp>&) const;
00088
00089 private:
00090 mask_array(_Array<_Tp>, size_t, _Array<bool>);
00091 friend class valarray<_Tp>;
00092
00093 const size_t _M_sz;
00094 const _Array<bool> _M_mask;
00095 const _Array<_Tp> _M_array;
00096
00097 mask_array (const mask_array&);
00098
00099
00100 mask_array();
00101 mask_array& operator=(const mask_array&);
00102 };
00103
00104
00105 template<typename _Tp>
00106 inline mask_array<_Tp>::mask_array(const mask_array<_Tp>& a)
00107 : _M_sz(a._M_sz), _M_mask(a._M_mask), _M_array(a._M_array) {}
00108
00109 template<typename _Tp>
00110 inline
00111 mask_array<_Tp>::mask_array(_Array<_Tp> __a, size_t __s, _Array<bool> __m)
00112 : _M_sz(__s), _M_mask(__m), _M_array(__a) {}
00113
00114 template<typename _Tp>
00115 inline void
00116 mask_array<_Tp>::operator=(const _Tp& __t) const
00117 { __valarray_fill(_M_array, _M_sz, _M_mask, __t); }
00118
00119 template<typename _Tp>
00120 inline void
00121 mask_array<_Tp>::operator=(const valarray<_Tp>& __v) const
00122 { __valarray_copy(_Array<_Tp>(__v), __v.size(), _M_array, _M_mask); }
00123
00124 template<typename _Tp>
00125 template<class _Ex>
00126 inline void
00127 mask_array<_Tp>::operator=(const _Expr<_Ex, _Tp>& __e) const
00128 { __valarray_copy(__e, __e.size(), _M_array, _M_mask); }
00129
00130 #undef _DEFINE_VALARRAY_OPERATOR
00131 #define _DEFINE_VALARRAY_OPERATOR(_Op, _Name) \
00132 template<typename _Tp> \
00133 inline void \
00134 mask_array<_Tp>::operator _Op##=(const valarray<_Tp>& __v) const \
00135 { \
00136 _Array_augmented_##_Name(_M_array, _M_mask, \
00137 _Array<_Tp>(__v), __v.size()); \
00138 } \
00139 \
00140 template<typename _Tp> \
00141 template<class _Dom> \
00142 inline void \
00143 mask_array<_Tp>::operator _Op##=(const _Expr<_Dom, _Tp>& __e) const\
00144 { \
00145 _Array_augmented_##_Name(_M_array, _M_mask, __e, __e.size()); \
00146 }
00147
00148 _DEFINE_VALARRAY_OPERATOR(*, __multiplies)
00149 _DEFINE_VALARRAY_OPERATOR(/, __divides)
00150 _DEFINE_VALARRAY_OPERATOR(%, __modulus)
00151 _DEFINE_VALARRAY_OPERATOR(+, __plus)
00152 _DEFINE_VALARRAY_OPERATOR(-, __minus)
00153 _DEFINE_VALARRAY_OPERATOR(^, __bitwise_xor)
00154 _DEFINE_VALARRAY_OPERATOR(&, __bitwise_and)
00155 _DEFINE_VALARRAY_OPERATOR(|, __bitwise_or)
00156 _DEFINE_VALARRAY_OPERATOR(<<, __shift_left)
00157 _DEFINE_VALARRAY_OPERATOR(>>, __shift_right)
00158
00159 #undef _DEFINE_VALARRAY_OPERATOR
00160
00161 }
00162
00163 #endif
00164
00165
00166
00167