Chris@49: // Copyright (C) 2008-2013 NICTA (www.nicta.com.au)
Chris@49: // Copyright (C) 2008-2013 Conrad Sanderson
Chris@49: // 
Chris@49: // This Source Code Form is subject to the terms of the Mozilla Public
Chris@49: // License, v. 2.0. If a copy of the MPL was not distributed with this
Chris@49: // file, You can obtain one at http://mozilla.org/MPL/2.0/.
Chris@49: 
Chris@49: 
Chris@49: //! \addtogroup diagview
Chris@49: //! @{
Chris@49: 
Chris@49: 
Chris@49: //! Class for storing data required to extract and set the diagonals of a matrix
Chris@49: template<typename eT>
Chris@49: class diagview : public Base<eT, diagview<eT> >
Chris@49:   {
Chris@49:   public:
Chris@49:   
Chris@49:   typedef eT                                elem_type;
Chris@49:   typedef typename get_pod_type<eT>::result pod_type;
Chris@49:   
Chris@49:   arma_aligned const Mat<eT>& m;
Chris@49:   
Chris@49:   static const bool is_row = false;
Chris@49:   static const bool is_col = true;
Chris@49:   
Chris@49:   const uword row_offset;
Chris@49:   const uword col_offset;
Chris@49:   
Chris@49:   const uword n_rows;     // equal to n_elem
Chris@49:   const uword n_elem;
Chris@49:   
Chris@49:   static const uword n_cols = 1;
Chris@49:   
Chris@49:   
Chris@49:   protected:
Chris@49:   
Chris@49:   arma_inline diagview(const Mat<eT>& in_m, const uword in_row_offset, const uword in_col_offset, const uword len);
Chris@49:   
Chris@49:   
Chris@49:   public:
Chris@49:   
Chris@49:   inline ~diagview();
Chris@49:   
Chris@49:   inline void operator=(const diagview& x);
Chris@49:   
Chris@49:   inline void operator+=(const eT val);
Chris@49:   inline void operator-=(const eT val);
Chris@49:   inline void operator*=(const eT val);
Chris@49:   inline void operator/=(const eT val);
Chris@49:   
Chris@49:   template<typename T1> inline void operator= (const Base<eT,T1>& x);
Chris@49:   template<typename T1> inline void operator+=(const Base<eT,T1>& x);
Chris@49:   template<typename T1> inline void operator-=(const Base<eT,T1>& x);
Chris@49:   template<typename T1> inline void operator%=(const Base<eT,T1>& x);
Chris@49:   template<typename T1> inline void operator/=(const Base<eT,T1>& x);
Chris@49:   
Chris@49:   
Chris@49:   arma_inline eT  at_alt    (const uword ii) const;
Chris@49:   
Chris@49:   arma_inline eT& operator[](const uword ii);
Chris@49:   arma_inline eT  operator[](const uword ii) const;
Chris@49:   
Chris@49:   arma_inline eT&         at(const uword ii);
Chris@49:   arma_inline eT          at(const uword ii) const;
Chris@49:   
Chris@49:   arma_inline eT& operator()(const uword ii);
Chris@49:   arma_inline eT  operator()(const uword ii) const;
Chris@49:   
Chris@49:   arma_inline eT&         at(const uword in_n_row, const uword);
Chris@49:   arma_inline eT          at(const uword in_n_row, const uword) const;
Chris@49:    
Chris@49:   arma_inline eT& operator()(const uword in_n_row, const uword in_n_col);
Chris@49:   arma_inline eT  operator()(const uword in_n_row, const uword in_n_col) const;
Chris@49:   
Chris@49:   
Chris@49:   arma_inline const Op<diagview<eT>,op_htrans>  t() const;
Chris@49:   arma_inline const Op<diagview<eT>,op_htrans> ht() const;
Chris@49:   arma_inline const Op<diagview<eT>,op_strans> st() const;
Chris@49:   
Chris@49:   inline void fill(const eT val);
Chris@49:   inline void zeros();
Chris@49:   inline void ones();
Chris@49:   
Chris@49:   inline static void extract(Mat<eT>& out, const diagview& in);
Chris@49:   
Chris@49:   inline static void  plus_inplace(Mat<eT>& out, const diagview& in);
Chris@49:   inline static void minus_inplace(Mat<eT>& out, const diagview& in);
Chris@49:   inline static void schur_inplace(Mat<eT>& out, const diagview& in);
Chris@49:   inline static void   div_inplace(Mat<eT>& out, const diagview& in);
Chris@49:   
Chris@49:   
Chris@49:   private:
Chris@49:   
Chris@49:   friend class Mat<eT>;
Chris@49:   friend class subview<eT>;
Chris@49:   
Chris@49:   diagview();
Chris@49:   //diagview(const diagview&);  // making this private causes an error under gcc 4.1/4.2, but not 4.3
Chris@49:   };
Chris@49: 
Chris@49: 
Chris@49: //! @}