xiaoguangbin/AnalysisSystemForRadionuclide_qt_alg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
fe493fb636
AnalysisSystemForRadionucli.../include/armadillo_bits/subview_each_meat.hpp

1396 lines
29 KiB
C++
Raw Normal View History

Initial commit on main branch
2024-06-04 15:25:02 +08:00
// Copyright (C) 2012-2015 National ICT Australia (NICTA)
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
// -------------------------------------------------------------------
//
// Written by Conrad Sanderson - http://conradsanderson.id.au
//! \addtogroup subview_each
//! @{
//
//
// subview_each_common
template<typename parent, unsigned int mode>
inline
subview_each_common<parent,mode>::subview_each_common(const parent& in_P)
: P(in_P)
{
arma_extra_debug_sigprint();
}
template<typename parent, unsigned int mode>
arma_inline
const Mat<typename parent::elem_type>&
subview_each_common<parent,mode>::get_mat_ref_helper(const Mat<typename parent::elem_type>& X) const
{
return X;
}
template<typename parent, unsigned int mode>
arma_inline
const Mat<typename parent::elem_type>&
subview_each_common<parent,mode>::get_mat_ref_helper(const subview<typename parent::elem_type>& X) const
{
return X.m;
}
template<typename parent, unsigned int mode>
arma_inline
const Mat<typename parent::elem_type>&
subview_each_common<parent,mode>::get_mat_ref() const
{
return get_mat_ref_helper(P);
}
template<typename parent, unsigned int mode>
inline
void
subview_each_common<parent,mode>::check_size(const Mat<typename parent::elem_type>& A) const
{
if(arma_config::debug == true)
{
if(mode == 0)
{
if( (A.n_rows != P.n_rows) || (A.n_cols != 1) )
{
arma_stop( incompat_size_string(A) );
}
}
else
{
if( (A.n_rows != 1) || (A.n_cols != P.n_cols) )
{
arma_stop( incompat_size_string(A) );
}
}
}
}
template<typename parent, unsigned int mode>
arma_cold
inline
const std::string
subview_each_common<parent,mode>::incompat_size_string(const Mat<typename parent::elem_type>& A) const
{
std::stringstream tmp;
if(mode == 0)
{
tmp << "each_col(): incompatible size; expected " << P.n_rows << "x1" << ", got " << A.n_rows << 'x' << A.n_cols;
}
else
{
tmp << "each_row(): incompatible size; expected 1x" << P.n_cols << ", got " << A.n_rows << 'x' << A.n_cols;
}
return tmp.str();
}
//
//
// subview_each1
template<typename parent, unsigned int mode>
inline
subview_each1<parent,mode>::~subview_each1()
{
arma_extra_debug_sigprint();
}
template<typename parent, unsigned int mode>
inline
subview_each1<parent,mode>::subview_each1(const parent& in_P)
: subview_each_common<parent,mode>::subview_each_common(in_P)
{
arma_extra_debug_sigprint();
}
template<typename parent, unsigned int mode>
template<typename T1>
inline
void
subview_each1<parent,mode>::operator= (const Base<eT,T1>& in)
{
arma_extra_debug_sigprint();
parent& p = access::rw(subview_each_common<parent,mode>::P);
const unwrap_check<T1> tmp( in.get_ref(), (*this).get_mat_ref() );
const Mat<eT>& A = tmp.M;
subview_each_common<parent,mode>::check_size(A);
const eT* A_mem = A.memptr();
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
if(mode == 0) // each column
{
for(uword i=0; i < p_n_cols; ++i)
{
arrayops::copy( p.colptr(i), A_mem, p_n_rows );
}
}
else // each row
{
for(uword i=0; i < p_n_cols; ++i)
{
arrayops::inplace_set( p.colptr(i), A_mem[i], p_n_rows);
}
}
}
template<typename parent, unsigned int mode>
template<typename T1>
inline
void
subview_each1<parent,mode>::operator+= (const Base<eT,T1>& in)
{
arma_extra_debug_sigprint();
parent& p = access::rw(subview_each_common<parent,mode>::P);
const unwrap_check<T1> tmp( in.get_ref(), (*this).get_mat_ref() );
const Mat<eT>& A = tmp.M;
subview_each_common<parent,mode>::check_size(A);
const eT* A_mem = A.memptr();
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
if(mode == 0) // each column
{
for(uword i=0; i < p_n_cols; ++i)
{
arrayops::inplace_plus( p.colptr(i), A_mem, p_n_rows );
}
}
else // each row
{
for(uword i=0; i < p_n_cols; ++i)
{
arrayops::inplace_plus( p.colptr(i), A_mem[i], p_n_rows);
}
}
}
template<typename parent, unsigned int mode>
template<typename T1>
inline
void
subview_each1<parent,mode>::operator-= (const Base<eT,T1>& in)
{
arma_extra_debug_sigprint();
parent& p = access::rw(subview_each_common<parent,mode>::P);
const unwrap_check<T1> tmp( in.get_ref(), (*this).get_mat_ref() );
const Mat<eT>& A = tmp.M;
subview_each_common<parent,mode>::check_size(A);
const eT* A_mem = A.memptr();
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
if(mode == 0) // each column
{
for(uword i=0; i < p_n_cols; ++i)
{
arrayops::inplace_minus( p.colptr(i), A_mem, p_n_rows );
}
}
else // each row
{
for(uword i=0; i < p_n_cols; ++i)
{
arrayops::inplace_minus( p.colptr(i), A_mem[i], p_n_rows);
}
}
}
template<typename parent, unsigned int mode>
template<typename T1>
inline
void
subview_each1<parent,mode>::operator%= (const Base<eT,T1>& in)
{
arma_extra_debug_sigprint();
parent& p = access::rw(subview_each_common<parent,mode>::P);
const unwrap_check<T1> tmp( in.get_ref(), (*this).get_mat_ref() );
const Mat<eT>& A = tmp.M;
subview_each_common<parent,mode>::check_size(A);
const eT* A_mem = A.memptr();
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
if(mode == 0) // each column
{
for(uword i=0; i < p_n_cols; ++i)
{
arrayops::inplace_mul( p.colptr(i), A_mem, p_n_rows );
}
}
else // each row
{
for(uword i=0; i < p_n_cols; ++i)
{
arrayops::inplace_mul( p.colptr(i), A_mem[i], p_n_rows);
}
}
}
template<typename parent, unsigned int mode>
template<typename T1>
inline
void
subview_each1<parent,mode>::operator/= (const Base<eT,T1>& in)
{
arma_extra_debug_sigprint();
parent& p = access::rw(subview_each_common<parent,mode>::P);
const unwrap_check<T1> tmp( in.get_ref(), (*this).get_mat_ref() );
const Mat<eT>& A = tmp.M;
subview_each_common<parent,mode>::check_size(A);
const eT* A_mem = A.memptr();
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
if(mode == 0) // each column
{
for(uword i=0; i < p_n_cols; ++i)
{
arrayops::inplace_div( p.colptr(i), A_mem, p_n_rows );
}
}
else // each row
{
for(uword i=0; i < p_n_cols; ++i)
{
arrayops::inplace_div( p.colptr(i), A_mem[i], p_n_rows);
}
}
}
//
//
// subview_each2
template<typename parent, unsigned int mode, typename TB>
inline
subview_each2<parent,mode,TB>::~subview_each2()
{
arma_extra_debug_sigprint();
}
template<typename parent, unsigned int mode, typename TB>
inline
subview_each2<parent,mode,TB>::subview_each2(const parent& in_P, const Base<uword, TB>& in_indices)
: subview_each_common<parent,mode>::subview_each_common(in_P)
, base_indices(in_indices)
{
arma_extra_debug_sigprint();
}
template<typename parent, unsigned int mode, typename TB>
inline
void
subview_each2<parent,mode,TB>::check_indices(const Mat<uword>& indices) const
{
if(mode == 0)
{
arma_debug_check( ((indices.is_vec() == false) && (indices.is_empty() == false)), "each_col(): list of indices must be a vector" );
}
else
{
arma_debug_check( ((indices.is_vec() == false) && (indices.is_empty() == false)), "each_row(): list of indices must be a vector" );
}
}
template<typename parent, unsigned int mode, typename TB>
template<typename T1>
inline
void
subview_each2<parent,mode,TB>::operator= (const Base<eT,T1>& in)
{
arma_extra_debug_sigprint();
parent& p = access::rw(subview_each_common<parent,mode>::P);
const unwrap_check<T1> tmp( in.get_ref(), (*this).get_mat_ref() );
const Mat<eT>& A = tmp.M;
subview_each_common<parent,mode>::check_size(A);
const unwrap_check_mixed<TB> U( base_indices.get_ref(), (*this).get_mat_ref() );
check_indices(U.M);
const eT* A_mem = A.memptr();
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
const uword* indices_mem = U.M.memptr();
const uword N = U.M.n_elem;
if(mode == 0) // each column
{
for(uword i=0; i < N; ++i)
{
const uword col = indices_mem[i];
arma_debug_check( (col >= p_n_cols), "each_col(): index out of bounds" );
arrayops::copy( p.colptr(col), A_mem, p_n_rows );
}
}
else // each row
{
for(uword i=0; i < N; ++i)
{
const uword row = indices_mem[i];
arma_debug_check( (row >= p_n_rows), "each_row(): index out of bounds" );
for(uword col=0; col < p_n_cols; ++col)
{
p.at(row,col) = A_mem[col];
}
}
}
}
template<typename parent, unsigned int mode, typename TB>
template<typename T1>
inline
void
subview_each2<parent,mode,TB>::operator+= (const Base<eT,T1>& in)
{
arma_extra_debug_sigprint();
parent& p = access::rw(subview_each_common<parent,mode>::P);
const unwrap_check<T1> tmp( in.get_ref(), (*this).get_mat_ref() );
const Mat<eT>& A = tmp.M;
subview_each_common<parent,mode>::check_size(A);
const unwrap_check_mixed<TB> U( base_indices.get_ref(), (*this).get_mat_ref() );
check_indices(U.M);
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
const uword* indices_mem = U.M.memptr();
const uword N = U.M.n_elem;
if(mode == 0) // each column
{
const eT* A_mem = A.memptr();
for(uword i=0; i < N; ++i)
{
const uword col = indices_mem[i];
arma_debug_check( (col >= p_n_cols), "each_col(): index out of bounds" );
arrayops::inplace_plus( p.colptr(col), A_mem, p_n_rows );
}
}
else // each row
{
for(uword i=0; i < N; ++i)
{
const uword row = indices_mem[i];
arma_debug_check( (row >= p_n_rows), "each_row(): index out of bounds" );
p.row(row) += A;
}
}
}
template<typename parent, unsigned int mode, typename TB>
template<typename T1>
inline
void
subview_each2<parent,mode,TB>::operator-= (const Base<eT,T1>& in)
{
arma_extra_debug_sigprint();
parent& p = access::rw(subview_each_common<parent,mode>::P);
const unwrap_check<T1> tmp( in.get_ref(), (*this).get_mat_ref() );
const Mat<eT>& A = tmp.M;
subview_each_common<parent,mode>::check_size(A);
const unwrap_check_mixed<TB> U( base_indices.get_ref(), (*this).get_mat_ref() );
check_indices(U.M);
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
const uword* indices_mem = U.M.memptr();
const uword N = U.M.n_elem;
if(mode == 0) // each column
{
const eT* A_mem = A.memptr();
for(uword i=0; i < N; ++i)
{
const uword col = indices_mem[i];
arma_debug_check( (col >= p_n_cols), "each_col(): index out of bounds" );
arrayops::inplace_minus( p.colptr(col), A_mem, p_n_rows );
}
}
else // each row
{
for(uword i=0; i < N; ++i)
{
const uword row = indices_mem[i];
arma_debug_check( (row >= p_n_rows), "each_row(): index out of bounds" );
p.row(row) -= A;
}
}
}
template<typename parent, unsigned int mode, typename TB>
template<typename T1>
inline
void
subview_each2<parent,mode,TB>::operator%= (const Base<eT,T1>& in)
{
arma_extra_debug_sigprint();
parent& p = access::rw(subview_each_common<parent,mode>::P);
const unwrap_check<T1> tmp( in.get_ref(), (*this).get_mat_ref() );
const Mat<eT>& A = tmp.M;
subview_each_common<parent,mode>::check_size(A);
const unwrap_check_mixed<TB> U( base_indices.get_ref(), (*this).get_mat_ref() );
check_indices(U.M);
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
const uword* indices_mem = U.M.memptr();
const uword N = U.M.n_elem;
if(mode == 0) // each column
{
const eT* A_mem = A.memptr();
for(uword i=0; i < N; ++i)
{
const uword col = indices_mem[i];
arma_debug_check( (col >= p_n_cols), "each_col(): index out of bounds" );
arrayops::inplace_mul( p.colptr(col), A_mem, p_n_rows );
}
}
else // each row
{
for(uword i=0; i < N; ++i)
{
const uword row = indices_mem[i];
arma_debug_check( (row >= p_n_rows), "each_row(): index out of bounds" );
p.row(row) %= A;
}
}
}
template<typename parent, unsigned int mode, typename TB>
template<typename T1>
inline
void
subview_each2<parent,mode,TB>::operator/= (const Base<eT,T1>& in)
{
arma_extra_debug_sigprint();
parent& p = access::rw(subview_each_common<parent,mode>::P);
const unwrap_check<T1> tmp( in.get_ref(), (*this).get_mat_ref() );
const Mat<eT>& A = tmp.M;
subview_each_common<parent,mode>::check_size(A);
const unwrap_check_mixed<TB> U( base_indices.get_ref(), (*this).get_mat_ref() );
check_indices(U.M);
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
const uword* indices_mem = U.M.memptr();
const uword N = U.M.n_elem;
if(mode == 0) // each column
{
const eT* A_mem = A.memptr();
for(uword i=0; i < N; ++i)
{
const uword col = indices_mem[i];
arma_debug_check( (col >= p_n_cols), "each_col(): index out of bounds" );
arrayops::inplace_div( p.colptr(col), A_mem, p_n_rows );
}
}
else // each row
{
for(uword i=0; i < N; ++i)
{
const uword row = indices_mem[i];
arma_debug_check( (row >= p_n_rows), "each_row(): index out of bounds" );
p.row(row) /= A;
}
}
}
//
//
// subview_each1_aux
template<typename parent, unsigned int mode, typename T2>
inline
Mat<typename parent::elem_type>
subview_each1_aux::operator_plus
(
const subview_each1<parent,mode>& X,
const Base<typename parent::elem_type,T2>& Y
)
{
arma_extra_debug_sigprint();
typedef typename parent::elem_type eT;
const parent& p = X.P;
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
Mat<eT> out(p_n_rows, p_n_cols);
const quasi_unwrap<T2> tmp(Y.get_ref());
const Mat<eT>& A = tmp.M;
X.check_size(A);
const eT* A_mem = A.memptr();
if(mode == 0) // each column
{
for(uword i=0; i < p_n_cols; ++i)
{
const eT* p_mem = p.colptr(i);
eT* out_mem = out.colptr(i);
for(uword row=0; row < p_n_rows; ++row)
{
out_mem[row] = p_mem[row] + A_mem[row];
}
}
}
if(mode == 1) // each row
{
for(uword i=0; i < p_n_cols; ++i)
{
const eT* p_mem = p.colptr(i);
eT* out_mem = out.colptr(i);
const eT A_val = A_mem[i];
for(uword row=0; row < p_n_rows; ++row)
{
out_mem[row] = p_mem[row] + A_val;
}
}
}
return out;
}
template<typename parent, unsigned int mode, typename T2>
inline
Mat<typename parent::elem_type>
subview_each1_aux::operator_minus
(
const subview_each1<parent,mode>& X,
const Base<typename parent::elem_type,T2>& Y
)
{
arma_extra_debug_sigprint();
typedef typename parent::elem_type eT;
const parent& p = X.P;
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
Mat<eT> out(p_n_rows, p_n_cols);
const quasi_unwrap<T2> tmp(Y.get_ref());
const Mat<eT>& A = tmp.M;
X.check_size(A);
const eT* A_mem = A.memptr();
if(mode == 0) // each column
{
for(uword i=0; i < p_n_cols; ++i)
{
const eT* p_mem = p.colptr(i);
eT* out_mem = out.colptr(i);
for(uword row=0; row < p_n_rows; ++row)
{
out_mem[row] = p_mem[row] - A_mem[row];
}
}
}
if(mode == 1) // each row
{
for(uword i=0; i < p_n_cols; ++i)
{
const eT* p_mem = p.colptr(i);
eT* out_mem = out.colptr(i);
const eT A_val = A_mem[i];
for(uword row=0; row < p_n_rows; ++row)
{
out_mem[row] = p_mem[row] - A_val;
}
}
}
return out;
}
template<typename T1, typename parent, unsigned int mode>
inline
Mat<typename parent::elem_type>
subview_each1_aux::operator_minus
(
const Base<typename parent::elem_type,T1>& X,
const subview_each1<parent,mode>& Y
)
{
arma_extra_debug_sigprint();
typedef typename parent::elem_type eT;
const parent& p = Y.P;
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
Mat<eT> out(p_n_rows, p_n_cols);
const quasi_unwrap<T1> tmp(X.get_ref());
const Mat<eT>& A = tmp.M;
Y.check_size(A);
const eT* A_mem = A.memptr();
if(mode == 0) // each column
{
for(uword i=0; i < p_n_cols; ++i)
{
const eT* p_mem = p.colptr(i);
eT* out_mem = out.colptr(i);
for(uword row=0; row < p_n_rows; ++row)
{
out_mem[row] = A_mem[row] - p_mem[row];
}
}
}
if(mode == 1) // each row
{
for(uword i=0; i < p_n_cols; ++i)
{
const eT* p_mem = p.colptr(i);
eT* out_mem = out.colptr(i);
const eT A_val = A_mem[i];
for(uword row=0; row < p_n_rows; ++row)
{
out_mem[row] = A_val - p_mem[row];
}
}
}
return out;
}
template<typename parent, unsigned int mode, typename T2>
inline
Mat<typename parent::elem_type>
subview_each1_aux::operator_schur
(
const subview_each1<parent,mode>& X,
const Base<typename parent::elem_type,T2>& Y
)
{
arma_extra_debug_sigprint();
typedef typename parent::elem_type eT;
const parent& p = X.P;
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
Mat<eT> out(p_n_rows, p_n_cols);
const quasi_unwrap<T2> tmp(Y.get_ref());
const Mat<eT>& A = tmp.M;
X.check_size(A);
const eT* A_mem = A.memptr();
if(mode == 0) // each column
{
for(uword i=0; i < p_n_cols; ++i)
{
const eT* p_mem = p.colptr(i);
eT* out_mem = out.colptr(i);
for(uword row=0; row < p_n_rows; ++row)
{
out_mem[row] = p_mem[row] * A_mem[row];
}
}
}
if(mode == 1) // each row
{
for(uword i=0; i < p_n_cols; ++i)
{
const eT* p_mem = p.colptr(i);
eT* out_mem = out.colptr(i);
const eT A_val = A_mem[i];
for(uword row=0; row < p_n_rows; ++row)
{
out_mem[row] = p_mem[row] * A_val;
}
}
}
return out;
}
template<typename parent, unsigned int mode, typename T2>
inline
Mat<typename parent::elem_type>
subview_each1_aux::operator_div
(
const subview_each1<parent,mode>& X,
const Base<typename parent::elem_type,T2>& Y
)
{
arma_extra_debug_sigprint();
typedef typename parent::elem_type eT;
const parent& p = X.P;
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
Mat<eT> out(p_n_rows, p_n_cols);
const quasi_unwrap<T2> tmp(Y.get_ref());
const Mat<eT>& A = tmp.M;
X.check_size(A);
const eT* A_mem = A.memptr();
if(mode == 0) // each column
{
for(uword i=0; i < p_n_cols; ++i)
{
const eT* p_mem = p.colptr(i);
eT* out_mem = out.colptr(i);
for(uword row=0; row < p_n_rows; ++row)
{
out_mem[row] = p_mem[row] / A_mem[row];
}
}
}
if(mode == 1) // each row
{
for(uword i=0; i < p_n_cols; ++i)
{
const eT* p_mem = p.colptr(i);
eT* out_mem = out.colptr(i);
const eT A_val = A_mem[i];
for(uword row=0; row < p_n_rows; ++row)
{
out_mem[row] = p_mem[row] / A_val;
}
}
}
return out;
}
template<typename T1, typename parent, unsigned int mode>
inline
Mat<typename parent::elem_type>
subview_each1_aux::operator_div
(
const Base<typename parent::elem_type,T1>& X,
const subview_each1<parent,mode>& Y
)
{
arma_extra_debug_sigprint();
typedef typename parent::elem_type eT;
const parent& p = Y.P;
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
Mat<eT> out(p_n_rows, p_n_cols);
const quasi_unwrap<T1> tmp(X.get_ref());
const Mat<eT>& A = tmp.M;
Y.check_size(A);
const eT* A_mem = A.memptr();
if(mode == 0) // each column
{
for(uword i=0; i < p_n_cols; ++i)
{
const eT* p_mem = p.colptr(i);
eT* out_mem = out.colptr(i);
for(uword row=0; row < p_n_rows; ++row)
{
out_mem[row] = A_mem[row] / p_mem[row];
}
}
}
if(mode == 1) // each row
{
for(uword i=0; i < p_n_cols; ++i)
{
const eT* p_mem = p.colptr(i);
eT* out_mem = out.colptr(i);
const eT A_val = A_mem[i];
for(uword row=0; row < p_n_rows; ++row)
{
out_mem[row] = A_val / p_mem[row];
}
}
}
return out;
}
//
//
// subview_each2_aux
template<typename parent, unsigned int mode, typename TB, typename T2>
inline
Mat<typename parent::elem_type>
subview_each2_aux::operator_plus
(
const subview_each2<parent,mode,TB>& X,
const Base<typename parent::elem_type,T2>& Y
)
{
arma_extra_debug_sigprint();
typedef typename parent::elem_type eT;
const parent& p = X.P;
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
Mat<eT> out = p;
const quasi_unwrap<T2> tmp(Y.get_ref());
const Mat<eT>& A = tmp.M;
const unwrap<TB> U(X.base_indices.get_ref());
X.check_size(A);
X.check_indices(U.M);
const uword* indices_mem = U.M.memptr();
const uword N = U.M.n_elem;
if(mode == 0) // process columns
{
const eT* A_mem = A.memptr();
for(uword i=0; i < N; ++i)
{
const uword col = indices_mem[i];
arma_debug_check( (col >= p_n_cols), "each_col(): index out of bounds" );
arrayops::inplace_plus( out.colptr(col), A_mem, p_n_rows );
}
}
if(mode == 1) // process rows
{
for(uword i=0; i < N; ++i)
{
const uword row = indices_mem[i];
arma_debug_check( (row >= p_n_rows), "each_row(): index out of bounds" );
out.row(row) += A;
}
}
return out;
}
template<typename parent, unsigned int mode, typename TB, typename T2>
inline
Mat<typename parent::elem_type>
subview_each2_aux::operator_minus
(
const subview_each2<parent,mode,TB>& X,
const Base<typename parent::elem_type,T2>& Y
)
{
arma_extra_debug_sigprint();
typedef typename parent::elem_type eT;
const parent& p = X.P;
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
Mat<eT> out = p;
const quasi_unwrap<T2> tmp(Y.get_ref());
const Mat<eT>& A = tmp.M;
const unwrap<TB> U(X.base_indices.get_ref());
X.check_size(A);
X.check_indices(U.M);
const uword* indices_mem = U.M.memptr();
const uword N = U.M.n_elem;
if(mode == 0) // process columns
{
const eT* A_mem = A.memptr();
for(uword i=0; i < N; ++i)
{
const uword col = indices_mem[i];
arma_debug_check( (col >= p_n_cols), "each_col(): index out of bounds" );
arrayops::inplace_minus( out.colptr(col), A_mem, p_n_rows );
}
}
if(mode == 1) // process rows
{
for(uword i=0; i < N; ++i)
{
const uword row = indices_mem[i];
arma_debug_check( (row >= p_n_rows), "each_row(): index out of bounds" );
out.row(row) -= A;
}
}
return out;
}
template<typename T1, typename parent, unsigned int mode, typename TB>
inline
Mat<typename parent::elem_type>
subview_each2_aux::operator_minus
(
const Base<typename parent::elem_type,T1>& X,
const subview_each2<parent,mode,TB>& Y
)
{
arma_extra_debug_sigprint();
typedef typename parent::elem_type eT;
const parent& p = Y.P;
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
Mat<eT> out = p;
const quasi_unwrap<T1> tmp(X.get_ref());
const Mat<eT>& A = tmp.M;
const unwrap<TB> U(Y.base_indices.get_ref());
Y.check_size(A);
Y.check_indices(U.M);
const uword* indices_mem = U.M.memptr();
const uword N = U.M.n_elem;
if(mode == 0) // process columns
{
const eT* A_mem = A.memptr();
for(uword i=0; i < N; ++i)
{
const uword col = indices_mem[i];
arma_debug_check( (col >= p_n_cols), "each_col(): index out of bounds" );
const eT* p_mem = p.colptr(col);
eT* out_mem = out.colptr(col);
for(uword row=0; row < p_n_rows; ++row)
{
out_mem[row] = A_mem[row] - p_mem[row];
}
}
}
if(mode == 1) // process rows
{
for(uword i=0; i < N; ++i)
{
const uword row = indices_mem[i];
arma_debug_check( (row >= p_n_rows), "each_row(): index out of bounds" );
out.row(row) = A - p.row(row);
}
}
return out;
}
template<typename parent, unsigned int mode, typename TB, typename T2>
inline
Mat<typename parent::elem_type>
subview_each2_aux::operator_schur
(
const subview_each2<parent,mode,TB>& X,
const Base<typename parent::elem_type,T2>& Y
)
{
arma_extra_debug_sigprint();
typedef typename parent::elem_type eT;
const parent& p = X.P;
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
Mat<eT> out = p;
const quasi_unwrap<T2> tmp(Y.get_ref());
const Mat<eT>& A = tmp.M;
const unwrap<TB> U(X.base_indices.get_ref());
X.check_size(A);
X.check_indices(U.M);
const uword* indices_mem = U.M.memptr();
const uword N = U.M.n_elem;
if(mode == 0) // process columns
{
const eT* A_mem = A.memptr();
for(uword i=0; i < N; ++i)
{
const uword col = indices_mem[i];
arma_debug_check( (col >= p_n_cols), "each_col(): index out of bounds" );
arrayops::inplace_mul( out.colptr(col), A_mem, p_n_rows );
}
}
if(mode == 1) // process rows
{
for(uword i=0; i < N; ++i)
{
const uword row = indices_mem[i];
arma_debug_check( (row >= p_n_rows), "each_row(): index out of bounds" );
out.row(row) %= A;
}
}
return out;
}
template<typename parent, unsigned int mode, typename TB, typename T2>
inline
Mat<typename parent::elem_type>
subview_each2_aux::operator_div
(
const subview_each2<parent,mode,TB>& X,
const Base<typename parent::elem_type,T2>& Y
)
{
arma_extra_debug_sigprint();
typedef typename parent::elem_type eT;
const parent& p = X.P;
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
Mat<eT> out = p;
const quasi_unwrap<T2> tmp(Y.get_ref());
const Mat<eT>& A = tmp.M;
const unwrap<TB> U(X.base_indices.get_ref());
X.check_size(A);
X.check_indices(U.M);
const uword* indices_mem = U.M.memptr();
const uword N = U.M.n_elem;
if(mode == 0) // process columns
{
const eT* A_mem = A.memptr();
for(uword i=0; i < N; ++i)
{
const uword col = indices_mem[i];
arma_debug_check( (col >= p_n_cols), "each_col(): index out of bounds" );
arrayops::inplace_div( out.colptr(col), A_mem, p_n_rows );
}
}
if(mode == 1) // process rows
{
for(uword i=0; i < N; ++i)
{
const uword row = indices_mem[i];
arma_debug_check( (row >= p_n_rows), "each_row(): index out of bounds" );
out.row(row) /= A;
}
}
return out;
}
template<typename T1, typename parent, unsigned int mode, typename TB>
inline
Mat<typename parent::elem_type>
subview_each2_aux::operator_div
(
const Base<typename parent::elem_type,T1>& X,
const subview_each2<parent,mode,TB>& Y
)
{
arma_extra_debug_sigprint();
typedef typename parent::elem_type eT;
const parent& p = Y.P;
const uword p_n_rows = p.n_rows;
const uword p_n_cols = p.n_cols;
Mat<eT> out = p;
const quasi_unwrap<T1> tmp(X.get_ref());
const Mat<eT>& A = tmp.M;
const unwrap<TB> U(Y.base_indices.get_ref());
Y.check_size(A);
Y.check_indices(U.M);
const uword* indices_mem = U.M.memptr();
const uword N = U.M.n_elem;
if(mode == 0) // process columns
{
const eT* A_mem = A.memptr();
for(uword i=0; i < N; ++i)
{
const uword col = indices_mem[i];
arma_debug_check( (col >= p_n_cols), "each_col(): index out of bounds" );
const eT* p_mem = p.colptr(col);
eT* out_mem = out.colptr(col);
for(uword row=0; row < p_n_rows; ++row)
{
out_mem[row] = A_mem[row] / p_mem[row];
}
}
}
if(mode == 1) // process rows
{
for(uword i=0; i < N; ++i)
{
const uword row = indices_mem[i];
arma_debug_check( (row >= p_n_rows), "each_row(): index out of bounds" );
out.row(row) = A / p.row(row);
}
}
return out;
}
//! @}
Reference in New Issue Copy Permalink