From eda5bc26f44ee9a6f83dcf8c91f17296d7fc509d Mon Sep 17 00:00:00 2001 From: Nao Pross Date: Mon, 12 Feb 2024 14:52:43 +0100 Subject: Move into version control --- .../include/armadillo_bits/subview_bones.hpp | 673 +++++++++++++++++++++ 1 file changed, 673 insertions(+) create mode 100644 src/armadillo/include/armadillo_bits/subview_bones.hpp (limited to 'src/armadillo/include/armadillo_bits/subview_bones.hpp') diff --git a/src/armadillo/include/armadillo_bits/subview_bones.hpp b/src/armadillo/include/armadillo_bits/subview_bones.hpp new file mode 100644 index 0000000..95553a1 --- /dev/null +++ b/src/armadillo/include/armadillo_bits/subview_bones.hpp @@ -0,0 +1,673 @@ +// SPDX-License-Identifier: Apache-2.0 +// +// Copyright 2008-2016 Conrad Sanderson (http://conradsanderson.id.au) +// Copyright 2008-2016 National ICT Australia (NICTA) +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. +// ------------------------------------------------------------------------ + + +//! \addtogroup subview +//! @{ + + +//! Class for storing data required to construct or apply operations to a submatrix +//! (ie. where the submatrix starts and ends as well as a reference/pointer to the original matrix), +template +class subview : public Base< eT, subview > + { + public: + + typedef eT elem_type; + typedef typename get_pod_type::result pod_type; + + arma_aligned const Mat& m; + + static constexpr bool is_row = false; + static constexpr bool is_col = false; + static constexpr bool is_xvec = false; + + 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& in_m, const uword in_row1, const uword in_col1, const uword in_n_rows, const uword in_n_cols); + + + public: + + inline ~subview(); + inline subview() = delete; + + inline subview(const subview& in); + inline subview( subview&& in); + + template inline void inplace_op(const eT val ); + template inline void inplace_op(const Base& x, const char* identifier); + template inline void inplace_op(const subview& x, const char* identifier); + + // deliberately returning void + + inline void operator= (const eT val); + inline void operator+= (const eT val); + inline void operator-= (const eT val); + inline void operator*= (const eT val); + inline void operator/= (const eT val); + + 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); + + template inline void operator= (const Base& x); + template inline void operator+= (const Base& x); + template inline void operator-= (const Base& x); + template inline void operator%= (const Base& x); + template inline void operator/= (const Base& x); + + template inline void operator= (const SpBase& x); + template inline void operator+= (const SpBase& x); + template inline void operator-= (const SpBase& x); + template inline void operator%= (const SpBase& x); + template inline void operator/= (const SpBase& x); + + template + inline typename enable_if2< is_same_type::value, void>::result operator=(const Gen& x); + + inline void operator=(const std::initializer_list& list); + inline void operator=(const std::initializer_list< std::initializer_list >& list); + + + inline static void extract(Mat& out, const subview& in); + + inline static void plus_inplace(Mat& out, const subview& in); + inline static void minus_inplace(Mat& out, const subview& in); + inline static void schur_inplace(Mat& out, const subview& in); + inline static void div_inplace(Mat& out, const subview& in); + + template inline void for_each(functor F); + template inline void for_each(functor F) const; + + template inline void transform(functor F); + template inline void imbue(functor F); + + inline void replace(const eT old_val, const eT new_val); + + inline void clean(const pod_type threshold); + + inline void clamp(const eT min_val, const eT max_val); + + inline void fill(const eT val); + inline void zeros(); + inline void ones(); + inline void eye(); + inline void randu(); + inline void randn(); + + arma_warn_unused inline eT at_alt (const uword ii) const; + + arma_warn_unused inline eT& operator[](const uword ii); + arma_warn_unused inline eT operator[](const uword ii) const; + + arma_warn_unused inline eT& operator()(const uword ii); + arma_warn_unused inline eT operator()(const uword ii) const; + + arma_warn_unused inline eT& operator()(const uword in_row, const uword in_col); + arma_warn_unused inline eT operator()(const uword in_row, const uword in_col) const; + + arma_warn_unused inline eT& at(const uword in_row, const uword in_col); + arma_warn_unused inline eT at(const uword in_row, const uword in_col) const; + + arma_warn_unused inline eT& front(); + arma_warn_unused inline eT front() const; + + arma_warn_unused inline eT& back(); + arma_warn_unused inline eT back() const; + + arma_inline eT* colptr(const uword in_col); + arma_inline const eT* colptr(const uword in_col) const; + + template + inline bool check_overlap(const subview& x) const; + + arma_warn_unused inline bool is_vec() const; + arma_warn_unused inline bool is_finite() const; + arma_warn_unused inline bool is_zero(const pod_type tol = 0) const; + + arma_warn_unused inline bool has_inf() const; + arma_warn_unused inline bool has_nan() const; + arma_warn_unused inline bool has_nonfinite() const; + + inline subview_row row(const uword row_num); + inline const subview_row row(const uword row_num) const; + + inline subview_row operator()(const uword row_num, const span& col_span); + inline const subview_row operator()(const uword row_num, const span& col_span) const; + + inline subview_col col(const uword col_num); + inline const subview_col col(const uword col_num) const; + + inline subview_col operator()(const span& row_span, const uword col_num); + inline const subview_col operator()(const span& row_span, const uword col_num) const; + + inline Col unsafe_col(const uword col_num); + inline const Col unsafe_col(const uword col_num) const; + + inline subview rows(const uword in_row1, const uword in_row2); + inline const subview rows(const uword in_row1, const uword in_row2) const; + + inline subview cols(const uword in_col1, const uword in_col2); + inline const subview cols(const uword in_col1, const uword in_col2) const; + + inline subview submat(const uword in_row1, const uword in_col1, const uword in_row2, const uword in_col2); + inline const subview submat(const uword in_row1, const uword in_col1, const uword in_row2, const uword in_col2) const; + + inline subview submat (const span& row_span, const span& col_span); + inline const subview submat (const span& row_span, const span& col_span) const; + + inline subview operator()(const span& row_span, const span& col_span); + inline const subview operator()(const span& row_span, const span& col_span) const; + + inline subview_each1< subview, 0 > each_col(); + inline subview_each1< subview, 1 > each_row(); + + template inline subview_each2< subview, 0, T1 > each_col(const Base& indices); + template inline subview_each2< subview, 1, T1 > each_row(const Base& indices); + + inline void each_col(const std::function< void( Col&) >& F); + inline void each_col(const std::function< void(const Col&) >& F) const; + + inline void each_row(const std::function< void( Row&) >& F); + inline void each_row(const std::function< void(const Row&) >& F) const; + + inline diagview diag(const sword in_id = 0); + inline const diagview 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); + + + class const_iterator; + + class iterator + { + public: + + inline iterator(); + inline iterator(const iterator& X); + inline iterator(subview& in_sv, const uword in_row, const uword in_col); + + arma_warn_unused inline eT& operator*(); + + inline iterator& operator++(); + arma_warn_unused inline iterator operator++(int); + + arma_warn_unused inline bool operator==(const iterator& rhs) const; + arma_warn_unused inline bool operator!=(const iterator& rhs) const; + arma_warn_unused inline bool operator==(const const_iterator& rhs) const; + arma_warn_unused inline bool operator!=(const const_iterator& rhs) const; + + typedef std::forward_iterator_tag iterator_category; + typedef eT value_type; + typedef std::ptrdiff_t difference_type; // TODO: not certain on this one + typedef eT* pointer; + typedef eT& reference; + + arma_aligned Mat* M; + arma_aligned eT* current_ptr; + arma_aligned uword current_row; + arma_aligned uword current_col; + + arma_aligned const uword aux_row1; + arma_aligned const uword aux_row2_p1; + }; + + + class const_iterator + { + public: + + inline const_iterator(); + inline const_iterator(const iterator& X); + inline const_iterator(const const_iterator& X); + inline const_iterator(const subview& in_sv, const uword in_row, const uword in_col); + + arma_warn_unused inline const eT& operator*(); + + inline const_iterator& operator++(); + arma_warn_unused inline const_iterator operator++(int); + + arma_warn_unused inline bool operator==(const iterator& rhs) const; + arma_warn_unused inline bool operator!=(const iterator& rhs) const; + arma_warn_unused inline bool operator==(const const_iterator& rhs) const; + arma_warn_unused inline bool operator!=(const const_iterator& rhs) const; + + // So that we satisfy the STL iterator types. + typedef std::forward_iterator_tag iterator_category; + typedef eT value_type; + typedef std::ptrdiff_t difference_type; // TODO: not certain on this one + typedef const eT* pointer; + typedef const eT& reference; + + arma_aligned const Mat* M; + arma_aligned const eT* current_ptr; + arma_aligned uword current_row; + arma_aligned uword current_col; + + arma_aligned const uword aux_row1; + arma_aligned const uword aux_row2_p1; + }; + + + class const_row_iterator; + + class row_iterator + { + public: + + inline row_iterator(); + inline row_iterator(const row_iterator& X); + inline row_iterator(subview& in_sv, const uword in_row, const uword in_col); + + arma_warn_unused inline eT& operator* (); + + inline row_iterator& operator++(); + arma_warn_unused inline row_iterator operator++(int); + + arma_warn_unused inline bool operator!=(const row_iterator& X) const; + arma_warn_unused inline bool operator==(const row_iterator& X) const; + arma_warn_unused inline bool operator!=(const const_row_iterator& X) const; + arma_warn_unused inline bool operator==(const const_row_iterator& X) const; + + typedef std::forward_iterator_tag iterator_category; + typedef eT value_type; + typedef std::ptrdiff_t difference_type; // TODO: not certain on this one + typedef eT* pointer; + typedef eT& reference; + + arma_aligned Mat* M; + arma_aligned uword current_row; + arma_aligned uword current_col; + + arma_aligned const uword aux_col1; + arma_aligned const uword aux_col2_p1; + }; + + + class const_row_iterator + { + public: + + inline const_row_iterator(); + inline const_row_iterator(const row_iterator& X); + inline const_row_iterator(const const_row_iterator& X); + inline const_row_iterator(const subview& in_sv, const uword in_row, const uword in_col); + + arma_warn_unused inline const eT& operator*() const; + + inline const_row_iterator& operator++(); + arma_warn_unused inline const_row_iterator operator++(int); + + arma_warn_unused inline bool operator!=(const row_iterator& X) const; + arma_warn_unused inline bool operator==(const row_iterator& X) const; + arma_warn_unused inline bool operator!=(const const_row_iterator& X) const; + arma_warn_unused inline bool operator==(const const_row_iterator& X) const; + + typedef std::forward_iterator_tag iterator_category; + typedef eT value_type; + typedef std::ptrdiff_t difference_type; // TODO: not certain on this one + typedef const eT* pointer; + typedef const eT& reference; + + arma_aligned const Mat* M; + arma_aligned uword current_row; + arma_aligned uword current_col; + + arma_aligned const uword aux_col1; + arma_aligned const uword aux_col2_p1; + }; + + + + inline iterator begin(); + inline const_iterator begin() const; + inline const_iterator cbegin() const; + + inline iterator end(); + inline const_iterator end() const; + inline const_iterator cend() const; + + + friend class Mat; + }; + + + +template +class subview_col : public subview + { + public: + + typedef eT elem_type; + typedef typename get_pod_type::result pod_type; + + static constexpr bool is_row = false; + static constexpr bool is_col = true; + static constexpr bool is_xvec = false; + + const eT* colmem; + + inline void operator= (const subview& x); + inline void operator= (const subview_col& x); + inline void operator= (const eT val); + inline void operator= (const std::initializer_list& list); + + template inline void operator= (const Base& x); + template inline void operator= (const SpBase& x); + + template + inline typename enable_if2< is_same_type::value, void>::result operator=(const Gen& x); + + arma_warn_unused arma_inline const Op,op_htrans> t() const; + arma_warn_unused arma_inline const Op,op_htrans> ht() const; + arma_warn_unused arma_inline const Op,op_strans> st() const; + + arma_warn_unused arma_inline const Op,op_strans> as_row() const; + + inline void fill(const eT val); + inline void zeros(); + inline void ones(); + + arma_inline eT at_alt (const uword i) const; + + arma_inline eT& operator[](const uword i); + arma_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 subview_col rows(const uword in_row1, const uword in_row2); + inline const subview_col rows(const uword in_row1, const uword in_row2) const; + + inline subview_col subvec(const uword in_row1, const uword in_row2); + inline const subview_col subvec(const uword in_row1, const uword in_row2) const; + + inline subview_col subvec(const uword start_row, const SizeMat& s); + inline const subview_col subvec(const uword start_row, const SizeMat& s) const; + + inline subview_col head(const uword N); + inline const subview_col head(const uword N) const; + + inline subview_col tail(const uword N); + inline const subview_col tail(const uword N) const; + + arma_warn_unused inline eT min() const; + arma_warn_unused inline eT max() const; + + inline eT min(uword& index_of_min_val) const; + inline eT max(uword& index_of_max_val) const; + + arma_warn_unused inline uword index_min() const; + arma_warn_unused inline uword index_max() const; + + inline subview_col(const subview_col& in); + inline subview_col( subview_col&& in); + + + protected: + + inline subview_col(const Mat& in_m, const uword in_col); + inline subview_col(const Mat& in_m, const uword in_col, const uword in_row1, const uword in_n_rows); + inline subview_col() = delete; + + + friend class Mat; + friend class Col; + friend class subview; + }; + + + +template +class subview_cols : public subview + { + public: + + typedef eT elem_type; + typedef typename get_pod_type::result pod_type; + + static constexpr bool is_row = false; + static constexpr bool is_col = false; + static constexpr bool is_xvec = false; + + inline subview_cols(const subview_cols& in); + inline subview_cols( subview_cols&& in); + + inline void operator= (const subview& x); + inline void operator= (const subview_cols& x); + inline void operator= (const eT val); + inline void operator= (const std::initializer_list& list); + inline void operator= (const std::initializer_list< std::initializer_list >& list); + + template inline void operator= (const Base& x); + template inline void operator= (const SpBase& x); + + template + inline typename enable_if2< is_same_type::value, void>::result operator=(const Gen& x); + + arma_warn_unused arma_inline const Op,op_htrans> t() const; + arma_warn_unused arma_inline const Op,op_htrans> ht() const; + arma_warn_unused arma_inline const Op,op_strans> st() const; + + arma_warn_unused arma_inline const Op,op_vectorise_col> as_col() const; + + arma_warn_unused inline eT at_alt (const uword ii) const; + + arma_warn_unused inline eT& operator[](const uword ii); + arma_warn_unused inline eT operator[](const uword ii) const; + + arma_warn_unused inline eT& operator()(const uword ii); + arma_warn_unused inline eT operator()(const uword ii) const; + + arma_warn_unused inline eT& operator()(const uword in_row, const uword in_col); + arma_warn_unused inline eT operator()(const uword in_row, const uword in_col) const; + + arma_warn_unused inline eT& at(const uword in_row, const uword in_col); + arma_warn_unused 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; + + protected: + + inline subview_cols(const Mat& in_m, const uword in_col1, const uword in_n_cols); + inline subview_cols() = delete; + + friend class Mat; + friend class subview; + }; + + + +template +class subview_row : public subview + { + public: + + typedef eT elem_type; + typedef typename get_pod_type::result pod_type; + + static constexpr bool is_row = true; + static constexpr bool is_col = false; + static constexpr bool is_xvec = false; + + inline void operator= (const subview& x); + inline void operator= (const subview_row& x); + inline void operator= (const eT val); + inline void operator= (const std::initializer_list& list); + + template inline void operator= (const Base& x); + template inline void operator= (const SpBase& x); + + template + inline typename enable_if2< is_same_type::value, void>::result operator=(const Gen& x); + + arma_warn_unused arma_inline const Op,op_htrans> t() const; + arma_warn_unused arma_inline const Op,op_htrans> ht() const; + arma_warn_unused arma_inline const Op,op_strans> st() const; + + arma_warn_unused arma_inline const Op,op_strans> as_col() const; + + inline eT at_alt (const uword i) const; + + 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; + + inline subview_row cols(const uword in_col1, const uword in_col2); + inline const subview_row cols(const uword in_col1, const uword in_col2) const; + + inline subview_row subvec(const uword in_col1, const uword in_col2); + inline const subview_row subvec(const uword in_col1, const uword in_col2) const; + + inline subview_row subvec(const uword start_col, const SizeMat& s); + inline const subview_row subvec(const uword start_col, const SizeMat& s) const; + + inline subview_row head(const uword N); + inline const subview_row head(const uword N) const; + + inline subview_row tail(const uword N); + inline const subview_row tail(const uword N) const; + + arma_warn_unused inline uword index_min() const; + arma_warn_unused inline uword index_max() const; + + inline typename subview::row_iterator begin(); + inline typename subview::const_row_iterator begin() const; + inline typename subview::const_row_iterator cbegin() const; + + inline typename subview::row_iterator end(); + inline typename subview::const_row_iterator end() const; + inline typename subview::const_row_iterator cend() const; + + inline subview_row(const subview_row& in); + inline subview_row( subview_row&& in); + + + protected: + + inline subview_row(const Mat& in_m, const uword in_row); + inline subview_row(const Mat& in_m, const uword in_row, const uword in_col1, const uword in_n_cols); + inline subview_row() = delete; + + + friend class Mat; + friend class Row; + friend class subview; + }; + + + +template +class subview_row_strans : public Base< eT, subview_row_strans > + { + public: + + typedef eT elem_type; + typedef typename get_pod_type::result pod_type; + + static constexpr bool is_row = false; + static constexpr bool is_col = true; + static constexpr bool is_xvec = false; + + arma_aligned const subview_row& sv_row; + + const uword n_rows; // equal to n_elem + const uword n_elem; + + static constexpr uword n_cols = 1; + + + inline explicit subview_row_strans(const subview_row& in_sv_row); + + inline void extract(Mat& out) const; + + inline eT at_alt (const uword i) const; + + inline eT operator[](const uword i) const; + inline eT operator()(const uword i) const; + + inline eT operator()(const uword in_row, const uword in_col) const; + inline eT at(const uword in_row, const uword in_col) const; + }; + + + +template +class subview_row_htrans : public Base< eT, subview_row_htrans > + { + public: + + typedef eT elem_type; + typedef typename get_pod_type::result pod_type; + + static constexpr bool is_row = false; + static constexpr bool is_col = true; + static constexpr bool is_xvec = false; + + arma_aligned const subview_row& sv_row; + + const uword n_rows; // equal to n_elem + const uword n_elem; + + static constexpr uword n_cols = 1; + + + inline explicit subview_row_htrans(const subview_row& in_sv_row); + + inline void extract(Mat& out) const; + + inline eT at_alt (const uword i) const; + + inline eT operator[](const uword i) const; + inline eT operator()(const uword i) const; + + inline eT operator()(const uword in_row, const uword in_col) const; + inline eT at(const uword in_row, const uword in_col) const; + }; + + + +//! @} -- cgit v1.2.1