// The template and inlines for the -*- C++ -*- mask_array class.

// Copyright (C) 1997-2001 Free Software Foundation, Inc.
//
// This file is part of the GNU ISO C++ Library.  This library is free
// software; you can redistribute it and/or modify it under the
// terms of the GNU General Public License as published by the
// Free Software Foundation; either version 2, or (at your option)
// any later version.

// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License along
// with this library; see the file COPYING.  If not, write to the Free
// Software Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307,
// USA.

// As a special exception, you may use this file as part of a free software
// library without restriction.  Specifically, if other files instantiate
// templates or use macros or inline functions from this file, or you compile
// this file and link it with other files to produce an executable, this
// file does not by itself cause the resulting executable to be covered by
// the GNU General Public License.  This exception does not however
// invalidate any other reasons why the executable file might be covered by
// the GNU General Public License.

// Written by Gabriel Dos Reis <Gabriel.Dos-Reis@DPTMaths.ENS-Cachan.Fr>

#ifndef _CPP_BITS_MASK_ARRAY_H
#define _CPP_BITS_MASK_ARRAY_H 1

#pragma GCC system_header

namespace std {

    template <class _Tp> class mask_array
    { 
    public:
        typedef _Tp value_type;
    
        void operator=  (const valarray<_Tp>&) const;
        void operator*= (const valarray<_Tp>&) const;
        void operator/= (const valarray<_Tp>&) const;
        void operator%= (const valarray<_Tp>&) const;
        void operator+= (const valarray<_Tp>&) const; 
        void operator-= (const valarray<_Tp>&) const;
        void operator^= (const valarray<_Tp>&) const;  
        void operator&= (const valarray<_Tp>&) const;
        void operator|= (const valarray<_Tp>&) const;
        void operator<<=(const valarray<_Tp>&) const;  
        void operator>>=(const valarray<_Tp>&) const; 
        void operator= (const _Tp&);
    
        //        ~mask_array ();
        
        template<class _Dom>
        void operator=  (const _Expr<_Dom,_Tp>&) const;
        template<class _Dom>
        void operator*= (const _Expr<_Dom,_Tp>&) const;
        template<class _Dom>
        void operator/= (const _Expr<_Dom,_Tp>&) const;
        template<class _Dom>
        void operator%= (const _Expr<_Dom,_Tp>&) const;
        template<class _Dom>
        void operator+= (const _Expr<_Dom,_Tp>&) const;
        template<class _Dom>
        void operator-= (const _Expr<_Dom,_Tp>&) const;
        template<class _Dom>
        void operator^= (const _Expr<_Dom,_Tp>&) const;
        template<class _Dom>
        void operator&= (const _Expr<_Dom,_Tp>&) const;
        template<class _Dom>
        void operator|= (const _Expr<_Dom,_Tp>&) const;
        template<class _Dom>
        void operator<<=(const _Expr<_Dom,_Tp>&) const;
        template<class _Dom>
        void operator>>=(const _Expr<_Dom,_Tp>&) const; 

    private:
        mask_array (_Array<_Tp>, size_t, _Array<bool>);
        friend class valarray<_Tp>;
        
        const size_t       _M_sz;
        const _Array<bool> _M_mask;
        const _Array<_Tp>   _M_array;
        
        mask_array (const mask_array&);
        
        // not implemented
        mask_array ();
        mask_array& operator= (const mask_array&);
    };


    template<typename _Tp>
    inline mask_array<_Tp>::mask_array (const mask_array<_Tp>& a)
            : _M_sz (a._M_sz), _M_mask (a._M_mask), _M_array (a._M_array) {}

    template<typename _Tp>
    inline 
    mask_array<_Tp>::mask_array (_Array<_Tp> __a, size_t __s, _Array<bool> __m)
            : _M_sz (__s), _M_mask (__m), _M_array (__a) {}
    
    //    template<typename _Tp>
    //    inline mask_array<_Tp>::~mask_array () {}
    
    template<typename _Tp>
    inline void
    mask_array<_Tp>::operator= (const _Tp& __t)
    { __valarray_fill (_M_array, _M_sz, _M_mask, __t); }
    
    template<typename _Tp>
    inline void
    mask_array<_Tp>::operator= (const valarray<_Tp>& __v) const
    { __valarray_copy (_Array<_Tp> (__v), __v.size (), _M_array, _M_mask); }

    template<typename _Tp>
    template<class E>
    inline void
    mask_array<_Tp>::operator= (const _Expr<E, _Tp>& __e) const
    { __valarray_copy (__e, __e.size (), _M_array, _M_mask); }

#undef _DEFINE_VALARRAY_OPERATOR
#define _DEFINE_VALARRAY_OPERATOR(op, name)				\
template<typename _Tp>							\
inline void								\
mask_array<_Tp>::operator op##= (const valarray<_Tp>& __v) const	\
{									\
  _Array_augmented_##name (_M_array, _M_mask, 				\
                           _Array<_Tp> (__v), __v.size ());		\
}									\
									\
template<typename _Tp> template<class E>				\
inline void								\
mask_array<_Tp>::operator op##= (const _Expr<E, _Tp>& __e) const	\
{									\
  _Array_augmented_##name (_M_array, _M_mask, __e, __e.size ());	\
}

_DEFINE_VALARRAY_OPERATOR(*, multiplies)
_DEFINE_VALARRAY_OPERATOR(/, divides)
_DEFINE_VALARRAY_OPERATOR(%, modulus)
_DEFINE_VALARRAY_OPERATOR(+, plus)
_DEFINE_VALARRAY_OPERATOR(-, minus)
_DEFINE_VALARRAY_OPERATOR(^, xor)
_DEFINE_VALARRAY_OPERATOR(&, and)
_DEFINE_VALARRAY_OPERATOR(|, or)
_DEFINE_VALARRAY_OPERATOR(<<, shift_left)
_DEFINE_VALARRAY_OPERATOR(>>, shift_right)

#undef _DEFINE_VALARRAY_OPERATOR    
    
} // std::

#endif /* _CPP_BITS_MASK_ARRAY_H */

// Local Variables:
// mode:c++
// End:
