Mercurial > hg > segmenter-vamp-plugin
diff armadillo-2.4.4/include/armadillo_bits/subview_bones.hpp @ 0:8b6102e2a9b0
Armadillo Library
author | maxzanoni76 <max.zanoni@eecs.qmul.ac.uk> |
---|---|
date | Wed, 11 Apr 2012 09:27:06 +0100 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/armadillo-2.4.4/include/armadillo_bits/subview_bones.hpp Wed Apr 11 09:27:06 2012 +0100 @@ -0,0 +1,254 @@ +// Copyright (C) 2008-2011 NICTA (www.nicta.com.au) +// Copyright (C) 2008-2011 Conrad Sanderson +// Copyright (C) 2011 James Sanders +// +// This file is part of the Armadillo C++ library. +// It is provided without any warranty of fitness +// for any purpose. You can redistribute this file +// and/or modify it under the terms of the GNU +// Lesser General Public License (LGPL) as published +// by the Free Software Foundation, either version 3 +// of the License or (at your option) any later version. +// (see http://www.opensource.org/licenses for more info) + + +//! \addtogroup subview +//! @{ + + +//! Class for storing data required to construct or apply operations to a submatrix +//! (i.e. where the submatrix starts and ends as well as a reference/pointer to the original matrix), +template<typename eT> +class subview : public Base<eT, subview<eT> > + { + public: arma_aligned const Mat<eT>& m; + protected: arma_aligned Mat<eT>* m_ptr; + + public: + + typedef eT elem_type; + typedef typename get_pod_type<elem_type>::result pod_type; + + const uword aux_row1; + const uword aux_col1; + + const uword n_rows; + const uword n_cols; + const uword n_elem; + + + protected: + + arma_inline subview(const Mat<eT>& in_m, const uword in_row1, const uword in_col1, const uword in_n_rows, const uword in_n_cols); + arma_inline subview( Mat<eT>& in_m, const uword in_row1, const uword in_col1, const uword in_n_rows, const uword in_n_cols); + + + public: + + inline ~subview(); + + inline void operator+= (const eT val); + inline void operator-= (const eT val); + inline void operator*= (const eT val); + inline void operator/= (const eT val); + + // deliberately returning void + template<typename T1> inline void operator= (const Base<eT,T1>& x); + template<typename T1> inline void operator+= (const Base<eT,T1>& x); + template<typename T1> inline void operator-= (const Base<eT,T1>& x); + template<typename T1> inline void operator%= (const Base<eT,T1>& x); + template<typename T1> inline void operator/= (const Base<eT,T1>& x); + + inline void operator= (const subview& x); + inline void operator+= (const subview& x); + inline void operator-= (const subview& x); + inline void operator%= (const subview& x); + inline void operator/= (const subview& x); + + inline static void extract(Mat<eT>& out, const subview& in); + + inline static void plus_inplace(Mat<eT>& out, const subview& in); + inline static void minus_inplace(Mat<eT>& out, const subview& in); + inline static void schur_inplace(Mat<eT>& out, const subview& in); + inline static void div_inplace(Mat<eT>& out, const subview& in); + + inline void fill(const eT val); + inline void zeros(); + inline void ones(); + inline void eye(); + + inline eT& operator[](const uword i); + inline eT operator[](const uword i) const; + + inline eT& operator()(const uword i); + inline eT operator()(const uword i) const; + + inline eT& operator()(const uword in_row, const uword in_col); + inline eT operator()(const uword in_row, const uword in_col) const; + + inline eT& at(const uword in_row, const uword in_col); + inline eT at(const uword in_row, const uword in_col) const; + + arma_inline eT* colptr(const uword in_col); + arma_inline const eT* colptr(const uword in_col) const; + + inline bool check_overlap(const subview& x) const; + + inline bool is_vec() const; + + inline subview_row<eT> row(const uword row_num); + inline const subview_row<eT> row(const uword row_num) const; + + inline subview_row<eT> operator()(const uword row_num, const span& col_span); + inline const subview_row<eT> operator()(const uword row_num, const span& col_span) const; + + inline subview_col<eT> col(const uword col_num); + inline const subview_col<eT> col(const uword col_num) const; + + inline subview_col<eT> operator()(const span& row_span, const uword col_num); + inline const subview_col<eT> operator()(const span& row_span, const uword col_num) const; + + inline Col<eT> unsafe_col(const uword col_num); + inline const Col<eT> unsafe_col(const uword col_num) const; + + inline subview<eT> rows(const uword in_row1, const uword in_row2); + inline const subview<eT> rows(const uword in_row1, const uword in_row2) const; + + inline subview<eT> cols(const uword in_col1, const uword in_col2); + inline const subview<eT> cols(const uword in_col1, const uword in_col2) const; + + inline subview<eT> submat(const uword in_row1, const uword in_col1, const uword in_row2, const uword in_col2); + inline const subview<eT> submat(const uword in_row1, const uword in_col1, const uword in_row2, const uword in_col2) const; + + inline subview<eT> submat (const span& row_span, const span& col_span); + inline const subview<eT> submat (const span& row_span, const span& col_span) const; + + inline subview<eT> operator()(const span& row_span, const span& col_span); + inline const subview<eT> operator()(const span& row_span, const span& col_span) const; + + inline diagview<eT> diag(const sword in_id = 0); + inline const diagview<eT> diag(const sword in_id = 0) const; + + inline void swap_rows(const uword in_row1, const uword in_row2); + inline void swap_cols(const uword in_col1, const uword in_col2); + + + // // primitive forward iterator + // class iter + // { + // public: + // + // inline iter(const subview<eT>& in_M); + // + // arma_inline eT operator* () const; + // + // inline void operator++(); + // inline void operator++(int); + // + // + // private: + // + // arma_aligned const eT* mem; + // + // arma_aligned uword n_rows; + // + // arma_aligned uword row_start; + // arma_aligned uword row_end_p1; + // + // arma_aligned uword row; + // arma_aligned uword col; + // arma_aligned uword i; + // }; + + + private: + + friend class Mat<eT>; + subview(); + }; + + + +template<typename eT> +class subview_col : public subview<eT> + { + public: + + typedef eT elem_type; + typedef typename get_pod_type<elem_type>::result pod_type; + + inline void operator= (const subview<eT>& x); + inline void operator= (const subview_col& x); + + template<typename T1> + inline void operator= (const Base<eT,T1>& x); + + inline subview_col<eT> rows(const uword in_row1, const uword in_row2); + inline const subview_col<eT> rows(const uword in_row1, const uword in_row2) const; + + inline subview_col<eT> subvec(const uword in_row1, const uword in_row2); + inline const subview_col<eT> subvec(const uword in_row1, const uword in_row2) const; + + + protected: + + inline subview_col(const Mat<eT>& in_m, const uword in_col); + inline subview_col( Mat<eT>& in_m, const uword in_col); + + inline subview_col(const Mat<eT>& in_m, const uword in_col, const uword in_row1, const uword in_n_rows); + inline subview_col( Mat<eT>& in_m, const uword in_col, const uword in_row1, const uword in_n_rows); + + + private: + + friend class Mat<eT>; + friend class Col<eT>; + friend class subview<eT>; + + subview_col(); + }; + + + +template<typename eT> +class subview_row : public subview<eT> + { + public: + + typedef eT elem_type; + typedef typename get_pod_type<elem_type>::result pod_type; + + inline void operator= (const subview<eT>& x); + inline void operator= (const subview_row& x); + + template<typename T1> + inline void operator= (const Base<eT,T1>& x); + + inline subview_row<eT> cols(const uword in_col1, const uword in_col2); + inline const subview_row<eT> cols(const uword in_col1, const uword in_col2) const; + + inline subview_row<eT> subvec(const uword in_col1, const uword in_col2); + inline const subview_row<eT> subvec(const uword in_col1, const uword in_col2) const; + + + protected: + + inline subview_row(const Mat<eT>& in_m, const uword in_row); + inline subview_row( Mat<eT>& in_m, const uword in_row); + + inline subview_row(const Mat<eT>& in_m, const uword in_row, const uword in_col1, const uword in_n_cols); + inline subview_row( Mat<eT>& in_m, const uword in_row, const uword in_col1, const uword in_n_cols); + + + private: + + friend class Mat<eT>; + friend class Row<eT>; + friend class subview<eT>; + + subview_row(); + }; + + + +//! @}