89 lines
3.0 KiB
C++
89 lines
3.0 KiB
C++
|
// Copyright (C) 2008-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 mtOp
|
||
|
|
//! @{
|
||
|
|
|
||
|
|
|
||
|
|
struct mtOp_dual_aux_indicator {};
|
||
|
|
|
||
|
|
|
||
|
|
template<typename out_eT, typename T1, typename op_type>
|
||
|
|
class mtOp : public Base<out_eT, mtOp<out_eT, T1, op_type> >
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
|
||
|
|
typedef out_eT elem_type;
|
||
|
|
typedef typename get_pod_type<out_eT>::result pod_type;
|
||
|
|
|
||
|
|
typedef typename T1::elem_type in_eT;
|
||
|
|
|
||
|
|
static const bool is_row = \
|
||
|
|
(
|
||
|
|
// operations which result in a row vector if the input is a row vector
|
||
|
|
T1::is_row &&
|
||
|
|
(
|
||
|
|
is_op_mixed_elem<op_type>::value
|
||
|
|
|| is_same_type<op_type, op_clamp>::value
|
||
|
|
|| is_same_type<op_type, op_hist>::value
|
||
|
|
|| is_same_type<op_type, op_real>::value
|
||
|
|
|| is_same_type<op_type, op_imag>::value
|
||
|
|
|| is_same_type<op_type, op_abs>::value
|
||
|
|
)
|
||
|
|
);
|
||
|
|
|
||
|
|
static const bool is_col = \
|
||
|
|
(
|
||
|
|
// operations which always result in a column vector
|
||
|
|
(is_same_type<op_type, op_find>::value)
|
||
|
|
|| (is_same_type<op_type, op_find_simple>::value)
|
||
|
|
|| (is_same_type<op_type, op_find_unique>::value)
|
||
|
|
|| (is_same_type<op_type, op_sort_index>::value)
|
||
|
|
|| (is_same_type<op_type, op_stable_sort_index>::value)
|
||
|
|
)
|
||
|
|
||
|
||
|
|
(
|
||
|
|
// operations which result in a column vector if the input is a column vector
|
||
|
|
T1::is_col &&
|
||
|
|
(
|
||
|
|
is_op_mixed_elem<op_type>::value
|
||
|
|
|| is_same_type<op_type, op_clamp>::value
|
||
|
|
|| is_same_type<op_type, op_hist>::value
|
||
|
|
|| is_same_type<op_type, op_real>::value
|
||
|
|
|| is_same_type<op_type, op_imag>::value
|
||
|
|
|| is_same_type<op_type, op_abs>::value
|
||
|
|
)
|
||
|
|
);
|
||
|
|
|
||
|
|
|
||
|
|
inline explicit mtOp(const T1& in_m);
|
||
|
|
inline mtOp(const T1& in_m, const in_eT in_aux);
|
||
|
|
inline mtOp(const T1& in_m, const uword in_aux_uword_a, const uword in_aux_uword_b);
|
||
|
|
inline mtOp(const T1& in_m, const in_eT in_aux, const uword in_aux_uword_a, const uword in_aux_uword_b);
|
||
|
|
|
||
|
|
inline mtOp(const char junk, const T1& in_m, const out_eT in_aux);
|
||
|
|
|
||
|
|
inline mtOp(const mtOp_dual_aux_indicator&, const T1& in_m, const in_eT in_aux_a, const out_eT in_aux_b);
|
||
|
|
|
||
|
|
inline ~mtOp();
|
||
|
|
|
||
|
|
|
||
|
|
arma_aligned const T1& m; //!< storage of reference to the operand (eg. a matrix)
|
||
|
|
arma_aligned in_eT aux; //!< storage of auxiliary data, using the element type as used by T1
|
||
|
|
arma_aligned out_eT aux_out_eT; //!< storage of auxiliary data, using the element type as specified by the out_eT template parameter
|
||
|
|
arma_aligned uword aux_uword_a; //!< storage of auxiliary data, uword format
|
||
|
|
arma_aligned uword aux_uword_b; //!< storage of auxiliary data, uword format
|
||
|
|
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
//! @}
|