/* -*- c++ -*- (enables emacs c++ mode) */ /*=========================================================================== Copyright (C) 2003-2017 Yves Renard This file is a part of GetFEM++ GetFEM++ is free software; you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation; either version 3 of the License, or (at your option) any later version along with the GCC Runtime Library Exception either version 3.1 or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License and GCC Runtime Library Exception for more details. You should have received a copy of the GNU Lesser General Public License along with this program; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA. As a special exception, you may use this file as it is a part of a free software library without restriction. Specifically, if other files instantiate templates or use macros or inline functions from this file, or you compile this file and link it with other files to produce an executable, this file does not by itself cause the resulting executable to be covered by the GNU Lesser General Public License. This exception does not however invalidate any other reasons why the executable file might be covered by the GNU Lesser General Public License. ===========================================================================*/ /**@file gmm_real_part.h @author Yves Renard @date September 18, 2003. @brief extract the real/imaginary part of vectors/matrices */ #ifndef GMM_REAL_PART_H #define GMM_REAL_PART_H #include "gmm_def.h" #include "gmm_vector.h" namespace gmm { struct linalg_real_part {}; struct linalg_imag_part {}; template struct which_part {}; template typename number_traits::magnitude_type real_or_imag_part(C x, linalg_real_part) { return gmm::real(x); } template typename number_traits::magnitude_type real_or_imag_part(C x, linalg_imag_part) { return gmm::imag(x); } template C complex_from(T x, C y, OP op, linalg_real_part) { return std::complex(op(std::real(y), x), std::imag(y)); } template C complex_from(T x, C y, OP op,linalg_imag_part) { return std::complex(std::real(y), op(std::imag(y), x)); } template struct project2nd { T operator()(T , T b) const { return b; } }; template class ref_elt_vector > { R r; public : operator T() const { return real_or_imag_part(std::complex(r), PART()); } ref_elt_vector(R r_) : r(r_) {} inline ref_elt_vector &operator =(T v) { r = complex_from(v, std::complex(r), gmm::project2nd(), PART()); return *this; } inline bool operator ==(T v) const { return (r == v); } inline bool operator !=(T v) const { return (r != v); } inline ref_elt_vector &operator +=(T v) { r = complex_from(v, std::complex(r), std::plus(), PART()); return *this; } inline ref_elt_vector &operator -=(T v) { r = complex_from(v, std::complex(r), std::minus(), PART()); return *this; } inline ref_elt_vector &operator /=(T v) { r = complex_from(v, std::complex(r), std::divides(), PART()); return *this; } inline ref_elt_vector &operator *=(T v) { r = complex_from(v, std::complex(r), std::multiplies(), PART()); return *this; } inline ref_elt_vector &operator =(const ref_elt_vector &re) { *this = T(re); return *this; } T operator +() { return T(*this); } // necessary for unknow reason T operator -() { return -T(*this); } // necessary for unknow reason T operator +(T v) { return T(*this)+ v; } // necessary for unknow reason T operator -(T v) { return T(*this)- v; } // necessary for unknow reason T operator *(T v) { return T(*this)* v; } // necessary for unknow reason T operator /(T v) { return T(*this)/ v; } // necessary for unknow reason }; template struct ref_or_value_type { template static W r(const T &x, linalg_real_part, W) { return gmm::real(x); } template static W r(const T &x, linalg_imag_part, W) { return gmm::imag(x); } }; template struct ref_or_value_type > > { template static const T &r(const T &x, linalg_real_part, W) { return x; } template static const T &r(const T &x, linalg_imag_part, W) { return x; } template static T &r(T &x, linalg_real_part, W) { return x; } template static T &r(T &x, linalg_imag_part, W) { return x; } }; /* ********************************************************************* */ /* Reference to the real part of (complex) vectors */ /* ********************************************************************* */ template struct part_vector_iterator { typedef typename std::iterator_traits::value_type vtype; typedef typename gmm::number_traits::magnitude_type value_type; typedef value_type *pointer; typedef ref_elt_vector::reference, PART> > reference; typedef typename std::iterator_traits::difference_type difference_type; typedef typename std::iterator_traits::iterator_category iterator_category; IT it; part_vector_iterator(void) {} explicit part_vector_iterator(const IT &i) : it(i) {} part_vector_iterator(const part_vector_iterator &i) : it(i.it) {} size_type index(void) const { return it.index(); } part_vector_iterator operator ++(int) { part_vector_iterator tmp = *this; ++it; return tmp; } part_vector_iterator operator --(int) { part_vector_iterator tmp = *this; --it; return tmp; } part_vector_iterator &operator ++() { ++it; return *this; } part_vector_iterator &operator --() { --it; return *this; } part_vector_iterator &operator +=(difference_type i) { it += i; return *this; } part_vector_iterator &operator -=(difference_type i) { it -= i; return *this; } part_vector_iterator operator +(difference_type i) const { part_vector_iterator itb = *this; return (itb += i); } part_vector_iterator operator -(difference_type i) const { part_vector_iterator itb = *this; return (itb -= i); } difference_type operator -(const part_vector_iterator &i) const { return difference_type(it - i.it); } reference operator *() const { return reference(*it); } reference operator [](size_type ii) const { return reference(it[ii]); } bool operator ==(const part_vector_iterator &i) const { return (i.it == it); } bool operator !=(const part_vector_iterator &i) const { return (i.it != it); } bool operator < (const part_vector_iterator &i) const { return (it < i.it); } }; template struct part_vector { typedef part_vector this_type; typedef typename std::iterator_traits::value_type V; typedef V * CPT; typedef typename select_ref::const_iterator, typename linalg_traits::iterator, PT>::ref_type iterator; typedef typename linalg_traits::reference reference; typedef typename linalg_traits::value_type value_type; typedef typename linalg_traits::porigin_type porigin_type; iterator begin_, end_; porigin_type origin; size_type size_; size_type size(void) const { return size_; } reference operator[](size_type i) const { return reference(ref_or_value_type::r( linalg_traits::access(origin, begin_, end_, i), PART(), value_type())); } part_vector(V &v) : begin_(vect_begin(v)), end_(vect_end(v)), origin(linalg_origin(v)), size_(gmm::vect_size(v)) {} part_vector(const V &v) : begin_(vect_begin(const_cast(v))), end_(vect_end(const_cast(v))), origin(linalg_origin(const_cast(v))), size_(gmm::vect_size(v)) {} part_vector() {} part_vector(const part_vector &cr) : begin_(cr.begin_),end_(cr.end_),origin(cr.origin), size_(cr.size_) {} }; template inline void set_to_begin(part_vector_iterator &it, ORG o, part_vector *, linalg_modifiable) { typedef part_vector VECT; typedef typename linalg_traits::V_reference ref_t; set_to_begin(it.it, o, typename linalg_traits::pV(), ref_t()); } template inline void set_to_begin(part_vector_iterator &it, ORG o, const part_vector *, linalg_modifiable) { typedef part_vector VECT; typedef typename linalg_traits::V_reference ref_t; set_to_begin(it.it, o, typename linalg_traits::pV(), ref_t()); } template inline void set_to_end(part_vector_iterator &it, ORG o, part_vector *, linalg_modifiable) { typedef part_vector VECT; typedef typename linalg_traits::V_reference ref_t; set_to_end(it.it, o, typename linalg_traits::pV(), ref_t()); } template inline void set_to_end(part_vector_iterator &it, ORG o, const part_vector *, linalg_modifiable) { typedef part_vector VECT; typedef typename linalg_traits::V_reference ref_t; set_to_end(it.it, o, typename linalg_traits::pV(), ref_t()); } template std::ostream &operator << (std::ostream &o, const part_vector& m) { gmm::write(o,m); return o; } /* ********************************************************************* */ /* Reference to the real or imaginary part of (complex) matrices */ /* ********************************************************************* */ template struct part_row_ref { typedef part_row_ref this_type; typedef typename std::iterator_traits::value_type M; typedef M * CPT; typedef typename std::iterator_traits::reference ref_M; typedef typename select_ref ::const_row_iterator, typename linalg_traits ::row_iterator, PT>::ref_type iterator; typedef typename linalg_traits::value_type value_type; typedef typename linalg_traits::reference reference; typedef typename linalg_traits::porigin_type porigin_type; iterator begin_, end_; porigin_type origin; size_type nr, nc; part_row_ref(ref_M m) : begin_(mat_row_begin(m)), end_(mat_row_end(m)), origin(linalg_origin(m)), nr(mat_nrows(m)), nc(mat_ncols(m)) {} part_row_ref(const part_row_ref &cr) : begin_(cr.begin_),end_(cr.end_), origin(cr.origin),nr(cr.nr),nc(cr.nc) {} reference operator()(size_type i, size_type j) const { return reference(ref_or_value_type::r( linalg_traits::access(begin_+i, j), PART(), value_type())); } }; template std::ostream &operator << (std::ostream &o, const part_row_ref& m) { gmm::write(o,m); return o; } template struct part_col_ref { typedef part_col_ref this_type; typedef typename std::iterator_traits::value_type M; typedef M * CPT; typedef typename std::iterator_traits::reference ref_M; typedef typename select_ref ::const_col_iterator, typename linalg_traits ::col_iterator, PT>::ref_type iterator; typedef typename linalg_traits::value_type value_type; typedef typename linalg_traits::reference reference; typedef typename linalg_traits::porigin_type porigin_type; iterator begin_, end_; porigin_type origin; size_type nr, nc; part_col_ref(ref_M m) : begin_(mat_col_begin(m)), end_(mat_col_end(m)), origin(linalg_origin(m)), nr(mat_nrows(m)), nc(mat_ncols(m)) {} part_col_ref(const part_col_ref &cr) : begin_(cr.begin_),end_(cr.end_), origin(cr.origin),nr(cr.nr),nc(cr.nc) {} reference operator()(size_type i, size_type j) const { return reference(ref_or_value_type::r( linalg_traits::access(begin_+j, i), PART(), value_type())); } }; template std::ostream &operator << (std::ostream &o, const part_col_ref& m) { gmm::write(o,m); return o; } template struct part_return_ { typedef abstract_null_type return_type; }; template struct part_return_ { typedef typename std::iterator_traits::value_type L; typedef typename select_return, part_row_ref< L *, PART>, PT>::return_type return_type; }; template struct part_return_ { typedef typename std::iterator_traits::value_type L; typedef typename select_return, part_col_ref, PT>::return_type return_type; }; template struct part_return__{ typedef abstract_null_type return_type; }; template struct part_return__ { typedef typename std::iterator_traits::value_type L; typedef typename part_return_::sub_orientation>::potype, PART, PT>::return_type return_type; }; template struct part_return__ { typedef typename std::iterator_traits::value_type L; typedef typename select_return, part_vector, PT>::return_type return_type; }; template struct part_return { typedef typename std::iterator_traits::value_type L; typedef typename part_return__::linalg_type>::return_type return_type; }; template inline typename part_return::return_type real_part(const L &l) { return typename part_return::return_type (linalg_cast(const_cast(l))); } template inline typename part_return::return_type real_part(L &l) { return typename part_return::return_type(linalg_cast(l)); } template inline typename part_return::return_type imag_part(const L &l) { return typename part_return::return_type (linalg_cast(const_cast(l))); } template inline typename part_return::return_type imag_part(L &l) { return typename part_return::return_type(linalg_cast(l)); } template struct linalg_traits > { typedef part_vector this_type; typedef this_type * pthis_type; typedef PT pV; typedef typename std::iterator_traits::value_type V; typedef typename linalg_traits::index_sorted index_sorted; typedef typename linalg_traits::is_reference V_reference; typedef typename linalg_traits::origin_type origin_type; typedef typename select_ref::ref_type porigin_type; typedef typename which_reference::is_reference is_reference; typedef abstract_vector linalg_type; typedef typename linalg_traits::value_type vtype; typedef typename number_traits::magnitude_type value_type; typedef typename select_ref::reference, PART> >, PT>::ref_type reference; typedef typename select_ref::const_iterator, typename linalg_traits::iterator, PT>::ref_type pre_iterator; typedef typename select_ref, PT>::ref_type iterator; typedef part_vector_iterator::const_iterator, pre_iterator, PART> const_iterator; typedef typename linalg_traits::storage_type storage_type; static size_type size(const this_type &v) { return v.size(); } static iterator begin(this_type &v) { iterator it; it.it = v.begin_; if (!is_const_reference(is_reference()) && is_sparse(storage_type())) set_to_begin(it, v.origin, pthis_type(), is_reference()); return it; } static const_iterator begin(const this_type &v) { const_iterator it(v.begin_); if (!is_const_reference(is_reference()) && is_sparse(storage_type())) { set_to_begin(it, v.origin, pthis_type(), is_reference()); } return it; } static iterator end(this_type &v) { iterator it(v.end_); if (!is_const_reference(is_reference()) && is_sparse(storage_type())) set_to_end(it, v.origin, pthis_type(), is_reference()); return it; } static const_iterator end(const this_type &v) { const_iterator it(v.end_); if (!is_const_reference(is_reference()) && is_sparse(storage_type())) set_to_end(it, v.origin, pthis_type(), is_reference()); return it; } static origin_type* origin(this_type &v) { return v.origin; } static const origin_type* origin(const this_type &v) { return v.origin; } static void clear(origin_type* o, const iterator &begin_, const iterator &end_, abstract_sparse) { std::deque ind; iterator it = begin_; for (; it != end_; ++it) ind.push_front(it.index()); for (; !(ind.empty()); ind.pop_back()) access(o, begin_, end_, ind.back()) = value_type(0); } static void clear(origin_type* o, const iterator &begin_, const iterator &end_, abstract_skyline) { clear(o, begin_, end_, abstract_sparse()); } static void clear(origin_type* o, const iterator &begin_, const iterator &end_, abstract_dense) { for (iterator it = begin_; it != end_; ++it) *it = value_type(0); } static void clear(origin_type* o, const iterator &begin_, const iterator &end_) { clear(o, begin_, end_, storage_type()); } static void do_clear(this_type &v) { clear(v.origin, begin(v), end(v)); } static value_type access(const origin_type *o, const const_iterator &it, const const_iterator &ite, size_type i) { return real_or_imag_part(linalg_traits::access(o, it.it, ite.it,i), PART()); } static reference access(origin_type *o, const iterator &it, const iterator &ite, size_type i) { return reference(linalg_traits::access(o, it.it, ite.it,i)); } }; template struct linalg_traits > { typedef part_row_ref this_type; typedef typename std::iterator_traits::value_type M; typedef typename linalg_traits::origin_type origin_type; typedef typename select_ref::ref_type porigin_type; typedef typename which_reference::is_reference is_reference; typedef abstract_matrix linalg_type; typedef typename linalg_traits::value_type vtype; typedef typename number_traits::magnitude_type value_type; typedef typename linalg_traits::storage_type storage_type; typedef abstract_null_type sub_col_type; typedef abstract_null_type const_sub_col_type; typedef abstract_null_type col_iterator; typedef abstract_null_type const_col_iterator; typedef typename org_type::const_sub_row_type>::t pre_const_sub_row_type; typedef typename org_type::sub_row_type>::t pre_sub_row_type; typedef part_vector const_sub_row_type; typedef typename select_ref, PT>::ref_type sub_row_type; typedef typename linalg_traits::const_row_iterator const_row_iterator; typedef typename select_ref::row_iterator, PT>::ref_type row_iterator; typedef typename select_ref< typename linalg_traits::reference, typename linalg_traits::reference, PT>::ref_type reference; typedef row_major sub_orientation; typedef typename linalg_traits::index_sorted index_sorted; static size_type ncols(const this_type &v) { return v.nc; } static size_type nrows(const this_type &v) { return v.nr; } static const_sub_row_type row(const const_row_iterator &it) { return const_sub_row_type(linalg_traits::row(it)); } static sub_row_type row(const row_iterator &it) { return sub_row_type(linalg_traits::row(it)); } static row_iterator row_begin(this_type &m) { return m.begin_; } static row_iterator row_end(this_type &m) { return m.end_; } static const_row_iterator row_begin(const this_type &m) { return m.begin_; } static const_row_iterator row_end(const this_type &m) { return m.end_; } static origin_type* origin(this_type &v) { return v.origin; } static const origin_type* origin(const this_type &v) { return v.origin; } static void do_clear(this_type &v); static value_type access(const const_row_iterator &itrow, size_type i) { return real_or_imag_part(linalg_traits::access(itrow, i), PART()); } static reference access(const row_iterator &itrow, size_type i) { return reference(ref_or_value_type::r( linalg_traits::access(itrow, i), PART(), value_type())); } }; template struct linalg_traits > { typedef part_col_ref this_type; typedef typename std::iterator_traits::value_type M; typedef typename linalg_traits::origin_type origin_type; typedef typename select_ref::ref_type porigin_type; typedef typename which_reference::is_reference is_reference; typedef abstract_matrix linalg_type; typedef typename linalg_traits::value_type vtype; typedef typename number_traits::magnitude_type value_type; typedef typename linalg_traits::storage_type storage_type; typedef abstract_null_type sub_row_type; typedef abstract_null_type const_sub_row_type; typedef abstract_null_type row_iterator; typedef abstract_null_type const_row_iterator; typedef typename org_type::const_sub_col_type>::t pre_const_sub_col_type; typedef typename org_type::sub_col_type>::t pre_sub_col_type; typedef part_vector const_sub_col_type; typedef typename select_ref, PT>::ref_type sub_col_type; typedef typename linalg_traits::const_col_iterator const_col_iterator; typedef typename select_ref::col_iterator, PT>::ref_type col_iterator; typedef typename select_ref< typename linalg_traits::reference, typename linalg_traits::reference, PT>::ref_type reference; typedef col_major sub_orientation; typedef typename linalg_traits::index_sorted index_sorted; static size_type nrows(const this_type &v) { return v.nr; } static size_type ncols(const this_type &v) { return v.nc; } static const_sub_col_type col(const const_col_iterator &it) { return const_sub_col_type(linalg_traits::col(it)); } static sub_col_type col(const col_iterator &it) { return sub_col_type(linalg_traits::col(it)); } static col_iterator col_begin(this_type &m) { return m.begin_; } static col_iterator col_end(this_type &m) { return m.end_; } static const_col_iterator col_begin(const this_type &m) { return m.begin_; } static const_col_iterator col_end(const this_type &m) { return m.end_; } static origin_type* origin(this_type &v) { return v.origin; } static const origin_type* origin(const this_type &v) { return v.origin; } static void do_clear(this_type &v); static value_type access(const const_col_iterator &itcol, size_type i) { return real_or_imag_part(linalg_traits::access(itcol, i), PART()); } static reference access(const col_iterator &itcol, size_type i) { return reference(ref_or_value_type::r( linalg_traits::access(itcol, i), PART(), value_type())); } }; template void linalg_traits >::do_clear(this_type &v) { col_iterator it = mat_col_begin(v), ite = mat_col_end(v); for (; it != ite; ++it) clear(col(it)); } template void linalg_traits >::do_clear(this_type &v) { row_iterator it = mat_row_begin(v), ite = mat_row_end(v); for (; it != ite; ++it) clear(row(it)); } } #endif // GMM_REAL_PART_H