AnalysisSystemForRadionucli.../include/armadillo_bits/op_flip_meat.hpp

109 lines
2.3 KiB
C++
Raw Normal View History

2024-06-04 15:25:02 +08:00
// Copyright (C) 2010-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 op_flip
//! @{
template<typename T1>
inline
void
op_flipud::apply(Mat<typename T1::elem_type>& out, const Op<T1,op_flipud>& in)
{
arma_extra_debug_sigprint();
typedef typename T1::elem_type eT;
const unwrap<T1> tmp(in.m);
const Mat<eT>& X = tmp.M;
const uword X_n_rows = T1::is_row ? uword(1) : X.n_rows;
const uword X_n_cols = T1::is_col ? uword(1) : X.n_cols;
if(&out != &X)
{
out.copy_size(X);
for(uword col=0; col<X_n_cols; ++col)
{
const eT* X_data = X.colptr(col);
eT* out_data = out.colptr(col);
for(uword row=0; row<X_n_rows; ++row)
{
out_data[row] = X_data[X_n_rows-1 - row];
}
}
}
else // in-place operation
{
const uword N = X_n_rows / 2;
for(uword col=0; col<X_n_cols; ++col)
{
eT* out_data = out.colptr(col);
for(uword row=0; row<N; ++row)
{
std::swap(out_data[row], out_data[X_n_rows-1 - row]);
}
}
}
}
template<typename T1>
inline
void
op_fliplr::apply(Mat<typename T1::elem_type>& out, const Op<T1,op_fliplr>& in)
{
arma_extra_debug_sigprint();
typedef typename T1::elem_type eT;
const unwrap<T1> tmp(in.m);
const Mat<eT>& X = tmp.M;
const uword X_n_cols = X.n_cols;
if(&out != &X)
{
out.copy_size(X);
if(T1::is_row || X.is_rowvec())
{
for(uword i=0; i<X_n_cols; ++i) { out[i] = X[X_n_cols-1 - i]; }
}
else
{
for(uword i=0; i<X_n_cols; ++i) { out.col(i) = X.col(X_n_cols-1 - i); }
}
}
else
{
const uword N = X_n_cols / 2;
if(T1::is_row || X.is_rowvec())
{
for(uword i=0; i<N; ++i) { std::swap(out[i], out[X_n_cols-1 - i]); }
}
else
{
for(uword i=0; i<N; ++i) { out.swap_cols(i, X_n_cols-1 - i); }
}
}
}
//! @}